vibezcheck 0.5.0 → 0.5.4

package/dist/ai-sdk/index.js

@@ -309,12 +309,38 @@ 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.type === "finish" ? chunk.usage : null) || chunk.usage || chunk.token_usage || chunk.x_groq?.usage || chunk.usageMetadata;
+  if (rawUsage) {
+    let inputTokens = 0;
+    let outputTokens = 0;
+    let reasoningTokens = 0;
+    let cachedTokens = 0;
+    if (typeof rawUsage.inputTokens === "object" && rawUsage.inputTokens !== null) {
+      inputTokens = rawUsage.inputTokens.total ?? rawUsage.inputTokens.noCache ?? 0;
+      cachedTokens = rawUsage.inputTokens.cacheRead ?? 0;
+    } else if (typeof rawUsage.promptTokens === "number") {
+      inputTokens = rawUsage.promptTokens;
+    } else if (typeof rawUsage.inputTokens === "number") {
+      inputTokens = rawUsage.inputTokens;
+    } else {
+      inputTokens = rawUsage.prompt_tokens ?? rawUsage.input_tokens ?? rawUsage.promptTokenCount ?? 0;
+    }
+    if (typeof rawUsage.outputTokens === "object" && rawUsage.outputTokens !== null) {
+      outputTokens = rawUsage.outputTokens.total ?? rawUsage.outputTokens.text ?? 0;
+      reasoningTokens = rawUsage.outputTokens.reasoning ?? 0;
+    } else if (typeof rawUsage.completionTokens === "number") {
+      outputTokens = rawUsage.completionTokens;
+    } else if (typeof rawUsage.outputTokens === "number") {
+      outputTokens = rawUsage.outputTokens;
+    } else {
+      outputTokens = rawUsage.completion_tokens ?? rawUsage.output_tokens ?? rawUsage.candidatesTokenCount ?? 0;
+    }
+    if (!reasoningTokens) {
+      reasoningTokens = rawUsage.completion_tokens_details?.reasoning_tokens ?? rawUsage.output_token_details?.reasoning_tokens ?? rawUsage.reasoning_tokens ?? rawUsage.reasoningTokens ?? rawUsage.thoughtsTokenCount ?? 0;
+    }
+    if (!cachedTokens) {
+      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 +355,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: {
@@ -460,10 +486,36 @@ function extractGenericResponseUsage(response, fallbackModel = "generic-llm", fa
   if (!response || typeof response !== "object") return null;
   const usage = response.usage || response.token_usage || response.usageMetadata;
   if (usage) {
-    const inputTokens = usage.prompt_tokens ?? usage.input_tokens ?? usage.promptTokenCount ?? usage.prompt_eval_count ?? 0;
-    const outputTokens = usage.completion_tokens ?? usage.output_tokens ?? usage.candidatesTokenCount ?? usage.eval_count ?? 0;
-    const reasoningTokens = usage.reasoning_tokens ?? usage.thoughtsTokenCount ?? usage.completion_tokens_details?.reasoning_tokens ?? 0;
-    const cachedTokens = usage.prompt_tokens_details?.cached_tokens ?? usage.cached_tokens ?? usage.cachedContentTokenCount ?? 0;
+    let inputTokens = 0;
+    let outputTokens = 0;
+    let reasoningTokens = 0;
+    let cachedTokens = 0;
+    if (typeof usage.inputTokens === "object" && usage.inputTokens !== null) {
+      inputTokens = usage.inputTokens.total ?? usage.inputTokens.noCache ?? 0;
+      cachedTokens = usage.inputTokens.cacheRead ?? 0;
+    } else if (typeof usage.promptTokens === "number") {
+      inputTokens = usage.promptTokens;
+    } else if (typeof usage.inputTokens === "number") {
+      inputTokens = usage.inputTokens;
+    } else {
+      inputTokens = usage.prompt_tokens ?? usage.input_tokens ?? usage.promptTokenCount ?? usage.prompt_eval_count ?? 0;
+    }
+    if (typeof usage.outputTokens === "object" && usage.outputTokens !== null) {
+      outputTokens = usage.outputTokens.total ?? usage.outputTokens.text ?? 0;
+      reasoningTokens = usage.outputTokens.reasoning ?? 0;
+    } else if (typeof usage.completionTokens === "number") {
+      outputTokens = usage.completionTokens;
+    } else if (typeof usage.outputTokens === "number") {
+      outputTokens = usage.outputTokens;
+    } else {
+      outputTokens = usage.completion_tokens ?? usage.output_tokens ?? usage.candidatesTokenCount ?? usage.eval_count ?? 0;
+    }
+    if (!reasoningTokens) {
+      reasoningTokens = usage.reasoning_tokens ?? usage.thoughtsTokenCount ?? usage.completion_tokens_details?.reasoning_tokens ?? 0;
+    }
+    if (!cachedTokens) {
+      cachedTokens = usage.prompt_tokens_details?.cached_tokens ?? usage.cached_tokens ?? usage.cachedContentTokenCount ?? 0;
+    }
     const model = response.model || fallbackModel;
     const provider = response.provider || fallbackProvider;
     return {
@@ -498,7 +550,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 +562,26 @@ 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 },
+  "gpt-6-astra": { inputPer1M: 5, outputPer1M: 25, cachedInputPer1M: 0.5 },
+  "gpt-6-astra-fast": { inputPer1M: 3, outputPer1M: 15, cachedInputPer1M: 0.3 },
+  "gpt-image-2.5-flare": { inputPer1M: 0, outputPer1M: 0.04 },
+  "gpt-image-2.5-sunburst": { inputPer1M: 0, outputPer1M: 0.08 },
+  // --- 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 +589,172 @@ 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-medium-3-5": { inputPer1M: 2, outputPer1M: 6, cachedInputPer1M: 0.4 },
+  "mistral-medium-2604": { inputPer1M: 2, outputPer1M: 6, cachedInputPer1M: 0.4 },
+  "magistral-small-latest": { inputPer1M: 0.5, outputPer1M: 1.5, cachedInputPer1M: 0.1 },
+  "magistral-medium-latest": { inputPer1M: 2, outputPer1M: 6, cachedInputPer1M: 0.4 },
   "mistral-small-latest": { inputPer1M: 0.2, outputPer1M: 0.6 },
+  "mistral-small-2603": { inputPer1M: 0.2, outputPer1M: 0.6, cachedInputPer1M: 0.05 },
+  "mistral-vibe-cli-fast": { inputPer1M: 0.2, outputPer1M: 0.6, cachedInputPer1M: 0.05 },
+  "mistral-vibe-cli-latest": { inputPer1M: 0.2, outputPer1M: 0.6, cachedInputPer1M: 0.05 },
+  "codestral-2508": { inputPer1M: 0.3, outputPer1M: 0.9, cachedInputPer1M: 0.08 },
+  "codestral-embed": { inputPer1M: 0.1, outputPer1M: 0 },
+  "mistral-embed-2312": { inputPer1M: 0.1, outputPer1M: 0 },
+  "zai-glm-5-2": { inputPer1M: 0.6, outputPer1M: 2.2, cachedInputPer1M: 0.15 },
+  "glm-5-2": { inputPer1M: 0.6, outputPer1M: 2.2, cachedInputPer1M: 0.15 },
+  "labs-leanstral-1-5": { inputPer1M: 0.2, outputPer1M: 0.6, cachedInputPer1M: 0.05 },
   "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.1-flash": { inputPer1M: 0.2, outputPer1M: 0.6, cachedInputPer1M: 0.05 },
   "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 },
+  // --- Gateway, Inception & InclusionAI Models ---
+  "mercury-2.5": { inputPer1M: 0.25, outputPer1M: 0.75, cachedInputPer1M: 0.05 },
+  "mercury-2": { inputPer1M: 0.25, outputPer1M: 0.75, cachedInputPer1M: 0.05 },
+  "ling-3.0-flash-sante": { inputPer1M: 0.15, outputPer1M: 0.45, cachedInputPer1M: 0.03 },
+  "ling-3.0-flash": { inputPer1M: 0.15, outputPer1M: 0.45, cachedInputPer1M: 0.03 },
   // --- 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",
+  "magistral": "magistral-small-latest",
+  "magistral-small": "magistral-small-latest",
+  "magistral-medium": "magistral-medium-latest",
+  "mistral-3.5": "mistral-medium-3-5",
+  "mistral-medium-3.5": "mistral-medium-3-5",
+  "gpt-image-2.5": "gpt-image-2.5-flare",
+  "gpt-image-sunburst": "gpt-image-2.5-sunburst",
+  "gpt-image-flare": "gpt-image-2.5-flare"
 };
 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 +771,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 +1053,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 +1075,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 +1182,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 +1224,49 @@ 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;
+    let inputTokens = 0;
+    let outputTokens = 0;
+    let reasoningTokens = 0;
+    let cachedTokens = 0;
+    let cacheWriteTokens = 0;
+    if (typeof rawUsage.inputTokens === "object" && rawUsage.inputTokens !== null) {
+      inputTokens = rawUsage.inputTokens.total ?? rawUsage.inputTokens.noCache ?? 0;
+      cachedTokens = rawUsage.inputTokens.cacheRead ?? 0;
+      cacheWriteTokens = rawUsage.inputTokens.cacheWrite ?? 0;
+    } else if (typeof rawUsage.promptTokens === "number") {
+      inputTokens = rawUsage.promptTokens;
+    } else if (typeof rawUsage.inputTokens === "number") {
+      inputTokens = rawUsage.inputTokens;
+    } else {
+      inputTokens = rawUsage.prompt_tokens ?? rawUsage.input_tokens ?? rawUsage.promptTokenCount ?? 0;
+    }
+    if (typeof rawUsage.outputTokens === "object" && rawUsage.outputTokens !== null) {
+      outputTokens = rawUsage.outputTokens.total ?? rawUsage.outputTokens.text ?? 0;
+      reasoningTokens = rawUsage.outputTokens.reasoning ?? 0;
+    } else if (typeof rawUsage.completionTokens === "number") {
+      outputTokens = rawUsage.completionTokens;
+    } else if (typeof rawUsage.outputTokens === "number") {
+      outputTokens = rawUsage.outputTokens;
+    } else {
+      outputTokens = rawUsage.completion_tokens ?? rawUsage.output_tokens ?? rawUsage.candidatesTokenCount ?? 0;
+    }
+    if (!reasoningTokens) {
+      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;
+    }
+    if (!cachedTokens) {
+      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;
+    }
+    if (!cacheWriteTokens) {
+      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 +1367,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 +1385,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 +1402,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 +1459,25 @@ 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;
+              if (chunk.type === "text-delta" && typeof chunk.delta === "string") {
+                accumulatedChars += chunk.delta.length;
+              } else if (chunk.type === "reasoning-delta" && typeof chunk.delta === "string") {
+                accumulatedChars += chunk.delta.length;
+              } else {
+                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.type === "finish" ? chunk.usage : null) || chunk.usage || chunk.token_usage || chunk.usageMetadata;
+              if (chunkUsage) {
                 streamCompleted = true;
-                handleUsage(chunk.usage);
+                handleUsage(chunkUsage);
               }
             },
             flush() {
@@ -1249,13 +1503,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 +1549,22 @@ 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("magistral-") || rawId.startsWith("voxtral-") || rawId.startsWith("devstral-") || 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";
+    } else if (rawId.startsWith("mercury-") || rawId.startsWith("ling-")) {
+      providerName = "gateway";
+    } else if (rawId.startsWith("gpt-image-") || rawId.startsWith("gpt-6-")) {
+      providerName = "openai";
     }
-    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 +1601,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 = {