pycoze 0.1.191__py3-none-any.whl → 0.1.193__py3-none-any.whl

pycoze/reference/bot.py

@@ -14,7 +14,7 @@ params = utils.read_params_file()
 
 def ref_bot(bot_id, as_agent_tool=False):
     global bot_index
-    tool_base_path = os.path.join(params["workspacePath"], "bot")
+    tool_base_path = os.path.join(params["workspacePath"], "User/Local/bot")
     module_path = os.path.join(tool_base_path, bot_id)
     module_path = os.path.normpath(os.path.abspath(module_path))
     

pycoze/reference/tool.py

@@ -11,7 +11,7 @@ from pycoze import utils
 params = utils.read_params_file()
 
 def ref_tools(tool_id, as_agent_tool=False):
-    tool_base_path = os.path.join(params["workspacePath"], "bot")
+    tool_base_path = os.path.join(params["workspacePath"], "User/Local/tool")
     module_path = os.path.join(tool_base_path, tool_id)
     module_path = os.path.normpath(os.path.abspath(module_path))
 

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

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

{pycoze-0.1.191.dist-info → pycoze-0.1.193.dist-info}/RECORD

@@ -15,9 +15,10 @@ pycoze/bot/agent/agent_types/__init__.py,sha256=zmU2Kmrv5mCdfg-QlPn2H6pWxbGeq8s7
 pycoze/bot/agent/agent_types/const.py,sha256=BfUKPrhAHREoMLHuFNG2bCIEkC1-f7K0LEqNg4RwiRE,70
 pycoze/bot/agent/agent_types/openai_func_call_agent.py,sha256=SnEm5MODHn2uMsaMNqgzULM_91vqLHC0TU6ovwCOqLU,6675
 pycoze/reference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycoze/reference/bot.py,sha256=42p4fXVLbktf8EjsgxiMdiPFSFv42jAADnG1-yHq1ws,2322
+pycoze/reference/bot.py,sha256=oSxlrqS0MWUwxNlTx-ZKI1p5Udj44IhrFZaPLiTMsw8,2333
 pycoze/reference/lib.py,sha256=0xQJTLTHedGzQBsjuTFNBVqYc4-8Yl65gGCrAhWyOX8,2155
-pycoze/reference/tool.py,sha256=OmrhRyi0yfvYpqoP-X8HEUEzL7UpmLmomCq6SsUjcSI,1618
+pycoze/reference/tool.py,sha256=gLffWRz6tXPiWjewplnky-yxXya_fGkx132PRYZmxpw,1630
+pycoze/reference/workflow.py,sha256=22UJPdnqK1RvyGE4fxkvQqDxVM6YbayLmbuZK3bpbrA,1923
 pycoze/ui/__init__.py,sha256=7xAfL2lfG7-jllPJEZUJO89xUE9sNzvo1y0WmBswjBI,458
 pycoze/ui/base.py,sha256=sbBZGMUtlosWHQJpxMULa1bGByeSlcldtE9QXNyiJmM,1093
 pycoze/ui/color.py,sha256=cT9Ib8uNzkOKxyW0IwVj46o4LwdB1xgNCj1_Rou9d_4,854
@@ -27,8 +28,8 @@ pycoze/utils/__init__.py,sha256=Gi5EnrWZGMD2JRejgV4c_VLCXyvA2wwBFI_niDF5MUE,110
 pycoze/utils/arg.py,sha256=GtfGbMTMdaK75Fwh6MpUe1pCA5X6Ep4LFG7a72YrzjI,525
 pycoze/utils/env.py,sha256=W04lhvTHhAAC6EldP6kk2xrctqtu8K6kl1vDLZDNeh8,561
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.191.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.191.dist-info/METADATA,sha256=iDKXf3S477IYK68SPgLT4pLVoUNDp07GFT7FwJz7_fs,726
-pycoze-0.1.191.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-pycoze-0.1.191.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.191.dist-info/RECORD,,
+pycoze-0.1.193.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.193.dist-info/METADATA,sha256=FeRBlJCfv9JYdmtXlVpQaSMvp4dQK2kBKWKFtXn0QCg,726
+pycoze-0.1.193.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+pycoze-0.1.193.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.193.dist-info/RECORD,,