pycoze 0.1.209__py3-none-any.whl → 0.1.211__py3-none-any.whl

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

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

{pycoze-0.1.209.dist-info → pycoze-0.1.211.dist-info}/RECORD

@@ -5,10 +5,10 @@ pycoze/automation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 pycoze/automation/browser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/automation/browser/edge_driver_manager.py,sha256=gpgseunph5owZH6EskSYthuhey2SU3UP204gY0yIcuI,3022
 pycoze/bot/__init__.py,sha256=6HHMxDQVOyZM9dtSjQm9tjGnhj4h7CixD0JOvEwTi48,41
-pycoze/bot/agent_chat.py,sha256=hr0jGTd-lvLSe4vCcvaN2rKhB7BMBcVBtI1FpVa6ngs,3016
-pycoze/bot/bot.py,sha256=MtQcntUrKqUhxMbInZYz0l3muuO2X_C9HFjYLXndFiE,1159
+pycoze/bot/agent_chat.py,sha256=4J-y7lSM2ULQQYIUB0L4i94m0KyeB4Gdf2U3zhVMOG4,3033
+pycoze/bot/bot.py,sha256=A1EryCW6WCnb6jIKc0V58Qzv6qQebKTfQxhcLp8Ngtc,1051
 pycoze/bot/agent/__init__.py,sha256=3wE8_FFQS8j2BY-g9Cr-onV0POEvDRZaw_NCzpqrNus,265
-pycoze/bot/agent/agent.py,sha256=Ueste4xWvkQFk5Giklv4c-kcxiPL-D-KafzZ4DONosY,3613
+pycoze/bot/agent/agent.py,sha256=chUgNZh6v6375L_Y2dBEAaLJyfmw4SygYjVVrDN8VIk,3548
 pycoze/bot/agent/assistant.py,sha256=3iLxnRvf_ia0cP-FHK5Fv4ylltlnzPq1KscRCFYqjkc,1147
 pycoze/bot/agent/chat.py,sha256=mubOCAHvA6VtyE6N40elI6KrP6A69uB_G6ihE3G_Vi4,860
 pycoze/bot/agent/agent_types/__init__.py,sha256=zmU2Kmrv5mCdfg-QlPn2H6pWxbGeq8s7YTqLhpzJC6k,179
@@ -28,8 +28,8 @@ pycoze/utils/__init__.py,sha256=Gi5EnrWZGMD2JRejgV4c_VLCXyvA2wwBFI_niDF5MUE,110
 pycoze/utils/arg.py,sha256=GtfGbMTMdaK75Fwh6MpUe1pCA5X6Ep4LFG7a72YrzjI,525
 pycoze/utils/env.py,sha256=W04lhvTHhAAC6EldP6kk2xrctqtu8K6kl1vDLZDNeh8,561
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.209.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.209.dist-info/METADATA,sha256=M10jwT4s0ZFcC1Cs2df86tpvX0D_cOb5AOfNnLEoT5o,726
-pycoze-0.1.209.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-pycoze-0.1.209.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.209.dist-info/RECORD,,
+pycoze-0.1.211.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.211.dist-info/METADATA,sha256=MPKRrqQHGS9v4n7MnO6JZ9eAvl1cf9rK2BVdmdzRp2g,726
+pycoze-0.1.211.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+pycoze-0.1.211.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.211.dist-info/RECORD,,