beswarm 0.2.10__tar.gz → 0.2.12__tar.gz

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

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

{beswarm-0.2.10 → beswarm-0.2.12}/beswarm/tools/__init__.py

@@ -1,6 +1,7 @@
 from .edit_file import edit_file
 from .worker import worker, worker_gen
 from .screenshot import save_screenshot_to_file
+from .request_input import request_admin_input
 
 from .search_arxiv import search_arxiv
 from .repomap import get_code_repo_map
@@ -47,4 +48,5 @@ __all__ = [
     "search_web",
     "save_screenshot_to_file",
     "set_readonly_path",
+    "request_admin_input",
 ]

beswarm-0.2.12/beswarm/tools/request_input.py

@@ -0,0 +1,21 @@
+from ..aient.src.aient.plugins import register_tool
+
+@register_tool()
+def request_admin_input(prompt: str) -> str:
+    """
+    当缺少必要信息（如 API Key、配置等）时，调用此工具向管理员发起请求以获取输入。
+
+    这个工具会向管理员显示一个提示问题 (`prompt`)，并同步等待管理员的文本输入。
+    它主要用于解决因信息不足而导致任务中断的情况，作为一个与外部操作人员交互的桥梁。
+
+    Args:
+        prompt (str): 需要向管理员显示的、用于请求信息的具体问题。
+
+    Returns:
+        str: 管理员输入的文本信息。
+    """
+    # 打印一个清晰的提示，告知管理员需要提供信息
+    print(f"\n[❗️] 智能体需要管理员的帮助来继续执行任务: {prompt}")
+    # 使用 input() 函数来捕获管理员的输入
+    admin_input = input("请管理员输入 > ")
+    return admin_input.strip()

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

@@ -15,8 +15,6 @@ 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):
     """
@@ -34,6 +32,7 @@ async def worker(goal, tools, work_dir, cache_messages=None):
     Returns:
         str: 当任务成功完成时，返回字符串 "任务已完成"。
     """
+    DEBUG = os.getenv("DEBUG", "false").lower() in ("true", "1", "t", "yes")
     if DEBUG:
         import sys
         log_file = open(Path(work_dir) / ".beswarm" / "history.log", "a")
@@ -234,6 +233,7 @@ async def worker(goal, tools, work_dir, cache_messages=None):
     return "任务已完成"
 
 async def worker_gen(goal, tools, work_dir, cache_messages=None):
+    DEBUG = os.getenv("DEBUG", "false").lower() in ("true", "1", "t", "yes")
     if DEBUG:
         import sys
         log_file = open(Path(work_dir) / ".beswarm" / "history.log", "a")

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

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

{beswarm-0.2.10 → beswarm-0.2.12}/beswarm.egg-info/SOURCES.txt

@@ -138,6 +138,7 @@ beswarm/tools/click.py
 beswarm/tools/edit_file.py
 beswarm/tools/planner.py
 beswarm/tools/repomap.py
+beswarm/tools/request_input.py
 beswarm/tools/screenshot.py
 beswarm/tools/search_arxiv.py
 beswarm/tools/search_web.py

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

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