pycoze 0.1.32__tar.gz → 0.1.34__tar.gz

{pycoze-0.1.32 → pycoze-0.1.34}/PKG-INFO

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

pycoze-0.1.34/pycoze/bot/agent/chat.py

@@ -0,0 +1,22 @@
+import json
+from langchain_core.messages import AIMessage
+
+
+INPUT_MESSAGE = "INPUT_MESSAGE=>"
+_OUTPUT_MESSAGE = "OUTPUT_MESSAGE=>"
+_INFOMATION_MESSAGE = "INFOMATION_MESSAGE=>"
+_LOG = "LOG=>"
+
+
+def log(content, *args, end="\n", **kwargs):
+    print(_LOG + content, *args, end=end, **kwargs)
+
+
+def output(role, content):
+    assert role == "assistant"
+    print(_OUTPUT_MESSAGE + json.dumps({"role": role, "content": content}))
+    return AIMessage(content=content)
+
+
+def info(role, content):
+    print(_INFOMATION_MESSAGE + json.dumps({"role": role, "content": content}))

{pycoze-0.1.32 → pycoze-0.1.34}/pycoze/bot/bot.py

@@ -3,7 +3,7 @@ from langchain_openai import ChatOpenAI
 from .base import import_tools
 from .agent import run_agent, Runnable, INPUT_MESSAGE, output
 import asyncio
-from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
+from langchain_core.messages import HumanMessage
 from pycoze import utils
 
 
@@ -26,7 +26,7 @@ def load_tools(bot_setting_file: str):
     return tools
 
 
-def run_agent(bot_setting_file, history):
+def agent_chat(bot_setting_file, history):
     role_setting = load_role_setting(bot_setting_file)
     tools = load_tools(bot_setting_file)
     with open(llm_file, "r", encoding="utf-8") as f:
@@ -48,20 +48,12 @@ def run_agent(bot_setting_file, history):
 
 
 def chat(bot_setting_file: str):
+    history = []
     while True:
         input_text = input()
         if not input_text.startswith(INPUT_MESSAGE):
             raise ValueError("Invalid message")
-        messages = json.loads(input_text[len(INPUT_MESSAGE) :])
-        history = []
-        for message in messages:
-            if message["role"] == "assistant":
-                history += [AIMessage(content=message["content"])]
-            elif message["role"] == "user":
-                history += [HumanMessage(content=message["content"])]
-            elif message["role"] == "system":
-                history += [SystemMessage(content=message["content"])]
-            else:
-                raise ValueError("Invalid message")
-        result = run_agent(bot_setting_file, history)
-        output("assistant", result, history)
+        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, history))

{pycoze-0.1.32 → pycoze-0.1.34}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.32 → pycoze-0.1.34}/setup.py

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

pycoze-0.1.32/pycoze/bot/agent/chat.py

@@ -1,28 +0,0 @@
-import json
-from langchain_core.messages import HumanMessage, AIMessage
-
-
-INPUT_MESSAGE = "INPUT_MESSAGE=>"
-_OUTPUT_MESSAGE = "OUTPUT_MESSAGE=>"
-_INFOMATION_MESSAGE = "INFOMATION_MESSAGE=>"
-_LOG = "LOG=>"
-
-
-
-def log(content, *args, end='\n', **kwargs):
-    print(_LOG + content, *args, end=end, **kwargs)
-
-
-def output(role, content, history):
-    print(_OUTPUT_MESSAGE + json.dumps({"role": role, "content": content}))
-    if role == "assistant":
-        history.append(AIMessage(content=content))
-        
-    elif role == "user":
-        history.append(HumanMessage(content=content))
-    else:
-        raise ValueError("Invalid role")
-    return history
-
-def info(role, content):
-     print(_INFOMATION_MESSAGE + json.dumps({"role": role, "content": content}))