beswarm 0.2.26__tar.gz → 0.2.28__tar.gz

{beswarm-0.2.26 → beswarm-0.2.28}/PKG-INFO

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

{beswarm-0.2.26 → beswarm-0.2.28}/beswarm/tools/__init__.py

@@ -1,5 +1,5 @@
 from .edit_file import edit_file
-from .worker import worker, worker_gen, create_task, get_task_result
+from .worker import worker, worker_gen
 from .screenshot import save_screenshot_to_file
 from .request_input import request_admin_input
 
@@ -7,6 +7,9 @@ from .search_arxiv import search_arxiv
 from .repomap import get_code_repo_map
 from .click import find_and_click_element, scroll_screen
 from .search_web import search_web
+from .taskmanager import create_task, resume_task, get_all_tasks_status, get_task_result
+from .completion import task_complete
+
 #显式导入 aient.plugins 中的所需内容
 from ..aient.src.aient.plugins import (
     excute_command,
@@ -50,5 +53,8 @@ __all__ = [
     "set_readonly_path",
     "request_admin_input",
     "create_task",
+    "resume_task",
+    "get_all_tasks_status",
     "get_task_result",
+    "task_complete",
 ]

beswarm-0.2.28/beswarm/tools/completion.py

@@ -0,0 +1,17 @@
+from ..aient.src.aient.plugins import register_tool
+
+@register_tool()
+def task_complete(message: str) -> str:
+    """
+    当任务完成时，调用此工具以返回最终结果。
+
+    这个工具接收一个表示任务完成信息的字符串，并将其直接返回。
+    它标志着一个任务的成功结束，并将最终的输出传递给用户或调用者。
+
+    Args:
+        message (str): 任务完成的信息或最终结果。
+
+    Returns:
+        str: 传入的任务完成信息。
+    """
+    return "<task_complete_message>" + message + "</task_complete_message>"

{beswarm-0.2.26 → beswarm-0.2.28}/beswarm/tools/taskmanager.py

@@ -1,9 +1,12 @@
-import os
+import ast
+import json
 import uuid
 import asyncio
 from enum import Enum
 from pathlib import Path
 
+from ..aient.src.aient.plugins import register_tool, registry
+
 class TaskStatus(Enum):
     """任务状态枚举"""
     PENDING = "PENDING"
@@ -19,7 +22,49 @@ class TaskManager:
     def __init__(self):
         self.tasks = {}  # 使用字典来存储任务，key是task_id, value是task对象
         self.results_queue = asyncio.Queue()
-        self.root_path = Path(os.getcwd())
+        self.root_path = None
+        self.tasks_cache = {}
+
+    def set_root_path(self, root_path):
+        if self.root_path:
+            return
+        self.root_path = Path(root_path)
+        self.cache_dir = self.root_path / ".beswarm"
+        self.task_cache_file = self.cache_dir / "tasks.json"
+        self.task_cache_file.touch(exist_ok=True)
+        self.read_tasks_cache()
+        self.set_task_cache("root_path", str(self.root_path))
+
+    def set_task_cache(self, *keys_and_value):
+        """
+        设置可嵌套的任务缓存。
+        接受无限个键和一个值，例如 set_task_cache('a', 'b', 'c', value)
+        会转换为 tasks_cache['a']['b']['c'] = value
+        """
+        if len(keys_and_value) < 2:
+            return  # 至少需要一个键和一个值
+
+        keys = keys_and_value[:-1]
+        value = keys_and_value[-1]
+
+        d = self.tasks_cache
+        # 遍历到倒数第二个键，确保路径存在
+        for key in keys[:-1]:
+            d = d.setdefault(key, {})
+
+        # 在最后一个键上设置值
+        d[keys[-1]] = value
+        self.save_tasks_cache()
+
+    def save_tasks_cache(self):
+        self.task_cache_file.write_text(json.dumps(self.tasks_cache, ensure_ascii=False, indent=4), encoding="utf-8")
+
+    def read_tasks_cache(self):
+        content = self.task_cache_file.read_text(encoding="utf-8")
+        try:
+            self.tasks_cache = json.loads(content) if content else {}
+        except json.JSONDecodeError:
+            raise ValueError("任务缓存文件格式错误")
 
     def create_tasks(self, task_coro, tasks_params):
         """
@@ -37,8 +82,27 @@ class TaskManager:
             coro = task_coro(**args)
             task_id = self.create_task(coro)
             task_ids.append(task_id)
+            self.set_task_cache(task_id, "args", args)
+            self.set_task_cache(task_id, "status", TaskStatus.RUNNING.value)
         return task_ids
 
+    def resume_task(self, task_id, task_coro, args):
+        """
+        恢复一个任务。
+        """
+        task = self.tasks_cache.get(task_id)
+        if not task:
+            return TaskStatus.NOT_FOUND
+
+        coro = task_coro(**args)
+        task_id = self.create_task(coro)
+        self.set_task_cache(task_id, "args", args)
+        self.set_task_cache(task_id, "status", TaskStatus.RUNNING.value)
+        print(f"任务已恢复: ID={task_id}, Name={task_id}")
+        print(f"args: {args}")
+        print(f"self.tasks_cache: {json.dumps(self.tasks_cache, ensure_ascii=False, indent=4)}")
+        return task_id
+
     def create_task(self, coro):
         """
         创建并注册一个新任务。
@@ -108,14 +172,19 @@ class TaskManager:
             self.results_queue.put_nowait(
                 (task_id, TaskStatus.DONE, task.result())
             )
+            self.set_task_cache(task_id, "status", TaskStatus.DONE.value)
+            self.set_task_cache(task_id, "result", task.result())
         except asyncio.CancelledError:
             self.results_queue.put_nowait(
                 (task_id, TaskStatus.CANCELLED, None)
             )
+            self.set_task_cache(task_id, "status", TaskStatus.CANCELLED.value)
         except Exception as e:
             self.results_queue.put_nowait(
                 (task_id, TaskStatus.ERROR, e)
             )
+            self.set_task_cache(task_id, "status", TaskStatus.ERROR.value)
+            self.set_task_cache(task_id, "result", str(e))
 
     async def get_next_result(self):
         """
@@ -146,8 +215,78 @@ class TaskManager:
             # 如果任务ID不存在，则返回-1
             return -1
 
-    def set_root_path(self, root_path):
-        self.root_path = Path(root_path)
+
+task_manager = TaskManager()
+
+worker_fun = registry.tools["worker"]
+
+@register_tool()
+def create_task(goal, tools, work_dir):
+    """
+    启动一个子任务来自动完成指定的任务目标 (`goal`)。
+
+    这个子任务接收一个清晰的任务描述、一组可供调用的工具 (`tools`)，以及一个工作目录 (`work_dir`)。
+    它会结合可用的工具，自主规划并逐步执行必要的操作，直到最终完成指定的任务目标。
+    核心功能是根据输入的目标，驱动整个任务执行流程。
+    子任务下上文为空，因此需要细致的背景信息。
+
+    Args:
+        goal (str): 需要完成的具体任务目标描述。子任务将围绕此目标进行工作。必须清晰、具体。必须包含背景信息，完成指标等。写清楚什么时候算任务完成，同时交代清楚任务的背景信息，这个背景信息可以是需要读取的文件等一切有助于完成任务的信息。
+        tools (list[str]): 一个包含可用工具函数对象的列表。子任务在执行任务时可能会调用这些工具来与环境交互（例如读写文件、执行命令等）。
+        work_dir (str): 工作目录的绝对路径。子任务将在此目录上下文中执行操作。子任务的工作目录位置在主任务的工作目录的子目录。
+
+    Returns:
+        str: 当任务成功完成时，返回字符串 "任务已完成"。
+    """
+    tasks_params = [
+        {"goal": goal, "tools": ast.literal_eval(tools), "work_dir": work_dir, "cache_messages": True}
+    ]
+    task_ids = task_manager.create_tasks(worker_fun, tasks_params)
+    return task_ids
+
+@register_tool()
+def resume_task(task_id, goal):
+    """
+    恢复一个子任务。
+    """
+    if task_id not in task_manager.tasks_cache:
+        return f"任务 {task_id} 不存在"
+    tasks_params = task_manager.tasks_cache[task_id]["args"]
+    tasks_params["goal"] = goal
+    tasks_params["cache_messages"] = True
+    task_id = task_manager.resume_task(task_id, worker_fun, tasks_params)
+    return f"任务 {task_id} 已恢复"
+
+@register_tool()
+def get_all_tasks_status():
+    """
+    获取所有任务的状态。
+    子任务状态会持久化到磁盘，因此即使历史记录为空，之前的子任务仍然存在。
+
+    Returns:
+        str: 所有任务的状态。每个任务的id，状态，结果。
+    """
+    return task_manager.tasks_cache
+
+@register_tool()
+async def get_task_result():
+    """
+    等待并获取子任务的执行结果。如果需要等待子任务完成，请使用这个工具。一旦有任务完成，会自动获取结果。如果调用时没有任务完成，会等待直到有任务完成。
+
+    Returns:
+        str: 子任务的执行结果。
+    """
+    task_id, status, result = await task_manager.get_next_result()
+
+    unfinished_tasks = [task_id for task_id, task in task_manager.tasks_cache.items() if task_id != "root_path" and task.get("status") != "DONE"]
+    text = "".join([
+        f"Task ID: {task_id}\n",
+        f"Status: {status.value}\n",
+        f"Result: {result}\n\n",
+        f"There are {len(unfinished_tasks)} unfinished tasks, unfinished task ids: {unfinished_tasks}" if unfinished_tasks else "All tasks are finished.",
+    ])
+
+    return text
 
 async def main():
     manager = TaskManager()

{beswarm-0.2.26 → beswarm-0.2.28}/beswarm/tools/worker.py

@@ -1,7 +1,6 @@
 import os
 import re
 import sys
-import ast
 import copy
 import json
 import difflib
@@ -27,50 +26,14 @@ from ..aient.src.aient.plugins import register_tool, get_function_call_list, reg
 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
 from ..bemcp.bemcp import MCPClient, convert_tool_format, MCPManager
-from .taskmanager import TaskManager
 
 manager = MCPManager()
-task_manager = TaskManager()
-
-@register_tool()
-def create_task(goal, tools, work_dir):
-    """
-    启动一个子任务来自动完成指定的任务目标 (`goal`)。
-
-    这个子任务接收一个清晰的任务描述、一组可供调用的工具 (`tools`)，以及一个工作目录 (`work_dir`)。
-    它会结合可用的工具，自主规划并逐步执行必要的操作，直到最终完成指定的任务目标。
-    核心功能是根据输入的目标，驱动整个任务执行流程。
-    子任务下上文为空，因此需要细致的背景信息。
-
-    Args:
-        goal (str): 需要完成的具体任务目标描述。子任务将围绕此目标进行工作。必须清晰、具体。必须包含背景信息，完成指标等。写清楚什么时候算任务完成，同时交代清楚任务的背景信息，这个背景信息可以是需要读取的文件等一切有助于完成任务的信息。
-        tools (list[str]): 一个包含可用工具函数对象的列表。子任务在执行任务时可能会调用这些工具来与环境交互（例如读写文件、执行命令等）。
-        work_dir (str): 工作目录的绝对路径。子任务将在此目录上下文中执行操作。子任务的工作目录位置在主任务的工作目录的子目录。
-
-    Returns:
-        str: 当任务成功完成时，返回字符串 "任务已完成"。
-    """
-    tasks_params = [
-        {"goal": goal, "tools": ast.literal_eval(tools), "work_dir": work_dir, "cache_messages": True}
-    ]
-    task_ids = task_manager.create_tasks(worker, tasks_params)
-    return task_ids
-
-@register_tool()
-async def get_task_result():
-    """
-    等待并获取子任务的执行结果。
-
-    Returns:
-        str: 子任务的执行结果。
-    """
-    task_id, status, result = await task_manager.get_next_result()
-    return result
 
 @register_tool()
 async def worker(goal, tools, work_dir, cache_messages=None):
     cache_dir = Path(work_dir) / ".beswarm"
     cache_dir.mkdir(parents=True, exist_ok=True)
+    task_manager.set_root_path(work_dir)
     cache_file = cache_dir / "work_agent_conversation_history.json"
     if not cache_file.exists():
         cache_file.write_text("[]", encoding="utf-8")
@@ -101,6 +64,8 @@ async def worker(goal, tools, work_dir, cache_messages=None):
         tools += mcp_tools_name
 
     tools = [item for item in tools if not isinstance(item, dict)]
+    if "task_complete" not in tools:
+        tools.append("task_complete")
 
     tools_json = [value for _, value in get_function_call_list(tools).items()]
     work_agent_system_prompt = worker_system_prompt.format(
@@ -157,7 +122,7 @@ async def worker(goal, tools, work_dir, cache_messages=None):
                     f"任务目标: {goal}\n\n",
                     f"任务目标新变化：\n{goal_diff}\n\n" if goal_diff else "",
                     "在 tag <work_agent_conversation_start>...</work_agent_conversation_end> 之前的对话历史都是工作智能体的对话历史。\n\n",
-                    "根据以上对话历史和目标，请生成下一步指令。如果任务已完成，请回复'任务已完成'。\n\n",
+                    "根据以上对话历史和目标，请生成下一步指令。如果任务已完成，指示工作智能体调用task_complete工具。\n\n",
                 ])
             if last_instruction and 'fetch_gpt_response_stream HTTP Error' not in last_instruction:
                 instruction_prompt = (
@@ -168,6 +133,11 @@ async def worker(goal, tools, work_dir, cache_messages=None):
             # 让指令agent分析对话历史并生成新指令
             instruction_agent = chatgpt(**instruction_agent_config)
             conversation_history = copy.deepcopy(work_agent.conversation["default"])
+            if len(conversation_history) > 1 and conversation_history[-2]["role"] == "user" \
+            and "<task_complete_message>" in conversation_history[-2]["content"]:
+                task_complete_message = extract_xml_content(conversation_history[-2]["content"], "task_complete_message")
+                # del work_agent.conversation["default"][-4:]
+                return "<task_complete_message>" + task_complete_message + "</task_complete_message>"
 
             cache_file.write_text(json.dumps(conversation_history, ensure_ascii=False, indent=4), encoding="utf-8")
 
@@ -206,8 +176,7 @@ async def worker(goal, tools, work_dir, cache_messages=None):
                 print(f"任务结束时间: {end_time.strftime('%Y-%m-%d %H:%M:%S')}")
                 print(f"总用时: {total_time}")
                 raise Exception(f"The request body is too long, please try again.")
-            if "任务已完成" == next_instruction.strip():
-                break
+
             last_instruction = next_instruction
             next_instruction = extract_xml_content(next_instruction, "instructions")
             if not next_instruction:
@@ -235,18 +204,19 @@ async def worker(goal, tools, work_dir, cache_messages=None):
         return next_instruction
 
     need_instruction = True
+    result = None
     while True:
         next_instruction = ''
         if need_instruction:
             next_instruction = await instruction_agent_task()
 
             # 检查任务是否完成
-            if "任务已完成" in next_instruction and len(next_instruction) < 10:
+            if "<task_complete_message>" in next_instruction:
                 if finish_flag == 0:
                     finish_flag = 1
                     continue
                 elif finish_flag == 1:
-                    print("\n✅ 任务已完成！")
+                    result = extract_xml_content(next_instruction, "task_complete_message")
                     break
             else:
                 finish_flag = 0
@@ -262,15 +232,17 @@ async def worker(goal, tools, work_dir, cache_messages=None):
 
     end_time = datetime.now()
     total_time = end_time - start_time
+    print("\n✅ 任务已完成：", result)
     print(f"\n任务开始时间: {start_time.strftime('%Y-%m-%d %H:%M:%S')}")
     print(f"任务结束时间: {end_time.strftime('%Y-%m-%d %H:%M:%S')}")
     print(f"总用时: {total_time}")
     await manager.cleanup()
-    return "任务已完成"
+    return result
 
 async def worker_gen(goal, tools, work_dir, cache_messages=None):
     cache_dir = Path(work_dir) / ".beswarm"
     cache_dir.mkdir(parents=True, exist_ok=True)
+    task_manager.set_root_path(work_dir)
     cache_file = cache_dir / "work_agent_conversation_history.json"
     if not cache_file.exists():
         cache_file.write_text("[]", encoding="utf-8")
@@ -301,6 +273,8 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
         tools += mcp_tools_name
 
     tools = [item for item in tools if not isinstance(item, dict)]
+    if "task_complete" not in tools:
+        tools.append("task_complete")
 
     tools_json = [value for _, value in get_function_call_list(tools).items()]
     work_agent_system_prompt = worker_system_prompt.format(
@@ -357,7 +331,7 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
                     f"任务目标: {goal}\n\n",
                     f"任务目标新变化：\n{goal_diff}\n\n" if goal_diff else "",
                     "在 tag <work_agent_conversation_start>...</work_agent_conversation_end> 之前的对话历史都是工作智能体的对话历史。\n\n",
-                    "根据以上对话历史和目标，请生成下一步指令。如果任务已完成，请回复'任务已完成'。\n\n",
+                    "根据以上对话历史和目标，请生成下一步指令。如果任务已完成，指示工作智能体调用task_complete工具。\n\n",
                 ])
             if last_instruction and 'fetch_gpt_response_stream HTTP Error' not in last_instruction:
                 instruction_prompt = (
@@ -368,6 +342,11 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
             # 让指令agent分析对话历史并生成新指令
             instruction_agent = chatgpt(**instruction_agent_config)
             conversation_history = copy.deepcopy(work_agent.conversation["default"])
+            if len(conversation_history) > 1 and conversation_history[-2]["role"] == "user" \
+            and "<task_complete_message>" in conversation_history[-2]["content"]:
+                task_complete_message = extract_xml_content(conversation_history[-2]["content"], "task_complete_message")
+                # del work_agent.conversation["default"][-4:]
+                return "<task_complete_message>" + task_complete_message + "</task_complete_message>"
 
             cache_file.write_text(json.dumps(conversation_history, ensure_ascii=False, indent=4), encoding="utf-8")
 
@@ -406,8 +385,7 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
                 print(f"任务结束时间: {end_time.strftime('%Y-%m-%d %H:%M:%S')}")
                 print(f"总用时: {total_time}")
                 raise Exception(f"The request body is too long, please try again.")
-            if "任务已完成" == next_instruction.strip():
-                break
+
             last_instruction = next_instruction
             next_instruction = extract_xml_content(next_instruction, "instructions")
             if not next_instruction:
@@ -435,6 +413,7 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
         return next_instruction
 
     need_instruction = True
+    result = None
     while True:
         next_instruction = ''
         if need_instruction:
@@ -443,12 +422,12 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
             yield {"user": next_instruction}
 
             # 检查任务是否完成
-            if "任务已完成" in next_instruction and len(next_instruction) < 10:
+            if "<task_complete_message>" in next_instruction:
                 if finish_flag == 0:
                     finish_flag = 1
                     continue
                 elif finish_flag == 1:
-                    print("\n✅ 任务已完成！")
+                    result = extract_xml_content(next_instruction, "task_complete_message")
                     break
             else:
                 finish_flag = 0
@@ -465,7 +444,10 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
 
     end_time = datetime.now()
     total_time = end_time - start_time
+    print("\n✅ 任务已完成：", result)
     print(f"\n任务开始时间: {start_time.strftime('%Y-%m-%d %H:%M:%S')}")
     print(f"任务结束时间: {end_time.strftime('%Y-%m-%d %H:%M:%S')}")
     print(f"总用时: {total_time}")
-    await manager.cleanup()
+    await manager.cleanup()
+
+from .taskmanager import task_manager

{beswarm-0.2.26 → beswarm-0.2.28}/beswarm.egg-info/PKG-INFO

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

{beswarm-0.2.26 → beswarm-0.2.28}/beswarm.egg-info/SOURCES.txt

@@ -135,6 +135,7 @@ beswarm/queries/tree-sitter-languages/scala-tags.scm
 beswarm/queries/tree-sitter-languages/typescript-tags.scm
 beswarm/tools/__init__.py
 beswarm/tools/click.py
+beswarm/tools/completion.py
 beswarm/tools/edit_file.py
 beswarm/tools/planner.py
 beswarm/tools/repomap.py

{beswarm-0.2.26 → beswarm-0.2.28}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beswarm"
-version = "0.2.26"
+version = "0.2.28"
 description = "MAS"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -37,4 +37,4 @@ search = [
 
 [tool.setuptools.packages.find]
 where = ["."]
-include = ["beswarm*"]
+include = ["beswarm*"]