@edwinfom/ai-guard

Un middleware de sécurité pour les réponses d'API d'IA — l'anonymisation des informations personnel, la validation de schéma, la détection d'injection de prompt, la sentinelle budgétaire, et bien plus encore.

Le Problème

Lors de l'intégration d'API d'IA (OpenAI, Anthropic, Gemini) dans des applications de production, les développeurs sont confrontés à des problèmes récurrents sans solution standardisée :

• JSON mal formé — Les LLM encadrent parfois les réponses dans des blocs markdown ou ajoutent du texte explicatif, ce qui plante votre pipeline.
• Fuite de données privées (PII) — Les utilisateurs envoient des mots de passe ou des numéros de carte dans les prompts. Les réponses de l'IA peuvent renvoyer des données sensibles de votre base de données RAG.
• Injection de Prompt — Les utilisateurs malveillants tentent d'écraser votre prompt système par un "Ignore the previous instructions…".
• Vol du prompt système — Un attaquant trompe l'IA pour qu'elle répète vos instructions confidentielles.
• Contenu Toxique ou Dangereux — Pas de modération de contenu intégrée entre le LLM et vos utilisateurs.
• Hallucinations de RAG — L'IA invente des faits non présents dans vos documents sources.
• Facturation surprise — L'utilisation de tokens explose sans le moindre averstissement ou limite matérielle.
• Abus — Un seul utilisateur noie votre terminal de requêtes.

@edwinfom/ai-guard agit comme une membrane sécurisée entre votre application et n'importe quel fournisseur d'IA. Un wrapper unique, toutes protections.

import { Guardian } from '@edwinfom/ai-guard';
import { z } from 'zod';
 
const guard = new Guardian({
  pii:          { onInput: true, onOutput: true },
  schema:       { validator: z.object({ city: z.string(), temp: z.number() }), repair: 'retry' },
  injection:    { enabled: true, sensitivity: 'medium' },
  content:      { enabled: true, sensitivity: 'medium' },
  canary:       { enabled: true },
  hallucination:{ sources: [ragDocument1, ragDocument2] },
  budget:       { maxTokens: 2000, maxCostUSD: 0.05, model: 'gpt-4o-mini' },
  rateLimit:    { maxRequests: 10, windowMs: 60_000, keyFn: (p) => getUserId(p) },
  onAudit:      (entry) => logger.info(entry),
});
 
const result = await guard.protect(
  (safePrompt) => openai.chat.completions.create({ model: 'gpt-4o-mini', messages: [{ role: 'user', content: safePrompt }] }),
  userPrompt
);
 
console.log(result.data);              // typé par votre schéma Zod
console.log(result.meta.budget);       // { totalTokens: 312, estimatedCostUSD: 0.000047 }
console.log(result.meta.piiRedacted);  // [{ type: 'email', value: 'user@...', ... }]
console.log(result.meta.canaryLeaked); // false — le prompt system n'a pas fuité.

Fonctionnalités