update

src/util.jl

@@ -1,12 +1,168 @@
 module util
 
-# export sendReceivePrompt
+export clearhistory, addNewMessage
 
 using UUIDs, Dates, DataStructures, HTTP, MQTTClient, JSON3
 using GeneralUtils
 using ..type
 
 # ---------------------------------------------- 100 --------------------------------------------- #
+
+""" Clear agent chat history.
+
+  Arguments\n
+  -----
+    a::agent
+      an agent
+
+  Return\n
+  -----
+    nothing
+
+  Example\n
+  -----
+  ```jldoctest
+  julia> using YiemAgent, MQTTClient, GeneralUtils
+  julia> client, connection = MakeConnection("test.mosquitto.org", 1883)
+  julia> connect(client, connection)
+  julia> msgMeta = GeneralUtils.generate_msgMeta("testtopic")
+  julia> agentConfig = Dict(
+            :receiveprompt=>Dict(
+              :mqtttopic=> "testtopic/receive",
+            ),
+            :receiveinternal=>Dict(  
+              :mqtttopic=> "testtopic/internal",
+            ),
+            :text2text=>Dict(
+              :mqtttopic=> "testtopic/text2text",
+            ),
+          )
+  julia> a = YiemAgent.sommelier(
+            client,
+            msgMeta,
+            agentConfig,
+            )
+  julia> YiemAgent.addNewMessage(a, "user", "hello")
+  julia> YiemAgent.clearhistory(a)
+  ```
+
+  Signature\n
+  -----
+"""
+function clearhistory(a::T)  where {T<:agent}
+  empty!(a.chathistory)
+  empty!(a.mctstree)
+  empty!(a.plan[:activeplan])
+  empty!(a.plan[:currenttrajectory])
+end
+
+
+""" Add new message to agent.
+
+  Arguments\n
+  -----
+    a::agent
+      an agent
+    role::String
+      message sender role i.e. system, user or assistant
+    text::String
+      message text
+
+  Return\n
+  -----
+    nothing
+
+  Example\n
+  -----
+  ```jldoctest
+  julia> using YiemAgent, MQTTClient, GeneralUtils
+  julia> client, connection = MakeConnection("test.mosquitto.org", 1883)
+  julia> connect(client, connection)
+  julia> msgMeta = GeneralUtils.generate_msgMeta("testtopic")
+  julia> agentConfig = Dict(
+            :receiveprompt=>Dict(
+              :mqtttopic=> "testtopic/receive",
+            ),
+            :receiveinternal=>Dict(  
+              :mqtttopic=> "testtopic/internal",
+            ),
+            :text2text=>Dict(
+              :mqtttopic=> "testtopic/text2text",
+            ),
+          )
+  julia> a = YiemAgent.sommelier(
+            client,
+            msgMeta,
+            agentConfig,
+            )
+  julia> YiemAgent.addNewMessage(a, "user", "hello")
+  ```
+
+  Signature\n
+  -----
+"""
+function addNewMessage(a::T1, role::String, text::T2; 
+                      maximumMsg::Integer=20) where {T1<:agent, T2<:AbstractString}
+  if role ∉ ["system", "user", "assistant"]   # guard against typo
+    error("role is not in agent.availableRole $(@__LINE__)")
+  end
+
+  #TODO summarize the oldest 10 message
+  if length(a.chathistory) > maximumMsg
+    summarize(a.chathistory)
+  else
+    d = Dict(:role=> role, :text=> text, :timestamp=> Dates.now())
+    push!(a.chathistory, d)
+  end
+end
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+