beswarm 0.2.72__tar.gz → 0.2.75__tar.gz

{beswarm-0.2.72 → beswarm-0.2.75}/PKG-INFO

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

{beswarm-0.2.72 → beswarm-0.2.75}/beswarm/agents/planact.py

@@ -12,10 +12,10 @@ from typing import List, Dict, Union
 
 from ..broker import MessageBroker
 from ..aient.aient.models import chatgpt
-from ..aient.aient.models.chatgpt import ModelNotFoundError, TaskComplete, RetryFailedError
 from ..aient.aient.plugins import get_function_call_list, registry
 from ..prompt import worker_system_prompt, instruction_system_prompt
 from ..utils import extract_xml_content, get_current_screen_image_message, replace_xml_content, register_mcp_tools, setup_logger
+from ..aient.aient.models.chatgpt import ModelNotFoundError, TaskComplete, RetryFailedError, InputTokenCountExceededError, BadRequestError
 
 try:
     from importlib import metadata
@@ -154,11 +154,12 @@ class InstructionAgent(BaseAgent):
             self.logger.error("❌ Commander retry failed, retrying...")
             self.broker.publish(message, self.error_topic)
             return
-
-        if "'status_code': 413" in raw_response or \
-        "'status_code': 400" in raw_response:
+        except InputTokenCountExceededError as e:
             self.broker.publish({"status": "error", "result": "The request body is too long, please try again."}, self.status_topic)
             return
+        except BadRequestError as e:
+            self.broker.publish({"status": "error", "result": "Bad request error!"}, self.status_topic)
+            return
 
         self.broker.publish({"status": "new_message", "result": "\n🤖 指令智能体:\n" + raw_response}, self.status_topic)
 

{beswarm-0.2.72 → beswarm-0.2.75}/beswarm/aient/aient/core/response.py

@@ -21,13 +21,12 @@ async def check_response(response, error_log):
         return {"error": f"{error_log} HTTP Error", "status_code": response.status_code, "details": error_json}
     return None
 
-async def gemini_json_poccess(response_str):
+async def gemini_json_poccess(response_json):
     promptTokenCount = 0
     candidatesTokenCount = 0
     totalTokenCount = 0
     image_base64 = None
 
-    response_json = await asyncio.to_thread(json.loads, response_str)
     json_data = safe_get(response_json, "candidates", 0, "content", default=None)
     finishReason = safe_get(response_json, "candidates", 0 , "finishReason", default=None)
     if finishReason:
@@ -48,7 +47,7 @@ async def gemini_json_poccess(response_str):
 
     function_call_name = safe_get(json_data, "functionCall", "name", default=None)
     function_full_response = safe_get(json_data, "functionCall", "args", default="")
-    function_full_response = json.dumps(function_full_response) if function_full_response else None
+    function_full_response = await asyncio.to_thread(json.dumps, function_full_response) if function_full_response else None
 
     blockReason = safe_get(json_data, 0, "promptFeedback", "blockReason", default=None)
 
@@ -77,7 +76,7 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model, tim
                 if line.startswith("data: "):
                     parts_json = line.lstrip("data: ").strip()
                     try:
-                        await asyncio.to_thread(json.loads, parts_json)
+                        response_json = await asyncio.to_thread(json.loads, parts_json)
                     except json.JSONDecodeError:
                         logger.error(f"JSON decode error: {parts_json}")
                         continue
@@ -85,12 +84,12 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model, tim
                     parts_json += line
                     parts_json = parts_json.lstrip("[,")
                     try:
-                        await asyncio.to_thread(json.loads, parts_json)
+                        response_json = await asyncio.to_thread(json.loads, parts_json)
                     except json.JSONDecodeError:
                         continue
 
                 # https://ai.google.dev/api/generate-content?hl=zh-cn#FinishReason
-                is_thinking, reasoning_content, content, image_base64, function_call_name, function_full_response, finishReason, blockReason, promptTokenCount, candidatesTokenCount, totalTokenCount = await gemini_json_poccess(parts_json)
+                is_thinking, reasoning_content, content, image_base64, function_call_name, function_full_response, finishReason, blockReason, promptTokenCount, candidatesTokenCount, totalTokenCount = await gemini_json_poccess(response_json)
 
                 if is_thinking:
                     sse_string = await generate_sse_response(timestamp, model, reasoning_content=reasoning_content)

{beswarm-0.2.72 → beswarm-0.2.75}/beswarm/aient/aient/models/chatgpt.py

@@ -37,10 +37,18 @@ class RateLimitError(Exception):
     """Custom exception for rate limit (429) errors."""
     pass
 
+class BadRequestError(Exception):
+    """Custom exception for bad request (400) errors."""
+    pass
+
 class HTTPError(Exception):
     """Custom exception for HTTP 500 errors."""
     pass
 
+class InputTokenCountExceededError(Exception):
+    """Custom exception for input token count exceeding the maximum."""
+    pass
+
 class ConfigurationError(Exception):
     """Custom exception for configuration errors."""
     pass
@@ -802,6 +810,10 @@ class chatgpt(BaseLLM):
                             raise ModelNotFoundError(f"Model: {model or self.engine} not found!")
                         if "HTTP Error', 'status_code': 429" in processed_chunk:
                             raise RateLimitError(f"Rate limit exceeded for model: {model or self.engine}")
+                        if "HTTP Error', 'status_code': 413" in processed_chunk:
+                            raise InputTokenCountExceededError(processed_chunk)
+                        if "HTTP Error', 'status_code': 400" in processed_chunk:
+                            raise BadRequestError(f"Bad Request: {processed_chunk}")
                         if "HTTP Error', 'status_code': " in processed_chunk:
                             raise HTTPError(f"HTTP Error: {processed_chunk}")
                     yield processed_chunk
@@ -814,7 +826,7 @@ class chatgpt(BaseLLM):
                 return # Stop iteration
             except httpx.RemoteProtocolError:
                 continue
-            except httpx.ReadError:
+            except httpx.ReadError as e:
                 self.logger.warning(f"{e}, retrying...")
                 continue
             except APITimeoutError:
@@ -826,6 +838,12 @@ class chatgpt(BaseLLM):
             except RateLimitError as e:
                 self.logger.warning(f"{e}, retrying...")
                 continue
+            except InputTokenCountExceededError as e:
+                self.logger.error(f"The request body is too long: {e}")
+                raise
+            except BadRequestError as e:
+                self.logger.error(f"Bad request error: {e}")
+                raise
             except ValidationError as e:
                 self.logger.warning(f"Validation failed: {e}. Retrying with corrective prompt.")
                 need_done_prompt = [

{beswarm-0.2.72 → beswarm-0.2.75}/beswarm/tools/subtasks.py

@@ -36,11 +36,14 @@ def create_task(goal, tools, work_dir):
         }
     ]
 
+    if work_dir == str(task_manager.root_path):
+        return f"<tool_error>子任务的工作目录位置在主任务的工作目录的子目录。子任务工作目录**禁止**设置为主任务目录本身。请重新创建子任务。当前主任务工作目录：{task_manager.root_path}</tool_error>"
+
     # 调用新的批量创建接口
     task_ids = task_manager.create_tasks_batch(worker_fun, tasks_params)
 
     # 返回新创建的单个任务ID
-    return f"子任务已提交到队列，ID: {task_ids[0]}" if task_ids else "任务提交失败"
+    return f"子任务已提交到队列，ID: {task_ids[0]}" if task_ids else "<tool_error>任务提交失败</tool_error>"
 
 @register_tool()
 def resume_task(task_id, goal):

{beswarm-0.2.72 → beswarm-0.2.75}/beswarm.egg-info/PKG-INFO

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

{beswarm-0.2.72 → beswarm-0.2.75}/pyproject.toml

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