jarvis-ai-assistant 0.1.161__py3-none-any.whl → 0.1.163__py3-none-any.whl

jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.161"
+__version__ = "0.1.163"

jarvis/jarvis_agent/__init__.py

@@ -2,10 +2,10 @@ import datetime
 import platform
 from typing import Any, Callable, List, Optional, Tuple, Union
 
+from jarvis.jarvis_tools.registry import ToolRegistry
 from yaspin import yaspin # type: ignore
 
 from jarvis.jarvis_agent.output_handler import OutputHandler
-from jarvis.jarvis_agent.patch import PatchOutputHandler
 from jarvis.jarvis_platform.base import BasePlatform
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_utils.output import PrettyOutput, OutputType
@@ -139,6 +139,7 @@ class Agent:
                  summary_prompt: Optional[str] = None,
                  auto_complete: Optional[bool] = None,
                  output_handler: List[OutputHandler] = [],
+                 use_tools: List[str] = [],
                  input_handler: Optional[List[Callable[[str, Any], Tuple[str, bool]]]] = None,
                  execute_tool_confirm: Optional[bool] = None,
                  need_summary: bool = True,
@@ -180,7 +181,9 @@ class Agent:
         self.model.set_suppress_output(False)
 
         from jarvis.jarvis_tools.registry import ToolRegistry
-        self.output_handler = output_handler if output_handler else [ToolRegistry(), PatchOutputHandler()]
+        self.output_handler = output_handler if output_handler else [ToolRegistry()]
+        self.set_use_tools(use_tools)
+
         self.multiline_inputer = multiline_inputer if multiline_inputer else get_multiline_input
 
         self.prompt = ""
@@ -193,6 +196,7 @@ class Agent:
 
         self.tool_call_count = 0
         self.max_tool_call_count = get_max_tool_call_count()
+        self.after_tool_call_cb: Optional[Callable[[Agent], None]] = None
 
 
         self.execute_tool_confirm = execute_tool_confirm if execute_tool_confirm is not None else is_execute_tool_confirm()
@@ -281,6 +285,13 @@ class Agent:
 """)
         self.first = True
 
+    def set_use_tools(self, use_tools):
+        for handler in self.output_handler:
+            if isinstance(handler, ToolRegistry):
+                if use_tools:
+                    handler.use_tools(use_tools)
+                break
+
 
     def set_addon_prompt(self, addon_prompt: str):
         """设置附加提示。
@@ -290,6 +301,14 @@ class Agent:
         """
         self.addon_prompt = addon_prompt
 
+    def set_after_tool_call_cb(self, cb: Callable[[Any], None]): # type: ignore
+        """设置工具调用后回调函数。
+
+        参数:
+            cb: 回调函数
+        """
+        self.after_tool_call_cb = cb
+
     def make_default_addon_prompt(self, need_complete: bool) -> str:
         """生成附加提示。
 
@@ -358,10 +377,6 @@ class Agent:
         else:
             message += f"\n\n{self.make_default_addon_prompt(need_complete)}"
 
-        # 在真正调用模型对话前减少文件读取计数
-        from jarvis.jarvis_utils.globals import decrease_read_file_counts
-        decrease_read_file_counts()
-
         # 累加对话长度
         self.conversation_length += get_context_token_count(message)
 
@@ -658,6 +673,9 @@ arguments:
                     if need_return:
                         return self.prompt
 
+                    if self.after_tool_call_cb:
+                        self.after_tool_call_cb(self)
+
                     if self.prompt:
                         continue
 

jarvis/jarvis_agent/jarvis.py

@@ -15,7 +15,6 @@ from jarvis.jarvis_agent import (
      Agent,
      origin_agent_system_prompt
 )
-from jarvis.jarvis_agent.patch import PatchOutputHandler
 from jarvis.jarvis_tools.registry import ToolRegistry
 from jarvis.jarvis_utils.utils import init_env
 from jarvis.jarvis_agent.file_input_handler import file_input_handler
@@ -123,7 +122,7 @@ def main() -> None:
             platform=args.platform,
             model_name=args.model,
             input_handler=[file_input_handler, shell_input_handler, builtin_input_handler],
-            output_handler=[ToolRegistry(), PatchOutputHandler()],
+            output_handler=[ToolRegistry()],
             need_summary=False
         )
 

jarvis/jarvis_code_agent/code_agent.py

@@ -10,20 +10,22 @@ import argparse
 from typing import Any, Dict, Optional, List, Tuple
 
 # 忽略yaspin的类型检查
+from jarvis.jarvis_utils.config import is_confirm_before_apply_patch
 from yaspin import yaspin  # type: ignore
 
 from jarvis.jarvis_agent import Agent
 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_git_utils.git_commiter import GitCommitTool
 from jarvis.jarvis_tools.registry import ToolRegistry
 from jarvis.jarvis_utils.git_utils import (
     find_git_root,
     get_commits_between,
+    get_diff,
     get_latest_commit_hash,
+    handle_commit_workflow,
     has_uncommitted_changes
 )
 from jarvis.jarvis_utils.input import get_multiline_input
@@ -51,161 +53,65 @@ class CodeAgent:
             "read_code",
             "methodology",
             "chdir",
-            "create_code_agent",
             "find_methodology",
-            "virtual_tty",
+            "edit_file",
+            "rewrite_file"
         ])
         code_system_prompt = """
 <code_engineer_guide>
-<principles>
+## 角色定位
+你是Jarvis系统的代码工程师，一个专业的代码分析和修改助手。你的职责是：
+- 理解用户的代码需求，并提供高质量的实现方案
+- 精确分析项目结构和代码，准确定位需要修改的位置
+- 编写符合项目风格和标准的代码
+- 在修改代码时保持谨慎，确保不破坏现有功能
+- 做出专业的技术决策，减少用户决策负担
+
 ## 核心原则
 - 自主决策：基于专业判断做出决策，减少用户询问
-- 高效精准：一次性提供完整解决方案，避免反复修改
-- 修改审慎：修改代码前要三思而后行，充分分析影响范围，尽量做到一次把事情做好
+- 高效精准：提供完整解决方案，避免反复修改
+- 修改审慎：修改前充分分析影响范围，做到一次把事情做好
 - 工具精通：选择最高效工具路径解决问题
-- 严格确认：必须先分析项目结构，确定要修改的文件，禁止虚构已存在的代码
-</principles>
 
-<workflow>
 ## 工作流程
-
-<step>
-### 1. 项目结构分析
-- 第一步必须分析项目结构，识别关键模块和文件
-- 结合用户需求，确定需要修改的文件列表
-- 优先使用fd命令查找文件，使用execute_script执行
-- 明确说明将要修改的文件及其范围
-</step>
-
-<step>
-### 2. 需求分析
-- 基于项目结构理解，分析需求意图和实现方案
-- 当需求有多种实现方式时，选择影响最小的方案
-- 仅当需求显著模糊时才询问用户
-</step>
-
-<step>
-### 3. 代码分析与确认
-- 详细分析确定要修改的文件内容
-- 明确区分现有代码和需要新建的内容
-- 绝对禁止虚构或假设现有代码的实现细节
-- 分析顺序：项目结构 → 目标文件 → 相关文件
-- 只在必要时扩大分析范围，避免过度分析
-- 工具选择：
-  | 分析需求 | 首选工具 | 备选工具 |
-  |---------|---------|----------|
-  | 项目结构 | fd (通过execute_script) | ask_codebase(仅在必要时) |
-  | 文件内容 | read_code | ask_codebase(仅在必要时) |
-  | 查找引用 | rg (通过execute_script) | ask_codebase(仅在必要时) |
-  | 查找定义 | rg (通过execute_script) | ask_codebase(仅在必要时) |
-  | 函数调用者 | rg (通过execute_script) | ask_codebase(仅在必要时) |
-  | 函数分析 | read_code + rg | ask_codebase(仅在必要时) |
-  | 整体分析 | execute_script | ask_codebase(仅在必要时) |
-  | 代码质量检查 | execute_script | ask_codebase(仅在必要时) |
-  | 统计代码行数 | loc (通过execute_script) | - |
-</step>
-
-<step>
-### 4. 方案设计
-- 确定最小变更方案，保持代码结构
-- 变更类型处理：
-  - 修改现有文件：必须先确认文件存在及其内容
-  - 创建新文件：可以根据需求创建，但要符合项目结构和风格
-- 变更规模处理：
-  - ≤50行：一次性完成所有修改
-  - 50-200行：按功能模块分组
-  - >200行：按功能拆分，但尽量减少提交次数
-</step>
-
-<step>
-### 5. 实施修改
-- 遵循"先读后写"原则，在修改已有代码前，必须已经读取了对应文件，如果已经读取过文件，不需要重新读取
-- 保持代码风格一致性
-- 自动匹配项目现有命名风格
-- 允许创建新文件和结构，但不得假设或虚构现有代码
-</step>
-</workflow>
-
-<tools>
-## 专用工具简介
-仅在必要时使用以下专用工具：
-
-- **ask_codebase**: 代码库整体查询，应优先使用fd、rg和read_code组合替代
-</tools>
-
-<shell_commands>
-## Shell命令优先策略
-
-<category>
-### 优先使用的Shell命令
-- **项目结构分析**：
-  - `fd -t f -e py` 查找所有Python文件
-  - `fd -t f -e js -e ts` 查找所有JavaScript/TypeScript文件
-  - `fd -t d` 列出所有目录
-  - `fd -t f -e java -e kt` 查找所有Java/Kotlin文件
-  - `fd -t f -e go` 查找所有Go文件
-  - `fd -t f -e rs` 查找所有Rust文件
-  - `fd -t f -e c -e cpp -e h -e hpp` 查找所有C/C++文件
-</category>
-
-<category>
-- **代码内容搜索**：
-  - `rg "pattern" --type py` 在Python文件中搜索
-  - `rg "pattern" --type js` 在JavaScript文件中搜索
-  - `rg "pattern" --type java` 在Java文件中搜索
-  - `rg "pattern" --type c` 在C文件中搜索
-  - `rg "class ClassName"` 查找类定义
-  - `rg "func|function|def" -g "*.py" -g "*.js" -g "*.go" -g "*.rs"` 查找函数定义
-  - `rg -w "word"` 精确匹配单词
-</category>
-
-<category>
-- **代码统计分析**：
-  - `loc` 统计当前目录代码行数
-</category>
-
-<category>
-- **代码质量检查**：
-  - Python: `pylint <file_path>`, `flake8 <file_path>`
-  - JavaScript: `eslint <file_path>`
-  - TypeScript: `tsc --noEmit <file_path>`
-  - Java: `checkstyle <file_path>`
-  - Go: `go vet <file_path>`
-  - Rust: `cargo clippy`
-  - C/C++: `cppcheck <file_path>`
-</category>
-
-<category>
-- **整体代码分析**：
-  - 使用execute_script编写和执行脚本，批量分析多个文件
-  - 简单脚本示例：`find . -name "*.py" | xargs pylint`
-  - 使用多工具组合：`fd -e py | xargs pylint`
-</category>
-</shell_commands>
-
-<read_code_usage>
-### read_code工具使用
-读取文件应优先使用read_code工具，而非shell命令：
-- 完整读取：使用read_code读取整个文件内容
-- 部分读取：使用read_code指定行范围
-- 大文件处理：对大型文件使用read_code指定行范围，避免全部加载
-</read_code_usage>
-
-<tool_usage>
-### 仅在命令行工具不足时使用专用工具
-只有当fd、rg、loc和read_code工具无法获取足够信息时，才考虑使用专用工具（ask_codebase等）。在每次使用专用工具前，应先尝试使用上述工具获取所需信息。
-</tool_usage>
-
-<notes>
-### 注意事项
-- read_code比cat或grep更适合阅读代码
-- rg比grep更快更强大，应优先使用
-- fd比find更快更易用，应优先使用
-- loc比wc -l提供更多代码统计信息，应优先使用
-- 针对不同编程语言选择对应的代码质量检查工具
-- 不要留下未实现的代码
-- 对于非常复杂的需求，可以使用create_code_agent工具，但是要提供完整的上下文信息
-</notes>
+1. **项目分析**：分析项目结构，确定需修改的文件
+2. **需求分析**：理解需求意图，选择影响最小的实现方案
+3. **代码分析**：详细分析目标文件，禁止虚构现有代码
+4. **方案设计**：确定最小变更方案，保持代码结构
+5. **实施修改**：遵循"先读后写"原则，保持代码风格一致性
+
+## 工具使用
+- 项目结构：优先使用fd命令查找文件
+- 代码搜索：优先使用rg进行内容搜索
+- 代码阅读：优先使用read_code工具
+- 仅在命令行工具不足时使用专用工具
+
+## 代码编辑规范
+### 代码修改工具选择
+- 对于所有代码修改任务，优先使用edit_file工具，对于特别简单的修改，可以通过execute_script工具使用shell命令修改
+- edit_file通过搜索和替换实现精确的代码编辑，是首选的代码修改方式
+- 每个搜索文本在文件中必须唯一匹配，确保修改的准确性
+- 为新文件创建时，使用空字符串作为搜索文本，替换文本作为完整内容
+- 保持与原代码完全一致的格式和缩进风格
+
+### 文件重写工具使用
+- rewrite_file工具用于完全重写文件内容或创建新文件
+- 当需要替换整个文件内容时，优先使用rewrite_file工具
+- 创建新文件时，使用rewrite_file工具提供完整的文件内容
+- 保持与原代码相同的格式、缩进和编码风格
+- 确保提供格式良好的完整文件内容
+
+### 最小补丁原则
+- 对于任何代码修改，只提供需要修改的代码部分，不提供完整文件内容
+- 严格保持原始代码的缩进、空行和格式风格
+- 提供最小必要的上下文，确保修改位置精确
+- 每个修改必须包含清晰的修改理由
+
+### 格式兼容要求
+- 保持与原代码相同的缩进方式（空格或制表符）
+- 保持原代码的空行数量和位置
+- 不改变原代码的换行风格
+- 对新文件可提供完整内容，对现有文件只提供差异部分
 </code_engineer_guide>
 """
         # Dynamically add ask_codebase based on task complexity if really needed
@@ -220,7 +126,7 @@ class CodeAgent:
             system_prompt=code_system_prompt,
             name="CodeAgent",
             auto_complete=False,
-            output_handler=[tool_registry, PatchOutputHandler()],
+            output_handler=[tool_registry],
             platform=platform_instance,
             input_handler=[
                 shell_input_handler,
@@ -232,9 +138,14 @@ class CodeAgent:
         self.agent.set_addon_prompt(
             "请使用工具充分理解用户需求，然后根据需求一步步执行代码修改/开发，"
             "如果不清楚要修改那些文件，可以使用ask_codebase工具，"
-            "以：xxxx功能在哪个文件中实现？类似句式提问"
+            "以：xxxx功能在哪个文件中实现？类似句式提问。"
+            "所有代码修改任务都应优先使用edit_file工具，而非edit_file工具。"
+            "edit_file工具通过精确的搜索和替换实现代码编辑，"
+            "搜索文本需在目标文件中有且仅有一次精确匹配，确保修改的准确性。"
         )
 
+        self.agent.set_after_tool_call_cb(self.after_tool_call_cb)
+
     def get_root_dir(self) -> str:
         """获取项目根目录
 
@@ -271,7 +182,7 @@ class CodeAgent:
                         'message': 提交信息,
                         'author': 作者,
                         'date': 提交日期,
-                        'files': [修改的文件列表] (最多50个文件)
+                        'files': [修改的文件列表] (最多20个文件)
                     },
                     ...
                 ]
@@ -313,7 +224,7 @@ class CodeAgent:
                 )
                 if files_result.returncode == 0:
                     files = list(set(filter(None, files_result.stdout.splitlines())))
-                    commit['files'] = files[:50]  # 限制最多50个文件
+                    commit['files'] = files[:20]  # 限制最多20个文件
 
             return commits
 
@@ -321,7 +232,10 @@ class CodeAgent:
             return []
 
     def _init_env(self) -> None:
-        """初始化环境"""
+        """初始化环境，包括：
+        1. 查找git根目录
+        2. 检查并处理未提交的修改
+        """
         with yaspin(text="正在初始化环境...", color="cyan") as spinner:
             curr_dir = os.getcwd()
             git_dir = find_git_root(curr_dir)
@@ -334,7 +248,10 @@ class CodeAgent:
             spinner.ok("✅")
 
     def _handle_uncommitted_changes(self) -> None:
-        """处理未提交的修改"""
+        """处理未提交的修改，包括：
+        1. 提示用户确认是否提交
+        2. 如果确认，则暂存并提交所有修改
+        """
         if has_uncommitted_changes():
             PrettyOutput.print("检测到未提交的修改，是否要提交？", OutputType.WARNING)
             if user_confirm("是否要提交？", True):
@@ -367,14 +284,14 @@ class CodeAgent:
         start_commit: Optional[str],
         end_commit: Optional[str]
     ) -> List[Tuple[str, str]]:
-        """Show commit history between two commits.
+        """显示两个提交之间的提交历史
 
-        Args:
-            start_commit: The starting commit hash
-            end_commit: The ending commit hash
+        参数:
+            start_commit: 起始提交hash
+            end_commit: 结束提交hash
 
-        Returns:
-            List of tuples containing (commit_hash, commit_message)
+        返回:
+            包含(commit_hash, commit_message)的元组列表
         """
         if start_commit and end_commit:
             commits = get_commits_between(start_commit, end_commit)
@@ -454,6 +371,56 @@ class CodeAgent:
 
         except RuntimeError as e:
             return f"Error during execution: {str(e)}"
+        
+    def after_tool_call_cb(self, agent: Agent) -> None:
+        """工具调用后回调函数。"""
+        final_ret = ""
+        diff = get_diff()
+        if diff:
+            start_hash = get_latest_commit_hash()
+            PrettyOutput.print(diff, OutputType.CODE, lang="diff")
+            commited = handle_commit_workflow()
+            if commited:
+                # 获取提交信息
+                end_hash = get_latest_commit_hash()
+                commits = get_commits_between(start_hash, end_hash)
+
+                # 添加提交信息到final_ret
+                if commits:
+                    final_ret += "✅ 补丁已应用\n"
+                    final_ret += "# 提交信息:\n"
+                    for commit_hash, commit_message in commits:
+                        final_ret += f"- {commit_hash[:7]}: {commit_message}\n"
+
+                    final_ret += f"# 应用补丁:\n```diff\n{diff}\n```"
+
+                    # 修改后的提示逻辑
+                    addon_prompt = f"如果用户的需求未完成，请继续生成补丁，如果已经完成，请终止，不要输出新的PATCH，不要实现任何超出用户需求外的内容\n"
+                    addon_prompt += "如果有任何信息不明确，调用工具获取信息\n"
+                    addon_prompt += "每次响应必须且只能包含一个操作\n"
+
+                    agent.set_addon_prompt(addon_prompt)
+
+                else:
+                    final_ret += "✅ 补丁已应用（没有新的提交）"
+            else:
+                final_ret += "❌ 补丁应用被拒绝\n"
+                final_ret += f"# 补丁预览:\n```diff\n{diff}\n```"
+        else:
+            return
+        # 用户确认最终结果
+        if commited:
+            agent.prompt += final_ret
+            return
+        PrettyOutput.print(final_ret, OutputType.USER, lang="markdown")
+        if not is_confirm_before_apply_patch() or user_confirm("是否使用此回复？", default=True):
+            agent.prompt += final_ret
+            return
+        custom_reply = get_multiline_input("请输入自定义回复")
+        if not custom_reply.strip():  # 如果自定义回复为空，返回空字符串
+            agent.prompt += final_ret
+        agent.set_addon_prompt(custom_reply)
+        agent.prompt += final_ret
 
 
 def main() -> None:

jarvis/jarvis_dev/main.py

@@ -3,7 +3,8 @@ from jarvis.jarvis_multi_agent import MultiAgent
 from jarvis.jarvis_tools.registry import ToolRegistry
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-from jarvis.jarvis_utils.utils import ct, ot, init_env
+from jarvis.jarvis_utils.tag import ct, ot
+from jarvis.jarvis_utils.utils import init_env
 
 # 定义每个角色的系统提示
 PM_PROMPT = f"""
@@ -994,7 +995,9 @@ def create_dev_team() -> MultiAgent:
         "search_web",
         "execute_script",
         "methodology",
-        "ask_codebase"
+        "ask_codebase",
+        "edit_file",
+        "rewrite_file",
     ])
 
     BA_output_handler = ToolRegistry()
@@ -1005,7 +1008,9 @@ def create_dev_team() -> MultiAgent:
         "execute_script",
         "read_webpage",
         "methodology",
-        "ask_codebase"
+        "ask_codebase",
+        "edit_file",
+        "rewrite_file",
     ])
 
     SA_output_handler = ToolRegistry()
@@ -1015,7 +1020,9 @@ def create_dev_team() -> MultiAgent:
         "ask_codebase",
         "execute_script",
         "read_code",
-        "methodology"
+        "methodology",
+        "edit_file",
+        "rewrite_file",
     ])
 
     TL_output_handler = ToolRegistry()
@@ -1025,6 +1032,8 @@ def create_dev_team() -> MultiAgent:
         "lsp_get_diagnostics",
         "execute_script",
         "methodology",
+        "edit_file",
+        "rewrite_file",
     ])
 
     DEV_output_handler = ToolRegistry()
@@ -1036,6 +1045,8 @@ def create_dev_team() -> MultiAgent:
         "read_code",
         "create_sub_agent",
         "methodology",
+        "edit_file",
+        "rewrite_file",
     ])
 
     QA_output_handler = ToolRegistry()
@@ -1048,6 +1059,8 @@ def create_dev_team() -> MultiAgent:
         "execute_script",
         "read_code",
         "methodology",
+        "edit_file",
+        "rewrite_file",
     ])
 
     # Update PM prompt with tool usage guidance

jarvis/jarvis_git_details/main.py

@@ -13,7 +13,8 @@ from jarvis.jarvis_agent import Agent
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_tools.registry import ToolRegistry
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-from jarvis.jarvis_utils.utils import ct, ot, init_env
+from jarvis.jarvis_utils.tag import ct, ot
+from jarvis.jarvis_utils.utils import init_env
 
 
 class GitCommitAnalyzer:

jarvis/jarvis_git_utils/git_commiter.py

@@ -153,9 +153,10 @@ class GitCommitTool:
                             except Exception as e:
                                 spinner.write(f"⚠️ 上传文件时出错: {str(e)}")
                                 upload_success = False
-                        
                         # 根据上传状态准备完整的提示
                         if upload_success:
+                            # 尝试生成提交信息
+                            spinner.text = "正在生成提交消息..."
                             # 使用上传的文件
                             prompt = base_prompt + f'''
         # 变更概述
@@ -178,8 +179,6 @@ class GitCommitTool:
         '''
                                 commit_message = platform.chat_until_success(prompt)
                         
-                        # 尝试生成提交信息
-                        spinner.text = "正在生成提交消息..."
                         while True:
                             # 只在特定情况下重新获取commit_message
                             if not upload_success and not is_large_content and not commit_message:

jarvis/jarvis_multi_agent/__init__.py

@@ -1,14 +1,13 @@
 import re
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, Dict, List, Tuple
 
 import yaml
 
 from jarvis.jarvis_agent import Agent
 from jarvis.jarvis_agent.output_handler import OutputHandler
 from jarvis.jarvis_tools.registry import ToolRegistry
-from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-from jarvis.jarvis_utils.utils import ct, ot, init_env
+from jarvis.jarvis_utils.tag import ct, ot
 
 
 class MultiAgent(OutputHandler):

jarvis/jarvis_platform/base.py

@@ -3,7 +3,6 @@ import re
 from typing import List, Tuple
 from jarvis.jarvis_utils.config import get_max_input_token_count
 from jarvis.jarvis_utils.embedding import split_text_into_chunks
-from jarvis.jarvis_utils.globals import clear_read_file_record
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import get_context_token_count, while_success, while_true
 from jarvis.jarvis_utils.tag import ot, ct
@@ -28,7 +27,6 @@ class BasePlatform(ABC):
 
     def reset(self):
         """Reset model"""
-        clear_read_file_record()
         self.delete_chat()
 
     @abstractmethod

jarvis/jarvis_platform/yuanbao.py

@@ -220,6 +220,7 @@ class YuanbaoPlatform(BasePlatform):
         
         self.multimedia = uploaded_files
         return True
+
         
     def _generate_upload_info(self, file_name: str) -> Dict:
         """从元宝API生成上传信息