pycoze 0.1.31__tar.gz → 0.1.32__tar.gz

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

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

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

@@ -26,31 +26,33 @@ def load_tools(bot_setting_file: str):
     return tools
 
 
-def chat(bot_setting_file: str):
+def run_agent(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:
+        cfg = json.load(f)
+        chat = ChatOpenAI(
+            api_key=cfg["apiKey"],
+            base_url=cfg["baseURL"],
+            model=cfg["model"],
+            temperature=role_setting["temperature"],
+        )
+
+    agent = Runnable(
+        agent_execution_mode="FuncCall",  # 'FuncCall' or 'ReAct'，大模型支持FuncCall的话就用FuncCall
+        tools=tools,
+        llm=chat,
+        assistant_message=role_setting["prompt"],
+    )
+    return asyncio.run(run_agent(agent, history))
+
 
+def chat(bot_setting_file: str):
     while True:
         input_text = input()
-        role_setting = load_role_setting(bot_setting_file)
-        tools = load_tools(bot_setting_file)
         if not input_text.startswith(INPUT_MESSAGE):
             raise ValueError("Invalid message")
         messages = json.loads(input_text[len(INPUT_MESSAGE) :])
-
-        with open(llm_file, "r", encoding="utf-8") as f:
-            cfg = json.load(f)
-            chat = ChatOpenAI(
-                api_key=cfg["apiKey"],
-                base_url=cfg["baseURL"],
-                model=cfg["model"],
-                temperature=role_setting["temperature"],
-            )
-
-        agent = Runnable(
-            agent_execution_mode="FuncCall",  # 'FuncCall' or 'ReAct'，大模型支持FuncCall的话就用FuncCall
-            tools=tools,
-            llm=chat,
-            assistant_message=role_setting["prompt"],
-        )
         history = []
         for message in messages:
             if message["role"] == "assistant":
@@ -61,5 +63,5 @@ def chat(bot_setting_file: str):
                 history += [SystemMessage(content=message["content"])]
             else:
                 raise ValueError("Invalid message")
-        result = asyncio.run(run_agent(agent, history))
+        result = run_agent(bot_setting_file, history)
         output("assistant", result, history)

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

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

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

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