@lastbrain/module-ai 2.0.26 → 2.0.30

package/dist/server/__tests__/billing.test.js

@@ -0,0 +1,223 @@
+/**
+ * Tests unitaires pour le système de billing centralisé
+ */
+import { describe, it, expect } from "vitest";
+import { computeAndDebitAIUsage } from "../billing";
+describe("computeAndDebitAIUsage - Système de billing centralisé", () => {
+    // Wallet mock standard (pack 5$ avec coef 2.5)
+    const mockWallet = {
+        tokenBalance: 100000,
+        walletProviderBudgetUsd: 2.0, // 5$ / 2.5
+        walletSellValueUsd: 5.0,
+    };
+    describe("Application des minimums par action", () => {
+        it("recette applique minimum 0.20$", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user-123",
+                actionType: "generate-recipe-text",
+                providerCostUsd: 0.001, // Très bas
+                wallet: mockWallet,
+                metadata: { model: "gpt-4o-mini", provider: "openai" },
+            });
+            expect(result.success).toBe(true);
+            expect(result.sellCostUsd).toBeGreaterThanOrEqual(0.2);
+            expect(result.sellCostUsd).toBe(0.2); // Exactement le minimum
+        });
+        it("image applique minimum 0.05$", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user-456",
+                actionType: "generate-image",
+                providerCostUsd: 0.001,
+                wallet: mockWallet,
+                metadata: { model: "dall-e-3", provider: "openai" },
+            });
+            expect(result.success).toBe(true);
+            expect(result.sellCostUsd).toBeGreaterThanOrEqual(0.05);
+        });
+        it("generate-text générique n'applique pas de minimum", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user-789",
+                actionType: "generate-text",
+                providerCostUsd: 0.001,
+                wallet: mockWallet,
+                metadata: { model: "gpt-4o-mini", provider: "openai" },
+            });
+            expect(result.success).toBe(true);
+            // Avec marge 60%, sell = 0.001 / 0.4 = 0.0025, arrondi ceil = 0.0025
+            expect(result.sellCostUsd).toBeLessThan(0.01);
+        });
+        it("autocomplete n'applique pas de minimum", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user-abc",
+                actionType: "autocomplete",
+                providerCostUsd: 0.0001,
+                wallet: mockWallet,
+                metadata: { model: "gpt-4o-mini", provider: "openai" },
+            });
+            expect(result.success).toBe(true);
+            expect(result.sellCostUsd).toBeLessThan(0.001);
+        });
+    });
+    describe("Calcul de la marge (60%)", () => {
+        it("marge correcte pour coût sans minimum", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user",
+                actionType: "generate-text",
+                providerCostUsd: 0.1,
+                wallet: mockWallet,
+                metadata: { model: "gpt-4o-mini" },
+            });
+            expect(result.success).toBe(true);
+            // sell = 0.1 / 0.4 = 0.25
+            expect(result.sellCostUsd).toBeCloseTo(0.25, 2);
+            expect(result.marginUsd).toBeCloseTo(0.15, 2);
+            expect(result.marginPercent).toBeCloseTo(60, 0);
+        });
+        it("marge élevée quand minimum appliqué", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user",
+                actionType: "generate-recipe-text",
+                providerCostUsd: 0.001, // Très bas
+                wallet: mockWallet,
+                metadata: { model: "gpt-4o-mini" },
+            });
+            expect(result.success).toBe(true);
+            expect(result.sellCostUsd).toBe(0.2); // Minimum
+            expect(result.marginUsd).toBeCloseTo(0.199, 3);
+            // Marge très élevée: (0.199 / 0.2) * 100 ≈ 99.5%
+            expect(result.marginPercent).toBeGreaterThan(99);
+        });
+    });
+    describe("Vérifications budget provider", () => {
+        it("refuse si budget provider insuffisant", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user",
+                actionType: "generate-recipe-text",
+                providerCostUsd: 3.0, // Plus que 2.0 disponible
+                wallet: mockWallet,
+                metadata: { model: "gpt-4" },
+            });
+            expect(result.success).toBe(false);
+            expect(result.error).toBe("insufficient_credits");
+        });
+        it("refuse si sell value insuffisante", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user",
+                actionType: "generate-recipe-text",
+                providerCostUsd: 0.5, // OK pour provider
+                wallet: {
+                    ...mockWallet,
+                    walletSellValueUsd: 0.1, // Insuffisant pour 0.2$ minimum
+                },
+                metadata: { model: "gpt-4o-mini" },
+            });
+            expect(result.success).toBe(false);
+            expect(result.error).toBe("insufficient_credits");
+        });
+        it("accepte si budgets suffisants", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user",
+                actionType: "generate-recipe-text",
+                providerCostUsd: 0.05,
+                wallet: mockWallet,
+                metadata: { model: "gpt-4o-mini" },
+            });
+            expect(result.success).toBe(true);
+            expect(result.newWallet).toBeDefined();
+            expect(result.newWallet.walletProviderBudgetUsd).toBeCloseTo(1.95, 2);
+            expect(result.newWallet.walletSellValueUsd).toBeCloseTo(4.8, 2);
+        });
+    });
+    describe("Calcul percent remaining", () => {
+        it("calcule correctement le pourcentage restant", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user",
+                actionType: "generate-recipe-text",
+                providerCostUsd: 0.1,
+                wallet: mockWallet,
+                metadata: { model: "gpt-4o-mini" },
+            });
+            expect(result.success).toBe(true);
+            // Débite 0.20$ (minimum) sur 5$ = reste 4.8$
+            // 4.8 / 5 = 96%
+            expect(result.newWallet.percentRemaining).toBeCloseTo(96, 0);
+        });
+        it("pourcentage proche de 0 quand presque épuisé", () => {
+            const lowWallet = {
+                tokenBalance: 1000,
+                walletProviderBudgetUsd: 0.1,
+                walletSellValueUsd: 0.25,
+            };
+            const result = computeAndDebitAIUsage({
+                userId: "test-user",
+                actionType: "generate-recipe-text",
+                providerCostUsd: 0.05,
+                wallet: lowWallet,
+                metadata: { model: "gpt-4o-mini" },
+            });
+            expect(result.success).toBe(true);
+            // 0.25 - 0.20 = 0.05 restant
+            // 0.05 / 0.25 = 20%
+            expect(result.newWallet.percentRemaining).toBeCloseTo(20, 0);
+        });
+    });
+    describe("Override minimum", () => {
+        it("permet de surcharger le minimum par action", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user",
+                actionType: "generate-recipe-text", // Min 0.20 par défaut
+                providerCostUsd: 0.001,
+                wallet: mockWallet,
+                metadata: { model: "gpt-4o-mini" },
+                minSellUsdOverride: 0.5, // Forcer 0.50$ minimum
+            });
+            expect(result.success).toBe(true);
+            expect(result.sellCostUsd).toBe(0.5);
+        });
+        it("override à 0 désactive le minimum", () => {
+            const result = computeAndDebitAIUsage({
+                userId: "test-user",
+                actionType: "generate-recipe-text", // Min 0.20 par défaut
+                providerCostUsd: 0.001,
+                wallet: mockWallet,
+                metadata: { model: "gpt-4o-mini" },
+                minSellUsdOverride: 0, // Désactiver minimum
+            });
+            expect(result.success).toBe(true);
+            // Marge 60% seulement: 0.001 / 0.4 = 0.0025
+            expect(result.sellCostUsd).toBeLessThan(0.01);
+        });
+    });
+    describe("Protection contre abus", () => {
+        it("empêche génération de 1000 recettes pour 5$", () => {
+            const wallet = { ...mockWallet };
+            let successCount = 0;
+            let totalCost = 0;
+            // Tenter de générer jusqu'à épuisement
+            for (let i = 0; i < 1000; i++) {
+                const result = computeAndDebitAIUsage({
+                    userId: "test-user",
+                    actionType: "generate-recipe-text",
+                    providerCostUsd: 0.001, // Coût très bas
+                    wallet,
+                    metadata: { model: "gpt-4o-mini" },
+                });
+                if (result.success) {
+                    successCount++;
+                    totalCost += result.sellCostUsd;
+                    // Mettre à jour wallet simulé
+                    wallet.walletProviderBudgetUsd =
+                        result.newWallet.walletProviderBudgetUsd;
+                    wallet.walletSellValueUsd = result.newWallet.walletSellValueUsd;
+                }
+                else {
+                    break;
+                }
+            }
+            // Avec minimum 0.20$ par recette, max = 5 / 0.20 = 25 recettes
+            expect(successCount).toBeLessThanOrEqual(25);
+            expect(successCount).toBeGreaterThanOrEqual(20); // Au moins 20
+            expect(totalCost).toBeLessThanOrEqual(5.01); // Légèrement plus pour arrondis
+        });
+    });
+});

package/dist/server/ai-client.d.ts

@@ -0,0 +1,59 @@
+/**
+ * AI client using Vercel AI SDK with Gateway
+ * Gateway key (vck_...) enables direct model string usage
+ */
+/**
+ * Text generation using Vercel AI Gateway
+ * Model format: "provider/model" (e.g., "openai/gpt-4o-mini", "anthropic/claude-sonnet-4.5")
+ */
+export interface TextAIParams {
+    prompt: string;
+    model?: string;
+    system?: string;
+    maxTokens?: number;
+    temperature?: number;
+}
+export interface TextAIResult {
+    text: string;
+    usage?: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+    raw?: any;
+}
+export declare function textAI(params: TextAIParams): Promise<TextAIResult>;
+/**
+ * Image generation using Vercel AI Gateway
+ * Model format: "provider/model" (e.g., "openai/dall-e-3")
+ */
+export interface ImageAIParams {
+    prompt: string;
+    model?: string;
+    size?: string;
+    n?: number;
+}
+export interface ImageAIResult {
+    imageUrl: string;
+    tokensUsed?: number;
+    raw?: any;
+}
+export declare function imageAI(params: ImageAIParams): Promise<ImageAIResult>;
+/**
+ * Embedding generation using Vercel AI Gateway
+ * Model format: "provider/model" (e.g., "openai/text-embedding-3-small")
+ */
+export interface EmbedAIParams {
+    input: string | string[];
+    model?: string;
+}
+export interface EmbedAIResult {
+    embedding?: number[];
+    embeddings?: number[][];
+    usage?: {
+        tokens: number;
+    };
+    raw?: any;
+}
+export declare function embedAI(params: EmbedAIParams): Promise<EmbedAIResult>;
+//# sourceMappingURL=ai-client.d.ts.map

package/dist/server/ai-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-client.d.ts","sourceRoot":"","sources":["../../src/server/ai-client.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAyBH;;;GAGG;AACH,MAAM,WAAW,YAAY;IAC3B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,GAAG,CAAC,EAAE,GAAG,CAAC;CACX;AAED,wBAAsB,MAAM,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC,CAkCxE;AAED;;;GAGG;AACH,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,CAAC,CAAC,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,GAAG,CAAC,EAAE,GAAG,CAAC;CACX;AAED,wBAAsB,OAAO,CAAC,MAAM,EAAE,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC,CA6C3E;AAED;;;GAGG;AACH,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,aAAa;IAC5B,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;IACxB,KAAK,CAAC,EAAE;QACN,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,GAAG,CAAC,EAAE,GAAG,CAAC;CACX;AAED,wBAAsB,OAAO,CAAC,MAAM,EAAE,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC,CAwB3E"}

package/dist/server/ai-client.js

@@ -0,0 +1,111 @@
+/**
+ * AI client using Vercel AI SDK with Gateway
+ * Gateway key (vck_...) enables direct model string usage
+ */
+import { generateText, generateImage, embed } from "ai";
+import { logger } from "@lastbrain/core";
+// Gateway API key - no need to create providers
+const gatewayKey = process.env.AI_GATEWAY_API_KEY;
+if (!gatewayKey) {
+    logger.error("[AI Client] AI_GATEWAY_API_KEY not configured");
+}
+// Get default models from environment with fallbacks
+function getDefaultTextModel() {
+    return process.env.AI_DEFAULT_TEXT_MODEL || "openai/gpt-4o-mini";
+}
+function getDefaultImageModel() {
+    return process.env.AI_DEFAULT_IMAGE_MODEL || "openai/dall-e-3";
+}
+function getDefaultEmbedModel() {
+    return process.env.AI_DEFAULT_EMBED_MODEL || "openai/text-embedding-3-small";
+}
+export async function textAI(params) {
+    const fullModel = params.model || getDefaultTextModel();
+    logger.info(`[textAI] Generating text with model: ${fullModel}`);
+    try {
+        // With Gateway, pass model string directly (e.g., "openai/gpt-4o-mini")
+        const result = await generateText({
+            model: fullModel,
+            prompt: params.prompt,
+            system: params.system,
+            temperature: params.temperature,
+            ...(params.maxTokens && { maxTokens: params.maxTokens }),
+        });
+        logger.info(`[textAI] Generation complete. Usage: ${JSON.stringify(result.usage)}`);
+        return {
+            text: result.text,
+            usage: result.usage
+                ? {
+                    promptTokens: result.usage.inputTokens ?? 0,
+                    completionTokens: result.usage.outputTokens ?? 0,
+                    totalTokens: result.usage.totalTokens ?? 0,
+                }
+                : undefined,
+            raw: result,
+        };
+    }
+    catch (error) {
+        logger.error(`[textAI] Error with model ${fullModel}:`, error);
+        throw new Error(`Text generation failed: ${error.message}`);
+    }
+}
+export async function imageAI(params) {
+    const fullModel = params.model || getDefaultImageModel();
+    logger.info(`[imageAI] Generating image with model: ${fullModel}`);
+    try {
+        // Use generateImage API for image generation
+        const result = await generateImage({
+            model: fullModel,
+            prompt: params.prompt,
+        });
+        // Image is available in result.image
+        if (!result.image) {
+            logger.error("[imageAI] No image generated", { result });
+            throw new Error("No image generated by the model");
+        }
+        // Convert to base64 data URL
+        const imageUrl = `data:${result.image.mediaType};base64,${result.image.base64}`;
+        // Log pour déboguer l'usage réel
+        logger.info(`[imageAI] Generated image with model ${fullModel}`, {
+            usage: result.usage,
+            rawUsage: JSON.stringify(result.usage),
+        });
+        // Use token usage from result - vérifier plusieurs formats possibles
+        const tokensUsed = result.usage?.totalTokens ||
+            result.usage?.total ||
+            result.usage?.tokens ||
+            6000;
+        logger.info(`[imageAI] Tokens used: ${tokensUsed}`);
+        return {
+            imageUrl,
+            tokensUsed,
+            raw: result,
+        };
+    }
+    catch (error) {
+        logger.error(`[imageAI] Error with model ${fullModel}:`, error);
+        throw new Error(`Image generation failed: ${error.message}`);
+    }
+}
+export async function embedAI(params) {
+    const fullModel = params.model || getDefaultEmbedModel();
+    logger.info(`[embedAI] Generating embedding with model: ${fullModel}`);
+    try {
+        // With Gateway, pass model string directly
+        const result = await embed({
+            model: fullModel,
+            value: typeof params.input === "string"
+                ? params.input
+                : params.input.join("\n"),
+        });
+        return {
+            embedding: result.embedding,
+            usage: result.usage ? { tokens: result.usage.tokens } : undefined,
+            raw: result,
+        };
+    }
+    catch (error) {
+        logger.error(`[embedAI] Error with model ${fullModel}:`, error);
+        throw new Error(`Embedding generation failed: ${error.message}`);
+    }
+}

package/dist/server/ai-generation-service.d.ts

@@ -0,0 +1,66 @@
+/**
+ * Centralized AI Generation Service
+ *
+ * SINGLE SOURCE OF TRUTH for all AI generation with billing
+ * Used by both internal auth endpoints AND public API v1
+ *
+ * This service handles:
+ * 1. Pricing validation
+ * 2. Wallet pre-check (> 0)
+ * 3. AI API call
+ * 4. Real cost computation with minimums
+ * 5. Wallet debit
+ * 6. Ledger recording
+ */
+import { type ActionType } from "./billing";
+export interface TextGenerationParams {
+    prompt: string;
+    model?: string;
+    systemPrompt?: string;
+    maxTokens?: number;
+    temperature?: number;
+    actionType?: ActionType;
+    context?: string;
+    promptUser?: string;
+    metadata?: Record<string, any>;
+}
+export interface ImageGenerationParams {
+    prompt: string;
+    model?: string;
+    size?: "256x256" | "512x512" | "1024x1024" | "1792x1024" | "1024x1792";
+    quality?: "standard" | "hd";
+    actionType?: ActionType;
+    metadata?: Record<string, any>;
+}
+export interface GenerationResult {
+    success: boolean;
+    error?: string;
+    errorCode?: "NO_CREDITS" | "NO_PRICING" | "MODEL_NOT_ALLOWED" | "PROVIDER_ERROR";
+    text?: string;
+    inputTokens?: number;
+    outputTokens?: number;
+    imageUrl?: string;
+    revisedPrompt?: string;
+    model?: string;
+    provider?: string;
+    providerCostUsd?: number;
+    sellCostUsd?: number;
+    marginPercent?: number;
+}
+/**
+ * Generate text with automatic billing
+ *
+ * @param userId User ID
+ * @param params Generation parameters
+ * @returns Generation result with billing info
+ */
+export declare function generateTextWithBilling(userId: string, params: TextGenerationParams): Promise<GenerationResult>;
+/**
+ * Generate image with automatic billing
+ *
+ * @param userId User ID
+ * @param params Generation parameters
+ * @returns Generation result with billing info
+ */
+export declare function generateImageWithBilling(userId: string, params: ImageGenerationParams): Promise<GenerationResult>;
+//# sourceMappingURL=ai-generation-service.d.ts.map

package/dist/server/ai-generation-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-generation-service.d.ts","sourceRoot":"","sources":["../../src/server/ai-generation-service.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAOH,OAAO,EAKL,KAAK,UAAU,EAChB,MAAM,WAAW,CAAC;AAMnB,MAAM,WAAW,oBAAoB;IACnC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,WAAW,GAAG,WAAW,CAAC;IACvE,OAAO,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC;IAC5B,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EACN,YAAY,GACZ,YAAY,GACZ,mBAAmB,GACnB,gBAAgB,CAAC;IAGrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IAGtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IAGvB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAMD;;;;;;GAMG;AACH,wBAAsB,uBAAuB,CAC3C,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,oBAAoB,GAC3B,OAAO,CAAC,gBAAgB,CAAC,CAqK3B;AAMD;;;;;;GAMG;AACH,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,gBAAgB,CAAC,CAwJ3B"}