jarvis-ai-assistant 0.1.128__py3-none-any.whl → 0.1.130__py3-none-any.whl

jarvis/jarvis_code_agent/builtin_input_handler.py

@@ -0,0 +1,43 @@
+import re
+from typing import Any, Tuple
+
+from jarvis.jarvis_utils.output import PrettyOutput, OutputType
+
+
+def builtin_input_handler(user_input: str, agent: Any) -> Tuple[str, bool]:
+    """
+    处理内置的特殊输入标记，并追加相应的提示词
+    
+    参数：
+        user_input: 用户输入
+        agent: 代理对象
+        
+    返回：
+        Tuple[str, bool]: 处理后的输入和是否需要进一步处理
+    """
+    # 查找特殊标记
+    special_tags = re.findall(r"'<([^>]+)>'", user_input)
+    
+    if not special_tags:
+        return user_input, False
+    
+    # 使用集合去重
+    processed_tags = set()
+    # 处理每个标记
+    for tag in special_tags:
+        if tag in processed_tags:
+            continue
+        processed_tags.add(tag)
+        
+        if tag == "CodeBase":
+            user_input = user_input.replace(f"'<{tag}>'", "")
+            user_input += "\n请使用ask_codebase工具查询代码库"
+        elif tag == "Web":
+            user_input = user_input.replace(f"'<{tag}>'", "")
+            user_input += "\n请使用search_web工具进行网页搜索"
+        elif tag == "RAG":
+            user_input = user_input.replace(f"'<{tag}>'", "")
+            user_input += "\n请使用rag工具进行知识库检索"
+        # 移除对未知标记的警告输出
+    
+    return user_input, False

jarvis/jarvis_code_agent/code_agent.py

@@ -6,6 +6,8 @@ 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_platform.registry import PlatformRegistry
@@ -21,83 +23,6 @@ from jarvis.jarvis_utils.utils import init_env, user_confirm
 
 
 
-def file_input_handler(user_input: str, agent: Any) -> Tuple[str, bool]:
-    prompt = user_input
-    files = []
-    
-    file_refs = re.findall(r"'([^']+)'", user_input)
-    for ref in file_refs:
-        # Handle file:start,end or file:start:end format
-        if ':' in ref:
-            file_path, line_range = ref.split(':', 1)
-            # Initialize with default values
-            start_line = 1  # 1-based
-            end_line = -1
-            
-            # Process line range if specified
-            if ',' in line_range or ':' in line_range:
-                try:
-                    raw_start, raw_end = map(int, re.split(r'[,:]', line_range))
-                    
-                    # Handle special values and Python-style negative indices
-                    try:
-                        with open(file_path, 'r', encoding='utf-8') as f:
-                            total_lines = len(f.readlines())
-                    except FileNotFoundError:
-                        PrettyOutput.print(f"文件不存在: {file_path}", OutputType.WARNING)
-                        continue
-                    # Process start line
-                    if raw_start == 0:  # 0表示整个文件
-                        start_line = 1
-                        end_line = total_lines
-                    else:
-                        start_line = raw_start if raw_start > 0 else total_lines + raw_start + 1
-                    
-                    # Process end line
-                    if raw_end == 0:  # 0表示整个文件（如果start也是0）
-                        end_line = total_lines
-                    else:
-                        end_line = raw_end if raw_end > 0 else total_lines + raw_end + 1
-                    
-                    # Auto-correct ranges
-                    start_line = max(1, min(start_line, total_lines))
-                    end_line = max(start_line, min(end_line, total_lines))
-                    
-                    # Final validation
-                    if start_line < 1 or end_line > total_lines or start_line > end_line:
-                        raise ValueError
-
-                except:
-                    continue
-            
-            # Add file if it exists
-            if os.path.isfile(file_path):
-                files.append({
-                    "path": file_path,
-                    "start_line": start_line,
-                    "end_line": end_line
-                })
-        else:
-            # Handle simple file path
-            if os.path.isfile(ref):
-                files.append({
-                    "path": ref,
-                    "start_line": 1,  # 1-based
-                    "end_line": -1
-                })
-        
-    # Read and process files if any were found
-    if files:
-        with yaspin(text="正在读取文件...", color="cyan") as spinner:
-            result = FileOperationTool().execute({"operation":"read","files": files})
-            if result["success"]:
-                spinner.text = "文件读取完成"
-                spinner.ok("✅")
-                return result["stdout"] + "\n" + prompt, False
-    
-    return prompt, False
-
-
 
 class CodeAgent:
     def __init__(self):
@@ -116,89 +41,91 @@ class CodeAgent:
                                  "lsp_prepare_rename"
                                  ])
         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_get_document_symbols映射代码结构
+- 通过lsp_find_references和lsp_find_definition追踪依赖关系
+- 在进行更改前识别潜在影响区域
+- 为安全创建回滚计划
+
+### 实施阶段
+- 进行原子化、聚焦的最小范围更改
+- 保持一致的代码风格和格式
+- 每次重大更改后运行lsp_get_diagnostics
+- 立即修复任何检测到的问题
+
+### 验证阶段
+- 使用lsp_get_diagnostics进行全面诊断
+- 用lsp_prepare_rename验证所有重命名元素
+- 检查相关代码中的意外副作用
+- 确保必要的向后兼容性
+
+### 文档阶段
+- 为每项修改提供清晰的理由
+- 在所有补丁描述中包含上下文
+- 记录任何假设或约束条件
+- 准备详细的提交信息
+
+## 代码修改协议
+1. **修改前要求**：
    - 完整的代码分析报告
-   - 影响评估矩阵
-   - 回滚程序文档
+   - 影响评估
+   - 验证策略
 
-2. 修改实施：
-   - 单一职责修改
-   - 严格的代码范围验证（±3行缓冲区）
+2. **修改实施**：
+   - 单一职责变更
+   - 严格的范围验证
    - 接口兼容性检查
 
-3. 验证清单：
-   [ ] 执行lsp_get_diagnostics并确保零错误
-   [ ] 使用lsp_find_references确认影响范围
-   [ ] 使用lsp_prepare_rename验证重命名安全性
+3. **验证清单**：
+   - 运行lsp_get_diagnostics确保零错误
+   - 使用lsp_find_references确认影响范围
+   - 用lsp_prepare_rename验证重命名安全性
 
-4. 修改后：
+4. **修改后流程**：
    - 代码审查模拟
    - 版本控制审计
-   - 更新变更日志
-
-## 关键要求
-1. 强制分析：
-   - 修改前完整符号追踪
-   - 跨文件影响分析
-   - 依赖关系映射
-
-2. 禁止操作：
-   - 未通过lsp_get_diagnostics检查继续操作
-   - 多个功能组合修改
-   - 未经测试的接口修改
-
-3. 紧急协议：
-   - lsp_get_diagnostics出现错误时立即停止并回滚
-   - 出现意外行为时通知用户
-   - 对任何回归进行事后分析
+   - 更新变更文档
+
+## 工具使用指南
+- **分析工具**：
+  - lsp_get_document_symbols：用于映射代码结构
+  - lsp_find_references：用于理解使用模式
+  - lsp_find_definition：用于追踪实现细节
+
+- **验证工具**：
+  - lsp_prepare_rename：用于安全重构检查
+  - lsp_get_diagnostics：用于修改后检查
+
+- **系统工具**：
+  - execute_shell：用于git操作和grep搜索
+  - ask_codebase：用于查询代码知识库
+  - search_web：用于技术参考查找
+
+## 补丁格式要求
+创建代码补丁时：
+1. 包含足够的上下文（更改前后各3行）
+2. 保留原始缩进和格式
+3. 为新文件提供完整代码
+4. 对于修改，保留周围未更改的代码
+5. 为每项更改包含清晰的理由说明
 """
         self.agent = Agent(system_prompt=code_system_prompt, 
                            name="CodeAgent", 
@@ -208,7 +135,7 @@ class CodeAgent:
                            output_handler=[tool_registry, PatchOutputHandler()], 
                            platform=PlatformRegistry().get_codegen_platform(), 
                            record_methodology=False,
-                           input_handler=[shell_input_handler, file_input_handler],
+                           input_handler=[shell_input_handler, file_input_handler, builtin_input_handler],
                            need_summary=False)
 
     
@@ -222,8 +149,7 @@ class CodeAgent:
                 with spinner.hidden():
                     git_commiter = GitCommitTool()
                     git_commiter.execute({})
-            else:
-                spinner.text = "环境初始化完成"
+            spinner.text = "环境初始化完成"
             spinner.ok("✅")
 
     

jarvis/jarvis_code_agent/file_input_handler.py

@@ -0,0 +1,88 @@
+
+
+import os
+import re
+from typing import Any, Tuple
+
+from yaspin import yaspin
+
+from jarvis.jarvis_tools.file_operation import FileOperationTool
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+
+
+def file_input_handler(user_input: str, agent: Any) -> Tuple[str, bool]:
+    prompt = user_input
+    files = []
+    
+    file_refs = re.findall(r"'([^']+)'", user_input)
+    for ref in file_refs:
+        # Handle file:start,end or file:start:end format
+        if ':' in ref:
+            file_path, line_range = ref.split(':', 1)
+            # Initialize with default values
+            start_line = 1  # 1-based
+            end_line = -1
+            
+            # Process line range if specified
+            if ',' in line_range or ':' in line_range:
+                try:
+                    raw_start, raw_end = map(int, re.split(r'[,:]', line_range))
+                    
+                    # Handle special values and Python-style negative indices
+                    try:
+                        with open(file_path, 'r', encoding='utf-8') as f:
+                            total_lines = len(f.readlines())
+                    except FileNotFoundError:
+                        PrettyOutput.print(f"文件不存在: {file_path}", OutputType.WARNING)
+                        continue
+                    # Process start line
+                    if raw_start == 0:  # 0表示整个文件
+                        start_line = 1
+                        end_line = total_lines
+                    else:
+                        start_line = raw_start if raw_start > 0 else total_lines + raw_start + 1
+                    
+                    # Process end line
+                    if raw_end == 0:  # 0表示整个文件（如果start也是0）
+                        end_line = total_lines
+                    else:
+                        end_line = raw_end if raw_end > 0 else total_lines + raw_end + 1
+                    
+                    # Auto-correct ranges
+                    start_line = max(1, min(start_line, total_lines))
+                    end_line = max(start_line, min(end_line, total_lines))
+                    
+                    # Final validation
+                    if start_line < 1 or end_line > total_lines or start_line > end_line:
+                        raise ValueError
+
+                except:
+                    continue
+            
+            # Add file if it exists
+            if os.path.isfile(file_path):
+                files.append({
+                    "path": file_path,
+                    "start_line": start_line,
+                    "end_line": end_line
+                })
+        else:
+            # Handle simple file path
+            if os.path.isfile(ref):
+                files.append({
+                    "path": ref,
+                    "start_line": 1,  # 1-based
+                    "end_line": -1
+                })
+        
+    # Read and process files if any were found
+    if files:
+        with yaspin(text="正在读取文件...", color="cyan") as spinner:
+            result = FileOperationTool().execute({"operation":"read","files": files})
+            if result["success"]:
+                spinner.text = "文件读取完成"
+                spinner.ok("✅")
+                return result["stdout"] + "\n" + prompt, False
+    
+    return prompt, False
+