import { query } from "@anthropic-ai/claude-agent-sdk";

for await (const message of query({
  prompt: "Analyze this codebase",
  options: {
    maxTurns: 10,
    allowedTools: ["Read", "Grep", "Glob"]
  }
})) {
  if (message.type === "result" && message.subtype === "success") {
    console.log(message.result);
  }
}

from claude_agent_sdk import query, ClaudeAgentOptions

async for message in query(
    prompt="Analyze this codebase",
    options=ClaudeAgentOptions(
        max_turns=10,
        allowed_tools=["Read", "Grep", "Glob"]
    )
):
    if message.type == "result":
        print(message.result)

for await (const message of query({
  prompt: "Your task",
  options: { maxTurns: 5, model: "claude-sonnet-4-5" }
})) {
  console.log(message);
}

import { tool, createSdkMcpServer } from "@anthropic-ai/claude-agent-sdk";
import { z } from "zod";

const server = createSdkMcpServer({
  name: "my-tools",
  version: "1.0.0",
  tools: [
    tool("get_data", "Fetch data", { id: z.string() }, async (args) => ({
      content: [{ type: "text", text: `Data for ${args.id}` }]
    }))
  ]
});

// Use with streaming input
async function* messages() {
  yield { type: "user", message: { role: "user", content: "Get data for user-123" } };
}

for await (const msg of query({
  prompt: messages(),
  options: {
    mcpServers: { "my-tools": server },
    allowedTools: ["mcp__my-tools__get_data"]
  }
})) {
  console.log(msg);
}

const result = query({
  prompt: "Review this code",
  options: {
    agents: {
      'reviewer': {
        description: 'Code review specialist',
        prompt: 'You are a code reviewer...',
        tools: ['Read', 'Grep'],
        model: 'sonnet'
      }
    }
  }
});

for await (const message of query({
  prompt: "Extract user info",
  options: {
    outputFormat: {
      type: 'json_schema',
      schema: {
        type: 'object',
        properties: {
          name: { type: 'string' },
          email: { type: 'string' }
        },
        required: ['name', 'email']
      }
    }
  }
})) {
  if (message.structured_output) {
    console.log(message.structured_output);
  }
}

// Get session ID
let sessionId: string;
for await (const msg of query({ prompt: "Hello" })) {
  if (msg.type === 'system' && msg.subtype === 'init') {
    sessionId = msg.session_id;
  }
}

// Resume later
for await (const msg of query({
  prompt: "Continue",
  options: { resume: sessionId }
})) {
  console.log(msg);
}

const result = query({
  prompt: "Make changes",
  options: {
    permissionMode: 'acceptEdits',  // Auto-accept file edits
    canUseTool: async (toolName, input) => {
      if (toolName === "Bash" && input.command?.includes("rm")) {
        return { behavior: "deny", message: "Deletion not allowed" };
      }
      return { behavior: "allow", updatedInput: input };
    }
  }
});