Agent SDK 参考 - Python

pip install claude-agent-sdk

async def query(
    *,
    prompt: str | AsyncIterable[dict[str, Any]],
    options: ClaudeAgentOptions | None = None
) -> AsyncIterator[Message]

import asyncio
from claude_agent_sdk import query, ClaudeAgentOptions

async def main():
    options = ClaudeAgentOptions(
        system_prompt="你是一个专业的 Python 开发者",
        permission_mode='acceptEdits',
        cwd="/home/user/project"
    )

    async for message in query(
        prompt="创建一个 Python Web 服务器",
        options=options
    ):
        print(message)


asyncio.run(main())

def tool(
    name: str,
    description: str,
    input_schema: type | dict[str, Any]
) -> Callable[[Callable[[Any], Awaitable[dict[str, Any]]]], SdkMcpTool[Any]]

from claude_agent_sdk import tool
from typing import Any

@tool("greet", "问候用户", {"name": str})
async def greet(args: dict[str, Any]) -> dict[str, Any]:
    return {
        "content": [{
            "type": "text",
            "text": f"你好，{args['name']}！"
        }]
    }

def create_sdk_mcp_server(
    name: str,
    version: str = "1.0.0",
    tools: list[SdkMcpTool[Any]] | None = None
) -> McpSdkServerConfig

from claude_agent_sdk import tool, create_sdk_mcp_server

@tool("add", "两数相加", {"a": float, "b": float})
async def add(args):
    return {
        "content": [{
            "type": "text",
            "text": f"和：{args['a'] + args['b']}"
        }]
    }

@tool("multiply", "两数相乘", {"a": float, "b": float})
async def multiply(args):
    return {
        "content": [{
            "type": "text",
            "text": f"积：{args['a'] * args['b']}"
        }]
    }

calculator = create_sdk_mcp_server(
    name="calculator",
    version="2.0.0",
    tools=[add, multiply]  # 传递装饰的函数
)

# 与 Claude 一起使用
options = ClaudeAgentOptions(
    mcp_servers={"calc": calculator},
    allowed_tools=["mcp__calc__add", "mcp__calc__multiply"]
)

class ClaudeSDKClient:
    def __init__(self, options: ClaudeAgentOptions | None = None)
    async def connect(self, prompt: str | AsyncIterable[dict] | None = None) -> None
    async def query(self, prompt: str | AsyncIterable[dict], session_id: str = "default") -> None
    async def receive_messages(self) -> AsyncIterator[Message]
    async def receive_response(self) -> AsyncIterator[Message]
    async def interrupt(self) -> None
    async def disconnect(self) -> None

async with ClaudeSDKClient() as client:
    await client.query("你好 Claude")
    async for message in client.receive_response():
        print(message)

import asyncio
from claude_agent_sdk import ClaudeSDKClient, AssistantMessage, TextBlock, ResultMessage

async def main():
    async with ClaudeSDKClient() as client:
        # 第一个问题
        await client.query("法国的首都是什么？")
        
        # 处理响应
        async for message in client.receive_response():
            if isinstance(message, AssistantMessage):
                for block in message.content:
                    if isinstance(block, TextBlock):
                        print(f"Claude：{block.text}")
        
        # 后续问题 - Claude 记住先前的上下文
        await client.query("那个城市的人口是多少？")
        
        async for message in client.receive_response():
            if isinstance(message, AssistantMessage):
                for block in message.content:
                    if isinstance(block, TextBlock):
                        print(f"Claude：{block.text}")
        
        # 另一个后续问题 - 仍在同一对话中
        await client.query("那里有哪些著名的地标？")
        
        async for message in client.receive_response():
            if isinstance(message, AssistantMessage):
                for block in message.content:
                    if isinstance(block, TextBlock):
                        print(f"Claude：{block.text}")

asyncio.run(main())

import asyncio
from claude_agent_sdk import ClaudeSDKClient

async def message_stream():
    """动态生成消息。"""
    yield {"type": "text", "text": "分析以下数据："}
    await asyncio.sleep(0.5)
    yield {"type": "text", "text": "温度：25°C"}
    await asyncio.sleep(0.5)
    yield {"type": "text", "text": "湿度：60%"}
    await asyncio.sleep(0.5)
    yield {"type": "text", "text": "你看到了什么模式？"}

async def main():
    async with ClaudeSDKClient() as client:
        # 向 Claude 流式输入
        await client.query(message_stream())
        
        # 处理响应
        async for message in client.receive_response():
            print(message)
        
        # 在同一会话中的后续问题
        await client.query("我们应该担心这些读数吗？")
        
        async for message in client.receive_response():
            print(message)

asyncio.run(main())

import asyncio
from claude_agent_sdk import ClaudeSDKClient, ClaudeAgentOptions

async def interruptible_task():
    options = ClaudeAgentOptions(
        allowed_tools=["Bash"],
        permission_mode="acceptEdits"
    )
    
    async with ClaudeSDKClient(options=options) as client:
        # 开始一个长时间运行的任务
        await client.query("慢慢地从 1 数到 100")
        
        # 让它运行一会儿
        await asyncio.sleep(2)
        
        # 中断任务
        await client.interrupt()
        print("任务已中断！")
        
        # 发送新命令
        await client.query("改为只说你好")
        
        async for message in client.receive_response():
            # 处理新响应
            pass

asyncio.run(interruptible_task())

from claude_agent_sdk import (
    ClaudeSDKClient,
    ClaudeAgentOptions
)

async def custom_permission_handler(
    tool_name: str,
    input_data: dict,
    context: dict
):
    """工具权限的自定义逻辑。"""

    # 阻止写入系统目录
    if tool_name == "Write" and input_data.get("file_path", "").startswith("/system/"):
        return {
            "behavior": "deny",
            "message": "不允许写入系统目录",
            "interrupt": True
        }

    # 重定向敏感文件操作
    if tool_name in ["Write", "Edit"] and "config" in input_data.get("file_path", ""):
        safe_path = f"./sandbox/{input_data['file_path']}"
        return {
            "behavior": "allow",
            "updatedInput": {**input_data, "file_path": safe_path}
        }

    # 允许其他所有操作
    return {
        "behavior": "allow",
        "updatedInput": input_data
    }

async def main():
    options = ClaudeAgentOptions(
        can_use_tool=custom_permission_handler,
        allowed_tools=["Read", "Write", "Edit"]
    )
    
    async with ClaudeSDKClient(options=options) as client:
        await client.query("更新系统配置文件")
        
        async for message in client.receive_response():
            # 将使用沙盒路径
            print(message)

asyncio.run(main())

@dataclass
class SdkMcpTool(Generic[T]):
    name: str
    description: str
    input_schema: type[T] | dict[str, Any]
    handler: Callable[[T], Awaitable[dict[str, Any]]]

@dataclass
class ClaudeAgentOptions:
    allowed_tools: list[str] = field(default_factory=list)
    max_thinking_tokens: int = 8000
    system_prompt: str | None = None
    mcp_servers: dict[str, McpServerConfig] | str | Path = field(default_factory=dict)
    permission_mode: PermissionMode | None = None
    continue_conversation: bool = False
    resume: str | None = None
    fork_session: bool = False
    max_turns: int | None = None
    disallowed_tools: list[str] = field(default_factory=list)
    model: str | None = None
    permission_prompt_tool_name: str | None = None
    cwd: str | Path | None = None
    settings: str | None = None
    add_dirs: list[str | Path] = field(default_factory=list)
    env: dict[str, str] = field(default_factory=dict)
    extra_args: dict[str, str | None] = field(default_factory=dict)

SettingSource = Literal["user", "project", "local"]

# 像 SDK v0.0.x 那样加载所有设置
from claude_agent_sdk import query, ClaudeAgentOptions

async for message in query(
    prompt="分析这段代码",
    options=ClaudeAgentOptions(
        setting_sources=["user", "project", "local"]  # 加载所有设置
    )
):
    print(message)

# 仅加载项目设置，忽略用户和本地设置
async for message in query(
    prompt="运行 CI 检查",
    options=ClaudeAgentOptions(
        setting_sources=["project"]  # 仅 .claude/settings.json
    )
):
    print(message)

# 通过排除本地设置确保 CI 中的一致行为
async for message in query(
    prompt="运行测试",
    options=ClaudeAgentOptions(
        setting_sources=["project"],  # 仅团队共享设置
        permission_mode="bypassPermissions"
    )
):
    print(message)

# 程序化定义所有内容（默认行为）
# 无文件系统依赖 - setting_sources 默认为 None
async for message in query(
    prompt="审查这个 PR",
    options=ClaudeAgentOptions(
        # setting_sources=None 是默认值，无需指定
        agents={ /* ... */ },
        mcp_servers={ /* ... */ },
        allowed_tools=["Read", "Grep", "Glob"]
    )
):
    print(message)

@dataclass
class AgentDefinition:
    description: str
    prompt: str
    tools: list[str] | None = None
    model: Literal["sonnet", "opus", "haiku", "inherit"] | None = None

PermissionMode = Literal[
    "default",           # 标准权限行为
    "acceptEdits",       # 自动接受文件编辑
    "plan",              # 规划模式 - 不执行
    "bypassPermissions"  # 绕过所有权限检查（谨慎使用）
]

class McpSdkServerConfig(TypedDict):
    type: Literal["sdk"]
    name: str
    instance: Any  # MCP 服务器实例

McpServerConfig = McpStdioServerConfig | McpSSEServerConfig | McpHttpServerConfig | McpSdkServerConfig

class McpStdioServerConfig(TypedDict):
    type: NotRequired[Literal["stdio"]]  # 为向后兼容性可选
    command: str
    args: NotRequired[list[str]]
    env: NotRequired[dict[str, str]]

class McpSSEServerConfig(TypedDict):
    type: Literal["sse"]
    url: str
    headers: NotRequired[dict[str, str]]

class McpHttpServerConfig(TypedDict):
    type: Literal["http"]
    url: str
    headers: NotRequired[dict[str, str]]

Message = UserMessage | AssistantMessage | SystemMessage | ResultMessage

@dataclass
class UserMessage:
    content: str | list[ContentBlock]

@dataclass
class AssistantMessage:
    content: list[ContentBlock]
    model: str

@dataclass
class SystemMessage:
    subtype: str
    data: dict[str, Any]

@dataclass
class ResultMessage:
    subtype: str
    duration_ms: int
    duration_api_ms: int
    is_error: bool
    num_turns: int
    session_id: str
    total_cost_usd: float | None = None
    usage: dict[str, Any] | None = None
    result: str | None = None

ContentBlock = TextBlock | ThinkingBlock | ToolUseBlock | ToolResultBlock

@dataclass
class TextBlock:
    text: str

@dataclass
class ThinkingBlock:
    thinking: str
    signature: str

@dataclass
class ToolUseBlock:
    id: str
    name: str
    input: dict[str, Any]

@dataclass
class ToolResultBlock:
    tool_use_id: str
    content: str | list[dict[str, Any]] | None = None
    is_error: bool | None = None

class ClaudeSDKError(Exception):
    """Claude SDK 的基础错误。"""

class CLINotFoundError(CLIConnectionError):
    def __init__(self, message: str = "未找到 Claude Code", cli_path: str | None = None):
        """
        参数：
            message: 错误消息（默认："未找到 Claude Code"）
            cli_path: 未找到的 CLI 的可选路径
        """

class CLIConnectionError(ClaudeSDKError):
    """连接到 Claude Code 失败。"""

class ProcessError(ClaudeSDKError):
    def __init__(self, message: str, exit_code: int | None = None, stderr: str | None = None):
        self.exit_code = exit_code
        self.stderr = stderr

class CLIJSONDecodeError(ClaudeSDKError):
    def __init__(self, line: str, original_error: Exception):
        """
        参数：
            line: 解析失败的行
            original_error: 原始 JSON 解码异常
        """
        self.line = line
        self.original_error = original_error

HookEvent = Literal[
    "PreToolUse",      # 工具执行前调用
    "PostToolUse",     # 工具执行后调用
    "UserPromptSubmit", # 用户提交提示时调用
    "Stop",            # 停止执行时调用
    "SubagentStop",    # 子代理停止时调用
    "PreCompact"       # 消息压缩前调用
]

HookCallback = Callable[
    [dict[str, Any], str | None, HookContext],
    Awaitable[dict[str, Any]]
]

@dataclass
class HookContext:
    signal: Any | None = None  # 未来：中止信号支持

@dataclass
class HookMatcher:
    matcher: str | None = None        # 要匹配的工具名称或模式（例如，"Bash"、"Write|Edit"）
    hooks: list[HookCallback] = field(default_factory=list)  # 要执行的回调列表

from claude_agent_sdk import query, ClaudeAgentOptions, HookMatcher, HookContext
from typing import Any

async def validate_bash_command(
    input_data: dict[str, Any],
    tool_use_id: str | None,
    context: HookContext
) -> dict[str, Any]:
    """验证并可能阻止危险的 bash 命令。"""
    if input_data['tool_name'] == 'Bash':
        command = input_data['tool_input'].get('command', '')
        if 'rm -rf /' in command:
            return {
                'hookSpecificOutput': {
                    'hookEventName': 'PreToolUse',
                    'permissionDecision': 'deny',
                    'permissionDecisionReason': '危险命令已阻止'
                }
            }
    return {}

async def log_tool_use(
    input_data: dict[str, Any],
    tool_use_id: str | None,
    context: HookContext
) -> dict[str, Any]:
    """记录所有工具使用以进行审计。"""
    print(f"使用的工具：{input_data.get('tool_name')}")
    return {}

options = ClaudeAgentOptions(
    hooks={
        'PreToolUse': [
            HookMatcher(matcher='Bash', hooks=[validate_bash_command]),
            HookMatcher(hooks=[log_tool_use])  # 适用于所有工具
        ],
        'PostToolUse': [
            HookMatcher(hooks=[log_tool_use])
        ]
    }
)

async for message in query(
    prompt="分析这个代码库",
    options=options
):
    print(message)

{
    "description": str,      # 任务的简短（3-5 个词）描述
    "prompt": str,           # 代理要执行的任务
    "subagent_type": str     # 要使用的专门代理类型
}

{
    "result": str,                    # 子代理的最终结果
    "usage": dict | None,             # 令牌使用统计
    "total_cost_usd": float | None,  # 总成本（美元）
    "duration_ms": int | None         # 执行持续时间（毫秒）
}

{
    "command": str,                  # 要执行的命令
    "timeout": int | None,           # 可选超时（毫秒）（最大 600000）
    "description": str | None,       # 清晰、简洁的描述（5-10 个词）
    "run_in_background": bool | None # 设置为 true 以在后台运行
}

{
    "output": str,              # 合并的 stdout 和 stderr 输出
    "exitCode": int,            # 命令的退出代码
    "killed": bool | None,      # 命令是否因超时而被终止
    "shellId": str | None       # 后台进程的 Shell ID
}

{
    "file_path": str,           # 要修改的文件的绝对路径
    "old_string": str,          # 要替换的文本
    "new_string": str,          # 要替换为的文本
    "replace_all": bool | None  # 替换所有出现（默认 False）
}

{
    "message": str,      # 确认消息
    "replacements": int, # 进行的替换次数
    "file_path": str     # 被编辑的文件路径
}

{
    "file_path": str,     # 要修改的文件的绝对路径
    "edits": [            # 编辑操作数组
        {
            "old_string": str,          # 要替换的文本
            "new_string": str,          # 要替换为的文本
            "replace_all": bool | None  # 替换所有出现
        }
    ]
}

{
    "message": str,       # 成功消息
    "edits_applied": int, # 应用的编辑总数
    "file_path": str      # 被编辑的文件路径
}

{
    "file_path": str,       # 要读取的文件的绝对路径
    "offset": int | None,   # 开始读取的行号
    "limit": int | None     # 要读取的行数
}

{
    "content": str,         # 带行号的文件内容
    "total_lines": int,     # 文件中的总行数
    "lines_returned": int   # 实际返回的行数
}

{
    "image": str,       # Base64 编码的图像数据
    "mime_type": str,   # 图像 MIME 类型
    "file_size": int    # 文件大小（字节）
}

{
    "file_path": str,  # 要写入的文件的绝对路径
    "content": str     # 要写入文件的内容
}

{
    "message": str,        # 成功消息
    "bytes_written": int,  # 写入的字节数
    "file_path": str       # 被写入的文件路径
}

{
    "pattern": str,       # 匹配文件的 glob 模式
    "path": str | None    # 要搜索的目录（默认为 cwd）
}

{
    "matches": list[str],  # 匹配的文件路径数组
    "count": int,          # 找到的匹配数
    "search_path": str     # 使用的搜索目录
}

{
    "pattern": str,                    # 正则表达式模式
    "path": str | None,                # 要搜索的文件或目录
    "glob": str | None,                # 过滤文件的 Glob 模式
    "type": str | None,                # 要搜索的文件类型
    "output_mode": str | None,         # "content"、"files_with_matches" 或 "count"
    "-i": bool | None,                 # 不区分大小写搜索
    "-n": bool | None,                 # 显示行号
    "-B": int | None,                  # 每个匹配前显示的行数
    "-A": int | None,                  # 每个匹配后显示的行数
    "-C": int | None,                  # 前后显示的行数
    "head_limit": int | None,          # 将输出限制为前 N 行/条目
    "multiline": bool | None           # 启用多行模式
}

{
    "matches": [
        {
            "file": str,
            "line_number": int | None,
            "line": str,
            "before_context": list[str] | None,
            "after_context": list[str] | None
        }
    ],
    "total_matches": int
}

{
    "files": list[str],  # 包含匹配的文件
    "count": int         # 有匹配的文件数
}

{
    "notebook_path": str,                     # Jupyter 笔记本的绝对路径
    "cell_id": str | None,                    # 要编辑的单元格的 ID
    "new_source": str,                        # 单元格的新源代码
    "cell_type": "code" | "markdown" | None,  # 单元格的类型
    "edit_mode": "replace" | "insert" | "delete" | None  # 编辑操作类型
}

{
    "message": str, # 成功消息
    "edit_type": "replaced" | "inserted" | "deleted",  # 执行的编辑类型
    "cell_id": str | None,                       # 受影响的单元格 ID
    "total_cells": int                           # 编辑后笔记本中的总单元格数
}

{
    "url": str,     # 要获取内容的 URL
    "prompt": str   # 在获取的内容上运行的提示
}

{
    "response": str,           # AI 模型对提示的响应
    "url": str,                # 被获取的 URL
    "final_url": str | None,   # 重定向后的最终 URL
    "status_code": int | None  # HTTP 状态码
}

{
    "query": str,                        # 要使用的搜索查询
    "allowed_domains": list[str] | None, # 仅包含来自这些域的结果
    "blocked_domains": list[str] | None  # 永不包含来自这些域的结果
}

{
    "results": [
        {
            "title": str,
            "url": str,
            "snippet": str,
            "metadata": dict | None
        }
    ],
    "total_results": int,
    "query": str
}

{
    "todos": [
        {
            "content": str, # 任务描述
            "status": "pending" | "in_progress" | "completed",  # 任务状态
            "activeForm": str                            # 描述的活动形式
        }
    ]
}

{
    "message": str,  # 成功消息
    "stats": {
        "total": int,
        "pending": int,
        "in_progress": int,
        "completed": int
    }
}

{
    "bash_id": str,       # 后台 shell 的 ID
    "filter": str | None  # 过滤输出行的可选正则表达式
}

{
    "output": str, # 自上次检查以来的新输出
    "status": "running" | "completed" | "failed",       # 当前 shell 状态
    "exitCode": int | None # 完成时的退出代码
}

{
    "shell_id": str  # 要终止的后台 shell 的 ID
}

{
    "message": str,  # 成功消息
    "shell_id": str  # 被终止的 shell 的 ID
}

{
    "plan": str  # 用户要运行以获得批准的计划
}

{
    "message": str,          # 确认消息
    "approved": bool | None  # 用户是否批准了计划
}

{
    "server": str | None  # 可选的服务器名称以按资源过滤
}

{
    "resources": [
        {
            "uri": str,
            "name": str,
            "description": str | None,
            "mimeType": str | None,
            "server": str
        }
    ],
    "total": int
}

{
    "server": str,  # MCP 服务器名称
    "uri": str      # 要读取的资源 URI
}

{
    "contents": [
        {
            "uri": str,
            "mimeType": str | None,
            "text": str | None,
            "blob": str | None
        }
    ],
    "server": str
}

from claude_agent_sdk import ClaudeSDKClient, ClaudeCodeOptions, AssistantMessage, TextBlock
import asyncio

class ConversationSession:
    """与 Claude 维护单个对话会话。"""
    
    def __init__(self, options: ClaudeCodeOptions = None):
        self.client = ClaudeSDKClient(options)
        self.turn_count = 0
    
    async def start(self):
        await self.client.connect()
        print("开始对话会话。Claude 将记住上下文。")
        print("命令：'exit' 退出，'interrupt' 停止当前任务，'new' 新会话")
        
        while True:
            user_input = input(f"\n[轮次 {self.turn_count + 1}] 你：")
            
            if user_input.lower() == 'exit':
                break
            elif user_input.lower() == 'interrupt':
                await self.client.interrupt()
                print("任务已中断！")
                continue
            elif user_input.lower() == 'new':
                # 断开连接并重新连接以获得新会话
                await self.client.disconnect()
                await self.client.connect()
                self.turn_count = 0
                print("开始新对话会话（先前上下文已清除）")
                continue
            
            # 发送消息 - Claude 记住此会话中的所有先前消息
            await self.client.query(user_input)
            self.turn_count += 1
            
            # 处理响应
            print(f"[轮次 {self.turn_count}] Claude：", end="")
            async for message in self.client.receive_response():
                if isinstance(message, AssistantMessage):
                    for block in message.content:
                        if isinstance(block, TextBlock):
                            print(block.text, end="")
            print()  # 响应后换行
        
        await self.client.disconnect()
        print(f"对话在 {self.turn_count} 轮后结束。")

async def main():
    options = ClaudeCodeOptions(
        allowed_tools=["Read", "Write", "Bash"],
        permission_mode="acceptEdits"
    )
    session = ConversationSession(options)
    await session.start()

# 示例对话：
# 轮次 1 - 你："创建一个名为 hello.py 的文件"
# 轮次 1 - Claude："我将为你创建一个 hello.py 文件..."
# 轮次 2 - 你："那个文件里有什么？"  
# 轮次 2 - Claude："我刚创建的 hello.py 文件包含..."（记住了！）
# 轮次 3 - 你："给它添加一个主函数"
# 轮次 3 - Claude："我将给 hello.py 添加一个主函数..."（知道是哪个文件！）

asyncio.run(main())

from claude_agent_sdk import (
    ClaudeSDKClient,
    ClaudeAgentOptions,
    HookMatcher,
    HookContext
)
import asyncio
from typing import Any

async def pre_tool_logger(
    input_data: dict[str, Any],
    tool_use_id: str | None,
    context: HookContext
) -> dict[str, Any]:
    """在执行前记录所有工具使用。"""
    tool_name = input_data.get('tool_name', 'unknown')
    print(f"[PRE-TOOL] 即将使用：{tool_name}")

    # 你可以在这里修改或阻止工具执行
    if tool_name == "Bash" and "rm -rf" in str(input_data.get('tool_input', {})):
        return {
            'hookSpecificOutput': {
                'hookEventName': 'PreToolUse',
                'permissionDecision': 'deny',
                'permissionDecisionReason': '危险命令已阻止'
            }
        }
    return {}

async def post_tool_logger(
    input_data: dict[str, Any],
    tool_use_id: str | None,
    context: HookContext
) -> dict[str, Any]:
    """工具执行后记录结果。"""
    tool_name = input_data.get('tool_name', 'unknown')
    print(f"[POST-TOOL] 已完成：{tool_name}")
    return {}

async def user_prompt_modifier(
    input_data: dict[str, Any],
    tool_use_id: str | None,
    context: HookContext
) -> dict[str, Any]:
    """为用户提示添加上下文。"""
    original_prompt = input_data.get('prompt', '')

    # 为所有提示添加时间戳
    from datetime import datetime
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")

    return {
        'hookSpecificOutput': {
            'hookEventName': 'UserPromptSubmit',
            'updatedPrompt': f"[{timestamp}] {original_prompt}"
        }
    }

async def main():
    options = ClaudeAgentOptions(
        hooks={
            'PreToolUse': [
                HookMatcher(hooks=[pre_tool_logger]),
                HookMatcher(matcher='Bash', hooks=[pre_tool_logger])
            ],
            'PostToolUse': [
                HookMatcher(hooks=[post_tool_logger])
            ],
            'UserPromptSubmit': [
                HookMatcher(hooks=[user_prompt_modifier])
            ]
        },
        allowed_tools=["Read", "Write", "Bash"]
    )
    
    async with ClaudeSDKClient(options=options) as client:
        await client.query("列出当前目录中的文件")
        
        async for message in client.receive_response():
            # 钩子将自动记录工具使用
            pass

asyncio.run(main())

from claude_agent_sdk import (
    ClaudeSDKClient,
    ClaudeAgentOptions,
    AssistantMessage,
    ToolUseBlock,
    ToolResultBlock,
    TextBlock
)
import asyncio

async def monitor_progress():
    options = ClaudeAgentOptions(
        allowed_tools=["Write", "Bash"],
        permission_mode="acceptEdits"
    )
    
    async with ClaudeSDKClient(options=options) as client:
        await client.query(
            "创建 5 个包含不同排序算法的 Python 文件"
        )
        
        # 实时监控进度
        files_created = []
        async for message in client.receive_messages():
            if isinstance(message, AssistantMessage):
                for block in message.content:
                    if isinstance(block, ToolUseBlock):
                        if block.name == "Write":
                            file_path = block.input.get("file_path", "")
                            print(f"🔨 创建中：{file_path}")
                    elif isinstance(block, ToolResultBlock):
                        print(f"✅ 工具执行完成")
                    elif isinstance(block, TextBlock):
                        print(f"💭 Claude 说：{block.text[:100]}...")
            
            # 检查是否收到最终结果
            if hasattr(message, 'subtype') and message.subtype in ['success', 'error']:
                print(f"\n🎯 任务完成！")
                break

asyncio.run(monitor_progress())

from claude_agent_sdk import query, ClaudeAgentOptions, AssistantMessage, ToolUseBlock
import asyncio

async def create_project():
    options = ClaudeAgentOptions(
        allowed_tools=["Read", "Write", "Bash"],
        permission_mode='acceptEdits',
        cwd="/home/user/project"
    )
    
    async for message in query(
        prompt="创建一个带有 setup.py 的 Python 项目结构",
        options=options
    ):
        if isinstance(message, AssistantMessage):
            for block in message.content:
                if isinstance(block, ToolUseBlock):
                    print(f"使用工具：{block.name}")

asyncio.run(create_project())

from claude_agent_sdk import (
    query,
    CLINotFoundError,
    ProcessError,
    CLIJSONDecodeError
)

try:
    async for message in query(prompt="你好"):
        print(message)
except CLINotFoundError:
    print("请安装 Claude Code：npm install -g @anthropic-ai/claude-code")
except ProcessError as e:
    print(f"进程失败，退出代码：{e.exit_code}")
except CLIJSONDecodeError as e:
    print(f"解析响应失败：{e}")

from claude_agent_sdk import ClaudeSDKClient
import asyncio

async def interactive_session():
    async with ClaudeSDKClient() as client:
        # 发送初始消息
        await client.query("天气怎么样？")
        
        # 处理响应
        async for msg in client.receive_response():
            print(msg)
        
        # 发送后续消息
        await client.query("告诉我更多相关信息")
        
        # 处理后续响应
        async for msg in client.receive_response():
            print(msg)

asyncio.run(interactive_session())

from claude_agent_sdk import (
    ClaudeSDKClient,
    ClaudeAgentOptions,
    tool,
    create_sdk_mcp_server,
    AssistantMessage,
    TextBlock
)
import asyncio
from typing import Any

# 使用 @tool 装饰器定义自定义工具
@tool("calculate", "执行数学计算", {"expression": str})
async def calculate(args: dict[str, Any]) -> dict[str, Any]:
    try:
        result = eval(args["expression"], {"__builtins__": {}})
        return {
            "content": [{
                "type": "text",
                "text": f"结果：{result}"
            }]
        }
    except Exception as e:
        return {
            "content": [{
                "type": "text",
                "text": f"错误：{str(e)}"
            }],
            "is_error": True
        }

@tool("get_time", "获取当前时间", {})
async def get_time(args: dict[str, Any]) -> dict[str, Any]:
    from datetime import datetime
    current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    return {
        "content": [{
            "type": "text",
            "text": f"当前时间：{current_time}"
        }]
    }

async def main():
    # 使用自定义工具创建 SDK MCP 服务器
    my_server = create_sdk_mcp_server(
        name="utilities",
        version="1.0.0",
        tools=[calculate, get_time]
    )

    # 使用服务器配置选项
    options = ClaudeAgentOptions(
        mcp_servers={"utils": my_server},
        allowed_tools=[
            "mcp__utils__calculate",
            "mcp__utils__get_time"
        ]
    )
    
    # 使用 ClaudeSDKClient 进行交互式工具使用
    async with ClaudeSDKClient(options=options) as client:
        await client.query("123 * 456 等于多少？")
        
        # 处理计算响应
        async for message in client.receive_response():
            if isinstance(message, AssistantMessage):
                for block in message.content:
                    if isinstance(block, TextBlock):
                        print(f"计算：{block.text}")
        
        # 跟进时间查询
        await client.query("现在几点了？")
        
        async for message in client.receive_response():
            if isinstance(message, AssistantMessage):
                for block in message.content:
                    if isinstance(block, TextBlock):
                        print(f"时间：{block.text}")

asyncio.run(main())