vault backup: 2026-06-09 23:15:17
This commit is contained in:
@@ -1,37 +1,42 @@
|
||||
---
|
||||
title: "计划模式 - Plan Mode 先看后做的安全机制"
|
||||
description: "基于源码解析 Claude Code Plan Mode 的完整实现:EnterPlanModeTool/ExitPlanModeV2Tool 的工具设计、权限上下文切换机制、Prompt-based 权限请求、计划文件持久化、Teammate 审批流程。"
|
||||
keywords: ["Plan Mode", "计划模式", "EnterPlanMode", "ExitPlanMode", "prepareContextForPlanMode", "allowedPrompts"]
|
||||
tags: [Plan-Mode, 计划模式, 权限模式, Claude-Code, 安全]
|
||||
create time: 2026-06-09 22:30
|
||||
---
|
||||
|
||||
{/* 本章目标:基于源码揭示 Plan Mode 的完整实现 */}
|
||||
# 计划模式 - Plan Mode 先看后做的安全机制
|
||||
|
||||
## 问题场景
|
||||
## 概述
|
||||
|
||||
Plan Mode 为复杂任务提供了一个"只读探索"阶段,通过 EnterPlanModeTool 和 ExitPlanModeV2Tool 两个工具实现闭环。AI 先在只读模式下充分理解代码库,形成计划方案后提交用户审阅,批准后才恢复全部权限执行。这解决了"AI 匆忙行动"的问题。
|
||||
|
||||
## 正文
|
||||
|
||||
### 问题场景
|
||||
|
||||
你说"重构这个模块",AI 立刻开始改代码——但你还没搞清楚它打算怎么改。等改了一半发现方向不对,已经来不及了。
|
||||
|
||||
## Plan Mode 的解决方案
|
||||
### Plan Mode 的解决方案
|
||||
|
||||
计划模式给对话加了一个"只读阶段",通过两个工具实现闭环:
|
||||
|
||||
<Steps>
|
||||
<Step title="EnterPlanMode — 进入计划模式">
|
||||
AI 自主判断(或用户触发)任务需要规划,调用 `EnterPlanModeTool`(`packages/builtin-tools/src/tools/EnterPlanModeTool/EnterPlanModeTool.ts:36`)。该工具需要**用户审批**(`checkPermissions` 返回 `ask`)。
|
||||
</Step>
|
||||
<Step title="探索阶段 — 只读工具集">
|
||||
权限模式切换为 `'plan'`,AI 只能使用 `isReadOnly()` 为 true 的工具(Read、Grep、Glob、Agent 等)。写操作被自动拒绝。
|
||||
</Step>
|
||||
<Step title="ExitPlanMode — 提交方案审批">
|
||||
AI 完成探索后,调用 `ExitPlanModeV2Tool`(`packages/builtin-tools/src/tools/ExitPlanModeTool/ExitPlanModeV2Tool.ts:147`),将计划文件提交给用户审阅。这是第二个**需要用户审批**的节点。
|
||||
</Step>
|
||||
<Step title="恢复执行 — 全部工具权限">
|
||||
用户批准后,权限模式恢复为进入前的状态,AI 按计划执行。
|
||||
</Step>
|
||||
</Steps>
|
||||
```mermaid
|
||||
flowchart TD
|
||||
A["EnterPlanMode\n用户审批进入"] --> B["探索阶段\n只读工具集 Read/Grep/Glob"]
|
||||
B --> C["ExitPlanMode\n提交方案审批"]
|
||||
C --> D["恢复执行\n全部工具权限"]
|
||||
```
|
||||
|
||||
## 权限的自动收窄与恢复
|
||||
1. **EnterPlanMode — 进入计划模式**:AI 自主判断(或用户触发)任务需要规划,调用 `EnterPlanModeTool`(`packages/builtin-tools/src/tools/EnterPlanModeTool/EnterPlanModeTool.ts:36`)。该工具需要**用户审批**(`checkPermissions` 返回 `ask`)。
|
||||
|
||||
### 进入:`prepareContextForPlanMode()`
|
||||
2. **探索阶段 — 只读工具集**:权限模式切换为 `'plan'`,AI 只能使用 `isReadOnly()` 为 true 的工具(Read、Grep、Glob、Agent 等)。写操作被自动拒绝。
|
||||
|
||||
3. **ExitPlanMode — 提交方案审批**:AI 完成探索后,调用 `ExitPlanModeV2Tool`(`packages/builtin-tools/src/tools/ExitPlanModeTool/ExitPlanModeV2Tool.ts:147`),将计划文件提交给用户审阅。这是第二个**需要用户审批**的节点。
|
||||
|
||||
4. **恢复执行 — 全部工具权限**:用户批准后,权限模式恢复为进入前的状态,AI 按计划执行。
|
||||
|
||||
### 权限的自动收窄与恢复
|
||||
|
||||
#### 进入:prepareContextForPlanMode()
|
||||
|
||||
`EnterPlanModeTool.call()`(第 77 行)的核心逻辑:
|
||||
|
||||
@@ -54,7 +59,7 @@ context.setAppState(prev => ({
|
||||
- 在 plan 模式下,工具的 `isReadOnly()` 检查成为唯一准入条件
|
||||
- 如果用户的默认模式是 `'auto'`,还会激活 classifier 的副作用
|
||||
|
||||
### 退出:权限恢复 + Prompt-based 权限
|
||||
#### 退出:权限恢复 + Prompt-based 权限
|
||||
|
||||
`ExitPlanModeV2Tool` 的退出逻辑做了两件关键的事:
|
||||
|
||||
@@ -76,7 +81,10 @@ allowedPrompts: z.array(z.object({
|
||||
|
||||
当 AI 提交计划时,如果声明了 `allowedPrompts: [{ tool: 'Bash', prompt: 'run tests' }]`,用户批准后,"run tests" 这类 Bash 命令会被自动放行——不再需要逐个确认。
|
||||
|
||||
## 计划文件的持久化
|
||||
> [!tip] Prompt-based 权限的价值
|
||||
> 这个设计让 AI 可以"预告"它将要执行的操作类别,用户在审批计划时一并授权,避免了执行阶段的频繁打断。
|
||||
|
||||
### 计划文件的持久化
|
||||
|
||||
计划内容被写入磁盘文件(由 `getPlanFilePath()` 确定路径),这与简单的"AI 说一段话然后开始执行"有本质区别:
|
||||
|
||||
@@ -85,7 +93,7 @@ allowedPrompts: z.array(z.object({
|
||||
3. `planWasEdited` 字段标记用户是否修改了计划,影响后续的 tool_result 回显
|
||||
4. `persistFileSnapshotIfRemote()` 在远程场景下保存文件快照
|
||||
|
||||
## Teammate 场景下的计划审批
|
||||
### Teammate 场景下的计划审批
|
||||
|
||||
在 Agent Swarms(`isAgentSwarmsEnabled()`)模式下,计划审批有额外的协作流程:
|
||||
|
||||
@@ -105,7 +113,7 @@ if (isTeammate()) {
|
||||
|
||||
这意味着在蜂群模式下,计划可能不是由直接用户审批,而是由 Team Leader 审批。
|
||||
|
||||
## 什么时候该用计划模式
|
||||
### 什么时候该用计划模式
|
||||
|
||||
`EnterPlanModeTool` 的 Prompt(`packages/builtin-tools/src/tools/EnterPlanModeTool/prompt.ts`)定义了两套触发标准——外部版本更积极(鼓励规划),内部版本更克制(仅在真正模糊时使用):
|
||||
|
||||
@@ -117,7 +125,7 @@ if (isTeammate()) {
|
||||
| "开始做 X" | — | **跳过**(直接开始) |
|
||||
| 架构决策(Redis vs 内存缓存) | **进入** | **进入**(真正模糊) |
|
||||
|
||||
## 计划模式 + 任务系统
|
||||
### 计划模式 + 任务系统
|
||||
|
||||
计划模式通常与任务系统配合使用:
|
||||
|
||||
@@ -126,26 +134,21 @@ if (isTeammate()) {
|
||||
3. 退出计划模式后,AI 按任务列表逐项执行
|
||||
4. 用户可以通过任务列表追踪进度
|
||||
|
||||
## 完整生命周期
|
||||
### 完整生命周期
|
||||
|
||||
```mermaid
|
||||
flowchart TD
|
||||
U["用户: 重构这个模块"] --> AI1["AI 判断需要规划\n调用 EnterPlanModeTool"]
|
||||
AI1 -->|"用户审批 Ask 对话框"| TRANS["handlePlanModeTransition(default, 'plan')\nprepareContextForPlanMode()"]
|
||||
TRANS --> EXPLORE["AI 使用 Read/Grep/Glob/Agent 探索代码库\n可能 10+ 轮只读工具调用"]
|
||||
EXPLORE --> EXIT["AI 形成方案\n调用 ExitPlanModeV2Tool\nallowedPrompts: run tests, install deps"]
|
||||
EXIT -->|"用户审批计划\n可编辑计划文件"| RESTORE["恢复权限模式\n注入 prompt-based 权限"]
|
||||
RESTORE --> EXEC["AI 使用全部工具执行计划\nrun tests 等命令自动放行"]
|
||||
```
|
||||
用户: "重构这个模块"
|
||||
↓
|
||||
AI 判断需要规划 → 调用 EnterPlanModeTool
|
||||
↓ 用户审批(Ask 对话框)
|
||||
handlePlanModeTransition(default, 'plan') // 保存 default
|
||||
prepareContextForPlanMode() // 创建只读上下文
|
||||
↓
|
||||
AI 使用 Read/Grep/Glob/Agent 探索代码库
|
||||
↓ (可能 10+ 轮只读工具调用)
|
||||
AI 形成方案 → 调用 ExitPlanModeV2Tool({
|
||||
allowedPrompts: [
|
||||
{ tool: 'Bash', prompt: 'run tests' },
|
||||
{ tool: 'Bash', prompt: 'install dependencies' }
|
||||
]
|
||||
})
|
||||
↓ 用户审批计划(可编辑计划文件)
|
||||
恢复权限模式 → 注入 prompt-based 权限
|
||||
↓
|
||||
AI 使用全部工具执行计划,"run tests" 等命令自动放行
|
||||
```
|
||||
|
||||
## 关联笔记
|
||||
|
||||
- [[why-safety-matters|AI 安全至关重要]]
|
||||
- [[permission-model|权限模型]]
|
||||
- [[auto-mode|Auto Mode]]
|
||||
- [[sandbox|沙箱机制]]
|
||||
|
||||
Reference in New Issue
Block a user