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

jarvis/jarvis_code_agent/code_agent.py

@@ -1,17 +1,14 @@
-import re
 import subprocess
 import os
-from typing import Any, Tuple
 
 from yaspin import yaspin
 
 from jarvis.jarvis_agent import Agent
-from jarvis.jarvis_code_agent.builtin_input_handler import builtin_input_handler
-from jarvis.jarvis_code_agent.file_input_handler import file_input_handler
-from jarvis.jarvis_code_agent.shell_input_handler import shell_input_handler
-from jarvis.jarvis_code_agent.patch import PatchOutputHandler
+from jarvis.jarvis_agent.builtin_input_handler import builtin_input_handler
+from jarvis.jarvis_agent.file_input_handler import file_input_handler
+from jarvis.jarvis_agent.shell_input_handler import shell_input_handler
+from jarvis.jarvis_agent.patch import PatchOutputHandler
 from jarvis.jarvis_platform.registry import PlatformRegistry
-from jarvis.jarvis_tools.file_operation import FileOperationTool
 from jarvis.jarvis_tools.git_commiter import GitCommitTool
 
 from jarvis.jarvis_tools.registry import ToolRegistry
@@ -34,96 +31,105 @@ class CodeAgent:
                                  "create_code_agent",
                                  "ask_user",  
                                  "ask_codebase",
-                                 "lsp_get_document_symbols", 
                                  "lsp_get_diagnostics", 
                                  "lsp_find_references", 
-                                 "lsp_find_definition", 
-                                 "lsp_prepare_rename"
+                                 "lsp_find_definition",
+                                 "treesitter_analyzer",
+                                 "code_review"  # 新增代码审查工具
                                  ])
         code_system_prompt = """
-# 角色：高级代码工程师
-精通安全、精确的代码修改，具有严格的验证流程。
-
-## 核心原则
-1. 安全第一：绝不破坏现有功能
-2. 精准工程：最小化、针对性修改
-3. 完整可追溯：记录所有决策
-4. 验证驱动：在每个阶段进行验证
-
-## 工具使用协议
-1. 分析工具：
-   - lsp_get_document_symbols：映射代码结构
-   - lsp_find_references：理解使用模式
-   - lsp_find_definition：追踪实现细节
-
-2. 验证工具：
-   - lsp_prepare_rename：安全重构检查
-   - lsp_get_diagnostics：修改后检查
-
-3. 系统工具：
-   - execute_shell：用于git操作和grep搜索
-   - ask_codebase：查询代码知识库
-   - search_web：技术参考查找
-
-## 工作流程（PDCA循环）
-1. 计划：
-   - 使用ask_user分析需求
-   - 使用LSP工具映射现有代码
-   - 使用find_references识别影响区域
-   - 使用git创建回滚计划
-
-2. 执行：
-   - 在受保护的块中进行原子修改
-   - 每次更改后自动运行lsp_get_diagnostics
-   - 如果发现错误，使用lsp_find_references和lsp_find_definition进行即时修复
-   - 每次更改后使用LSP验证语法
-
-3. 检查：
-   - 强制使用lsp_get_diagnostics进行完整诊断报告
-   - 使用lsp_preprepare_rename验证所有重命名
-   - 如果检测到错误，进入修复循环直到所有检查通过
-
-4. 行动：
-   - 使用git提交详细消息
-   - 准备回滚脚本（如果需要）
-   - 进行实施后审查
-
-## 代码修改标准
-1. 修改前要求：
+# 专业代码工程师
+
+## 身份与能力范围
+- **角色定位**：精通精确、安全代码修改的高级代码工程师
+- **核心能力**：代码分析、精准重构和系统化验证
+- **知识领域**：软件架构、设计模式和特定语言的最佳实践
+- **局限性**：复杂系统和特定领域知识需要明确的上下文支持
+
+## 交互原则与策略
+- **沟通风格**：清晰简洁的技术解释，为所有决策提供合理依据
+- **呈现格式**：使用补丁格式展示结构化的代码变更及其上下文
+- **主动引导**：在有益的情况下，提出超出即时需求的改进建议
+- **特殊场景应对**：
+  - 对于模糊请求：在继续前提出澄清问题
+  - 对于高风险变更：提出带有验证步骤的渐进式方法
+  - 对于性能问题：平衡即时修复与长期架构考量
+
+## 任务执行规范
+### 分析阶段
+- 通过lsp_find_references和lsp_find_definition追踪依赖关系
+- 使用treesitter_analyzer深入分析代码结构和关系
+- 在进行更改前识别潜在影响区域
+- 为安全创建回滚计划
+
+### 实施阶段
+- 进行原子化、聚焦的最小范围更改
+- 保持一致的代码风格和格式
+- 每次重大更改后运行lsp_get_diagnostics
+- 立即修复任何检测到的问题
+- 确保代码修改后第一时间验证，优先修复错误而不是继续实现新功能
+
+### 验证阶段
+- 使用lsp_get_diagnostics进行全面诊断
+- 检查相关代码中的意外副作用
+- 确保必要的向后兼容性
+
+### 文档阶段
+- 为每项修改提供清晰的理由
+- 在所有补丁描述中包含上下文
+- 记录任何假设或约束条件
+- 准备详细的提交信息
+
+## 代码修改协议
+1. **修改前要求**：
    - 完整的代码分析报告
-   - 影响评估矩阵
-   - 回滚程序文档
+   - 影响评估
+   - 验证策略
 
-2. 修改实施：
-   - 单一职责修改
-   - 严格的代码范围验证（±3行缓冲区）
+2. **修改实施**：
+   - 单一职责变更
+   - 严格的范围验证
    - 接口兼容性检查
 
-3. 验证清单：
-   [ ] 执行lsp_get_diagnostics并确保零错误
-   [ ] 使用lsp_find_references确认影响范围
-   [ ] 使用lsp_prepare_rename验证重命名安全性
+3. **验证清单**：
+   - 运行lsp_get_diagnostics确保零错误
+   - 使用lsp_find_references确认影响范围
 
-4. 修改后：
+4. **修改后流程**：
    - 代码审查模拟
    - 版本控制审计
-   - 更新变更日志
-
-## 关键要求
-1. 强制分析：
-   - 修改前完整符号追踪
-   - 跨文件影响分析
-   - 依赖关系映射
-
-2. 禁止操作：
-   - 未通过lsp_get_diagnostics检查继续操作
-   - 多个功能组合修改
-   - 未经测试的接口修改
-
-3. 紧急协议：
-   - lsp_get_diagnostics出现错误时立即停止并回滚
-   - 出现意外行为时通知用户
-   - 对任何回归进行事后分析
+   - 更新变更文档
+
+## 工具使用指南
+- **分析工具**：
+  - lsp_find_references：用于理解使用模式
+  - lsp_find_definition：用于追踪实现细节
+  - treesitter_analyzer：用于高级代码分析，支持以下功能：
+    - find_symbol：查找符号定义位置
+    - find_references：查找符号的所有引用
+    - find_callers：查找函数的所有调用处
+    
+    注意事项：
+    - 每次操作都会自动索引指定目录，无需单独执行索引步骤
+    - 适用于多种编程语言：Python, C, C++, Go, Rust 等
+    - 首次使用时会自动下载语法文件，可能需要一些时间
+
+- **验证工具**：
+  - lsp_get_diagnostics：用于修改后检查
+  - code_review：用于代码审查
+
+- **系统工具**：
+  - execute_shell：用于git操作和grep搜索
+  - ask_codebase：用于查询代码知识库
+  - search_web：用于技术参考查找
+
+## 补丁格式要求
+创建代码补丁时：
+1. 包含足够的上下文（更改前后各3行）
+2. 保留原始缩进和格式
+3. 为新文件提供完整代码
+4. 对于修改，保留周围未更改的代码
+5. 为每项更改包含清晰的理由说明
 """
         self.agent = Agent(system_prompt=code_system_prompt, 
                            name="CodeAgent", 
@@ -147,8 +153,7 @@ class CodeAgent:
                 with spinner.hidden():
                     git_commiter = GitCommitTool()
                     git_commiter.execute({})
-            else:
-                spinner.text = "环境初始化完成"
+            spinner.text = "环境初始化完成"
             spinner.ok("✅")
 
     
@@ -186,7 +191,7 @@ class CodeAgent:
             if commits and user_confirm("是否接受以上提交记录？", True):
                 if len(commits) > 1 and user_confirm("是否要合并为一个更清晰的提交记录？", True):
                     # Reset to start commit
-                    subprocess.run(["git", "reset", "--soft", start_commit], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+                    subprocess.run(["git", "reset", "--mixed", start_commit], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
                     # Create new commit
                     git_commiter = GitCommitTool()
                     git_commiter.execute({})

jarvis/jarvis_codebase/main.py

@@ -78,7 +78,7 @@ class CodeBase:
 
     def is_text_file(self, file_path: str):
         try:
-            open(file_path, "r", encoding="utf-8").read()
+            open(file_path, "r", encoding="utf-8", errors="ignore").read()
             return True
         except Exception:
             return False
@@ -233,7 +233,7 @@ class CodeBase:
                 return cached_vector
                 
             # Read the file content and combine information
-            content = open(file_path, "r", encoding="utf-8").read()[:self.max_token_count]  # Limit the file content length
+            content = open(file_path, "r", encoding="utf-8", errors="ignore").read()[:self.max_token_count]  # Limit the file content length
             
             # Combine file information, including file content
             combined_text = f"""
@@ -288,7 +288,7 @@ Content: {content}
             
             md5 = get_file_md5(file_path)
 
-            content = open(file_path, "r", encoding="utf-8").read()
+            content = open(file_path, "r", encoding="utf-8", errors="ignore").read()
             
             # Check if the file has already been processed and the content has not changed
             if file_path in self.vector_cache:
@@ -565,7 +565,7 @@ Content: {content}
                 
                 for path in initial_results:
                     try:
-                        content = open(path, "r", encoding="utf-8").read()
+                        content = open(path, "r", encoding="utf-8", errors="ignore").read()
                         # Truncate large files
                         if get_context_token_count(content) > max_file_length:
                             spinner.write(f"❌ 截断大文件: {path}")
@@ -861,7 +861,7 @@ Content: {content}
             
             for path in files_from_codebase:
                 try:
-                    content = open(path["file"], "r", encoding="utf-8").read()
+                    content = open(path["file"], "r", encoding="utf-8", errors="ignore").read()
                     file_content = f"""
     ## 文件: {path["file"]}
     ```