beswarm 0.2.9__tar.gz → 0.2.10__tar.gz

{beswarm-0.2.9 → beswarm-0.2.10}/PKG-INFO

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

{beswarm-0.2.9 → beswarm-0.2.10}/beswarm/aient/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aient",
-    version="1.1.41",
+    version="1.1.42",
     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.2.9 → beswarm-0.2.10}/beswarm/aient/src/aient/core/request.py

@@ -239,15 +239,17 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
         if m:
             try:
                 val = int(m.group(1))
-                if val < 0:
-                    val = 0
-                elif val > 32768 and "gemini-2.5-pro" in original_model:
+                if val > 32768 and "gemini-2.5-pro" in original_model:
                     val = 32768
                 elif val < 128 and "gemini-2.5-pro" in original_model:
                     val = 128
+                elif val <= 0:
+                    val = 0
                 elif val > 24576:
                     val = 24576
                 payload["generationConfig"]["thinkingConfig"]["thinkingBudget"] = val
+                if val == 0:
+                    payload["generationConfig"].pop("thinkingConfig", None)
             except ValueError:
                 # 如果转换为整数失败，忽略思考预算设置
                 pass
@@ -535,15 +537,17 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
         if m:
             try:
                 val = int(m.group(1))
-                if val < 0:
-                    val = 0
-                elif val > 32768 and "gemini-2.5-pro" in original_model:
+                if val > 32768 and "gemini-2.5-pro" in original_model:
                     val = 32768
                 elif val < 128 and "gemini-2.5-pro" in original_model:
                     val = 128
+                elif val <= 0:
+                    val = 0
                 elif val > 24576:
                     val = 24576
                 payload["generationConfig"]["thinkingConfig"]["thinkingBudget"] = val
+                if val == 0:
+                    payload["generationConfig"].pop("thinkingConfig", None)
             except ValueError:
                 # 如果转换为整数失败，忽略思考预算设置
                 pass

{beswarm-0.2.9 → beswarm-0.2.10}/beswarm/tools/worker.py

@@ -15,6 +15,8 @@ from ..bemcp.bemcp import MCPClient, convert_tool_format, MCPManager
 
 manager = MCPManager()
 
+DEBUG = os.getenv("DEBUG", "false").lower() in ("true", "1", "t", "yes")
+
 @register_tool()
 async def worker(goal, tools, work_dir, cache_messages=None):
     """
@@ -32,6 +34,13 @@ async def worker(goal, tools, work_dir, cache_messages=None):
     Returns:
         str: 当任务成功完成时，返回字符串 "任务已完成"。
     """
+    if DEBUG:
+        import sys
+        log_file = open(Path(work_dir) / ".beswarm" / "history.log", "a")
+        log_file.write(f"========== {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} ==========\n")
+        sys.stdout = log_file
+        sys.stderr = log_file
+
     start_time = datetime.now()
     os.chdir(Path(work_dir).absolute())
     finish_flag = 0
@@ -93,7 +102,7 @@ async def worker(goal, tools, work_dir, cache_messages=None):
         "api_url": os.getenv("BASE_URL"),
         "engine": os.getenv("MODEL"),
         "system_prompt": instruction_system_prompt.format(os_version=platform.platform(), tools_list=tools_json, workspace_path=work_dir, current_time=datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
-        "print_log": False,
+        "print_log": DEBUG,
         # "max_tokens": 4000,
         "temperature": 0.7,
         "use_plugins": False,
@@ -225,6 +234,13 @@ async def worker(goal, tools, work_dir, cache_messages=None):
     return "任务已完成"
 
 async def worker_gen(goal, tools, work_dir, cache_messages=None):
+    if DEBUG:
+        import sys
+        log_file = open(Path(work_dir) / ".beswarm" / "history.log", "a")
+        log_file.write(f"========== {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} ==========\n")
+        sys.stdout = log_file
+        sys.stderr = log_file
+
     start_time = datetime.now()
     os.chdir(Path(work_dir).absolute())
     finish_flag = 0
@@ -286,7 +302,7 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
         "api_url": os.getenv("BASE_URL"),
         "engine": os.getenv("MODEL"),
         "system_prompt": instruction_system_prompt.format(os_version=platform.platform(), tools_list=tools_json, workspace_path=work_dir, current_time=datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
-        "print_log": False,
+        "print_log": DEBUG,
         # "max_tokens": 4000,
         "temperature": 0.7,
         "use_plugins": False,

{beswarm-0.2.9 → beswarm-0.2.10}/beswarm.egg-info/PKG-INFO

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

{beswarm-0.2.9 → beswarm-0.2.10}/pyproject.toml

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