pycoze 0.1.174__py3-none-any.whl → 0.1.175__py3-none-any.whl

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/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/bot/bot.py

@@ -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.174.dist-info → pycoze-0.1.175.dist-info}/METADATA

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

{pycoze-0.1.174.dist-info → pycoze-0.1.175.dist-info}/RECORD

@@ -2,12 +2,12 @@ pycoze/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/ai/__init__.py,sha256=odM2lgYSnApxw4AzLV_5AyaxL5_MLfydOSB1VmLGFyA,75
 pycoze/ai/vram_reserve.py,sha256=QbqaA8qv87cnEpOVDMygi0BNMxuhLYwj1UKfR_D5BD4,4340
 pycoze/bot/__init__.py,sha256=6HHMxDQVOyZM9dtSjQm9tjGnhj4h7CixD0JOvEwTi48,41
-pycoze/bot/agent_chat.py,sha256=Q5uPcLre-x9oxwGvZYWIYXl7GzNSI2fgSEPg4jPb25I,2876
-pycoze/bot/bot.py,sha256=yRWOYbZKWyaO7uO3QkRSrYIh0cuzbWrz46QX1P-ZyxM,3136
+pycoze/bot/agent_chat.py,sha256=3LIsLwlZ1oLvtx92NkLLGPt8QWDwjeUXz9oqhOyxAxU,2914
+pycoze/bot/bot.py,sha256=NV7OlU9JW5h6ObHakiEfjwLQtluOjX4PHRPoOTx5u00,3101
 pycoze/bot/agent/__init__.py,sha256=3wE8_FFQS8j2BY-g9Cr-onV0POEvDRZaw_NCzpqrNus,265
 pycoze/bot/agent/agent.py,sha256=7dyAsHSVnBmFUK9GR_BumkjSgrvQA0KLj0e3aDN_JU4,3592
 pycoze/bot/agent/assistant.py,sha256=3iLxnRvf_ia0cP-FHK5Fv4ylltlnzPq1KscRCFYqjkc,1147
-pycoze/bot/agent/chat.py,sha256=fgsQGYXGSB_TO4iKfWMr85Rx3tqW4MFawid2dgcXrE4,691
+pycoze/bot/agent/chat.py,sha256=CxIoTYJ2CXzukTQDXP2YhS-wnJTiRNoPXq-_i0CNfgA,748
 pycoze/bot/agent/agent_types/__init__.py,sha256=zmU2Kmrv5mCdfg-QlPn2H6pWxbGeq8s7YTqLhpzJC6k,179
 pycoze/bot/agent/agent_types/const.py,sha256=BfUKPrhAHREoMLHuFNG2bCIEkC1-f7K0LEqNg4RwiRE,70
 pycoze/bot/agent/agent_types/openai_func_call_agent.py,sha256=SnEm5MODHn2uMsaMNqgzULM_91vqLHC0TU6ovwCOqLU,6675
@@ -23,8 +23,8 @@ pycoze/ui/ui_def.py,sha256=UhhU_yB3GV9ISbvTWT48hsHPHI250BhMILh6bu5Uioo,4206
 pycoze/utils/__init__.py,sha256=ET0W5wzq4zlY3dr1wHVbbeRKlKdC_zqHt9b0jZyHohw,94
 pycoze/utils/arg.py,sha256=orLVEGw3x2f3l7bZDbBPkkdDPSdqXo-_Rs-4ZhzmrEw,984
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.174.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.174.dist-info/METADATA,sha256=IDJg2useghHBEoLce7nCxkkV9nftgWIAoD-UdfmMlP4,726
-pycoze-0.1.174.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-pycoze-0.1.174.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.174.dist-info/RECORD,,
+pycoze-0.1.175.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.175.dist-info/METADATA,sha256=2_ItfL7g_r5Vf9m3E6sD9KMuC3Q629M-WPAo0QVEI4w,726
+pycoze-0.1.175.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+pycoze-0.1.175.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.175.dist-info/RECORD,,