embed.ts

/**
 * Génère des embeddings via l'API Voyage AI (recommandée par Anthropic
 * pour le RAG). Un compte gratuit suffit pour tester : https://voyageai.com
 *
 * Vous pouvez remplacer ce fichier par n'importe quel autre fournisseur
 * d'embeddings (OpenAI, Cohere, un modèle local...) — le reste du code
 * ne dépend que de la fonction embed() ci-dessous.
 */

const VOYAGE_API_KEY = process.env.VOYAGE_API_KEY;
const VOYAGE_URL = "https://api.voyageai.com/v1/embeddings";

export async function embed(
  texts: string[],
  inputType: "document" | "query"
): Promise<number[][]> {
  if (!VOYAGE_API_KEY) {
    throw new Error("VOYAGE_API_KEY manquant dans .env");
  }

  const res = await fetch(VOYAGE_URL, {
    method: "POST",
    headers: {
      "Content-Type": "application/json",
      Authorization: `Bearer ${VOYAGE_API_KEY}`,
    },
    body: JSON.stringify({
      input: texts,
      model: "voyage-3.5", // modèle généraliste, bon rapport qualité/prix
      input_type: inputType, // "document" à l'ingestion, "query" à la recherche
    }),
  });

  if (!res.ok) {
    throw new Error(`Voyage API error ${res.status}: ${await res.text()}`);
  }

  const data = (await res.json()) as { data: { embedding: number[] }[] };
  return data.data.map((d) => d.embedding);
}