update

test/runtest.jl

@@ -13,60 +13,60 @@ client, connection = MakeConnection(config[:mqttServerInfo][:broker],
 
 receiveUserMsgChannel = Channel{Dict}(4)
 receiveInternalMsgChannel = Channel{Dict}(4)
-
+println(typeof(connection))
 msgMeta = GeneralUtils.generate_msgMeta(
   "N/A", 
   replyTopic = config[:servicetopic][:mqtttopic]  # ask frontend reply to this instance_chat_topic
   )  
 
-agentConfig = Dict(
-  :mqttServerInfo=> config[:mqttServerInfo],
-  :receivemsg=>  Dict(
-    :prompt=> config[:servicetopic][:mqtttopic], # topic to receive prompt i.e. frontend send msg to this topic
-    :internal=> instanceInternalTopic,
-  ),
-  :externalservice=> config[:externalservice],
-)
+# commConfig = Dict(
+#   :mqttServerInfo=> config[:mqttServerInfo],
+#   :receivemsg=>  Dict(
+#     :prompt=> config[:servicetopic][:mqtttopic], # topic to receive prompt i.e. frontend send msg to this topic
+#   ),
+#   :externalservice=> config[:externalservice],
+# )
 
-# 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,
-    ),
+function text2textInstructLLM(prompt::String)
+  msgMeta = GeneralUtils.generate_msgMeta(
+    config[:externalservice][:text2textinstruct][:mqtttopic],
+    senderName= "yiemagent",
+    senderId= string(uuid4()),
+    receiverName= "text2textinstruct",
+    mqttBroker= config[:mqttServerInfo][:broker],
+    mqttBrokerPort= config[:mqttServerInfo][:port],
   )
 
+  outgoingMsg = Dict(
+    :msgMeta=> msgMeta,
+    :payload=> Dict(
+      :text=> prompt,
+      :kwargs=> Dict(
+        :max_tokens=> 512,
+        :stop=> ["<|eot_id|>"],
+        :temperature=> 0.2,
+        )
+      )
+    )
+
+  _response = GeneralUtils.sendReceiveMqttMsg(outgoingMsg)
+  response = _response[:response][:text]
+
+  return response
+end
+
+  # Instantiate an agent
   a = YiemAgent.sommelier(
-    agentConfig, 
+    text2textInstructLLM,
     name="assistant",
     id="testingSessionID",  # agent instance id
-    tools=tools,
   )
 
 # response = YiemAgent.conversation(a, Dict(:text=> "newtopic",) )
 
-response = YiemAgent.conversation(a, Dict(:text=> "Hello, I would like a get a bottle of wine",
-                                          :select=> nothing,
-                                          :reward=> 0,
-                                          :isterminal=> false,
-                                ) )
+response = YiemAgent.conversation(a, Dict(:text=> "Hello, I would like a get a bottle of wine."))
 println("---> YiemAgent: ", response)
 
-#BUG mcts do not start at current chat history
 response = YiemAgent.conversation(a, Dict(:text=> "I'm having a graduation party this evening. I'll pay at most 30 bucks.",
                                           :select=> nothing,
                                           :reward=> 0,
@@ -96,9 +96,6 @@ response = YiemAgent.winestock(a, dummyinput)
 
 
 
-
-
-