jarvis-ai-assistant 0.1.164__py3-none-any.whl → 0.1.166__py3-none-any.whl

jarvis/jarvis_platform/yuanbao.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, List, Tuple
 import requests
 import json
@@ -43,7 +44,7 @@ class YuanbaoPlatform(BasePlatform):
                 "   • 打开浏览器开发者工具 (F12 或右键 -> 检查)\n"
                 "   • 切换到网络标签\n"
                 "   • 发送任意消息\n"
-                "   • 在请求中找到 X-Uskey 和 T-UserID 头部值\n"
+                "   • 查看请求中的 Cookie 和 AgentID 值（具体位置见README.md中截图）\n"
                 "2. 设置环境变量:\n"
                 "   • 方法 1: 创建或编辑配置文件:\n"
                 f"   echo 'YUANBAO_COOKIES=your_cookies_here' >> {get_data_dir()}/env\n"
@@ -212,6 +213,7 @@ class YuanbaoPlatform(BasePlatform):
                     uploaded_files.append(file_metadata)
                     spinner.text = f"文件 {file_name} 上传成功"
                     spinner.ok("✅")
+                    time.sleep(3) # 上传成功后等待3秒
                 
                 except Exception as e:
                     spinner.text = f"上传文件 {file_path} 时出错: {str(e)}"

jarvis/jarvis_platform_manager/main.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import os
 from jarvis.jarvis_platform.registry import PlatformRegistry
 import asyncio

jarvis/jarvis_smart_shell/main.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# -*- coding: utf-8 -*-
 import argparse
 import os
 import sys

jarvis/jarvis_tools/ask_codebase.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 import os
 

jarvis/jarvis_tools/ask_user.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 # 导入所需的类型注解模块
 from typing import Dict, Any
 

jarvis/jarvis_tools/base.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any, Callable
 import json
 class Tool:

jarvis/jarvis_tools/chdir.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 import os
 

jarvis/jarvis_tools/code_plan.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """代码修改规划工具模块
 
 该模块提供CodePlanTool类，用于分析代码修改需求并制定详细的修改计划。

jarvis/jarvis_tools/create_code_agent.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 import os
 from jarvis.jarvis_code_agent.code_agent import CodeAgent

jarvis/jarvis_tools/create_sub_agent.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 import os
 

jarvis/jarvis_tools/edit_file.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """
 文件搜索替换工具类
 
@@ -13,7 +14,17 @@
 - 完善的错误处理和回滚机制
 - 严格的格式保持要求
 """
-from typing import Any, Dict
+import re
+from typing import Any, Dict, Tuple
+
+import yaml
+from yaspin import yaspin
+
+from jarvis.jarvis_platform.registry import PlatformRegistry
+from jarvis.jarvis_tools.file_operation import FileOperationTool
+from jarvis.jarvis_utils.git_utils import revert_file
+from jarvis.jarvis_utils.tag import ct, ot
+from jarvis.jarvis_utils.utils import is_context_overflow
 
 
 class FileSearchReplaceTool:
@@ -23,43 +34,27 @@ class FileSearchReplaceTool:
 # 代码编辑规范
 
 ## 重要提示
-此工具可以查看和修改单个文件的代码，只需提供要修改的代码片段即可。应尽量精简内容，只包含必要的上下文和修改部分。特别注意：不要提供完整文件内容，只提供需要修改的部分！
+此工具可以查看和修改单个文件的代码，只需提供要修改的代码片段即可。应尽量精简内容，只包含必要的上下文和修改部分。特别注意：不要提供完整文件内容，只提供需要修改的部分及其上下文！
 
 ## 基本使用
 1. 指定需要修改的文件路径
-2. 提供一组或多组"搜索文本"和"替换文本"对
-3. 每个搜索文本需在目标文件中有且仅有一次精确匹配
-4. 创建新文件时，使用空字符串("")作为搜索文本，替换文本作为完整文件内容
-5. 所有修改要么全部成功，要么全部失败并回滚
+2. 提供一组或多组"reason"和"patch"对
+3. 每个patch必须包含修改后的代码和1-2行上下文用于精确定位
 
 ## 核心原则
-1. **精准修改**：只提供需要修改的代码部分，不需要展示整个文件内容
+1. **精准修改**：只提供需要修改的代码部分及其上下文，不需要展示整个文件内容
 2. **最小补丁原则**：始终生成最小范围的补丁，只包含必要的上下文和实际修改
-3. **格式严格保持**：
-   - 严格保持原始代码的缩进方式（空格或制表符）
-   - 保持原始代码的空行数量和位置
-   - 保持原始代码的行尾空格处理方式
-   - 不改变原始代码的换行风格
-4. **新旧区分**：
-   - 对于新文件：提供完整的代码内容
-   - 对于现有文件：只提供修改部分，不要提供整个文件
-
-## 格式兼容性要求
-1. **缩进一致性**：
-   - 如果原代码使用4个空格缩进，替换文本也必须使用4个空格缩进
-   - 如果原代码使用制表符缩进，替换文本也必须使用制表符
-2. **空行保留**：
-   - 如果原代码在函数之间有两个空行，替换文本也必须保留这两个空行
-   - 如果原代码在类方法之间有一个空行，替换文本也必须保留这一个空行
-3. **行尾处理**：
-   - 不要改变原代码的行尾空格或换行符风格
-   - 保持原有的换行符类型(CR、LF或CRLF)
+3. **上下文定位**：确保提供的上下文能唯一标识修改位置
 
 ## 最佳实践
 1. 每个修改应专注于单一职责，避免包含过多无关代码
-2. 设计唯一的搜索文本，避免多处匹配的风险
-3. 创建新文件时提供完整、格式良好的内容
-4. 不要出现未实现的代码，如：TODO
+2. 不要出现未实现的代码，如：TODO
+3. 示例格式：
+  ```
+  # 原有上下文行
+  if condition:  # 修改这行
+      return new_value
+  ```
 """
     parameters = {
         "type": "object",
@@ -70,20 +65,20 @@ class FileSearchReplaceTool:
             },
             "changes": {
                 "type": "array",
-                "description": "一组或多组搜索替换对",
+                "description": "一组或多组修改，每个修改必须包含1-2行上下文用于精确定位",
                 "items": {
                     "type": "object",
                     "properties": {
-                        "search": {
+                        "reason": {
                             "type": "string",
-                            "description": "需要被替换的文本，必须在文件中唯一匹配，新文件用空字符串"
+                            "description": "修改的原因"
                         },
-                        "replace": {
-                            "type": "string", 
-                            "description": "替换的目标文本，需保持与原代码相同的缩进和格式风格"
+                        "patch": {
+                            "type": "string",
+                            "description": "修改后的代码片段，必须包含1-2行上下文代码用于精确定位修改位置，不需要传入完整文件内容"
                         }
                     },
-                    "required": ["search", "replace"]
+                    "required": ["reason", "patch"]
                 }
             }
         },
@@ -95,31 +90,31 @@ class FileSearchReplaceTool:
         pass
 
     def execute(self, args: Dict) -> Dict[str, Any]:
-        """
-        执行文件搜索替换操作，每个搜索文本只允许有且只有一次匹配，否则失败
-        特殊支持不存在的文件和空文件处理
+        """执行文件编辑操作，包含错误处理和回滚机制。
+
+        主要功能:
+        1. 处理文件创建或修改
+        2. 原子操作：所有修改要么全部成功，要么全部回滚
+        3. 保存修改前后的文件状态以便回滚
+        4. 提供详细的执行状态输出
 
         参数:
-            file (str): 文件路径
-            changes (list): 一组或多组搜索替换对，格式如下：
-            [
-                {
-                    "search": "搜索文本1",
-                    "replace": "替换文本1"
-                },
-                {
-                    "search": "搜索文本2",
-                    "replace": "替换文本2"
-                }
-            ]
+            args: 包含以下键的字典:
+                - file: 要修改的文件路径
+                - changes: 修改列表，每个修改包含:
+                    - reason: 修改原因描述
+                    - patch: 修改后的代码片段
 
         返回:
-            dict: 包含执行结果的字典
-            {
-                "success": bool,  # 是否成功完成所有替换
-                "stdout": str,    # 标准输出信息
-                "stderr": str     # 错误信息
-            }
+            Dict[str, Any] 包含:
+                - success: 操作是否成功
+                - stdout: 成功时的输出消息
+                - stderr: 失败时的错误消息
+
+        异常处理:
+        1. 捕获并记录文件操作异常
+        2. 失败的修改尝试回滚到原始状态
+        3. 新创建的文件在失败时会被删除
         """
         import os
         from jarvis.jarvis_utils.output import PrettyOutput, OutputType
@@ -146,49 +141,7 @@ class FileSearchReplaceTool:
                         content = f.read()
                         original_content = content
                 
-                # 创建一个临时内容副本进行操作
-                temp_content = content
-                replaced_count = 0
-                
-                # 处理所有搜索替换对
-                for change in changes:
-                    search_text = change["search"]
-                    replace_text = change["replace"]
-                    
-                    # 特殊处理不存在的文件或空文件
-                    if not file_exists or content == "":
-                        if search_text == "":
-                            # 对于不存在的文件或空文件，如果搜索文本为空，则直接使用替换文本作为内容
-                            temp_content = replace_text
-                            replaced_count += 1
-                            # 只允许有一个空字符串搜索
-                            break
-                        else:
-                            stderr_message = f"文件 {file_path} {'不存在' if not file_exists else '为空'}，但搜索文本非空: '{search_text}'"
-                            stderr_messages.append(stderr_message)
-                            PrettyOutput.print(stderr_message, OutputType.WARNING)
-                            success = False
-                            break
-                    else:
-                        # 正常文件处理 - 检查匹配次数
-                        match_count = temp_content.count(search_text)
-                        
-                        if match_count == 0:
-                            stderr_message = f"文件 {file_path} 中未找到匹配文本: '{search_text}'"
-                            stderr_messages.append(stderr_message)
-                            PrettyOutput.print(stderr_message, OutputType.WARNING)
-                            success = False
-                            break
-                        elif match_count > 1:
-                            stderr_message = f"文件 {file_path} 中匹配到多个 ({match_count}) '{search_text}'，搜索文本只允许一次匹配"
-                            stderr_messages.append(stderr_message)
-                            PrettyOutput.print(stderr_message, OutputType.WARNING)
-                            success = False
-                            break
-                        else:
-                            # 只有一个匹配，执行替换
-                            temp_content = temp_content.replace(search_text, replace_text, 1)
-                            replaced_count += 1
+                success, temp_content = patch_apply(file_path, yaml.safe_dump(changes))
 
                 # 只有当所有替换操作都成功时，才写回文件
                 if success and (temp_content != original_content or not file_exists):
@@ -200,14 +153,17 @@ class FileSearchReplaceTool:
                     
                     processed = True
                     
-                    action = "创建并写入" if not file_exists else "成功替换"
-                    stdout_message = f"文件 {file_path} {action} {replaced_count} 处"
+                    action = "创建并写入" if not file_exists else "成功修改"
+                    stdout_message = f"文件 {file_path} {action} 完成"
                     stdout_messages.append(stdout_message)
                     PrettyOutput.print(stdout_message, OutputType.SUCCESS)
                 elif success:
                     stdout_message = f"文件 {file_path} 没有找到需要替换的内容"
                     stdout_messages.append(stdout_message)
                     PrettyOutput.print(stdout_message, OutputType.INFO)
+                else:
+                    stdout_message = f"文件 {file_path} 修改失败"
+                    stdout_messages.append(stdout_message)
                 
             except Exception as e:
                 stderr_message = f"处理文件 {file_path} 时出错: {str(e)}"
@@ -215,35 +171,10 @@ class FileSearchReplaceTool:
                 PrettyOutput.print(stderr_message, OutputType.WARNING)
                 success = False
 
-            # 如果操作失败，回滚已修改的文件
-            if not success and processed:
-                rollback_message = "操作失败，正在回滚修改..."
-                stderr_messages.append(rollback_message)
-                PrettyOutput.print(rollback_message, OutputType.WARNING)
-                
-                try:
-                    if original_content is None:
-                        # 如果是新创建的文件，则删除
-                        if os.path.exists(file_path):
-                            os.remove(file_path)
-                        rollback_file_message = f"已删除新创建的文件: {file_path}"
-                    else:
-                        # 如果是修改的文件，则恢复原内容
-                        with open(file_path, 'w', encoding='utf-8') as f:
-                            f.write(original_content)
-                        rollback_file_message = f"已回滚文件: {file_path}"
-                        
-                    stderr_messages.append(rollback_file_message)
-                    PrettyOutput.print(rollback_file_message, OutputType.INFO)
-                except Exception as e:
-                    rollback_error = f"回滚文件 {file_path} 失败: {str(e)}"
-                    stderr_messages.append(rollback_error)
-                    PrettyOutput.print(rollback_error, OutputType.WARNING)
-
             return {
                 "success": success,
-                "stdout": "\n".join(stdout_messages),
-                "stderr": "\n".join(stderr_messages)
+                "stdout": "\n".join(stdout_messages) if success else "",
+                "stderr": "\n".join(stderr_messages) if not success else ""
             }
             
         except Exception as e:
@@ -276,4 +207,145 @@ class FileSearchReplaceTool:
                 "stderr": error_msg + "\n" + "\n".join(stderr_messages)
             }
 
-            
+
+def patch_apply(filepath: str, patch_content: str) -> Tuple[bool, str]:
+    """执行精确的文件编辑操作，使用AI模型生成差异补丁并应用。
+
+    功能概述:
+    1. 使用AI模型分析补丁内容并生成精确的代码差异
+    2. 应用生成的差异补丁到目标文件
+    3. 提供重试机制确保操作可靠性
+
+    参数:
+        filepath: 要编辑的文件路径
+        patch_content: YAML格式的补丁内容，包含修改原因和代码片段
+
+    返回值:
+        Tuple[bool, str]: 
+            - 第一个元素表示操作是否成功
+            - 第二个元素是修改后的文件内容(成功时)或错误信息(失败时)
+
+    异常处理:
+    1. 捕获并处理文件操作异常
+    2. 失败时自动回滚文件修改
+    3. 提供详细的执行状态输出
+    """
+    model = PlatformRegistry().get_normal_platform()
+    with yaspin(text=f"正在处理文件 {filepath}...", color="cyan") as spinner:
+        try:
+            file_content = FileOperationTool().execute({"operation":"read", "files":[{"path":filepath}]})["stdout"]
+            need_upload_file = is_context_overflow(file_content)
+            upload_success = False
+            # 读取原始文件内容
+            with spinner.hidden():  
+                if need_upload_file and model.upload_files([filepath]):
+                    upload_success = True
+
+
+            model.set_suppress_output(True)
+
+            main_prompt = f"""
+# 代码补丁生成专家指南
+
+## 任务描述
+你是一位精确的代码补丁生成专家，需要根据补丁描述生成精确的代码差异。
+
+### 补丁内容
+```
+{patch_content}
+```
+
+## 补丁生成要求
+1. **精确性**：严格按照补丁的意图修改代码
+2. **格式一致性**：严格保持原始代码的格式风格，如果补丁中缩进或者空行与原代码不一致，则需要修正补丁中的缩进或者空行
+3. **最小化修改**：只修改必要的代码部分，保持其他部分不变
+4. **上下文完整性**：提供足够的上下文，确保补丁能准确应用
+
+## 输出格式规范
+- 使用{ot("DIFF")}块包围每个需要修改的代码段
+- 每个{ot("DIFF")}块必须包含SEARCH部分和REPLACE部分
+- SEARCH部分是需要查找的原始代码
+- REPLACE部分是替换后的新代码
+- 确保SEARCH部分能在原文件中**唯一匹配**
+- 如果修改较大，可以使用多个{ot("DIFF")}块
+
+## 输出模板
+{ot("DIFF")}
+{">" * 5} SEARCH
+[需要查找的原始代码，包含足够上下文，避免出现可匹配多处的情况]
+{'='*5}
+[替换后的新代码]
+{"<" * 5} REPLACE
+{ct("DIFF")}
+
+{ot("DIFF")}
+{">" * 5} SEARCH
+[另一处需要查找的原始代码，包含足够上下文，避免出现可匹配多处的情况]
+{'='*5}
+[另一处替换后的新代码]
+{"<" * 5} REPLACE
+{ct("DIFF")}
+"""
+            
+            for _ in range(3):
+                file_prompt = ""
+                if not need_upload_file:
+                    file_prompt = f"""
+    # 原始代码
+    {file_content}
+    """
+                    
+                    response = model.chat_until_success(main_prompt + file_prompt)
+                else:
+                    if upload_success:
+                        response = model.chat_until_success(main_prompt)
+                    else:
+                        response = model.chat_big_content(file_content, main_prompt)
+
+                # 解析差异化补丁
+                diff_blocks = re.finditer(ot("DIFF")+r'\s*>{4,} SEARCH\n?(.*?)\n?={4,}\n?(.*?)\s*<{4,} REPLACE\n?'+ct("DIFF"),
+                                        response, re.DOTALL)
+
+                # 读取原始文件内容
+                with open(filepath, 'r', encoding='utf-8', errors="ignore") as f:
+                    file_content = f.read()
+
+                # 应用所有差异化补丁
+                modified_content = file_content
+                patch_count = 0
+                success = True
+                for match in diff_blocks:
+                    search_text = match.group(1).strip()
+                    replace_text = match.group(2).strip()
+                    patch_count += 1
+                    # 检查搜索文本是否存在于文件中
+                    if search_text in modified_content:
+                        # 如果有多处，报错
+                        if modified_content.count(search_text) > 1:
+                            spinner.write(f"❌ 补丁 #{patch_count} 应用失败：找到多个匹配的代码段")
+                            success = False
+                            break
+                        # 应用替换
+                        modified_content = modified_content.replace(
+                            search_text, replace_text)
+                        spinner.write(f"✅ 补丁 #{patch_count} 应用成功")
+                    else:
+                        spinner.write(f"❌ 补丁 #{patch_count} 应用失败：无法找到匹配的代码段")
+                        success = False
+                        break
+                if not success:
+                    revert_file(filepath)
+                    continue
+
+
+                spinner.text = f"文件 {filepath} 修改完成，应用了 {patch_count} 个补丁"
+                spinner.ok("✅")
+                return True, modified_content
+            spinner.text = f"文件 {filepath} 修改失败"
+            spinner.fail("❌")
+            return False, ""
+
+        except Exception as e:
+            spinner.text = f"文件修改失败: {str(e)}"
+            spinner.fail("❌")
+            return False, ""

jarvis/jarvis_tools/execute_script.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 import os
 import tempfile

jarvis/jarvis_tools/file_analyzer.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 import os
 

jarvis/jarvis_tools/file_operation.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 import os
 from pathlib import Path

jarvis/jarvis_tools/find_methodology.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from typing import Dict, Any
 from yaspin import yaspin
 
@@ -34,10 +35,14 @@ class FindMethodologyTool:
                     "stdout": "",
                     "stderr": "参数中必须包含查询文本"
                 }
+            
+            agent = args.get("agent", None)
+
+            tool_registry = agent.get_tool_registry() if agent else None
 
             with yaspin(text="搜索相关方法论...", color="cyan") as spinner:
                 with spinner.hidden():
-                    methodology_prompt = load_methodology(args["query"])
+                    methodology_prompt = load_methodology(args["query"], tool_registry)
 
                 if methodology_prompt:
                     spinner.text = "找到相关方法论"