update

test/runtest.jl

@@ -0,0 +1,109 @@
+using Revise      # remove when this package is completed
+using YiemAgent, GeneralUtils, JSON3, MQTTClient, Dates, UUIDs
+using Base.Threads
+
+# ---------------------------------------------- 100 --------------------------------------------- #
+
+config = copy(JSON3.read("config.json"))
+
+instanceInternalTopic = config[:serviceInternalTopic][:value] * "/1" 
+
+client, connection = MakeConnection(config[:mqttServerInfo][:value][:broker], 
+                                    config[:mqttServerInfo][:value][:port])
+
+receiveUserMsgChannel = Channel{Dict}(4)
+receiveInternalMsgChannel = Channel{Dict}(4)
+
+msgMeta = GeneralUtils.generate_msgMeta(
+  "N/A", 
+  replyTopic = config[:servicetopic][:value]  # ask frontend reply to this instance_chat_topic
+  )  
+
+agentConfig = Dict(
+    :receiveprompt=>Dict(
+      :mqtttopic=> config[:servicetopic][:value], # topic to receive prompt i.e. frontend send msg to this topic
+    ),
+    :receiveinternal=>Dict(  
+      :mqtttopic=> instanceInternalTopic, # receive topic for model's internal 
+    ),
+    :text2text=>Dict(
+      :mqtttopic=> config[:text2text][:value],
+    ),
+  )
+
+# Instantiate an agent
+tools=Dict( # update input format
+    "askbox"=> 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 => "" ,
+      :func => nothing,
+    ),
+    # "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.""",
+    #   :func => ChatAgent.winestock,
+    # ),
+    "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,
+    ),
+  )
+
+  a = YiemAgent.sommelier(
+    receiveUserMsgChannel,
+    receiveInternalMsgChannel,
+    msgMeta,
+    agentConfig, 
+    name= "assistant",
+    id= "randomSessionID",  # agent instance id
+    tools=tools,
+  )
+
+response = YiemAgent.conversation(a, Dict(:text=> "newtopic", ) )
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+