pycoze 0.1.130__tar.gz → 0.1.132__tar.gz

{pycoze-0.1.130 → pycoze-0.1.132}/PKG-INFO

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

{pycoze-0.1.130 → pycoze-0.1.132}/pycoze/access/bot_for_bot.py

@@ -45,6 +45,7 @@ def {random_name}(command:str) -> str:
     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)

{pycoze-0.1.130 → pycoze-0.1.132}/pycoze/access/lib.py

@@ -54,7 +54,7 @@ def wrapped_func(item, module_path):
         try:
             print(f"调用了 {item.name}")
         except:
-            pass
+            print(f"called unknown")
         with ChangeDirectoryAndPath(module_path):
             result = original_tool_function(*args, **kwargs)
         try:

{pycoze-0.1.130 → pycoze-0.1.132}/pycoze/bot/agent/agent.py

@@ -10,9 +10,10 @@ from langchain_core.messages import (
     SystemMessage,
 )
 from langchain_core.agents import AgentFinish
+from .agent_types.const import HumanToolString
 
 
-async def run_agent(agent, inputs: list):
+async def run_agent(agent, inputs: list, tool_compatibility_mode:bool):
     exist_ids = set()
     content_list = []
     async for event in agent.astream_events(inputs, version="v2"):
@@ -27,10 +28,9 @@ async def run_agent(agent, inputs: list):
                 ):
                     input_list = event["data"]["input"]
                     for msg in input_list:
-                        if isinstance(msg, HumanMessage) or isinstance(
-                            msg, SystemMessage
-                        ):
-                            content_list = []
+                        if isinstance(msg, HumanMessage) or isinstance(msg, SystemMessage):
+                            if not tool_compatibility_mode or not msg.content.startswith(HumanToolString):
+                                content_list = [] # 防止内容重复
                         if isinstance(msg, AIMessage) and not isinstance(
                             msg, AIMessageChunk
                         ):
@@ -87,4 +87,4 @@ if __name__ == "__main__":
     )
 
     inputs = [HumanMessage(content="计算根号7+根号88")]
-    print(asyncio.run(run_agent(agent, inputs)))
+    print(asyncio.run(run_agent(agent, inputs, True)))

pycoze-0.1.132/pycoze/bot/agent/agent_types/__init__.py

@@ -0,0 +1,4 @@
+from .openai_func_call_agent import create_openai_func_call_agent_executor
+from .const import HumanToolString
+
+__all__ = [create_openai_func_call_agent_executor, HumanToolString]

pycoze-0.1.132/pycoze/bot/agent/agent_types/const.py

@@ -0,0 +1 @@
+HumanToolString = 'The tool call is done, the result is as follows:\n'

{pycoze-0.1.130 → pycoze-0.1.132}/pycoze/bot/agent/agent_types/openai_func_call_agent.py

@@ -11,7 +11,7 @@ from langgraph.prebuilt import ToolExecutor, ToolInvocation
 import re
 import json
 import random
-
+from .const import HumanToolString
 
 def get_all_markdown_json(content):
     # Find all markdown json blocks
@@ -160,10 +160,7 @@ def create_openai_func_call_agent_executor(
             # HumanMessage
             tool_msgs_str = repr(tool_messages)
             tool_messages = [
-                HumanMessage(
-                    content="The tool call is done, the result is as follows:\n"
-                    + tool_msgs_str
-                )
+                HumanMessage(content=HumanToolString + tool_msgs_str)
             ]
         return tool_messages
 

{pycoze-0.1.130 → pycoze-0.1.132}/pycoze/bot/agent/assistant.py

@@ -1,35 +1,35 @@
-from typing import Sequence
-from langchain.tools import BaseTool
-from langchain_core.language_models.base import LanguageModelLike
-from langchain_core.runnables import RunnableBinding
-from .agent_types import create_openai_func_call_agent_executor
-
-
-class Runnable(RunnableBinding):
-    agent_execution_mode: str
-    tools: Sequence[BaseTool]
-    llm: LanguageModelLike
-    assistant_message: str
-
-    def __init__(
-        self,
-        *,
-        agent_execution_mode: str,
-        tools: Sequence[BaseTool],
-        llm: LanguageModelLike,
-        assistant_message: str,
-        tool_compatibility_mode: bool
-    ) -> None:
-
-        agent_executor = create_openai_func_call_agent_executor(
-            tools, llm, assistant_message, tool_compatibility_mode
-        )
-        agent_executor = agent_executor.with_config({"recursion_limit": 50})
-        super().__init__(
-            tools=tools,
-            llm=llm,
-            agent_execution_mode=agent_execution_mode,
-            assistant_message=assistant_message,
-            bound=agent_executor,
-            return_intermediate_steps=True,
-        )
+from typing import Sequence
+from langchain.tools import BaseTool
+from langchain_core.language_models.base import LanguageModelLike
+from langchain_core.runnables import RunnableBinding
+from .agent_types import create_openai_func_call_agent_executor
+
+
+class Runnable(RunnableBinding):
+    agent_execution_mode: str
+    tools: Sequence[BaseTool]
+    llm: LanguageModelLike
+    assistant_message: str
+
+    def __init__(
+        self,
+        *,
+        agent_execution_mode: str,
+        tools: Sequence[BaseTool],
+        llm: LanguageModelLike,
+        assistant_message: str,
+        tool_compatibility_mode: bool
+    ) -> None:
+
+        agent_executor = create_openai_func_call_agent_executor(
+            tools, llm, assistant_message, tool_compatibility_mode
+        )
+        agent_executor = agent_executor.with_config({"recursion_limit": 50})
+        super().__init__(
+            tools=tools,
+            llm=llm,
+            agent_execution_mode=agent_execution_mode,
+            assistant_message=assistant_message,
+            bound=agent_executor,
+            return_intermediate_steps=True,
+        )

{pycoze-0.1.130 → pycoze-0.1.132}/pycoze/bot/bot.py

@@ -53,8 +53,8 @@ def agent_chat(bot_setting_file, history):
     )
     prompt = role_setting["prompt"]
     if (
-        cfg["model"].startswith("deepseek")
-        or cfg["toolCompatibilityMode"]
+        (cfg["model"].startswith("deepseek")
+        or cfg["toolCompatibilityMode"])
         and len(abilities) > 0
     ):
         prompt += """
@@ -76,7 +76,7 @@ def agent_chat(bot_setting_file, history):
         assistant_message=prompt,
         tool_compatibility_mode=cfg["toolCompatibilityMode"],
     )
-    return asyncio.run(run_agent(agent, history))
+    return asyncio.run(run_agent(agent, history, cfg["toolCompatibilityMode"]))
 
 
 def chat(bot_setting_file: str):

{pycoze-0.1.130 → pycoze-0.1.132}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.130 → pycoze-0.1.132}/pycoze.egg-info/SOURCES.txt

@@ -20,6 +20,7 @@ pycoze/bot/agent/agent.py
 pycoze/bot/agent/assistant.py
 pycoze/bot/agent/chat.py
 pycoze/bot/agent/agent_types/__init__.py
+pycoze/bot/agent/agent_types/const.py
 pycoze/bot/agent/agent_types/openai_func_call_agent.py
 pycoze/ui/__init__.py
 pycoze/ui/base.py

{pycoze-0.1.130 → pycoze-0.1.132}/setup.py

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

pycoze-0.1.130/pycoze/bot/agent/agent_types/__init__.py

@@ -1,4 +0,0 @@
-from .openai_func_call_agent import create_openai_func_call_agent_executor
-
-
-__all__ = [create_openai_func_call_agent_executor]