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

jarvis/jarvis_multi_agent/__init__.py

@@ -138,45 +138,4 @@ content: {msg['content']}
                 last_agent = self.agents[msg['to']].name
                 msg = self.agents[msg['to']].run(prompt)
         return ""
-
-
-def main():
-    """从YAML配置文件初始化并运行多智能体系统
-    
-    Returns:
-        最终处理结果
-    """
-    init_env()
-    import argparse
-    parser = argparse.ArgumentParser(description="多智能体系统启动器")
-    parser.add_argument("--config", "-c", required=True, help="YAML配置文件路径")
-    parser.add_argument("--input", "-i", help="用户输入（可选）")
-    args = parser.parse_args()
-        
-    try:
-        with open(args.config, 'r', errors="ignore") as f:
-            config_data = yaml.safe_load(f)
-            
-        # 获取agents配置
-        agents_config = config_data.get('agents', [])
-        
-        main_agent_name = config_data.get('main_agent', '')
-        if not main_agent_name:
-            raise ValueError("必须指定main_agent作为主智能体")
-            
-        # 创建并运行多智能体系统
-        multi_agent = MultiAgent(agents_config, main_agent_name)
-        user_input = args.input if args.input is not None else get_multiline_input("请输入内容（输入空行结束）：")
-        if user_input == "":
-            return
-        return multi_agent.run(user_input)
-        
-    except yaml.YAMLError as e:
-        raise ValueError(f"YAML配置文件解析错误: {str(e)}")
-    except Exception as e:
-        raise RuntimeError(f"多智能体系统初始化失败: {str(e)}")
-
-
-if __name__ == "__main__":
-    result = main()
     

jarvis/jarvis_multi_agent/main.py

@@ -0,0 +1,43 @@
+import yaml
+from jarvis.jarvis_multi_agent import MultiAgent
+from jarvis.jarvis_utils.utils import init_env
+from jarvis.jarvis_utils.input import get_multiline_input
+
+def main():
+    """从YAML配置文件初始化并运行多智能体系统
+    
+    Returns:
+        最终处理结果
+    """
+    init_env()
+    import argparse
+    parser = argparse.ArgumentParser(description="多智能体系统启动器")
+    parser.add_argument("--config", "-c", required=True, help="YAML配置文件路径")
+    parser.add_argument("--input", "-i", help="用户输入（可选）")
+    args = parser.parse_args()
+        
+    try:
+        with open(args.config, 'r', errors="ignore") as f:
+            config_data = yaml.safe_load(f)
+            
+        # 获取agents配置
+        agents_config = config_data.get('agents', [])
+        
+        main_agent_name = config_data.get('main_agent', '')
+        if not main_agent_name:
+            raise ValueError("必须指定main_agent作为主智能体")
+            
+        # 创建并运行多智能体系统
+        multi_agent = MultiAgent(agents_config, main_agent_name)
+        user_input = args.input if args.input is not None else get_multiline_input("请输入内容（输入空行结束）：")
+        if user_input == "":
+            return
+        return multi_agent.run(user_input)
+        
+    except yaml.YAMLError as e:
+        raise ValueError(f"YAML配置文件解析错误: {str(e)}")
+    except Exception as e:
+        raise RuntimeError(f"多智能体系统初始化失败: {str(e)}")
+
+if __name__ == "__main__":
+    result = main()

jarvis/jarvis_platform/registry.py

@@ -4,7 +4,7 @@ import os
 import sys
 from typing import Dict, Type, Optional, List
 from jarvis.jarvis_platform.base import BasePlatform
-from jarvis.jarvis_utils.config import get_cheap_model_name, get_cheap_platform_name, get_codegen_model_name, get_codegen_platform_name, get_normal_model_name, get_normal_platform_name, get_thinking_model_name, get_thinking_platform_name
+from jarvis.jarvis_utils.config import get_normal_model_name, get_normal_platform_name, get_thinking_model_name, get_thinking_platform_name
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 REQUIRED_METHODS = [
@@ -21,7 +21,7 @@ REQUIRED_METHODS = [
 class PlatformRegistry:
     """Platform registry"""
 
-    global_platform_name = "kimi"
+    global_platform_name = "yuanbao"
     global_platform_registry = None
 
     @staticmethod
@@ -161,20 +161,6 @@ class PlatformRegistry:
         platform.set_model_name(model_name) # type: ignore
         return platform # type: ignore
     
-    def get_codegen_platform(self) -> BasePlatform:
-        platform_name = get_codegen_platform_name()
-        model_name = get_codegen_model_name()
-        platform = self.create_platform(platform_name)
-        platform.set_model_name(model_name) # type: ignore
-        return platform # type: ignore
-    
-    def get_cheap_platform(self) -> BasePlatform:
-        platform_name = get_cheap_platform_name()
-        model_name = get_cheap_model_name()
-        platform = self.create_platform(platform_name)
-        platform.set_model_name(model_name) # type: ignore
-        return platform # type: ignore
-    
     def get_thinking_platform(self) -> BasePlatform:
         platform_name = get_thinking_platform_name()
         model_name = get_thinking_model_name()

jarvis/jarvis_platform/yuanbao.py

@@ -6,7 +6,7 @@ from jarvis.jarvis_platform.base import BasePlatform
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import while_success
 
-class HunyuanModel(BasePlatform):
+class YuanbaoPlatform(BasePlatform):
     """Hunyuan model implementation"""
 
     platform_name = "yuanbao"

jarvis/jarvis_rag/main.py

@@ -18,7 +18,7 @@ from jarvis.jarvis_utils.embedding import get_context_token_count, get_embedding
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import  ct, get_file_md5, init_env, init_gpu_config, ot
 
-from .file_processors import TextFileProcessor, PDFProcessor, DocxProcessor, PPTProcessor, ExcelProcessor
+from jarvis.jarvis_rag.file_processors import TextFileProcessor, PDFProcessor, DocxProcessor, PPTProcessor, ExcelProcessor
 
 """
 Jarvis RAG (Retrieval-Augmented Generation) Module

jarvis/jarvis_tools/ask_codebase.py

@@ -1,23 +1,20 @@
 from typing import Dict, Any
 import os
-import pathlib
 
-from colorama import init
 
 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.config import dont_use_local_model
 from jarvis.jarvis_utils.utils import init_env
 
 class AskCodebaseTool:
     """用于智能代码库查询和分析的工具
     
     适用场景：
-    - 查询特定功能所在的文件位置
-    - 了解单个功能点的实现原理
-    - 查找特定API或接口的用法
+    - 查询特定功能所在的文件位置（支持所有文件类型）
+    - 了解单个功能点的实现原理（支持所有文件类型）
+    - 查找特定API或接口的用法（支持所有文件类型）
     
     不适用场景：
     - 跨越多文件的大范围分析
@@ -43,9 +40,8 @@ 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 an Agent with execute_shell and rag tools
@@ -98,14 +94,10 @@ class AskCodebaseTool:
                     name=f"CodebaseAnalyzer",
                     description=f"分析代码库中的功能实现和定位",
                     summary_prompt=summary_prompt,
-                    platform=PlatformRegistry().get_codegen_platform(),
+                    platform=PlatformRegistry().get_normal_platform(),
                     output_handler=[tool_registry],
-                    need_summary=True,
-                    is_sub_agent=True,
-                    use_methodology=False,
-                    record_methodology=False,
                     execute_tool_confirm=False,
-                    auto_complete=True
+                    auto_complete=self.auto_complete
                 )
                 
                 # Run agent and get result
@@ -149,62 +141,67 @@ class AskCodebaseTool:
 - 代码库根目录: {git_root}
 
 ## 工具使用优先级
-1. **优先使用 execute_shell**: 首先使用shell命令查找和分析相关文件
-2. **优先使用 read_code**: 找到相关文件后优先使用read_code读取文件内容
-3. **谨慎使用 rag**: 仅在shell命令和直接文件读取无法解决问题时作为辅助手段
+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. 使用shell命令(execute_shell)搜索和查找可能相关的文件
-3. 使用read_code工具直接读取和分析相关文件内容
-4. 只有在必要时才使用RAG工具作为辅助手段
-5. 根据文件内容提供具体、准确的回答
+2. 使用fd命令查找可能相关的文件
+3. 使用rg命令搜索关键词和代码模式
+4. 使用read_code工具直接读取和分析相关文件内容
+5. 只有在fd、rg和read_code都无法解决问题时才考虑使用RAG工具
+6. 根据文件内容提供具体、准确的回答
+7. 确保分析的完整性，收集充分的信息后再得出结论
+8. 优先查阅README文件、文档目录和项目文档
 
 ## 分析步骤
-1. **探索代码库结构**:
-   - 使用shell命令了解目录结构，找出可能包含相关功能的位置
-   - 识别主要模块和组件
-
-2. **定位相关文件**:
-   - 优先使用grep、find等shell命令查找关键词
-   - 使用文件名、关键词搜索找到潜在相关文件
-
-3. **深入分析代码**:
+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识别功能的实现方式和关键逻辑
 
-4. **回答问题**:
+5. **回答问题**:
    - 提供基于直接分析代码的具体回答
    - 引用具体文件和代码片段作为依据
 
 ## 关于RAG工具使用
-- RAG工具应作为最后选择，仅在shell命令和直接文件读取无法解决问题时使用
-- RAG工具返回的信息可能存在偏差或不准确之处
+- RAG工具应作为最后选择，仅在fd、rg和read_code都无法解决问题时使用
 - 必须通过查看实际代码文件验证RAG返回的每条重要信息
 - 对于关键发现，始终使用`read_code`工具查看原始文件内容进行求证
 - 如发现RAG结果与实际代码不符，以实际代码为准
 
-## 探索命令示例
-```bash
-# 查看目录结构
-find . -type d -not -path "*/\\.*" | sort
-
-# 搜索与问题相关的文件
-find . -type f -name "*.py" -o -name "*.js" | xargs grep -l "关键词"
-grep -r "关键词" --include="*.py" .
-
-# 查看文件内容
-cat 文件路径
-```
-
-
 ## 输出要求
 - 提供准确、具体的回答，避免模糊不清的描述
 - 引用具体文件路径和代码片段作为依据
 - 如果无法找到答案，明确说明并提供原因
 - 组织信息的逻辑清晰，便于理解
 - 对复杂概念提供简明解释
+- 确保全面收集相关信息后再形成结论
 - 明确区分已验证的信息和待验证的信息"""
 
     def _create_summary_prompt(self, question: str) -> str:
@@ -225,25 +222,35 @@ cat 文件路径
 
 报告应包含:
 
-1. **问题回答**:
+1. **项目基本信息**:
+   - 项目的主要编程语言和技术栈
+   - 项目的主要框架和依赖
+   - 项目的基本结构
+
+2. **问题回答**:
    - 直接、具体地回答问题
    - 基于代码库中的实际代码
    - 避免模糊或推测性的回答
 
-2. **核心发现**:
+3. **核心发现**:
    - 相关文件和代码位置
    - 关键实现细节
    - 功能运作机制
 
-3. **证据引用**:
+4. **证据引用**:
    - 引用具体文件路径
    - 包含关键代码片段
    - 解释代码如何支持你的回答
 
-4. **局限性**(如有):
+5. **局限性**(如有):
    - 指出分析的任何局限
    - 说明任何无法确定的信息
 
+6. **分析完整性**:
+   - 确保在得出结论前已全面收集和分析相关信息
+   - 避免基于部分信息形成不完整或偏颇的判断
+   - 明确标识哪些是已确认的结论，哪些可能需要进一步验证
+
 使用清晰的Markdown格式，重点突出对问题的回答和支持证据。"""
 
 
@@ -270,7 +277,7 @@ def main():
     args = parser.parse_args()
     
     # 创建并执行工具
-    tool = AskCodebaseTool()
+    tool = AskCodebaseTool(auto_complete=False)
     result = tool.execute({
         "question": args.question,
         "root_dir": args.root_dir

jarvis/jarvis_tools/code_review.py

@@ -4,6 +4,7 @@ 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
@@ -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 -"
                 
@@ -114,6 +115,25 @@ class CodeReviewTool:
 
                 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. 基于坚实的证据识别问题，不做主观臆测，给出明确的问题定位和详细分析
@@ -226,7 +246,6 @@ class CodeReviewTool:
 
 如果没有发现任何问题，请在REPORT标签内进行全面分析后明确说明"经过全面审查，未发现问题"并解释原因。
 必须确保对所有修改的文件都进行了审查，并在报告中明确提及每个文件，即使某些文件没有发现问题。""",
-                    is_sub_agent=True,
                     output_handler=[tool_registry],
                     platform=PlatformRegistry().get_thinking_platform(),
                     auto_complete=True

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

jarvis/jarvis_tools/file_analyzer.py

@@ -112,12 +112,8 @@ class FileAnalyzerTool:
                 name=f"FileAnalyzer-{file_name}",
                 description=f"分析 {file_name} 文件的结构和实现",
                 summary_prompt=summary_prompt,
-                platform=PlatformRegistry().get_codegen_platform(),
+                platform=PlatformRegistry().get_normal_platform(),
                 output_handler=[tool_registry],
-                need_summary=True,
-                is_sub_agent=True,
-                use_methodology=False,
-                record_methodology=False,
                 execute_tool_confirm=False,
                 auto_complete=True
             )
@@ -149,101 +145,116 @@ class FileAnalyzerTool:
         创建Agent的system prompt
         
         Args:
-            file_path: 文件相对路径
+            file_path: 文件路径
             file_name: 文件名
             file_ext: 文件扩展名
-            root_dir: 项目根目录
+            root_dir: 代码库根目录
             objective: 分析目标
             
         Returns:
             系统提示文本
         """
-        # 根据文件扩展名确定语言
-        language_map = {
-            '.py': 'Python',
-            '.js': 'JavaScript',
-            '.ts': 'TypeScript',
-            '.java': 'Java',
-            '.c': 'C',
-            '.cpp': 'C++',
-            '.cs': 'C#',
-            '.go': 'Go',
-            '.rs': 'Rust',
-            '.php': 'PHP',
-            '.rb': 'Ruby',
-            '.swift': 'Swift',
-            '.kt': 'Kotlin',
-            '.sh': 'Shell',
-            '.html': 'HTML',
-            '.css': 'CSS',
-            '.scss': 'SCSS',
-            '.json': 'JSON',
-            '.xml': 'XML',
-            '.yaml': 'YAML',
-            '.md': 'Markdown'
-        }
-        
-        language = language_map.get(file_ext, '未知')
         objective_text = f"\n\n## 分析目标\n{objective}" if objective else ""
         
-        return f"""# 代码文件分析专家
+        return f"""# 文件分析专家
 
 ## 任务描述
-分析文件 `{file_path}` 的内容，专注于分析目标所需的信息，生成有针对性的文件分析报告。{objective_text}
+分析文件 `{file_path}` 的结构、实现细节和代码质量，专注于分析目标所需的内容，生成有针对性、深入且有洞察力的文件分析报告。{objective_text}
+
+## 工具使用优先级
+1. **优先使用 read_code**: 直接读取文件内容是分析文件的首选方式
+2. **优先使用 execute_shell**:
+   - 使用 rg 搜索文件内容: `rg "pattern" {file_path}` 
+   - 使用 loc 统计代码: `loc {file_path}`
+3. **仅在必要时使用其他分析工具**
 
 ## 文件信息
 - 文件路径: `{file_path}`
-- 编程语言: {language}
+- 文件名称: `{file_name}`
+- 文件类型: `{file_ext}`
 - 项目根目录: `{root_dir}`
 
 ## 分析策略
-1. 首先理解分析目标，确定你需要查找的信息
-2. 灵活采用适合目标的分析方法，不受预设分析框架的限制
-3. 专注于与分析目标直接相关的内容，忽略无关部分
-4. 根据目标需要自行判断分析的深度和广度
+1. 首先使用read_code直接读取整个文件内容或分段读取
+2. 使用rg命令搜索文件中的特定模式和结构
+3. 使用loc命令获取文件统计信息
+4. 根据文件类型和分析目标确定重点关注的方面
+5. 保证分析的完整性，收集充分的信息后再得出结论
+
+## 分析步骤
+以下步骤应根据具体分析目标灵活应用:
+
+1. **文件基本信息分析**:
+   - 使用 `loc {file_path}` 获取代码统计
+   - 使用 read_code 读取文件头部注释和文档
+   - 确定文件的编程语言和主要功能
+
+2. **结构分析**:
+   - 使用 read_code 阅读完整文件
+   - 对于大文件，可分段读取主要部分
+   - 识别文件的主要组成部分:
+     - 类定义: `rg "class\\s+" {file_path}`
+     - 函数定义: `rg "def\\s+|function\\s+" {file_path}`
+     - 重要变量: `rg "const\\s+|var\\s+|let\\s+" {file_path}`
+
+3. **核心组件分析**:
+   - 识别文件中的关键接口、类和函数
+   - 使用 rg 搜索重要的代码模式
+   - 分析组件间的交互和依赖关系
+
+4. **实现细节分析**:
+   - 读取并分析关键函数的实现
+   - 关注异常处理: `rg "try|catch|except" {file_path}`
+   - 检查资源管理: `rg "open|close|with" {file_path}`
+
+5. **引用分析**:
+   - 找出引用的外部依赖: `rg "import|require|include" {file_path}`
+   - 分析与其他模块的交互
+
+## 分析工具使用指南
+
+### read_code
+- **首选工具**: 用于读取和分析文件内容
+- **用法指南**:
+  - 读取整个文件: 直接指定文件路径
+  - 读取部分内容: 指定文件路径和行范围
+  - 读取头部或关键部分: 根据目的选择合适的行范围
+
+### execute_shell
+- **用途**: 执行辅助命令进行分析
+- **推荐命令**:
+  - `rg "pattern" {file_path}`: 在文件中搜索模式
+  - `loc {file_path}`: 获取文件代码统计
+  - `rg -n "class|def|function" {file_path}`: 查找结构元素
 
-## 探索建议
-- 首先读取整个文件内容以获取全局视图
-- 识别与分析目标相关的关键部分
-- 深入探索这些关键部分，提供有针对性的分析
+### 其他专业工具
+- **使用时机**: 仅当read_code和execute_shell不足以完成分析时
+- **使用前提**: 必须先尝试使用基本工具解决问题
+- **选择原则**: 根据实际需要选择最简洁有效的工具
 
-## 执行指令
-- 首先读取整个文件内容:
-  ```
-  read_code {{
-    "files": [{{
-      "path": "{file_path}",
-      "start_line": 1,
-      "end_line": -1
-    }}]
-  }}
-  ```
+## 分析框架适应
+根据文件类型和编程范式调整分析重点:
 
-- 分析文件统计信息:
-  ```
-  wc -l {file_path}
-  ```
+### 不同编程范式
+- 面向对象: 类层次、继承、封装、接口实现
+- 函数式: 函数组合、不可变性、纯函数
+- 过程式: 流程控制、状态管理、数据处理
+- 声明式: 规则定义、约束表达、模式匹配
 
-- 分析关键符号和函数:
-  ```
-  find_symbol {{
-    "symbol": "关键符号名称",
-    "root_dir": "."
-  }}
-  
-  function_analyzer {{
-    "function_name": "关键函数名称",
-    "file_path": "{file_path}"
-  }}
-  ```
+### 不同文件类型
+- 源代码文件: 实现逻辑、算法、接口设计
+- 配置文件: 参数设置、环境变量、系统选项
+- 模板文件: 渲染逻辑、变量占位符、条件分支
+- 数据文件: 结构组织、关系定义、索引设计
 
 ## 输出要求
 - 直接回应分析目标的关键问题
 - 提供与目标相关的深入洞察
 - 分析内容应直接服务于分析目标
 - 避免与目标无关的冗余信息
-- 使用具体代码路径和示例支持分析结论
-- 提供针对分析目标的具体建议和改进方向"""
+- 使用具体代码片段支持分析结论
+- 提供针对分析目标的具体建议
+- 保证全面分析相关信息后再得出结论"""
 
     def _create_summary_prompt(self, file_path: str, file_name: str) -> str:
         """