pycoze 0.1.85__tar.gz → 0.1.86__tar.gz

{pycoze-0.1.85 → pycoze-0.1.86}/PKG-INFO

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

{pycoze-0.1.85 → pycoze-0.1.86}/pycoze/bot/agent/agent.py

@@ -83,6 +83,7 @@ if __name__ == "__main__":
         tools=[python_tool],
         llm=chat,
         assistant_message="请以女友的口吻回答，输出不小于100字，可以随便说点其他的",
+        support_tools=True,
     )
 
     inputs = [HumanMessage(content="计算根号7+根号88")]

{pycoze-0.1.85 → pycoze-0.1.86}/pycoze/bot/agent/agent_types/openai_func_call_agent.py

@@ -4,7 +4,7 @@ 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
-from langchain_core.messages import SystemMessage, ToolMessage
+from langchain_core.messages import SystemMessage, ToolMessage, HumanMessage
 from langgraph.graph import END
 from langgraph.graph.message import MessageGraph
 from langgraph.prebuilt import ToolExecutor, ToolInvocation
@@ -31,7 +31,11 @@ def get_all_markdown_json(content):
 
 
 def create_openai_func_call_agent_executor(
-    tools: list[BaseTool], llm: LanguageModelLike, system_message: str, **kwargs
+    tools: list[BaseTool],
+    llm: LanguageModelLike,
+    system_message: str,
+    support_tools: str,
+    **kwargs
 ):
 
     async def _get_messages(messages):
@@ -58,7 +62,6 @@ def create_openai_func_call_agent_executor(
     def should_continue(messages):
         # If there is no FuncCall, then we finish
         last_message = messages[-1]
-        print("last_message", last_message)
         if last_message.content.strip().endswith("```"):
             last_message.content = last_message.content + "\n\n"  # 避免影响阅读
         # if not last_message.tool_calls:
@@ -128,7 +131,6 @@ def create_openai_func_call_agent_executor(
         # Based on the continue condition
         # we know the last message involves a FuncCall
         last_message = messages[-1]
-        print("last_message", last_message)
         for tool_call in last_message.additional_kwargs["tool_calls"]:
             function = tool_call["function"]
             function_name = function["name"]
@@ -167,44 +169,32 @@ def create_openai_func_call_agent_executor(
                 additional_kwargs={"name": tool_call["function"]["name"]},
             )
             tool_messages.append(message)
+        if not support_tools:
+            # HumanMessage
+            tool_msgs_str = repr(tool_messages)
+            tool_messages = [
+                HumanMessage(content="工具调用结束，结果如下：\n" + tool_msgs_str)
+            ]
         return tool_messages
 
     workflow = MessageGraph()
 
     # Define the two nodes we will cycle between
     workflow.add_node("agent", agent)
-    workflow.add_node("action", call_tool)
+    workflow.add_node("call_tool", call_tool)
 
-    # Set the entrypoint as `agent`
-    # This means that this node is the first one called
     workflow.set_entry_point("agent")
 
-    # We now add a conditional edge
     workflow.add_conditional_edges(
-        # First, we define the start node. We use `agent`.
-        # This means these are the edges taken after the `agent` node is called.
         "agent",
-        # Next, we pass in the function that will determine which node is called next.
         should_continue,
-        # Finally we pass in a mapping.
-        # The keys are strings, and the values are other nodes.
-        # END is a special node marking that the graph should finish.
-        # What will happen is we will call `should_continue`, and then the output of that
-        # will be matched against the keys in this mapping.
-        # Based on which one it matches, that node will then be called.
         {
-            # If `tools`, then we call the tool node.
-            "continue": "action",
-            # Otherwise we finish.
+            "continue": "call_tool",
             "end": END,
         },
     )
 
-    # We now add a normal edge from `tools` to `agent`.
-    # This means that after `tools` is called, `agent` node is called next.
-    workflow.add_edge("action", "agent")
+    # 调用完工具后，再次调用agent
+    workflow.add_edge("call_tool", "agent")
 
-    # Finally, we compile it!
-    # This compiles it into a LangChain Runnable,
-    # meaning you can use it as you would any other runnable
     return workflow.compile()

{pycoze-0.1.85 → pycoze-0.1.86}/pycoze/bot/agent/assistant.py

@@ -18,10 +18,11 @@ class Runnable(RunnableBinding):
         tools: Sequence[BaseTool],
         llm: LanguageModelLike,
         assistant_message: str,
+        support_tools: bool
     ) -> None:
 
         agent_executor = create_openai_func_call_agent_executor(
-            tools, llm, assistant_message
+            tools, llm, assistant_message, support_tools
         )
         agent_executor = agent_executor.with_config({"recursion_limit": 50})
         super().__init__(

{pycoze-0.1.85 → pycoze-0.1.86}/pycoze/bot/bot.py

@@ -9,6 +9,9 @@ from langchain_core.utils.function_calling import convert_to_openai_tool
 
 params = utils.arg.read_params_file()
 llm_file = params["appPath"] + "/JsonStorage/llm.json"
+with open(llm_file, "r", encoding="utf-8") as f:
+    cfg = json.load(f)
+support_tools = not cfg["model"].startswith("yi-")
 
 
 def load_role_setting(bot_setting_file: str):
@@ -29,28 +32,23 @@ def load_tools(bot_setting_file: str):
 def agent_chat(bot_setting_file, history):
     role_setting = load_role_setting(bot_setting_file)
     tools = load_tools(bot_setting_file)
-    with open(llm_file, "r", encoding="utf-8") as f:
-        cfg = json.load(f)
-        chat = ChatOpenAI(
-            api_key=cfg["apiKey"],
-            base_url=cfg["baseURL"],
-            model=cfg["model"],
-            temperature=(
-                role_setting["temperature"] * 2
-                if cfg["model"].startswith("deepseek") or cfg["model"].startswith("yi-")
-                else role_setting["temperature"]
-            ),
-            stop_sequences=[
-                "tool▁calls▁end",
-                "tool▁call▁end",
-            ],  # 停用deepseek的工具调用标记，不然会虚构工具调用过程和结果
-        )
+
+    chat = ChatOpenAI(
+        api_key=cfg["apiKey"],
+        base_url=cfg["baseURL"],
+        model=cfg["model"],
+        temperature=(
+            role_setting["temperature"] * 2
+            if cfg["model"].startswith("deepseek")
+            else role_setting["temperature"]
+        ),
+        stop_sequences=[
+            "tool▁calls▁end",
+            "tool▁call▁end",
+        ],  # 停用deepseek的工具调用标记，不然会虚构工具调用过程和结果
+    )
     prompt = role_setting["prompt"]
-    if (
-        cfg["model"].startswith("deepseek")
-        or cfg["model"].startswith("yi-")
-        and len(tools) > 0
-    ):
+    if cfg["model"].startswith("deepseek") or not support_tools and len(tools) > 0:
         prompt += """
 如果需要调用工具，请使用以正确的json格式进行结尾（务必保证json格式正确，不要出现反斜杠未转义等问题）：
 ```json
@@ -66,6 +64,7 @@ def agent_chat(bot_setting_file, history):
         tools=tools,
         llm=chat,
         assistant_message=prompt,
+        support_tools=support_tools,
     )
     return asyncio.run(run_agent(agent, history))
 

{pycoze-0.1.85 → pycoze-0.1.86}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.85 → pycoze-0.1.86}/setup.py

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