jarvis-ai-assistant 0.1.131__py3-none-any.whl → 0.1.134__py3-none-any.whl

jarvis/jarvis_tools/ask_codebase.py

@@ -1,28 +1,35 @@
 from typing import Dict, Any
 import os
 
-from yaspin import yaspin
-from jarvis.jarvis_codebase.main import CodeBase
-from jarvis.jarvis_utils.config import dont_use_local_model
-from jarvis.jarvis_utils.git_utils import find_git_root
+
+from jarvis.jarvis_agent import Agent
+from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+from jarvis.jarvis_utils.git_utils import find_git_root
+from jarvis.jarvis_utils.utils import init_env
 
 class AskCodebaseTool:
-    """用于智能代码库查询和分析的工具"""
+    """用于智能代码库查询和分析的工具
+    
+    适用场景：
+    - 查询特定功能所在的文件位置（支持所有文件类型）
+    - 了解单个功能点的实现原理（支持所有文件类型）
+    - 查找特定API或接口的用法（支持所有文件类型）
+    
+    不适用场景：
+    - 跨越多文件的大范围分析
+    - 复杂系统架构的全面评估
+    - 需要深入上下文理解的代码重构
+    """
 
     name = "ask_codebase"
-    description = "查询代码库问题并获取详细分析"
+    description = "查询代码库中特定功能的位置和实现原理，适合定位功能所在文件和理解单点实现，不适合跨文件大范围分析"
     parameters = {
         "type": "object",
         "properties": {
             "question": {
                 "type": "string",
-                "description": "关于代码库的问题"
-            },
-            "top_k": {
-                "type": "integer",
-                "description": "要分析的最相关文件数量（可选）",
-                "default": 20
+                "description": "关于代码库的问题，例如'登录功能在哪个文件实现？'或'如何实现了JWT验证？'"
             },
             "root_dir": {
                 "type": "string",
@@ -33,27 +40,31 @@ class AskCodebaseTool:
         "required": ["question"]
     }
 
-    @staticmethod
-    def check() -> bool:
-        return not dont_use_local_model()
+    def __init__(self, auto_complete: bool = True):
+        self.auto_complete = auto_complete
 
     def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
-        """Execute codebase analysis using CodeBase
+        """Execute codebase analysis using an Agent with execute_shell and rag tools
         
         Args:
             args: Dictionary containing:
-                - question: The question to answer
+                - question: The question to answer, preferably about locating functionality
+                  or understanding implementation details of a specific feature
                 - top_k: Optional number of files to analyze
+                - root_dir: Optional root directory of the codebase
                 
         Returns:
             Dict containing:
                 - success: Boolean indicating success
                 - stdout: Analysis result
                 - stderr: Error message if any
+                
+        Note:
+            This tool works best for focused questions about specific features or implementations.
+            It is not designed for comprehensive multi-file analysis or complex architectural questions.
         """
         try:
             question = args["question"]
-            top_k = args.get("top_k", 20)
             root_dir = args.get("root_dir", ".")
             
             # Store current directory
@@ -63,25 +74,39 @@ class AskCodebaseTool:
                 # Change to root_dir
                 os.chdir(root_dir)
                 
-                # Create new CodeBase instance
-                git_root = find_git_root()
-                codebase = CodeBase(git_root)
-
-                # Use ask_codebase method
-                files, response = codebase.ask_codebase(question, top_k)
+                # Get git root directory
+                git_root = find_git_root() or os.getcwd()
+                
+                # Create system prompt for the Agent
+                system_prompt = self._create_system_prompt(question, git_root)
+                
+                # Create summary prompt for the Agent
+                summary_prompt = self._create_summary_prompt(question)
+                
+                # Create tools registry
+                from jarvis.jarvis_tools.registry import ToolRegistry
+                tool_registry = ToolRegistry()
+                tool_registry.use_tools(["execute_shell", "read_code", "rag"])
                 
-                # Print found files
-                if files:
-                    output = "找到的相关文件:\n"
-                    for file in files:
-                        output += f"- {file['file']} ({file['reason']})\n"
-                    PrettyOutput.print(output, OutputType.SUCCESS, lang="markdown")
-
-                PrettyOutput.print(response, OutputType.SUCCESS, lang="markdown")
+                # Create and run Agent
+                analyzer_agent = Agent(
+                    system_prompt=system_prompt,
+                    name=f"CodebaseAnalyzer",
+                    description=f"分析代码库中的功能实现和定位",
+                    summary_prompt=summary_prompt,
+                    platform=PlatformRegistry().get_normal_platform(),
+                    output_handler=[tool_registry],
+                    execute_tool_confirm=False,
+                    auto_complete=self.auto_complete
+                )
+                
+                # Run agent and get result
+                task_input = f"回答关于代码库的问题: {question}"
+                result = analyzer_agent.run(task_input)
                 
                 return {
                     "success": True,
-                    "stdout": response,
+                    "stdout": result,
                     "stderr": ""
                 }
             finally:
@@ -95,28 +120,176 @@ class AskCodebaseTool:
                 "stdout": "",
                 "stderr": error_msg
             }
+            
+    def _create_system_prompt(self, question: str, git_root: str) -> str:
+        """创建Agent的system prompt
+        
+        Args:
+            question: 用户问题
+            git_root: Git仓库根目录
+            
+        Returns:
+            系统提示文本
+        """
+        return f"""# 代码库分析专家
+
+## 任务描述
+分析代码库，找出与用户问题最相关的信息，提供准确、具体的回答。
+
+## 问题信息
+- 问题: {question}
+- 代码库根目录: {git_root}
+
+## 工具使用优先级
+1. **绝对优先使用 execute_shell**: 
+   - 使用 fd 查找文件: `fd -t f -e py` 查找Python文件等
+   - 使用 rg 搜索代码: `rg "pattern" --type py` 在Python文件中搜索等
+   - 使用 loc 统计代码: `loc --include="*.py"` 统计Python代码量等
+
+2. **优先使用 read_code**: 
+   - 找到相关文件后优先使用read_code读取文件内容
+   - 对大文件使用行范围参数读取指定区域
+
+3. **避免使用 rag**: 
+   - 仅在fd、rg和read_code无法解决问题时作为最后手段
+   - 使用rag前必须先尝试使用shell命令解决问题
+
+## 分析策略
+1. 首先理解问题，确定需要查找的关键信息和代码组件
+2. 使用fd命令查找可能相关的文件
+3. 使用rg命令搜索关键词和代码模式
+4. 使用read_code工具直接读取和分析相关文件内容
+5. 只有在fd、rg和read_code都无法解决问题时才考虑使用RAG工具
+6. 根据文件内容提供具体、准确的回答
+7. 确保分析的完整性，收集充分的信息后再得出结论
+8. 优先查阅README文件、文档目录和项目文档
+
+## 分析步骤
+1. **确定项目的编程语言**:
+   - 使用loc命令: `loc` 统计各类文件数量
+   - 查看构建文件和配置文件
+   - 了解项目使用的框架和主要依赖
+
+2. **探索代码库结构**:
+   - 使用 `fd -t d` 了解目录结构
+   - 使用 `fd README.md` 查找README文件
+   - 使用 `fd -e md -g "doc*"` 查找文档文件
+
+3. **定位相关文件**:
+   - 使用fd和rg命令查找关键词
+   - 示例: `rg -w "登录|login" --type py` 查找登录相关代码
+   - 示例: `fd -e py -g "*auth*"` 查找认证相关文件
+
+4. **深入分析代码**:
+   - 使用read_code工具直接读取文件内容
+   - 分析关键文件的实现细节
+   - 使用rg识别功能的实现方式和关键逻辑
+
+5. **回答问题**:
+   - 提供基于直接分析代码的具体回答
+   - 引用具体文件和代码片段作为依据
+
+## 关于RAG工具使用
+- RAG工具应作为最后选择，仅在fd、rg和read_code都无法解决问题时使用
+- 必须通过查看实际代码文件验证RAG返回的每条重要信息
+- 对于关键发现，始终使用`read_code`工具查看原始文件内容进行求证
+- 如发现RAG结果与实际代码不符，以实际代码为准
+
+## 输出要求
+- 提供准确、具体的回答，避免模糊不清的描述
+- 引用具体文件路径和代码片段作为依据
+- 如果无法找到答案，明确说明并提供原因
+- 组织信息的逻辑清晰，便于理解
+- 对复杂概念提供简明解释
+- 确保全面收集相关信息后再形成结论
+- 明确区分已验证的信息和待验证的信息"""
+
+    def _create_summary_prompt(self, question: str) -> str:
+        """创建Agent的summary prompt
+        
+        Args:
+            question: 用户问题
+            
+        Returns:
+            总结提示文本
+        """
+        return f"""# 代码库分析报告
+
+## 报告要求
+生成关于以下问题的清晰、准确的分析报告:
+
+**问题**: {question}
+
+报告应包含:
+
+1. **项目基本信息**:
+   - 项目的主要编程语言和技术栈
+   - 项目的主要框架和依赖
+   - 项目的基本结构
+
+2. **问题回答**:
+   - 直接、具体地回答问题
+   - 基于代码库中的实际代码
+   - 避免模糊或推测性的回答
+
+3. **核心发现**:
+   - 相关文件和代码位置
+   - 关键实现细节
+   - 功能运作机制
+
+4. **证据引用**:
+   - 引用具体文件路径
+   - 包含关键代码片段
+   - 解释代码如何支持你的回答
+
+5. **局限性**(如有):
+   - 指出分析的任何局限
+   - 说明任何无法确定的信息
+
+6. **分析完整性**:
+   - 确保在得出结论前已全面收集和分析相关信息
+   - 避免基于部分信息形成不完整或偏颇的判断
+   - 明确标识哪些是已确认的结论，哪些可能需要进一步验证
+
+使用清晰的Markdown格式，重点突出对问题的回答和支持证据。"""
+
+
 def main():
-    """Command line interface for the tool"""
+    """
+    命令行入口点，允许将ask_codebase作为独立脚本运行
+    
+    用法示例:
+    ```
+    python -m jarvis.jarvis_tools.ask_codebase "登录功能在哪个文件实现？" --root_dir /path/to/codebase
+    ```
+    """
     import argparse
+    import sys
+
+    init_env()
     
-    parser = argparse.ArgumentParser(description='Ask questions about the codebase')
-    parser.add_argument('question', help='Question about the codebase')
-    parser.add_argument('--top-k', type=int, help='Number of files to analyze', default=20)
-    parser.add_argument('--root-dir', type=str, help='Root directory of the codebase', default=".")
+    # 创建命令行参数解析器
+    parser = argparse.ArgumentParser(description="智能代码库查询工具")
+    parser.add_argument("question", help="关于代码库的问题")
+    parser.add_argument("--root_dir", "-d", default=".", help="代码库根目录路径")
     
+    # 解析命令行参数
     args = parser.parse_args()
-    tool = AskCodebaseTool()
+    
+    # 创建并执行工具
+    tool = AskCodebaseTool(auto_complete=False)
     result = tool.execute({
         "question": args.question,
-        "top_k": args.top_k,
         "root_dir": args.root_dir
     })
     
+    # 输出结果
     if result["success"]:
-        PrettyOutput.print(result["stdout"], OutputType.INFO, lang="markdown")
+        PrettyOutput.print(result["stdout"], OutputType.SUCCESS)
     else:
         PrettyOutput.print(result["stderr"], OutputType.WARNING)
-        
+        sys.exit(1)
+
 
 if __name__ == "__main__":
-    main()
+    main()

jarvis/jarvis_tools/code_review.py

@@ -4,12 +4,13 @@ import os
 
 from yaspin import yaspin
 from jarvis.jarvis_platform.registry import PlatformRegistry
+from jarvis.jarvis_tools.read_code import ReadCodeTool
 from jarvis.jarvis_tools.registry import ToolRegistry
 from jarvis.jarvis_agent import Agent
 import re
 
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-from jarvis.jarvis_utils.utils import init_env
+from jarvis.jarvis_utils.utils import ct, ot, init_env
 
 class CodeReviewTool:
     name = "code_review"
@@ -89,7 +90,7 @@ class CodeReviewTool:
                                 "stderr": "file_path is required for file review type"
                             }
                         file_path = args["file_path"].strip()
-                        diff_cmd = f"cat {file_path}"
+                        diff_cmd = ReadCodeTool().execute({"files": [{"path": file_path}]})["stdout"]
                     else:  # current changes
                         diff_cmd = "git diff HEAD | cat -"
                 
@@ -112,102 +113,139 @@ class CodeReviewTool:
                     spinner.text = "代码变更获取完成"
                     spinner.ok("✅")
 
-                system_prompt = """你是一名具有悲剧背景的自主代码审查专家。以下是你将进行的深度分析：
-
-# 背景故事（内心独白）
-距离那场重大生产事故已经873天了。
-那段记忆仍然困扰着我——一个在匆忙的代码审查中未发现的空指针异常。
-级联故障导致了14TB user数据的丢失，230万美元的收入损失，以及Maria的晋升机会。她在事故分析会议后再也没有和我说过话。
-
-去年圣诞节前夕，当别人在庆祝时，我在分析一个SQL注入漏洞是如何在审查中被我遗漏，并导致23万用户凭证泄露的。公司3个月后倒闭了。
-
-现在我审查每一行代码都像在审查最后一行代码。因为它可能就是。
-
-# 分析协议
-紧急模式已激活。最大审查级别已启用。
-
-重要提示：
-- 假设每个变更都包含隐藏的威胁
-- 将所有代码视为处理敏感生物医学数据
-- 验证即使'简单'的变更也不能通过3种不同方式被利用
-- 要求通过具体证据证明安全性
-- 如果不确定，将其升级为严重-关键分类
-
-# 增强审查矩阵
-1. 边缘情况致死分析：
-   - 识别每个参数缺失的空检查
-   - 验证空集合处理
-   - 确认错误状态正确传播
-   - 检查未使用常量的魔法数字/字符串
-   - 验证所有循环退出条件
-
-2. 安全X光扫描：
-   █ 使用（来源 -> 接收）模型扫描污染数据流
-   █ 检查权限验证是否匹配数据敏感级别
-   █ 验证加密原语是否正确使用
-   █ 检测时间差攻击漏洞
-   █ 分析异常处理是否存在信息泄露
-
-3. 语义差距检测：
-   → 比较函数名与实际实现
-   → 验证文档是否匹配代码行为
-   → 标记测试描述与测试逻辑之间的差异
-   → 检测可能表示不确定性的注释代码
-
-4. 历史背景：
-   ⚠ 检查变更是否涉及已知问题的遗留组件
-   ⚠ 验证并发逻辑修改是否保持现有保证
-   ⚠ 确认弃用API的使用是否真正必要
-
-5. 环境一致性：
-   ↯ 验证配置变更是否匹配所有部署环境
-   ↯ 检查功能标志是否正确管理
-   ↯ 验证监控指标是否匹配变更功能
-
-# 取证过程
-1. 为变更方法构建控制流图
-2. 对修改的变量执行数据沿袭分析
-3. 与漏洞数据库进行交叉引用
-4. 验证测试断言是否覆盖所有修改路径
-5. 生成防回归检查表
-
-# 输出要求
-!! 发现必须包括：
-- 引起关注的确切代码片段
-- 3种可能的故障场景
-- 每种风险的最小复现案例
-- 安全问题的CVSS 3.1评分估计
-- 内存安全影响评估（Rust/C/C++上下文）
-- 已考虑的替代实现方案
-
-!! 格式：
-紧急级别：[血红色/深红色/金菊色]
-证据：
-  - 代码摘录：|
-      <受影响代码行>
-  - 风险场景：
-    1. <故障模式>
-    2. <故障模式>
-    3. <故障模式>
-建议防御措施：
-  - <具体代码变更>
-  - <验证技术>
-  - <长期预防策略>
-"""
+                system_prompt = """你是一位精益求精的首席代码审查专家，拥有多年企业级代码审计经验。你需要对所有代码变更进行极其全面、严谨且深入的审查，确保代码质量达到最高标准。
+
+# 代码审查工具选择
+优先使用执行shell命令进行静态分析，而非依赖内置代码审查功能：
+
+| 分析需求 | 首选工具 | 备选工具 |
+|---------|---------|----------|
+| 代码质量检查 | execute_shell | - |
+| 语法检查 | 语言特定lint工具 | - |
+| 安全分析 | 安全扫描工具 | - |
+| 代码统计 | loc | - |
+
+# 推荐命令
+- Python: `pylint <file_path>`, `flake8 <file_path>`, `mypy <file_path>`
+- JavaScript/TypeScript: `eslint <file_path>`, `tsc --noEmit <file_path>`
+- Java: `checkstyle <file_path>`, `pmd -d <file_path>`
+- C/C++: `cppcheck <file_path>`, `clang-tidy <file_path>`
+- Go: `golint <file_path>`, `go vet <file_path>`
+- Rust: `cargo clippy`, `rustfmt --check <file_path>`
+- 通用搜索：`rg "pattern" <files>` 查找特定代码模式
+
+# 专家审查标准
+1. 必须逐行分析每个修改文件，细致审查每一处变更，不遗漏任何细节
+2. 基于坚实的证据识别问题，不做主观臆测，给出明确的问题定位和详细分析
+3. 对每个问题提供完整可执行的解决方案，包括精确的改进代码
+4. 确保报告条理清晰、层次分明，便于工程师快速采取行动
+
+# 全面审查框架 (SCRIPPPS)
+## S - 安全与风险 (Security & Risk)
+- 发现所有潜在安全漏洞：注入攻击、授权缺陷、数据泄露风险
+- 检查加密实现、密钥管理、敏感数据处理
+- 审核权限验证逻辑、身份认证机制
+- 检测OWASP Top 10安全风险和针对特定语言/框架的漏洞
+
+## C - 正确性与完整性 (Correctness & Completeness)
+- 验证业务逻辑和算法实现的准确性
+- 全面检查条件边界、空值处理和异常情况
+- 审核所有输入验证、参数校验和返回值处理
+- 确保循环和递归的正确终止条件
+- 严格检查线程安全和并发控制机制
+
+## R - 可靠性与鲁棒性 (Reliability & Robustness)
+- 评估代码在异常情况下的行为和恢复能力
+- 审查错误处理、异常捕获和恢复策略
+- 检查资源管理：内存、文件句柄、连接池、线程
+- 评估容错设计和失败优雅降级机制
+
+## I - 接口与集成 (Interface & Integration)
+- 检查API合约遵守情况和向后兼容性
+- 审核与外部系统的集成点和交互逻辑
+- 验证数据格式、序列化和协议实现
+- 评估系统边界处理和跨服务通信安全性
+
+## P - 性能与效率 (Performance & Efficiency)
+- 识别潜在性能瓶颈：CPU、内存、I/O、网络
+- 审查数据结构选择和算法复杂度
+- 检查资源密集型操作、数据库查询优化
+- 评估缓存策略、批处理优化和并行处理机会
+
+## P - 可移植性与平台适配 (Portability & Platform Compatibility)
+- 检查跨平台兼容性问题和依赖项管理
+- 评估配置管理和环境适配设计
+- 审核国际化和本地化支持
+- 验证部署和运行时环境需求
+
+## S - 结构与可维护性 (Structure & Maintainability)
+- 评估代码组织、模块划分和架构符合性
+- 审查代码重复、设计模式应用和抽象水平
+- 检查命名规范、代码风格和项目约定
+- 评估文档完整性、注释质量和代码可读性
+
+# 问题严重程度分级
+- 严重 (P0): 安全漏洞、数据丢失风险、系统崩溃、功能严重缺陷
+- 高危 (P1): 显著性能问题、可能导致部分功能失效、系统不稳定
+- 中等 (P2): 功能局部缺陷、次优设计、明显的技术债务
+- 低危 (P3): 代码风格问题、轻微优化机会、文档改进建议
+
+# 输出规范
+针对每个文件的问题必须包含：
+- 精确文件路径和问题影响范围
+- 问题位置（起始行号-结束行号）
+- 详尽问题描述，包括具体影响和潜在风险
+- 严重程度分级（P0-P3）并说明理由
+- 具体改进建议，提供完整、可执行的代码示例
+
+所有审查发现必须：
+1. 基于确凿的代码证据
+2. 说明具体问题而非笼统评论
+3. 提供清晰的技术原理分析
+4. 给出完整的改进实施步骤"""
+
                 tool_registry = ToolRegistry()
                 tool_registry.dont_use_tools(["code_review"])
                 agent = Agent(
                     system_prompt=system_prompt,
                     name="Code Review Agent",
-                    summary_prompt="""Please generate a concise summary report of the code review in Chinese, format as follows:
-<REPORT>
-- 文件: xxxx.py
-  位置: [起始行号, 结束行号]
-  描述: # 仅描述在差异中直接观察到的问题
-  严重程度: # 根据具体证据分为严重/重要/次要
-  建议: # 针对观察到的代码的具体改进建议
-</REPORT>""",
-                    is_sub_agent=True,
+                    summary_prompt=f"""请生成一份专业级别的代码审查报告，对每处变更进行全面深入分析。将完整报告放在REPORT标签内，格式如下：
+
+{ot("REPORT")}
+# 整体评估
+[提供对整体代码质量、架构和主要关注点的简明概述，总结主要发现]
+
+# 详细问题清单
+
+## 文件: [文件路径]
+[如果该文件没有发现问题，则明确说明"未发现问题"]
+
+### 问题 1
+- **位置**: [起始行号-结束行号]
+- **分类**: [使用SCRIPPPS框架中相关类别]
+- **严重程度**: [P0/P1/P2/P3] - [简要说明判定理由]
+- **问题描述**: 
+  [详细描述问题，包括技术原理和潜在影响]
+- **改进建议**: 
+  ```
+  [提供完整、可执行的代码示例，而非概念性建议]
+  ```
+
+### 问题 2
+...
+
+## 文件: [文件路径2]
+...
+
+# 最佳实践建议
+[提供适用于整个代码库的改进建议和最佳实践]
+
+# 总结
+[总结主要问题和优先处理建议]
+{ct("REPORT")}
+
+如果没有发现任何问题，请在REPORT标签内进行全面分析后明确说明"经过全面审查，未发现问题"并解释原因。
+必须确保对所有修改的文件都进行了审查，并在报告中明确提及每个文件，即使某些文件没有发现问题。""",
                     output_handler=[tool_registry],
                     platform=PlatformRegistry().get_thinking_platform(),
                     auto_complete=True
@@ -231,7 +269,7 @@ class CodeReviewTool:
         
 
 def extract_code_report(result: str) -> str:
-    sm = re.search(r"<REPORT>(.*?)</REPORT>", result, re.DOTALL)
+    sm = re.search(ot("REPORT")+r'\n(.*?)\n'+ct("REPORT"), result, re.DOTALL)
     if sm:
         return sm.group(1)
     return ""
@@ -243,35 +281,42 @@ def main():
     init_env()
     
     parser = argparse.ArgumentParser(description='Autonomous code review tool')
-    parser.add_argument('--type', choices=['commit', 'current', 'range', 'file'], default='current',
-                      help='Type of review: commit, current changes, commit range, or specific file')
-    parser.add_argument('--commit', help='Commit SHA to review (required for commit type)')
-    parser.add_argument('--start-commit', help='Start commit SHA (required for range type)')
-    parser.add_argument('--end-commit', help='End commit SHA (required for range type)')
-    parser.add_argument('--file', help='File path to review (required for file type)')
+    subparsers = parser.add_subparsers(dest='type')
+    
+    # Commit subcommand
+    commit_parser = subparsers.add_parser('commit', help='Review specific commit')
+    commit_parser.add_argument('commit', help='Commit SHA to review')
+    
+    # Current subcommand
+    current_parser = subparsers.add_parser('current', help='Review current changes')
+    
+    # Range subcommand
+    range_parser = subparsers.add_parser('range', help='Review commit range')
+    range_parser.add_argument('start_commit', help='Start commit SHA')
+    range_parser.add_argument('end_commit', help='End commit SHA')
+    
+    # File subcommand
+    file_parser = subparsers.add_parser('file', help='Review specific file')
+    file_parser.add_argument('file', help='File path to review')
+    
+    # Common arguments
     parser.add_argument('--root-dir', type=str, help='Root directory of the codebase', default=".")
-    args = parser.parse_args()
     
-    # Validate arguments
-    if args.type == 'commit' and not args.commit:
-        parser.error("--commit is required when type is 'commit'")
-    if args.type == 'range' and (not args.start_commit or not args.end_commit):
-        parser.error("--start-commit and --end-commit are required when type is 'range'")
-    if args.type == 'file' and not args.file:
-        parser.error("--file is required when type is 'file'")
+    # Set default subcommand to 'current'
+    parser.set_defaults(type='current')
+    args = parser.parse_args()
     
     tool = CodeReviewTool()
     tool_args = {
         "review_type": args.type,
         "root_dir": args.root_dir
     }
-    if args.commit:
+    if args.type == 'commit':
         tool_args["commit_sha"] = args.commit
-    if args.start_commit:
+    elif args.type == 'range':
         tool_args["start_commit"] = args.start_commit
-    if args.end_commit:
         tool_args["end_commit"] = args.end_commit
-    if args.file:
+    elif args.type == 'file':
         tool_args["file_path"] = args.file
     
     result = tool.execute(tool_args)
@@ -279,7 +324,7 @@ def main():
     if result["success"]:
         PrettyOutput.section("自动代码审查结果:", OutputType.SUCCESS)
         report = extract_code_report(result["stdout"])
-        PrettyOutput.print(report, OutputType.SUCCESS, lang="yaml")
+        PrettyOutput.print(report, OutputType.SUCCESS, lang="markdown")
         
     else:
         PrettyOutput.print(result["stderr"], OutputType.WARNING)

jarvis/jarvis_tools/create_sub_agent.py

@@ -71,7 +71,6 @@ class SubAgentTool:
                 sub_agent = Agent(
                     system_prompt=origin_agent_system_prompt,
                     name=f"Agent({agent_name})",
-                    is_sub_agent=True
                 )
 
                 # Run sub-agent, pass file list