vibezcheck 0.4.2 → 0.5.0

package/dist/ai-sdk/index.d.mts

@@ -1,54 +1,13 @@
-import { c as CircuitBreakerOptions, C as CustomerParam, B as BillingConfig, g as PricingConfig, e as InlineRateConfig, U as UsageEvent } from '../types-Mozk4-Mr.mjs';
-import { V as VibezMeter } from '../client-w_hWhQ6g.mjs';
+import { W as WithBillingOptions } from '../with-billing-mci2vtFS.mjs';
+export { m as meteredModel, w as withBilling } from '../with-billing-mci2vtFS.mjs';
+import { C as CustomerParam } from '../types-BVGqbKnY.mjs';
+import { V as VibezMeter } from '../client-Dmwb3hQt.mjs';
 import 'stripe';
 
-interface WithBillingOptions extends CircuitBreakerOptions {
-    /** Customer email, user ID, or rich customer object with metadata */
-    customer?: CustomerParam;
-    /** Direct Stripe customer ID */
-    customerId?: string;
-    /** Billing mode configuration (Universal Auto-Debit, postpaid vs prepaid, pluggable providers) */
-    billing?: BillingConfig;
-    /** Profit margin & minimum charge configuration */
-    pricing?: PricingConfig;
-    /** 1-line inline pricing rate card */
-    rate?: InlineRateConfig;
-    /** Pluggable Database Sink (Supabase, Postgres, Firebase, or 1-line callback) */
-    database?: any | ((event: UsageEvent) => void | Promise<void>);
-    /** Whether to capture tokens if the client aborts or closes tab mid-stream (default: true) */
-    captureOnAbort?: boolean;
-    /** Execution runtime environment (default: 'auto') */
-    runtime?: 'auto' | 'serverless' | 'edge' | 'node';
-    /** Stripe API Key override (uses STRIPE_SECRET_KEY env by default) */
-    stripeApiKey?: string;
-    /** Existing VibezMeter instance (optional) */
-    meter?: VibezMeter;
-    /** Custom meter event name */
-    eventName?: string;
-    /** Usage callback fired upon generation/stream completion */
-    onUsage?: (event: UsageEvent) => void | Promise<void>;
-    /** Custom developer metadata */
-    metadata?: Record<string, string | number | boolean>;
-}
-/**
- * Wraps any Vercel AI SDK LanguageModel (v2 or v3) with automated Stripe billing, token metering, and agent circuit breakers.
- *
- * Sane Defaults Built-In:
- * - $0.50 safety ceiling per call (overridable)
- * - Automatic in-flight token capture on abort
- * - Automatic prompt caching discount parsing
- * - Serverless lifecycle preservation
- */
-declare function withBilling<T extends object>(model: T, options?: WithBillingOptions): T;
-/**
- * Convenient alias for withBilling
- */
-declare const meteredModel: typeof withBilling;
-
 interface VibezCheckModelOptions extends WithBillingOptions {
-    /** OpenAI / AI Gateway API Key override */
+    /** OpenAI / Anthropic / AI Gateway API Key override */
     apiKey?: string;
-    /** AI Gateway / OpenAI Base URL override */
+    /** AI Gateway / Provider Base URL override */
     baseURL?: string;
 }
 interface VibezSessionOptions {
@@ -72,23 +31,33 @@ interface VibezSession {
 /**
  * Creates or resolves an AI SDK compatible LanguageModel with built-in VibezCheck billing & metering.
  *
+ * Supports:
+ * - Direct instances: `vibezcheck(openai('gpt-4o'))`, `vibezcheck(anthropic('claude-3-7-sonnet'))`
+ * - String identifiers: `vibezcheck('openai/gpt-4o-mini')`, `vibezcheck('anthropic/claude-3-5-sonnet')`, `vibezcheck('gpt-4o')`
+ * - Vercel AI SDK primitives: `generateText`, `streamText`, `generateObject`, `streamObject`
+ *
  * @example
  * ```typescript
- * import { generateText, streamText } from 'ai';
+ * import { streamText } from 'ai';
+ * import { openai } from '@ai-sdk/openai';
+ * import { anthropic } from '@ai-sdk/anthropic';
  * import { vibezcheck } from 'vibezcheck';
  *
- * // 1. Declarative string model identifier:
- * const { text } = await generateText({
- *   model: vibezcheck('openai/gpt-4o-mini', { customer: 'alex@example.com' }),
- *   prompt: 'What is love?',
+ * // 1. Pass wrapped OpenAI instance:
+ * const result = streamText({
+ *   model: vibezcheck(openai('gpt-4o-mini'), { customer: 'alex@example.com' }),
+ *   messages,
  * });
  *
- * // 2. Works with all Vercel AI SDK primitives (streamText, generateObject, streamObject):
- * const result = streamText({
- *   model: vibezcheck('gpt-4o', {
- *     customer: 'alex@example.com',
- *     pricing: { margin: 1.5 }, // 50% profit margin
- *   }),
+ * // 2. Pass wrapped Anthropic instance:
+ * const result2 = streamText({
+ *   model: vibezcheck(anthropic('claude-3-7-sonnet'), { customer: 'alex@example.com' }),
+ *   messages,
+ * });
+ *
+ * // 3. Use 1-line string identifier:
+ * const result3 = streamText({
+ *   model: vibezcheck('openai/gpt-4o-mini', { customer: 'alex@example.com' }),
  *   messages,
  * });
  * ```
@@ -102,4 +71,4 @@ declare const vibezcheck: typeof createVibezModel & {
     session: typeof createVibezSession;
 };
 
-export { type VibezCheckModelOptions, type VibezSession, type VibezSessionOptions, type WithBillingOptions, createVibezModel, createVibezSession, meteredModel, vibezcheck, withBilling };
+export { type VibezCheckModelOptions, type VibezSession, type VibezSessionOptions, WithBillingOptions, createVibezModel, createVibezSession, vibezcheck };

package/dist/ai-sdk/index.d.ts

@@ -1,54 +1,13 @@
-import { c as CircuitBreakerOptions, C as CustomerParam, B as BillingConfig, g as PricingConfig, e as InlineRateConfig, U as UsageEvent } from '../types-Mozk4-Mr.js';
-import { V as VibezMeter } from '../client-ynfWcHX6.js';
+import { W as WithBillingOptions } from '../with-billing-CDqKIAQ6.js';
+export { m as meteredModel, w as withBilling } from '../with-billing-CDqKIAQ6.js';
+import { C as CustomerParam } from '../types-BVGqbKnY.js';
+import { V as VibezMeter } from '../client-CJ2Dj8ID.js';
 import 'stripe';
 
-interface WithBillingOptions extends CircuitBreakerOptions {
-    /** Customer email, user ID, or rich customer object with metadata */
-    customer?: CustomerParam;
-    /** Direct Stripe customer ID */
-    customerId?: string;
-    /** Billing mode configuration (Universal Auto-Debit, postpaid vs prepaid, pluggable providers) */
-    billing?: BillingConfig;
-    /** Profit margin & minimum charge configuration */
-    pricing?: PricingConfig;
-    /** 1-line inline pricing rate card */
-    rate?: InlineRateConfig;
-    /** Pluggable Database Sink (Supabase, Postgres, Firebase, or 1-line callback) */
-    database?: any | ((event: UsageEvent) => void | Promise<void>);
-    /** Whether to capture tokens if the client aborts or closes tab mid-stream (default: true) */
-    captureOnAbort?: boolean;
-    /** Execution runtime environment (default: 'auto') */
-    runtime?: 'auto' | 'serverless' | 'edge' | 'node';
-    /** Stripe API Key override (uses STRIPE_SECRET_KEY env by default) */
-    stripeApiKey?: string;
-    /** Existing VibezMeter instance (optional) */
-    meter?: VibezMeter;
-    /** Custom meter event name */
-    eventName?: string;
-    /** Usage callback fired upon generation/stream completion */
-    onUsage?: (event: UsageEvent) => void | Promise<void>;
-    /** Custom developer metadata */
-    metadata?: Record<string, string | number | boolean>;
-}
-/**
- * Wraps any Vercel AI SDK LanguageModel (v2 or v3) with automated Stripe billing, token metering, and agent circuit breakers.
- *
- * Sane Defaults Built-In:
- * - $0.50 safety ceiling per call (overridable)
- * - Automatic in-flight token capture on abort
- * - Automatic prompt caching discount parsing
- * - Serverless lifecycle preservation
- */
-declare function withBilling<T extends object>(model: T, options?: WithBillingOptions): T;
-/**
- * Convenient alias for withBilling
- */
-declare const meteredModel: typeof withBilling;
-
 interface VibezCheckModelOptions extends WithBillingOptions {
-    /** OpenAI / AI Gateway API Key override */
+    /** OpenAI / Anthropic / AI Gateway API Key override */
     apiKey?: string;
-    /** AI Gateway / OpenAI Base URL override */
+    /** AI Gateway / Provider Base URL override */
     baseURL?: string;
 }
 interface VibezSessionOptions {
@@ -72,23 +31,33 @@ interface VibezSession {
 /**
  * Creates or resolves an AI SDK compatible LanguageModel with built-in VibezCheck billing & metering.
  *
+ * Supports:
+ * - Direct instances: `vibezcheck(openai('gpt-4o'))`, `vibezcheck(anthropic('claude-3-7-sonnet'))`
+ * - String identifiers: `vibezcheck('openai/gpt-4o-mini')`, `vibezcheck('anthropic/claude-3-5-sonnet')`, `vibezcheck('gpt-4o')`
+ * - Vercel AI SDK primitives: `generateText`, `streamText`, `generateObject`, `streamObject`
+ *
  * @example
  * ```typescript
- * import { generateText, streamText } from 'ai';
+ * import { streamText } from 'ai';
+ * import { openai } from '@ai-sdk/openai';
+ * import { anthropic } from '@ai-sdk/anthropic';
  * import { vibezcheck } from 'vibezcheck';
  *
- * // 1. Declarative string model identifier:
- * const { text } = await generateText({
- *   model: vibezcheck('openai/gpt-4o-mini', { customer: 'alex@example.com' }),
- *   prompt: 'What is love?',
+ * // 1. Pass wrapped OpenAI instance:
+ * const result = streamText({
+ *   model: vibezcheck(openai('gpt-4o-mini'), { customer: 'alex@example.com' }),
+ *   messages,
  * });
  *
- * // 2. Works with all Vercel AI SDK primitives (streamText, generateObject, streamObject):
- * const result = streamText({
- *   model: vibezcheck('gpt-4o', {
- *     customer: 'alex@example.com',
- *     pricing: { margin: 1.5 }, // 50% profit margin
- *   }),
+ * // 2. Pass wrapped Anthropic instance:
+ * const result2 = streamText({
+ *   model: vibezcheck(anthropic('claude-3-7-sonnet'), { customer: 'alex@example.com' }),
+ *   messages,
+ * });
+ *
+ * // 3. Use 1-line string identifier:
+ * const result3 = streamText({
+ *   model: vibezcheck('openai/gpt-4o-mini', { customer: 'alex@example.com' }),
  *   messages,
  * });
  * ```
@@ -102,4 +71,4 @@ declare const vibezcheck: typeof createVibezModel & {
     session: typeof createVibezSession;
 };
 
-export { type VibezCheckModelOptions, type VibezSession, type VibezSessionOptions, type WithBillingOptions, createVibezModel, createVibezSession, meteredModel, vibezcheck, withBilling };
+export { type VibezCheckModelOptions, type VibezSession, type VibezSessionOptions, WithBillingOptions, createVibezModel, createVibezSession, vibezcheck };

package/dist/ai-sdk/index.js

@@ -602,35 +602,57 @@ function calculateCost(params) {
   } else {
     rates = getModelPricing(params.model);
   }
-  const inputTokens = params.inputTokens ?? 0;
-  const outputTokens = params.outputTokens ?? 0;
-  const reasoningTokens = params.reasoningTokens ?? 0;
-  const cachedTokens = params.cachedTokens ?? 0;
-  const regularInputTokens = Math.max(0, inputTokens - cachedTokens);
-  const regularInputCost = regularInputTokens / 1e6 * rates.inputPer1M;
-  const cachedRate = rates.cachedInputPer1M ?? rates.inputPer1M * 0.15;
-  const cachedInputCost = cachedTokens / 1e6 * cachedRate;
-  const inputCostUSD = regularInputCost + cachedInputCost;
-  const outputCostUSD = outputTokens / 1e6 * rates.outputPer1M;
-  const reasoningRate = rates.reasoningPer1M ?? rates.outputPer1M;
-  const reasoningCostUSD = reasoningTokens / 1e6 * reasoningRate;
-  const standardCacheCost = cachedTokens / 1e6 * rates.inputPer1M;
-  const cachedDiscountUSD = Math.max(0, standardCacheCost - cachedInputCost);
-  let totalUSD = inputCostUSD + outputCostUSD;
+  const inputTokens = BigInt(Math.max(0, params.inputTokens ?? 0));
+  const outputTokens = BigInt(Math.max(0, params.outputTokens ?? 0));
+  const reasoningTokens = BigInt(Math.max(0, params.reasoningTokens ?? 0));
+  const cachedTokens = BigInt(Math.max(0, params.cachedTokens ?? 0));
+  const regularInputTokens = inputTokens > cachedTokens ? inputTokens - cachedTokens : 0n;
+  const rateInputNano = BigInt(Math.round(rates.inputPer1M * 1e3));
+  const cachedPer1M = rates.cachedInputPer1M ?? rates.inputPer1M * 0.15;
+  const rateCachedNano = BigInt(Math.round(cachedPer1M * 1e3));
+  const rateOutputNano = BigInt(Math.round(rates.outputPer1M * 1e3));
+  const reasoningPer1M = rates.reasoningPer1M ?? rates.outputPer1M;
+  const rateReasoningNano = BigInt(Math.round(reasoningPer1M * 1e3));
+  const regularInputCostNano = regularInputTokens * rateInputNano;
+  const cachedInputCostNano = cachedTokens * rateCachedNano;
+  const inputCostNano = regularInputCostNano + cachedInputCostNano;
+  const outputCostNano = outputTokens * rateOutputNano;
+  const reasoningCostNano = reasoningTokens * rateReasoningNano;
+  const standardCacheCostNano = cachedTokens * rateInputNano;
+  const cachedDiscountNano = standardCacheCostNano > cachedInputCostNano ? standardCacheCostNano - cachedInputCostNano : 0n;
+  const wholesaleNano = inputCostNano + outputCostNano;
   const markup = params.markupMultiplier ?? 1;
-  let retailUSD = void 0;
+  let billedNano = wholesaleNano;
+  let hasRetail = false;
   if (markup !== 1 || params.minimumChargeUSD !== void 0) {
-    retailUSD = totalUSD * markup;
-    if (params.minimumChargeUSD !== void 0 && retailUSD < params.minimumChargeUSD) {
-      retailUSD = params.minimumChargeUSD;
+    hasRetail = true;
+    const markupMultiplierNano = BigInt(Math.round(markup * 1e3));
+    billedNano = wholesaleNano * markupMultiplierNano / 1000n;
+    if (params.minimumChargeUSD !== void 0 && params.minimumChargeUSD > 0) {
+      const minChargeNano = BigInt(Math.round(params.minimumChargeUSD * 1e9));
+      if (billedNano < minChargeNano) {
+        billedNano = minChargeNano;
+      }
     }
   }
+  const inputCostUSD = Number(inputCostNano) / 1e9;
+  const outputCostUSD = Number(outputCostNano) / 1e9;
+  const reasoningCostUSD = Number(reasoningCostNano) / 1e9;
+  const cachedDiscountUSD = Number(cachedDiscountNano) / 1e9;
+  const totalUSD = Number(wholesaleNano) / 1e9;
+  const wholesaleTotalUSD = totalUSD;
+  const billedUSD = Number(billedNano) / 1e9;
+  const profitUSD = Math.max(0, billedUSD - wholesaleTotalUSD);
+  const retailUSD = hasRetail ? billedUSD : void 0;
   return {
     inputCostUSD: Number(inputCostUSD.toFixed(8)),
     outputCostUSD: Number(outputCostUSD.toFixed(8)),
-    reasoningCostUSD: reasoningTokens > 0 ? Number(reasoningCostUSD.toFixed(8)) : void 0,
-    cachedDiscountUSD: cachedTokens > 0 ? Number(cachedDiscountUSD.toFixed(8)) : void 0,
+    reasoningCostUSD: reasoningTokens > 0n ? Number(reasoningCostUSD.toFixed(8)) : void 0,
+    cachedDiscountUSD: cachedTokens > 0n ? Number(cachedDiscountUSD.toFixed(8)) : void 0,
     totalUSD: Number(totalUSD.toFixed(8)),
+    wholesaleTotalUSD: Number(wholesaleTotalUSD.toFixed(8)),
+    billedUSD: Number(billedUSD.toFixed(8)),
+    profitUSD: Number(profitUSD.toFixed(8)),
     retailUSD: retailUSD !== void 0 ? Number(retailUSD.toFixed(8)) : void 0,
     currency: rates.currency || "USD"
   };
@@ -858,7 +880,7 @@ var VibezMeter = class {
       this.stripeClient = new import_stripe.default(key, {
         appInfo: {
           name: "vibezcheck",
-          version: "0.4.2",
+          version: "0.5.0",
           url: "https://vibezcheck.xyz"
         }
       });
@@ -1253,23 +1275,48 @@ function createVibezModel(modelOrId, options = {}) {
       const parts = rawId.split("/");
       providerName = parts[0].toLowerCase();
       cleanModelId = parts.slice(1).join("/");
+    } else if (rawId.startsWith("claude-")) {
+      providerName = "anthropic";
+    } else if (rawId.startsWith("gemini-")) {
+      providerName = "google";
     }
-    const apiKey = options.apiKey || process.env.AI_GATEWAY_API_KEY || process.env.OPENAI_API_KEY || process.env.ANTHROPIC_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+    const apiKey = options.apiKey || process.env.AI_GATEWAY_API_KEY || (providerName === "anthropic" ? process.env.ANTHROPIC_API_KEY : void 0) || (providerName === "google" ? process.env.GOOGLE_GENERATIVE_AI_API_KEY : void 0) || process.env.OPENAI_API_KEY;
     const baseURL = options.baseURL || process.env.AI_GATEWAY_BASE_URL || (providerName === "openai" ? "https://api.openai.com/v1" : void 0);
     let baseModelInstance = null;
     try {
       if (providerName === "openai" || providerName === "gateway" || !providerName) {
-        const { createOpenAI } = require("@ai-sdk/openai");
-        const openaiProvider = createOpenAI({ apiKey, baseURL });
-        baseModelInstance = openaiProvider(cleanModelId);
+        const mod = require("@ai-sdk/openai");
+        const factory = mod.createOpenAI || mod.openai || mod.default?.createOpenAI || mod.default?.openai;
+        if (typeof factory === "function") {
+          if (mod.createOpenAI) {
+            const provider = factory({ apiKey, baseURL });
+            baseModelInstance = provider(cleanModelId);
+          } else {
+            baseModelInstance = factory(cleanModelId);
+          }
+        }
       } else if (providerName === "anthropic") {
-        const { createAnthropic } = require("@ai-sdk/anthropic");
-        const anthropicProvider = createAnthropic({ apiKey, baseURL });
-        baseModelInstance = anthropicProvider(cleanModelId);
+        const mod = require("@ai-sdk/anthropic");
+        const factory = mod.createAnthropic || mod.anthropic || mod.default?.createAnthropic || mod.default?.anthropic;
+        if (typeof factory === "function") {
+          if (mod.createAnthropic) {
+            const provider = factory({ apiKey, baseURL });
+            baseModelInstance = provider(cleanModelId);
+          } else {
+            baseModelInstance = factory(cleanModelId);
+          }
+        }
       } else if (providerName === "google") {
-        const { createGoogleGenerativeAI } = require("@ai-sdk/google");
-        const googleProvider = createGoogleGenerativeAI({ apiKey, baseURL });
-        baseModelInstance = googleProvider(cleanModelId);
+        const mod = require("@ai-sdk/google");
+        const factory = mod.createGoogleGenerativeAI || mod.google || mod.default?.createGoogleGenerativeAI || mod.default?.google;
+        if (typeof factory === "function") {
+          if (mod.createGoogleGenerativeAI) {
+            const provider = factory({ apiKey, baseURL });
+            baseModelInstance = provider(cleanModelId);
+          } else {
+            baseModelInstance = factory(cleanModelId);
+          }
+        }
       }
     } catch {
       baseModelInstance = {