pycoze 0.1.192__tar.gz → 0.1.193__tar.gz

{pycoze-0.1.192 → pycoze-0.1.193}/PKG-INFO

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

pycoze-0.1.193/pycoze/reference/workflow.py

@@ -0,0 +1,54 @@
+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_func
+from pycoze import utils
+
+
+params = utils.read_params_file()
+
+def _ref_workflows(workflow_id, as_agent_tool=False):
+    tool_base_path = os.path.join(params["workspacePath"], "User/Local/workflow")
+    module_path = os.path.join(tool_base_path, workflow_id)
+    module_path = os.path.normpath(os.path.abspath(module_path))
+
+    if not os.path.exists(module_path):
+        print(f"Workflow {workflow_id} not found")
+        return []
+
+    try:
+        with ModuleManager(module_path) as manager:
+            module = importlib.import_module("workflow")
+            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 workflow {workflow_id}: {e}")
+        return []
+
+    for tool in export_tools:
+        tool.func = wrapped_func(tool, module_path)
+        if tool.description is None:
+            tool.description = "This tool is used to " + tool.name + "."
+
+    return export_tools if as_agent_tool else [tool.func for tool in export_tools]
+
+
+def ref_workflows(workflow_id, as_agent_tool=False):
+    tools = _ref_workflows(workflow_id, as_agent_tool=as_agent_tool)
+    if len(tools) > 0:
+        return tools[0]
+    else:
+        return None
+

{pycoze-0.1.192 → pycoze-0.1.193}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.192 → pycoze-0.1.193}/pycoze.egg-info/SOURCES.txt

@@ -25,6 +25,7 @@ pycoze/reference/__init__.py
 pycoze/reference/bot.py
 pycoze/reference/lib.py
 pycoze/reference/tool.py
+pycoze/reference/workflow.py
 pycoze/ui/__init__.py
 pycoze/ui/base.py
 pycoze/ui/color.py

{pycoze-0.1.192 → pycoze-0.1.193}/setup.py

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