beswarm 0.1.29__tar.gz → 0.1.31__tar.gz

{beswarm-0.1.29 → beswarm-0.1.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.1.29
+Version: 0.1.31
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.1.29 → beswarm-0.1.31}/beswarm/aient/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aient",
-    version="1.0.88",
+    version="1.0.90",
     description="Aient: The Awakening of Agent.",
     long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
     long_description_content_type="text/markdown",

{beswarm-0.1.29 → beswarm-0.1.31}/beswarm/aient/src/aient/core/request.py

@@ -995,6 +995,10 @@ async def get_gpt_payload(request, engine, provider, api_key=None):
                     }
                 })
 
+    if safe_get(provider, "preferences", "post_body_parameter_overrides", default=None):
+        for key, value in safe_get(provider, "preferences", "post_body_parameter_overrides", default={}).items():
+            payload[key] = value
+
     return url, headers, payload
 
 def build_azure_endpoint(base_url, deployment_id, api_version="2025-01-01-preview"):
@@ -1085,6 +1089,10 @@ async def get_azure_payload(request, engine, provider, api_key=None):
         payload.pop("tools", None)
         payload.pop("tool_choice", None)
 
+    if safe_get(provider, "preferences", "post_body_parameter_overrides", default=None):
+        for key, value in safe_get(provider, "preferences", "post_body_parameter_overrides", default={}).items():
+            payload[key] = value
+
     return url, headers, payload
 
 async def get_openrouter_payload(request, engine, provider, api_key=None):

{beswarm-0.1.29 → beswarm-0.1.31}/beswarm/aient/src/aient/core/response.py

@@ -327,7 +327,10 @@ async def fetch_azure_response_stream(client, url, headers, payload):
                         continue
 
                     if no_stream_content or content or sse_string:
-                        sse_string = await generate_sse_response(timestamp, safe_get(line, "model", default=None), content=no_stream_content or content)
+                        input_tokens = safe_get(line, "usage", "prompt_tokens", default=0)
+                        output_tokens = safe_get(line, "usage", "completion_tokens", default=0)
+                        total_tokens = safe_get(line, "usage", "total_tokens", default=0)
+                        sse_string = await generate_sse_response(timestamp, safe_get(line, "model", default=None), content=no_stream_content or content, total_tokens=total_tokens, prompt_tokens=input_tokens, completion_tokens=output_tokens)
                         yield sse_string
                     else:
                         if no_stream_content:

{beswarm-0.1.29 → beswarm-0.1.31}/beswarm/aient/src/aient/models/chatgpt.py

@@ -431,26 +431,21 @@ class chatgpt(BaseLLM):
 
         function_parameter = parse_function_xml(full_response)
         if function_parameter:
-            print("function_parameter1", function_parameter)
-            function_parameter = [tool_dict for tool_dict in function_parameter if tool_dict.get("function_name", "") in self.plugins.keys()]
             invalid_tools = [tool_dict for tool_dict in function_parameter if tool_dict.get("function_name", "") not in self.plugins.keys()]
+            function_parameter = [tool_dict for tool_dict in function_parameter if tool_dict.get("function_name", "") in self.plugins.keys()]
             for tool_dict in invalid_tools:
                 full_response = full_response + f"\n\nFunction: {tool_dict.get('function_name', '')} does not exist! I must use existing functions. I need to try again."
-            print("function_parameter2", self.print_log, function_parameter)
-            if self.print_log:
+            if self.print_log and invalid_tools:
                 print("invalid_tools", invalid_tools)
+                print("function_parameter", function_parameter)
                 print("full_response", full_response)
             if function_parameter:
                 need_function_call = True
             else:
                 need_function_call = False
                 if self.print_log:
-                    print("Failed to parse function_parameter", function_parameter)
-                    print("full_response", full_response)
-                full_response = (
-                    f"{full_response}"
-                    "\n\nFailed to parse function parameter, I need to try again."
-                )
+                    print("Failed to parse function_parameter full_response", full_response)
+                full_response = ""
 
         # 处理函数调用
         if need_function_call and self.use_plugins != False:
@@ -458,11 +453,6 @@ class chatgpt(BaseLLM):
                 print("function_parameter", function_parameter)
                 print("function_full_response", function_full_response)
 
-            function_parameter = [tool_dict for tool_dict in function_parameter if tool_dict.get("function_name", "") in self.plugins.keys()]
-
-            if self.print_log:
-                print("function_parameter3", function_parameter)
-
             function_response = ""
             # 定义处理单个工具调用的辅助函数
             async def process_single_tool_call(tool_name, tool_args, tool_id):

{beswarm-0.1.29 → beswarm-0.1.31}/beswarm/aient/src/aient/plugins/write_file.py

@@ -44,7 +44,7 @@ Example: Requesting to write to frontend-config.json
     # 确保目录存在
     os.makedirs(os.path.dirname(path) or '.', exist_ok=True)
 
-    if content.startswith("## ") and path.endswith(".md"):
+    if content.startswith("##") and (path.endswith(".md") or path.endswith(".txt")):
         content = "\n\n" + content
 
     # 写入文件

{beswarm-0.1.29 → beswarm-0.1.31}/beswarm.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.1.29
+Version: 0.1.31
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.1.29 → beswarm-0.1.31}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beswarm"
-version = "0.1.29"
+version = "0.1.31"
 description = "MAS"
 readme = "README.md"
 requires-python = ">=3.11"