Aprobación humana antes de actuar

import json

import anthropic

MODEL = "claude-sonnet-5-5"
client = anthropic.Anthropic()
MAX_PASOS = 10

PEDIDOS = {"P-1001": {"cliente": "Ana López", "importe": 120.0, "estado": "entregado", "reembolsado": 0.0}}

HERRAMIENTAS = [
    {
        "name": "buscar_pedido",
        "description": "Devuelve cliente, importe, estado e importe ya reembolsado de un pedido.",
        "input_schema": {
            "type": "object",
            "properties": {"pedido_id": {"type": "string", "description": "Formato P-NNNN"}},
            "required": ["pedido_id"],
        },
    },
    {
        "name": "reembolsar",
        "description": "Reembolsa un importe de un pedido. Requiere aprobación humana. "
                       "Busca antes el pedido para conocer el importe máximo.",
        "input_schema": {
            "type": "object",
            "properties": {
                "pedido_id": {"type": "string"},
                "importe": {"type": "number", "description": "En euros"},
                "motivo": {"type": "string"},
            },
            "required": ["pedido_id", "importe", "motivo"],
        },
    },
]

# la lista la decide tu código, nunca el modelo
REQUIEREN_APROBACION = {"reembolsar"}


def buscar_pedido(pedido_id: str) -> dict:
    if pedido_id not in PEDIDOS:
        raise ValueError(f"No existe el pedido {pedido_id}")
    return PEDIDOS[pedido_id]


def reembolsar(pedido_id: str, importe: float, motivo: str) -> dict:
    pedido = buscar_pedido(pedido_id)
    disponible = pedido["importe"] - pedido["reembolsado"]
    if not 0 < importe <= disponible:  # el límite vive en código, no en el prompt
        raise ValueError(f"Importe no válido: máximo reembolsable {disponible:.2f} €")
    pedido["reembolsado"] += importe
    return {"ok": True, "pedido_id": pedido_id, "reembolsado": importe}


FUNCIONES = {"buscar_pedido": buscar_pedido, "reembolsar": reembolsar}


def pedir_aprobacion(nombre: str, args: dict, razon: str) -> tuple[bool, str]:
    print("\n--- APROBACIÓN NECESARIA ---")
    print(f"Acción: {nombre}")
    print(json.dumps(args, ensure_ascii=False, indent=2))
    if razon:
        print(f"El agente dice: {razon}")
    respuesta = input("¿Aprobar? [s/N] ").strip().lower()
    if respuesta == "s":
        return True, ""
    return False, input("Motivo del rechazo: ").strip() or "Rechazado sin motivo"


def ejecutar(bloque, razon: str) -> dict:
    """Devuelve el tool_result de un bloque tool_use, pasando por aprobación si hace falta."""
    resultado = {"type": "tool_result", "tool_use_id": bloque.id}
    if bloque.name in REQUIEREN_APROBACION:
        aprobado, motivo = pedir_aprobacion(bloque.name, bloque.input, razon)
        if not aprobado:
            return {**resultado, "is_error": True,
                    "content": f"Una persona ha rechazado la acción. Motivo: {motivo}. No la repitas sin cambios."}
    try:
        salida = FUNCIONES[bloque.name](**bloque.input)
        return {**resultado, "content": json.dumps(salida, ensure_ascii=False)}
    except Exception as e:
        return {**resultado, "is_error": True, "content": str(e)}


def agente(peticion: str) -> str:
    mensajes = [{"role": "user", "content": peticion}]
    for _ in range(MAX_PASOS):
        resp = client.messages.create(
            model=MODEL,
            max_tokens=1024,
            system="Eres un agente de soporte. Usa las herramientas para resolver la petición.",
            tools=HERRAMIENTAS,
            messages=mensajes,
        )
        mensajes.append({"role": "assistant", "content": resp.content})

        if resp.stop_reason == "end_turn":
            return "".join(b.text for b in resp.content if b.type == "text")
        if resp.stop_reason != "tool_use":
            return f"[parada inesperada: {resp.stop_reason}]"

        razon = " ".join(b.text for b in resp.content if b.type == "text")
        # un tool_result por cada tool_use, todos en el mismo mensaje
        resultados = [ejecutar(b, razon) for b in resp.content if b.type == "tool_use"]
        mensajes.append({"role": "user", "content": resultados})
    return "[límite de pasos alcanzado]"


if __name__ == "__main__":
    print(agente("El pedido P-1001 llegó roto. Quiero que me devolváis el dinero."))

# salida de ejemplo
--- APROBACIÓN NECESARIA ---
Acción: reembolsar
{
  "pedido_id": "P-1001",
  "importe": 120.0,
  "motivo": "Producto roto"
}
El agente dice: El pedido llegó roto, así que propongo el reembolso completo.
¿Aprobar? [s/N] n
Motivo del rechazo: Ofrece primero un reembolso parcial de 50 €
...
¿Aprobar? [s/N] s
He solicitado un reembolso de 50 € para el pedido P-1001.