beswarm 0.2.10__py3-none-any.whl → 0.2.12__py3-none-any.whl

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/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/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.dist-info → beswarm-0.2.12.dist-info}/METADATA

@@ -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.dist-info → beswarm-0.2.12.dist-info}/RECORD

@@ -125,16 +125,17 @@ beswarm/queries/tree-sitter-languages/ruby-tags.scm,sha256=vIidsCeE2A0vdFN18yXKq
 beswarm/queries/tree-sitter-languages/rust-tags.scm,sha256=9ljM1nzhfPs_ZTRw7cr2P9ToOyhGcKkCoN4_HPXSWi4,1451
 beswarm/queries/tree-sitter-languages/scala-tags.scm,sha256=UxQjz80JIrrJ7Pm56uUnQyThfmQNvwk7aQzPNypB-Ao,1761
 beswarm/queries/tree-sitter-languages/typescript-tags.scm,sha256=OMdCeedPiA24ky82DpgTMKXK_l2ySTuF2zrQ2fJAi9E,1253
-beswarm/tools/__init__.py,sha256=xTA2aKUlQj5sAglk71i5xV2Y32SH8T8vMb-0zoe4cBc,1160
+beswarm/tools/__init__.py,sha256=JkrgZVtTPeBfI6Y39-RJ1g9z_VyALP7skx3W0n0f9kE,1234
 beswarm/tools/click.py,sha256=TygaekCXTmU3fIu6Uom7ZcyzEgYMlCC_GX-5SmWHuLI,20762
 beswarm/tools/edit_file.py,sha256=iwWl7a8sTVq4vj0e1ny3H6UGcHfYnxALRGcLuk5hZS8,9155
 beswarm/tools/planner.py,sha256=lguBCS6kpwNPoXQvqH-WySabVubT82iyWOkJnjt6dXw,1265
 beswarm/tools/repomap.py,sha256=YsTPq5MXfn_Ds5begcvHDnY_Xp2d4jH-xmWqNMHnNHY,45239
+beswarm/tools/request_input.py,sha256=gXNAJPOJektMqxJVyzNTFOeMQ7xUkO-wWMYH-r2Rdwk,942
 beswarm/tools/screenshot.py,sha256=u6t8FCgW5YHJ_Oc4coo8e0F3wTusWE_-H8dFh1rBq9Q,1011
 beswarm/tools/search_arxiv.py,sha256=GpuIOYX8T0iRC-X-hmuR9AUJVn15WWZq864DaoC7BUc,8004
 beswarm/tools/search_web.py,sha256=w0T0aCqOVlb6Of5hn_TtpnrGXo6bMtw2aKZdkrYjycI,12069
-beswarm/tools/worker.py,sha256=iWhqema9bmn5dSDYqE1tKZ7HMCSZKYxUEklAQhc_Ow4,22919
-beswarm-0.2.10.dist-info/METADATA,sha256=EjUnjz_W9CzDce_JCxVObIQJdln9MD46XbOFVFECP3c,3847
-beswarm-0.2.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-beswarm-0.2.10.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.2.10.dist-info/RECORD,,
+beswarm/tools/worker.py,sha256=uUlaJ06Ys3iFVTAb9LEPd4XJ8HQ9tB1pyxUiy0wCe9o,22999
+beswarm-0.2.12.dist-info/METADATA,sha256=ysrKwf7d0b6x_ELfF8K_GSbrg71N6PLAogqCT710U-A,3847
+beswarm-0.2.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beswarm-0.2.12.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.2.12.dist-info/RECORD,,