Agent SDK 参考 - TypeScript

npm install @anthropic-ai/claude-agent-sdk

function query({
  prompt,
  options
}: {
  prompt: string | AsyncIterable<SDKUserMessage>;
  options?: Options;
}): Query

function tool<Schema extends ZodRawShape>(
  name: string,
  description: string,
  inputSchema: Schema,
  handler: (args: z.infer<ZodObject<Schema>>, extra: unknown) => Promise<CallToolResult>
): SdkMcpToolDefinition<Schema>

function createSdkMcpServer(options: {
  name: string;
  version?: string;
  tools?: Array<SdkMcpToolDefinition<any>>;
}): McpSdkServerConfigWithInstance

interface Query extends AsyncGenerator<SDKMessage, void> {
  interrupt(): Promise<void>;
  setPermissionMode(mode: PermissionMode): Promise<void>;
}

type AgentDefinition = {
  description: string;
  tools?: string[];
  prompt: string;
  model?: 'sonnet' | 'opus' | 'haiku' | 'inherit';
}

type SettingSource = 'user' | 'project' | 'local';

// 像 SDK v0.0.x 一样加载所有设置
const result = query({
  prompt: "分析这段代码",
  options: {
    settingSources: ['user', 'project', 'local']  // 加载所有设置
  }
});

// 仅加载项目设置，忽略用户和本地设置
const result = query({
  prompt: "运行 CI 检查",
  options: {
    settingSources: ['project']  // 仅 .claude/settings.json
  }
});

// 通过排除本地设置确保 CI 中的一致行为
const result = query({
  prompt: "运行测试",
  options: {
    settingSources: ['project'],  // 仅团队共享设置
    permissionMode: 'bypassPermissions'
  }
});

// 以编程方式定义所有内容（默认行为）
// 无文件系统依赖 - settingSources 默认为 []
const result = query({
  prompt: "审查此 PR",
  options: {
    // settingSources: [] 是默认值，无需指定
    agents: { /* ... */ },
    mcpServers: { /* ... */ },
    allowedTools: ['Read', 'Grep', 'Glob']
  }
});

// 加载项目设置以包含 CLAUDE.md 文件
const result = query({
  prompt: "按照项目约定添加新功能",
  options: {
    systemPrompt: {
      type: 'preset',
      preset: 'claude_code'  // 需要使用 CLAUDE.md
    },
    settingSources: ['project'],  // 从项目目录加载 CLAUDE.md
    allowedTools: ['Read', 'Write', 'Edit']
  }
});

type PermissionMode =
  | 'default'           // 标准权限行为
  | 'acceptEdits'       // 自动接受文件编辑
  | 'bypassPermissions' // 绕过所有权限检查
  | 'plan'              // 规划模式 - 无执行

type CanUseTool = (
  toolName: string,
  input: ToolInput,
  options: {
    signal: AbortSignal;
    suggestions?: PermissionUpdate[];
  }
) => Promise<PermissionResult>;

type PermissionResult = 
  | {
      behavior: 'allow';
      updatedInput: ToolInput;
      updatedPermissions?: PermissionUpdate[];
    }
  | {
      behavior: 'deny';
      message: string;
      interrupt?: boolean;
    }

type McpServerConfig = 
  | McpStdioServerConfig
  | McpSSEServerConfig
  | McpHttpServerConfig
  | McpSdkServerConfigWithInstance;

type McpStdioServerConfig = {
  type?: 'stdio';
  command: string;
  args?: string[];
  env?: Record<string, string>;
}

type McpSSEServerConfig = {
  type: 'sse';
  url: string;
  headers?: Record<string, string>;
}

type McpHttpServerConfig = {
  type: 'http';
  url: string;
  headers?: Record<string, string>;
}

type McpSdkServerConfigWithInstance = {
  type: 'sdk';
  name: string;
  instance: McpServer;
}

type SdkPluginConfig = {
  type: 'local';
  path: string;
}

plugins: [
  { type: 'local', path: './my-plugin' },
  { type: 'local', path: '/absolute/path/to/plugin' }
]

type SDKMessage = 
  | SDKAssistantMessage
  | SDKUserMessage
  | SDKUserMessageReplay
  | SDKResultMessage
  | SDKSystemMessage
  | SDKPartialAssistantMessage
  | SDKCompactBoundaryMessage;

type SDKAssistantMessage = {
  type: 'assistant';
  uuid: UUID;
  session_id: string;
  message: APIAssistantMessage; // 来自 Anthropic SDK
  parent_tool_use_id: string | null;
}

type SDKUserMessage = {
  type: 'user';
  uuid?: UUID;
  session_id: string;
  message: APIUserMessage; // 来自 Anthropic SDK
  parent_tool_use_id: string | null;
}

type SDKUserMessageReplay = {
  type: 'user';
  uuid: UUID;
  session_id: string;
  message: APIUserMessage;
  parent_tool_use_id: string | null;
}

type SDKResultMessage = 
  | {
      type: 'result';
      subtype: 'success';
      uuid: UUID;
      session_id: string;
      duration_ms: number;
      duration_api_ms: number;
      is_error: boolean;
      num_turns: number;
      result: string;
      total_cost_usd: number;
      usage: NonNullableUsage;
      permission_denials: SDKPermissionDenial[];
    }
  | {
      type: 'result';
      subtype: 'error_max_turns' | 'error_during_execution';
      uuid: UUID;
      session_id: string;
      duration_ms: number;
      duration_api_ms: number;
      is_error: boolean;
      num_turns: number;
      total_cost_usd: number;
      usage: NonNullableUsage;
      permission_denials: SDKPermissionDenial[];
    }

type SDKSystemMessage = {
  type: 'system';
  subtype: 'init';
  uuid: UUID;
  session_id: string;
  apiKeySource: ApiKeySource;
  cwd: string;
  tools: string[];
  mcp_servers: {
    name: string;
    status: string;
  }[];
  model: string;
  permissionMode: PermissionMode;
  slash_commands: string[];
  output_style: string;
}

type SDKPartialAssistantMessage = {
  type: 'stream_event';
  event: RawMessageStreamEvent; // 来自 Anthropic SDK
  parent_tool_use_id: string | null;
  uuid: UUID;
  session_id: string;
}

type SDKCompactBoundaryMessage = {
  type: 'system';
  subtype: 'compact_boundary';
  uuid: UUID;
  session_id: string;
  compact_metadata: {
    trigger: 'manual' | 'auto';
    pre_tokens: number;
  };
}

type SDKPermissionDenial = {
  tool_name: string;
  tool_use_id: string;
  tool_input: ToolInput;
}

type HookEvent = 
  | 'PreToolUse'
  | 'PostToolUse'
  | 'Notification'
  | 'UserPromptSubmit'
  | 'SessionStart'
  | 'SessionEnd'
  | 'Stop'
  | 'SubagentStop'
  | 'PreCompact';

type HookCallback = (
  input: HookInput, // 所有钩子输入类型的联合
  toolUseID: string | undefined,
  options: { signal: AbortSignal }
) => Promise<HookJSONOutput>;

interface HookCallbackMatcher {
  matcher?: string;
  hooks: HookCallback[];
}

type HookInput = 
  | PreToolUseHookInput
  | PostToolUseHookInput
  | NotificationHookInput
  | UserPromptSubmitHookInput
  | SessionStartHookInput
  | SessionEndHookInput
  | StopHookInput
  | SubagentStopHookInput
  | PreCompactHookInput;

type BaseHookInput = {
  session_id: string;
  transcript_path: string;
  cwd: string;
  permission_mode?: string;
}

type PreToolUseHookInput = BaseHookInput & {
  hook_event_name: 'PreToolUse';
  tool_name: string;
  tool_input: ToolInput;
}

type PostToolUseHookInput = BaseHookInput & {
  hook_event_name: 'PostToolUse';
  tool_name: string;
  tool_input: ToolInput;
  tool_response: ToolOutput;
}

type NotificationHookInput = BaseHookInput & {
  hook_event_name: 'Notification';
  message: string;
  title?: string;
}

type UserPromptSubmitHookInput = BaseHookInput & {
  hook_event_name: 'UserPromptSubmit';
  prompt: string;
}

type SessionStartHookInput = BaseHookInput & {
  hook_event_name: 'SessionStart';
  source: 'startup' | 'resume' | 'clear' | 'compact';
}

type SessionEndHookInput = BaseHookInput & {
  hook_event_name: 'SessionEnd';
  reason: 'clear' | 'logout' | 'prompt_input_exit' | 'other';
}

type StopHookInput = BaseHookInput & {
  hook_event_name: 'Stop';
  stop_hook_active: boolean;
}

type SubagentStopHookInput = BaseHookInput & {
  hook_event_name: 'SubagentStop';
  stop_hook_active: boolean;
}

type PreCompactHookInput = BaseHookInput & {
  hook_event_name: 'PreCompact';
  trigger: 'manual' | 'auto';
  custom_instructions: string | null;
}

type HookJSONOutput = AsyncHookJSONOutput | SyncHookJSONOutput;

type AsyncHookJSONOutput = {
  async: true;
  asyncTimeout?: number;
}

type SyncHookJSONOutput = {
  continue?: boolean;
  suppressOutput?: boolean;
  stopReason?: string;
  decision?: 'approve' | 'block';
  systemMessage?: string;
  reason?: string;
  hookSpecificOutput?:
    | {
        hookEventName: 'PreToolUse';
        permissionDecision?: 'allow' | 'deny' | 'ask';
        permissionDecisionReason?: string;
      }
    | {
        hookEventName: 'UserPromptSubmit';
        additionalContext?: string;
      }
    | {
        hookEventName: 'SessionStart';
        additionalContext?: string;
      }
    | {
        hookEventName: 'PostToolUse';
        additionalContext?: string;
      };
}

type ToolInput = 
  | AgentInput
  | BashInput
  | BashOutputInput
  | FileEditInput
  | FileReadInput
  | FileWriteInput
  | GlobInput
  | GrepInput
  | KillShellInput
  | NotebookEditInput
  | WebFetchInput
  | WebSearchInput
  | TodoWriteInput
  | ExitPlanModeInput
  | ListMcpResourcesInput
  | ReadMcpResourceInput;

interface AgentInput {
  /**
   * 任务的简短描述（3-5 个单词）
   */
  description: string;
  /**
   * 代理要执行的任务
   */
  prompt: string;
  /**
   * 用于此任务的专门代理类型
   */
  subagent_type: string;
}

interface BashInput {
  /**
   * 要执行的命令
   */
  command: string;
  /**
   * 可选的超时时间（以毫秒为单位，最大 600000）
   */
  timeout?: number;
  /**
   * 清晰简洁的描述，说明此命令在 5-10 个单词内的作用
   */
  description?: string;
  /**
   * 设置为 true 以在后台运行此命令
   */
  run_in_background?: boolean;
}

interface BashOutputInput {
  /**
   * 要从中检索输出的后台 shell 的 ID
   */
  bash_id: string;
  /**
   * 可选的正则表达式以过滤输出行
   */
  filter?: string;
}

interface FileEditInput {
  /**
   * 要修改的文件的绝对路径
   */
  file_path: string;
  /**
   * 要替换的文本
   */
  old_string: string;
  /**
   * 用来替换它的文本（必须与 old_string 不同）
   */
  new_string: string;
  /**
   * 替换 old_string 的所有出现（默认 false）
   */
  replace_all?: boolean;
}

interface FileReadInput {
  /**
   * 要读取的文件的绝对路径
   */
  file_path: string;
  /**
   * 开始读取的行号
   */
  offset?: number;
  /**
   * 要读取的行数
   */
  limit?: number;
}

interface FileWriteInput {
  /**
   * 要写入的文件的绝对路径
   */
  file_path: string;
  /**
   * 要写入文件的内容
   */
  content: string;
}

interface GlobInput {
  /**
   * 用于匹配文件的 glob 模式
   */
  pattern: string;
  /**
   * 要搜索的目录（默认为 cwd）
   */
  path?: string;
}

interface GrepInput {
  /**
   * 要搜索的正则表达式模式
   */
  pattern: string;
  /**
   * 要搜索的文件或目录（默认为 cwd）
   */
  path?: string;
  /**
   * 用于过滤文件的 glob 模式（例如 "*.js"）
   */
  glob?: string;
  /**
   * 要搜索的文件类型（例如 "js"、"py"、"rust"）
   */
  type?: string;
  /**
   * 输出模式："content"、"files_with_matches" 或 "count"
   */
  output_mode?: 'content' | 'files_with_matches' | 'count';
  /**
   * 不区分大小写的搜索
   */
  '-i'?: boolean;
  /**
   * 显示行号（用于内容模式）
   */
  '-n'?: boolean;
  /**
   * 每个匹配项之前显示的行数
   */
  '-B'?: number;
  /**
   * 每个匹配项之后显示的行数
   */
  '-A'?: number;
  /**
   * 每个匹配项之前和之后显示的行数
   */
  '-C'?: number;
  /**
   * 将输出限制为前 N 行/条目
   */
  head_limit?: number;
  /**
   * 启用多行模式
   */
  multiline?: boolean;
}

interface KillShellInput {
  /**
   * 要终止的后台 shell 的 ID
   */
  shell_id: string;
}

interface NotebookEditInput {
  /**
   * Jupyter notebook 文件的绝对路径
   */
  notebook_path: string;
  /**
   * 要编辑的单元格的 ID
   */
  cell_id?: string;
  /**
   * 单元格的新源代码
   */
  new_source: string;
  /**
   * 单元格的类型（代码或 markdown）
   */
  cell_type?: 'code' | 'markdown';
  /**
   * 编辑类型（替换、插入、删除）
   */
  edit_mode?: 'replace' | 'insert' | 'delete';
}

interface WebFetchInput {
  /**
   * 要从中获取内容的 URL
   */
  url: string;
  /**
   * 在获取的内容上运行的提示
   */
  prompt: string;
}

interface WebSearchInput {
  /**
   * 要使用的搜索查询
   */
  query: string;
  /**
   * 仅包含来自这些域的结果
   */
  allowed_domains?: string[];
  /**
   * 永远不包含来自这些域的结果
   */
  blocked_domains?: string[];
}

interface TodoWriteInput {
  /**
   * 更新的待办事项列表
   */
  todos: Array<{
    /**
     * 任务描述
     */
    content: string;
    /**
     * 任务状态
     */
    status: 'pending' | 'in_progress' | 'completed';
    /**
     * 任务描述的主动形式
     */
    activeForm: string;
  }>;
}

interface ExitPlanModeInput {
  /**
   * 用户批准要运行的计划
   */
  plan: string;
}

interface ListMcpResourcesInput {
  /**
   * 可选的服务器名称以按其过滤资源
   */
  server?: string;
}

interface ReadMcpResourceInput {
  /**
   * MCP 服务器名称
   */
  server: string;
  /**
   * 要读取的资源 URI
   */
  uri: string;
}

type ToolOutput = 
  | TaskOutput
  | BashOutput
  | BashOutputToolOutput
  | EditOutput
  | ReadOutput
  | WriteOutput
  | GlobOutput
  | GrepOutput
  | KillBashOutput
  | NotebookEditOutput
  | WebFetchOutput
  | WebSearchOutput
  | TodoWriteOutput
  | ExitPlanModeOutput
  | ListMcpResourcesOutput
  | ReadMcpResourceOutput;

interface TaskOutput {
  /**
   * 子代理的最终结果消息
   */
  result: string;
  /**
   * 令牌使用统计
   */
  usage?: {
    input_tokens: number;
    output_tokens: number;
    cache_creation_input_tokens?: number;
    cache_read_input_tokens?: number;
  };
  /**
   * 以美元计的总成本
   */
  total_cost_usd?: number;
  /**
   * 执行持续时间（以毫秒为单位）
   */
  duration_ms?: number;
}

interface BashOutput {
  /**
   * 合并的 stdout 和 stderr 输出
   */
  output: string;
  /**
   * 命令的退出代码
   */
  exitCode: number;
  /**
   * 命令是否因超时而被终止
   */
  killed?: boolean;
  /**
   * 后台进程的 Shell ID
   */
  shellId?: string;
}

interface BashOutputToolOutput {
  /**
   * 自上次检查以来的新输出
   */
  output: string;
  /**
   * 当前 shell 状态
   */
  status: 'running' | 'completed' | 'failed';
  /**
   * 退出代码（完成时）
   */
  exitCode?: number;
}

interface EditOutput {
  /**
   * 确认消息
   */
  message: string;
  /**
   * 进行的替换次数
   */
  replacements: number;
  /**
   * 被编辑的文件路径
   */
  file_path: string;
}

type ReadOutput = 
  | TextFileOutput
  | ImageFileOutput
  | PDFFileOutput
  | NotebookFileOutput;

interface TextFileOutput {
  /**
   * 带行号的文件内容
   */
  content: string;
  /**
   * 文件中的总行数
   */
  total_lines: number;
  /**
   * 实际返回的行数
   */
  lines_returned: number;
}

interface ImageFileOutput {
  /**
   * Base64 编码的图像数据
   */
  image: string;
  /**
   * 图像 MIME 类型
   */
  mime_type: string;
  /**
   * 文件大小（以字节为单位）
   */
  file_size: number;
}

interface PDFFileOutput {
  /**
   * 页面内容数组
   */
  pages: Array<{
    page_number: number;
    text?: string;
    images?: Array<{
      image: string;
      mime_type: string;
    }>;
  }>;
  /**
   * 总页数
   */
  total_pages: number;
}

interface NotebookFileOutput {
  /**
   * Jupyter notebook 单元格
   */
  cells: Array<{
    cell_type: 'code' | 'markdown';
    source: string;
    outputs?: any[];
    execution_count?: number;
  }>;
  /**
   * Notebook 元数据
   */
  metadata?: Record<string, any>;
}

interface WriteOutput {
  /**
   * 成功消息
   */
  message: string;
  /**
   * 写入的字节数
   */
  bytes_written: number;
  /**
   * 被写入的文件路径
   */
  file_path: string;
}

interface GlobOutput {
  /**
   * 匹配的文件路径数组
   */
  matches: string[];
  /**
   * 找到的匹配数
   */
  count: number;
  /**
   * 使用的搜索目录
   */
  search_path: string;
}

type GrepOutput = 
  | GrepContentOutput
  | GrepFilesOutput
  | GrepCountOutput;

interface GrepContentOutput {
  /**
   * 带上下文的匹配行
   */
  matches: Array<{
    file: string;
    line_number?: number;
    line: string;
    before_context?: string[];
    after_context?: string[];
  }>;
  /**
   * 匹配总数
   */
  total_matches: number;
}

interface GrepFilesOutput {
  /**
   * 包含匹配的文件
   */
  files: string[];
  /**
   * 包含匹配的文件数
   */
  count: number;
}

interface GrepCountOutput {
  /**
   * 每个文件的匹配计数
   */
  counts: Array<{
    file: string;
    count: number;
  }>;
  /**
   * 所有文件中的总匹配数
   */
  total: number;
}

interface KillBashOutput {
  /**
   * 成功消息
   */
  message: string;
  /**
   * 被终止的 shell 的 ID
   */
  shell_id: string;
}

interface NotebookEditOutput {
  /**
   * 成功消息
   */
  message: string;
  /**
   * 执行的编辑类型
   */
  edit_type: 'replaced' | 'inserted' | 'deleted';
  /**
   * 受影响的单元格 ID
   */
  cell_id?: string;
  /**
   * 编辑后 notebook 中的总单元格数
   */
  total_cells: number;
}

interface WebFetchOutput {
  /**
   * AI 模型对提示的响应
   */
  response: string;
  /**
   * 被获取的 URL
   */
  url: string;
  /**
   * 重定向后的最终 URL
   */
  final_url?: string;
  /**
   * HTTP 状态代码
   */
  status_code?: number;
}

interface WebSearchOutput {
  /**
   * 搜索结果
   */
  results: Array<{
    title: string;
    url: string;
    snippet: string;
    /**
     * 如果可用的其他元数据
     */
    metadata?: Record<string, any>;
  }>;
  /**
   * 结果总数
   */
  total_results: number;
  /**
   * 被搜索的查询
   */
  query: string;
}

interface TodoWriteOutput {
  /**
   * 成功消息
   */
  message: string;
  /**
   * 当前待办事项统计
   */
  stats: {
    total: number;
    pending: number;
    in_progress: number;
    completed: number;
  };
}

interface ExitPlanModeOutput {
  /**
   * 确认消息
   */
  message: string;
  /**
   * 用户是否批准了计划
   */
  approved?: boolean;
}

interface ListMcpResourcesOutput {
  /**
   * 可用资源
   */
  resources: Array<{
    uri: string;
    name: string;
    description?: string;
    mimeType?: string;
    server: string;
  }>;
  /**
   * 资源总数
   */
  total: number;
}

interface ReadMcpResourceOutput {
  /**
   * 资源内容
   */
  contents: Array<{
    uri: string;
    mimeType?: string;
    text?: string;
    blob?: string;
  }>;
  /**
   * 提供资源的服务器
   */
  server: string;
}

type PermissionUpdate = 
  | {
      type: 'addRules';
      rules: PermissionRuleValue[];
      behavior: PermissionBehavior;
      destination: PermissionUpdateDestination;
    }
  | {
      type: 'replaceRules';
      rules: PermissionRuleValue[];
      behavior: PermissionBehavior;
      destination: PermissionUpdateDestination;
    }
  | {
      type: 'removeRules';
      rules: PermissionRuleValue[];
      behavior: PermissionBehavior;
      destination: PermissionUpdateDestination;
    }
  | {
      type: 'setMode';
      mode: PermissionMode;
      destination: PermissionUpdateDestination;
    }
  | {
      type: 'addDirectories';
      directories: string[];
      destination: PermissionUpdateDestination;
    }
  | {
      type: 'removeDirectories';
      directories: string[];
      destination: PermissionUpdateDestination;
    }

type PermissionBehavior = 'allow' | 'deny' | 'ask';

type PermissionUpdateDestination = 
  | 'userSettings'     // 全局用户设置
  | 'projectSettings'  // 按目录项目设置
  | 'localSettings'    // Gitignored 本地设置
  | 'session'          // 仅当前会话

type PermissionRuleValue = {
  toolName: string;
  ruleContent?: string;
}

type ApiKeySource = 'user' | 'project' | 'org' | 'temporary';

type ConfigScope = 'local' | 'user' | 'project';

type NonNullableUsage = {
  [K in keyof Usage]: NonNullable<Usage[K]>;
}

type Usage = {
  input_tokens: number | null;
  output_tokens: number | null;
  cache_creation_input_tokens?: number | null;
  cache_read_input_tokens?: number | null;
}

type CallToolResult = {
  content: Array<{
    type: 'text' | 'image' | 'resource';
    // 其他字段因类型而异
  }>;
  isError?: boolean;
}

class AbortError extends Error {}