Fixes chat history

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='runai',
-    version="v1.4.0",
+    version="v1.4.1",
     author="Capsize LLC",
     description="Run a socket server for AI models.",
     long_description=open("README.md", "r", encoding="utf-8").read(),

src/runai/client.py

@@ -132,6 +132,11 @@ def update_history(self, name: str, message: str):
         })
 
     def do_query(self, user_prompt, instructions):
+        if len(self.history):
+            instructions += "\nThe conversation so far:\n"
+            for turn in self.history:
+                instructions += f"{turn['name']}: {turn['message']}\n"
+
         self.send_message(json.dumps({
             "history": self.history,
             "listener": self.user_agent.to_dict() if self.user_agent else None,
@@ -181,6 +186,7 @@ def run(self):
                 self.handle_res(res)
 
     def handle_res(self, res):
+        self.update_history(self.bot_agent.name, res)
         print(res)
 
     def connect(self):

src/runai/server.py

@@ -441,7 +441,6 @@ def watch_connection(self):
 
     def query_llm(self, data: dict):
         for text in self.llm_handler.query_model(data):
-            print(text)
             self.send_message(text)
         self.send_end_message()