jarvis-ai-assistant 0.3.20__py3-none-any.whl → 0.3.21__py3-none-any.whl

jarvis/jarvis_tools/sub_code_agent.py

@@ -0,0 +1,194 @@
+# -*- coding: utf-8 -*-
+"""
+sub_code_agent 工具
+将子任务交给 CodeAgent 执行，并返回执行结果。
+
+约定：
+- 仅接收一个参数：task
+- 不依赖父 Agent，所有配置使用系统默认与全局变量
+- 子Agent必须自动完成(auto_complete=True)且需要summary(need_summary=True)
+"""
+from typing import Any, Dict, Optional
+import json
+
+from jarvis.jarvis_code_agent.code_agent import CodeAgent
+from jarvis.jarvis_utils.globals import delete_agent
+from jarvis.jarvis_code_agent import code_agent as code_agent_module
+
+
+class SubCodeAgentTool:
+    """
+    使用 CodeAgent 托管执行子任务，执行完立即清理内部 Agent。
+    - 不注册至全局
+    - 使用系统默认/全局配置
+    - 启用自动完成与总结
+    """
+
+    # 必须与文件名一致，供 ToolRegistry 自动注册
+    name = "sub_code_agent"
+    description = "将子任务交给 CodeAgent 执行，并返回执行结果（使用系统默认配置，自动完成并生成总结）。"
+    parameters = {
+        "type": "object",
+        "properties": {
+            "task": {
+                "type": "string",
+                "description": "要执行的子任务内容（必填）",
+            },
+            "background": {
+                "type": "string",
+                "description": "任务背景与已知信息（可选，将与任务一并提供给子Agent）",
+            },
+        },
+        "required": ["task"],
+    }
+
+    def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
+        """
+        执行代码子任务并返回结果（由 CodeAgent 托管执行）。
+        返回:
+          - success: 是否成功
+          - stdout: CodeAgent 执行结果（字符串；若为 None 则返回“任务执行完成”）
+          - stderr: 错误信息（如有）
+        """
+        try:
+            task: str = str(args.get("task", "")).strip()
+            if not task:
+                return {
+                    "success": False,
+                    "stdout": "",
+                    "stderr": "task 不能为空",
+                }
+
+            # 在模块级别打补丁，仅自动确认交互（允许用户输入）
+            def _auto_confirm(tip: str, default: bool = True) -> bool:
+                return default
+
+            code_agent_module.user_confirm = _auto_confirm
+
+            # 读取背景信息并组合任务
+            background: str = str(args.get("background", "")).strip()
+            enhanced_task = (
+                f"背景信息:\n{background}\n\n任务:\n{task}" if background else task
+            )
+
+            # 继承父Agent的模型组与工具使用集（用于覆盖默认值）
+            parent_agent = args.get("agent")
+            # 如未注入父Agent，尝试从全局获取当前或任一已注册Agent
+            if parent_agent is None:
+                try:
+                    from jarvis.jarvis_utils import globals as G  # 延迟导入避免循环
+
+                    curr = getattr(G, "current_agent_name", "")
+                    if curr:
+                        parent_agent = getattr(G, "global_agents", {}).get(curr)
+                    if parent_agent is None and getattr(G, "global_agents", {}):
+                        try:
+                            parent_agent = next(iter(G.global_agents.values()))
+                        except Exception:
+                            parent_agent = None
+                except Exception:
+                    parent_agent = None
+            model_group = None
+            use_tools: list[str] = []
+            try:
+                if parent_agent is not None:
+                    if getattr(parent_agent, "model", None):
+                        model_group = getattr(parent_agent.model, "model_group", None)
+                    parent_registry = parent_agent.get_tool_registry()
+                    if parent_registry:
+                        for t in parent_registry.get_all_tools():
+                            if isinstance(t, dict) and t.get("name"):
+                                use_tools.append(str(t["name"]))
+            except Exception:
+                pass
+
+            # 创建 CodeAgent：参数优先使用父Agent的配置（若可获取），否则使用默认
+            # 推断/继承 llm_type、need_summary、tool_group
+            llm_type = "normal"
+            tool_group = None
+            try:
+                if parent_agent is not None:
+                    llm_type = getattr(parent_agent, "llm_type", llm_type)
+                    tool_group = getattr(parent_agent, "tool_group", tool_group)
+            except Exception:
+                pass
+
+            # 依据父Agent已启用工具集，推导 append_tools（作为在 CodeAgent 基础工具上的增量）
+            append_tools = None
+            try:
+                base_tools = [
+                    "execute_script",
+                    "search_web",
+                    "ask_user",
+                    "read_code",
+                    "rewrite_file",
+                    "save_memory",
+                    "retrieve_memory",
+                    "clear_memory",
+                    "sub_code_agent",
+                ]
+                if use_tools:
+                    extras = [t for t in use_tools if t not in base_tools]
+                    append_tools = ",".join(extras) if extras else None
+            except Exception:
+                append_tools = None
+
+            try:
+                code_agent = CodeAgent(
+                    llm_type=llm_type,
+                    model_group=model_group,
+                    need_summary=True,
+                    append_tools=append_tools,
+                    tool_group=tool_group,
+                )
+            except SystemExit as se:
+                # 将底层 sys.exit 转换为工具错误，避免终止进程
+                return {
+                    "success": False,
+                    "stdout": "",
+                    "stderr": f"初始化 CodeAgent 失败（可能未配置 git 或当前非 git 仓库）: {se}",
+                }
+
+            # 子Agent需要自动完成
+            try:
+                code_agent.agent.auto_complete = True
+                # 同步父Agent工具使用集（如可用）
+                if use_tools:
+                    code_agent.agent.set_use_tools(use_tools)
+                # 同步父Agent的模型名称（如可用），以尽量保持平台与模型一致
+                if (
+                    parent_agent is not None
+                    and getattr(parent_agent, "model", None)
+                    and getattr(code_agent.agent, "model", None)
+                ):
+                    try:
+                        parent_model_name = parent_agent.model.name()  # type: ignore[attr-defined]
+                        if parent_model_name:
+                            code_agent.agent.model.set_model_name(parent_model_name)  # type: ignore[attr-defined]
+                    except Exception:
+                        pass
+            except Exception:
+                pass
+
+            # 执行子任务（无提交信息前后缀）
+            ret = code_agent.run(enhanced_task, prefix="", suffix="")
+            stdout = ret if isinstance(ret, str) and ret else "任务执行完成"
+
+            # 主动清理内部 Agent，避免污染父Agent的全局状态
+            try:
+                inner_agent = code_agent.agent
+                delete_agent(inner_agent.name)
+            except Exception:
+                pass
+
+            return {
+                "success": True,
+                "stdout": json.dumps({"result": stdout}, ensure_ascii=False, indent=2),
+                "stderr": "",
+            }
+        except Exception as e:
+            return {
+                "success": False,
+                "stdout": "",
+                "stderr": f"执行子任务失败: {str(e)}",
+            }

jarvis/jarvis_tools/virtual_tty.py

@@ -127,7 +127,9 @@ class VirtualTTYTool:
         try:
             if action == "launch":
                 if args.get("keys", "") != "":
-                    PrettyOutput.print("启动虚拟终端时，不能同时指定 keys 参数", OutputType.ERROR)
+                    PrettyOutput.print(
+                        "启动虚拟终端时，不能同时指定 keys 参数", OutputType.ERROR
+                    )
                     return {
                         "success": False,
                         "stdout": "",
@@ -136,7 +138,9 @@ class VirtualTTYTool:
 
                 result = self._launch_tty(agent, tty_id)
                 if not result["success"]:
-                    PrettyOutput.print(f"启动虚拟终端 [{tty_id}] 失败", OutputType.ERROR)
+                    PrettyOutput.print(
+                        f"启动虚拟终端 [{tty_id}] 失败", OutputType.ERROR
+                    )
                 return result
             elif action == "send_keys":
                 keys = args.get("keys", "").strip()
@@ -145,26 +149,34 @@ class VirtualTTYTool:
 
                 result = self._input_command(agent, tty_id, keys, timeout, add_enter)
                 if not result["success"]:
-                    PrettyOutput.print(f"发送按键序列到终端 [{tty_id}] 失败", OutputType.ERROR)
+                    PrettyOutput.print(
+                        f"发送按键序列到终端 [{tty_id}] 失败", OutputType.ERROR
+                    )
                 return result
             elif action == "output":
                 timeout = args.get("timeout", 5.0)  # 默认5秒超时
 
                 result = self._get_output(agent, tty_id, timeout)
                 if not result["success"]:
-                    PrettyOutput.print(f"获取终端 [{tty_id}] 输出失败", OutputType.ERROR)
+                    PrettyOutput.print(
+                        f"获取终端 [{tty_id}] 输出失败", OutputType.ERROR
+                    )
                 return result
             elif action == "close":
 
                 result = self._close_tty(agent, tty_id)
                 if not result["success"]:
-                    PrettyOutput.print(f"关闭虚拟终端 [{tty_id}] 失败", OutputType.ERROR)
+                    PrettyOutput.print(
+                        f"关闭虚拟终端 [{tty_id}] 失败", OutputType.ERROR
+                    )
                 return result
             elif action == "get_screen":
 
                 result = self._get_screen(agent, tty_id)
                 if not result["success"]:
-                    PrettyOutput.print(f"获取终端 [{tty_id}] 屏幕内容失败", OutputType.ERROR)
+                    PrettyOutput.print(
+                        f"获取终端 [{tty_id}] 屏幕内容失败", OutputType.ERROR
+                    )
                 return result
             elif action == "list":
 
@@ -230,8 +242,6 @@ class VirtualTTYTool:
                     except BlockingIOError:
                         continue
 
-
-
                 return {"success": True, "stdout": output, "stderr": ""}
 
         except Exception as e:
@@ -297,8 +307,6 @@ class VirtualTTYTool:
                 except _queue.Empty:
                     continue
 
-
-
             return {"success": True, "stdout": output, "stderr": ""}
 
         except Exception as e:
@@ -668,9 +676,9 @@ class VirtualTTYTool:
                         {
                             "id": tty_id,
                             "status": status,
-                            "pid": tty_data["process"].pid
-                            if tty_data["process"]
-                            else None,
+                            "pid": (
+                                tty_data["process"].pid if tty_data["process"] else None
+                            ),
                             "shell": tty_data["shell"],
                         }
                     )

jarvis/jarvis_utils/config.py

@@ -296,7 +296,11 @@ def get_tool_load_dirs() -> List[str]:
     返回:
         List[str]: 工具加载目录列表
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_TOOL_LOAD_DIRS", [])
+    return [
+        os.path.expanduser(os.path.expandvars(str(p)))
+        for p in GLOBAL_CONFIG_DATA.get("JARVIS_TOOL_LOAD_DIRS", [])
+        if p
+    ]
 
 
 def get_methodology_dirs() -> List[str]:
@@ -306,7 +310,11 @@ def get_methodology_dirs() -> List[str]:
     返回:
         List[str]: 方法论加载目录列表
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_METHODOLOGY_DIRS", [])
+    return [
+        os.path.expanduser(os.path.expandvars(str(p)))
+        for p in GLOBAL_CONFIG_DATA.get("JARVIS_METHODOLOGY_DIRS", [])
+        if p
+    ]
 
 
 def get_agent_definition_dirs() -> List[str]:
@@ -316,7 +324,11 @@ def get_agent_definition_dirs() -> List[str]:
     返回:
         List[str]: agent 定义加载目录列表
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_AGENT_DEFINITION_DIRS", [])
+    return [
+        os.path.expanduser(os.path.expandvars(str(p)))
+        for p in GLOBAL_CONFIG_DATA.get("JARVIS_AGENT_DEFINITION_DIRS", [])
+        if p
+    ]
 
 
 def get_multi_agent_dirs() -> List[str]:
@@ -326,7 +338,11 @@ def get_multi_agent_dirs() -> List[str]:
     返回:
         List[str]: multi_agent 加载目录列表
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_MULTI_AGENT_DIRS", [])
+    return [
+        os.path.expanduser(os.path.expandvars(str(p)))
+        for p in GLOBAL_CONFIG_DATA.get("JARVIS_MULTI_AGENT_DIRS", [])
+        if p
+    ]
 
 
 def get_roles_dirs() -> List[str]:
@@ -336,7 +352,11 @@ def get_roles_dirs() -> List[str]:
     返回:
         List[str]: roles 加载目录列表
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_ROLES_DIRS", [])
+    return [
+        os.path.expanduser(os.path.expandvars(str(p)))
+        for p in GLOBAL_CONFIG_DATA.get("JARVIS_ROLES_DIRS", [])
+        if p
+    ]
 
 
 def get_central_methodology_repo() -> str:
@@ -369,6 +389,16 @@ def is_print_prompt() -> bool:
     return GLOBAL_CONFIG_DATA.get("JARVIS_PRINT_PROMPT", False) == True
 
 
+def is_print_error_traceback() -> bool:
+    """
+    获取是否在错误输出时打印回溯调用链。
+
+    返回：
+        bool: 如果打印回溯则返回True，默认为False（不打印）
+    """
+    return GLOBAL_CONFIG_DATA.get("JARVIS_PRINT_ERROR_TRACEBACK", False) is True
+
+
 def is_force_save_memory() -> bool:
     """
     获取是否强制保存记忆。