jarvis-ai-assistant 0.1.162__tar.gz → 0.1.164__tar.gz

{jarvis_ai_assistant-0.1.162/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.164}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.162
+Version: 0.1.164
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -77,6 +77,7 @@ Dynamic: requires-python
 
 *您的智能开发和系统交互助手*
 
+[核心特色](#core-features) •
 [视频介绍](#video-introduction) •
 [快速开始](#quick-start) •
 [配置说明](#configuration) •
@@ -89,6 +90,15 @@ Dynamic: requires-python
 
 ---
 
+## 🌟 核心特色 <a id="core-features"></a>
+
+- 🆓 零成本接入：无缝集成元宝、Kimi等优质模型，无需支付API费用，同时保留强大的文件处理、搜索和推理能力
+- 🛠️ 工具驱动：内置丰富工具集，涵盖脚本执行、代码开发、网页搜索、终端操作等核心功能
+- 👥 人机协作：支持实时交互，用户可随时介入指导，确保AI行为符合预期
+- 🔌 高度可扩展：支持自定义工具和平台，轻松集成MCP协议
+- 📈 智能进化：内置方法论系统，持续学习优化，越用越智能
+
+
 ## 📺 视频介绍<a id="video-introduction"></a>
 
 [![视频介绍](docs/images/intro.png)](https://player.bilibili.com/player.html?isOutside=true&aid=114306578382907&bvid=BV1x2dAYeEpM&cid=29314583629&p=1)
@@ -205,7 +215,6 @@ jarvis-methodology --help
 | `JARVIS_EXECUTE_TOOL_CONFIRM` | false | 执行工具前是否需要确认 |
 | `JARVIS_CONFIRM_BEFORE_APPLY_PATCH` | true | 应用补丁前是否需要确认 |
 | `JARVIS_MAX_TOOL_CALL_COUNT` | 20 | 最大连续工具调用次数 |
-| `JARVIS_CODE_EMBEDDING_MODEL_SIZE` | small | 代码嵌入模型大小，可选值：small、large |
 
 
 ---
@@ -216,6 +225,8 @@ jarvis-methodology --help
 | ask_codebase | 智能代码库查询和分析，用于定位功能所在文件和理解单点实现，适合查询特定功能位置和实现原理 |
 | ask_user | 交互式用户输入收集 |
 | chdir | 更改当前工作目录 |
+| rewrite_file | 文件重写工具，用于完全重写或创建文件，提供完整的文件内容替换 |
+| edit_file | 代码编辑工具，用于精确修改文件内容，支持搜索替换方式编辑 |
 | code_plan | 理解需求并制定详细的代码修改计划，在修改前获取用户确认 |
 | create_code_agent | 代码开发工具，当需要修改代码时使用 |
 | create_sub_agent | 创建子代理以处理特定任务，子代理将生成任务总结报告 |

{jarvis_ai_assistant-0.1.162 → jarvis_ai_assistant-0.1.164}/README.md

@@ -8,6 +8,7 @@
 
 *您的智能开发和系统交互助手*
 
+[核心特色](#core-features) •
 [视频介绍](#video-introduction) •
 [快速开始](#quick-start) •
 [配置说明](#configuration) •
@@ -20,6 +21,15 @@
 
 ---
 
+## 🌟 核心特色 <a id="core-features"></a>
+
+- 🆓 零成本接入：无缝集成元宝、Kimi等优质模型，无需支付API费用，同时保留强大的文件处理、搜索和推理能力
+- 🛠️ 工具驱动：内置丰富工具集，涵盖脚本执行、代码开发、网页搜索、终端操作等核心功能
+- 👥 人机协作：支持实时交互，用户可随时介入指导，确保AI行为符合预期
+- 🔌 高度可扩展：支持自定义工具和平台，轻松集成MCP协议
+- 📈 智能进化：内置方法论系统，持续学习优化，越用越智能
+
+
 ## 📺 视频介绍<a id="video-introduction"></a>
 
 [![视频介绍](docs/images/intro.png)](https://player.bilibili.com/player.html?isOutside=true&aid=114306578382907&bvid=BV1x2dAYeEpM&cid=29314583629&p=1)
@@ -136,7 +146,6 @@ jarvis-methodology --help
 | `JARVIS_EXECUTE_TOOL_CONFIRM` | false | 执行工具前是否需要确认 |
 | `JARVIS_CONFIRM_BEFORE_APPLY_PATCH` | true | 应用补丁前是否需要确认 |
 | `JARVIS_MAX_TOOL_CALL_COUNT` | 20 | 最大连续工具调用次数 |
-| `JARVIS_CODE_EMBEDDING_MODEL_SIZE` | small | 代码嵌入模型大小，可选值：small、large |
 
 
 ---
@@ -147,6 +156,8 @@ jarvis-methodology --help
 | ask_codebase | 智能代码库查询和分析，用于定位功能所在文件和理解单点实现，适合查询特定功能位置和实现原理 |
 | ask_user | 交互式用户输入收集 |
 | chdir | 更改当前工作目录 |
+| rewrite_file | 文件重写工具，用于完全重写或创建文件，提供完整的文件内容替换 |
+| edit_file | 代码编辑工具，用于精确修改文件内容，支持搜索替换方式编辑 |
 | code_plan | 理解需求并制定详细的代码修改计划，在修改前获取用户确认 |
 | create_code_agent | 代码开发工具，当需要修改代码时使用 |
 | create_sub_agent | 创建子代理以处理特定任务，子代理将生成任务总结报告 |

{jarvis_ai_assistant-0.1.162 → jarvis_ai_assistant-0.1.164}/pyproject.toml

@@ -8,7 +8,7 @@ default = true
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.162"
+version = "0.1.164"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]

{jarvis_ai_assistant-0.1.162 → jarvis_ai_assistant-0.1.164}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.162",
+    version="0.1.164",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",

{jarvis_ai_assistant-0.1.162 → jarvis_ai_assistant-0.1.164}/src/jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.162"
+__version__ = "0.1.164"

{jarvis_ai_assistant-0.1.162 → jarvis_ai_assistant-0.1.164}/src/jarvis/jarvis_agent/__init__.py

@@ -2,6 +2,7 @@ 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
@@ -138,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,6 +182,8 @@ class Agent:
 
         from jarvis.jarvis_tools.registry import ToolRegistry
         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 = ""
@@ -192,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()
@@ -280,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):
         """设置附加提示。
@@ -289,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:
         """生成附加提示。
 
@@ -357,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)
 
@@ -657,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_ai_assistant-0.1.164/src/jarvis/jarvis_agent/output_handler.py

@@ -0,0 +1,47 @@
+from abc import ABC, abstractmethod
+from typing import Any, Tuple
+
+
+class OutputHandler(ABC):
+    @abstractmethod
+    def handle(self, response: str, agent: Any) -> Tuple[bool, Any]:
+        """处理响应数据
+        
+        Args:
+            response: 需要处理的响应字符串
+            agent: 执行处理的agent实例
+        
+        Returns:
+            Tuple[bool, Any]: 返回处理结果元组，第一个元素表示是否处理成功，第二个元素为处理后的数据
+        """
+        pass
+
+    @abstractmethod
+    def can_handle(self, response: str) -> bool:
+        """判断是否能处理给定的响应
+        
+        Args:
+            response: 需要判断的响应字符串
+        
+        Returns:
+            bool: 返回是否能处理该响应
+        """
+        pass
+
+    @abstractmethod
+    def prompt(self) -> str:
+        """获取处理器的提示信息
+        
+        Returns:
+            str: 返回处理器的提示字符串
+        """
+        pass
+
+    @abstractmethod
+    def name(self) -> str:
+        """获取处理器的名称
+        
+        Returns:
+            str: 返回处理器的名称字符串
+        """
+        pass

{jarvis_ai_assistant-0.1.162 → jarvis_ai_assistant-0.1.164}/src/jarvis/jarvis_code_agent/code_agent.py

@@ -10,6 +10,7 @@ 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
@@ -22,7 +23,9 @@ 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
@@ -52,164 +55,63 @@ class CodeAgent:
             "chdir",
             "find_methodology",
             "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提供更多代码统计信息，应优先使用
-- 针对不同编程语言选择对应的代码质量检查工具
-- 不要留下未实现的代码
-
-### 代码编辑规范
-- 使用edit_file工具进行代码修改时，必须遵循最小补丁原则
-- 只提供需要修改的代码部分，不要提供完整文件内容
-- 保持原始代码的缩进、空行和格式风格
+1. **项目分析**：分析项目结构，确定需修改的文件
+2. **需求分析**：理解需求意图，选择影响最小的实现方案
+3. **代码分析**：详细分析目标文件，禁止虚构现有代码
+4. **方案设计**：确定最小变更方案，保持代码结构
+5. **实施修改**：遵循"先读后写"原则，保持代码风格一致性
+
+## 工具使用
+- 项目结构：优先使用fd命令查找文件
+- 代码搜索：优先使用rg进行内容搜索
+- 代码阅读：优先使用read_code工具
+- 仅在命令行工具不足时使用专用工具
+
+## 代码编辑规范
+### 代码修改工具选择
+- 对于所有代码修改任务，优先使用edit_file工具，对于特别简单的修改，可以通过execute_script工具使用shell命令修改
+- edit_file通过搜索和替换实现精确的代码编辑，是首选的代码修改方式
+- 每个搜索文本在文件中必须唯一匹配，确保修改的准确性
+- 为新文件创建时，使用空字符串作为搜索文本，替换文本作为完整内容
+- 保持与原代码完全一致的格式和缩进风格
+
+### 文件重写工具使用
+- rewrite_file工具用于完全重写文件内容或创建新文件
+- 当需要替换整个文件内容时，优先使用rewrite_file工具
+- 创建新文件时，使用rewrite_file工具提供完整的文件内容
+- 保持与原代码相同的格式、缩进和编码风格
+- 确保提供格式良好的完整文件内容
+
+### 最小补丁原则
+- 对于任何代码修改，只提供需要修改的代码部分，不提供完整文件内容
+- 严格保持原始代码的缩进、空行和格式风格
+- 提供最小必要的上下文，确保修改位置精确
 - 每个修改必须包含清晰的修改理由
-- 新建文件时可以提供完整内容，修改现有文件时只提供差异部分
-</notes>
+
+### 格式兼容要求
+- 保持与原代码相同的缩进方式（空格或制表符）
+- 保持原代码的空行数量和位置
+- 不改变原代码的换行风格
+- 对新文件可提供完整内容，对现有文件只提供差异部分
 </code_engineer_guide>
 """
         # Dynamically add ask_codebase based on task complexity if really needed
@@ -236,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:
         """获取项目根目录
 
@@ -275,7 +182,7 @@ class CodeAgent:
                         'message': 提交信息,
                         'author': 作者,
                         'date': 提交日期,
-                        'files': [修改的文件列表] (最多50个文件)
+                        'files': [修改的文件列表] (最多20个文件)
                     },
                     ...
                 ]
@@ -317,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
 
@@ -325,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)
@@ -338,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):
@@ -371,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)
@@ -458,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_ai_assistant-0.1.162 → jarvis_ai_assistant-0.1.164}/src/jarvis/jarvis_dev/main.py

@@ -997,6 +997,7 @@ def create_dev_team() -> MultiAgent:
         "methodology",
         "ask_codebase",
         "edit_file",
+        "rewrite_file",
     ])
 
     BA_output_handler = ToolRegistry()
@@ -1009,6 +1010,7 @@ def create_dev_team() -> MultiAgent:
         "methodology",
         "ask_codebase",
         "edit_file",
+        "rewrite_file",
     ])
 
     SA_output_handler = ToolRegistry()
@@ -1020,6 +1022,7 @@ def create_dev_team() -> MultiAgent:
         "read_code",
         "methodology",
         "edit_file",
+        "rewrite_file",
     ])
 
     TL_output_handler = ToolRegistry()
@@ -1030,6 +1033,7 @@ def create_dev_team() -> MultiAgent:
         "execute_script",
         "methodology",
         "edit_file",
+        "rewrite_file",
     ])
 
     DEV_output_handler = ToolRegistry()
@@ -1042,6 +1046,7 @@ def create_dev_team() -> MultiAgent:
         "create_sub_agent",
         "methodology",
         "edit_file",
+        "rewrite_file",
     ])
 
     QA_output_handler = ToolRegistry()
@@ -1055,6 +1060,7 @@ def create_dev_team() -> MultiAgent:
         "read_code",
         "methodology",
         "edit_file",
+        "rewrite_file",
     ])
 
     # Update PM prompt with tool usage guidance

{jarvis_ai_assistant-0.1.162 → jarvis_ai_assistant-0.1.164}/src/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: