pycoze 0.1.33__py3-none-any.whl → 0.1.35__py3-none-any.whl

pycoze/bot/agent/chat.py

@@ -1,28 +1,22 @@
-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}))
+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/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
 
 
@@ -48,20 +48,12 @@ def agent_chat(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")
+        message = json.loads(input_text[len(INPUT_MESSAGE) :])
+        history.append(HumanMessage(message["content"]))
         result = agent_chat(bot_setting_file, history)
-        output("assistant", result, history)
+        history.append(output("assistant", result))

{pycoze-0.1.33.dist-info → pycoze-0.1.35.dist-info}/METADATA

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

{pycoze-0.1.33.dist-info → pycoze-0.1.35.dist-info}/RECORD

@@ -2,11 +2,11 @@ pycoze/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/module.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/bot/__init__.py,sha256=pciDtfcIXda7iFt9uI5Fpm0JKpGBhdXHmJv4966WTVU,21
 pycoze/bot/base.py,sha256=GWYDVGGtiCpk6gv-163cAbDid_IsnMe5jTj7eZUMJQU,2679
-pycoze/bot/bot.py,sha256=y-VRF8F75tdoYDPIkSca1jcQYgWJIw-_67koSGgo68g,2253
+pycoze/bot/bot.py,sha256=45HHv9PmUDHfP0pg5zZi23Io7bCZetJSneTZfFaSjow,1838
 pycoze/bot/agent/__init__.py,sha256=IaYqQCJ3uBor92JdOxI_EY4HtYOHgej8lijr3UrN1Vc,161
 pycoze/bot/agent/agent.py,sha256=vDbTCorUL6Eh2Az4uzwGsLa3Hp4EPcOkq62JpQXt8-s,3435
 pycoze/bot/agent/assistant.py,sha256=QLeWaPi415P9jruYOm8qcIbC94cXXAhJYmLTkyC9NTQ,1267
-pycoze/bot/agent/chat.py,sha256=C2X0meUcIPbn5FCxvhkhxozldPG7qdb2jVR-WnPqqnQ,791
+pycoze/bot/agent/chat.py,sha256=kc0qgcrBSXdiMy49JwThZTV-0PAvzAhiUvbI5ILiSnU,571
 pycoze/bot/agent/agent_types/__init__.py,sha256=W2jTNMLqUMqgCMG0Tw0d8n7WpsbsnIonqaPR-YLegLU,210
 pycoze/bot/agent/agent_types/openai_func_call_agent.py,sha256=LBCkcPbFHpI9ijrze2X15O-9tXxCbi7IKbJRtrBQjwg,4504
 pycoze/bot/agent/agent_types/react_agent.py,sha256=AnjHwHXVwLAm77ndglJGi4rQhqDGWaLuUfl46uZVSzM,6749
@@ -21,8 +21,8 @@ pycoze/ui/ui_def.py,sha256=CNFYH8NC-WYmbceIPpxsRr9H6O006pMKukx7U-BOE1Q,3744
 pycoze/utils/__init__.py,sha256=KExBkotf23dr2NfTEouWke5nJB1q2IuDXgHrmuyd95k,73
 pycoze/utils/arg.py,sha256=rRujm1zKc0XlnNlpIJ6JAAaFiTzDGmL_RliIpSc5OD8,724
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.33.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.33.dist-info/METADATA,sha256=eZ3PUpQoFCtWtNaVjXlJdhNo1u6PKoA9LVdaWhHtzYE,719
-pycoze-0.1.33.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
-pycoze-0.1.33.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.33.dist-info/RECORD,,
+pycoze-0.1.35.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.35.dist-info/METADATA,sha256=zNyvESPcN0ZDZrIb8r_b50-l8FgGWVSJ33p5WeshqW8,719
+pycoze-0.1.35.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
+pycoze-0.1.35.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.35.dist-info/RECORD,,