Referencia del SDK del Agente - 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="You are an expert Python developer",
        permission_mode='acceptEdits',
        cwd="/home/user/project"
    )

    async for message in query(
        prompt="Create a Python web server",
        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", "Greet a user", {"name": str})
async def greet(args: dict[str, Any]) -> dict[str, Any]:
    return {
        "content": [{
            "type": "text",
            "text": f"Hello, {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", "Add two numbers", {"a": float, "b": float})
async def add(args):
    return {
        "content": [{
            "type": "text",
            "text": f"Sum: {args['a'] + args['b']}"
        }]
    }

@tool("multiply", "Multiply two numbers", {"a": float, "b": float})
async def multiply(args):
    return {
        "content": [{
            "type": "text",
            "text": f"Product: {args['a'] * args['b']}"
        }]
    }

calculator = create_sdk_mcp_server(
    name="calculator",
    version="2.0.0",
    tools=[add, multiply]  # Pass decorated functions
)

# Use with 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 rewind_files(self, user_message_uuid: str) -> None
    async def disconnect(self) -> None

async with ClaudeSDKClient() as client:
    await client.query("Hello 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:
        # First question
        await client.query("What's the capital of France?")

        # Process response
        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}")

        # Follow-up question - Claude remembers the previous context
        await client.query("What's the population of that city?")

        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}")

        # Another follow-up - still in the same conversation
        await client.query("What are some famous landmarks there?")

        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():
    """Generate messages dynamically."""
    yield {"type": "text", "text": "Analyze the following data:"}
    await asyncio.sleep(0.5)
    yield {"type": "text", "text": "Temperature: 25°C"}
    await asyncio.sleep(0.5)
    yield {"type": "text", "text": "Humidity: 60%"}
    await asyncio.sleep(0.5)
    yield {"type": "text", "text": "What patterns do you see?"}

async def main():
    async with ClaudeSDKClient() as client:
        # Stream input to Claude
        await client.query(message_stream())

        # Process response
        async for message in client.receive_response():
            print(message)

        # Follow-up in same session
        await client.query("Should we be concerned about these readings?")

        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:
        # Start a long-running task
        await client.query("Count from 1 to 100 slowly")

        # Let it run for a bit
        await asyncio.sleep(2)

        # Interrupt the task
        await client.interrupt()
        print("Task interrupted!")

        # Send a new command
        await client.query("Just say hello instead")

        async for message in client.receive_response():
            # Process the new 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
):
    """Custom logic for tool permissions."""

    # Block writes to system directories
    if tool_name == "Write" and input_data.get("file_path", "").startswith("/system/"):
        return {
            "behavior": "deny",
            "message": "System directory write not allowed",
            "interrupt": True
        }

    # Redirect sensitive file operations
    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}
        }

    # Allow everything else
    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("Update the system config file")

        async for message in client.receive_response():
            # Will use sandbox path instead
            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)
    system_prompt: str | SystemPromptPreset | 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
    max_turns: int | None = None
    disallowed_tools: list[str] = field(default_factory=list)
    model: str | None = None
    output_format: OutputFormat | 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)
    max_buffer_size: int | None = None
    debug_stderr: Any = sys.stderr  # Deprecated
    stderr: Callable[[str], None] | None = None
    can_use_tool: CanUseTool | None = None
    hooks: dict[HookEvent, list[HookMatcher]] | None = None
    user: str | None = None
    include_partial_messages: bool = False
    fork_session: bool = False
    agents: dict[str, AgentDefinition] | None = None
    setting_sources: list[SettingSource] | None = None

class OutputFormat(TypedDict):
    type: Literal["json_schema"]
    schema: dict[str, Any]

class SystemPromptPreset(TypedDict):
    type: Literal["preset"]
    preset: Literal["claude_code"]
    append: NotRequired[str]

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

# Load all settings like SDK v0.0.x did
from claude_agent_sdk import query, ClaudeAgentOptions

async for message in query(
    prompt="Analyze this code",
    options=ClaudeAgentOptions(
        setting_sources=["user", "project", "local"]  # Load all settings
    )
):
    print(message)

# Load only project settings, ignore user and local
async for message in query(
    prompt="Run CI checks",
    options=ClaudeAgentOptions(
        setting_sources=["project"]  # Only .claude/settings.json
    )
):
    print(message)

# Ensure consistent behavior in CI by excluding local settings
async for message in query(
    prompt="Run tests",
    options=ClaudeAgentOptions(
        setting_sources=["project"],  # Only team-shared settings
        permission_mode="bypassPermissions"
    )
):
    print(message)

# Define everything programmatically (default behavior)
# No filesystem dependencies - setting_sources defaults to None
async for message in query(
    prompt="Review this PR",
    options=ClaudeAgentOptions(
        # setting_sources=None is the default, no need to specify
        agents={ /* ... */ },
        mcp_servers={ /* ... */ },
        allowed_tools=["Read", "Grep", "Glob"]
    )
):
    print(message)

# Load project settings to include CLAUDE.md files
async for message in query(
    prompt="Add a new feature following project conventions",
    options=ClaudeAgentOptions(
        system_prompt={
            "type": "preset",
            "preset": "claude_code"  # Use Claude Code's system prompt
        },
        setting_sources=["project"],  # Required to load CLAUDE.md from project
        allowed_tools=["Read", "Write", "Edit"]
    )
):
    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",           # Standard permission behavior
    "acceptEdits",       # Auto-accept file edits
    "plan",              # Planning mode - no execution
    "bypassPermissions"  # Bypass all permission checks (use with caution)
]

class McpSdkServerConfig(TypedDict):
    type: Literal["sdk"]
    name: str
    instance: Any  # MCP Server instance

McpServerConfig = McpStdioServerConfig | McpSSEServerConfig | McpHttpServerConfig | McpSdkServerConfig

class McpStdioServerConfig(TypedDict):
    type: NotRequired[Literal["stdio"]]  # Optional for backwards compatibility
    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]]

class SdkPluginConfig(TypedDict):
    type: Literal["local"]
    path: str

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

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):
    """Error base para Claude SDK."""

class CLINotFoundError(CLIConnectionError):
    def __init__(self, message: str = "Claude Code not found", cli_path: str | None = None):
        """
        Args:
            message: Mensaje de error (predeterminado: "Claude Code not found")
            cli_path: Ruta opcional a la CLI que no se encontró
        """

class CLIConnectionError(ClaudeSDKError):
    """Error al conectar con 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):
        """
        Args:
            line: La línea que no se pudo analizar
            original_error: La excepción original de decodificación JSON
        """
        self.line = line
        self.original_error = original_error

HookEvent = Literal[
    "PreToolUse",      # Se llama antes de la ejecución de la herramienta
    "PostToolUse",     # Se llama después de la ejecución de la herramienta
    "UserPromptSubmit", # Se llama cuando el usuario envía un mensaje
    "Stop",            # Se llama cuando se detiene la ejecución
    "SubagentStop",    # Se llama cuando un subaagente se detiene
    "PreCompact"       # Se llama antes de la compactación de mensajes
]

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

@dataclass
class HookContext:
    signal: Any | None = None  # Futuro: soporte de señal de aborto

@dataclass
class HookMatcher:
    matcher: str | None = None        # Nombre de herramienta o patrón a coincidir (p. ej., "Bash", "Write|Edit")
    hooks: list[HookCallback] = field(default_factory=list)  # Lista de devoluciones de llamada a ejecutar
    timeout: float | None = None        # Tiempo de espera en segundos para todos los hooks en este matcher (predeterminado: 60)

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]:
    """Validar y potencialmente bloquear comandos bash peligrosos."""
    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': 'Comando peligroso bloqueado'
                }
            }
    return {}

async def log_tool_use(
    input_data: dict[str, Any],
    tool_use_id: str | None,
    context: HookContext
) -> dict[str, Any]:
    """Registrar todo el uso de herramientas para auditoría."""
    print(f"Herramienta utilizada: {input_data.get('tool_name')}")
    return {}

options = ClaudeAgentOptions(
    hooks={
        'PreToolUse': [
            HookMatcher(matcher='Bash', hooks=[validate_bash_command], timeout=120),  # 2 min para validación
            HookMatcher(hooks=[log_tool_use])  # Se aplica a todas las herramientas (tiempo de espera predeterminado de 60s)
        ],
        'PostToolUse': [
            HookMatcher(hooks=[log_tool_use])
        ]
    }
)

async for message in query(
    prompt="Analizar este repositorio de código",
    options=options
):
    print(message)

{
    "description": str,      # Una descripción breve (3-5 palabras) de la tarea
    "prompt": str,           # La tarea para que el agente realice
    "subagent_type": str     # El tipo de agente especializado a utilizar
}

{
    "result": str,                    # Resultado final del subaagente
    "usage": dict | None,             # Estadísticas de uso de tokens
    "total_cost_usd": float | None,  # Costo total en USD
    "duration_ms": int | None         # Duración de ejecución en milisegundos
}

{
    "command": str,                  # El comando a ejecutar
    "timeout": int | None,           # Tiempo de espera opcional en milisegundos (máximo 600000)
    "description": str | None,       # Descripción clara y concisa (5-10 palabras)
    "run_in_background": bool | None # Establecer en true para ejecutar en segundo plano
}

{
    "output": str,              # Salida combinada de stdout y stderr
    "exitCode": int,            # Código de salida del comando
    "killed": bool | None,      # Si el comando fue terminado debido a tiempo de espera
    "shellId": str | None       # ID de shell para procesos en segundo plano
}

{
    "file_path": str,           # La ruta absoluta del archivo a modificar
    "old_string": str,          # El texto a reemplazar
    "new_string": str,          # El texto para reemplazarlo
    "replace_all": bool | None  # Reemplazar todas las ocurrencias (predeterminado False)
}

{
    "message": str,      # Mensaje de confirmación
    "replacements": int, # Número de reemplazos realizados
    "file_path": str     # Ruta del archivo que fue editado
}

{
    "file_path": str,       # La ruta absoluta del archivo a leer
    "offset": int | None,   # El número de línea desde donde comenzar a leer
    "limit": int | None     # El número de líneas a leer
}

{
    "content": str,         # Contenido del archivo con números de línea
    "total_lines": int,     # Número total de líneas en el archivo
    "lines_returned": int   # Líneas realmente devueltas
}

{
    "image": str,       # Datos de imagen codificados en Base64
    "mime_type": str,   # Tipo MIME de imagen
    "file_size": int    # Tamaño del archivo en bytes
}

{
    "file_path": str,  # La ruta absoluta del archivo a escribir
    "content": str     # El contenido a escribir en el archivo
}

{
    "message": str,        # Mensaje de éxito
    "bytes_written": int,  # Número de bytes escritos
    "file_path": str       # Ruta del archivo que fue escrito
}

{
    "pattern": str,       # El patrón glob para coincidir archivos
    "path": str | None    # El directorio a buscar (predeterminado al cwd)
}

{
    "matches": list[str],  # Matriz de rutas de archivos coincidentes
    "count": int,          # Número de coincidencias encontradas
    "search_path": str     # Directorio de búsqueda utilizado
}

{
    "pattern": str,                    # El patrón de expresión regular
    "path": str | None,                # Archivo o directorio a buscar
    "glob": str | None,                # Patrón glob para filtrar archivos
    "type": str | None,                # Tipo de archivo a buscar
    "output_mode": str | None,         # "content", "files_with_matches", o "count"
    "-i": bool | None,                 # Búsqueda sin distinción de mayúsculas/minúsculas
    "-n": bool | None,                 # Mostrar números de línea
    "-B": int | None,                  # Líneas a mostrar antes de cada coincidencia
    "-A": int | None,                  # Líneas a mostrar después de cada coincidencia
    "-C": int | None,                  # Líneas a mostrar antes y después
    "head_limit": int | None,          # Limitar salida a las primeras N líneas/entradas
    "multiline": bool | None           # Habilitar modo multilínea
}

{
    "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],  # Archivos que contienen coincidencias
    "count": int         # Número de archivos con coincidencias
}

{
    "notebook_path": str,                     # Ruta absoluta al cuaderno Jupyter
    "cell_id": str | None,                    # El ID de la celda a editar
    "new_source": str,                        # La nueva fuente para la celda
    "cell_type": "code" | "markdown" | None,  # El tipo de celda
    "edit_mode": "replace" | "insert" | "delete" | None  # Tipo de operación de edición
}

{
    "message": str,                              # Mensaje de éxito
    "edit_type": "replaced" | "inserted" | "deleted",  # Tipo de edición realizada
    "cell_id": str | None,                       # ID de celda que fue afectado
    "total_cells": int                           # Total de celdas en el cuaderno después de la edición
}

{
    "url": str,     # La URL de la que obtener contenido
    "prompt": str   # El mensaje a ejecutar en el contenido obtenido
}

{
    "response": str,           # Respuesta del modelo de IA al mensaje
    "url": str,                # URL que fue obtenida
    "final_url": str | None,   # URL final después de redirecciones
    "status_code": int | None  # Código de estado HTTP
}

{
    "query": str,                        # La consulta de búsqueda a utilizar
    "allowed_domains": list[str] | None, # Solo incluir resultados de estos dominios
    "blocked_domains": list[str] | None  # Nunca incluir resultados de estos dominios
}

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

{
    "todos": [
        {
            "content": str,                              # La descripción de la tarea
            "status": "pending" | "in_progress" | "completed",  # Estado de la tarea
            "activeForm": str                            # Forma activa de la descripción
        }
    ]
}

{
    "message": str,  # Mensaje de éxito
    "stats": {
        "total": int,
        "pending": int,
        "in_progress": int,
        "completed": int
    }
}

{
    "bash_id": str,       # El ID del shell en segundo plano
    "filter": str | None  # Expresión regular opcional para filtrar líneas de salida
}

{
    "output": str,                                      # Nueva salida desde la última verificación
    "status": "running" | "completed" | "failed",       # Estado actual del shell
    "exitCode": int | None                              # Código de salida cuando se completa
}

{
    "shell_id": str  # El ID del shell en segundo plano a terminar
}

{
    "message": str,  # Mensaje de éxito
    "shell_id": str  # ID del shell terminado
}

{
    "plan": str  # El plan a ejecutar por el usuario para aprobación
}

{
    "message": str,          # Mensaje de confirmación
    "approved": bool | None  # Si el usuario aprobó el plan
}

{
    "server": str | None  # Nombre de servidor opcional para filtrar recursos por
}

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

{
    "server": str,  # El nombre del servidor MCP
    "uri": str      # El URI del recurso a leer
}

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

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

class ConversationSession:
    """Mantiene una única sesión de conversación con Claude."""

    def __init__(self, options: ClaudeAgentOptions = None):
        self.client = ClaudeSDKClient(options)
        self.turn_count = 0

    async def start(self):
        await self.client.connect()
        print("Iniciando sesión de conversación. Claude recordará el contexto.")
        print("Comandos: 'exit' para salir, 'interrupt' para detener la tarea actual, 'new' para nueva sesión")

        while True:
            user_input = input(f"\n[Turno {self.turn_count + 1}] Tú: ")

            if user_input.lower() == 'exit':
                break
            elif user_input.lower() == 'interrupt':
                await self.client.interrupt()
                print("¡Tarea interrumpida!")
                continue
            elif user_input.lower() == 'new':
                # Desconectar y reconectar para una sesión nueva
                await self.client.disconnect()
                await self.client.connect()
                self.turn_count = 0
                print("Sesión de conversación nueva iniciada (contexto anterior borrado)")
                continue

            # Enviar mensaje - Claude recuerda todos los mensajes anteriores en esta sesión
            await self.client.query(user_input)
            self.turn_count += 1

            # Procesar respuesta
            print(f"[Turno {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()  # Nueva línea después de la respuesta

        await self.client.disconnect()
        print(f"Conversación finalizada después de {self.turn_count} turnos.")

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

# Ejemplo de conversación:
# Turno 1 - Tú: "Crear un archivo llamado hello.py"
# Turno 1 - Claude: "Crearé un archivo hello.py para ti..."
# Turno 2 - Tú: "¿Qué hay en ese archivo?"
# Turno 2 - Claude: "El archivo hello.py que acabo de crear contiene..." (¡recuerda!)
# Turno 3 - Tú: "Agregar una función main a él"
# Turno 3 - Claude: "Agregaré una función main a hello.py..." (¡sabe qué archivo!)

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]:
    """Registrar todo el uso de herramientas antes de la ejecución."""
    tool_name = input_data.get('tool_name', 'unknown')
    print(f"[PRE-TOOL] A punto de usar: {tool_name}")

    # Puedes modificar o bloquear la ejecución de la herramienta aquí
    if tool_name == "Bash" and "rm -rf" in str(input_data.get('tool_input', {})):
        return {
            'hookSpecificOutput': {
                'hookEventName': 'PreToolUse',
                'permissionDecision': 'deny',
                'permissionDecisionReason': 'Comando peligroso bloqueado'
            }
        }
    return {}

async def post_tool_logger(
    input_data: dict[str, Any],
    tool_use_id: str | None,
    context: HookContext
) -> dict[str, Any]:
    """Registrar resultados después de la ejecución de la herramienta."""
    tool_name = input_data.get('tool_name', 'unknown')
    print(f"[POST-TOOL] Completado: {tool_name}")
    return {}

async def user_prompt_modifier(
    input_data: dict[str, Any],
    tool_use_id: str | None,
    context: HookContext
) -> dict[str, Any]:
    """Agregar contexto a los mensajes del usuario."""
    original_prompt = input_data.get('prompt', '')

    # Agregar marca de tiempo a todos los mensajes
    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("Listar archivos en el directorio actual")

        async for message in client.receive_response():
            # Los hooks registrarán automáticamente el uso de herramientas
            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(
            "Crear 5 archivos Python con diferentes algoritmos de ordenamiento"
        )

        # Monitorear progreso en tiempo real
        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"🔨 Creando: {file_path}")
                    elif isinstance(block, ToolResultBlock):
                        print(f"✅ Ejecución de herramienta completada")
                    elif isinstance(block, TextBlock):
                        print(f"💭 Claude dice: {block.text[:100]}...")

            # Verificar si hemos recibido el resultado final
            if hasattr(message, 'subtype') and message.subtype in ['success', 'error']:
                print(f"\n🎯 ¡Tarea completada!")
                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="Crear una estructura de proyecto Python con setup.py",
        options=options
    ):
        if isinstance(message, AssistantMessage):
            for block in message.content:
                if isinstance(block, ToolUseBlock):
                    print(f"Usando herramienta: {block.name}")

asyncio.run(create_project())

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

try:
    async for message in query(prompt="Hola"):
        print(message)
except CLINotFoundError:
    print("Por favor instala Claude Code: npm install -g @anthropic-ai/claude-code")
except ProcessError as e:
    print(f"El proceso falló con código de salida: {e.exit_code}")
except CLIJSONDecodeError as e:
    print(f"Error al analizar la respuesta: {e}")

from claude_agent_sdk import ClaudeSDKClient
import asyncio

async def interactive_session():
    async with ClaudeSDKClient() as client:
        # Enviar mensaje inicial
        await client.query("¿Cómo está el clima?")

        # Procesar respuestas
        async for msg in client.receive_response():
            print(msg)

        # Enviar seguimiento
        await client.query("Cuéntame más sobre eso")

        # Procesar respuesta de seguimiento
        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

# Definir herramientas personalizadas con decorador @tool
@tool("calculate", "Realizar cálculos matemáticos", {"expression": str})
async def calculate(args: dict[str, Any]) -> dict[str, Any]:
    try:
        result = eval(args["expression"], {"__builtins__": {}})
        return {
            "content": [{
                "type": "text",
                "text": f"Resultado: {result}"
            }]
        }
    except Exception as e:
        return {
            "content": [{
                "type": "text",
                "text": f"Error: {str(e)}"
            }],
            "is_error": True
        }

@tool("get_time", "Obtener la hora actual", {})
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"Hora actual: {current_time}"
        }]
    }

async def main():
    # Crear servidor SDK MCP con herramientas personalizadas
    my_server = create_sdk_mcp_server(
        name="utilities",
        version="1.0.0",
        tools=[calculate, get_time]
    )

    # Configurar opciones con el servidor
    options = ClaudeAgentOptions(
        mcp_servers={"utils": my_server},
        allowed_tools=[
            "mcp__utils__calculate",
            "mcp__utils__get_time"
        ]
    )

    # Usar ClaudeSDKClient para uso interactivo de herramientas
    async with ClaudeSDKClient(options=options) as client:
        await client.query("¿Cuánto es 123 * 456?")

        # Procesar respuesta de cálculo
        async for message in client.receive_response():
            if isinstance(message, AssistantMessage):
                for block in message.content:
                    if isinstance(block, TextBlock):
                        print(f"Cálculo: {block.text}")

        # Seguimiento con consulta de hora
        await client.query("¿Qué hora es ahora?")

        async for message in client.receive_response():
            if isinstance(message, AssistantMessage):
                for block in message.content:
                    if isinstance(block, TextBlock):
                        print(f"Hora: {block.text}")

asyncio.run(main())

class SandboxSettings(TypedDict, total=False):
    enabled: bool
    autoAllowBashIfSandboxed: bool
    excludedCommands: list[str]
    allowUnsandboxedCommands: bool
    network: SandboxNetworkConfig
    ignoreViolations: SandboxIgnoreViolations
    enableWeakerNestedSandbox: bool

from claude_agent_sdk import query, ClaudeAgentOptions, SandboxSettings

sandbox_settings: SandboxSettings = {
    "enabled": True,
    "autoAllowBashIfSandboxed": True,
    "excludedCommands": ["docker"],
    "network": {
        "allowLocalBinding": True,
        "allowUnixSockets": ["/var/run/docker.sock"]
    }
}

async for message in query(
    prompt="Construir y probar mi proyecto",
    options=ClaudeAgentOptions(sandbox=sandbox_settings)
):
    print(message)

class SandboxNetworkConfig(TypedDict, total=False):
    allowLocalBinding: bool
    allowUnixSockets: list[str]
    allowAllUnixSockets: bool
    httpProxyPort: int
    socksProxyPort: int

class SandboxIgnoreViolations(TypedDict, total=False):
    file: list[str]
    network: list[str]

from claude_agent_sdk import query, ClaudeAgentOptions

async def can_use_tool(tool: str, input: dict) -> bool:
    # Verificar si el modelo está solicitando omitir el sandbox
    if tool == "Bash" and input.get("dangerouslyDisableSandbox"):
        # El modelo quiere ejecutar este comando fuera del sandbox
        print(f"Comando sin sandbox solicitado: {input.get('command')}")

        # Devolver True para permitir, False para denegar
        return is_command_authorized(input.get("command"))
    return True

async def main():
    async for message in query(
        prompt="Desplegar mi aplicación",
        options=ClaudeAgentOptions(
            sandbox={
                "enabled": True,
                "allowUnsandboxedCommands": True  # El modelo puede solicitar ejecución sin sandbox
            },
            permission_mode="default",
            can_use_tool=can_use_tool
        )
    ):
        print(message)