pycoze 0.1.156__tar.gz → 0.1.157__tar.gz

{pycoze-0.1.156 → pycoze-0.1.157}/PKG-INFO

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

{pycoze-0.1.156 → pycoze-0.1.157}/pycoze/bot/bot.py

@@ -2,9 +2,8 @@ from langchain_core.messages import HumanMessage
 from .agent import INPUT_MESSAGE, output
 from .agent_chat import agent_chat
 import json
-import queue
 import threading
-
+from .chat import CHAT_DATA, clear_chat_data
 
 def chat(bot_setting_file: str):
     history = []
@@ -14,17 +13,16 @@ def chat(bot_setting_file: str):
             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))
+        clear_chat_data()
+        agent_chat_thread = threading.Thread(target=agent_chat, args=(bot_setting_file, history))
         agent_chat_thread.start()
         agent_chat_thread.join()
-        result = result_queue.get()
-        history.append(output("assistant", result))
+        history.append(output("assistant", CHAT_DATA["output"]))
 
 
 
 def get_chat_response(bot_setting_file: str, input_text: str):
     history = [HumanMessage(input_text)]
-    result_queue = queue.Queue()
-    agent_chat(bot_setting_file, history, result_queue)
-    return result_queue.get()
+    clear_chat_data()
+    agent_chat(bot_setting_file, history)
+    return CHAT_DATA["output"]

{pycoze-0.1.156/pycoze/bot/agent → pycoze-0.1.157/pycoze/bot}/chat.py

@@ -8,15 +8,24 @@ _INFOMATION_MESSAGE = "INFOMATION_MESSAGE=>"
 _LOG = "LOG=>"
 
 
+CHAT_DATA = {output:"",info:""}
+
+
 def log(content, *args, end="\n", **kwargs):
     print(_LOG + content, *args, end=end, **kwargs)
 
 
+def clear_chat_data():
+    CHAT_DATA = {output:"",info:""}
+
+
 def output(role, content):
     assert role == "assistant"
     print(_OUTPUT_MESSAGE + json.dumps({"role": role, "content": content}))
+    CHAT_DATA["output"] = content
     return AIMessage(content=content)
 
 
 def info(role, content):
+    CHAT_DATA["info"] += content
     print(_INFOMATION_MESSAGE + json.dumps({"role": role, "content": content}))

{pycoze-0.1.156 → pycoze-0.1.157}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.156 → pycoze-0.1.157}/pycoze.egg-info/SOURCES.txt

@@ -11,10 +11,10 @@ pycoze/ai/vram_reserve.py
 pycoze/bot/__init__.py
 pycoze/bot/agent_chat.py
 pycoze/bot/bot.py
+pycoze/bot/chat.py
 pycoze/bot/agent/__init__.py
 pycoze/bot/agent/agent.py
 pycoze/bot/agent/assistant.py
-pycoze/bot/agent/chat.py
 pycoze/bot/agent/agent_types/__init__.py
 pycoze/bot/agent/agent_types/const.py
 pycoze/bot/agent/agent_types/openai_func_call_agent.py

{pycoze-0.1.156 → pycoze-0.1.157}/setup.py

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