pycoze 0.1.151__tar.gz → 0.1.153__tar.gz

{pycoze-0.1.151 → pycoze-0.1.153}/PKG-INFO

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

{pycoze-0.1.151 → pycoze-0.1.153}/pycoze/bot/agent_chat.py

@@ -32,7 +32,7 @@ def load_abilities(bot_setting_file: str):
     return abilities
 
 
-def agent_chat(bot_setting_file, history):
+def agent_chat(bot_setting_file, history, result_queue):
     role_setting = load_role_setting(bot_setting_file)
     abilities = load_abilities(bot_setting_file)
 
@@ -75,4 +75,5 @@ def agent_chat(bot_setting_file, history):
         assistant_message=prompt,
         tool_compatibility_mode=cfg["toolCompatibilityMode"],
     )
-    return asyncio.run(run_agent(agent, history, cfg["toolCompatibilityMode"]))
+    result = asyncio.run(run_agent(agent, history, cfg["toolCompatibilityMode"]))
+    result_queue.put(result_queue)

{pycoze-0.1.151 → pycoze-0.1.153}/pycoze/bot/bot.py

@@ -1,21 +1,29 @@
-from langchain_core.messages import HumanMessage
-from .agent import INPUT_MESSAGE, output
-from .agent_chat import agent_chat
-import json
-
-def chat(bot_setting_file: str):
-    history = []
-    while True:
-        input_text = input()
-        if not input_text.startswith(INPUT_MESSAGE):
-            raise ValueError("Invalid message")
-        message = json.loads(input_text[len(INPUT_MESSAGE) :])
-        history.append(HumanMessage(message["content"]))
-        result = agent_chat(bot_setting_file, history)
-        history.append(output("assistant", result))
-
-
-def get_chat_response(bot_setting_file: str, input_text: str):
-    history = [HumanMessage(input_text)]
-    result = agent_chat(bot_setting_file, history)
-    return result
+from langchain_core.messages import HumanMessage
+from .agent import INPUT_MESSAGE, output
+from .agent_chat import agent_chat
+import json
+import queue
+import threading
+
+
+def chat(bot_setting_file: str):
+    history = []
+    while True:
+        input_text = input()
+        if not input_text.startswith(INPUT_MESSAGE):
+            raise ValueError("Invalid message")
+        message = json.loads(input_text[len(INPUT_MESSAGE) :])
+        history.append(HumanMessage(message["content"]))
+        result_queue = queue.Queue()
+        agent_chat_thread = threading.Thread(target=agent_chat, args=(bot_setting_file, history, result_queue))
+
+        result = agent_chat(bot_setting_file, history)
+        history.append(output("assistant", result))
+
+
+
+def get_chat_response(bot_setting_file: str, input_text: str):
+    history = [HumanMessage(input_text)]
+    result_queue = queue.Queue()
+    result = agent_chat(bot_setting_file, history, result_queue)
+    return result

{pycoze-0.1.151 → pycoze-0.1.153}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.151 → pycoze-0.1.153}/pycoze.egg-info/SOURCES.txt

@@ -2,7 +2,6 @@ LICENSE
 README.md
 setup.py
 pycoze/__init__.py
-pycoze/module.py
 pycoze.egg-info/PKG-INFO
 pycoze.egg-info/SOURCES.txt
 pycoze.egg-info/dependency_links.txt

{pycoze-0.1.151 → pycoze-0.1.153}/setup.py

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