@fallom/trace 0.1.11 → 0.1.12

package/dist/index.d.mts

@@ -214,12 +214,20 @@ declare function wrapGoogleAI<T extends {
  * }); // Automatically traced!
  * ```
  */
+/** Options for wrapAISDK */
+interface WrapAISDKOptions {
+    /**
+     * Enable debug logging to see the raw Vercel AI SDK response structure.
+     * Useful for debugging missing usage/token data.
+     */
+    debug?: boolean;
+}
 declare function wrapAISDK<T extends {
     generateText: (...args: any[]) => Promise<any>;
     streamText: (...args: any[]) => any;
     generateObject?: (...args: any[]) => Promise<any>;
     streamObject?: (...args: any[]) => any;
-}>(ai: T): {
+}>(ai: T, options?: WrapAISDKOptions): {
     generateText: T["generateText"];
     streamText: T["streamText"];
     generateObject: T["generateObject"];

package/dist/index.d.ts

@@ -214,12 +214,20 @@ declare function wrapGoogleAI<T extends {
  * }); // Automatically traced!
  * ```
  */
+/** Options for wrapAISDK */
+interface WrapAISDKOptions {
+    /**
+     * Enable debug logging to see the raw Vercel AI SDK response structure.
+     * Useful for debugging missing usage/token data.
+     */
+    debug?: boolean;
+}
 declare function wrapAISDK<T extends {
     generateText: (...args: any[]) => Promise<any>;
     streamText: (...args: any[]) => any;
     generateObject?: (...args: any[]) => Promise<any>;
     streamObject?: (...args: any[]) => any;
-}>(ai: T): {
+}>(ai: T, options?: WrapAISDKOptions): {
     generateText: T["generateText"];
     streamText: T["streamText"];
     generateObject: T["generateObject"];

package/dist/index.js

@@ -1194,7 +1194,13 @@ function wrapOpenAI(client) {
         response?.choices?.[0]?.message,
         response?.model || params?.model,
         response?.id
-      ) : void 0;
+      ) : {};
+      if (response?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(response.usage);
+      }
+      if (response?.choices?.[0]?.finish_reason) {
+        attributes["gen_ai.response.finish_reason"] = response.choices[0].finish_reason;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1212,7 +1218,7 @@ function wrapOpenAI(client) {
         prompt_tokens: response?.usage?.prompt_tokens,
         completion_tokens: response?.usage?.completion_tokens,
         total_tokens: response?.usage?.total_tokens,
-        attributes,
+        attributes: Object.keys(attributes).length > 0 ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1285,10 +1291,16 @@ function wrapAnthropic(client) {
         { role: "assistant", content: response?.content?.[0]?.text || "" },
         response?.model || params?.model,
         response?.id
-      ) : void 0;
-      if (attributes && params?.system) {
+      ) : {};
+      if (params?.system) {
         attributes["gen_ai.system_prompt"] = params.system;
       }
+      if (response?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(response.usage);
+      }
+      if (response?.stop_reason) {
+        attributes["gen_ai.response.finish_reason"] = response.stop_reason;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1306,7 +1318,7 @@ function wrapAnthropic(client) {
         prompt_tokens: response?.usage?.input_tokens,
         completion_tokens: response?.usage?.output_tokens,
         total_tokens: (response?.usage?.input_tokens || 0) + (response?.usage?.output_tokens || 0),
-        attributes,
+        attributes: Object.keys(attributes).length > 0 ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1399,6 +1411,13 @@ function wrapGoogleAI(model) {
           attributes["gen_ai.completion.0.content"] = outputText;
         }
       }
+      if (usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(usage);
+      }
+      const candidate = result?.candidates?.[0];
+      if (candidate?.finishReason) {
+        attributes["gen_ai.response.finish_reason"] = candidate.finishReason;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1416,7 +1435,7 @@ function wrapGoogleAI(model) {
         prompt_tokens: usage?.promptTokenCount,
         completion_tokens: usage?.candidatesTokenCount,
         total_tokens: usage?.totalTokenCount,
-        attributes: captureContent ? attributes : void 0,
+        attributes: Object.keys(attributes).length > 0 ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1464,8 +1483,37 @@ function wrapGoogleAI(model) {
   };
   return model;
 }
-function wrapAISDK(ai) {
+var aiSdkDebug = false;
+function extractUsageFromResult(result, directUsage) {
+  let usage = directUsage ?? result?.usage;
+  const isValidNumber = (v) => v !== null && v !== void 0 && !Number.isNaN(v);
+  let promptTokens = isValidNumber(usage?.promptTokens) ? usage.promptTokens : void 0;
+  let completionTokens = isValidNumber(usage?.completionTokens) ? usage.completionTokens : void 0;
+  let totalTokens = isValidNumber(usage?.totalTokens) ? usage.totalTokens : void 0;
+  let cost;
+  const orUsage = result?.experimental_providerMetadata?.openrouter?.usage;
+  if (orUsage) {
+    if (promptTokens === void 0 && isValidNumber(orUsage.promptTokens)) {
+      promptTokens = orUsage.promptTokens;
+    }
+    if (completionTokens === void 0 && isValidNumber(orUsage.completionTokens)) {
+      completionTokens = orUsage.completionTokens;
+    }
+    if (totalTokens === void 0 && isValidNumber(orUsage.totalTokens)) {
+      totalTokens = orUsage.totalTokens;
+    }
+    if (isValidNumber(orUsage.cost)) {
+      cost = orUsage.cost;
+    }
+  }
+  if (totalTokens === void 0 && (promptTokens !== void 0 || completionTokens !== void 0)) {
+    totalTokens = (promptTokens ?? 0) + (completionTokens ?? 0);
+  }
+  return { promptTokens, completionTokens, totalTokens, cost };
+}
+function wrapAISDK(ai, options) {
   const aiModule = ai;
+  aiSdkDebug = options?.debug ?? false;
   return {
     generateText: createGenerateTextWrapper(aiModule),
     streamText: createStreamTextWrapper(aiModule),
@@ -1494,6 +1542,28 @@ function createGenerateTextWrapper(aiModule) {
     try {
       const result = await aiModule.generateText(...args);
       const endTime = Date.now();
+      if (aiSdkDebug) {
+        console.log(
+          "\n\u{1F50D} [Fallom Debug] generateText result keys:",
+          Object.keys(result || {})
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.usage:",
+          JSON.stringify(result?.usage, null, 2)
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.response keys:",
+          Object.keys(result?.response || {})
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.response.usage:",
+          JSON.stringify(result?.response?.usage, null, 2)
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.experimental_providerMetadata:",
+          JSON.stringify(result?.experimental_providerMetadata, null, 2)
+        );
+      }
       const modelId = result?.response?.modelId || params?.model?.modelId || String(params?.model || "unknown");
       const attributes = {};
       if (captureContent) {
@@ -1517,6 +1587,18 @@ function createGenerateTextWrapper(aiModule) {
           attributes["gen_ai.response.id"] = result.response.id;
         }
       }
+      if (result?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(result.usage);
+      }
+      if (result?.experimental_providerMetadata) {
+        attributes["fallom.raw.providerMetadata"] = JSON.stringify(
+          result.experimental_providerMetadata
+        );
+      }
+      if (result?.finishReason) {
+        attributes["gen_ai.response.finish_reason"] = result.finishReason;
+      }
+      const usage = extractUsageFromResult(result);
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1531,9 +1613,9 @@ function createGenerateTextWrapper(aiModule) {
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "OK",
-        prompt_tokens: result?.usage?.promptTokens,
-        completion_tokens: result?.usage?.completionTokens,
-        total_tokens: result?.usage?.totalTokens,
+        prompt_tokens: usage.promptTokens,
+        completion_tokens: usage.completionTokens,
+        total_tokens: usage.totalTokens,
         attributes: captureContent ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
@@ -1592,9 +1674,31 @@ function createStreamTextWrapper(aiModule) {
     } catch {
     }
     if (result?.usage) {
-      result.usage.then((usage) => {
+      result.usage.then(async (rawUsage) => {
         const endTime = Date.now();
-        log2("\u{1F4CA} streamText usage:", JSON.stringify(usage, null, 2));
+        if (aiSdkDebug) {
+          console.log(
+            "\n\u{1F50D} [Fallom Debug] streamText usage:",
+            JSON.stringify(rawUsage, null, 2)
+          );
+          console.log(
+            "\u{1F50D} [Fallom Debug] streamText result keys:",
+            Object.keys(result || {})
+          );
+        }
+        log2("\u{1F4CA} streamText usage:", JSON.stringify(rawUsage, null, 2));
+        let providerMetadata = result?.experimental_providerMetadata;
+        if (providerMetadata && typeof providerMetadata.then === "function") {
+          try {
+            providerMetadata = await providerMetadata;
+          } catch {
+            providerMetadata = void 0;
+          }
+        }
+        const usage = extractUsageFromResult(
+          { experimental_providerMetadata: providerMetadata },
+          rawUsage
+        );
         const attributes = {};
         if (captureContent) {
           attributes["gen_ai.request.model"] = modelId;
@@ -1606,6 +1710,12 @@ function createStreamTextWrapper(aiModule) {
         if (firstTokenTime) {
           attributes["gen_ai.time_to_first_token_ms"] = firstTokenTime - startTime;
         }
+        if (rawUsage) {
+          attributes["fallom.raw.usage"] = JSON.stringify(rawUsage);
+        }
+        if (providerMetadata) {
+          attributes["fallom.raw.providerMetadata"] = JSON.stringify(providerMetadata);
+        }
         const tracePayload = {
           config_key: ctx.configKey,
           session_id: ctx.sessionId,
@@ -1620,9 +1730,9 @@ function createStreamTextWrapper(aiModule) {
           end_time: new Date(endTime).toISOString(),
           duration_ms: endTime - startTime,
           status: "OK",
-          prompt_tokens: usage?.promptTokens,
-          completion_tokens: usage?.completionTokens,
-          total_tokens: usage?.totalTokens,
+          prompt_tokens: usage.promptTokens,
+          completion_tokens: usage.completionTokens,
+          total_tokens: usage.totalTokens,
           time_to_first_token_ms: firstTokenTime ? firstTokenTime - startTime : void 0,
           attributes: captureContent ? attributes : void 0,
           prompt_key: promptCtx?.promptKey,
@@ -1702,6 +1812,24 @@ function createGenerateObjectWrapper(aiModule) {
     try {
       const result = await aiModule.generateObject(...args);
       const endTime = Date.now();
+      if (aiSdkDebug) {
+        console.log(
+          "\n\u{1F50D} [Fallom Debug] generateObject result keys:",
+          Object.keys(result || {})
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.usage:",
+          JSON.stringify(result?.usage, null, 2)
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.response keys:",
+          Object.keys(result?.response || {})
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.response.usage:",
+          JSON.stringify(result?.response?.usage, null, 2)
+        );
+      }
       const modelId = result?.response?.modelId || params?.model?.modelId || String(params?.model || "unknown");
       const attributes = {};
       if (captureContent) {
@@ -1714,6 +1842,18 @@ function createGenerateObjectWrapper(aiModule) {
           );
         }
       }
+      if (result?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(result.usage);
+      }
+      if (result?.experimental_providerMetadata) {
+        attributes["fallom.raw.providerMetadata"] = JSON.stringify(
+          result.experimental_providerMetadata
+        );
+      }
+      if (result?.finishReason) {
+        attributes["gen_ai.response.finish_reason"] = result.finishReason;
+      }
+      const usage = extractUsageFromResult(result);
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1728,9 +1868,9 @@ function createGenerateObjectWrapper(aiModule) {
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "OK",
-        prompt_tokens: result?.usage?.promptTokens,
-        completion_tokens: result?.usage?.completionTokens,
-        total_tokens: result?.usage?.totalTokens,
+        prompt_tokens: usage.promptTokens,
+        completion_tokens: usage.completionTokens,
+        total_tokens: usage.totalTokens,
         attributes: captureContent ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
@@ -1790,9 +1930,31 @@ function createStreamObjectWrapper(aiModule) {
     } catch {
     }
     if (result?.usage) {
-      result.usage.then((usage) => {
+      result.usage.then(async (rawUsage) => {
         const endTime = Date.now();
-        log2("\u{1F4CA} streamObject usage:", JSON.stringify(usage, null, 2));
+        if (aiSdkDebug) {
+          console.log(
+            "\n\u{1F50D} [Fallom Debug] streamObject usage:",
+            JSON.stringify(rawUsage, null, 2)
+          );
+          console.log(
+            "\u{1F50D} [Fallom Debug] streamObject result keys:",
+            Object.keys(result || {})
+          );
+        }
+        log2("\u{1F4CA} streamObject usage:", JSON.stringify(rawUsage, null, 2));
+        let providerMetadata = result?.experimental_providerMetadata;
+        if (providerMetadata && typeof providerMetadata.then === "function") {
+          try {
+            providerMetadata = await providerMetadata;
+          } catch {
+            providerMetadata = void 0;
+          }
+        }
+        const usage = extractUsageFromResult(
+          { experimental_providerMetadata: providerMetadata },
+          rawUsage
+        );
         const attributes = {};
         if (captureContent) {
           attributes["gen_ai.request.model"] = modelId;
@@ -1800,6 +1962,12 @@ function createStreamObjectWrapper(aiModule) {
         if (firstTokenTime) {
           attributes["gen_ai.time_to_first_token_ms"] = firstTokenTime - startTime;
         }
+        if (rawUsage) {
+          attributes["fallom.raw.usage"] = JSON.stringify(rawUsage);
+        }
+        if (providerMetadata) {
+          attributes["fallom.raw.providerMetadata"] = JSON.stringify(providerMetadata);
+        }
         sendTrace({
           config_key: ctx.configKey,
           session_id: ctx.sessionId,
@@ -1814,9 +1982,9 @@ function createStreamObjectWrapper(aiModule) {
           end_time: new Date(endTime).toISOString(),
           duration_ms: endTime - startTime,
           status: "OK",
-          prompt_tokens: usage?.promptTokens,
-          completion_tokens: usage?.completionTokens,
-          total_tokens: usage?.totalTokens,
+          prompt_tokens: usage.promptTokens,
+          completion_tokens: usage.completionTokens,
+          total_tokens: usage.totalTokens,
           attributes: captureContent ? attributes : void 0,
           prompt_key: promptCtx?.promptKey,
           prompt_version: promptCtx?.promptVersion,

package/dist/index.mjs

@@ -917,7 +917,13 @@ function wrapOpenAI(client) {
         response?.choices?.[0]?.message,
         response?.model || params?.model,
         response?.id
-      ) : void 0;
+      ) : {};
+      if (response?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(response.usage);
+      }
+      if (response?.choices?.[0]?.finish_reason) {
+        attributes["gen_ai.response.finish_reason"] = response.choices[0].finish_reason;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -935,7 +941,7 @@ function wrapOpenAI(client) {
         prompt_tokens: response?.usage?.prompt_tokens,
         completion_tokens: response?.usage?.completion_tokens,
         total_tokens: response?.usage?.total_tokens,
-        attributes,
+        attributes: Object.keys(attributes).length > 0 ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1008,10 +1014,16 @@ function wrapAnthropic(client) {
         { role: "assistant", content: response?.content?.[0]?.text || "" },
         response?.model || params?.model,
         response?.id
-      ) : void 0;
-      if (attributes && params?.system) {
+      ) : {};
+      if (params?.system) {
         attributes["gen_ai.system_prompt"] = params.system;
       }
+      if (response?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(response.usage);
+      }
+      if (response?.stop_reason) {
+        attributes["gen_ai.response.finish_reason"] = response.stop_reason;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1029,7 +1041,7 @@ function wrapAnthropic(client) {
         prompt_tokens: response?.usage?.input_tokens,
         completion_tokens: response?.usage?.output_tokens,
         total_tokens: (response?.usage?.input_tokens || 0) + (response?.usage?.output_tokens || 0),
-        attributes,
+        attributes: Object.keys(attributes).length > 0 ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1122,6 +1134,13 @@ function wrapGoogleAI(model) {
           attributes["gen_ai.completion.0.content"] = outputText;
         }
       }
+      if (usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(usage);
+      }
+      const candidate = result?.candidates?.[0];
+      if (candidate?.finishReason) {
+        attributes["gen_ai.response.finish_reason"] = candidate.finishReason;
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1139,7 +1158,7 @@ function wrapGoogleAI(model) {
         prompt_tokens: usage?.promptTokenCount,
         completion_tokens: usage?.candidatesTokenCount,
         total_tokens: usage?.totalTokenCount,
-        attributes: captureContent ? attributes : void 0,
+        attributes: Object.keys(attributes).length > 0 ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
         prompt_ab_test_key: promptCtx?.abTestKey,
@@ -1187,8 +1206,37 @@ function wrapGoogleAI(model) {
   };
   return model;
 }
-function wrapAISDK(ai) {
+var aiSdkDebug = false;
+function extractUsageFromResult(result, directUsage) {
+  let usage = directUsage ?? result?.usage;
+  const isValidNumber = (v) => v !== null && v !== void 0 && !Number.isNaN(v);
+  let promptTokens = isValidNumber(usage?.promptTokens) ? usage.promptTokens : void 0;
+  let completionTokens = isValidNumber(usage?.completionTokens) ? usage.completionTokens : void 0;
+  let totalTokens = isValidNumber(usage?.totalTokens) ? usage.totalTokens : void 0;
+  let cost;
+  const orUsage = result?.experimental_providerMetadata?.openrouter?.usage;
+  if (orUsage) {
+    if (promptTokens === void 0 && isValidNumber(orUsage.promptTokens)) {
+      promptTokens = orUsage.promptTokens;
+    }
+    if (completionTokens === void 0 && isValidNumber(orUsage.completionTokens)) {
+      completionTokens = orUsage.completionTokens;
+    }
+    if (totalTokens === void 0 && isValidNumber(orUsage.totalTokens)) {
+      totalTokens = orUsage.totalTokens;
+    }
+    if (isValidNumber(orUsage.cost)) {
+      cost = orUsage.cost;
+    }
+  }
+  if (totalTokens === void 0 && (promptTokens !== void 0 || completionTokens !== void 0)) {
+    totalTokens = (promptTokens ?? 0) + (completionTokens ?? 0);
+  }
+  return { promptTokens, completionTokens, totalTokens, cost };
+}
+function wrapAISDK(ai, options) {
   const aiModule = ai;
+  aiSdkDebug = options?.debug ?? false;
   return {
     generateText: createGenerateTextWrapper(aiModule),
     streamText: createStreamTextWrapper(aiModule),
@@ -1217,6 +1265,28 @@ function createGenerateTextWrapper(aiModule) {
     try {
       const result = await aiModule.generateText(...args);
       const endTime = Date.now();
+      if (aiSdkDebug) {
+        console.log(
+          "\n\u{1F50D} [Fallom Debug] generateText result keys:",
+          Object.keys(result || {})
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.usage:",
+          JSON.stringify(result?.usage, null, 2)
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.response keys:",
+          Object.keys(result?.response || {})
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.response.usage:",
+          JSON.stringify(result?.response?.usage, null, 2)
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.experimental_providerMetadata:",
+          JSON.stringify(result?.experimental_providerMetadata, null, 2)
+        );
+      }
       const modelId = result?.response?.modelId || params?.model?.modelId || String(params?.model || "unknown");
       const attributes = {};
       if (captureContent) {
@@ -1240,6 +1310,18 @@ function createGenerateTextWrapper(aiModule) {
           attributes["gen_ai.response.id"] = result.response.id;
         }
       }
+      if (result?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(result.usage);
+      }
+      if (result?.experimental_providerMetadata) {
+        attributes["fallom.raw.providerMetadata"] = JSON.stringify(
+          result.experimental_providerMetadata
+        );
+      }
+      if (result?.finishReason) {
+        attributes["gen_ai.response.finish_reason"] = result.finishReason;
+      }
+      const usage = extractUsageFromResult(result);
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1254,9 +1336,9 @@ function createGenerateTextWrapper(aiModule) {
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "OK",
-        prompt_tokens: result?.usage?.promptTokens,
-        completion_tokens: result?.usage?.completionTokens,
-        total_tokens: result?.usage?.totalTokens,
+        prompt_tokens: usage.promptTokens,
+        completion_tokens: usage.completionTokens,
+        total_tokens: usage.totalTokens,
         attributes: captureContent ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
@@ -1315,9 +1397,31 @@ function createStreamTextWrapper(aiModule) {
     } catch {
     }
     if (result?.usage) {
-      result.usage.then((usage) => {
+      result.usage.then(async (rawUsage) => {
         const endTime = Date.now();
-        log("\u{1F4CA} streamText usage:", JSON.stringify(usage, null, 2));
+        if (aiSdkDebug) {
+          console.log(
+            "\n\u{1F50D} [Fallom Debug] streamText usage:",
+            JSON.stringify(rawUsage, null, 2)
+          );
+          console.log(
+            "\u{1F50D} [Fallom Debug] streamText result keys:",
+            Object.keys(result || {})
+          );
+        }
+        log("\u{1F4CA} streamText usage:", JSON.stringify(rawUsage, null, 2));
+        let providerMetadata = result?.experimental_providerMetadata;
+        if (providerMetadata && typeof providerMetadata.then === "function") {
+          try {
+            providerMetadata = await providerMetadata;
+          } catch {
+            providerMetadata = void 0;
+          }
+        }
+        const usage = extractUsageFromResult(
+          { experimental_providerMetadata: providerMetadata },
+          rawUsage
+        );
         const attributes = {};
         if (captureContent) {
           attributes["gen_ai.request.model"] = modelId;
@@ -1329,6 +1433,12 @@ function createStreamTextWrapper(aiModule) {
         if (firstTokenTime) {
           attributes["gen_ai.time_to_first_token_ms"] = firstTokenTime - startTime;
         }
+        if (rawUsage) {
+          attributes["fallom.raw.usage"] = JSON.stringify(rawUsage);
+        }
+        if (providerMetadata) {
+          attributes["fallom.raw.providerMetadata"] = JSON.stringify(providerMetadata);
+        }
         const tracePayload = {
           config_key: ctx.configKey,
           session_id: ctx.sessionId,
@@ -1343,9 +1453,9 @@ function createStreamTextWrapper(aiModule) {
           end_time: new Date(endTime).toISOString(),
           duration_ms: endTime - startTime,
           status: "OK",
-          prompt_tokens: usage?.promptTokens,
-          completion_tokens: usage?.completionTokens,
-          total_tokens: usage?.totalTokens,
+          prompt_tokens: usage.promptTokens,
+          completion_tokens: usage.completionTokens,
+          total_tokens: usage.totalTokens,
           time_to_first_token_ms: firstTokenTime ? firstTokenTime - startTime : void 0,
           attributes: captureContent ? attributes : void 0,
           prompt_key: promptCtx?.promptKey,
@@ -1425,6 +1535,24 @@ function createGenerateObjectWrapper(aiModule) {
     try {
       const result = await aiModule.generateObject(...args);
       const endTime = Date.now();
+      if (aiSdkDebug) {
+        console.log(
+          "\n\u{1F50D} [Fallom Debug] generateObject result keys:",
+          Object.keys(result || {})
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.usage:",
+          JSON.stringify(result?.usage, null, 2)
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.response keys:",
+          Object.keys(result?.response || {})
+        );
+        console.log(
+          "\u{1F50D} [Fallom Debug] result.response.usage:",
+          JSON.stringify(result?.response?.usage, null, 2)
+        );
+      }
       const modelId = result?.response?.modelId || params?.model?.modelId || String(params?.model || "unknown");
       const attributes = {};
       if (captureContent) {
@@ -1437,6 +1565,18 @@ function createGenerateObjectWrapper(aiModule) {
           );
         }
       }
+      if (result?.usage) {
+        attributes["fallom.raw.usage"] = JSON.stringify(result.usage);
+      }
+      if (result?.experimental_providerMetadata) {
+        attributes["fallom.raw.providerMetadata"] = JSON.stringify(
+          result.experimental_providerMetadata
+        );
+      }
+      if (result?.finishReason) {
+        attributes["gen_ai.response.finish_reason"] = result.finishReason;
+      }
+      const usage = extractUsageFromResult(result);
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,
@@ -1451,9 +1591,9 @@ function createGenerateObjectWrapper(aiModule) {
         end_time: new Date(endTime).toISOString(),
         duration_ms: endTime - startTime,
         status: "OK",
-        prompt_tokens: result?.usage?.promptTokens,
-        completion_tokens: result?.usage?.completionTokens,
-        total_tokens: result?.usage?.totalTokens,
+        prompt_tokens: usage.promptTokens,
+        completion_tokens: usage.completionTokens,
+        total_tokens: usage.totalTokens,
         attributes: captureContent ? attributes : void 0,
         prompt_key: promptCtx?.promptKey,
         prompt_version: promptCtx?.promptVersion,
@@ -1513,9 +1653,31 @@ function createStreamObjectWrapper(aiModule) {
     } catch {
     }
     if (result?.usage) {
-      result.usage.then((usage) => {
+      result.usage.then(async (rawUsage) => {
         const endTime = Date.now();
-        log("\u{1F4CA} streamObject usage:", JSON.stringify(usage, null, 2));
+        if (aiSdkDebug) {
+          console.log(
+            "\n\u{1F50D} [Fallom Debug] streamObject usage:",
+            JSON.stringify(rawUsage, null, 2)
+          );
+          console.log(
+            "\u{1F50D} [Fallom Debug] streamObject result keys:",
+            Object.keys(result || {})
+          );
+        }
+        log("\u{1F4CA} streamObject usage:", JSON.stringify(rawUsage, null, 2));
+        let providerMetadata = result?.experimental_providerMetadata;
+        if (providerMetadata && typeof providerMetadata.then === "function") {
+          try {
+            providerMetadata = await providerMetadata;
+          } catch {
+            providerMetadata = void 0;
+          }
+        }
+        const usage = extractUsageFromResult(
+          { experimental_providerMetadata: providerMetadata },
+          rawUsage
+        );
         const attributes = {};
         if (captureContent) {
           attributes["gen_ai.request.model"] = modelId;
@@ -1523,6 +1685,12 @@ function createStreamObjectWrapper(aiModule) {
         if (firstTokenTime) {
           attributes["gen_ai.time_to_first_token_ms"] = firstTokenTime - startTime;
         }
+        if (rawUsage) {
+          attributes["fallom.raw.usage"] = JSON.stringify(rawUsage);
+        }
+        if (providerMetadata) {
+          attributes["fallom.raw.providerMetadata"] = JSON.stringify(providerMetadata);
+        }
         sendTrace({
           config_key: ctx.configKey,
           session_id: ctx.sessionId,
@@ -1537,9 +1705,9 @@ function createStreamObjectWrapper(aiModule) {
           end_time: new Date(endTime).toISOString(),
           duration_ms: endTime - startTime,
           status: "OK",
-          prompt_tokens: usage?.promptTokens,
-          completion_tokens: usage?.completionTokens,
-          total_tokens: usage?.totalTokens,
+          prompt_tokens: usage.promptTokens,
+          completion_tokens: usage.completionTokens,
+          total_tokens: usage.totalTokens,
           attributes: captureContent ? attributes : void 0,
           prompt_key: promptCtx?.promptKey,
           prompt_version: promptCtx?.promptVersion,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fallom/trace",
-  "version": "0.1.11",
+  "version": "0.1.12",
   "description": "Model A/B testing and tracing for LLM applications. Zero latency, production-ready.",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",