pycoze 0.1.89__tar.gz → 0.1.93__tar.gz

{pycoze-0.1.89 → pycoze-0.1.93}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.89
+Version: 0.1.93
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com
@@ -18,7 +18,7 @@ Package for pycoze only!
 
 
 <!-- For author only -->
-<!-- pip install twine -->
+<!-- pip install twine wheel -->
 
 <!-- 递增setup.py的版本 -->
 <!-- 删除build和dist和pycoze.egg-info文件夹 -->

{pycoze-0.1.89 → pycoze-0.1.93}/README.md

@@ -5,7 +5,7 @@ Package for pycoze only!
 
 
 <!-- For author only -->
-<!-- pip install twine -->
+<!-- pip install twine wheel -->
 
 <!-- 递增setup.py的版本 -->
 <!-- 删除build和dist和pycoze.egg-info文件夹 -->

pycoze-0.1.93/pycoze/ai/__init__.py

@@ -0,0 +1 @@
+from .vram_reserve import reserve_vram, reserve_vram_retry, unreserve_vram

{pycoze-0.1.89 → pycoze-0.1.93}/pycoze/ai/vram_reserve.py

@@ -88,33 +88,25 @@ def reserve_vram_retry(gb, retry=None, uid=None):
         retry = sys.maxsize
     for i in range(retry):
         time.sleep(1)
-        if i % 10 == 0 or i < 10:
+        if i % 10 == 0 or i < 10 and i != 0:
             print(f"重试第{i}次")
         if reserve_vram(gb, uid):
             return True
     return False
 
 
-def release_vram(uid=None):
+def unreserve_vram(uid=None):
     if uid is None:
         uid = f"pid:{os.getpid()}"
     with sqlite3.connect(DATABASE_PATH) as conn:
         cursor = conn.cursor()
         cursor.execute(f"DELETE FROM {TABLE_NAME} WHERE uid = ?", (uid,))
         conn.commit()
-        # 计算释放后的剩余VRAM大小
-        cursor.execute(f"SELECT SUM(reserved_gb) FROM {TABLE_NAME}")
-        total_reserved = cursor.fetchone()[0]
-        if total_reserved is None:
-            total_reserved = 0.0
-        available_gb = get_vram_resources() - total_reserved
-        print(f"释放成功，剩余VRAM大小: {available_gb} GB")
 
 
 # 注册退出时的清理函数
 def cleanup():
-    release_vram()
-    print("程序退出，VRAM资源已释放")
+    unreserve_vram()
 
 
 def initialize_and_check():

{pycoze-0.1.89 → pycoze-0.1.93}/pycoze/bot/agent/agent_types/openai_func_call_agent.py

@@ -30,40 +30,11 @@ def get_all_markdown_json(content):
     return json_list
 
 
-def create_openai_func_call_agent_executor(
-    tools: list[BaseTool],
-    llm: LanguageModelLike,
-    system_message: str,
-    tool_compatibility_mode: str,
-    **kwargs
-):
-
-    async def _get_messages(messages):
-        msgs = []
-        for m in messages:
-            if isinstance(m, ToolMessage):
-                _dict = m.dict()
-                _dict["content"] = str(_dict["content"])
-                m_c = ToolMessage(**_dict)
-                msgs.append(m_c)
-            else:
-                msgs.append(m)
-
-        return [SystemMessage(content=system_message)] + msgs
-
-    if tools:
-        llm_with_tools = llm.bind(tools=[convert_to_openai_tool(t) for t in tools])
+def get_tools(last_message):
+    if "tool_calls" in last_message.additional_kwargs:
+        return last_message.additional_kwargs["tool_calls"]
     else:
-        llm_with_tools = llm
-    agent = _get_messages | llm_with_tools
-    tool_executor = ToolExecutor(tools)
-
-    # Define the function that determines whether to continue or not
-    def should_continue(messages):
-        # If there is no FuncCall, then we finish
-        last_message = messages[-1]
-        if last_message.content.strip().endswith("```"):
-            last_message.content = last_message.content + "\n\n"  # 避免影响阅读
+        tool_calls = None
         if '"name"' in last_message.content and '"parameters":' in last_message.content:
             print("name 和 paremeters 模式")
             all_json = get_all_markdown_json(last_message.content)
@@ -79,9 +50,6 @@ def create_openai_func_call_agent_executor(
                         "id": random.randint(0, 1000000),
                     }
                 )
-            last_message.tool_calls = tool_calls
-            last_message.additional_kwargs["tool_calls"] = tool_calls
-            return "continue"
         if "<｜tool▁sep｜>" in last_message.content:
             print("deepseek的bug: <｜tool▁sep｜> 模式")
             name = (
@@ -100,15 +68,49 @@ def create_openai_func_call_agent_executor(
                     }
                 )
 
-            last_message.additional_kwargs["tool_calls"] = tool_calls
-            last_message.tool_calls = tool_calls
-            return "continue"
+        return tool_calls
+
+
+def create_openai_func_call_agent_executor(
+    tools: list[BaseTool],
+    llm: LanguageModelLike,
+    system_message: str,
+    tool_compatibility_mode: str,
+    **kwargs
+):
+
+    async def _get_messages(messages):
+        msgs = []
+        for m in messages:
+            if isinstance(m, ToolMessage):
+                _dict = m.dict()
+                _dict["content"] = str(_dict["content"])
+                m_c = ToolMessage(**_dict)
+                msgs.append(m_c)
+            else:
+                msgs.append(m)
+
+        return [SystemMessage(content=system_message)] + msgs
+
+    if tools:
+        llm_with_tools = llm.bind(tools=[convert_to_openai_tool(t) for t in tools])
+    else:
+        llm_with_tools = llm
+    agent = _get_messages | llm_with_tools
+    tool_executor = ToolExecutor(tools)
+
+    # Define the function that determines whether to continue or not
+    def should_continue(messages):
+        # If there is no FuncCall, then we finish
+        last_message = messages[-1]
+        if last_message.content.strip().endswith("```"):
+            last_message.content = last_message.content + "\n\n"  # 避免影响阅读
+        tools = get_tools(last_message)
+        if last_message.tool_calls or tools:
+            return 'continue'
+        return 'end'
 
-        if not last_message.tool_calls:
-            return "end"
-        # Otherwise if there is, we continue
-        else:
-            return "continue"
+        
 
     # Define the function to execute tools
     async def call_tool(messages):
@@ -116,7 +118,7 @@ def create_openai_func_call_agent_executor(
         # Based on the continue condition
         # we know the last message involves a FuncCall
         last_message = messages[-1]
-        for tool_call in last_message.additional_kwargs["tool_calls"]:
+        for tool_call in get_tools(last_message):
             function = tool_call["function"]
             function_name = function["name"]
             if function_name == "a_delay_function":
@@ -141,7 +143,7 @@ def create_openai_func_call_agent_executor(
         # We use the response to create a ToolMessage
         tool_messages = []
         for tool_call, response in zip(
-            last_message.additional_kwargs["tool_calls"], responses
+            get_tools(last_message), responses
         ):
             if not isinstance(response, (str, int, float, bool, list, tuple)):
                 response = repr(

{pycoze-0.1.89 → pycoze-0.1.93}/pycoze/bot/bot.py

@@ -53,8 +53,9 @@ def agent_chat(bot_setting_file, history):
         and len(tools) > 0
     ):
         prompt += """
-如果不确定结果，请务必使用工具查询。
-如果需要调用工具，请使用以正确markdown中的json代码格式进行结尾（务必保证json格式正确，不要出现反斜杠未转义等问题）：
+作为一个AI，你如果不确定结果，请务必使用工具查询。
+你可以通过下面的方式使用工具，并耐心等待工具返回结果。
+如果你需要调用工具，请使用以正确markdown中的json代码格式进行结尾（务必保证json格式正确，不要出现反斜杠未转义等问题）：
 ```json
 {"name": 函数名, "parameters": 参数词典}
 ```

{pycoze-0.1.89 → pycoze-0.1.93}/pycoze.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.89
+Version: 0.1.93
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com
@@ -18,7 +18,7 @@ Package for pycoze only!
 
 
 <!-- For author only -->
-<!-- pip install twine -->
+<!-- pip install twine wheel -->
 
 <!-- 递增setup.py的版本 -->
 <!-- 删除build和dist和pycoze.egg-info文件夹 -->

{pycoze-0.1.89 → pycoze-0.1.93}/setup.py

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

pycoze-0.1.89/pycoze/ai/__init__.py

@@ -1 +0,0 @@
-from .vram_reserve import reserve_vram_retry, release_vram, reserve_vram