Agent SDK 设计模式

# 智能体 SDK 模式 — Python ## 基础智能体 python import anyio from claude_agent_sdk import query, ClaudeAgentOptions, ResultMessage async def main(): async for message in query( prompt="Explain what this repository does", options=ClaudeAgentOptions( cwd="/path/to/project", allowed_tools=["Read", "Glob", "Grep"] ) ): if isinstance(message, ResultMessage): print(message.result) anyio.run(main) --- ## 自定义工具 自定义工具需要 MCP 服务器。使用 `ClaudeSDKClient` 进行完全控制（自定义 SDK MCP 工具需要 `ClaudeSDKClient` — `query()` 仅支持外部 stdio/http MCP 服务器）。 python import anyio from claude_agent_sdk import ( tool, create_sdk_mcp_server, ClaudeSDKClient, ClaudeAgentOptions, AssistantMessage, TextBlock, ) @tool("get_weather", "Get the current weather for a location", {"location": str}) async def get_weather(args): location = args["location"] return {"content": [{"type": "text", "text": f"The weather in {location} is sunny and 72°F."}]} server = create_sdk_mcp_server("weather-tools", tools=[get_weather]) async def main(): options = ClaudeAgentOptions(mcp_servers={"weather": server}) async with ClaudeSDKClient(options=options) as client: await client.query("What's the weather in Paris?") async for message in client.receive_response(): if isinstance(message, AssistantMessage): for block in message.content: if isinstance(block, TextBlock): print(block.text) anyio.run(main) --- ## 钩子 (Hooks) ### 工具使用后钩子 (After Tool Use Hook) 在任何编辑操作后记录文件更改： python import anyio from datetime import datetime from claude_agent_sdk import query, ClaudeAgentOptions, HookMatcher, ResultMessage async def log_file_change(input_data, tool_use_id, context): file_path = input_data.get('tool_input', {}).get('file_path', 'unknown') with open('./audit.log', 'a') as f: f.write(f"{datetime.now()}: modified {file_path}n") return {} async def main(): async for message in query( prompt="Refactor utils.py to improve readability", options=ClaudeAgentOptions( allowed_tools=["Read", "Edit", "Write"], permission_mode="acceptEdits", hooks={ "PostToolUse": [HookMatcher(matcher="Edit|Write", hooks=[log_file_change])] } )