update

src/type.jl

@@ -80,12 +80,8 @@ julia> agent = YiemAgent.bsommelier(
 mutable struct sommelier <: agent
   name::String  # agent name
   id::String    # agent id
-  config::Dict  # agent config
   tools::Dict   
-  maxiterations::Integer  # how many thinking round
-  totalsample::Integer    # how many sample in each thinking round
-  maxDepth::Integer       # how many step ahead to be simulated start from current state into the future
-  maxHistoryMsg::Integer  # 21th and earlier messages will get summarized
+  maxHistoryMsg::Integer  # e.g. 21th and earlier messages will get summarized
 
   """ Memory
     Ref: Chat prompt format https://huggingface.co/TheBloke/Llama-2-7B-Chat-GGML/discussions/3
@@ -97,76 +93,48 @@ mutable struct sommelier <: agent
 
   """
   chathistory::Vector{Dict{Symbol, Any}}
-  keywordinfo::Dict{Symbol, Any}
-  
-  # 1-historyPoint is in Dict{Symbol, Any} and compose of: 
-  # state, statevalue, thought, action, observation
-  plan::Dict{Symbol, Any}
+  shortmem::Dict{Symbol, Any}
+
+  # communication function
+  text2textInstructLLM::Function
 end
 
 function sommelier(
-  config::Dict = Dict(
-    :mqttServerInfo=> Dict(
-      :broker=> nothing,
-      :port=> nothing,
-      ),
-    :receivemsg=>  Dict(
-      :prompt=> nothing, # topic to receive prompt i.e. frontend send msg to this topic
-      :internal=> nothing,
-      ),
-    :thirdPartyService=> Dict(
-      :text2textinstruct=> nothing,
-      :text2textchat=> nothing,
-      ),
-    )
+  text2textInstructLLM::Function
   ;
   name::String= "Assistant",
   id::String= string(uuid4()),
-  tools::Dict= Dict(
-    :chatbox=> Dict(
-      :name => "chatbox",
-      :description => "Useful for when you need to communicate with the user.",
-      :input => "Input should be a conversation to the user.",
-      :output => "" ,
-      :func => nothing,
-      ),
-    ),
-  maxiterations::Integer= 3,
-  totalsample::Integer= 3,
-  maxDepth::Integer= 3,
   maxHistoryMsg::Integer= 20,
   chathistory::Vector{Dict{Symbol, Any}} = Vector{Dict{Symbol, Any}}(),
-  keywordinfo::Dict{Symbol, Any} = Dict{Symbol, Any}(
-    :customerinfo => Dict{Symbol, Any}(),
-    :storeinfo => Dict{Symbol, Any}(),
-    ),
-  plan::Dict{Symbol, Any} = Dict{Symbol, Any}(
-
-    # store 3 to 5 best plan AI frequently used to avoid having to search MCTS all the time
-    # each plan is in [historyPoint_1, historyPoint_2, ...] format
-    :existingplan => Vector(),
-
-    :activeplan => Dict{Symbol, Any}(),  # current using plan
-    :currenttrajectory=> Dict{Symbol, Any}(),  # store question, thought, action, observation, ...
-    )
   )
 
-  #[NEXTVERSION] publish to a.config[:configtopic] to get a config.
-  #[NEXTVERSION] get a config message in a.mqttMsg_internal
-  #[NEXTVERSION] set agent according to config
+  tools = Dict( # update input format
+      "chatbox"=> Dict(
+        :description => "<askbox tool description>Useful for when you need to ask the user for more context. Do not ask the user their own question.</askbox tool description>",
+        :input => """<input>Input is a text in JSON format.</input><input example>{\"Q1\": \"How are you doing?\", \"Q2\": \"How may I help you?\"}</input example>""",
+        :output => "" ,
+      ),
+      "winestock"=> Dict(
+        :description => "<winestock tool description>A handy tool for searching wine in your inventory that match the user preferences.</winestock tool description>",
+        :input => """<input>Input is a JSON-formatted string that contains a detailed and precise search query.</input><input example>{\"wine type\": \"rose\", \"price\": \"max 35\", \"sweetness level\": \"sweet\", \"intensity level\": \"light bodied\", \"Tannin level\": \"low\", \"Acidity level\": \"low\"}</input example>""",
+        :output => """<output>Output are wines that match the search query in JSON format.""",
+      ),
+      # "finalanswer"=> Dict(
+      #   :description => "<tool description>Useful for when you are ready to recommend wines to the user.</tool description>",
+      #   :input => """<input format>{\"finalanswer\": \"some text\"}.</input format><input example>{\"finalanswer\": \"I recommend Zena Crown Vista\"}</input example>""",
+      #   :output => "" ,
+      #   :func => nothing,
+      # ),
+    )
 
   newAgent = sommelier(
     name,
     id,
-    config,
     tools,
-    maxiterations,
-    totalsample,
-    maxDepth,
     maxHistoryMsg,
     chathistory,
-    keywordinfo,
-    plan,
+    Dict{Symbol, Any}(),
+    text2textInstructLLM,
     )
   
   return newAgent