update

src/type.jl

@@ -77,13 +77,15 @@ julia> agent = YiemAgent.bsommelier(
 
 # Signature
 """
-@kwdef mutable struct sommelier <: agent
+mutable struct sommelier <: agent
   name::String  # agent name
   id::String    # agent id
   config::Dict  # agent config
   tools::Dict   
-  thinkinglimit::Integer # thinking round limit
-  thinkingcount::Integer      # used to count attempted round of a task
+  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
 
   """ Memory
     Ref: Chat prompt format https://huggingface.co/TheBloke/Llama-2-7B-Chat-GGML/discussions/3
@@ -94,17 +96,50 @@ julia> agent = YiemAgent.bsommelier(
     ]
 
   """
-  chathistory::Vector{Dict{Symbol, Any}} = Vector{Dict{Symbol, Any}}() 
-
-  maxHistoryMsg::Integer  # 21th and earlier messages will get summarized
-  keywordinfo::Dict{Symbol, Any} = Dict{Symbol, Any}(
-    :customerinfo => Dict{Symbol, Any}(),
-    :storeinfo => Dict{Symbol, Any}(),
-    )
-  mctstree::Dict{Symbol, Any} = Dict{Symbol, Any}()
+  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}
+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,
+      ),
+    )
+  ;
+  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
@@ -113,45 +148,7 @@ julia> agent = YiemAgent.bsommelier(
 
     :activeplan => Dict{Symbol, Any}(),  # current using plan
     :currenttrajectory=> Dict{Symbol, Any}(),  # store question, thought, action, observation, ...
-  )
-
-  # put incoming message here. waiting for further processing
-  receiveUserMsgChannel::Channel{Dict} = Channel{Dict}(8)  # for incoming user communication
-  receiveInternalMsgChannel::Channel{Dict} = Channel{Dict}(8)  # for internal communication
-end
-
-function sommelier(
-    receiveUserMsgChannel::Channel,
-    receiveInternalMsgChannel::Channel,
-    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,
-      ),
     )
-    ;
-    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,
-      ),
-    ),
-    maxHistoryMsg::Integer= 20,
-    thinkinglimit::Integer= 5,
-    thinkingcount::Integer= 0,
   )
 
   #[NEXTVERSION] publish to a.config[:configtopic] to get a config.
@@ -159,16 +156,18 @@ function sommelier(
   #[NEXTVERSION] set agent according to config
 
   newAgent = sommelier(
-    receiveUserMsgChannel= receiveUserMsgChannel,
-    receiveInternalMsgChannel= receiveInternalMsgChannel,
-    name= name,
-    id= id,
-    config= config,
-    maxHistoryMsg= maxHistoryMsg,
-    tools= tools,
-    thinkinglimit= thinkinglimit,
-    thinkingcount= thinkingcount,
-  )
+    name,
+    id,
+    config,
+    tools,
+    maxiterations,
+    totalsample,
+    maxDepth,
+    maxHistoryMsg,
+    chathistory,
+    keywordinfo,
+    plan,
+    )
   
   return newAgent
 end