pycoze 0.1.164__tar.gz → 0.1.166__tar.gz

{pycoze-0.1.164 → pycoze-0.1.166}/PKG-INFO

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

{pycoze-0.1.164 → pycoze-0.1.166}/pycoze/bot/agent/agent.py

@@ -13,10 +13,13 @@ from langchain_core.agents import AgentFinish
 from .agent_types.const import HumanToolString
 
 
-async def run_agent(agent, inputs: list, tool_compatibility_mode:bool):
+async def run_agent(agent, inputs: list, tool_compatibility_mode: bool, should_exit):
     exist_ids = set()
     content_list = []
     async for event in agent.astream_events(inputs, version="v2"):
+        if should_exit.is_set():
+            break
+
         kind = event["event"]
         if kind == "on_chain_end":
             if "data" in event:
@@ -30,7 +33,7 @@ async def run_agent(agent, inputs: list, tool_compatibility_mode:bool):
                     for msg in input_list:
                         if isinstance(msg, HumanMessage) or isinstance(msg, SystemMessage):
                             if not tool_compatibility_mode or not msg.content.startswith(HumanToolString):
-                                content_list = [] # 防止内容重复
+                                content_list = []  # 防止内容重复
                         if isinstance(msg, AIMessage) and not isinstance(
                             msg, AIMessageChunk
                         ):
@@ -87,4 +90,4 @@ if __name__ == "__main__":
     )
 
     inputs = [HumanMessage(content="计算根号7+根号88")]
-    print(asyncio.run(run_agent(agent, inputs, True)))
+    print(asyncio.run(run_agent(agent, inputs, True, threading.Event())))

{pycoze-0.1.164 → pycoze-0.1.166}/pycoze/bot/agent/chat.py

@@ -9,7 +9,7 @@ _INFOMATION_MESSAGE = "INFOMATION_MESSAGE=>"
 _LOG = "LOG=>"
 
 
-CHAT_DATA = {"output":"", "info":""}
+CHAT_DATA = {"output":""}
 
 
 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":"", "info":""}
+    CHAT_DATA = {"output":""}
 
 
 def output(role, content):
@@ -26,5 +26,4 @@ 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.164 → pycoze-0.1.166}/pycoze/bot/agent_chat.py

@@ -75,20 +75,5 @@ async def agent_chat(bot_setting_file, history, should_exit):
         assistant_message=prompt,
         tool_compatibility_mode=cfg["toolCompatibilityMode"],
     )
-    
-    async def run_agent_with_exit_check():
-        result = await run_agent(agent, history, cfg["toolCompatibilityMode"])
-        if not should_exit.is_set():
-            output("assistant", result)
-
-    task = asyncio.create_task(run_agent_with_exit_check())
-
-    while not task.done():
-        if should_exit.is_set():
-            task.cancel()
-            try:
-                await task
-            except asyncio.CancelledError:
-                pass
-            break
-        await asyncio.sleep(0.1)
+    result = asyncio.run(run_agent(agent, history, cfg["toolCompatibilityMode"], should_exit))
+    output("assistant", result)

{pycoze-0.1.164 → pycoze-0.1.166}/pycoze/bot/bot.py

@@ -54,11 +54,9 @@ def chat(bot_setting_file: str):
                             next_input_text = input_queue.peek()
                             if next_input_text == INTERRUPT_MESSAGE:
                                 should_exit.set()  # 设置退出标志
-                                history.append(AIMessage(content=CHAT_DATA["info"]))
                                 break
                         time.sleep(0.1)  # 每隔 0.1 秒检查一次
-                    if next_input_text != INTERRUPT_MESSAGE:
-                        history.append(AIMessage(content=CHAT_DATA["output"]))
+                    history.append(AIMessage(content=CHAT_DATA["output"]))
                 except json.JSONDecodeError:
                     print("Invalid JSON format in input message.")
                 except KeyError:
@@ -73,11 +71,6 @@ def chat(bot_setting_file: str):
 def get_chat_response(bot_setting_file: str, input_text: str):
     history = [HumanMessage(input_text)]
     clear_chat_data()
-    
-    # 为每个线程创建一个独立的退出标志
     should_exit = threading.Event()
-    
-    # 使用 asyncio.run 来运行异步函数，并等待其完成
     asyncio.run(agent_chat(bot_setting_file, history, should_exit))
-    
     return CHAT_DATA["output"]

{pycoze-0.1.164 → pycoze-0.1.166}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.164 → pycoze-0.1.166}/setup.py

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