vibezcheck 0.5.0 → 0.5.3

package/README.md

@@ -39,7 +39,7 @@ const model = vibezcheck('openai/gpt-4o', { customer: 'cus_alex' });
 ## 📦 Installation
 
 ```bash
-npm install vibezcheck ai @ai-sdk/openai stripe
+npm install vibezcheck ai @ai-sdk/openai @ai-sdk/react stripe
 ```
 
 ---
@@ -74,38 +74,47 @@ export async function POST(req: Request) {
 
 ```tsx
 'use client';
-import { useChat } from 'ai/react';
+
+import { useChat } from '@ai-sdk/react';
+import { useState } from 'react';
 import { VibezCheck } from 'vibezcheck/ui';
 
-export default function ChatPage() {
-  const { messages, input, handleInputChange, handleSubmit } = useChat();
+export default function Chat() {
+  const [input, setInput] = useState('');
+  const { messages, sendMessage } = useChat();
 
   return (
-    <main className="max-w-2xl mx-auto p-6 space-y-4">
-      <div className="space-y-3">
-        {messages.map((m) => (
-          <div key={m.id} className="p-3 rounded-lg bg-slate-100 dark:bg-slate-800">
-            <strong>{m.role}: </strong>
-            <span>{m.content}</span>
-          </div>
-        ))}
-      </div>
-
-      <form onSubmit={handleSubmit} className="flex gap-2">
+    <div className="flex flex-col w-full max-w-md py-24 mx-auto stretch">
+      {messages.map(message => (
+        <div key={message.id} className="whitespace-pre-wrap">
+          {message.role === 'user' ? 'User: ' : 'AI: '}
+          {message.parts.map((part, i) => {
+            switch (part.type) {
+              case 'text':
+                return <div key={`${message.id}-${i}`}>{part.text}</div>;
+            }
+          })}
+        </div>
+      ))}
+
+      <form
+        onSubmit={e => {
+          e.preventDefault();
+          sendMessage({ text: input });
+          setInput('');
+        }}
+      >
         <input
+          className="fixed dark:bg-zinc-900 bottom-0 w-full max-w-md p-2 mb-8 border border-zinc-300 dark:border-zinc-800 rounded shadow-xl"
           value={input}
-          onChange={handleInputChange}
-          placeholder="Ask anything..."
-          className="flex-1 p-2 border rounded-lg"
+          placeholder="Say something..."
+          onChange={e => setInput(e.currentTarget.value)}
         />
-        <button type="submit" className="px-4 py-2 bg-blue-600 text-white rounded-lg">
-          Send
-        </button>
       </form>
 
-      {/* ✦ 1-Line HUD: Works with zero database & zero Stripe in Dev Mode */}
-      <VibezCheck messages={messages} model="gpt-4o-mini" margin={1.3} />
-    </main>
+      {/* ✦ 1-Line Token Meter & Financial HUD: Works with zero database & zero Stripe in Dev Mode */}
+      <VibezCheck messages={messages} />
+    </div>
   );
 }
 ```
@@ -225,6 +234,16 @@ npx vibezcheck audit --ci
 
 ---
 
+## 📂 Examples
+
+* **[Next.js App Router + OpenAI](examples/nextjs-app-router-openai)**: Direct implementation of the official [AI SDK Next.js App Router Tutorial](https://ai-sdk.dev/docs/getting-started/nextjs-app-router) with 1-line VibezCheck token metering.
+* **[Next.js 15 AI SaaS Starter](examples/nextjs-saas-starter)**: Complete SaaS template with Stripe Checkout, Top-Up Modal, and `<VibezReceipt />`.
+* **[Local Cost Meter](examples/01-local-cost-meter.ts)**: Offline micro-transaction calculations with BigInt Nano-USD precision.
+* **[Native OpenAI Streaming](examples/03-native-openai-stream.ts)**: Zero-latency stream wrapper for the official `openai` Node SDK.
+* **[Claude 3.7 Reasoning Tokens](examples/04-claude-thinking.ts)**: Tracking hidden thought tokens with Anthropic SDK.
+
+---
+
 ## 📄 License
 
 MIT © [VibezCheck](https://vibezcheck.xyz)

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

@@ -1,7 +1,7 @@
-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 { W as WithBillingOptions } from '../with-billing-DndWGxL8.mjs';
+export { m as meteredModel, w as withBilling } from '../with-billing-DndWGxL8.mjs';
+import { C as CustomerParam } from '../types-6P71CXAD.mjs';
+import { V as VibezMeter } from '../client-CbuXmoXz.mjs';
 import 'stripe';
 
 interface VibezCheckModelOptions extends WithBillingOptions {

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

@@ -1,7 +1,7 @@
-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 { W as WithBillingOptions } from '../with-billing-Bj6Ya_Iy.js';
+export { m as meteredModel, w as withBilling } from '../with-billing-Bj6Ya_Iy.js';
+import { C as CustomerParam } from '../types-6P71CXAD.js';
+import { V as VibezMeter } from '../client-BORmJy8s.js';
 import 'stripe';
 
 interface VibezCheckModelOptions extends WithBillingOptions {

package/dist/ai-sdk/index.js

@@ -309,12 +309,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: {
@@ -329,11 +329,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: {
@@ -498,7 +498,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 },
@@ -510,11 +510,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 },
@@ -522,48 +533,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) {
@@ -580,6 +688,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,
@@ -858,7 +970,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);
@@ -880,7 +992,7 @@ var VibezMeter = class {
       this.stripeClient = new import_stripe.default(key, {
         appInfo: {
           name: "vibezcheck",
-          version: "0.5.0",
+          version: "0.5.3",
           url: "https://vibezcheck.xyz"
         }
       });
@@ -987,10 +1099,20 @@ function createMeter(options = {}) {
 }
 
 // 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
@@ -1019,17 +1141,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,
@@ -1130,7 +1254,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") {
@@ -1146,7 +1272,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") {
@@ -1161,7 +1289,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") {
@@ -1216,14 +1346,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() {
@@ -1249,13 +1384,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);
@@ -1279,8 +1430,18 @@ function createVibezModel(modelOrId, options = {}) {
       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 || (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 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 {
@@ -1317,6 +1478,36 @@ function createVibezModel(modelOrId, options = {}) {
             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 = {