pycoze 0.1.39__tar.gz → 0.1.41__tar.gz

{pycoze-0.1.39 → pycoze-0.1.41}/PKG-INFO

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

pycoze-0.1.41/pycoze/access/tool_for_bot.py

@@ -0,0 +1,75 @@
+import sys
+import os
+import importlib
+from langchain.agents import tool as _tool
+import types
+import langchain_core
+
+
+def wrapped_tool(tool, module_path):
+    old_tool_fun = tool.func
+
+    def _wrapped_tool(*args, **kwargs):
+        print(f"调用了{tool.name}")
+        old_path = os.getcwd()
+        try:
+            sys.path.insert(0, module_path)  # 插入到第一个位置
+            os.chdir(module_path)
+            result = old_tool_fun(*args, **kwargs)
+        finally:
+            sys.path.remove(module_path)
+            os.chdir(old_path)
+        print(f"{tool.name}调用完毕,结果为:", result)
+        return result
+
+    return _wrapped_tool
+
+
+def import_tools(tool_id):
+    tool_path = "../../tool"
+    old_path = os.getcwd()
+    module_path = os.path.join(tool_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 []
+
+    # 保存当前的 sys.modules 状态
+    original_modules = sys.modules.copy()
+
+    try:
+        sys.path.insert(0, module_path)  # 插入到第一个位置
+        os.chdir(module_path)
+        module = importlib.import_module("tool")
+        export_tools = getattr(module, "export_tools")
+        temp_list = []
+        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 isinstance(tool, types.FunctionType) and not isinstance(
+                tool, langchain_core.tools.StructuredTool
+            ):
+                temp_list.append(_tool(tool))
+        export_tools = temp_list
+
+    except Exception as e:
+        print(f"Error loading tool {tool_id}: {e}")
+        sys.path.remove(module_path)
+        os.chdir(old_path)
+        return []
+
+    # 卸载模块并恢复 sys.modules 状态
+    importlib.invalidate_caches()
+    for key in list(sys.modules.keys()):
+        if key not in original_modules:
+            del sys.modules[key]
+
+    sys.path.remove(module_path)
+    os.chdir(old_path)
+
+    for tool in export_tools:
+        tool.func = wrapped_tool(tool, module_path)
+
+    return export_tools

{pycoze-0.1.39 → pycoze-0.1.41}/pycoze/bot/bot.py

@@ -4,7 +4,7 @@ from .agent import run_agent, Runnable, INPUT_MESSAGE, output
 import asyncio
 from langchain_core.messages import HumanMessage
 from pycoze import utils
-from pycoze.access.
+from pycoze.access.tool_for_bot import import_tools
 
 params = utils.arg.read_params()
 llm_file = params["appPath"] + "/JsonStorage/llm.json"

{pycoze-0.1.39 → pycoze-0.1.41}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.39 → pycoze-0.1.41}/pycoze.egg-info/SOURCES.txt

@@ -7,6 +7,8 @@ pycoze.egg-info/PKG-INFO
 pycoze.egg-info/SOURCES.txt
 pycoze.egg-info/dependency_links.txt
 pycoze.egg-info/top_level.txt
+pycoze/access/__init__.py
+pycoze/access/tool_for_bot.py
 pycoze/bot/__init__.py
 pycoze/bot/bot.py
 pycoze/bot/agent/__init__.py

{pycoze-0.1.39 → pycoze-0.1.41}/setup.py

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