pycoze 0.1.98__py3-none-any.whl → 0.1.99__py3-none-any.whl

pycoze/access/bot_for_bot.py

@@ -0,0 +1,40 @@
+import sys
+import os
+import importlib
+from langchain.agents import tool as to_agent_tool
+import types
+import langchain_core
+from .lib import ChangeDirectoryAndPath, ModuleManager, wrapped_fun
+
+
+def import_tools(tool_id):
+    tool_base_path = "../../tool"
+    module_path = os.path.join(tool_base_path, tool_id)
+    module_path = os.path.normpath(os.path.abspath(module_path))
+
+    if not os.path.exists(module_path):
+        print(f"Tool {tool_id} not found")
+        return []
+
+    try:
+        with ModuleManager(module_path) as manager:
+            module = importlib.import_module("tool")
+            export_tools = getattr(module, "export_tools")
+            valid_tools = []
+            for tool in export_tools:
+                assert isinstance(tool, langchain_core.tools.StructuredTool) or isinstance(
+                    tool, types.FunctionType
+                ), f"Tool is not a StructuredTool or function: {tool}"
+                if not isinstance(tool, langchain_core.tools.StructuredTool):
+                    tool = to_agent_tool(tool)
+                valid_tools.append(tool)
+            export_tools = valid_tools
+
+    except Exception as e:
+        print(f"Error loading tool {tool_id}: {e}")
+        return []
+
+    for tool in export_tools:
+        tool.func = wrapped_fun(tool, module_path)
+
+    return export_tools

pycoze/access/lib.py

@@ -44,4 +44,17 @@ class ModuleManager:
         importlib.invalidate_caches()
         for key in list(sys.modules.keys()):
             if key not in self.original_modules:
-                del sys.modules[key]
+                del sys.modules[key]
+                
+
+def wrapped_fun(item, module_path):
+    original_tool_function = item.func
+
+    def _wrapped(*args, **kwargs):
+        print(f"调用了 {item.name}")
+        with ChangeDirectoryAndPath(module_path):
+            result = original_tool_function(*args, **kwargs)
+        print(f"{item.name} 调用完毕,结果为:", result)
+        return result
+
+    return _wrapped

pycoze/access/tool_for_bot.py

@@ -4,26 +4,10 @@ import importlib
 from langchain.agents import tool as to_agent_tool
 import types
 import langchain_core
-from .lib import ChangeDirectoryAndPath, ModuleManager
-
-
-
-def wrapped_tool(tool, module_path):
-    """Wrap the tool function to include additional logging and path management."""
-    original_tool_function = tool.func
-
-    def _wrapped_tool(*args, **kwargs):
-        print(f"调用了{tool.name}")
-        with ChangeDirectoryAndPath(module_path):
-            result = original_tool_function(*args, **kwargs)
-        print(f"{tool.name}调用完毕,结果为:", result)
-        return result
-
-    return _wrapped_tool
+from .lib import ChangeDirectoryAndPath, ModuleManager, wrapped_fun
 
 
 def import_tools(tool_id):
-    """Import tools from a specified tool_id."""
     tool_base_path = "../../tool"
     module_path = os.path.join(tool_base_path, tool_id)
     module_path = os.path.normpath(os.path.abspath(module_path))
@@ -51,6 +35,6 @@ def import_tools(tool_id):
         return []
 
     for tool in export_tools:
-        tool.func = wrapped_tool(tool, module_path)
+        tool.func = wrapped_fun(tool, module_path)
 
     return export_tools

{pycoze-0.1.98.dist-info → pycoze-0.1.99.dist-info}/METADATA

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

{pycoze-0.1.98.dist-info → pycoze-0.1.99.dist-info}/RECORD

@@ -1,8 +1,9 @@
 pycoze/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/module.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/access/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycoze/access/lib.py,sha256=yeWWFauplfv9AVWbNXL4hAGqgdTXygqs42a23peDN6M,1514
-pycoze/access/tool_for_bot.py,sha256=iRBqtxku9BGng8AyWIscipLrMrVyvpB0RDGp_qQ0OGU,1837
+pycoze/access/bot_for_bot.py,sha256=p7BvQk_y4Hh8L9Et39EUnURpy4enRUvsMlBLNfYFb8o,1377
+pycoze/access/lib.py,sha256=RTQvEee1kfdbap-tzrcQbh-mZg_Y5ydmak7RRB7EkJ8,1912
+pycoze/access/tool_for_bot.py,sha256=F1QrGpGbJ3ucsRYWiJw-My4XiE8QuVb8oj7p9ZMmZXE,1337
 pycoze/ai/__init__.py,sha256=odM2lgYSnApxw4AzLV_5AyaxL5_MLfydOSB1VmLGFyA,75
 pycoze/ai/vram_reserve.py,sha256=QbqaA8qv87cnEpOVDMygi0BNMxuhLYwj1UKfR_D5BD4,4340
 pycoze/bot/__init__.py,sha256=6HHMxDQVOyZM9dtSjQm9tjGnhj4h7CixD0JOvEwTi48,41
@@ -21,8 +22,8 @@ pycoze/ui/ui_def.py,sha256=UhhU_yB3GV9ISbvTWT48hsHPHI250BhMILh6bu5Uioo,4206
 pycoze/utils/__init__.py,sha256=TNJhFfY7JYdLlzuP9GvgxfNXUtbgH_NUUJSqHXCxJn4,78
 pycoze/utils/arg.py,sha256=kA3KBQzXc2WlH5XbF8kfikfpqljiKaW7oY_GE4Qyffc,753
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.98.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.98.dist-info/METADATA,sha256=jL1CNsIYUUczQoslbVCvFMVBw6In5Bc8XUYX-RPCkDk,725
-pycoze-0.1.98.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-pycoze-0.1.98.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.98.dist-info/RECORD,,
+pycoze-0.1.99.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.99.dist-info/METADATA,sha256=3HaGjTXmeXlkFAK-1coDK6dpZreMIBRXpQlvPyhOW3U,725
+pycoze-0.1.99.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+pycoze-0.1.99.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.99.dist-info/RECORD,,