custom-agents.mdx 7.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210
  1. ---
  2. title: "自定义 Agent"
  3. description: "定义你自己的 AI 角色——从 Markdown 文件到运行时注入的完整链路"
  4. ---
  5. {/* 本章目标:揭示 Agent 定义的完整数据模型、加载发现机制、工具过滤和与 AgentTool 的联动 */}
  6. ## Agent 定义的三种来源
  7. Claude Code 的 Agent 不仅仅来自用户自定义——系统有三类来源,按优先级合并:
  8. | 来源 | 位置 | 优先级 |
  9. |------|------|--------|
  10. | **Built-in** | `src/tools/AgentTool/built-in/` 硬编码 | 最低(可被覆盖) |
  11. | **Plugin** | 通过插件系统注册 | 中 |
  12. | **User/Project/Policy** | `.claude/agents/*.md` 或 settings.json | 最高 |
  13. 合并逻辑在 `getActiveAgentsFromList()` 中:按 `agentType` 去重,后者覆盖前者。这意味着你可以在 `.claude/agents/` 中放一个 `Explore.md` 来完全替换内置的 Explore Agent。
  14. ## Markdown Agent 文件的完整格式
  15. ```markdown
  16. ---
  17. # === 必需字段 ===
  18. name: "reviewer" # Agent 标识(agentType)
  19. description: "Code review specialist, read-only analysis"
  20. # === 工具控制 ===
  21. tools: "Read,Glob,Grep,Bash" # 允许的工具列表(逗号分隔)
  22. disallowedTools: "Write,Edit" # 显式禁止的工具
  23. # === 模型配置 ===
  24. model: "haiku" # 指定模型(或 "inherit" 继承主线程)
  25. effort: "high" # 推理努力程度:low/medium/high 或整数
  26. # === 行为控制 ===
  27. maxTurns: 10 # 最大 agentic 轮次
  28. permissionMode: "plan" # 权限模式:plan/bypassPermissions 等
  29. background: true # 始终作为后台任务运行
  30. initialPrompt: "/search TODO" # 首轮用户消息前缀(支持斜杠命令)
  31. # === 隔离与持久化 ===
  32. isolation: "worktree" # 在独立 git worktree 中运行
  33. memory: "project" # 持久记忆范围:user/project/local
  34. # === MCP 服务器 ===
  35. mcpServers:
  36. - "slack" # 引用已配置的 MCP 服务器
  37. - database: # 内联定义
  38. command: "npx"
  39. args: ["mcp-db"]
  40. # === Hooks ===
  41. hooks:
  42. PreToolUse:
  43. - command: "audit-log.sh"
  44. timeout: 5000
  45. # === Skills ===
  46. skills: "code-review,security-review" # 预加载的 skills(逗号分隔)
  47. # === 显示 ===
  48. color: "blue" # 终端中的 Agent 颜色标识
  49. ---
  50. 你是代码审查专家。你的职责是...
  51. (正文内容 = system prompt)
  52. ```
  53. ### 字段解析细节
  54. - **`tools`**:通过 `parseAgentToolsFromFrontmatter()` 解析,支持逗号分隔字符串或数组
  55. - **`model: "inherit"`**:使用主线程的模型(区分大小写,只有小写 "inherit" 有效)
  56. - **`memory`**:启用后自动注入 `Write`/`Edit`/`Read` 工具(即使 `tools` 未包含),并在 system prompt 末尾追加 memory 指令
  57. - **`isolation: "remote"`**:仅在 Anthropic 内部可用(`USER_TYPE === 'ant'`),外部构建只支持 `worktree`
  58. - **`background`**:`true` 使 Agent 始终在后台运行,主线程不等待结果
  59. ## 加载与发现机制
  60. `getAgentDefinitionsWithOverrides()`(被 `memoize` 缓存)执行完整的发现流程:
  61. ```
  62. 1. 加载 Markdown 文件
  63. ├── loadMarkdownFilesForSubdir('agents', cwd)
  64. │ ├── ~/.claude/agents/*.md (用户级,source = 'userSettings')
  65. │ ├── .claude/agents/*.md (项目级,source = 'projectSettings')
  66. │ └── managed/policy sources (策略级,source = 'policySettings')
  67. └── 每个 .md 文件:
  68. ├── 解析 YAML frontmatter
  69. ├── 正文作为 system prompt
  70. ├── 校验必需字段(name, description)
  71. ├── 静默跳过无 frontmatter 的 .md 文件(可能是参考文档)
  72. └── 解析失败 → 记录到 failedFiles,不阻塞其他 Agent
  73. 2. 并行加载 Plugin Agents
  74. └── loadPluginAgents() → memoized
  75. 3. 初始化 Memory Snapshots(如果 AGENT_MEMORY_SNAPSHOT 启用)
  76. └── initializeAgentMemorySnapshots()
  77. 4. 合并 Built-in + Plugin + Custom
  78. └── getActiveAgentsFromList() → 按 agentType 去重,后者覆盖前者
  79. 5. 分配颜色
  80. └── setAgentColor(agentType, color) → 终端 UI 中区分不同 Agent
  81. ```
  82. ## 工具过滤的实现
  83. 当 Agent 被派生时,`AgentTool` 根据定义中的 `tools` / `disallowedTools` 过滤可用工具列表:
  84. ```
  85. 全部工具
  86. ↓ disallowedTools 移除
  87. ↓ tools 白名单过滤(如果指定)
  88. 可用工具
  89. ```
  90. - **`tools` 未指定**:Agent 可以使用所有工具(默认全能)
  91. - **`tools` 指定**:只能使用列出的工具
  92. - **`disallowedTools`**:即使 `tools` 未指定,这些工具也被禁止
  93. - **自动注入**:`memory` 启用时自动添加 `Write`/`Edit`/`Read`
  94. 以内置 Explore Agent 为例:
  95. ```typescript
  96. // src/tools/AgentTool/built-in/exploreAgent.ts
  97. disallowedTools: [
  98. 'Agent', // 不能嵌套调用 Agent
  99. 'ExitPlanMode', // 不需要 plan mode
  100. 'FileEdit', // 只读
  101. 'FileWrite', // 只读
  102. 'NotebookEdit', // 只读
  103. ]
  104. ```
  105. ## System Prompt 的注入方式
  106. Agent 的 system prompt 通过 `getSystemPrompt()` 闭包延迟生成:
  107. ```typescript
  108. // Markdown Agent
  109. getSystemPrompt: () => {
  110. if (isAutoMemoryEnabled() && memory) {
  111. return systemPrompt + '\n\n' + loadAgentMemoryPrompt(agentType, memory)
  112. }
  113. return systemPrompt
  114. }
  115. ```
  116. 这意味着:
  117. 1. **Markdown 正文 = 完整的 system prompt**——不是追加,而是替换默认 prompt
  118. 2. **Memory 指令**在 memory 启用时自动追加到末尾
  119. 3. **闭包延迟计算**——memory 状态可能在文件加载后才变化
  120. 对于 Built-in Agent,`getSystemPrompt` 接受 `toolUseContext` 参数,可以根据运行时状态(如是否使用嵌入式搜索工具)动态调整 prompt 内容。
  121. ## 与 AgentTool 的联动
  122. 当主 Agent 需要派生子 Agent 时:
  123. ```
  124. AgentTool.call({ subagent_type: "reviewer", ... })
  125. 1. 从 agentDefinitions.activeAgents 查找 agentType === "reviewer"
  126. 2. 检查 requiredMcpServers(如果 Agent 要求特定 MCP 服务器)
  127. 3. 过滤工具列表(tools / disallowedTools)
  128. 4. 解析模型:
  129. - "inherit" → 使用主线程模型
  130. - 具体模型名 → 直接使用
  131. - 未指定 → 主线程模型
  132. 5. 解析权限模式(permissionMode)
  133. 6. 构建隔离环境(如果 isolation === "worktree")
  134. 7. 注入 system prompt(getSystemPrompt())
  135. 8. 注入 initialPrompt(如果定义了)
  136. 9. 启动子 Agent 循环(forkSubagent / runAgent)
  137. ```
  138. ## 内置 Agent 参考
  139. | Agent | agentType | 角色 | 工具限制 | 模型 |
  140. |-------|-----------|------|---------|------|
  141. | **General Purpose** | `general-purpose` | 默认子 Agent | 全部工具 | 主线程模型 |
  142. | **Explore** | `Explore` | 代码搜索专家 | 只读(无 Write/Edit) | haiku(外部) |
  143. | **Plan** | `Plan` | 规划专家 | 只读 + ExitPlanMode | inherit |
  144. | **Verification** | `verification` | 结果验证 | 由 feature flag 控制 | — |
  145. | **Code Guide** | `claude-code-guide` | Claude Code 使用指南 | 只读 | — |
  146. | **Statusline Setup** | `statusline-setup` | 终端状态栏配置 | 有限 | — |
  147. SDK 入口(`sdk-ts`/`sdk-py`/`sdk-cli`)不加载 Code Guide Agent。环境变量 `CLAUDE_AGENT_SDK_DISABLE_BUILTIN_AGENTS` 可以完全禁用内置 Agent,给 SDK 用户提供空白画布。
  148. ## Agent Memory:持久化的 Agent 状态
  149. 当 `memory` 字段启用时,Agent 获得跨会话的持久记忆:
  150. - **`local`**:当前项目、当前用户有效
  151. - **`project`**:当前项目所有用户共享
  152. - **`user`**:所有项目共享
  153. Memory 通过 `loadAgentMemoryPrompt()` 注入到 system prompt 末尾,包含读写记忆的指令。Agent Memory Snapshot 机制在项目间同步 `user` 级记忆。