jarvis-ai-assistant 0.1.120__py3-none-any.whl → 0.1.122__py3-none-any.whl

jarvis/jarvis_code_agent/patch.py

@@ -21,246 +21,112 @@ class PatchOutputHandler(OutputHandler):
     
     def prompt(self) -> str:
         return """
-# 🔄 REPLACE: Modify existing code
-<REPLACE>
-File: path/to/file
-Lines: [start,end] or [start,end)
-[new content]
-...
-</REPLACE>
-
-# ➕ INSERT: Add new code
-<INSERT>
-File: path/to/file
-Line: position
-[new content]
-...
-</INSERT>
-
-# 🗑️ DELETE: Remove existing code
-<DELETE>
-File: path/to/file
-Lines: [start,end] or [start,end)
-</DELETE>
-
-# 🆕 NEW_FILE: Create new file
-<NEW_FILE>
-File: path/to/file
-[new content]
-...
-</NEW_FILE>
-
-# ➡️ MOVE_FILE: Relocate a file
-<MOVE_FILE>
-File: path/to/source/file
-NewPath: path/to/destination/file
-</MOVE_FILE>
-
-# ❌ REMOVE_FILE: Delete entire file
-<REMOVE_FILE>
-File: path/to/file
-</REMOVE_FILE>
-
-# 📋 Formatting Rules
-1. File Paths
-   - Use relative paths from project root
-   - Must be exact and case-sensitive
-   - Example: src/module/file.py
-   
-2. Line Numbers
-   - Format: [start,end] (inclusive) or [start,end) (right-exclusive)
-   - 1-based line numbers
-   - Single number for INSERT
-   - Omit for NEW_FILE/REMOVE_FILE
-
-3. Content
-   - Maintain original indentation
-   - Follow existing code style
-
-# 📌 Usage Examples
-## REPLACE Example (Closed Interval)
-<REPLACE>
-File: src/utils.py
-Lines: [9,13]
+# 🛠️ Simplified Patch Format
+<PATCH>
+File path [Operation parameters]
+Code content
+</PATCH>
+
+Operation types:
+- Replace: [Start line,End line] Replace line range (e.g. [5,8] replaces lines 5-8)
+- Delete: [Start line,End line] Delete line range (e.g. [10,10] deletes line 10)
+- Insert: [Line number] Insert before specified line (e.g. [3] inserts before line 3)
+- New file: [1] Create new file
+
+Examples:
+# Replace operation
+<PATCH>
+src/app.py [5,8]
 def updated_function():
-    # Replaces lines 9-13 inclusive
-    return "new_implementation"
-</REPLACE>
+    print("Replaced lines 5-8")
+    return new_value * 2
+</PATCH>
 
-## REPLACE Example (Left-Closed Right-Open)
-<REPLACE>
-File: src/calculator.py
-Lines: [5,8)
-def new_calculation():
-    # Replaces lines 5-7 (excludes line 8)
-    return 42
-</REPLACE>
+# Delete operation
+<PATCH>
+src/old.py [10,10]
+</PATCH>
 
-## INSERT Example
-<INSERT>
-File: src/main.py
-Line: 19
-    # Inserted before line 19
-    new_feature()
-</INSERT>
+# Insert operation
+<PATCH>
+utils/logger.py [3]
+print("Inserted before original line 3")
+</PATCH>
 
-## NEW_FILE Example
-<NEW_FILE>
-File: src/new_module.py
 # New file creation
-def feature_entry():
-    pass
-</NEW_FILE>
-
-## DELETE Example
-<DELETE>
-File: src/utils.py
-Lines: [9,13]
-</DELETE>
-
-## MOVE_FILE Example
-<MOVE_FILE>
-File: src/old_dir/file.py
-NewPath: src/new_dir/file.py
-</MOVE_FILE>
-
-## REMOVE_FILE Example
-<REMOVE_FILE>
-File: src/obsolete.py
-</REMOVE_FILE>
-
-# 🚨 Critical Requirements
-1. One change per block
-2. Use correct operation type
-3. Match existing code style
-4. Preserve indentation levels
-5. Exact file paths required
-6. Handle edge cases properly
-7. Include error handling
-8. Maintain code consistency
+<PATCH>
+config.yaml [1]
+database:
+  host: localhost
+  port: 5432
+</PATCH>
 """
 
 
 def _parse_patch(patch_str: str) -> Dict[str, List[Dict[str, Any]]]:
-    """Parse patches from string with optimized format"""
+    """解析补丁格式"""
     result = {}
-    patches = re.findall(r"<(REPLACE|INSERT|DELETE|NEW_FILE|REMOVE_FILE|MOVE_FILE)>(.*?)</\1>", patch_str, re.DOTALL)
+    header_pattern = re.compile(
+        r'^\s*"?(.+?)"?\s*\[(\d+)(?:,(\d+))?\]\s*$'  # Match file path and line number
+    )
+    patches = re.findall(r'<PATCH>\n?(.*?)\n?</PATCH>', patch_str, re.DOTALL)
     
-    for patch_type, patch in patches:
-        lines = patch.strip().split('\n')
-        if not lines:
+    for patch in patches:
+        # 分割首行和内容
+        parts = patch.split('\n', 1)
+        if len(parts) < 1:
             continue
+        header_line = parts[0].strip()
+        content = parts[1] if len(parts) > 1 else ''
+        
+        # 仅在内容非空时添加换行符
+        if content and not content.endswith('\n'):
+            content += '\n'
             
-        # Parse file path
-        file_match = re.match(r"File:\s*([^\s]+)", lines[0])
-        if not file_match:
+        # 解析文件路径和行号
+        header_match = header_pattern.match(header_line)
+        if not header_match:
             continue
-        filepath = file_match.group(1).strip()
-        
-        # Initialize line numbers
-        start_line = end_line = 0
-        
-        # Parse line numbers based on operation type
-        if patch_type in ['REPLACE', 'DELETE']:
-            # 增强正则表达式兼容性
-            line_match = re.match(
-                r"^Lines:\s*\[\s*(\d+)\s*,\s*(\d+)\s*([\]\)])\s*$",  # 匹配行尾
-                lines[1].strip(),  # 去除前后空格
-                re.IGNORECASE
-            )
-            if line_match:
-                start_line = int(line_match.group(1))
-                end_value = int(line_match.group(2))
-                bracket_type = line_match.group(3).strip()
-                
-                # 根据括号类型处理区间
-                if bracket_type == ')':  # [m,n)
-                    end_line = end_value - 1
-                else:  # [m,n]
-                    end_line = end_value
-                
-                # 确保 end_line >= start_line
-                end_line = max(end_line, start_line)
-            else:
-                PrettyOutput.print(f"无法解析行号格式: {lines[1]}", OutputType.WARNING)
-                continue
-        elif patch_type == 'INSERT':
-            line_match = re.match(r"Line:\s*(\d+)", lines[1])
-            if line_match:
-                start_line = int(line_match.group(1))  # 1-based
-                end_line = start_line
-        elif patch_type == 'MOVE_FILE':
-            new_path_match = re.match(r"NewPath:\s*([^\s]+)", lines[1])
-            if new_path_match:
-                new_path = new_path_match.group(1).strip()
-            else:
-                continue
-        
-        # Get content (after metadata)
-        if patch_type in ['REPLACE', 'DELETE']:
-            content_start = 2  # File + Lines
-        elif patch_type == 'INSERT':
-            content_start = 2   # File + Line
-        elif patch_type == 'NEW_FILE':
-            content_start = 1   # File
-        elif patch_type == 'MOVE_FILE':
-            content_start = 2   # File + NewPath
-        elif patch_type == 'REMOVE_FILE':
-            content_start = 1   # File
-        
-        content_lines = lines[content_start:]
-        content = '\n'.join(content_lines).strip()
 
+        filepath = header_match.group(1)
+        start = int(header_match.group(2))       # 保持1-based行号
+        end = int(header_match.group(3)) + 1 if header_match.group(3) else start
+
+        # 存储参数
         if filepath not in result:
             result[filepath] = []
-        
-        # Handle MOVE_FILE specially
-        if patch_type == 'MOVE_FILE':
-            result[filepath].append({
-                'type': patch_type,
-                'new_path': new_path,
-                'content': content
-            })
-        else:
-            result[filepath].append({
-                'type': patch_type,
-                'start_line': start_line,
-                'end_line': end_line,
-                'content': content
-            })
-    
-    # Sort patches by start line in reverse order to apply from bottom to top
-    for filepath in result:
-        result[filepath].sort(key=lambda x: x.get('start_line', 0), reverse=True)
-    
+        result[filepath].append({
+            'filepath': filepath,
+            'start': start,
+            'end': end,
+            'content': content  # 保留原始内容（可能为空）
+        })
+    for filepath in result.keys():
+        result[filepath] = sorted(result[filepath], key=lambda x: x['start'], reverse=True)
     return result
 
 
 def apply_patch(output_str: str) -> str:
     """Apply patches to files"""
-    patches = _parse_patch(output_str)
-    ret = ""  # Initialize return value
-
-    for filepath, patch_info in patches.items():
-        try:
-            for patch in patch_info:
-                patch_type = patch['type']
-                
-                if patch_type == 'MOVE_FILE':
-                    handle_move_file(filepath, patch)
-                elif patch_type == 'NEW_FILE':
-                    handle_new_file(filepath, patch)
-                elif patch_type == 'REMOVE_FILE':
-                    handle_remove_file(filepath)
-                else:
-                    handle_code_operation(filepath, patch)
-            
-        except Exception as e:
-            PrettyOutput.print(f"应用 {patch_type} 操作到 {filepath} 失败: {str(e)}", OutputType.ERROR)
-            continue
+    try:
+        patches = _parse_patch(output_str)
+    except Exception as e:
+        PrettyOutput.print(f"解析补丁失败: {str(e)}", OutputType.ERROR)
+        return ""
 
+    ret = ""
+    
+    for filepath, patch_list in patches.items():
+        for patch in patch_list:
+            try:
+                handle_code_operation(filepath, patch)
+                PrettyOutput.print(f"成功处理 操作", OutputType.SUCCESS)
+            except Exception as e:
+                PrettyOutput.print(f"操作失败: {str(e)}", OutputType.ERROR)
+    
     if has_uncommitted_changes():
-        if handle_commit_workflow():
+        diff = get_diff()
+        if handle_commit_workflow(diff):
             ret += "Successfully applied the patch\n"
             # Get modified line ranges
             modified_ranges = get_modified_line_ranges()
@@ -269,25 +135,37 @@ def apply_patch(output_str: str) -> str:
                 ret += "New code:\n"
                 ret += modified_code["stdout"]
         else:
-            ret += "User rejected the patch"
-        user_input = get_multiline_input("你可以继续输入: ")
+            ret += "User rejected the patch\nThis is your patch preview:\n"
+            ret += diff
+        user_input = get_multiline_input("你可以继续输入（输入空行重试，Ctrl+C退出）: ")
         if user_input:
             ret += "\n" + user_input
         else:
-            return ""
+            ret = ""
 
     return ret  # Ensure a string is always returned
-    
-def handle_commit_workflow()->bool:
+
+def get_diff() -> str:
+    """使用更安全的subprocess代替os.system"""
+    import subprocess
+    try:
+        subprocess.run(['git', 'add', '.'], check=True)
+        result = subprocess.run(
+            ['git', 'diff', 'HEAD'],
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        return result.stdout
+    finally:
+        subprocess.run(['git', 'reset', 'HEAD'], check=True)
+
+def handle_commit_workflow(diff:str)->bool:
     """Handle the git commit workflow and return the commit details.
     
     Returns:
         tuple[bool, str, str]: (continue_execution, commit_id, commit_message)
     """
-    os.system("git add .")
-    diff = os.popen("git diff HEAD").read()
-    os.system("git reset HEAD")
-    PrettyOutput.print(diff, OutputType.CODE, lang="diff")
     if not user_confirm("是否要提交代码？", default=True):
         os.system("git reset HEAD")
         os.system("git checkout -- .")
@@ -303,7 +181,7 @@ def get_modified_line_ranges() -> Dict[str, Tuple[int, int]]:
     
     Returns:
         Dictionary mapping file paths to tuple with (start_line, end_line) ranges
-        for modified sections. Line numbers are 0-based.
+        for modified sections. Line numbers are 1-based.
     """
     # Get git diff for all files
     diff_output = os.popen("git show").read()
@@ -322,91 +200,70 @@ def get_modified_line_ranges() -> Dict[str, Tuple[int, int]]:
         # Match lines like "@@ -100,5 +100,7 @@" where the + part shows new lines
         range_match = re.match(r"^@@ -\d+(?:,\d+)? \+(\d+)(?:,(\d+))? @@", line)
         if range_match and current_file:
-            start_line = int(range_match.group(1)) - 1  # Convert to 0-based
+            start_line = int(range_match.group(1))  # Keep as 1-based
             line_count = int(range_match.group(2)) if range_match.group(2) else 1
-            end_line = start_line + line_count
+            end_line = start_line + line_count - 1
             result[current_file] = (start_line, end_line)
     
     return result
-
 # New handler functions below ▼▼▼
 
-def handle_move_file(filepath: str, patch: Dict[str, Any]):
-    """Handle file moving operation"""
-    new_path = patch['new_path']
-    os.makedirs(os.path.dirname(new_path), exist_ok=True)
-    if os.path.exists(filepath):
-        os.rename(filepath, new_path)
-        PrettyOutput.print(f"成功移动文件 {filepath} -> {new_path}", OutputType.SUCCESS)
-    else:
-        PrettyOutput.print(f"源文件不存在: {filepath}", OutputType.WARNING)
-
 def handle_new_file(filepath: str, patch: Dict[str, Any]):
-    """Handle new file creation"""
-    new_content = patch.get('content', '').splitlines(keepends=True)
-    if new_content and new_content[-1] and new_content[-1][-1] != '\n':
-        new_content[-1] += '\n'
-    
+    """统一参数格式处理新文件"""
     os.makedirs(os.path.dirname(filepath), exist_ok=True)
     with open(filepath, 'w', encoding='utf-8') as f:
-        f.writelines(new_content)
-    PrettyOutput.print(f"成功创建新文件 {filepath}", OutputType.SUCCESS)
-
-def handle_remove_file(filepath: str):
-    """Handle file removal"""
-    if os.path.exists(filepath):
-        os.remove(filepath)
-        PrettyOutput.print(f"成功删除文件 {filepath}", OutputType.SUCCESS)
-    else:
-        PrettyOutput.print(f"文件不存在: {filepath}", OutputType.WARNING)
+        f.write(patch['content'])
 
 def handle_code_operation(filepath: str, patch: Dict[str, Any]):
-    """Handle code modification operations (REPLACE/INSERT/DELETE)"""
-    patch_type = patch['type']
-    start_line = patch.get('start_line', 0)
-    end_line = patch.get('end_line', 0)
-    new_content = patch.get('content', '').splitlines(keepends=True)
-
-    if not new_content:
-        new_content = ['']
-
-    PrettyOutput.print(f"patch_type: {patch_type}\nstart_line: {start_line}\nend_line: {end_line}\nnew_content:\n{new_content}", OutputType.INFO)
-
-    if new_content and new_content[-1] and new_content[-1][-1] != '\n':
-        new_content[-1] += '\n'
-
-    if not os.path.exists(filepath):
-        PrettyOutput.print(f"文件不存在: {filepath}", OutputType.WARNING)
-        return
+    """处理紧凑格式补丁"""
+    try:
+        # 新建文件时强制覆盖
+        os.makedirs(os.path.dirname(filepath), exist_ok=True)
+        if not os.path.exists(filepath):
+            open(filepath, 'w', encoding='utf-8').close()
+        with open(filepath, 'r+', encoding='utf-8') as f:
+            lines = f.readlines()
+            
+            new_lines = validate_and_apply_changes(
+                lines,
+                patch['start'],
+                patch['end'],
+                patch['content']
+            )
+            
+            f.seek(0)
+            f.writelines(new_lines)
+            f.truncate()
 
-    with open(filepath, 'r+', encoding='utf-8') as f:
-        lines = f.readlines()
-        validate_and_apply_changes(patch_type, lines, start_line, end_line, new_content)
-        f.seek(0)
-        f.writelines(lines)
-        f.truncate()
+        PrettyOutput.print(f"成功更新 {filepath}", OutputType.SUCCESS)
 
-    PrettyOutput.print(f"成功对 {filepath} 执行 {patch_type} 操作", OutputType.SUCCESS)
+    except Exception as e:
+        PrettyOutput.print(f"操作失败: {str(e)}", OutputType.ERROR)
 
 def validate_and_apply_changes(
-    patch_type: str,
     lines: List[str],
-    start_line: int,
-    end_line: int,
-    new_content: List[str]
-):
-    """Validate and apply code changes to in-memory file content"""
-    if patch_type in ['REPLACE', 'DELETE']:
-        if start_line < 1 or end_line > len(lines) or start_line > end_line:
-            raise ValueError(f"Invalid line range [{start_line}, {end_line}] (total lines: {len(lines)})")
-        
-        if patch_type == 'REPLACE':
-            lines[start_line-1:end_line] = new_content
-        else:  # DELETE
-            lines[start_line-1:end_line] = []
-            
-    elif patch_type == 'INSERT':
-        if start_line < 1 or start_line > len(lines) + 1:
-            raise ValueError(f"Invalid insertion position [{start_line}] (valid range: 1-{len(lines)+1})")
-        
-        lines[start_line-1:start_line-1] = new_content
+    start: int,
+    end: int,
+    content: str
+) -> List[str]:
+
+    new_content = content.splitlines(keepends=True)
+    
+    # 插入操作处理
+    if start == end:
+        if start < 1 or start > len(lines)+1:
+            raise ValueError(f"无效插入位置: {start}")
+        # 在指定位置前插入
+        return lines[:start-1] + new_content + lines[start-1:]
+    
+    # 范围替换/删除操作
+    if start > end:
+        raise ValueError(f"起始行{start}不能大于结束行{end}")
+    
+    max_line = len(lines)
+    # 自动修正行号范围
+    start = max(1, min(start, max_line+1))
+    end = max(start, min(end, max_line+1))
+    
+    # 执行替换
+    return lines[:start-1] + new_content + lines[end-1:]

jarvis/jarvis_code_agent/relevant_files.py

@@ -60,14 +60,12 @@ Identify customization options:
 - "Should we create a new class?"
 
 # 🎨 Question Template
-```
+3-5 specific questions about existing implementations:
 <QUESTION>
-[3-5 specific questions about existing implementations:
 1. System architecture question
 2. Implementation details question
-3. Integration or extension question]
+3. Integration or extension question
 </QUESTION>
-```
 
 # 🔎 Investigation Focus
 1. Current System

jarvis/jarvis_codebase/main.py

@@ -336,7 +336,7 @@ Content: {content}
                 vectors = np.vstack(vectors)
                 if len(vectors) != len(ids):
                     PrettyOutput.print(f"向量数量不匹配: {len(vectors)} 个向量 vs {len(ids)} 个ID", 
-                                     output_type=OutputType.ERROR)
+                                     output_type=OutputType.WARNING)
                     self.index = None
                     return
                     
@@ -441,7 +441,7 @@ Content: {content}
                     output_lines.append("删除的文件:")
                     output_lines.extend(f"  {f}" for f in deleted_files)
                 
-                PrettyOutput.print("\n".join(output_lines), output_type=OutputType.WARNING)
+                PrettyOutput.print("\n".join(output_lines), output_type=OutputType.INFO)
 
                 # If force is True, continue directly
                 if not force:

jarvis/jarvis_lsp/registry.py

@@ -59,8 +59,8 @@ class LSPRegistry:
         
         if missing_methods:
             PrettyOutput.print(
-                f"LSP {lsp_class.__name__} is missing necessary methods: {', '.join(missing_methods)}", 
-                OutputType.ERROR
+                f"LSP {lsp_class.__name__} 缺少必要的方法: {', '.join(missing_methods)}", 
+                OutputType.WARNING
             )
             return False
             
@@ -72,7 +72,7 @@ class LSPRegistry:
         lsp_servers = {}
         
         if not os.path.exists(directory):
-            PrettyOutput.print(f"LSP 目录不存在: {directory}", OutputType.ERROR)
+            PrettyOutput.print(f"LSP 目录不存在: {directory}", OutputType.WARNING)
             return lsp_servers
             
         package_name = None
@@ -138,7 +138,7 @@ class LSPRegistry:
     def create_lsp(self, language: str) -> Optional[BaseLSP]:
         """Create LSP instance for specified language."""
         if language not in self.lsp_servers:
-            PrettyOutput.print(f"没有找到 LSP 支持的语言: {language}", OutputType.ERROR)
+            PrettyOutput.print(f"没有找到 LSP 支持的语言: {language}", OutputType.WARNING)
             return None
             
         try:
@@ -186,11 +186,11 @@ def main():
     lsp = registry.create_lsp(args.language)
     
     if not lsp:
-        PrettyOutput.print(f"没有 LSP 支持的语言: {args.language}", OutputType.ERROR)
+        PrettyOutput.print(f"没有 LSP 支持的语言: {args.language}", OutputType.WARNING)
         return 1
         
     if not lsp.initialize(os.path.dirname(os.path.abspath(args.file))):
-        PrettyOutput.print("LSP 初始化失败", OutputType.ERROR)
+        PrettyOutput.print("LSP 初始化失败", OutputType.WARNING)
         return 1
     
     try:
@@ -208,7 +208,7 @@ def main():
                 
         elif args.action in ('references', 'definition'):
             if args.line is None or args.character is None:
-                PrettyOutput.print("需要行和字符位置用于 references/definition", OutputType.ERROR)
+                PrettyOutput.print("需要行和字符位置用于 references/definition", OutputType.WARNING)
                 return 1
                 
             if args.action == 'references':