pycoze 0.1.173__tar.gz → 0.1.175__tar.gz

{pycoze-0.1.173 → pycoze-0.1.175}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.173
+Version: 0.1.175
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.173 → pycoze-0.1.175}/pycoze/bot/agent/chat.py

@@ -9,7 +9,7 @@ _INFOMATION_MESSAGE = "INFOMATION_MESSAGE=>"
 _LOG = "LOG=>"
 
 
-CHAT_DATA = {"output":""}
+CHAT_DATA = {"output":"", "info": ""}
 
 
 def log(content, *args, end="\n", **kwargs):
@@ -17,7 +17,7 @@ def log(content, *args, end="\n", **kwargs):
 
 
 def clear_chat_data():
-    CHAT_DATA = {"output":""}
+    CHAT_DATA = {"output":"", "info": ""}
 
 
 def output(role, content):
@@ -26,4 +26,5 @@ def output(role, content):
     print(_OUTPUT_MESSAGE + json.dumps({"role": role, "content": content}))
 
 def info(role, content):
+    CHAT_DATA["info"] += content
     print(_INFOMATION_MESSAGE + json.dumps({"role": role, "content": content}))

{pycoze-0.1.173 → pycoze-0.1.175}/pycoze/bot/agent_chat.py

@@ -76,4 +76,5 @@ async def agent_chat(bot_setting_file, history, should_exit):
         tool_compatibility_mode=cfg["toolCompatibilityMode"],
     )
     result = await run_agent(agent, history, cfg["toolCompatibilityMode"], should_exit)
-    output("assistant", result)
+    if not should_exit.is_set():
+        output("assistant", result)

{pycoze-0.1.173 → pycoze-0.1.175}/pycoze/bot/bot.py

@@ -10,10 +10,10 @@ import asyncio
 
 class PeekableQueue(queue.Queue):
     def peek(self):
-        with self._mutex:
-            if self.empty():
-                return None
-            return self.queue[0]
+        try:
+            return self._get()
+        except queue.Empty:
+            return None
 
 def chat(bot_setting_file: str):
     history = []
@@ -51,15 +51,14 @@ def chat(bot_setting_file: str):
                     while agent_chat_thread.is_alive():
                         print("alive")
                         if not input_queue.empty():
-                            print("not input_queue.empty()")
                             next_input_text = input_queue.peek()
-                            print("next_input_text", next_input_text)
                             if next_input_text == INTERRUPT_MESSAGE:
-                                print("should_exit")
+                                history.append(AIMessage(content=CHAT_DATA["info"]))
                                 should_exit.set()  # 设置退出标志
                                 break
                         time.sleep(0.1)  # 每隔 0.1 秒检查一次
-                    history.append(AIMessage(content=CHAT_DATA["output"]))
+                    if next_input_text != INTERRUPT_MESSAGE:
+                        history.append(AIMessage(content=CHAT_DATA["output"]))
                 except json.JSONDecodeError:
                     print("Invalid JSON format in input message.")
                 except KeyError:

{pycoze-0.1.173 → pycoze-0.1.175}/pycoze.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.173
+Version: 0.1.175
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.173 → pycoze-0.1.175}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pycoze",
-    version="0.1.173",
+    version="0.1.175",
     packages=find_packages(),
     install_requires=[],
     author="Yuan Jie Xiong",