pycoze 0.1.192__py3-none-any.whl → 0.1.194__py3-none-any.whl

pycoze/reference/bot.py

@@ -12,9 +12,11 @@ bot_index  = 0
 
 params = utils.read_params_file()
 
-def ref_bot(bot_id, as_agent_tool=False):
+def ref_bot(bot_id, as_agent_tool=False, workspace_path=None):
     global bot_index
-    tool_base_path = os.path.join(params["workspacePath"], "User/Local/bot")
+    if workspace_path is None:
+        workspace_path = params["workspacePath"]
+    tool_base_path = os.path.join(workspace_path, "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

@@ -10,8 +10,10 @@ 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"], "User/Local/tool")
+def ref_tools(tool_id, as_agent_tool=False, workspace_path=None):
+    if workspace_path is None:
+        workspace_path = params["workspacePath"]
+    tool_base_path = os.path.join(workspace_path, "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,56 @@
+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, workspace_path=None):
+    if workspace_path is None:
+        workspace_path = params["workspacePath"]
+    tool_base_path = os.path.join(workspace_path, "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.dist-info → pycoze-0.1.194.dist-info}/METADATA

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

{pycoze-0.1.192.dist-info → pycoze-0.1.194.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=oSxlrqS0MWUwxNlTx-ZKI1p5Udj44IhrFZaPLiTMsw8,2333
+pycoze/reference/bot.py,sha256=BDflTV3zYoZqWnJpD5lMM_1vU_5b20M3XiFt1p-RHWM,2427
 pycoze/reference/lib.py,sha256=0xQJTLTHedGzQBsjuTFNBVqYc4-8Yl65gGCrAhWyOX8,2155
-pycoze/reference/tool.py,sha256=gLffWRz6tXPiWjewplnky-yxXya_fGkx132PRYZmxpw,1630
+pycoze/reference/tool.py,sha256=9OxHawxjkh6S8eAOlGNrbfhkk5oRlRc7LBpnV6RmhYI,1722
+pycoze/reference/workflow.py,sha256=rjjVrwk_Dym3u4bb461tKyETNiXi58DLwSSd5yAB3kg,2017
 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.192.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.192.dist-info/METADATA,sha256=QKCQpr1BfWEt_Lp737155V1XaTQxm-nnAaXrTg49L5c,726
-pycoze-0.1.192.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-pycoze-0.1.192.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.192.dist-info/RECORD,,
+pycoze-0.1.194.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.194.dist-info/METADATA,sha256=9GXgPZYP28cEGIPuSGa55XX2rHrJHue1qjcNwv94nrc,726
+pycoze-0.1.194.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+pycoze-0.1.194.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.194.dist-info/RECORD,,