@mux/ai 0.25.0 → 0.27.1

package/dist/primitives/index.js

@@ -414,7 +414,7 @@ var DEFAULT_LANGUAGE_MODELS = {
 var LANGUAGE_MODELS = {
   openai: ["gpt-5.1", "gpt-5-mini"],
   anthropic: ["claude-sonnet-4-5"],
-  google: ["gemini-3-flash-preview", "gemini-3.1-flash-lite", "gemini-3.1-flash-lite-preview", "gemini-2.5-flash"]
+  google: ["gemini-3-flash-preview", "gemini-3.1-flash-lite", "gemini-2.5-flash"]
 };
 var LANGUAGE_MODEL_DEPRECATIONS = [
   {
@@ -430,10 +430,10 @@ var LANGUAGE_MODEL_DEPRECATIONS = [
     provider: "google",
     modelId: "gemini-3.1-flash-lite-preview",
     replacementModelId: "gemini-3.1-flash-lite",
-    phase: "warn",
+    phase: "blocked",
     deprecatedOn: "2026-05-12",
     sunsetOn: "2026-05-25",
-    reason: "Preview model graduated to the stable Gemini 3.1 Flash-Lite release."
+    reason: "Google shut down Gemini 3.1 Flash-Lite Preview on 2026-05-25."
   }
 ];
 var warnedDeprecatedLanguageModels = /* @__PURE__ */ new Set();
@@ -792,6 +792,45 @@ function detectLeakReason(text) {
   return null;
 }
 
+// src/lib/token-usage.ts
+var TOKEN_USAGE_FIELDS = [
+  "inputTokens",
+  "outputTokens",
+  "totalTokens",
+  "reasoningTokens",
+  "cachedInputTokens"
+];
+function getErrorTokenUsage(error) {
+  if (typeof error !== "object" || error === null) {
+    return void 0;
+  }
+  const usage = error.usage;
+  if (typeof usage !== "object" || usage === null) {
+    return void 0;
+  }
+  const source = usage;
+  const extracted = {};
+  for (const field of TOKEN_USAGE_FIELDS) {
+    const value = source[field];
+    if (typeof value === "number") {
+      extracted[field] = value;
+    }
+  }
+  if (extracted.reasoningTokens === void 0) {
+    const reasoningTokens = source.outputTokenDetails?.reasoningTokens;
+    if (typeof reasoningTokens === "number") {
+      extracted.reasoningTokens = reasoningTokens;
+    }
+  }
+  if (extracted.cachedInputTokens === void 0) {
+    const cacheReadTokens = source.inputTokenDetails?.cacheReadTokens;
+    if (typeof cacheReadTokens === "number") {
+      extracted.cachedInputTokens = cacheReadTokens;
+    }
+  }
+  return Object.keys(extracted).length > 0 ? extracted : void 0;
+}
+
 // src/lib/mux-ai-error.ts
 var MuxAiError = class extends Error {
   constructor(message, opts) {
@@ -804,18 +843,38 @@ var MuxAiError = class extends Error {
     this.retryable = opts?.retryable ?? false;
   }
 };
+function extractErrorDetail(error) {
+  if (error instanceof Error && error.message) {
+    return error.message;
+  }
+  if (typeof error === "string" && error.length > 0) {
+    return error;
+  }
+  if (typeof error === "object" && error !== null) {
+    const message = error.message;
+    if (typeof message === "string" && message.length > 0) {
+      return message;
+    }
+  }
+  return "Unknown error";
+}
 function wrapError(error, message) {
   if (error instanceof MuxAiError) {
     throw error;
   }
-  const rawDetail = error instanceof Error ? error.message : "Unknown error";
+  const rawDetail = extractErrorDetail(error);
   const leakReason = detectLeakReason(rawDetail);
   const shouldSuppress = leakReason === "canary" || leakReason === "prompt_tag";
   const detail = shouldSuppress ? "Upstream error details suppressed by safety filter" : rawDetail;
   if (shouldSuppress) {
     console.warn(`[@mux/ai] Suppressed suspected prompt leak in wrapped error (context: ${message}, reason: ${leakReason}).`);
   }
-  throw new Error(`${message}: ${detail}`);
+  const wrapped = new Error(`${message}: ${detail}`);
+  const usage = getErrorTokenUsage(error);
+  if (usage) {
+    wrapped.usage = usage;
+  }
+  throw wrapped;
 }
 
 // src/primitives/shots.ts