jarvis-ai-assistant 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

jarvis/__init__.py

@@ -1,4 +1,4 @@
 # -*- coding: utf-8 -*-
 """Jarvis AI Assistant"""
 
-__version__ = "0.2.2"
+__version__ = "0.2.4"

jarvis/jarvis_agent/edit_file_handler.py

@@ -55,6 +55,11 @@ class EditFileHandler(OutputHandler):
         patches = self._parse_patches(response)
         if not patches:
             return False, "未找到有效的文件编辑指令"
+        
+        # 记录 edit_file 工具调用统计
+        from jarvis.jarvis_stats.stats import StatsManager
+        stats_manager = StatsManager()
+        stats_manager.increment("edit_file", group="tool")
 
         results = []
 

jarvis/jarvis_agent/jarvis.py

@@ -24,7 +24,7 @@ from jarvis.jarvis_tools.registry import ToolRegistry
 from jarvis.jarvis_utils.config import get_data_dir
 from jarvis.jarvis_utils.utils import init_env
 
-app = typer.Typer(help="Jarvis AI assistant")
+app = typer.Typer(help="Jarvis AI 助手")
 
 
 def _load_tasks() -> Dict[str, str]:
@@ -91,19 +91,13 @@ def _select_task(tasks: Dict[str, str]) -> str:
                 selected_task = tasks[task_names[choice - 1]]
                 PrettyOutput.print(f"将要执行任务:\n {selected_task}", OutputType.INFO)
                 # 询问是否需要补充信息
-                need_additional = user_confirm(
-                    "需要为此任务添加补充信息吗？", default=False
-                )
+                need_additional = user_confirm("需要为此任务添加补充信息吗？", default=False)
                 if need_additional:
                     additional_input = get_multiline_input("请输入补充信息：")
                     if additional_input:
-                        selected_task = (
-                            f"{selected_task}\n\n补充信息:\n{additional_input}"
-                        )
+                        selected_task = f"{selected_task}\n\n补充信息:\n{additional_input}"
                 return selected_task
-            PrettyOutput.print(
-                "无效的选择。请选择列表中的一个号码。", OutputType.WARNING
-            )
+            PrettyOutput.print("无效的选择。请选择列表中的一个号码。", OutputType.WARNING)
 
         except (KeyboardInterrupt, EOFError):
             return ""
@@ -121,7 +115,16 @@ def _handle_edit_mode(edit: bool, config_file: Optional[str]) -> None:
         if config_file
         else Path(os.path.expanduser("~/.jarvis/config.yaml"))
     )
-    editors = ["nvim", "vim", "vi"]
+    # 根据操作系统选择合适的编辑器
+    import platform
+
+    if platform.system() == "Windows":
+        # 优先级：终端工具 -> 代码编辑器 -> 通用文本编辑器
+        editors = ["nvim", "vim", "nano", "code", "notepad++", "notepad"]
+    else:
+        # 优先级：终端工具 -> 代码编辑器 -> 通用文本编辑器
+        editors = ["nvim", "vim", "vi", "nano", "emacs", "code", "gedit", "kate"]
+
     editor = next((e for e in editors if shutil.which(e)), None)
 
     if editor:
@@ -133,7 +136,7 @@ def _handle_edit_mode(edit: bool, config_file: Optional[str]) -> None:
             raise typer.Exit(code=1)
     else:
         PrettyOutput.print(
-            "No suitable editor found (nvim, vim, vi).", OutputType.ERROR
+            f"No suitable editor found. Tried: {', '.join(editors)}", OutputType.ERROR
         )
         raise typer.Exit(code=1)
 
@@ -186,30 +189,24 @@ def run_cli(
     llm_type: str = typer.Option(
         "normal",
         "--llm_type",
-        help="LLM type to use, choices are 'normal' and 'thinking'",
-    ),
-    task: Optional[str] = typer.Option(
-        None, "-t", "--task", help="Directly input task content from command line"
+        help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
+    task: Optional[str] = typer.Option(None, "-t", "--task", help="从命令行直接输入任务内容"),
     model_group: Optional[str] = typer.Option(
-        None, "--model_group", help="Model group to use, overriding config"
-    ),
-    config_file: Optional[str] = typer.Option(
-        None, "-f", "--config", help="Path to custom config file"
+        None, "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
     ),
+    config_file: Optional[str] = typer.Option(None, "-f", "--config", help="自定义配置文件路径"),
     restore_session: bool = typer.Option(
         False,
         "--restore-session",
-        help="Restore session from .jarvis/saved_session.json",
-    ),
-    edit: bool = typer.Option(
-        False, "-e", "--edit", help="Edit the configuration file"
+        help="从 .jarvis/saved_session.json 恢复会话",
     ),
+    edit: bool = typer.Option(False, "-e", "--edit", help="编辑配置文件"),
 ) -> None:
     """Jarvis AI assistant command-line interface."""
     if ctx.invoked_subcommand is not None:
         return
-        
+
     _handle_edit_mode(edit, config_file)
 
     init_env("欢迎使用 Jarvis AI 助手，您的智能助理已准备就绪！", config_file=config_file)

jarvis/jarvis_agent/main.py

@@ -10,7 +10,7 @@ from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import init_env
 
-app = typer.Typer(help="Jarvis AI assistant")
+app = typer.Typer(help="Jarvis AI 助手")
 
 
 def load_config(config_path: str) -> dict:
@@ -40,21 +40,21 @@ def load_config(config_path: str) -> dict:
 @app.command()
 def cli(
     config_file: Optional[str] = typer.Option(
-        None, "-f", "--config", help="Path to agent config file"
+        None, "-f", "--config", help="代理配置文件路径"
     ),
     agent_definition: Optional[str] = typer.Option(
-        None, "-c", "--agent_definition", help="Path to agent definition file"
+        None, "-c", "--agent_definition", help="代理定义文件路径"
     ),
     task: Optional[str] = typer.Option(
-        None, "-t", "--task", help="Initial task to execute"
+        None, "-t", "--task", help="初始任务内容"
     ),
     llm_type: str = typer.Option(
         "normal",
         "--llm_type",
-        help="LLM type to use, overriding config",
+        help="使用的LLM类型，覆盖配置文件中的设置",
     ),
     model_group: Optional[str] = typer.Option(
-        None, "--model_group", help="Model group to use, overriding config"
+        None, "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
     ),
 ):
     """Main entry point for Jarvis agent"""

jarvis/jarvis_agent/prompts.py

@@ -113,6 +113,31 @@ TASK_ANALYSIS_PROMPT = f"""<task_analysis>
            "stderr": f"操作失败: {{str(e)}}"
        }}
    ```
+4. **在工具中调用大模型**：如果工具需要调用大模型来完成子任务（例如，生成代码、分析文本等），为了避免干扰主对话流程，建议创建一个独立的大模型实例。
+   ```python
+    # 通过 agent 实例获取模型配置
+    agent = args.get("agent")
+    if not agent:
+        return {{"success": False, "stderr": "Agent not found."}}
+    
+    current_model = agent.model
+    platform_name = current_model.platform_name()
+    model_name = current_model.name()
+
+    # 创建独立的模型实例
+    from jarvis.jarvis_platform.registry import PlatformRegistry
+    llm = PlatformRegistry().create_platform(platform_name)
+    if not llm:
+        return {{"success": False, "stderr": f"Platform {{platform_name}} not found."}}
+    
+    llm.set_model_name(model_name)
+    llm.set_suppress_output(True) # 工具内的调用通常不需要流式输出
+
+    # 使用新实例调用大模型
+    PrettyOutput.print("正在执行子任务...", OutputType.INFO)
+    response = llm.chat_until_success("你的提示词")
+    PrettyOutput.print("子任务完成", OutputType.SUCCESS)
+   ```
 </tool_requirements>
 <methodology_requirements>
 方法论格式要求:
@@ -139,10 +164,7 @@ arguments:
     from jarvis.jarvis_utils.output import PrettyOutput, OutputType
     class 工具名称:
         name = "工具名称"
-        description = "Tool for text transformation"
-                Tool description
-        适用场景：1. 格式化文本; 2. 处理标题; 3. 标准化输出
-        \"\"\"
+        description = "Tool description"
         parameters = {{
             "type": "object",
             "properties": {{

jarvis/jarvis_code_agent/code_agent.py

@@ -38,7 +38,7 @@ from jarvis.jarvis_utils.input import get_multiline_input, user_confirm
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import get_loc_stats, init_env
 
-app = typer.Typer(help="Jarvis Code Agent")
+app = typer.Typer(help="Jarvis 代码助手")
 
 
 class CodeAgent:
@@ -54,6 +54,9 @@ class CodeAgent:
         need_summary: bool = True,
     ):
         self.root_dir = os.getcwd()
+
+        # 检测 git username 和 email 是否已设置
+        self._check_git_config()
         tool_registry = ToolRegistry()  # type: ignore
         tool_registry.use_tools(
             [
@@ -84,18 +87,18 @@ class CodeAgent:
 1. **项目分析**：分析项目结构，确定需修改的文件
 2. **需求分析**：理解需求意图，选择影响最小的实现方案
 3. **代码分析**：详细分析目标文件，禁止虚构现有代码
-   - 结构分析：优先使用 fd 命令或 find 工具快速定位文件和目录结构
-   - 内容搜索：优先使用 rg（ripgrep）进行函数、类、变量等内容的全文搜索，避免遗漏
-   - 依赖关系：如需分析依赖、调用关系，可结合 grep、ctags、pyan3 等工具辅助
+   - 结构分析：优先使用文件搜索工具快速定位文件和目录结构
+   - 内容搜索：优先使用全文搜索工具进行函数、类、变量等内容的搜索，避免遗漏
+   - 依赖关系：如需分析依赖、调用关系，可结合代码分析工具辅助
    - 代码阅读：使用 read_code 工具获取目标文件的完整内容或指定范围内容，禁止凭空假设代码
-   - 变更影响：如需分析变更影响范围，可结合 git diff、git log 等命令辅助判断
+   - 变更影响：如需分析变更影响范围，可结合版本控制工具辅助判断
    - 工具优先级：优先使用自动化工具，减少人工推断，确保分析结果准确
 4. **方案设计**：确定最小变更方案，保持代码结构
 5. **实施修改**：遵循"先读后写"原则，保持代码风格一致性
 
 ## 工具使用
-- 项目结构：优先使用fd命令查找文件
-- 代码搜索：优先使用rg进行内容搜索
+- 项目结构：优先使用文件搜索命令查找文件
+- 代码搜索：优先使用内容搜索工具
 - 代码阅读：优先使用read_code工具
 - 仅在命令行工具不足时使用专用工具
 
@@ -133,6 +136,52 @@ class CodeAgent:
 
         self.agent.set_after_tool_call_cb(self.after_tool_call_cb)
 
+    def _check_git_config(self) -> None:
+        """检查 git username 和 email 是否已设置，如果没有则提示并退出"""
+        try:
+            # 检查 git user.name
+            result = subprocess.run(
+                ["git", "config", "--get", "user.name"],
+                capture_output=True,
+                text=True,
+                check=False,
+            )
+            username = result.stdout.strip()
+
+            # 检查 git user.email
+            result = subprocess.run(
+                ["git", "config", "--get", "user.email"],
+                capture_output=True,
+                text=True,
+                check=False,
+            )
+            email = result.stdout.strip()
+
+            # 如果任一配置未设置，提示并退出
+            if not username or not email:
+                missing_configs = []
+                if not username:
+                    missing_configs.append(
+                        '  git config --global user.name "Your Name"'
+                    )
+                if not email:
+                    missing_configs.append(
+                        '  git config --global user.email "your.email@example.com"'
+                    )
+
+                message = "❌ Git 配置不完整\n\n请运行以下命令配置 Git：\n" + "\n".join(
+                    missing_configs
+                )
+                PrettyOutput.print(message, OutputType.WARNING)
+                sys.exit(1)
+
+        except FileNotFoundError:
+            PrettyOutput.print("❌ 未找到 git 命令，请先安装 Git", OutputType.ERROR)
+            sys.exit(1)
+        except Exception as e:
+            PrettyOutput.print(f"❌ 检查 Git 配置时出错: {str(e)}", OutputType.ERROR)
+            sys.exit(1)
+
     def _find_git_root(self) -> str:
         """查找并切换到git根目录
 
@@ -185,13 +234,138 @@ class CodeAgent:
         1. 查找git根目录
         2. 检查并更新.gitignore文件
         3. 处理未提交的修改
+        4. 配置git对换行符变化不敏感
         """
         print("🚀 正在初始化环境...")
         git_dir = self._find_git_root()
         self._update_gitignore(git_dir)
         self._handle_git_changes()
+        # 配置git对换行符变化不敏感
+        self._configure_line_ending_settings()
         print("✅ 环境初始化完成")
 
+    def _configure_line_ending_settings(self) -> None:
+        """配置git对换行符变化不敏感，只在当前设置与目标设置不一致时修改"""
+        target_settings = {
+            "core.autocrlf": "false",
+            "core.safecrlf": "false",
+            "core.whitespace": "cr-at-eol",  # 忽略行尾的CR
+        }
+
+        # 获取当前设置并检查是否需要修改
+        need_change = False
+        current_settings = {}
+        for key, target_value in target_settings.items():
+            result = subprocess.run(
+                ["git", "config", "--get", key], capture_output=True, text=True
+            )
+            current_value = result.stdout.strip()
+            current_settings[key] = current_value
+            if current_value != target_value:
+                need_change = True
+
+        if not need_change:
+            print("✅ git换行符敏感设置已符合要求")
+            return
+
+        PrettyOutput.print(
+            "⚠️ 即将修改git换行符敏感设置，这会影响所有文件的换行符处理方式",
+            OutputType.WARNING,
+        )
+        print("将进行以下设置：")
+        for key, value in target_settings.items():
+            current = current_settings.get(key, "未设置")
+            print(f"  {key}: {current} -> {value}")
+
+        if user_confirm("是否继续修改git换行符敏感设置？", True):
+            for key, value in target_settings.items():
+                subprocess.run(["git", "config", key, value], check=True)
+
+            # 对于Windows系统，提示用户可以创建.gitattributes文件
+            if sys.platform.startswith("win"):
+                self._handle_windows_line_endings()
+
+            print("✅ git换行符敏感设置已更新")
+        else:
+            print("❌ 用户取消修改git换行符敏感设置")
+            sys.exit(0)
+
+    def _handle_windows_line_endings(self) -> None:
+        """在Windows系统上处理换行符问题，提供建议而非强制修改"""
+        gitattributes_path = os.path.join(self.root_dir, ".gitattributes")
+
+        # 检查是否已存在.gitattributes文件
+        if os.path.exists(gitattributes_path):
+            with open(gitattributes_path, "r", encoding="utf-8") as f:
+                content = f.read()
+            # 如果已经有换行符相关配置，就不再提示
+            if any(keyword in content for keyword in ["text=", "eol=", "binary"]):
+                return
+
+        print(
+            "\n💡 提示：在Windows系统上，建议配置.gitattributes文件来避免换行符问题。"
+        )
+        print("这可以防止仅因换行符不同而导致整个文件被标记为修改。")
+
+        if user_confirm("是否要创建一个最小化的.gitattributes文件？", False):
+            # 最小化的内容，只影响特定类型的文件
+            minimal_content = """# Jarvis建议的最小化换行符配置
+# 默认所有文本文件使用LF，只有Windows特定文件使用CRLF
+
+# 默认所有文本文件使用LF
+* text=auto eol=lf
+
+# Windows批处理文件需要CRLF
+*.bat text eol=crlf
+*.cmd text eol=crlf
+*.ps1 text eol=crlf
+"""
+
+            if not os.path.exists(gitattributes_path):
+                with open(gitattributes_path, "w", encoding="utf-8", newline="\n") as f:
+                    f.write(minimal_content)
+                print("✅ 已创建最小化的.gitattributes文件")
+            else:
+                print("📝 将以下内容追加到现有.gitattributes文件：")
+                print(minimal_content)
+                if user_confirm("是否追加到现有文件？", True):
+                    with open(
+                        gitattributes_path, "a", encoding="utf-8", newline="\n"
+                    ) as f:
+                        f.write("\n" + minimal_content)
+                    print("✅ 已更新.gitattributes文件")
+        else:
+            print(
+                "ℹ️ 跳过.gitattributes文件创建。如果遇到换行符问题，可以手动创建此文件。"
+            )
+
+    def _record_code_changes_stats(self, diff_text: str) -> None:
+        """记录代码变更的统计信息。
+
+        Args:
+            diff_text: git diff的文本输出
+        """
+        from jarvis.jarvis_stats.stats import StatsManager
+        import re
+
+        stats_manager = StatsManager()
+
+        # 匹配插入行数
+        insertions_match = re.search(r"(\d+)\s+insertions?\(\+\)", diff_text)
+        if insertions_match:
+            insertions = int(insertions_match.group(1))
+            stats_manager.increment(
+                "code_lines_inserted", amount=insertions, group="code_agent"
+            )
+
+        # 匹配删除行数
+        deletions_match = re.search(r"(\d+)\s+deletions?\(\-\)", diff_text)
+        if deletions_match:
+            deletions = int(deletions_match.group(1))
+            stats_manager.increment(
+                "code_lines_deleted", amount=deletions, group="code_agent"
+            )
+
     def _handle_uncommitted_changes(self) -> None:
         """处理未提交的修改，包括：
         1. 提示用户确认是否提交
@@ -201,10 +375,31 @@ class CodeAgent:
         5. 暂存并提交所有修改
         """
         if has_uncommitted_changes():
+            # 获取代码变更统计
+            try:
+                diff_result = subprocess.run(
+                    ["git", "diff", "HEAD", "--shortstat"],
+                    capture_output=True,
+                    text=True,
+                    encoding="utf-8",
+                    errors="replace",
+                    check=True,
+                )
+                if diff_result.returncode == 0 and diff_result.stdout:
+                    self._record_code_changes_stats(diff_result.stdout)
+            except subprocess.CalledProcessError:
+                pass
+
             PrettyOutput.print("检测到未提交的修改，是否要提交？", OutputType.WARNING)
             if not user_confirm("是否要提交？", True):
                 return
 
+            # 用户确认修改，统计修改次数
+            from jarvis.jarvis_stats.stats import StatsManager
+
+            stats_manager = StatsManager()
+            stats_manager.increment("code_modification_confirmed", group="code_agent")
+
             try:
                 confirm_add_new_files()
 
@@ -216,6 +411,8 @@ class CodeAgent:
                     ["git", "rev-list", "--count", "HEAD"],
                     capture_output=True,
                     text=True,
+                    encoding="utf-8",
+                    errors="replace",
                     check=True,
                 )
                 if commit_result.returncode != 0:
@@ -231,6 +428,9 @@ class CodeAgent:
                     ["git", "commit", "-m", f"CheckPoint #{commit_count + 1}"],
                     check=True,
                 )
+
+                # 统计提交次数
+                stats_manager.increment("code_commits_accepted", group="code_agent")
             except subprocess.CalledProcessError as e:
                 PrettyOutput.print(f"提交失败: {str(e)}", OutputType.ERROR)
 
@@ -252,6 +452,12 @@ class CodeAgent:
             commits = []
 
         if commits:
+            # 统计生成的commit数量
+            from jarvis.jarvis_stats.stats import StatsManager
+
+            stats_manager = StatsManager()
+            stats_manager.increment("commits_generated", group="code_agent")
+
             commit_messages = "检测到以下提交记录:\n" + "\n".join(
                 f"- {commit_hash[:7]}: {message}" for commit_hash, message in commits
             )
@@ -263,6 +469,12 @@ class CodeAgent:
     ) -> None:
         """处理提交确认和可能的重置"""
         if commits and user_confirm("是否接受以上提交记录？", True):
+            # 统计接受的commit数量
+            from jarvis.jarvis_stats.stats import StatsManager
+
+            stats_manager = StatsManager()
+            stats_manager.increment("commits_accepted", group="code_agent")
+
             subprocess.run(
                 ["git", "reset", "--mixed", str(start_commit)],
                 stdout=subprocess.DEVNULL,
@@ -350,6 +562,28 @@ class CodeAgent:
             modified_files = get_diff_file_list()
             commited = handle_commit_workflow()
             if commited:
+                # 统计代码行数变化
+                # 获取diff的统计信息
+                try:
+                    diff_result = subprocess.run(
+                        ["git", "diff", "HEAD~1", "HEAD", "--shortstat"],
+                        capture_output=True,
+                        text=True,
+                        encoding="utf-8",
+                        errors="replace",
+                        check=True,
+                    )
+                    if diff_result.returncode == 0 and diff_result.stdout:
+                        self._record_code_changes_stats(diff_result.stdout)
+                except subprocess.CalledProcessError:
+                    pass
+
+                # 统计修改次数
+                from jarvis.jarvis_stats.stats import StatsManager
+
+                stats_manager = StatsManager()
+                stats_manager.increment("code_modifications", group="code_agent")
+
                 # 获取提交信息
                 end_hash = get_latest_commit_hash()
                 commits = get_commits_between(start_hash, end_hash)
@@ -409,23 +643,57 @@ def cli(
     llm_type: str = typer.Option(
         "normal",
         "--llm_type",
-        help="LLM type to use, choices are 'normal' and 'thinking'",
+        help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
     model_group: Optional[str] = typer.Option(
-        None, "--model_group", help="Model group to use, overriding config"
+        None, "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
     ),
     requirement: Optional[str] = typer.Option(
-        None, "-r", "--requirement", help="Requirement to process"
+        None, "-r", "--requirement", help="要处理的需求描述"
     ),
     restore_session: bool = typer.Option(
         False,
         "--restore-session",
-        help="Restore session from .jarvis/saved_session.json",
+        help="从 .jarvis/saved_session.json 恢复会话状态",
     ),
 ) -> None:
     """Jarvis主入口点。"""
     init_env("欢迎使用 Jarvis-CodeAgent，您的代码工程助手已准备就绪！")
 
+    try:
+        subprocess.run(
+            ["git", "rev-parse", "--git-dir"],
+            check=True,
+            stdout=subprocess.DEVNULL,
+            stderr=subprocess.DEVNULL,
+        )
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        curr_dir_path = os.getcwd()
+        PrettyOutput.print(
+            f"警告：当前目录 '{curr_dir_path}' 不是一个git仓库。", OutputType.WARNING
+        )
+        if user_confirm(
+            f"是否要在 '{curr_dir_path}' 中初始化一个新的git仓库？", default=True
+        ):
+            try:
+                subprocess.run(
+                    ["git", "init"],
+                    check=True,
+                    capture_output=True,
+                    text=True,
+                    encoding="utf-8",
+                    errors="replace",
+                )
+                PrettyOutput.print("✅ 已成功初始化git仓库。", OutputType.SUCCESS)
+            except (subprocess.CalledProcessError, FileNotFoundError) as e:
+                PrettyOutput.print(f"❌ 初始化git仓库失败: {e}", OutputType.ERROR)
+                sys.exit(1)
+        else:
+            PrettyOutput.print(
+                "操作已取消。Jarvis需要在git仓库中运行。", OutputType.INFO
+            )
+            sys.exit(0)
+
     curr_dir = os.getcwd()
     git_dir = find_git_root_and_cd(curr_dir)
     PrettyOutput.print(f"当前目录: {git_dir}", OutputType.INFO)