196 lines
7.0 KiB
Markdown
196 lines
7.0 KiB
Markdown
# FORK_SUBAGENT — 上下文继承子 Agent
|
||
|
||
> Feature Flag: `FEATURE_FORK_SUBAGENT=1`
|
||
> 实现状态：完整可用
|
||
> 引用数：4
|
||
|
||
## 一、功能概述
|
||
|
||
FORK_SUBAGENT 让 AgentTool 生成"fork 子 agent"，继承父级完整对话上下文。子 agent 看到父级的所有历史消息、工具集和系统提示，并且与父级共享 API 请求前缀以最大化 prompt cache 命中率。
|
||
|
||
### 核心优势
|
||
|
||
- **Prompt Cache 最大化**：多个并行 fork 共享相同的 API 请求前缀，只有最后的 directive 文本块不同
|
||
- **上下文完整性**：子 agent 继承父级的完整对话历史（包括 thinking config）
|
||
- **权限冒泡**：子 agent 的权限提示上浮到父级终端显示
|
||
- **Worktree 隔离**：支持 git worktree 隔离，子 agent 在独立分支工作
|
||
|
||
## 二、用户交互
|
||
|
||
### 触发方式
|
||
|
||
当 `FORK_SUBAGENT` 启用时，AgentTool 调用不指定 `subagent_type` 时自动走 fork 路径：
|
||
|
||
```
|
||
// Fork 路径（继承上下文）
|
||
Agent({ prompt: "修复这个 bug" }) // 无 subagent_type
|
||
|
||
// 普通 agent 路径（全新上下文）
|
||
Agent({ subagent_type: "general-purpose", prompt: "..." })
|
||
```
|
||
|
||
### /fork 命令
|
||
|
||
注册了 `/fork` 斜杠命令（当前为 stub）。当 FORK_SUBAGENT 开启时，`/branch` 命令失去 `fork` 别名，避免冲突。
|
||
|
||
## 三、实现架构
|
||
|
||
### 3.1 门控与互斥
|
||
|
||
文件：`src/tools/AgentTool/forkSubagent.ts:32-39`
|
||
|
||
```ts
|
||
export function isForkSubagentEnabled(): boolean {
|
||
if (feature('FORK_SUBAGENT')) {
|
||
if (isCoordinatorMode()) return false // Coordinator 有自己的委派模型
|
||
if (getIsNonInteractiveSession()) return false // pipe/SDK 模式禁用
|
||
return true
|
||
}
|
||
return false
|
||
}
|
||
```
|
||
|
||
### 3.2 FORK_AGENT 定义
|
||
|
||
```ts
|
||
export const FORK_AGENT = {
|
||
agentType: 'fork',
|
||
tools: ['*'], // 通配符：使用父级完整工具集
|
||
maxTurns: 200,
|
||
model: 'inherit', // 继承父级模型
|
||
permissionMode: 'bubble', // 权限冒泡到父级终端
|
||
getSystemPrompt: () => '', // 不使用：直接传递父级已渲染 prompt
|
||
}
|
||
```
|
||
|
||
### 3.3 核心调用流程
|
||
|
||
```
|
||
AgentTool.call({ prompt, name })
|
||
│
|
||
▼
|
||
isForkSubagentEnabled() && !subagent_type?
|
||
│
|
||
├── No → 普通 agent 路径
|
||
│
|
||
└── Yes → Fork 路径
|
||
│
|
||
▼
|
||
递归防护检查
|
||
├── querySource === 'agent:builtin:fork' → 拒绝
|
||
└── isInForkChild(messages) → 拒绝
|
||
│
|
||
▼
|
||
获取父级 system prompt
|
||
├── toolUseContext.renderedSystemPrompt（首选）
|
||
└── buildEffectiveSystemPrompt（回退）
|
||
│
|
||
▼
|
||
buildForkedMessages(prompt, assistantMessage)
|
||
├── 克隆父级 assistant 消息
|
||
├── 生成占位符 tool_result
|
||
└── 附加 directive 文本块
|
||
│
|
||
▼
|
||
[可选] buildWorktreeNotice()
|
||
│
|
||
▼
|
||
runAgent({
|
||
useExactTools: true,
|
||
override.systemPrompt: 父级,
|
||
forkContextMessages: 父级消息,
|
||
availableTools: 父级工具,
|
||
})
|
||
```
|
||
|
||
### 3.4 消息构建：buildForkedMessages
|
||
|
||
文件：`src/tools/AgentTool/forkSubagent.ts:107-169`
|
||
|
||
构建的消息结构：
|
||
|
||
```
|
||
[
|
||
...history (filterIncompleteToolCalls), // 父级完整历史
|
||
assistant(所有 tool_use 块), // 父级当前 turn 的 assistant 消息
|
||
user(
|
||
占位符 tool_result × N + // 相同占位符文本
|
||
<fork-boilerplate> directive // 每个 fork 不同
|
||
)
|
||
]
|
||
```
|
||
|
||
**所有 fork 使用相同的占位符文本**：`"Fork started — processing in background"`。这确保多个并行 fork 的 API 请求前缀完全一致，最大化 prompt cache 命中。
|
||
|
||
### 3.5 递归防护
|
||
|
||
两层检查防止 fork 嵌套：
|
||
|
||
1. **querySource 检查**：`toolUseContext.options.querySource === 'agent:builtin:fork'`。在 `context.options` 上设置，抗自动压缩（autocompact 只重写消息不改 options）
|
||
2. **消息扫描**：`isInForkChild()` 扫描消息历史中的 `<fork-boilerplate>` 标签
|
||
|
||
### 3.6 Worktree 隔离通知
|
||
|
||
当 fork + worktree 组合时，追加通知告知子 agent：
|
||
|
||
> "你继承了父 agent 在 `{parentCwd}` 的对话上下文，但你在独立的 git worktree `{worktreeCwd}` 中操作。路径需要转换，编辑前重新读取。"
|
||
|
||
### 3.7 强制异步
|
||
|
||
当 `isForkSubagentEnabled()` 为 true 时，所有 agent 启动都强制异步。`run_in_background` 参数从 schema 中移除。统一通过 `<task-notification>` XML 消息交互。
|
||
|
||
## 四、Prompt Cache 优化
|
||
|
||
这是整个 fork 设计的核心优化目标：
|
||
|
||
| 优化点 | 实现 |
|
||
|--------|------|
|
||
| **相同 system prompt** | 直传 `renderedSystemPrompt`，避免重新渲染（GrowthBook 状态可能不一致） |
|
||
| **相同工具集** | `useExactTools: true` 直接使用父级工具，不经过 `resolveAgentTools` 过滤 |
|
||
| **相同 thinking config** | 继承父级 thinking 配置（非 fork agent 默认禁用 thinking） |
|
||
| **相同占位符结果** | 所有 fork 使用 `FORK_PLACEHOLDER_RESULT` 相同文本 |
|
||
| **ContentReplacementState 克隆** | 默认克隆父级替换状态，保持 wire prefix 一致 |
|
||
|
||
## 五、子 Agent 指令
|
||
|
||
`buildChildMessage()` 生成 `<fork-boilerplate>` 包裹的指令：
|
||
|
||
- 你是 fork worker，不是主 agent
|
||
- 禁止再次 spawn sub-agent（直接执行）
|
||
- 不要闲聊、不要元评论
|
||
- 直接使用工具
|
||
- 修改文件后要 commit，报告 commit hash
|
||
- 报告格式：`Scope:` / `Result:` / `Key files:` / `Files changed:` / `Issues:`
|
||
|
||
## 六、关键设计决策
|
||
|
||
1. **Fork ≠ 普通 agent**：fork 继承完整上下文，普通 agent 从零开始。选择依据是 `subagent_type` 是否存在
|
||
2. **renderedSystemPrompt 直传**：避免 fork 时重新调用 `getSystemPrompt()`。父级在 turn 开始时冻结 prompt 字节
|
||
3. **占位符结果共享**：多个并行 fork 使用完全相同的占位符，只有 directive 不同
|
||
4. **Coordinator 互斥**：Coordinator 模式下禁用 fork，两者有不兼容的委派模型
|
||
5. **非交互式禁用**：pipe 模式和 SDK 模式下禁用，避免不可见的 fork 嵌套
|
||
|
||
## 七、使用方式
|
||
|
||
```bash
|
||
# 启用 feature
|
||
FEATURE_FORK_SUBAGENT=1 bun run dev
|
||
|
||
# 在 REPL 中使用（不指定 subagent_type 即走 fork）
|
||
# Agent({ prompt: "研究这个模块的结构" })
|
||
# Agent({ prompt: "实现这个功能" })
|
||
```
|
||
|
||
## 八、文件索引
|
||
|
||
| 文件 | 行数 | 职责 |
|
||
|------|------|------|
|
||
| `src/tools/AgentTool/forkSubagent.ts` | ~210 | 核心定义 + 消息构建 + 递归防护 |
|
||
| `src/tools/AgentTool/AgentTool.tsx` | — | Fork 路由 + 强制异步 |
|
||
| `src/tools/AgentTool/prompt.ts` | — | "When to Fork" 提示词段落 |
|
||
| `src/tools/AgentTool/runAgent.ts` | — | useExactTools 路径 |
|
||
| `src/tools/AgentTool/resumeAgent.ts` | — | Fork agent 恢复 |
|
||
| `src/constants/xml.ts` | — | XML 标签常量 |
|
||
| `src/utils/forkedAgent.ts` | — | CacheSafeParams + ContentReplacementState 克隆 |
|
||
| `src/commands/fork/index.ts` | — | /fork 命令（stub） |
|