pycoze 0.1.209__tar.gz → 0.1.210__tar.gz

{pycoze-0.1.209 → pycoze-0.1.210}/PKG-INFO

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

{pycoze-0.1.209 → pycoze-0.1.210}/pycoze/bot/agent/agent.py

@@ -13,13 +13,10 @@ from langchain_core.agents import AgentFinish
 from .agent_types.const import HumanToolString
 
 
-async def run_agent(agent, inputs: list, tool_compatibility_mode: bool, should_exit):
+async def run_agent(agent, inputs: list, tool_compatibility_mode: bool):
     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:

{pycoze-0.1.209 → pycoze-0.1.210}/pycoze/bot/agent_chat.py

@@ -34,7 +34,7 @@ def load_abilities(bot_setting_file: str):
     return abilities
 
 
-async def agent_chat(bot_setting_file, history, should_exit):
+async def agent_chat(bot_setting_file, history):
     role_setting = load_role_setting(bot_setting_file)
     abilities = load_abilities(bot_setting_file)
 
@@ -77,6 +77,9 @@ async def agent_chat(bot_setting_file, history, should_exit):
         assistant_message=prompt,
         tool_compatibility_mode=cfg["toolCompatibilityMode"],
     )
-    result = await run_agent(agent, history, cfg["toolCompatibilityMode"], should_exit)
-    if not should_exit.is_set():
-        output("assistant", result)
+    try:
+        result = await run_agent(agent, history, cfg["toolCompatibilityMode"])
+        return result
+    except KeyboardInterrupt:
+        return CHAT_DATA["info"]
+        

{pycoze-0.1.209 → pycoze-0.1.210}/pycoze/bot/bot.py

@@ -22,9 +22,8 @@ def chat(bot_setting_file: str):
             raise ValueError("Invalid message")
         message = json.loads(input_text[len(INPUT_MESSAGE) :])
         history.append(HumanMessage(message["content"]))
-        should_exit = threading.Event()
-        asyncio.run(agent_chat(bot_setting_file, history, should_exit))
-        history.append(AIMessage(CHAT_DATA["output"]))
+        result = asyncio.run(agent_chat(bot_setting_file, history))
+        history.append(AIMessage(result))
 
 
 
@@ -32,6 +31,5 @@ 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(agent_chat(bot_setting_file, history, should_exit))
+    asyncio.run(agent_chat(bot_setting_file, history))
     return CHAT_DATA["output"]

{pycoze-0.1.209 → pycoze-0.1.210}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.209 → pycoze-0.1.210}/setup.py

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