Uso de herramientas

import json
import anthropic

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

TOOLS = [
    {
        "name": "buscar_pedido",
        "description": "Devuelve estado y fecha de un pedido a partir de su id (formato P-123456).",
        "input_schema": {
            "type": "object",
            "properties": {"pedido_id": {"type": "string"}},
            "required": ["pedido_id"],
        },
    },
    {
        "name": "consultar_politica",
        "description": "Devuelve el texto de una política de la tienda: envios o devoluciones.",
        "input_schema": {
            "type": "object",
            "properties": {"tema": {"type": "string", "enum": ["envios", "devoluciones"]}},
            "required": ["tema"],
        },
    },
]

PEDIDOS = {"P-100200": {"estado": "entregado", "fecha_entrega": "2026-10-02"}}
POLITICAS = {"envios": "Envío en 48-72 h laborables.", "devoluciones": "30 días desde la entrega."}

def ejecutar(nombre: str, args: dict) -> str:
    if nombre == "buscar_pedido":
        pedido = PEDIDOS.get(args["pedido_id"])
        if pedido is None:
            raise LookupError(f"No existe el pedido {args['pedido_id']}")
        return json.dumps(pedido)
    if nombre == "consultar_politica":
        return POLITICAS[args["tema"]]
    raise ValueError(f"Herramienta desconocida: {nombre}")

def agente(pregunta: str) -> str:
    mensajes = [{"role": "user", "content": pregunta}]
    for _ in range(MAX_PASOS):
        resp = client.messages.create(model=MODEL, max_tokens=1024, tools=TOOLS, messages=mensajes)
        mensajes.append({"role": "assistant", "content": resp.content})

        if resp.stop_reason != "tool_use":
            if resp.stop_reason == "max_tokens":
                return "[respuesta cortada por max_tokens]"
            return "".join(b.text for b in resp.content if b.type == "text")

        resultados = []
        for bloque in resp.content:
            if bloque.type != "tool_use":
                continue
            try:
                salida, error = ejecutar(bloque.name, bloque.input), False
            except Exception as e:
                salida, error = f"Error: {e}", True
            resultados.append({
                "type": "tool_result",
                "tool_use_id": bloque.id,
                "content": salida,
                "is_error": error,
            })
        mensajes.append({"role": "user", "content": resultados})
    return "[se alcanzó el límite de pasos]"

print(agente("Mi pedido P-100200, ¿todavía puedo devolverlo?"))

import Anthropic from "@anthropic-ai/sdk";

const MODEL = "claude-sonnet-5-5";
const MAX_PASOS = 8;
const client = new Anthropic();

const TOOLS: Anthropic.Tool[] = [
  {
    name: "buscar_pedido",
    description: "Devuelve estado y fecha de un pedido a partir de su id (formato P-123456).",
    input_schema: {
      type: "object",
      properties: { pedido_id: { type: "string" } },
      required: ["pedido_id"],
    },
  },
  {
    name: "consultar_politica",
    description: "Devuelve el texto de una política de la tienda: envios o devoluciones.",
    input_schema: {
      type: "object",
      properties: { tema: { type: "string", enum: ["envios", "devoluciones"] } },
      required: ["tema"],
    },
  },
];

const PEDIDOS: Record<string, object> = { "P-100200": { estado: "entregado", fecha_entrega: "2026-10-02" } };
const POLITICAS: Record<string, string> = {
  envios: "Envío en 48-72 h laborables.",
  devoluciones: "30 días desde la entrega.",
};

function ejecutar(nombre: string, args: any): string {
  if (nombre === "buscar_pedido") {
    const pedido = PEDIDOS[args.pedido_id];
    if (!pedido) throw new Error(`No existe el pedido ${args.pedido_id}`);
    return JSON.stringify(pedido);
  }
  if (nombre === "consultar_politica") return POLITICAS[args.tema];
  throw new Error(`Herramienta desconocida: ${nombre}`);
}

async function agente(pregunta: string): Promise<string> {
  const mensajes: Anthropic.MessageParam[] = [{ role: "user", content: pregunta }];
  for (let paso = 0; paso < MAX_PASOS; paso++) {
    const resp = await client.messages.create({ model: MODEL, max_tokens: 1024, tools: TOOLS, messages: mensajes });
    mensajes.push({ role: "assistant", content: resp.content });

    if (resp.stop_reason !== "tool_use") {
      if (resp.stop_reason === "max_tokens") return "[respuesta cortada por max_tokens]";
      return resp.content.map((b) => (b.type === "text" ? b.text : "")).join("");
    }

    const resultados: Anthropic.ToolResultBlockParam[] = [];
    for (const bloque of resp.content) {
      if (bloque.type !== "tool_use") continue;
      try {
        resultados.push({ type: "tool_result", tool_use_id: bloque.id, content: ejecutar(bloque.name, bloque.input) });
      } catch (e) {
        resultados.push({ type: "tool_result", tool_use_id: bloque.id, content: `Error: ${e}`, is_error: true });
      }
    }
    mensajes.push({ role: "user", content: resultados });
  }
  return "[se alcanzó el límite de pasos]";
}

console.log(await agente("Mi pedido P-100200, ¿todavía puedo devolverlo?"));