pycoze 0.1.31__py3-none-any.whl → 0.1.32__py3-none-any.whl

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.dist-info → pycoze-0.1.32.dist-info}/METADATA

@@ -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.dist-info → pycoze-0.1.32.dist-info}/RECORD

@@ -2,7 +2,7 @@ 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=c8xje38_BMf4RgkWNDDuGBelHHszosMdYZevGD8dk-c,2225
+pycoze/bot/bot.py,sha256=mus0g9OMsH0VKOhhk3xmwIxhPkb92xfRsyce0BumQao,2251
 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
@@ -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.31.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.31.dist-info/METADATA,sha256=uUXKqs4ukQy4dj2MHEabHQPWu8IdK-sckxW0R4GV5kY,719
-pycoze-0.1.31.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
-pycoze-0.1.31.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.31.dist-info/RECORD,,
+pycoze-0.1.32.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.32.dist-info/METADATA,sha256=n98a0u3ipHtexw7qRCkCDJG5r8rCG_OKLphc-t5J4VY,719
+pycoze-0.1.32.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
+pycoze-0.1.32.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.32.dist-info/RECORD,,