jarvis-ai-assistant 0.1.179__py3-none-any.whl → 0.1.181__py3-none-any.whl

jarvis/jarvis_tools/execute_script.py

@@ -61,6 +61,30 @@ class ScriptTool:
         "g++": "cpp",
     }
 
+    def get_display_output(self, file_path: str) -> str:
+        """消除控制字符，得到用户实际看到的文本，去除script命令首尾行"""
+        import re
+        # 读取文件内容并尝试多种编码
+        with open(file_path, 'rb') as f:
+            data = f.read()
+
+        import pyte
+        screen = pyte.Screen(300, 100000)
+        stream = pyte.ByteStream(screen)
+        stream.feed(data)
+        
+        # 清理每行右侧空格，并过滤空行
+        cleaned = []
+        cleaned = []
+        for y in range(screen.lines):
+            line = screen.buffer[y]
+            stripped = "".join(
+                char.data for char in line.values()
+            ).rstrip()
+            if stripped:
+                cleaned.append(stripped)
+        return "\n".join(cleaned[1:-1])
+
     def _execute_script_with_interpreter(self, interpreter: str, script_content: str) -> Dict[str, Any]:
         """Execute a script with the specified interpreter
         
@@ -90,13 +114,8 @@ class ScriptTool:
                 
                 # Read and process output file
                 try:
-                    with open(output_file, 'r', encoding='utf-8', errors='ignore') as f:
-                        output = f.read()
-                        # Remove header and footer added by script command (if any)
-                        if output:
-                            lines = output.splitlines()
-                            if len(lines) > 2:
-                                output = "\n".join(lines[1:-1])
+                    # 消除控制字符，得到用户实际看到的文本
+                    output = self.get_display_output(output_file)
                 except Exception as e:
                     output = f"读取输出文件失败: {str(e)}"
                     
@@ -150,4 +169,8 @@ class ScriptTool:
                 "success": False,
                 "stdout": "",
                 "stderr": str(e)
-            } 
+            } 
+        
+if __name__ == "__main__":
+    script_tool = ScriptTool()
+    print(script_tool.get_display_output("/home/wangmaobin/code/Jarvis/a.txt"))

jarvis/jarvis_tools/registry.py

@@ -198,7 +198,7 @@ class ToolRegistry(OutputHandlerProtocol):
         stats_file = Path(get_data_dir()) / "tool_stat.yaml"
         try:
             with open(stats_file, "w", encoding="utf-8") as f:
-                yaml.safe_dump(stats, f)
+                yaml.safe_dump(stats, f, allow_unicode=True)
         except Exception as e:
             PrettyOutput.print(
                 f"保存工具调用统计失败: {str(e)}", OutputType.WARNING
@@ -229,14 +229,36 @@ class ToolRegistry(OutputHandlerProtocol):
         }
 
     def _load_mcp_tools(self) -> None:
-        """从jarvis_data/tools/mcp加载工具"""
+        """加载MCP工具，优先从配置获取，其次从目录扫描"""
+        from jarvis.jarvis_utils.config import get_mcp_config
+        
+        # 优先从配置获取MCP工具配置
+        mcp_configs = get_mcp_config()
+        if mcp_configs:
+            for config in mcp_configs:
+                self.register_mcp_tool_by_config(config)
+            return
+
+        # 如果配置中没有，则扫描目录
         mcp_tools_dir = Path(get_data_dir()) / "mcp"
         if not mcp_tools_dir.exists():
             return
 
+        # 添加警告信息
+        PrettyOutput.print(
+            "警告: 从文件目录加载MCP工具的方式将在未来版本中废弃，请尽快迁移到JARVIS_MCP配置方式",
+            OutputType.WARNING
+        )
+
         # 遍历目录中的所有.yaml文件
         for file_path in mcp_tools_dir.glob("*.yaml"):
-            self.register_mcp_tool_by_file(str(file_path))
+            try:
+                config = yaml.safe_load(open(file_path, "r", encoding="utf-8"))
+                self.register_mcp_tool_by_config(config)
+            except Exception as e:
+                PrettyOutput.print(
+                    f"文件 {file_path} 加载失败: {str(e)}", OutputType.WARNING
+                )
 
     def _load_builtin_tools(self) -> None:
         """从内置工具目录加载工具"""
@@ -264,29 +286,26 @@ class ToolRegistry(OutputHandlerProtocol):
 
             self.register_tool_by_file(str(file_path))
 
-    def register_mcp_tool_by_file(self, file_path: str) -> bool:
-        """从指定文件加载并注册工具
+    def register_mcp_tool_by_config(self, config: Dict[str, Any]) -> bool:
+        """从配置字典加载并注册工具
 
         参数:
-            file_path: 工具文件的路径
+            config: MCP工具配置字典
 
         返回:
             bool: 工具是否加载成功
         """
         try:
-            config = yaml.safe_load(open(file_path, "r", encoding="utf-8"))
             if "type" not in config:
-                PrettyOutput.print(f"文件 {file_path} 缺少type字段", OutputType.WARNING)
+                PrettyOutput.print(f"配置{config.get('name', '')}缺少type字段", OutputType.WARNING)
                 return False
 
             # 检查enable标志
             if not config.get("enable", True):
-                PrettyOutput.print(
-                    f"文件 {file_path} 已禁用(enable=false)，跳过注册", OutputType.INFO
-                )
+                PrettyOutput.print(f"MCP配置{config.get('name', '')}已禁用(enable=false)，跳过注册", OutputType.INFO)
                 return False
 
-            name = config.get("name", Path(file_path).stem)
+            name = config.get("name", "mcp")
 
             # 注册资源工具
             def create_resource_list_func(client: McpClient):
@@ -296,11 +315,11 @@ class ToolRegistry(OutputHandlerProtocol):
                     args.pop("want", None)
                     ret = client.get_resource_list()
                     PrettyOutput.print(
-                        f"MCP {name} 资源列表:\n{yaml.safe_dump(ret)}", OutputType.TOOL
+                        f"MCP {name} 资源列表:\n{yaml.safe_dump(ret, allow_unicode=True)}", OutputType.TOOL
                     )
                     return {
                         "success": True,
-                        "stdout": yaml.safe_dump(ret),
+                        "stdout": yaml.safe_dump(ret, allow_unicode=True),
                         "stderr": "",
                     }
 
@@ -319,7 +338,7 @@ class ToolRegistry(OutputHandlerProtocol):
                         }
                     ret = client.get_resource(args["uri"])
                     PrettyOutput.print(
-                        f"MCP {name} 获取资源:\n{yaml.safe_dump(ret)}", OutputType.TOOL
+                        f"MCP {name} 获取资源:\n{yaml.safe_dump(ret, allow_unicode=True)}", OutputType.TOOL
                     )
                     return ret
 
@@ -332,7 +351,7 @@ class ToolRegistry(OutputHandlerProtocol):
                     args.pop("want", None)
                     ret = client.execute(tool_name, args)
                     PrettyOutput.print(
-                        f"MCP {name} {tool_name} 执行结果:\n{yaml.safe_dump(ret)}",
+                        f"MCP {name} {tool_name} 执行结果:\n{yaml.safe_dump(ret, allow_unicode=True)}",
                         OutputType.TOOL,
                     )
                     return ret
@@ -341,26 +360,18 @@ class ToolRegistry(OutputHandlerProtocol):
 
             if config["type"] == "stdio":
                 if "command" not in config:
-                    PrettyOutput.print(
-                        f"文件 {file_path} 缺少command字段", OutputType.WARNING
-                    )
+                    PrettyOutput.print(f"配置{config.get('name', '')}缺少command字段", OutputType.WARNING)
                     return False
             elif config["type"] == "sse":
                 if "base_url" not in config:
-                    PrettyOutput.print(
-                        f"文件 {file_path} 缺少base_url字段", OutputType.WARNING
-                    )
+                    PrettyOutput.print(f"配置{config.get('name', '')}缺少base_url字段", OutputType.WARNING)
                     return False
             elif config["type"] == "streamable":
                 if "base_url" not in config:
-                    PrettyOutput.print(
-                        f"文件 {file_path} 缺少base_url字段", OutputType.WARNING
-                    )
+                    PrettyOutput.print(f"配置{config.get('name', '')}缺少base_url字段", OutputType.WARNING)
                     return False
             else:
-                PrettyOutput.print(
-                    f"文件 {file_path} 类型错误: {config['type']}", OutputType.WARNING
-                )
+                PrettyOutput.print(f"不支持的MCP客户端类型: {config['type']}", OutputType.WARNING)
                 return False
 
             # 创建MCP客户端
@@ -376,14 +387,11 @@ class ToolRegistry(OutputHandlerProtocol):
             # 获取工具信息
             tools = mcp_client.get_tool_list()
             if not tools:
-                PrettyOutput.print(
-                    f"从 {file_path} 获取工具列表失败", OutputType.WARNING
-                )
+                PrettyOutput.print(f"从配置{config.get('name', '')}获取工具列表失败", OutputType.WARNING)
                 return False
 
             # 注册每个工具
             for tool in tools:
-
                 # 注册工具
                 self.register_tool(
                     name=f"{name}.tool_call.{tool['name']}",
@@ -417,9 +425,7 @@ class ToolRegistry(OutputHandlerProtocol):
             return True
 
         except Exception as e:
-            PrettyOutput.print(
-                f"文件 {file_path} 加载失败: {str(e)}", OutputType.WARNING
-            )
+            PrettyOutput.print(f"MCP配置{config.get('name', '')}加载失败: {str(e)}", OutputType.WARNING)
             return False
 
     def register_tool_by_file(self, file_path: str) -> bool:

jarvis/jarvis_utils/builtin_replace_map.py

@@ -8,27 +8,6 @@
 from jarvis.jarvis_utils.tag import ct, ot
 
 BUILTIN_REPLACE_MAP = {
-    "CodeBase": {
-        "append": True,
-        "template": f"""
-请使用ask_codebase工具查询代码库，必须严格遵守以下工具调用格式：
-
-{ot("TOOL_CALL")}
-want: 想要从执行结果中获取到的信息
-name: ask_codebase
-arguments:
-    question: "与xxx功能相关的文件有哪些？"
-{ct("TOOL_CALL")}
-
-可以使用的提问格式包括：
-1. 与xxx功能相关的文件有哪些？
-2. 要实现xxx，应该要修改哪些文件？
-3. xxx功能是怎么实现的？
-4. xxx模块的入口函数是什么？
-5. xxx功能的测试用例在哪里？
-""",
-        "description": "查询代码库"
-    },
     "Web": {
         "append": True,
         "template": f"""
@@ -50,27 +29,6 @@ arguments:
 """,
         "description": "网页搜索"
     },
-    "Methodology": {
-        "append": True,
-        "template": f"""
-请使用find_methodology工具查找相关方法论，必须严格遵守以下工具调用格式：
-
-{ot("TOOL_CALL")}
-want: 想要从执行结果中获取到的信息
-name: find_methodology
-arguments:
-    query: "关于xxx的方法论有哪些？"
-{ct("TOOL_CALL")}
-
-可以使用的提问格式包括：
-1. 关于xxx的方法论有哪些？
-2. 如何解决xxx问题？
-3. xxx的最佳实践是什么？
-4. 处理xxx的标准流程是什么？
-5. 实现xxx的参考方案有哪些？
-""",
-        "description": "查找相关方法论"
-    },
     "Plan": {
         "append": True,
         "template": f"""
@@ -111,12 +69,6 @@ code_plan工具将：
         "append": False,
         "template": f"""
 请使用工具在当前目录下查找与以下功能相关的文件：
-"""
-    },
-    "FindMethodology": {
-        "append": False,
-        "template": f"""
-请使用find_methodology工具查找相关方法论：
 """
     },
     "Dev": {

jarvis/jarvis_utils/config.py

@@ -2,10 +2,30 @@
 import os
 from functools import lru_cache
 
+from typing import Any, Dict, List
+
+
 import yaml
 
+
 from jarvis.jarvis_utils.builtin_replace_map import BUILTIN_REPLACE_MAP
 
+
+# 全局环境变量存储
+
+GLOBAL_CONFIG_DATA: Dict[str, Any] = {}
+
+
+def set_global_env_data(env_data: Dict[str, Any]) -> None:
+    """设置全局环境变量数据"""
+    global GLOBAL_CONFIG_DATA
+    GLOBAL_CONFIG_DATA = env_data
+
+def set_config(key: str, value: Any) -> None:
+    """设置配置"""
+    GLOBAL_CONFIG_DATA[key] = value
+
+
 """配置管理模块。
 
 该模块提供了获取Jarvis系统各种配置设置的函数。
@@ -19,7 +39,7 @@ def get_git_commit_prompt() -> str:
     返回:
         str: Git提交信息生成提示模板，如果未配置则返回空字符串
     """
-    return os.getenv("JARVIS_GIT_COMMIT_PROMPT", "")
+    return GLOBAL_CONFIG_DATA.get("JARVIS_GIT_COMMIT_PROMPT", "")
 
 # 输出窗口预留大小
 INPUT_WINDOW_REVERSE_SIZE = 2048
@@ -29,16 +49,27 @@ def get_replace_map() -> dict:
     """
     获取替换映射表。
     
-    从数据目录下的replace_map.yaml文件中读取替换映射表，
+    优先使用GLOBAL_CONFIG_DATA['JARVIS_REPLACE_MAP']的配置，
+    如果没有则从数据目录下的replace_map.yaml文件中读取替换映射表，
     如果文件不存在则返回内置替换映射表。
     
     返回:
         dict: 合并后的替换映射表字典(内置+文件中的映射表)
     """
+    if 'JARVIS_REPLACE_MAP' in GLOBAL_CONFIG_DATA:
+        return {**BUILTIN_REPLACE_MAP, **GLOBAL_CONFIG_DATA['JARVIS_REPLACE_MAP']}
+        
     replace_map_path = os.path.join(get_data_dir(), 'replace_map.yaml')
     if not os.path.exists(replace_map_path):
         return BUILTIN_REPLACE_MAP.copy()
     
+    from jarvis.jarvis_utils.output import PrettyOutput, OutputType
+    PrettyOutput.print(
+        "警告：使用replace_map.yaml进行配置的方式已被弃用，将在未来版本中移除。"
+        "请迁移到使用GLOBAL_CONFIG_DATA中的JARVIS_REPLACE_MAP配置。",
+        output_type=OutputType.WARNING
+    )
+    
     with open(replace_map_path, 'r', encoding='utf-8', errors='ignore') as file:
         file_map = yaml.safe_load(file) or {}
         return {**BUILTIN_REPLACE_MAP, **file_map}
@@ -50,7 +81,7 @@ def get_max_token_count() -> int:
     返回:
         int: 模型能处理的最大token数量。
     """
-    return int(os.getenv('JARVIS_MAX_TOKEN_COUNT', '960000'))
+    return int(GLOBAL_CONFIG_DATA.get('JARVIS_MAX_TOKEN_COUNT', '960000'))
 
 def get_max_input_token_count() -> int:
     """
@@ -59,7 +90,7 @@ def get_max_input_token_count() -> int:
     返回:
         int: 模型能处理的最大输入token数量。
     """
-    return int(os.getenv('JARVIS_MAX_INPUT_TOKEN_COUNT', '32000'))
+    return int(GLOBAL_CONFIG_DATA.get('JARVIS_MAX_INPUT_TOKEN_COUNT', '32000'))
 
 
 def is_auto_complete() -> bool:
@@ -69,7 +100,7 @@ def is_auto_complete() -> bool:
     返回：
         bool: 如果启用了自动补全则返回True，默认为False
     """
-    return os.getenv('JARVIS_AUTO_COMPLETE', 'false') == 'true'
+    return GLOBAL_CONFIG_DATA.get('JARVIS_AUTO_COMPLETE', False) == True
 
 
 def get_shell_name() -> str:
@@ -79,7 +110,7 @@ def get_shell_name() -> str:
     返回：
         str: Shell名称（例如bash, zsh），默认为bash
     """
-    shell_path = os.getenv('SHELL', '/bin/bash')
+    shell_path = GLOBAL_CONFIG_DATA.get('SHELL', '/bin/bash')
     return os.path.basename(shell_path)
 def get_normal_platform_name() -> str:
     """
@@ -88,7 +119,7 @@ def get_normal_platform_name() -> str:
     返回：
         str: 平台名称，默认为'yuanbao'
     """
-    return os.getenv('JARVIS_PLATFORM', 'yuanbao')
+    return GLOBAL_CONFIG_DATA.get('JARVIS_PLATFORM', 'yuanbao')
 def get_normal_model_name() -> str:
     """
     获取正常操作的模型名称。
@@ -96,7 +127,7 @@ def get_normal_model_name() -> str:
     返回：
         str: 模型名称，默认为'deep_seek'
     """
-    return os.getenv('JARVIS_MODEL', 'deep_seek_v3')
+    return GLOBAL_CONFIG_DATA.get('JARVIS_MODEL', 'deep_seek_v3')
 
 
 def get_thinking_platform_name() -> str:
@@ -106,7 +137,7 @@ def get_thinking_platform_name() -> str:
     返回：
         str: 平台名称，默认为'yuanbao'
     """
-    return os.getenv('JARVIS_THINKING_PLATFORM', os.getenv('JARVIS_PLATFORM', 'yuanbao'))
+    return GLOBAL_CONFIG_DATA.get('JARVIS_THINKING_PLATFORM', GLOBAL_CONFIG_DATA.get('JARVIS_PLATFORM', 'yuanbao'))
 def get_thinking_model_name() -> str:
     """
     获取思考操作的模型名称。
@@ -114,7 +145,7 @@ def get_thinking_model_name() -> str:
     返回：
         str: 模型名称，默认为'deep_seek'
     """
-    return os.getenv('JARVIS_THINKING_MODEL', os.getenv('JARVIS_MODEL', 'deep_seek'))
+    return GLOBAL_CONFIG_DATA.get('JARVIS_THINKING_MODEL', GLOBAL_CONFIG_DATA.get('JARVIS_MODEL', 'deep_seek'))
 
 def is_execute_tool_confirm() -> bool:
     """
@@ -123,7 +154,7 @@ def is_execute_tool_confirm() -> bool:
     返回：
         bool: 如果需要确认则返回True，默认为False
     """
-    return os.getenv('JARVIS_EXECUTE_TOOL_CONFIRM', 'false') == 'true'
+    return GLOBAL_CONFIG_DATA.get('JARVIS_EXECUTE_TOOL_CONFIRM', False) == True
 def is_confirm_before_apply_patch() -> bool:
     """
     检查应用补丁前是否需要确认。
@@ -131,7 +162,7 @@ def is_confirm_before_apply_patch() -> bool:
     返回：
         bool: 如果需要确认则返回True，默认为False
     """
-    return os.getenv('JARVIS_CONFIRM_BEFORE_APPLY_PATCH', 'true') == 'true'
+    return GLOBAL_CONFIG_DATA.get('JARVIS_CONFIRM_BEFORE_APPLY_PATCH', True) == True
 
 def get_max_tool_call_count() -> int:
     """
@@ -140,7 +171,7 @@ def get_max_tool_call_count() -> int:
     返回：
         int: 最大连续工具调用次数，默认为20
     """
-    return int(os.getenv('JARVIS_MAX_TOOL_CALL_COUNT', '20'))
+    return int(GLOBAL_CONFIG_DATA.get('JARVIS_MAX_TOOL_CALL_COUNT', '20'))
 
 
 def get_data_dir() -> str:
@@ -151,7 +182,7 @@ def get_data_dir() -> str:
         str: 数据目录路径，优先从JARVIS_DATA_PATH环境变量获取，
              如果未设置或为空，则使用~/.jarvis作为默认值
     """
-    data_path = os.getenv('JARVIS_DATA_PATH', '').strip()
+    data_path = GLOBAL_CONFIG_DATA.get('JARVIS_DATA_PATH', '').strip()
     if not data_path:
         return os.path.expanduser('~/.jarvis')
     return data_path
@@ -163,7 +194,7 @@ def get_auto_update() -> bool:
     返回：
         bool: 如果需要自动更新则返回True，默认为True
     """
-    return os.getenv('JARVIS_AUTO_UPDATE', 'true') == 'true'
+    return GLOBAL_CONFIG_DATA.get('JARVIS_AUTO_UPDATE', True) == True
 
 def get_max_big_content_size() -> int:
     """
@@ -172,7 +203,7 @@ def get_max_big_content_size() -> int:
     返回：
         int: 最大大内容大小
     """
-    return int(os.getenv('JARVIS_MAX_BIG_CONTENT_SIZE', '1024000'))
+    return int(GLOBAL_CONFIG_DATA.get('JARVIS_MAX_BIG_CONTENT_SIZE', '1024000'))
 
 def get_pretty_output() -> bool:
     """
@@ -181,7 +212,7 @@ def get_pretty_output() -> bool:
     返回：
         bool: 如果启用PrettyOutput则返回True，默认为True
     """
-    return os.getenv('JARVIS_PRETTY_OUTPUT', 'false') == 'true'
+    return GLOBAL_CONFIG_DATA.get('JARVIS_PRETTY_OUTPUT', False) == True
 
 def is_use_methodology() -> bool:
     """
@@ -190,7 +221,7 @@ def is_use_methodology() -> bool:
     返回：
         bool: 如果启用方法论则返回True，默认为True
     """
-    return os.getenv('JARVIS_USE_METHODOLOGY', 'false') == 'true'
+    return GLOBAL_CONFIG_DATA.get('JARVIS_USE_METHODOLOGY', True) == True
 
 def is_use_analysis() -> bool:
     """
@@ -199,7 +230,7 @@ def is_use_analysis() -> bool:
     返回：
         bool: 如果启用任务分析则返回True，默认为True
     """
-    return os.getenv('JARVIS_USE_ANALYSIS', 'false') == 'true'
+    return GLOBAL_CONFIG_DATA.get('JARVIS_USE_ANALYSIS', True) == True
 
 def is_print_prompt() -> bool:
     """
@@ -208,4 +239,14 @@ def is_print_prompt() -> bool:
     返回：
         bool: 如果打印提示则返回True，默认为True
     """
-    return os.getenv('JARVIS_PRINT_PROMPT', 'false') == 'true'
+    return GLOBAL_CONFIG_DATA.get('JARVIS_PRINT_PROMPT', False) == True
+
+
+def get_mcp_config() -> List[Dict[str, Any]]:
+    """
+    获取MCP配置列表。
+
+    返回:
+        List[Dict[str, Any]]: MCP配置项列表，如果未配置则返回空列表
+    """
+    return GLOBAL_CONFIG_DATA.get("JARVIS_MCP", [])