pycoze 0.1.146__tar.gz → 0.1.148__tar.gz

{pycoze-0.1.146 → pycoze-0.1.148}/PKG-INFO

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

{pycoze-0.1.146 → pycoze-0.1.148}/pycoze/bot/bot.py

@@ -25,11 +25,11 @@ def load_abilities(bot_setting_file: str):
 
     abilities = []
     for bot_id in role_setting["bots"]:
-        bot = ref_bot(bot_id)
+        bot = ref_bot(bot_id, as_agent_tool=True)
         if bot:
             abilities.append(bot)
     for tool_id in role_setting["tools"]:
-        abilities.extend(ref_tools(tool_id))
+        abilities.extend(ref_tools(tool_id, as_agent_tool=True))
     return abilities
 
 

{pycoze-0.1.146 → pycoze-0.1.148}/pycoze/reference/bot.py

@@ -9,7 +9,7 @@ import json
 
 bot_index  = 0
 
-def ref_bot(bot_id):
+def ref_bot(bot_id, as_agent_tool=False):
     global bot_index
     tool_base_path = "../../bot"
     module_path = os.path.join(tool_base_path, bot_id)
@@ -53,7 +53,7 @@ def {random_name}(command:str) -> str:
             tool.func = wrapped_func(tool, module_path)
             if tool.description is None:
                 tool.description = "This tool is used to " + tool.name + "."
-            return tool
+            return tool if as_agent_tool else tool.func
     except Exception as e:
         print(f"Error loading bot {bot_id}: {e}")
         return None

{pycoze-0.1.146 → pycoze-0.1.148}/pycoze/reference/lib.py

@@ -1,6 +1,7 @@
 import os
 import sys
 import importlib
+import types
 
 
 class ChangeDirectoryAndPath:
@@ -58,7 +59,7 @@ def wrapped_func(item, module_path):
         with ChangeDirectoryAndPath(module_path):
             result = original_tool_function(*args, **kwargs)
         try:
-            print(f"{item.name} 调用完毕,结果为:", result)
+            print(f"{item.name} 调用完毕,结果:", result, type result, isinstance(result, types.GeneratorType))
         except:
             pass
         return result

{pycoze-0.1.146 → pycoze-0.1.148}/pycoze/reference/tool.py

@@ -7,7 +7,7 @@ import langchain_core
 from .lib import ChangeDirectoryAndPath, ModuleManager, wrapped_func
 
 
-def ref_tools(tool_id):
+def ref_tools(tool_id, as_agent_tool=False):
     tool_base_path = "../../tool"
     module_path = os.path.join(tool_base_path, tool_id)
     module_path = os.path.normpath(os.path.abspath(module_path))
@@ -39,4 +39,4 @@ def ref_tools(tool_id):
         if tool.description is None:
             tool.description = "This tool is used to " + tool.name + "."
 
-    return export_tools
+    return export_tools if as_agent_tool else [tool.func for tool in export_tools]

{pycoze-0.1.146 → pycoze-0.1.148}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.146 → pycoze-0.1.148}/setup.py

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