vibezcheck 0.4.2 → 0.5.3

package/dist/index.mjs

@@ -269,12 +269,12 @@ function extractOpenAIResponseUsage(response) {
 function inspectOpenAIStreamChunk(chunk) {
   if (!chunk || typeof chunk !== "object") return {};
   const model = chunk.model;
-  if (chunk.usage) {
-    const rawUsage = chunk.usage;
-    const inputTokens = rawUsage.prompt_tokens ?? rawUsage.input_tokens ?? 0;
-    const outputTokens = rawUsage.completion_tokens ?? rawUsage.output_tokens ?? 0;
-    const reasoningTokens = rawUsage.completion_tokens_details?.reasoning_tokens ?? rawUsage.output_token_details?.reasoning_tokens ?? 0;
-    const cachedTokens = rawUsage.prompt_tokens_details?.cached_tokens ?? rawUsage.input_token_details?.cached_tokens ?? 0;
+  const rawUsage = chunk.usage || chunk.token_usage || chunk.x_groq?.usage || chunk.usageMetadata;
+  if (rawUsage) {
+    const inputTokens = rawUsage.prompt_tokens ?? rawUsage.input_tokens ?? rawUsage.promptTokens ?? rawUsage.inputTokens ?? rawUsage.promptTokenCount ?? 0;
+    const outputTokens = rawUsage.completion_tokens ?? rawUsage.output_tokens ?? rawUsage.completionTokens ?? rawUsage.outputTokens ?? rawUsage.candidatesTokenCount ?? 0;
+    const reasoningTokens = rawUsage.completion_tokens_details?.reasoning_tokens ?? rawUsage.output_token_details?.reasoning_tokens ?? rawUsage.reasoning_tokens ?? rawUsage.reasoningTokens ?? rawUsage.thoughtsTokenCount ?? 0;
+    const cachedTokens = rawUsage.prompt_tokens_details?.cached_tokens ?? rawUsage.input_token_details?.cached_tokens ?? rawUsage.prompt_cache_hit_tokens ?? rawUsage.cached_tokens ?? rawUsage.cachedTokens ?? rawUsage.cachedContentTokenCount ?? 0;
     return {
       model,
       usage: {
@@ -289,11 +289,11 @@ function inspectOpenAIStreamChunk(chunk) {
   }
   if (chunk.type === "response.completed" || chunk.type === "response.done") {
     if (chunk.response?.usage) {
-      const rawUsage = chunk.response.usage;
-      const inputTokens = rawUsage.input_tokens ?? 0;
-      const outputTokens = rawUsage.output_tokens ?? 0;
-      const reasoningTokens = rawUsage.output_token_details?.reasoning_tokens ?? 0;
-      const cachedTokens = rawUsage.input_token_details?.cached_tokens ?? 0;
+      const rawUsage2 = chunk.response.usage;
+      const inputTokens = rawUsage2.input_tokens ?? 0;
+      const outputTokens = rawUsage2.output_tokens ?? 0;
+      const reasoningTokens = rawUsage2.output_token_details?.reasoning_tokens ?? 0;
+      const cachedTokens = rawUsage2.input_token_details?.cached_tokens ?? 0;
       return {
         model: chunk.response.model || model,
         usage: {
@@ -458,7 +458,7 @@ function detectAndExtractUsage(response, fallbackModel, fallbackProvider) {
 
 // src/pricing/table.ts
 var MODEL_PRICING_TABLE = {
-  // --- OpenAI ---
+  // --- OpenAI (Modern & Reasoning) ---
   "gpt-5.6-sol": { inputPer1M: 4, outputPer1M: 20, cachedInputPer1M: 0.4 },
   "gpt-5.6-terra": { inputPer1M: 2, outputPer1M: 12, cachedInputPer1M: 0.2 },
   "gpt-5.6-luna": { inputPer1M: 0.2, outputPer1M: 1.2, cachedInputPer1M: 0.02 },
@@ -470,11 +470,22 @@ var MODEL_PRICING_TABLE = {
   "o3-mini": { inputPer1M: 1.1, outputPer1M: 4.4, cachedInputPer1M: 0.55 },
   "gpt-4o": { inputPer1M: 2.5, outputPer1M: 10, cachedInputPer1M: 1.25 },
   "gpt-4o-mini": { inputPer1M: 0.15, outputPer1M: 0.6, cachedInputPer1M: 0.075 },
+  "gpt-4.5": { inputPer1M: 75, outputPer1M: 150, cachedInputPer1M: 37.5 },
+  "gpt-4.5-preview": { inputPer1M: 75, outputPer1M: 150, cachedInputPer1M: 37.5 },
+  "chatgpt-4o-latest": { inputPer1M: 5, outputPer1M: 15 },
   "gpt-4.1": { inputPer1M: 2, outputPer1M: 8, cachedInputPer1M: 1 },
   "gpt-4.1-nano": { inputPer1M: 0.1, outputPer1M: 0.4, cachedInputPer1M: 0.05 },
+  // --- OpenAI (Legacy & Backward Compatibility) ---
+  "gpt-4-turbo": { inputPer1M: 10, outputPer1M: 30, cachedInputPer1M: 5 },
+  "gpt-4-turbo-preview": { inputPer1M: 10, outputPer1M: 30, cachedInputPer1M: 5 },
+  "gpt-4": { inputPer1M: 30, outputPer1M: 60 },
+  "gpt-4-32k": { inputPer1M: 60, outputPer1M: 120 },
+  "gpt-3.5-turbo": { inputPer1M: 0.5, outputPer1M: 1.5 },
+  "gpt-3.5-turbo-16k": { inputPer1M: 3, outputPer1M: 4 },
   "text-embedding-3-small": { inputPer1M: 0.02, outputPer1M: 0 },
   "text-embedding-3-large": { inputPer1M: 0.13, outputPer1M: 0 },
-  // --- Anthropic ---
+  "text-embedding-ada-002": { inputPer1M: 0.1, outputPer1M: 0 },
+  // --- Anthropic (Modern & Extended Thinking) ---
   "claude-3-7-sonnet": { inputPer1M: 0.59, outputPer1M: 2.93, cachedInputPer1M: 0.3 },
   "claude-sonnet-5": { inputPer1M: 2, outputPer1M: 10, cachedInputPer1M: 0.3 },
   "claude-3-5-sonnet": { inputPer1M: 3, outputPer1M: 15, cachedInputPer1M: 0.3 },
@@ -482,48 +493,145 @@ var MODEL_PRICING_TABLE = {
   "haiku-4.5": { inputPer1M: 1, outputPer1M: 5, cachedInputPer1M: 0.1 },
   "claude-opus-5": { inputPer1M: 5, outputPer1M: 25, cachedInputPer1M: 1.5 },
   "claude-3-opus": { inputPer1M: 15, outputPer1M: 75, cachedInputPer1M: 1.5 },
-  // --- Google Gemini ---
+  // --- Anthropic (Legacy & Backward Compatibility) ---
+  "claude-3-sonnet": { inputPer1M: 3, outputPer1M: 15, cachedInputPer1M: 0.3 },
+  "claude-3-haiku": { inputPer1M: 0.25, outputPer1M: 1.25, cachedInputPer1M: 0.025 },
+  "claude-2.1": { inputPer1M: 8, outputPer1M: 24 },
+  "claude-2.0": { inputPer1M: 8, outputPer1M: 24 },
+  "claude-instant-1.2": { inputPer1M: 1.63, outputPer1M: 5.51 },
+  // --- Google Gemini (Modern & Thoughts) ---
   "gemini-3.7-flash": { inputPer1M: 0.75, outputPer1M: 3.75, cachedInputPer1M: 0.18 },
   "gemini-3.1-pro": { inputPer1M: 2, outputPer1M: 12, cachedInputPer1M: 0.5 },
   "gemini-3.5-flash": { inputPer1M: 1.5, outputPer1M: 9, cachedInputPer1M: 0.38 },
   "gemini-3.1-flash-lite": { inputPer1M: 0.25, outputPer1M: 1.5, cachedInputPer1M: 0.06 },
+  "gemini-2.5-pro": { inputPer1M: 1.25, outputPer1M: 5, cachedInputPer1M: 0.3125 },
+  "gemini-2.5-flash": { inputPer1M: 0.15, outputPer1M: 0.6, cachedInputPer1M: 0.0375 },
   "gemini-2.0-flash": { inputPer1M: 0.1, outputPer1M: 0.4, cachedInputPer1M: 0.025 },
+  "gemini-2.0-flash-lite": { inputPer1M: 0.075, outputPer1M: 0.3, cachedInputPer1M: 0.01875 },
+  "gemini-2.0-pro": { inputPer1M: 1.25, outputPer1M: 5, cachedInputPer1M: 0.3125 },
+  // --- Google Gemini (Legacy & Backward Compatibility) ---
   "gemini-1.5-pro": { inputPer1M: 1.25, outputPer1M: 5, cachedInputPer1M: 0.3125 },
   "gemini-1.5-flash": { inputPer1M: 0.075, outputPer1M: 0.3, cachedInputPer1M: 0.01875 },
+  "gemini-1.5-flash-8b": { inputPer1M: 0.0375, outputPer1M: 0.15, cachedInputPer1M: 9375e-6 },
+  "gemini-1.0-pro": { inputPer1M: 0.5, outputPer1M: 1.5 },
   // --- xAI Grok ---
   "grok-4.6": { inputPer1M: 3, outputPer1M: 15 },
+  "grok-3": { inputPer1M: 3, outputPer1M: 15 },
+  "grok-3-mini": { inputPer1M: 0.3, outputPer1M: 1.5 },
   "grok-2": { inputPer1M: 2, outputPer1M: 10 },
   "grok-2-vision": { inputPer1M: 2, outputPer1M: 10 },
   "grok-beta": { inputPer1M: 5, outputPer1M: 15 },
   // --- Mistral ---
   "mistral-large-3": { inputPer1M: 2, outputPer1M: 6 },
   "mistral-large-latest": { inputPer1M: 2, outputPer1M: 6 },
+  "mistral-large-2411": { inputPer1M: 2, outputPer1M: 6 },
   "codestral-latest": { inputPer1M: 0.3, outputPer1M: 0.9 },
+  "mistral-medium-latest": { inputPer1M: 2.7, outputPer1M: 8.1 },
   "mistral-small-latest": { inputPer1M: 0.2, outputPer1M: 0.6 },
   "ministral-8b-latest": { inputPer1M: 0.1, outputPer1M: 0.1 },
-  // --- Groq LPUs ---
+  "ministral-3b-latest": { inputPer1M: 0.04, outputPer1M: 0.04 },
+  "open-mistral-7b": { inputPer1M: 0.2, outputPer1M: 0.2 },
+  "open-mixtral-8x7b": { inputPer1M: 0.7, outputPer1M: 0.7 },
+  "open-mixtral-8x22b": { inputPer1M: 2, outputPer1M: 6 },
+  // --- Groq & Meta Llama LPUs ---
   "llama-3.3-70b-versatile": { inputPer1M: 0.59, outputPer1M: 0.79 },
+  "llama-3.1-405b": { inputPer1M: 3, outputPer1M: 3 },
+  "llama-3.1-70b-versatile": { inputPer1M: 0.59, outputPer1M: 0.79 },
   "llama-3.1-8b-instant": { inputPer1M: 0.05, outputPer1M: 0.08 },
+  "llama-3.2-1b-preview": { inputPer1M: 0.04, outputPer1M: 0.04 },
+  "llama-3.2-3b-preview": { inputPer1M: 0.06, outputPer1M: 0.06 },
+  "llama-3.2-11b-vision": { inputPer1M: 0.18, outputPer1M: 0.18 },
+  "llama-3.2-90b-vision": { inputPer1M: 0.9, outputPer1M: 0.9 },
   "deepseek-r1-distill-llama-70b": { inputPer1M: 0.75, outputPer1M: 0.99 },
+  "deepseek-r1-distill-qwen-32b": { inputPer1M: 0.49, outputPer1M: 0.49 },
   "qwen-2.5-32b": { inputPer1M: 0.29, outputPer1M: 0.39 },
+  "qwen-2.5-72b": { inputPer1M: 0.35, outputPer1M: 0.4 },
+  "mixtral-8x7b-32768": { inputPer1M: 0.24, outputPer1M: 0.24 },
+  "gemma2-9b-it": { inputPer1M: 0.2, outputPer1M: 0.2 },
   // --- DeepSeek ---
   "deepseek-v4-pro": { inputPer1M: 0.66, outputPer1M: 1.98, cachedInputPer1M: 0.15 },
   "deepseek-v4-flash": { inputPer1M: 0.22, outputPer1M: 0.66, cachedInputPer1M: 0.05 },
-  "deepseek-chat": { inputPer1M: 0.22, outputPer1M: 0.66, cachedInputPer1M: 0.05 },
-  "deepseek-reasoner": { inputPer1M: 0.66, outputPer1M: 1.98, cachedInputPer1M: 0.15 },
+  "deepseek-v3": { inputPer1M: 0.14, outputPer1M: 0.28, cachedInputPer1M: 0.014 },
+  "deepseek-chat": { inputPer1M: 0.14, outputPer1M: 0.28, cachedInputPer1M: 0.014 },
+  "deepseek-r1": { inputPer1M: 0.55, outputPer1M: 2.19, cachedInputPer1M: 0.14 },
+  "deepseek-reasoner": { inputPer1M: 0.55, outputPer1M: 2.19, cachedInputPer1M: 0.14 },
   // --- Cohere ---
   "command-r-plus": { inputPer1M: 2.5, outputPer1M: 10 },
-  "command-r": { inputPer1M: 0.15, outputPer1M: 0.6 }
+  "command-r": { inputPer1M: 0.15, outputPer1M: 0.6 },
+  "command": { inputPer1M: 1, outputPer1M: 2 },
+  "command-light": { inputPer1M: 0.3, outputPer1M: 0.6 },
+  "embed-english-v3.0": { inputPer1M: 0.1, outputPer1M: 0 },
+  // --- Perplexity ---
+  "sonar": { inputPer1M: 1, outputPer1M: 1 },
+  "sonar-pro": { inputPer1M: 3, outputPer1M: 15 },
+  "sonar-reasoning": { inputPer1M: 1, outputPer1M: 5 },
+  "sonar-reasoning-pro": { inputPer1M: 2, outputPer1M: 8 }
+};
+var MODEL_ALIASES = {
+  // OpenAI Shorthands & Aliases
+  "gpt4": "gpt-4",
+  "gpt4o": "gpt-4o",
+  "gpt-4-preview": "gpt-4-turbo",
+  "gpt-4-0125-preview": "gpt-4-turbo",
+  "gpt-4-1106-preview": "gpt-4-turbo",
+  "gpt-3.5": "gpt-3.5-turbo",
+  "gpt-3.5-turbo-0125": "gpt-3.5-turbo",
+  "gpt-3.5-turbo-1106": "gpt-3.5-turbo",
+  "gpt-3.5-turbo-16k-0613": "gpt-3.5-turbo-16k",
+  // Anthropic Shorthands
+  "sonnet": "claude-3-5-sonnet",
+  "sonnet-3.7": "claude-3-7-sonnet",
+  "sonnet-3.5": "claude-3-5-sonnet",
+  "haiku": "claude-3-5-haiku",
+  "haiku-3.5": "claude-3-5-haiku",
+  "opus": "claude-3-opus",
+  "claude-2": "claude-2.0",
+  "claude-instant": "claude-instant-1.2",
+  // DeepSeek Shorthands
+  "r1": "deepseek-r1",
+  "v3": "deepseek-v3",
+  // Gemini Shorthands
+  "flash": "gemini-2.0-flash",
+  "pro": "gemini-1.5-pro",
+  // Meta / Groq Shorthands
+  "llama-3.3-70b": "llama-3.3-70b-versatile",
+  "llama-3.1-70b": "llama-3.1-70b-versatile",
+  "llama-3.1-8b": "llama-3.1-8b-instant",
+  "llama-3-70b": "llama-3.1-70b-versatile",
+  "llama-3-8b": "llama-3.1-8b-instant",
+  // Mistral Shorthands
+  "codestral": "codestral-latest",
+  "mistral-large": "mistral-large-latest",
+  "mistral-small": "mistral-small-latest"
 };
 var customPricingRegistry = {};
 function normalizeModelKey(rawModel) {
   if (!rawModel) return "unknown";
   let model = rawModel.toLowerCase().trim();
+  model = model.replace(/^[a-z0-9_-]+\.(anthropic|meta|amazon|cohere|mistral|ai21)\./i, "");
+  model = model.replace(/^(anthropic|meta|amazon|cohere|mistral|ai21)\./i, "");
+  model = model.replace(/-v\d+(:\d+)?$/, "");
+  model = model.replace(/:\d+$/, "");
   if (model.includes("/")) {
-    model = model.split("/")[1] || model;
+    model = model.split("/").slice(1).join("/");
   }
+  model = model.replace(/:(latest|free|beta)$/, "");
   model = model.replace(/-\d{8}$/, "");
   model = model.replace(/-\d{4}-\d{2}-\d{2}$/, "");
+  model = model.replace(/llama(\d+)-(\d+)-/g, "llama-$1.$2-");
+  model = model.replace(/llama(\d+)\.(\d+)-/g, "llama-$1.$2-");
+  if (MODEL_ALIASES[model]) {
+    return MODEL_ALIASES[model];
+  }
+  if (!MODEL_PRICING_TABLE[model]) {
+    const withoutInstruct = model.replace(/-(instruct|chat|preview)$/, "");
+    if (MODEL_PRICING_TABLE[withoutInstruct]) {
+      return withoutInstruct;
+    }
+    if (MODEL_ALIASES[withoutInstruct]) {
+      return MODEL_ALIASES[withoutInstruct];
+    }
+  }
   return model;
 }
 function getModelPricing(modelName) {
@@ -540,6 +648,10 @@ function getModelPricing(modelName) {
   if (MODEL_PRICING_TABLE[modelName]) {
     return MODEL_PRICING_TABLE[modelName];
   }
+  const alias = MODEL_ALIASES[modelName.toLowerCase().trim()];
+  if (alias && MODEL_PRICING_TABLE[alias]) {
+    return MODEL_PRICING_TABLE[alias];
+  }
   return {
     inputPer1M: 1,
     outputPer1M: 3,
@@ -567,35 +679,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"
   };
@@ -801,7 +935,7 @@ function wrapUniversalStream(stream, options = {}, onComplete) {
     return wrapGeminiStream(stream, options, onComplete);
   }
   if (Symbol.asyncIterator in stream) {
-    if (options.provider === "anthropic") {
+    if (options.provider === "anthropic" || options.model?.toLowerCase().includes("claude")) {
       return wrapAnthropicStream(stream, options, onComplete);
     }
     return wrapOpenAIStream(stream, options, onComplete);
@@ -823,7 +957,7 @@ var VibezMeter = class {
       this.stripeClient = new Stripe(key, {
         appInfo: {
           name: "vibezcheck",
-          version: "0.4.2",
+          version: "0.5.3",
           url: "https://vibezcheck.xyz"
         }
       });
@@ -940,10 +1074,20 @@ var VibezCircuitBreakerError = class extends Error {
 };
 
 // src/ai-sdk/with-billing.ts
-function withBilling(model, options = {}) {
+function withBilling(model, optionsOrCustomer, extraOptions = {}) {
   if (!model || typeof model !== "object") {
     return model;
   }
+  let options = {};
+  if (typeof optionsOrCustomer === "string") {
+    options = { ...extraOptions, customer: optionsOrCustomer };
+  } else if (typeof optionsOrCustomer === "object" && optionsOrCustomer !== null) {
+    if ("userId" in optionsOrCustomer || "email" in optionsOrCustomer || "id" in optionsOrCustomer) {
+      options = { ...extraOptions, customer: optionsOrCustomer };
+    } else {
+      options = { ...optionsOrCustomer, ...extraOptions };
+    }
+  }
   const meter = options.meter || createMeter({
     apiKey: options.stripeApiKey,
     eventName: options.eventName
@@ -972,17 +1116,19 @@ function withBilling(model, options = {}) {
   };
   const handleUsage = (rawUsage, extraMeta) => {
     if (!rawUsage) return;
-    const inputTokens = rawUsage.promptTokens ?? rawUsage.inputTokens ?? 0;
-    const outputTokens = rawUsage.completionTokens ?? rawUsage.outputTokens ?? 0;
-    const reasoningTokens = rawUsage.reasoningTokens ?? rawUsage.completionTokensDetails?.reasoningTokens ?? rawUsage.outputTokenDetails?.reasoningTokens ?? 0;
-    const cachedTokens = rawUsage.promptTokensDetails?.cachedTokens ?? rawUsage.inputTokenDetails?.cachedTokens ?? 0;
+    const inputTokens = rawUsage.promptTokens ?? rawUsage.inputTokens ?? rawUsage.prompt_tokens ?? rawUsage.input_tokens ?? rawUsage.promptTokenCount ?? 0;
+    const outputTokens = rawUsage.completionTokens ?? rawUsage.outputTokens ?? rawUsage.completion_tokens ?? rawUsage.output_tokens ?? rawUsage.candidatesTokenCount ?? 0;
+    const reasoningTokens = rawUsage.reasoningTokens ?? rawUsage.completionTokensDetails?.reasoningTokens ?? rawUsage.outputTokenDetails?.reasoningTokens ?? rawUsage.reasoning_tokens ?? rawUsage.completion_tokens_details?.reasoning_tokens ?? rawUsage.output_token_details?.reasoning_tokens ?? rawUsage.thoughtsTokenCount ?? 0;
+    const cachedTokens = rawUsage.promptTokensDetails?.cachedTokens ?? rawUsage.inputTokenDetails?.cachedTokens ?? rawUsage.cachedTokens ?? rawUsage.cached_tokens ?? rawUsage.prompt_tokens_details?.cached_tokens ?? rawUsage.input_token_details?.cached_tokens ?? rawUsage.cachedContentTokenCount ?? rawUsage.cache_read_input_tokens ?? 0;
+    const cacheWriteTokens = rawUsage.cacheWriteTokens ?? rawUsage.cache_creation_input_tokens ?? rawUsage.cacheCreationTokens ?? 0;
     const usage = {
       inputTokens,
       outputTokens,
       totalTokens: inputTokens + outputTokens,
       reasoningTokens: reasoningTokens > 0 ? reasoningTokens : void 0,
       visibleOutputTokens: Math.max(0, outputTokens - reasoningTokens),
-      cachedTokens: cachedTokens > 0 ? cachedTokens : void 0
+      cachedTokens: cachedTokens > 0 ? cachedTokens : void 0,
+      cacheWriteTokens: cacheWriteTokens > 0 ? cacheWriteTokens : void 0
     };
     const cost = calculateUsageCost(modelId, usage, {
       markupMultiplier: options.pricing?.margin,
@@ -1083,7 +1229,9 @@ function withBilling(model, options = {}) {
             return;
           }
         } catch (err) {
-          console.warn(`[vibezcheck] Database record failed safely:`, err?.message || err);
+          if (process.env.NODE_ENV !== "test" && !options.silent) {
+            console.warn(`[vibezcheck] Database record failed safely:`, err?.message || err);
+          }
         }
       };
       if (typeof globalThis !== "undefined" && typeof globalThis.after === "function") {
@@ -1099,7 +1247,9 @@ function withBilling(model, options = {}) {
         try {
           await chargeFn(cost.totalUSD, event);
         } catch (err) {
-          console.warn(`[vibezcheck] Custom charge handler failed safely:`, err?.message || err);
+          if (process.env.NODE_ENV !== "test" && !options.silent) {
+            console.warn(`[vibezcheck] Custom charge handler failed safely:`, err?.message || err);
+          }
         }
       };
       if (typeof globalThis !== "undefined" && typeof globalThis.after === "function") {
@@ -1114,7 +1264,9 @@ function withBilling(model, options = {}) {
         try {
           await providerCharge(cost.totalUSD, event);
         } catch (err) {
-          console.warn(`[vibezcheck] Payment provider charge failed safely:`, err?.message || err);
+          if (process.env.NODE_ENV !== "test" && !options.silent) {
+            console.warn(`[vibezcheck] Payment provider charge failed safely:`, err?.message || err);
+          }
         }
       };
       if (typeof globalThis !== "undefined" && typeof globalThis.after === "function") {
@@ -1169,14 +1321,19 @@ function withBilling(model, options = {}) {
           const transformStream = new TransformStream({
             transform(chunk, controller) {
               controller.enqueue(chunk);
-              if (chunk.type === "text-delta" && chunk.textDelta) {
-                accumulatedChars += chunk.textDelta.length;
-              } else if (chunk.type === "reasoning" && (chunk.textDelta || chunk.reasoning)) {
-                accumulatedChars += (chunk.textDelta || chunk.reasoning || "").length;
+              const delta = chunk.textDelta || chunk.text || chunk.delta;
+              if (delta && typeof delta === "string") {
+                accumulatedChars += delta.length;
+              } else if (chunk.type === "reasoning" || chunk.type === "reasoning-delta") {
+                const reasoning = chunk.reasoning || chunk.textDelta || chunk.text || chunk.delta || "";
+                if (typeof reasoning === "string") {
+                  accumulatedChars += reasoning.length;
+                }
               }
-              if (chunk.type === "finish" && chunk.usage) {
+              const chunkUsage = chunk.usage || chunk.token_usage || chunk.usageMetadata;
+              if (chunkUsage) {
                 streamCompleted = true;
-                handleUsage(chunk.usage);
+                handleUsage(chunkUsage);
               }
             },
             flush() {
@@ -1202,13 +1359,29 @@ function withBilling(model, options = {}) {
       if (prop === "doGenerate" && typeof originalValue === "function") {
         return async function(...args) {
           const result = await originalValue.apply(target, args);
-          if (result && result.usage) {
-            handleUsage(result.usage);
+          const usage = result?.usage || result?.tokenUsage || result?.usageMetadata;
+          if (usage) {
+            handleUsage(usage);
           }
           return result;
         };
       }
       return originalValue;
+    },
+    apply(target, thisArg, argArray) {
+      if (typeof target === "function") {
+        return Reflect.apply(target, thisArg, argArray);
+      }
+      return target;
+    },
+    has(target, prop) {
+      return Reflect.has(target, prop);
+    },
+    ownKeys(target) {
+      return Reflect.ownKeys(target);
+    },
+    getPrototypeOf(target) {
+      return Reflect.getPrototypeOf(target);
     }
   };
   return new Proxy(model, handler);
@@ -1228,23 +1401,88 @@ 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";
+    } else if (rawId.startsWith("deepseek-")) {
+      providerName = "deepseek";
+    } else if (rawId.startsWith("grok-")) {
+      providerName = "xai";
+    } else if (rawId.startsWith("mistral-") || rawId.startsWith("codestral-") || rawId.startsWith("ministral-") || rawId.startsWith("open-mistral") || rawId.startsWith("open-mixtral")) {
+      providerName = "mistral";
+    } else if (rawId.startsWith("llama-") || rawId.startsWith("mixtral-") || rawId.startsWith("gemma") || rawId.startsWith("qwen-")) {
+      providerName = "groq";
+    } else if (rawId.startsWith("command-")) {
+      providerName = "cohere";
     }
-    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) || (providerName === "mistral" ? process.env.MISTRAL_API_KEY : void 0) || (providerName === "groq" ? process.env.GROQ_API_KEY : void 0) || (providerName === "deepseek" ? process.env.DEEPSEEK_API_KEY : void 0) || (providerName === "xai" ? process.env.XAI_API_KEY : void 0) || (providerName === "cohere" ? process.env.COHERE_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);
+          }
+        }
+      } else if (providerName === "mistral") {
+        const mod = __require("@ai-sdk/mistral");
+        const factory = mod.createMistral || mod.mistral || mod.default?.createMistral || mod.default?.mistral;
+        if (typeof factory === "function") {
+          baseModelInstance = mod.createMistral ? factory({ apiKey, baseURL })(cleanModelId) : factory(cleanModelId);
+        }
+      } else if (providerName === "groq") {
+        const mod = __require("@ai-sdk/groq");
+        const factory = mod.createGroq || mod.groq || mod.default?.createGroq || mod.default?.groq;
+        if (typeof factory === "function") {
+          baseModelInstance = mod.createGroq ? factory({ apiKey, baseURL })(cleanModelId) : factory(cleanModelId);
+        }
+      } else if (providerName === "deepseek") {
+        const mod = __require("@ai-sdk/deepseek");
+        const factory = mod.createDeepSeek || mod.deepseek || mod.default?.createDeepSeek || mod.default?.deepseek;
+        if (typeof factory === "function") {
+          baseModelInstance = mod.createDeepSeek ? factory({ apiKey, baseURL })(cleanModelId) : factory(cleanModelId);
+        }
+      } else if (providerName === "xai") {
+        const mod = __require("@ai-sdk/xai");
+        const factory = mod.createXai || mod.xai || mod.default?.createXai || mod.default?.xai;
+        if (typeof factory === "function") {
+          baseModelInstance = mod.createXai ? factory({ apiKey, baseURL })(cleanModelId) : factory(cleanModelId);
+        }
+      } else if (providerName === "cohere") {
+        const mod = __require("@ai-sdk/cohere");
+        const factory = mod.createCohere || mod.cohere || mod.default?.createCohere || mod.default?.cohere;
+        if (typeof factory === "function") {
+          baseModelInstance = mod.createCohere ? factory({ apiKey, baseURL })(cleanModelId) : factory(cleanModelId);
+        }
       }
     } catch {
       baseModelInstance = {
@@ -1323,6 +1561,23 @@ var vibezcheck = Object.assign(createVibezModel, {
   session: createVibezSession
 });
 
+// src/ai-sdk/middleware.ts
+function vibezcheckMiddleware(options = {}) {
+  return {
+    specificationVersion: "v1",
+    wrapGenerate: async ({ doGenerate, params, model }) => {
+      const target = { ...model, doGenerate };
+      const metered = withBilling(target, options);
+      return metered.doGenerate(params);
+    },
+    wrapStream: async ({ doStream, params, model }) => {
+      const target = { ...model, doStream };
+      const metered = withBilling(target, options);
+      return metered.doStream(params);
+    }
+  };
+}
+
 // src/customers/manager.ts
 import Stripe2 from "stripe";
 
@@ -1488,8 +1743,8 @@ import * as crypto from "crypto";
 import Stripe3 from "stripe";
 var ApiKeyAuth = class {
   stripe;
-  constructor(stripe) {
-    this.stripe = stripe;
+  constructor(stripe2) {
+    this.stripe = stripe2;
   }
   /**
    * Hashes a raw API key using SHA-256
@@ -1654,6 +1909,164 @@ function createBillingHelper(options = {}) {
   return new BillingHelper(options);
 }
 
+// src/billing/session.ts
+var AgentSession = class {
+  sessionId;
+  customer;
+  maxCostUSD;
+  totalCostUSD = 0;
+  totalTokens = 0;
+  toolCalls = [];
+  options;
+  constructor(options = {}) {
+    this.sessionId = `vibez_sess_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
+    this.customer = options.customer;
+    this.maxCostUSD = options.sessionBudgetUSD ?? options.maxCostUSD ?? 1;
+    this.options = options;
+  }
+  /**
+   * Returns an AI model instance bound to this agent session with cumulative budget enforcement
+   */
+  model(modelOrId, modelOptions = {}) {
+    const session = this;
+    return withBilling(modelOrId, {
+      ...modelOptions,
+      customer: this.customer,
+      pricing: modelOptions.pricing || this.options.pricing,
+      billing: modelOptions.billing || this.options.billing,
+      onUsage: async (event) => {
+        const callCost = event.cost.retailUSD ?? event.cost.billedUSD ?? event.cost.totalUSD;
+        session.totalCostUSD = Number((session.totalCostUSD + callCost).toFixed(8));
+        session.totalTokens += event.usage.totalTokens;
+        if (session.totalCostUSD > session.maxCostUSD) {
+          throw new VibezCircuitBreakerError({
+            reason: "cost_per_call_exceeded",
+            limit: session.maxCostUSD,
+            current: session.totalCostUSD,
+            model: event.model,
+            customerId: event.customerId,
+            message: `[vibezcheck] Agent Session Budget exceeded: Cumulative cost ($${session.totalCostUSD.toFixed(4)}) exceeded limit ($${session.maxCostUSD.toFixed(4)}).`
+          });
+        }
+        if (session.options.onUsage) {
+          await session.options.onUsage(event);
+        }
+      }
+    });
+  }
+  /**
+   * Tracks an external tool invocation (e.g. web search, Python code sandbox, vector query, API call)
+   */
+  async trackTool(toolName, costOrOptions, fn) {
+    const costUSD = typeof costOrOptions === "number" ? costOrOptions : costOrOptions.costUSD;
+    const latencyMs = typeof costOrOptions === "object" ? costOrOptions.latencyMs : void 0;
+    const metadata = typeof costOrOptions === "object" ? costOrOptions.metadata : void 0;
+    if (this.totalCostUSD + costUSD > this.maxCostUSD) {
+      throw new VibezCircuitBreakerError({
+        reason: "cost_per_call_exceeded",
+        limit: this.maxCostUSD,
+        current: this.totalCostUSD + costUSD,
+        model: `tool:${toolName}`,
+        message: `[vibezcheck] Agent Session Budget exceeded on tool '${toolName}': ($${(this.totalCostUSD + costUSD).toFixed(4)}) exceeded limit ($${this.maxCostUSD.toFixed(4)}).`
+      });
+    }
+    let result = void 0;
+    const start = performance.now();
+    if (fn) {
+      result = await fn();
+    }
+    const elapsed = latencyMs ?? Math.round(performance.now() - start);
+    this.totalCostUSD = Number((this.totalCostUSD + costUSD).toFixed(8));
+    this.toolCalls.push({
+      name: toolName,
+      costUSD,
+      latencyMs: elapsed,
+      metadata
+    });
+    return result;
+  }
+  getCurrentCostUSD() {
+    return this.totalCostUSD;
+  }
+  getTotalTokens() {
+    return this.totalTokens;
+  }
+  /**
+   * Concludes session and returns final financial summary
+   */
+  async conclude() {
+    return this.getSummary();
+  }
+  /**
+   * Returns current session summary
+   */
+  getSummary() {
+    return {
+      sessionId: this.sessionId,
+      customer: this.customer,
+      totalCostUSD: this.totalCostUSD,
+      totalTokens: this.totalTokens,
+      toolCallCount: this.toolCalls.length,
+      toolCalls: [...this.toolCalls]
+    };
+  }
+};
+function createAgentSession(options = {}) {
+  return new AgentSession(options);
+}
+
+// src/billing/tools.ts
+function wrapTool(options) {
+  const { name, costUSD, tool, execute } = options;
+  const targetFn = execute || tool?.execute;
+  if (!targetFn && !tool) {
+    return options;
+  }
+  const base = tool || {};
+  return {
+    ...base,
+    name,
+    costUSD,
+    execute: async (...args) => {
+      const startTime = performance.now();
+      try {
+        const result = await targetFn(...args);
+        const latencyMs = Math.round(performance.now() - startTime);
+        if (typeof globalThis.__vibezCurrentSession?.trackTool === "function") {
+          await globalThis.__vibezCurrentSession.trackTool(name, {
+            costUSD,
+            latencyMs,
+            metadata: { args: args[0] }
+          });
+        }
+        return result;
+      } catch (error) {
+        const latencyMs = Math.round(performance.now() - startTime);
+        if (typeof globalThis.__vibezCurrentSession?.trackTool === "function") {
+          await globalThis.__vibezCurrentSession.trackTool(name, {
+            costUSD,
+            latencyMs,
+            metadata: { error: String(error) }
+          });
+        }
+        throw error;
+      }
+    }
+  };
+}
+function instrumentToolKit(tools, options = {}) {
+  const instrumented = {};
+  const cost = options.costPerActionUSD ?? 5e-3;
+  for (const [toolName, toolDef] of Object.entries(tools)) {
+    instrumented[toolName] = wrapTool({
+      name: toolName,
+      costUSD: cost,
+      tool: toolDef
+    });
+  }
+  return instrumented;
+}
+
 // src/meter/index.ts
 var defaultMeter = createMeter();
 function wrapStream(stream, options) {
@@ -1663,6 +2076,299 @@ function trackTokens(response, options) {
   return defaultMeter.trackUsage(response, options);
 }
 
+// src/compat/token-meter.ts
+function createTokenMeter(stripeApiKey, config = {}) {
+  const apiKey = stripeApiKey || (typeof process !== "undefined" ? process.env?.STRIPE_SECRET_KEY || process.env?.STRIPE_API_KEY : void 0);
+  const meter = new VibezMeter({
+    apiKey,
+    eventName: config.eventName
+  });
+  const resolveCustomer = (cust) => {
+    if (typeof cust === "string") return cust;
+    return cust.customerId || cust.customer || "unknown-customer";
+  };
+  return {
+    trackUsage(response, customer) {
+      if (!response) return;
+      const customerId = resolveCustomer(customer);
+      meter.trackUsage(response, { customer: customerId });
+    },
+    track(response, customer) {
+      if (!response) return;
+      const customerId = resolveCustomer(customer);
+      meter.trackUsage(response, { customer: customerId });
+    },
+    trackUsageStreamOpenAI(stream, customer) {
+      if (!stream || typeof stream[Symbol.asyncIterator] !== "function") {
+        return stream;
+      }
+      const stripeCustomerId = resolveCustomer(customer);
+      const originalIterator = stream[Symbol.asyncIterator].bind(stream);
+      let capturedUsage = null;
+      let capturedModel = "gpt-4o";
+      const wrappedAsyncGenerator = async function* () {
+        try {
+          for await (const chunk of originalIterator()) {
+            if (chunk.model) {
+              capturedModel = chunk.model;
+            }
+            if (chunk.usage || chunk.token_usage) {
+              capturedUsage = chunk.usage || chunk.token_usage;
+            }
+            yield chunk;
+          }
+        } finally {
+          if (capturedUsage) {
+            meter.recordUsage({
+              model: capturedModel,
+              provider: "openai",
+              inputTokens: capturedUsage.prompt_tokens ?? capturedUsage.inputTokens ?? 0,
+              outputTokens: capturedUsage.completion_tokens ?? capturedUsage.outputTokens ?? 0,
+              reasoningTokens: capturedUsage.completion_tokens_details?.reasoning_tokens ?? capturedUsage.output_token_details?.reasoning_tokens ?? 0,
+              cachedTokens: capturedUsage.prompt_tokens_details?.cached_tokens ?? capturedUsage.input_token_details?.cached_tokens ?? 0,
+              customer: stripeCustomerId
+            });
+          }
+        }
+      };
+      return wrappedAsyncGenerator();
+    },
+    trackUsageStreamAnthropic(stream, customer) {
+      if (!stream || typeof stream[Symbol.asyncIterator] !== "function") {
+        return stream;
+      }
+      const stripeCustomerId = resolveCustomer(customer);
+      const originalIterator = stream[Symbol.asyncIterator].bind(stream);
+      let capturedUsage = null;
+      let capturedModel = "claude-3-5-sonnet";
+      const wrappedAsyncGenerator = async function* () {
+        try {
+          for await (const chunk of originalIterator()) {
+            if (chunk.type === "message_start" && chunk.message?.model) {
+              capturedModel = chunk.message.model;
+              if (chunk.message.usage) {
+                capturedUsage = { ...capturedUsage, ...chunk.message.usage };
+              }
+            } else if (chunk.type === "message_delta" && chunk.usage) {
+              capturedUsage = { ...capturedUsage, ...chunk.usage };
+            }
+            yield chunk;
+          }
+        } finally {
+          if (capturedUsage) {
+            meter.recordUsage({
+              model: capturedModel,
+              provider: "anthropic",
+              inputTokens: capturedUsage.input_tokens ?? 0,
+              outputTokens: capturedUsage.output_tokens ?? 0,
+              cachedTokens: capturedUsage.cache_read_input_tokens ?? 0,
+              customer: stripeCustomerId
+            });
+          }
+        }
+      };
+      return wrappedAsyncGenerator();
+    },
+    trackUsageStreamGemini(streamResult, customer, modelName = "gemini-2.0-flash") {
+      const stripeCustomerId = resolveCustomer(customer);
+      const originalStream = streamResult?.stream || streamResult;
+      if (!originalStream || typeof originalStream[Symbol.asyncIterator] !== "function") {
+        return streamResult;
+      }
+      const wrappedAsyncGenerator = async function* () {
+        let lastUsageMetadata = null;
+        try {
+          for await (const chunk of originalStream) {
+            if (chunk.usageMetadata) {
+              lastUsageMetadata = chunk.usageMetadata;
+            }
+            yield chunk;
+          }
+        } finally {
+          if (lastUsageMetadata) {
+            meter.recordUsage({
+              model: modelName,
+              provider: "google",
+              inputTokens: lastUsageMetadata.promptTokenCount ?? 0,
+              outputTokens: (lastUsageMetadata.candidatesTokenCount ?? 0) + (lastUsageMetadata.thoughtsTokenCount ?? 0),
+              reasoningTokens: lastUsageMetadata.thoughtsTokenCount ?? 0,
+              customer: stripeCustomerId
+            });
+          }
+        }
+      };
+      if (streamResult && streamResult.stream) {
+        return {
+          ...streamResult,
+          stream: wrappedAsyncGenerator()
+        };
+      }
+      return wrappedAsyncGenerator();
+    },
+    trackUsageStreamDeepSeek(stream, customer, modelName = "deepseek-chat") {
+      const stripeCustomerId = resolveCustomer(customer);
+      if (!stream || typeof stream[Symbol.asyncIterator] !== "function") {
+        return stream;
+      }
+      const originalIterator = stream[Symbol.asyncIterator].bind(stream);
+      let capturedUsage = null;
+      let capturedModel = modelName;
+      const wrappedAsyncGenerator = async function* () {
+        try {
+          for await (const chunk of originalIterator()) {
+            if (chunk.model) capturedModel = chunk.model;
+            if (chunk.usage) capturedUsage = chunk.usage;
+            yield chunk;
+          }
+        } finally {
+          if (capturedUsage) {
+            meter.recordUsage({
+              model: capturedModel,
+              provider: "deepseek",
+              inputTokens: capturedUsage.prompt_tokens ?? capturedUsage.inputTokens ?? 0,
+              outputTokens: capturedUsage.completion_tokens ?? capturedUsage.outputTokens ?? 0,
+              reasoningTokens: capturedUsage.completion_tokens_details?.reasoning_tokens ?? 0,
+              cachedTokens: capturedUsage.prompt_cache_hit_tokens ?? capturedUsage.prompt_tokens_details?.cached_tokens ?? 0,
+              customer: stripeCustomerId
+            });
+          }
+        }
+      };
+      return wrappedAsyncGenerator();
+    },
+    trackUsageStreamGroq(stream, customer, modelName = "llama-3.3-70b-versatile") {
+      const stripeCustomerId = resolveCustomer(customer);
+      if (!stream || typeof stream[Symbol.asyncIterator] !== "function") {
+        return stream;
+      }
+      const originalIterator = stream[Symbol.asyncIterator].bind(stream);
+      let capturedUsage = null;
+      let capturedModel = modelName;
+      const wrappedAsyncGenerator = async function* () {
+        try {
+          for await (const chunk of originalIterator()) {
+            if (chunk.model) capturedModel = chunk.model;
+            if (chunk.usage || chunk.x_groq?.usage) {
+              capturedUsage = chunk.usage || chunk.x_groq?.usage;
+            }
+            yield chunk;
+          }
+        } finally {
+          if (capturedUsage) {
+            meter.recordUsage({
+              model: capturedModel,
+              provider: "groq",
+              inputTokens: capturedUsage.prompt_tokens ?? 0,
+              outputTokens: capturedUsage.completion_tokens ?? 0,
+              customer: stripeCustomerId
+            });
+          }
+        }
+      };
+      return wrappedAsyncGenerator();
+    },
+    trackUsageStreamMistral(stream, customer, modelName = "mistral-large-latest") {
+      const stripeCustomerId = resolveCustomer(customer);
+      if (!stream || typeof stream[Symbol.asyncIterator] !== "function") {
+        return stream;
+      }
+      const originalIterator = stream[Symbol.asyncIterator].bind(stream);
+      let capturedUsage = null;
+      let capturedModel = modelName;
+      const wrappedAsyncGenerator = async function* () {
+        try {
+          for await (const chunk of originalIterator()) {
+            if (chunk.model) capturedModel = chunk.model;
+            if (chunk.usage) capturedUsage = chunk.usage;
+            yield chunk;
+          }
+        } finally {
+          if (capturedUsage) {
+            meter.recordUsage({
+              model: capturedModel,
+              provider: "mistral",
+              inputTokens: capturedUsage.prompt_tokens ?? 0,
+              outputTokens: capturedUsage.completion_tokens ?? 0,
+              customer: stripeCustomerId
+            });
+          }
+        }
+      };
+      return wrappedAsyncGenerator();
+    },
+    trackUsageStream(stream, customer, options = {}) {
+      const stripeCustomerId = resolveCustomer(customer);
+      return meter.wrapStream(stream, {
+        customer: stripeCustomerId,
+        model: options.model,
+        provider: options.provider
+      });
+    }
+  };
+}
+
+// src/compat/stripe-provider.ts
+function createStripe(config = {}) {
+  const apiKey = config.apiKey || (typeof process !== "undefined" ? process.env?.STRIPE_API_KEY || process.env?.STRIPE_SECRET_KEY : void 0);
+  const provider = function(modelId, settings = {}) {
+    const customerId = settings.customerId || config.customerId || "anonymous";
+    const options = {
+      customer: customerId,
+      stripeApiKey: apiKey
+    };
+    if (settings.model && typeof settings.model === "object") {
+      return withBilling(settings.model, options);
+    }
+    return createVibezModel(modelId, {
+      ...settings,
+      customer: customerId,
+      stripeApiKey: apiKey,
+      baseURL: config.baseURL
+    });
+  };
+  provider.languageModel = provider;
+  provider.specificationVersion = "v3";
+  return provider;
+}
+var stripe = createStripe();
+var createStripeV3 = createStripe;
+var stripeV3 = stripe;
+
+// src/compat/stripe-meter.ts
+function meteredModel2(model, arg2, arg3, arg4) {
+  if (!model || typeof model !== "object") {
+    throw new Error("[vibezcheck / Stripe AI] Invalid model provided to meteredModel().");
+  }
+  let stripeApiKey;
+  let stripeCustomerId;
+  let options = {};
+  if (typeof arg2 === "object" && arg2 !== null) {
+    options = { ...arg2 };
+    stripeCustomerId = options.customerId || (typeof options.customer === "string" ? options.customer : options.customer?.id);
+    stripeApiKey = options.stripeApiKey;
+  } else if (typeof arg2 === "string") {
+    if (typeof arg3 === "string") {
+      stripeApiKey = arg2;
+      stripeCustomerId = arg3;
+      options = { ...arg4 || {} };
+    } else {
+      stripeCustomerId = arg2;
+      options = { ...typeof arg3 === "object" ? arg3 : {} };
+    }
+  }
+  stripeApiKey = stripeApiKey || options.stripeApiKey || (typeof process !== "undefined" ? process.env?.STRIPE_API_KEY || process.env?.STRIPE_SECRET_KEY : void 0);
+  if (typeof process !== "undefined" && stripeApiKey && !process.env.STRIPE_API_KEY && !process.env.STRIPE_SECRET_KEY) {
+    process.env.STRIPE_API_KEY = stripeApiKey;
+  }
+  const mergedOptions = {
+    ...options,
+    customer: stripeCustomerId || options.customer,
+    customerId: stripeCustomerId,
+    stripeApiKey
+  };
+  return withBilling(model, mergedOptions);
+}
+
 // src/index.ts
 var VibezCheckClient = class {
   meter;
@@ -1738,21 +2444,31 @@ function vibezcheck2(firstArg, secondArg) {
   return new VibezCheckClient(firstArg || {});
 }
 vibezcheck2.calculateCost = calculateCost;
+vibezcheck2.calculateUsageCost = calculateUsageCost;
 vibezcheck2.getModelPricing = getModelPricing;
 vibezcheck2.registerModelPricing = registerModelPricing;
 vibezcheck2.create = createVibezCheck;
 vibezcheck2.withBilling = withBilling;
 vibezcheck2.createMeter = createMeter;
-vibezcheck2.session = createVibezSession;
+vibezcheck2.session = createAgentSession;
+vibezcheck2.Session = AgentSession;
+vibezcheck2.wrapTool = wrapTool;
+vibezcheck2.instrumentToolKit = instrumentToolKit;
+vibezcheck2.middleware = vibezcheckMiddleware;
+vibezcheck2.Billing = BillingHelper;
+vibezcheck2.Auth = ApiKeyAuth;
+vibezcheck2.Customers = CustomerManager;
 var vibez = new VibezCheckClient();
 var vibescheck = vibezcheck2;
 var vibes = vibez;
 export {
+  AgentSession,
   AnthropicStreamAccumulator,
   ApiKeyAuth,
   BillingHelper,
   CustomerCache,
   CustomerManager,
+  MODEL_ALIASES,
   MODEL_PRICING_TABLE,
   MeterBatcher,
   VibezCheckClient,
@@ -1760,10 +2476,14 @@ export {
   VibezMeter,
   calculateCost,
   calculateUsageCost,
+  createAgentSession,
   createApiKeyAuth,
   createBillingHelper,
   createCustomerManager,
   createMeter,
+  createStripe,
+  createStripeV3,
+  createTokenMeter,
   createVibezCheck,
   createVibezModel,
   createVibezSession,
@@ -1775,20 +2495,26 @@ export {
   extractOpenAIResponseUsage,
   getModelPricing,
   inspectOpenAIStreamChunk,
+  instrumentToolKit,
   meteredModel,
   normalizeCustomer,
   normalizeModelKey,
   registerModelPricing,
+  stripe,
+  meteredModel2 as stripeMeteredModel,
+  stripeV3,
   trackTokens,
   vibes,
   vibescheck,
   vibez,
   vibezcheck2 as vibezcheck,
+  vibezcheckMiddleware,
   withBilling,
   wrapAnthropicStream,
   wrapGeminiStream,
   wrapOpenAIStream,
   wrapStream,
+  wrapTool,
   wrapUniversalStream
 };
 //# sourceMappingURL=index.mjs.map