Un agente con varias herramientas

ver_pedidos: Obtiene pedidos.

ver_pedidos: Lista los pedidos de un cliente ya identificado, del más reciente al más
antiguo, con id, estado (pendiente, enviado, entregado, devuelto), fecha y total.
Úsala cuando necesites el estado o el historial de compras. Requiere el id de cliente
que devuelve buscar_cliente; no acepta emails. Devuelve como máximo 10 pedidos.

TOOLS = [
    {
        "name": "buscar_cliente",
        "description": (
            "Busca un cliente por email o por teléfono y devuelve su id, nombre y fecha de alta. "
            "Úsala siempre antes de ver_pedidos o abrir_incidencia. Si no hay coincidencia, devuelve un error."
        ),
        "input_schema": {
            "type": "object",
            "properties": {
                "email": {"type": "string", "description": "Email exacto, por ejemplo ana@example.com"},
                "telefono": {"type": "string", "description": "Teléfono con prefijo, por ejemplo +34600111222"},
            },
        },
    },
    {
        "name": "ver_pedidos",
        "description": (
            "Lista los pedidos de un cliente ya identificado, del más reciente al más antiguo, con id, "
            "estado, fecha y total. Requiere el id de cliente de buscar_cliente; no acepta emails. "
            "Devuelve como máximo 10 pedidos."
        ),
        "input_schema": {
            "type": "object",
            "properties": {
                "cliente_id": {"type": "string", "description": "Id con formato C-1234"},
                "estado": {"type": "string", "enum": ["pendiente", "enviado", "entregado", "devuelto"],
                           "description": "Opcional: filtra por estado"},
            },
            "required": ["cliente_id"],
        },
    },
    {
        "name": "buscar_ayuda",
        "description": (
            "Busca en el centro de ayuda y devuelve hasta 3 fragmentos con su título. Úsala para "
            "políticas generales (devoluciones, plazos de envío, pagos). No contiene datos de clientes ni pedidos."
        ),
        "input_schema": {
            "type": "object",
            "properties": {"consulta": {"type": "string", "description": "Pregunta en lenguaje natural"}},
            "required": ["consulta"],
        },
    },
    {
        "name": "abrir_incidencia",
        "description": (
            "Abre una incidencia para el equipo humano. Úsala solo si el problema no se resuelve con "
            "información (pedido perdido, cobro duplicado, producto dañado). Requiere aprobación del usuario."
        ),
        "input_schema": {
            "type": "object",
            "properties": {
                "cliente_id": {"type": "string"},
                "pedido_id": {"type": "string", "description": "Id con formato P-1234"},
                "tipo": {"type": "string", "enum": ["perdido", "cobro_duplicado", "danado", "otro"]},
                "resumen": {"type": "string", "description": "Una o dos frases para el equipo"},
            },
            "required": ["cliente_id", "tipo", "resumen"],
        },
    },
]

import json

def ver_pedidos(cliente_id: str, estado: str | None = None) -> str:
    if not cliente_id.startswith("C-"):
        raise ValueError(f"cliente_id debe tener formato C-1234; has enviado {cliente_id!r}. "
                         "Usa primero buscar_cliente.")
    pedidos = [p for p in BD_PEDIDOS.get(cliente_id, []) if estado in (None, p["estado"])]
    visibles = [{k: p[k] for k in ("id", "estado", "fecha", "total")} for p in pedidos[:10]]
    nota = f"Mostrando {len(visibles)} de {len(pedidos)} pedidos."
    return json.dumps({"pedidos": visibles, "nota": nota}, ensure_ascii=False)

DESPACHO = {
    "buscar_cliente": buscar_cliente,
    "ver_pedidos": ver_pedidos,
    "buscar_ayuda": buscar_ayuda,
    "abrir_incidencia": abrir_incidencia,
}

def ejecutar(nombre: str, args: dict) -> str:
    funcion = DESPACHO.get(nombre)
    if funcion is None:
        raise ValueError(f"No existe la herramienta {nombre}. Disponibles: {', '.join(DESPACHO)}")
    return funcion(**args)

CASOS = [
    ("Soy ana@example.com, ¿dónde está mi pedido?", "buscar_cliente"),
    ("¿Cuántos días tengo para devolver algo?", "buscar_ayuda"),
    ("Mi teléfono es +34600111222 y me habéis cobrado dos veces", "buscar_cliente"),
    ("¿Aceptáis pago con transferencia?", "buscar_ayuda"),
]

aciertos = 0
for consulta, esperada in CASOS:
    resp = client.messages.create(model=MODEL, max_tokens=1024, system=SYSTEM, tools=TOOLS,
                                  messages=[{"role": "user", "content": consulta}])
    usadas = [b.name for b in resp.content if b.type == "tool_use"]
    ok = esperada in usadas
    aciertos += ok
    print(f"{'OK ' if ok else 'MAL'}  {consulta!r} → {usadas or resp.stop_reason}")
print(f"{aciertos}/{len(CASOS)}")