pycoze 0.1.284__py3-none-any.whl → 0.1.286__py3-none-any.whl

pycoze/bot/agent/agent_types/openai_func_call_agent.py

@@ -1,6 +1,5 @@
 # reference：https://github.com/maxtheman/opengpts/blob/d3425b1ba80aec48953a327ecd9a61b80efb0e69/backend/app/agent_types/openai_agent.py
 import json
-
 from langchain.tools import BaseTool
 from langchain_core.utils.function_calling import convert_to_openai_tool
 from langchain_core.language_models.base import LanguageModelLike
@@ -13,6 +12,7 @@ import json
 import random
 from .const import HumanToolString
 
+
 def get_all_markdown_json(content):
     # Find all markdown json blocks
     markdown_json_blocks = re.findall(r"```json(.*?)```", content, re.DOTALL)
@@ -107,8 +107,8 @@ def create_openai_func_call_agent_executor(
             last_message.content = last_message.content + "\n\n"  # 避免影响阅读
         tools = get_tools(last_message)
         if last_message.tool_calls or tools:
-            return 'continue'
-        return 'end'
+            return "continue"
+        return "end"
 
     # Define the function to execute tools
     async def call_tool(messages):
@@ -140,9 +140,7 @@ def create_openai_func_call_agent_executor(
         responses = await tool_executor.abatch(actions, **kwargs)
         # We use the response to create a ToolMessage
         tool_messages = []
-        for tool_call, response in zip(
-            get_tools(last_message), responses
-        ):
+        for tool_call, response in zip(get_tools(last_message), responses):
             if not isinstance(response, (str, int, float, bool, list, tuple)):
                 response = repr(
                     response
@@ -157,9 +155,7 @@ def create_openai_func_call_agent_executor(
         if tool_compatibility_mode:
             # HumanMessage
             tool_msgs_str = repr(tool_messages)
-            tool_messages = [
-                HumanMessage(content=HumanToolString + tool_msgs_str)
-            ]
+            tool_messages = [HumanMessage(content=HumanToolString + tool_msgs_str)]
         return tool_messages
 
     workflow = MessageGraph()

pycoze/bot/agent/chat.py

@@ -1,5 +1,4 @@
 import json
-from langchain_core.messages import AIMessage
 
 
 INPUT_MESSAGE = "INPUT_MESSAGE=>"
@@ -9,7 +8,7 @@ _INFOMATION_MESSAGE = "INFOMATION_MESSAGE=>"
 _LOG = "LOG=>"
 
 
-CHAT_DATA = {"output":"", "info": ""}
+CHAT_DATA = {"output": "", "info": ""}
 
 
 def log(content, *args, end="\n", **kwargs):
@@ -27,6 +26,7 @@ def output(role, content):
     CHAT_DATA["output"] = content
     print(_OUTPUT_MESSAGE + json.dumps({"role": role, "content": content}))
 
+
 def info(role, content):
     CHAT_DATA["info"] += content
     print(_INFOMATION_MESSAGE + json.dumps({"role": role, "content": content}))

pycoze/reference/bot.py

@@ -1,64 +1,61 @@
-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
-import json
-from pycoze import utils
-
-bot_index  = 0
-
-params = utils.params
-
-def ref_bot(bot_id, as_agent_tool=False, workspace_path=None):
-    global bot_index
-    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))
-    
-    if not os.path.exists(module_path):
-        print(f"Bot {bot_id} not found in:"+ module_path)
-        return None
-    
-    try:
-        with ModuleManager(module_path) as manager:
-            info = module_path + "/info.json"
-            with open(info, "r", encoding="utf-8") as f:
-                info = json.load(f)
-            name = info["name"]
-            random_name = "bot_" + str(bot_index)
-            bot_index += 1
-            function_code = f"""
-def {random_name}(command:str) -> str:
-    \"\"\"接收任意指令字符串，并返回AI角色（{name}专家）深入思考和执行指令后的字符串结果。
-    AI角色（{name}专家）擅长使用各种工具，并会给出专业且更为准确的结果。
-
-    Args:
-        command (str): AI角色（{name}专家）需要执行的指令字符串。
-
-    Returns:
-        str: AI角色（{name}专家）执行指令后的结果。
-    \"\"\"
-    from pycoze import bot
-    from pycoze import utils
-    import tempfile
-    
-    with tempfile.NamedTemporaryFile(delete=True, mode='w+t') as temp_file:
-        sys.stdout = temp_file  # 将输出重定向到临时文件，防止影响AI结果
-        result = bot.get_chat_response("botSetting.json", command)
-        sys.stdout = sys.__stdout__  # 恢复标准输出
-    return result
-"""
-            exec(function_code)
-            tool = to_agent_tool(eval(random_name))
-            tool.func = wrapped_func(tool, module_path)
-            if tool.description is None:
-                tool.description = "This tool is used to " + tool.name + "."
-            return tool if as_agent_tool else tool.func
-    except Exception as e:
-        print(f"Error loading bot {bot_id}: {e}")
-        return None
+import os
+from langchain.agents import tool as to_agent_tool
+from .lib import ModuleManager, wrapped_func
+import json
+from pycoze import utils
+
+bot_index = 0
+
+params = utils.params
+
+
+def ref_bot(bot_id, as_agent_tool=False, workspace_path=None):
+    global bot_index
+    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))
+
+    if not os.path.exists(module_path):
+        print(f"Bot {bot_id} not found in:" + module_path)
+        return None
+
+    try:
+        with ModuleManager(module_path) as manager:
+            info = module_path + "/info.json"
+            with open(info, "r", encoding="utf-8") as f:
+                info = json.load(f)
+            name = info["name"]
+            random_name = "bot_" + str(bot_index)
+            bot_index += 1
+            function_code = f"""
+def {random_name}(command:str) -> str:
+    \"\"\"接收任意指令字符串，并返回AI角色（{name}专家）深入思考和执行指令后的字符串结果。
+    AI角色（{name}专家）擅长使用各种工具，并会给出专业且更为准确的结果。
+
+    Args:
+        command (str): AI角色（{name}专家）需要执行的指令字符串。
+
+    Returns:
+        str: AI角色（{name}专家）执行指令后的结果。
+    \"\"\"
+    from pycoze import bot
+    from pycoze import utils
+    import tempfile
+
+    with tempfile.NamedTemporaryFile(delete=True, mode='w+t') as temp_file:
+        sys.stdout = temp_file  # 将输出重定向到临时文件，防止影响AI结果
+        result = bot.get_chat_response("botSetting.json", command)
+        sys.stdout = sys.__stdout__  # 恢复标准输出
+    return result
+"""
+            exec(function_code)
+            tool = to_agent_tool(eval(random_name))
+            tool.func = wrapped_func(tool, module_path)
+            if tool.description is None:
+                tool.description = "This tool is used to " + tool.name + "."
+            return tool if as_agent_tool else tool.func
+    except Exception as e:
+        print(f"Error loading bot {bot_id}: {e}")
+        return None

pycoze/reference/tool.py

@@ -1,15 +1,15 @@
-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 .lib import ModuleManager, wrapped_func
 from pycoze import utils
 
 
 params = utils.params
 
+
 def ref_tools(tool_id, as_agent_tool=False, workspace_path=None):
     if workspace_path is None:
         workspace_path = params["workspacePath"]
@@ -27,7 +27,9 @@ def ref_tools(tool_id, as_agent_tool=False, workspace_path=None):
             export_tools = getattr(module, "export_tools")
             valid_tools = []
             for tool in export_tools:
-                assert isinstance(tool, langchain_core.tools.StructuredTool) or isinstance(
+                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):

pycoze/reference/workflow.py

@@ -1,56 +1,59 @@
-import sys
-import os
-import importlib
-from langchain.agents import tool as to_agent_tool
-import types
-import langchain_core
-from .lib import ModuleManager, wrapped_func
-from pycoze import utils
-
-
-params = utils.params
-
-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("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 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_workflow(workflow_id, as_agent_tool=False, workspace_path=None):
-    tools = _ref_workflows(workflow_id, as_agent_tool=as_agent_tool, workspace_path=workspace_path)
-    if len(tools) > 0:
-        return tools[0]
-    else:
-        return None
-
+import os
+import importlib
+from langchain.agents import tool as to_agent_tool
+import types
+import langchain_core
+from .lib import ModuleManager, wrapped_func
+from pycoze import utils
+
+
+params = utils.params
+
+
+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("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 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_workflow(workflow_id, as_agent_tool=False, workspace_path=None):
+    tools = _ref_workflows(
+        workflow_id, as_agent_tool=as_agent_tool, workspace_path=workspace_path
+    )
+    if len(tools) > 0:
+        return tools[0]
+    else:
+        return None

{pycoze-0.1.284.dist-info → pycoze-0.1.286.dist-info}/METADATA

@@ -1,18 +1,22 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: pycoze
-Version: 0.1.284
+Version: 0.1.286
 Summary: Package for pycoze only!
-Home-page: UNKNOWN
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com
-License: UNKNOWN
-Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-python
+Dynamic: summary
 
 # PYCOZE
 
@@ -28,5 +32,3 @@ Package for pycoze only!
 <!-- python setup.py sdist bdist_wheel -->
 <!-- twine upload dist/* -->
 <!-- 修改src\renderer\layout\init\const.ts中的库版本要求 -->
-
-

{pycoze-0.1.284.dist-info → pycoze-0.1.286.dist-info}/RECORD

@@ -16,15 +16,15 @@ pycoze/bot/bot.py,sha256=_qmUTZ09FmRLifHrW5stDZWGVK6yuMEMBC3fmeYJnqk,844
 pycoze/bot/agent/__init__.py,sha256=3wE8_FFQS8j2BY-g9Cr-onV0POEvDRZaw_NCzpqrNus,265
 pycoze/bot/agent/agent.py,sha256=qkLIRgSMNT1VD_UD0e6kYUuOTOqylQiYSCay6HZ12LA,3653
 pycoze/bot/agent/assistant.py,sha256=5LIgPIVVzx6uIOWT5S_XDDyPPjPHRBBNpIU3GiOkVHc,1186
-pycoze/bot/agent/chat.py,sha256=mubOCAHvA6VtyE6N40elI6KrP6A69uB_G6ihE3G_Vi4,860
+pycoze/bot/agent/chat.py,sha256=9wZ24CPdSbSnPCWmCQJle05U5VlDGgZhZ9z1mezLst0,816
 pycoze/bot/agent/agent_types/__init__.py,sha256=zmU2Kmrv5mCdfg-QlPn2H6pWxbGeq8s7YTqLhpzJC6k,179
 pycoze/bot/agent/agent_types/const.py,sha256=BfUKPrhAHREoMLHuFNG2bCIEkC1-f7K0LEqNg4RwiRE,70
-pycoze/bot/agent/agent_types/openai_func_call_agent.py,sha256=6aKnUQDINyUaCW24oa9Qjkm5w3ctZ6lxAgcE4m9YHwE,6701
+pycoze/bot/agent/agent_types/openai_func_call_agent.py,sha256=3qOyrddujtJ50W9SbH5bapbVTwjgE_LC2TnYJWUH9yc,6649
 pycoze/reference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycoze/reference/bot.py,sha256=EQUngYFn5K7DdQWsPHKOXOsrMAz-3nMjiZEPc3yHfvo,2415
+pycoze/reference/bot.py,sha256=d5uB0OH1MF3LXGoBAF6vX6y50IV3sOrNi_5hXAHAmGU,2255
 pycoze/reference/lib.py,sha256=0xQJTLTHedGzQBsjuTFNBVqYc4-8Yl65gGCrAhWyOX8,2155
-pycoze/reference/tool.py,sha256=CYUS95M_XGnUcl4uWayjzKOM9jGpylS7vWJ66JfNTjI,1710
-pycoze/reference/workflow.py,sha256=LiubvvhGa6KoVIP8W8D8zTZaL595M4Dorsa_kxZ8UgQ,2028
+pycoze/reference/tool.py,sha256=_dggSWn-oC_reB8TuNOPl48Xr-pgwJHF7XmzlIw6lsQ,1714
+pycoze/reference/workflow.py,sha256=22y8yyJgBMcGShBbqKpUGFl2CM9dh646LsM8r-1UKcs,2013
 pycoze/ui/__init__.py,sha256=uaXet23wUk64TcZjpBX8qOx4aUhwA_ucrmcxy7Q4Qr4,929
 pycoze/ui/base.py,sha256=bz9mHZwIXA8LErEHTIonH347u6LP7rxV2EADMMjNZos,1081
 pycoze/ui/color.py,sha256=cT9Ib8uNzkOKxyW0IwVj46o4LwdB1xgNCj1_Rou9d_4,854
@@ -35,8 +35,8 @@ pycoze/utils/arg.py,sha256=jop1tBfe5hYkHW1NSpCeaZBEznkgguBscj_7M2dWfrs,503
 pycoze/utils/env.py,sha256=5pWlXfM1F5ZU9hhv1rHlDEanjEW5wf0nbyez9bNRqqA,559
 pycoze/utils/socket.py,sha256=bZbFFRH4mfThzRqt55BAAGQ6eICx_ja4x8UGGrUdAm8,2428
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.284.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.284.dist-info/METADATA,sha256=7NRzJB0uGt2arHh5jYvBXhcspsqvN7RQmLtr7BUsaa4,755
-pycoze-0.1.284.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
-pycoze-0.1.284.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.284.dist-info/RECORD,,
+pycoze-0.1.286.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.286.dist-info/METADATA,sha256=CE7H9CpBdsQrmZo7FaGjGSQNBcOAO0rkp7BsVQ3ZCYU,854
+pycoze-0.1.286.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+pycoze-0.1.286.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.286.dist-info/RECORD,,

{pycoze-0.1.284.dist-info → pycoze-0.1.286.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.37.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any