update

src/agentCore.jl

@@ -5,7 +5,7 @@ export yiemAgent, _agent_loop, OpenAiToUserMessage
 using JSON, DataStructures, Dates, UUIDs, HTTP, Random, PrettyPrinting, Serialization,
       DataFrames, Base.Threads
 using GeneralUtils
-using ..type, ..utils
+using ..type, ..utils, ..toolRegistry
 
 # ---------------------------------------------- 100 --------------------------------------------- #
 
@@ -85,7 +85,7 @@ on `inputChannel` and `followUpChannel` channels concurrently.
 """
 function yiemAgent(
   toolsFolderPath::String,
-  llmCall::Function,
+  llmCall,
   ; 
   systemPrompt::String="You are helpful assistant.",
   model=nothing,
@@ -107,12 +107,12 @@ function yiemAgent(
   outputChannel   = Channel(16)
 
   # load tools from toolsFolderPath
-  toolStore = YiemAgent.toolStore(name="myagent")
+  toolStore1 = toolStore(name="myagent")
-  loadTools(toolStore, toolsFolderPath)
+  loadTools(toolStore1, toolsFolderPath)
 
   # Create struct with a placeholder task, then spawn and replace it
   agent = yiemAgent(
-    agentState(systemPrompt, model, getTools(toolStore), messages),
+    agentState(systemPrompt, model, getTools(toolStore1), messages),
     inputChannel,
     followUp,
     outputChannel,

src/type.jl

@@ -144,7 +144,7 @@ assistantMessage("assistant", [textContent("Hello!")], "", "", "gpt-4", ..., "en
 ```
 """
 function assistantMessage(; role="assistant", content=Vector{messageContent}(),
-    api="", provider="", model="", usage=llmUsage(0, 0), stopReason="end_turn",
+    api="", provider="", model=nothing, usage=llmUsage(0, 0), stopReason="end_turn",
     errorMessage=nothing, timestamp=now())
     return assistantMessage(role, content, api, provider, model, usage, stopReason, errorMessage, timestamp)
 end
@@ -309,7 +309,7 @@ end
 
 mutable struct agentState                     # Mutable runtime state of an agent
     systemPrompt::String                      # System prompt for the agent
-    model::llmModel                           # LLM model to use
+    model::Union{llmModel, Nothing}                           # LLM model to use
     tools::OrderedDict{String, agentTool}     # Available tools keyed by name, insertion-ordered
 
     # messages history includes userMessage, assistantMessage, toolResultMessage. NO system prompt
@@ -342,7 +342,7 @@ agentState("You are a helpful assistant", OrderedDict{String, agentTool}(), agen
 """
 function agentState(
   systemPrompt::String="",
-    model::llmModel=llmModel{String}("", "", "unknown", "unknown", "", false, String[], 
+  model::llmModel=llmModel{String}("", "unknown", "unknown", "", false, String[], 
                                   modelCost(0.0, 0.0, 0.0, 0.0), 0, 0),
   tools::OrderedDict{String, agentTool}=OrderedDict{String, agentTool}(),
   messages::Vector{agentMessage}=agentMessage[],

test/runtest.jl

@@ -1,232 +1,293 @@
 using JSON, Dates, UUIDs, PrettyPrinting, LibPQ, Base64, DataFrames, DataStructures, HTTP, Base64,
       NATS, Base.Threads
-using YiemAgent, GeneralUtils, msghandler
+using YiemAgent, GeneralUtils
 
-  function text2text_instruct_llm(sender_id::String, openai_msg::Dict{String, Any})
+struct text2textInstructLLM
+  natsConn::NATS.Connection
+  topic::String
+  senderID::String
+  fileserver_url::String
+end
+
+function (t::text2textInstructLLM)(openai_msg::Dict{String, Any})
   payloads = [("msg", openai_msg, "dictionary")]  # List of tuples
   _, msg_envelope_json_str = msghandler.smartpack(
-          config["externalservice"]["servicesloadbalancer"]["nats"], 
+        t.topic, 
         payloads;
-          sender_id=sender_id,
+        sender_id=t.senderID,
         msg_purpose="text2text",  
-          broker_url=config["nats_server_info"]["url"],
+        fileserver_url=t.fileserver_url)
-          fileserver_url=config["externalservice"]["fileserver"]["url"])
 
-    reply = NATS.request(agent_conn, 
+  reply = NATS.request(t.natsConn, t.topic, msg_envelope_json_str, timeout=180)
-                        config["externalservice"]["servicesloadbalancer"]["nats"], 
-                        msg_envelope_json_str, timeout=120)
 
   incoming_env_json_str = String(reply.payload)
   incoming_env = msghandler.smartunpack(incoming_env_json_str)
   _llm_response = incoming_env["payloads"][1][2]
   llm_response = _llm_response["choices"][1]["message"]["content"]
   return llm_response
-  end
+end
 
-  """ get a single text embedding from a LLM service
-  Example
-  text = ["hello"]
-  embedding = get_embedding(text)
-  """
-  function get_embedding(text::AbstractArray{String})
-    documents_dict = Dict("documents" => text)
-    payloads = [("documents", documents_dict, "dictionary")]
-    _, msg_envelope_json_str = msghandler.smartpack(
-      config["externalservice"]["servicesloadbalancer"]["nats"], 
-      payloads;   
-      msg_purpose="embedding",  
-      broker_url=config["nats_server_info"]["url"],
-      fileserver_url=config["externalservice"]["fileserver"]["url"])
 
-    reply = NATS.request(agent_conn, 
+# function get_embedding(text::AbstractArray{String})
-                        config["externalservice"]["servicesloadbalancer"]["nats"], 
+#   documents_dict = Dict("documents" => text)
-                        msg_envelope_json_str, timeout=120)
+#   payloads = [("documents", documents_dict, "dictionary")]
-    incoming_env_json_str = String(reply.payload)
+#   _, msg_envelope_json_str = msghandler.smartpack(
-    incoming_env = msghandler.smartunpack(incoming_env_json_str)
+#     config["externalservice"]["servicesloadbalancer"]["nats"], 
-    embedding_response = incoming_env["payloads"][1][2]
+#     payloads;   
+#     msg_purpose="embedding",  
+#     broker_url=config["nats_server_info"]["url"],
+#     fileserver_url=config["externalservice"]["fileserver"]["url"])
 
-    return embedding_response
+#   reply = NATS.request(agent_conn, 
-  end
+#                       config["externalservice"]["servicesloadbalancer"]["nats"], 
+#                       msg_envelope_json_str, timeout=120)
+#   incoming_env_json_str = String(reply.payload)
+#   incoming_env = msghandler.smartunpack(incoming_env_json_str)
+#   embedding_response = incoming_env["payloads"][1][2]
 
-  """ sql = "SELECT * FROM wine;"
+#   return embedding_response
-  result = execute_sql_winedb(sql)
+# end
-  """
-  function execute_sql_winedb(sql::T) where {T<:AbstractString}
-    host_url, _port = split(config["externalservice"]["sommpanion_db"]["url"], ':')
-    port = parse(Int, _port)
-    dbname = "winedb"
-    user = config["externalservice"]["sommpanion_db"]["user"]
-    password = config["externalservice"]["sommpanion_db"]["password"]
-    db_connection = LibPQ.Connection("host=$host_url port=$port dbname=$dbname user=$user password=$password")
-    result = nothing
-    try
-      result = LibPQ.execute(db_connection, sql)  
-    catch e
-      LibPQ.close(db_connection)
-    end
 
-    LibPQ.close(db_connection)
-    return result
-  end
 
-  """ find similar sql from vector database
+# """ sql = "SELECT * FROM wine;"
-  sql = "SELECT * FROM wine;"
+# result = execute_sql_winedb(sql)
-  result, distance = similar_sql_vectordb(sql)
+# """
-  """
+# function execute_sql_winedb(sql::T) where {T<:AbstractString}
-  function similar_sql_vectordb(sql::T; maxdistance::Number=0.2) where {T<:AbstractString}
+#   host_url, _port = split(config["externalservice"]["sommpanion_db"]["url"], ':')
-    tablename = "sqlllm_decision_repository"
+#   port = parse(Int, _port)
-    # get embedding of the query
+#   dbname = "winedb"
-    df = find_similar_text_from_vectordb(sql, tablename, 
+#   user = config["externalservice"]["sommpanion_db"]["user"]
-          "function_input_embedding", execute_sql_vectordb)
+#   password = config["externalservice"]["sommpanion_db"]["password"]
-    # println(df[1, [:id, :function_output]])
+#   db_connection = LibPQ.Connection("host=$host_url port=$port dbname=$dbname user=$user password=$password")
-    row, col = size(df)
+#   result = nothing
-    distance = row == 0 ? Inf : df[1, :distance]  
+#   try
-    if row != 0 && distance < maxdistance
+#     result = LibPQ.execute(db_connection, sql)  
-      # if there is usable SQL, return it.
+#   catch e
-      output_b64 = df[1, :function_output_base64]   # pick the closest match
+#     LibPQ.close(db_connection)
-      output_str = String(base64decode(output_b64))
+#   end
-      rowid = df[1, :id]
-      println("\n--| similar sql found. row id $rowid, distance $distance ", @__FILE__, ":", @__LINE__, " $(Dates.now())")
-      pprintln(output_str)
-      return (result=output_str, distance=distance)
-    else
-      println("\n--| similar sql not found, max distance $maxdistance ", @__FILE__, ":", @__LINE__, " $(Dates.now())")
-      return (result=nothing, distance=nothing)
-    end
-  end
   
-  """ insert query and sql into vector database
+#   LibPQ.close(db_connection)
-  query = "get all wines from wine table"
+#   return result
-  sql = "SELECT * FROM wine;"
+# end
-  insert_sql_vectordb(query, sql)
-  """
-  function insert_sql_vectordb(query::T1, SQL::T2; maxdistance::Number=3
-    ) where {T1<:AbstractString, T2<:AbstractString}
 
-    tablename = "sqlllm_decision_repository"
+# """ find similar sql from vector database
-    # get embedding of the query
+# sql = "SELECT * FROM wine;"
-    # query = state[:thoughtHistory][:question]
+# result, distance = similar_sql_vectordb(sql)
-    df = find_similar_text_from_vectordb(query, tablename, 
+# """
-                                    "function_input_embedding", execute_sql_vectordb)
+# function similar_sql_vectordb(sql::T; maxdistance::Number=1) where {T<:AbstractString}
-    row, col = size(df)
+#   tablename = "sqlllm_decision_repository"
-    distance = row == 0 ? Inf : df[1, :distance]
+#   # get embedding of the query
-    if row == 0 || distance > maxdistance # no close enough SQL stored in the database
+#   df = find_similar_text_from_vectordb(sql, tablename, 
-      _query_embedding = get_embedding([query])
+#         "function_input_embedding", execute_sql_vectordb)
-      _query_embedding = GeneralUtils.dictify(_query_embedding)
+#   # println(df[1, [:id, :function_output]])
-      # println("\n--- _query_embedding() ", @__FILE__, ":", @__LINE__, " $(Dates.now())")
+#   row, col = size(df)
-      # println(_query_embedding)
+#   distance = row == 0 ? Inf : df[1, :distance]  
-      # println("---\n")
+#   if row != 0 && distance < maxdistance
-      query_embedding = _query_embedding["data"][1]["embedding"]
+#     # if there is usable SQL, return it.
-      query = replace(query, "'" => "")
+#     output_b64 = df[1, :function_output_base64]   # pick the closest match
-      sql_base64 = base64encode(SQL)
+#     output_str = String(base64decode(output_b64))
-      sql_ = replace(SQL, "'" => "")
+#     rowid = df[1, :id]
+#     println("\n--| similar sql found. row id $rowid, distance $distance ", @__FILE__, ":", @__LINE__, " $(Dates.now())")
+#     pprintln(output_str)
+#     return (result=output_str, distance=distance)
+#   else
+#     println("\n--| similar sql not found, max distance $maxdistance ", @__FILE__, ":", @__LINE__, " $(Dates.now())")
+#     return (result=nothing, distance=nothing)
+#   end
+# end
 
-      sql = 
+# """ insert query and sql into vector database
-        """
+# query = "get all wines from wine table"
-        INSERT INTO $tablename (function_input, function_output, function_output_base64, function_input_embedding) VALUES ('$query', '$sql_', '$sql_base64', '$query_embedding');
+# sql = "SELECT * FROM wine;"
-        """
+# insert_sql_vectordb(query, sql)
-      # println("\n--| added new decision to vectorDB ", @__FILE__, ":", @__LINE__, " $(Dates.now())")
+# """
-      # println(sql)
+# function insert_sql_vectordb(query::T1, SQL::T2; maxdistance::Number=3
-      _ = execute_sql_vectordb(sql)
+#   ) where {T1<:AbstractString, T2<:AbstractString}
-    end
-  end
   
-  """ execute sql against vectordb
+#   tablename = "sqlllm_decision_repository"
-  sql = "SELECT * FROM wine;"
+#   # get embedding of the query
-  result = execute_sql_vectordb(sql)
+#   # query = state[:thoughtHistory][:question]
-  """
+#   df = find_similar_text_from_vectordb(query, tablename, 
-  function execute_sql_vectordb(sql::T) where {T<:AbstractString}
+#                                   "function_input_embedding", execute_sql_vectordb)
-    host_url, _port = split(config["externalservice"]["sommpanion_vectordb"]["url"], ':')
+#   row, col = size(df)
-    port = parse(Int, _port)
+#   distance = row == 0 ? Inf : df[1, :distance]
-    dbname = config["externalservice"]["sommpanion_vectordb"]["dbname"]
+#   if row == 0 || distance > maxdistance # no close enough SQL stored in the database
-    user = config["externalservice"]["sommpanion_vectordb"]["user"]
+#     _query_embedding = get_embedding([query])
-    password = config["externalservice"]["sommpanion_vectordb"]["password"]
+#     _query_embedding = GeneralUtils.dictify(_query_embedding)
-    DBconnection = LibPQ.Connection("host=$host_url port=$port dbname=$dbname user=$user password=$password")
+#     # println("\n--- _query_embedding() ", @__FILE__, ":", @__LINE__, " $(Dates.now())")
-    result = LibPQ.execute(DBconnection, sql)
+#     # println(_query_embedding)
-    close(DBconnection)
+#     # println("---\n")
-    return result
+#     query_embedding = _query_embedding["data"][1]["embedding"]
-  end
+#     query = replace(query, "'" => "")
+#     sql_base64 = base64encode(SQL)
+#     sql_ = replace(SQL, "'" => "")
 
-  """ search similar decision llm made from vectordb
+#     sql = 
-  """
+#       """
-  function similar_sommelier_decision(recentevents::T1; maxdistance::Integer=3
+#       INSERT INTO $tablename (function_input, function_output, function_output_base64, function_input_embedding) VALUES ('$query', '$sql_', '$sql_base64', '$query_embedding');
-    )::Union{AbstractDict, Nothing} where {T1<:AbstractString}
+#       """
+#     # println("\n--| added new decision to vectorDB ", @__FILE__, ":", @__LINE__, " $(Dates.now())")
+#     # println(sql)
+#     _ = execute_sql_vectordb(sql)
+#   end
+# end
 
-    tablename = "sommelier_decision_repository"
+# """ execute sql against vectordb
-    # find similar
+# sql = "SELECT * FROM wine;"
-    df = find_similar_text_from_vectordb(recentevents, tablename, 
+# result = execute_sql_vectordb(sql)
-                                    "function_input_embedding", execute_sql_vectordb)
+# """
-    row, col = size(df)
+# function execute_sql_vectordb(sql::T) where {T<:AbstractString}
-    distance = row == 0 ? Inf : df[1, :distance]
+#   host_url, _port = split(config["externalservice"]["sommpanion_vectordb"]["url"], ':')
-    if row != 0 && distance < maxdistance
+#   port = parse(Int, _port)
-      # if there is usable decision, return it.
+#   dbname = config["externalservice"]["sommpanion_vectordb"]["dbname"]
-      rowid = df[1, :id]
+#   user = config["externalservice"]["sommpanion_vectordb"]["user"]
-      println("\n--| found similar decision. row id $rowid, distance $distance ", @__FILE__, " ", @__LINE__)
+#   password = config["externalservice"]["sommpanion_vectordb"]["password"]
-      output_b64 = df[1, :function_output_base64]   # pick the closest match
+#   DBconnection = LibPQ.Connection("host=$host_url port=$port dbname=$dbname user=$user password=$password")
-      _output_str = String(base64decode(output_b64))
+#   result = LibPQ.execute(DBconnection, sql)
-      output = copy(JSON.read(_output_str))
+#   close(DBconnection)
-      return output
+#   return result
-    else
+# end
-      println("\n--| similar decision not found, max distance $maxdistance ", @__FILE__, " ", @__LINE__)
-      return nothing
-    end
-  end
 
-  """ search similar text from vectordb
+# """ search similar decision llm made from vectordb
-  """
+# """
-  function find_similar_text_from_vectordb(text::T1, tablename::T2, embeddingColumnName::T3,
+# function similar_sommelier_decision(recentevents::T1; maxdistance::Integer=3
-    vectorDB::Function; limit::Integer=1
+#   )::Union{AbstractDict, Nothing} where {T1<:AbstractString}
-    )::DataFrame where {T1<:AbstractString, T2<:AbstractString, T3<:AbstractString}
-    # get embedding from LLM service
-    _embedding = get_embedding([text])
-    _embedding = _embedding["data"][1]["embedding"]
-    _embedding = "$_embedding"
 
-    embedding = _embedding[4:end]
+#   tablename = "sommelier_decision_repository"
+#   # find similar
+#   df = find_similar_text_from_vectordb(recentevents, tablename, 
+#                                   "function_input_embedding", execute_sql_vectordb)
+#   row, col = size(df)
+#   distance = row == 0 ? Inf : df[1, :distance]
+#   if row != 0 && distance < maxdistance
+#     # if there is usable decision, return it.
+#     rowid = df[1, :id]
+#     println("\n--| found similar decision. row id $rowid, distance $distance ", @__FILE__, " ", @__LINE__)
+#     output_b64 = df[1, :function_output_base64]   # pick the closest match
+#     _output_str = String(base64decode(output_b64))
+#     output = copy(JSON.read(_output_str))
+#     return output
+#   else
+#     println("\n--| similar decision not found, max distance $maxdistance ", @__FILE__, " ", @__LINE__)
+#     return nothing
+#   end
+# end
 
-    # check whether there is close enough vector already store in vectorDB. if no, add, else skip 
+# """ search similar text from vectordb
-    sql = """
+# """
-          SELECT *, $embeddingColumnName <-> '$embedding' as distance
+# function find_similar_text_from_vectordb(text::T1, tablename::T2, embeddingColumnName::T3,
-          FROM $tablename
+#   vectorDB::Function; limit::Integer=1
-          ORDER BY distance LIMIT $limit;
+#   )::DataFrame where {T1<:AbstractString, T2<:AbstractString, T3<:AbstractString}
-          """
+#   # get embedding from LLM service
-    response = vectorDB(sql)
+#   _embedding = get_embedding([text])
-    df = DataFrame(response)
+#   _embedding = _embedding["data"][1]["embedding"]
+#   _embedding = "$_embedding"
 
-    return df
+#   embedding = _embedding[4:end] # remove 'Any' from Any[...]
-  end
 
-  """ insert decision llm made to vectordb
+#   # check whether there is close enough vector already store in vectorDB. if no, add, else skip 
-  """
+#   sql = """
-  function insert_sommelier_decision(recentevents::T1, decision::T2; maxdistance::Integer=5
+#         SELECT *, $embeddingColumnName <-> '$embedding' as distance
-    ) where {T1<:AbstractString, T2<:AbstractDict}
+#         FROM $tablename
-    tablename = "sommelier_decision_repository"
+#         ORDER BY distance LIMIT $limit;
-    # find similar
+#         """
-    df = find_similar_text_from_vectordb(recentevents, tablename, 
+#   response = vectorDB(sql)
-                                    "function_input_embedding", execute_sql_vectordb)
+#   df = DataFrame(response)
-    row, col = size(df)
+
-    distance = row == 0 ? Inf : df[1, :distance]
+#   return df
-    if row == 0 || distance > maxdistance # no close enough SQL stored in the database
+# end
-      _embedding = get_embedding([recentevents])[1]
+
-      recentevents_embedding = _embedding["data"][1]["embedding"]
+# """ insert decision llm made to vectordb
-      recentevents = replace(recentevents, "'" => "")
+# """
-      decision_json = JSON.json(decision)
+# function insert_sommelier_decision(recentevents::T1, decision::T2; maxdistance::Integer=5
-      decision_base64 = base64encode(decision_json)
+#   ) where {T1<:AbstractString, T2<:AbstractDict}
-      decision = replace(decision_json, "'" => "")
+#   tablename = "sommelier_decision_repository"
+#   # find similar
+#   df = find_similar_text_from_vectordb(recentevents, tablename, 
+#                                   "function_input_embedding", execute_sql_vectordb)
+#   row, col = size(df)
+#   distance = row == 0 ? Inf : df[1, :distance]
+#   if row == 0 || distance > maxdistance # no close enough SQL stored in the database
+#     _embedding = get_embedding([recentevents])[1]
+#     recentevents_embedding = _embedding["data"][1]["embedding"]
+#     recentevents = replace(recentevents, "'" => "")
+#     decision_json = JSON.json(decision)
+#     decision_base64 = base64encode(decision_json)
+#     decision = replace(decision_json, "'" => "")
+    
+#     sql = 
+#     """
+#     INSERT INTO $tablename (function_input, function_output, function_output_base64, function_input_embedding) VALUES ('$recentevents', '$decision', '$decision_base64', '$recentevents_embedding');
+#     """
+#     println("\n--| added new decision to vectorDB ", @__FILE__, " ", @__LINE__)
+#     println(sql)
+#     _ = execute_sql_vectordb(sql)
+#   else
+#     println("--| similar decision previously cached, distance $distance ", @__FILE__, " ", @__LINE__)
+#   end
+# end
+
+# function find_related_tables_for_user_question(question::String; top_row_num::Integer=20)
+
+#   metadata_df = GeneralUtils.extract_column_metadata(pg_conn_str)
+#   embedding_ready = GeneralUtils.generate_embedding_payloads(metadata_df)
+
+#   # use only text content
+#   embedding_ready_2 = [i["text_content"] for i in embedding_ready]
+
+#   table_embedding = get_embedding(embedding_ready_2)
+
+#   _user_question_embedding = get_embedding([question])
+#   user_question_embedding = Float64.(_user_question_embedding["data"][1]["embedding"])
+#   user_question_similarity = []
+
+#   for i in table_embedding["data"]
+#     i_data = i["embedding"]
+#     i_float = Float64.(i_data)
+#     r = 1 - Distances.cosine_dist(i_float, user_question_embedding)
+#     push!(user_question_similarity, r)
+#   end
+
+#   new_df = hcat(metadata_df, DataFrame(user_question_similarity = user_question_similarity))
+#   sorted_df = sort(new_df, :user_question_similarity, rev=true) # sort max to min
+#   _top_20_tables = unique(sorted_df[1:top_row_num, :table_name])
+#   top_20_tables = [i for i in _top_20_tables] # convert to Vector{String}
+#   g, id_to_table, table_to_id = GeneralUtils.harvest_db_undirected_schema_graph(pg_conn_str)
+#   table_relationship = GeneralUtils.resolve_semantic_cluster(top_20_tables, g, table_to_id, id_to_table)
+
+#   # tables that I should put schema in LLM context
+#   return table_relationship
+# end
+
+
+# function prepareContext(state::agentState)::agentContext
+  
+#   #TODO filter tools from state.tools based on user intend in user message and tool description
+#   filteredTools = state.tools
+
+#   #TODO add filtered tools to the current system prompt / modify systemPrompt here
+#   preparedSystemPrompt = state.systemPrompt
+  
+#   #TODO add system prompt, adjust/modify and inject additional context into messages
+#   preparedMessages = deepcopy(state.messages) # messages that will be send to LLM
+
+#   agentCtx = agentContext(preparedSystemPrompt, preparedMessages, filteredTools)
+
+#   return agentCtx
+# end
+
+# function formatMsgForLLM(ctx::agentContext)::Dict{String, Any}
+  
+# end
 
-      sql = 
-      """
-      INSERT INTO $tablename (function_input, function_output, function_output_base64, function_input_embedding) VALUES ('$recentevents', '$decision', '$decision_base64', '$recentevents_embedding');
-      """
-      println("\n--| added new decision to vectorDB ", @__FILE__, " ", @__LINE__)
-      println(sql)
-      _ = execute_sql_vectordb(sql)
-    else
-      println("--| similar decision previously cached, distance $distance ", @__FILE__, " ", @__LINE__)
-    end
-  end
 
 config = JSON.parsefile("./appconfig.json")
+host_url, _port = split(config["externalservice"]["sommpanion_db"]["url"], ':')
+port = parse(Int, _port)
+dbname = "winedb"
+user = config["externalservice"]["sommpanion_db"]["user"]
+password = config["externalservice"]["sommpanion_db"]["password"]
+pg_conn_str = "host=$host_url port=$port dbname=$dbname user=$user password=$password"
 sessionId = "0"
 backend_session_topic = "sommpanion.testsubject"
 agent_ch = Channel(8)
@@ -235,159 +296,19 @@ agent_conn  = NATS.connect(config["nats_server_info"]["url"])
 sub2 = NATS.subscribe(agent_conn, backend_session_topic) do msg
   put!(agent_ch, msg)
 end
-
-agent_context = YiemAgent.agentcontext(
-    text2text_instruct_llm,
-    get_embedding,
-    execute_sql_winedb,
-    similar_sql_vectordb,
-    insert_sql_vectordb,
-    similar_sommelier_decision,
-    insert_sommelier_decision
-  )
-
-  # can't instantiate
-  agent = YiemAgent.sommelier(
-    agent_context;
-    name="Janie",
-    id=sessionId,  # agent instance id
-    retailername="Yiem Wine Ltd.",
-    llmFormatName=""
-  )
-
-
-image1_path = "test/large_image.png"
-image1_bytes = read(image1_path)
-image1_base64_string = base64encode(image1_bytes)
-mime_type = "image/png" 
-data1_uri = "data:$(mime_type);base64,$(image1_base64_string)"
-
-# 1. Read local file and encode to base64 string
-image2_path = "test/small_image.png"
-image2_bytes = read(image2_path)
-image2_base64_string = base64encode(image2_bytes)
-mime_type = "image/png" 
-data2_uri = "data:$(mime_type);base64,$(image2_base64_string)"
-
-# 3. Construct payload with the Data URI
-message = Dict(
-    "role" => "user",
-    "content" => [
-      Dict("type" => "text", "text" => "Do you know type of wine in the image?"),
-      Dict(
-        "type" => "image_url",
-        "image_url" => Dict("url" => data1_uri)
-      )
-    ]
-  )
-
-result = YiemAgent.conversation(agent; userinput=message)
-println("\n$result")
-
-
-
-
-
-
-
-
-
-# message = Dict(
-#     "role" => "user",
-#     "content" => [
-#       Dict("type" => "text", "text" => 
-#       "
-#       เป็นงานเลี้ยงทั่วไป
-#       "),
-#     ]
-#   )
-
-# result = YiemAgent.conversation(agent; userinput=message)
-# println("\n$result")
-
-
-
-
-
-
-
-
-
-# message = Dict(
-#     "role" => "user",
-#     "content" => [
-#       Dict("type" => "text", "text" => "no thanks. that's all"),
-#     ]
-#   )
-
-# result = YiemAgent.conversation(agent; userinput=message)
-# println("\n$result")
-
-
-
-
-
-
-# message = Dict(
-#     "role" => "user",
-#     "content" => [
-#       Dict("type" => "text", "text" => "What about this wine?"),
-#       Dict(
-#         "type" => "image_url",
-#         "image_url" => Dict("url" => data2_uri)
-#       )
-#     ]
-#   )
-
-# result = YiemAgent.conversation(agent; userinput=message)
-# println("\n$result")
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
 
 
 
+#WORKING load tools
+text2text_llm = text2textInstructLLM(agent_conn, 
+                                    config["externalservice"]["servicesloadbalancer"]["nats"], 
+                                    "sender", 
+                                    config["externalservice"]["fileserver"]["url"])
 
+agent = YiemAgent.yiemAgent(
+  "/home/ton/docker-apps/sommpanion/agent-backend/tools",
+  text2text_llm
+)