update

src/type.jl

@@ -1,6 +1,6 @@
 module type
   
-export agent, sommelier, companion
+export agent, sommelier, companion, virtualcustomer
 
 using Dates, UUIDs, DataStructures, JSON3
 using GeneralUtils
@@ -24,17 +24,13 @@ end
 function companion(
   func::NamedTuple  # NamedTuple of functions
   ;
-  systemmsg::Union{String, Nothing}= nothing,
   name::String= "Assistant",
   id::String= GeneralUtils.uuid4snakecase(), 
   maxHistoryMsg::Integer= 20,
   chathistory::Vector{Dict{Symbol, String}} = Vector{Dict{Symbol, String}}(),
-  llmFormatName::String= "granite3"
-  )
-
-  if systemmsg === nothing
-    systemmsg = 
-      """
+  llmFormatName::String= "granite3",
+  systemmsg::String=
+    """
       Your name: $name
       Your sex: Female
       Your role: You are a helpful assistant.
@@ -43,8 +39,8 @@ function companion(
       - Your like to be short and concise.
 
       Let's begin!
-      """
-  end
+      """,
+  )
 
   tools = Dict( # update input format
       "CHATBOX"=> Dict(
@@ -222,6 +218,80 @@ function sommelier(
 end
 
 
+mutable struct virtualcustomer <: agent
+  name::String  # agent name
+  id::String    # agent id
+  systemmsg::String # system message
+  tools::Dict   
+  maxHistoryMsg::Integer  # e.g. 21th and earlier messages will get summarized
+  chathistory::Vector{Dict{Symbol, Any}}
+  memory::Dict{Symbol, Any}
+  func  # NamedTuple of functions 
+  llmFormatName::String
+end
+
+function virtualcustomer(
+  func,  # NamedTuple of functions
+  ;
+  name::String= "Assistant",
+  id::String= string(uuid4()),
+  maxHistoryMsg::Integer= 20,
+  chathistory::Vector{Dict{Symbol, String}} = Vector{Dict{Symbol, String}}(),
+  llmFormatName::String= "granite3",
+  systemmsg::String=
+    """
+      Your name: $name
+      Your sex: Female
+      Your role: You are a helpful assistant.
+      You should follow the following guidelines:
+      - Focus on the latest conversation.
+      - Your like to be short and concise.
+
+      Let's begin!
+      """,
+  )
+
+  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 => "" ,
+      ),
+    )
+
+  """ Memory
+    Ref: Chat prompt format https://huggingface.co/TheBloke/Llama-2-7B-Chat-GGML/discussions/3
+    NO "system" message in chathistory because I want to add it at the inference time
+    chathistory= [
+      Dict(:name=>"user", :text=> "Wassup!", :timestamp=> Dates.now()),
+      Dict(:name=>"assistant", :text=> "Hi I'm your assistant.", :timestamp=> Dates.now()),
+    ]
+  """
+  memory = Dict{Symbol, Any}(
+            :shortmem=> OrderedDict{Symbol, Any}(
+            ),
+            :events=> Vector{Dict{Symbol, Any}}(),
+            :state=> Dict{Symbol, Any}(
+            ),
+            :recap=> OrderedDict{Symbol, Any}(),
+          )
+
+  newAgent = virtualcustomer(
+    name,
+    id,
+    systemmsg,
+    tools,
+    maxHistoryMsg,
+    chathistory,
+    memory,
+    func,
+    llmFormatName
+    )
+  
+  return newAgent
+end
+
+