Tool Calling & Agents

import { generateText, tool } from "ai";
import { z } from "zod";

const result = await generateText({
  model: "anthropic/claude-sonnet-4.6",
  tools: {
    comparerOffres: tool({
      description: "Compare les offres d'electricite pour une adresse et une consommation donnees",
      inputSchema: z.object({
        codePostal: z.string().describe("Code postal du logement"),
        consommationKwh: z.number().describe("Consommation annuelle en kWh"),
        type: z.enum(["electricite", "gaz"]).describe("Type d'energie"),
      }),
      execute: async ({ codePostal, consommationKwh, type }) => {
        // En prod, ca appelle une vraie API
        return {
          offres: [
            { fournisseur: "TotalEnergies", prix: 0.19, abonnement: 12, total: consommationKwh * 0.19 + 144 },
            { fournisseur: "Engie", prix: 0.21, abonnement: 10, total: consommationKwh * 0.21 + 120 },
            { fournisseur: "EDF", prix: 0.22, abonnement: 11, total: consommationKwh * 0.22 + 132 },
          ]
        };
      },
    }),
  },
  prompt: "Je consomme 6000 kWh par an et j'habite a Lyon (69000). Quelle est la meilleure offre d'electricite ?",
});

// Le LLM va :
// 1. Voir qu'il a un outil "comparerOffres" qui correspond au besoin
// 2. L'appeler avec { codePostal: "69000", consommationKwh: 6000, type: "electricite" }
// 3. Recevoir les 3 offres
// 4. Presenter la meilleure au client

Client: "J'habite au 12 rue de la Paix a Paris, je veux changer de fournisseur"

── Step 1 (Reflexion) ──
Agent: Je dois d'abord trouver le numero PDL de cette adresse.

── Step 1 (Action) ──
→ Tool call: chercherPDL({ adresse: "12 rue de la Paix, 75002 Paris" })

── Step 1 (Observation) ──
← Resultat: { pdl: "09876543210987", puissance: 6, compteur: "Linky" }

── Step 2 (Reflexion) ──
Agent: J'ai le PDL. Maintenant j'ai besoin de la consommation pour comparer.
Le client ne me l'a pas dit. Je vais estimer avec la puissance.

── Step 2 (Action) ──
→ Tool call: estimerConsommation({ puissance: 6, type: "appartement", ville: "Paris" })

── Step 2 (Observation) ──
← Resultat: { estimationKwh: 5500 }

── Step 3 (Reflexion) ──
Agent: J'ai le PDL et la conso. Je peux comparer les offres.

── Step 3 (Action) ──
→ Tool call: comparerOffres({ codePostal: "75002", consommationKwh: 5500, type: "electricite" })

── Step 3 (Observation) ──
← Resultat: { offres: [...] }

── Step 4 (Reponse finale) ──
Agent: "Voici les 3 meilleures offres pour votre logement au 12 rue de la Paix..."

import { Agent, tool, stepCountIs } from "ai";
import { z } from "zod";

const selectraAgent = new Agent({
  model: "anthropic/claude-sonnet-4.6",
  instructions: `Tu es l'agent de qualification Selectra.
    Tu collectes les informations du client et cherches les meilleures offres.
    Utilise les outils disponibles pour trouver le PDL, estimer la conso, et comparer les offres.
    Pose UNE question a la fois au client.`,
  tools: {
    chercherPDL: tool({
      description: "Recherche le numero PDL a partir d'une adresse",
      inputSchema: z.object({
        adresse: z.string().describe("Adresse complete du logement"),
      }),
      execute: async ({ adresse }) => {
        // Appel API Enedis/Grdf
        return { pdl: "09876543210987", puissance: 6 };
      },
    }),
    comparerOffres: tool({
      description: "Compare les offres d'energie pour un client",
      inputSchema: z.object({
        codePostal: z.string(),
        consommationKwh: z.number(),
      }),
      execute: async ({ codePostal, consommationKwh }) => {
        // Appel API comparateur Selectra
        return { offres: [/* ... */] };
      },
    }),
    transfererConseiller: tool({
      description: "Transfere le client a un conseiller humain avec les infos collectees",
      inputSchema: z.object({
        ficheClient: z.object({
          nom: z.string().optional(),
          adresse: z.string(),
          pdl: z.string(),
          intent: z.string(),
          offresProposees: z.array(z.string()),
        }),
      }),
      execute: async ({ ficheClient }) => {
        // Envoie la fiche au CRM
        return { status: "transfert_en_cours", conseiller: "Marie D." };
      },
    }),
  },
});

// Utilisation
const result = await selectraAgent.generate({
  prompt: "Bonjour, je demenage au 42 rue Victor Hugo a Lyon",
  stopWhen: stepCountIs(10), // max 10 etapes
});