pycoze 0.1.233__tar.gz → 0.1.237__tar.gz

{pycoze-0.1.233 → pycoze-0.1.237}/PKG-INFO

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

{pycoze-0.1.233 → pycoze-0.1.237}/pycoze/bot/agent/agent.py

@@ -13,7 +13,7 @@ from langchain_core.agents import AgentFinish
 from .agent_types.const import HumanToolString
 
 
-async def run_agent(agent, inputs: list, tool_compatibility_mode: bool):
+async def run_agent(agent, inputs: list):
     exist_ids = set()
     content_list = []
     async for event in agent.astream_events(inputs, version="v2"):
@@ -29,7 +29,7 @@ async def run_agent(agent, inputs: list, tool_compatibility_mode: bool):
                     input_list = event["data"]["input"]
                     for msg in input_list:
                         if isinstance(msg, HumanMessage) or isinstance(msg, SystemMessage):
-                            if not tool_compatibility_mode or not msg.content.startswith(HumanToolString):
+                            if not msg.content.startswith(HumanToolString):
                                 content_list = []  # 防止内容重复
                         if isinstance(msg, AIMessage) and not isinstance(
                             msg, AIMessageChunk
@@ -84,7 +84,6 @@ if __name__ == "__main__":
         tools=[python_tool],
         llm=chat,
         assistant_message="请以女友的口吻回答，输出不小于100字，可以随便说点其他的",
-        tool_compatibility_mode=False,
     )
 
     inputs = [HumanMessage(content="计算根号7+根号88")]

{pycoze-0.1.233 → pycoze-0.1.237}/pycoze/bot/agent/agent_types/openai_func_call_agent.py

@@ -75,7 +75,6 @@ def create_openai_func_call_agent_executor(
     tools: list[BaseTool],
     llm: LanguageModelLike,
     system_message: str,
-    tool_compatibility_mode: str,
     **kwargs
 ):
 
@@ -156,12 +155,7 @@ def create_openai_func_call_agent_executor(
                 additional_kwargs={"name": tool_call["function"]["name"]},
             )
             tool_messages.append(message)
-        if tool_compatibility_mode:
-            # HumanMessage
-            tool_msgs_str = repr(tool_messages)
-            tool_messages = [
-                HumanMessage(content=HumanToolString + tool_msgs_str)
-            ]
+
         return tool_messages
 
     workflow = MessageGraph()

{pycoze-0.1.233 → pycoze-0.1.237}/pycoze/bot/agent/assistant.py

@@ -18,11 +18,10 @@ class Runnable(RunnableBinding):
         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
+            tools, llm, assistant_message
         )
         agent_executor = agent_executor.with_config({"recursion_limit": 50})
         super().__init__(

{pycoze-0.1.233 → pycoze-0.1.237}/pycoze/bot/agent_chat.py

@@ -10,8 +10,6 @@ from langchain_core.utils.function_calling import convert_to_openai_tool
 import os
 
 cfg = utils.read_json_file("llm.json")
-if cfg["model"].startswith("qwen-") or cfg["model"].startswith("deepseek"):
-    cfg["toolCompatibilityMode"] = False
 
 
 def load_role_setting(bot_setting_file: str):
@@ -45,10 +43,10 @@ async def check_interrupt_file(interval, interrupt_file,agent_task):
             agent_task.cancel()
             break 
         
-async def run_with_interrupt_check(agent, history, tool_compatibility_mode, interrupt_file, check_interval=1):
+async def run_with_interrupt_check(agent, history, interrupt_file, check_interval=1):
     clear_chat_data()
     try:
-        agent_task = asyncio.create_task(run_agent(agent, history, tool_compatibility_mode))
+        agent_task = asyncio.create_task(run_agent(agent, history))
         check_task = asyncio.create_task(check_interrupt_file(check_interval, interrupt_file, agent_task))
         result = await agent_task
         return result
@@ -59,6 +57,8 @@ async def run_with_interrupt_check(agent, history, tool_compatibility_mode, inte
         print(traceback.format_exc())
         return None  # 返回 None 或者处理异常后的结果
     finally:
+        if not agent_task.done():
+            agent_task.cancel()
         # 确保即使发生异常也会取消检查任务
         if not check_task.done():
             check_task.cancel()
@@ -86,35 +86,18 @@ async def agent_chat(bot_setting_file, history):
         ],  # 停用deepseek的工具调用标记，不然会虚构工具调用过程和结果
     )
     prompt = role_setting["prompt"]
-    if (
-        (cfg["model"].startswith("deepseek")
-        or cfg["toolCompatibilityMode"])
-        and len(abilities) > 0
-    ):
-        prompt += """
-作为一个AI，你如果不确定结果，请务必使用工具查询。
-你可以通过下面的方式使用工具，并耐心等待工具返回结果。
-如果你需要调用工具，请使用以正确markdown中的json代码格式进行结尾（务必保证json格式正确，不要出现反斜杠未转义等问题）：
-```json
-{"name": 函数名, "parameters": 参数词典}
-```
-"""
-        if cfg["model"].startswith("yi-"):
-            prompt += "\nAvailable functions:\n"
-            for t in abilities:
-                prompt += f"\n```json\n{json.dumps(convert_to_openai_tool(t))}\n```"
+
     agent = Runnable(
         agent_execution_mode="FuncCall",
         tools=abilities,
         llm=chat,
         assistant_message=prompt,
-        tool_compatibility_mode=cfg["toolCompatibilityMode"],
     )
     params = utils.read_params_file()
     if "interruptFile" in params:
         interrupt_file_path = params["interruptFile"]
-        result = await run_with_interrupt_check(agent, history, cfg["toolCompatibilityMode"], interrupt_file_path)
+        result = await run_with_interrupt_check(agent, history,interrupt_file_path)
     else:
-        result = await run_agent(agent, history, cfg["toolCompatibilityMode"])
+        result = await run_agent(agent, history)
     return result
 

{pycoze-0.1.233 → pycoze-0.1.237}/pycoze/bot/bot.py

@@ -1,5 +1,4 @@
 from langchain_core.messages import HumanMessage, AIMessage
-import threading
 import json
 from .agent import INPUT_MESSAGE, output, CHAT_DATA, clear_chat_data
 from .agent_chat import agent_chat

{pycoze-0.1.233 → pycoze-0.1.237}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.233 → pycoze-0.1.237}/setup.py

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