@fallom/trace 0.1.10 → 0.1.12

package/dist/index.mjs

@@ -17,6 +17,7 @@ __export(trace_exports, {
   wrapAISDK: () => wrapAISDK,
   wrapAnthropic: () => wrapAnthropic,
   wrapGoogleAI: () => wrapGoogleAI,
+  wrapMastraAgent: () => wrapMastraAgent,
   wrapOpenAI: () => wrapOpenAI
 });
 import { AsyncLocalStorage } from "async_hooks";
@@ -26,7 +27,7 @@ import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-http";
 // node_modules/@opentelemetry/resources/build/esm/Resource.js
 import { diag } from "@opentelemetry/api";
 
-// node_modules/@opentelemetry/semantic-conventions/build/esm/resource/SemanticResourceAttributes.js
+// node_modules/@opentelemetry/resources/node_modules/@opentelemetry/semantic-conventions/build/esm/resource/SemanticResourceAttributes.js
 var SemanticResourceAttributes = {
   /**
    * Name of the cloud provider.
@@ -406,35 +407,9 @@ var SemanticResourceAttributes = {
    */
   WEBENGINE_DESCRIPTION: "webengine.description"
 };
-var TelemetrySdkLanguageValues = {
-  /** cpp. */
-  CPP: "cpp",
-  /** dotnet. */
-  DOTNET: "dotnet",
-  /** erlang. */
-  ERLANG: "erlang",
-  /** go. */
-  GO: "go",
-  /** java. */
-  JAVA: "java",
-  /** nodejs. */
-  NODEJS: "nodejs",
-  /** php. */
-  PHP: "php",
-  /** python. */
-  PYTHON: "python",
-  /** ruby. */
-  RUBY: "ruby",
-  /** webjs. */
-  WEBJS: "webjs"
-};
-
-// node_modules/@opentelemetry/core/build/esm/version.js
-var VERSION = "1.19.0";
 
-// node_modules/@opentelemetry/core/build/esm/platform/node/sdk-info.js
-var _a;
-var SDK_INFO = (_a = {}, _a[SemanticResourceAttributes.TELEMETRY_SDK_NAME] = "opentelemetry", _a[SemanticResourceAttributes.PROCESS_RUNTIME_NAME] = "node", _a[SemanticResourceAttributes.TELEMETRY_SDK_LANGUAGE] = TelemetrySdkLanguageValues.NODEJS, _a[SemanticResourceAttributes.TELEMETRY_SDK_VERSION] = VERSION, _a);
+// node_modules/@opentelemetry/resources/build/esm/Resource.js
+import { SDK_INFO } from "@opentelemetry/core";
 
 // node_modules/@opentelemetry/resources/build/esm/platform/node/default-service-name.js
 function defaultServiceName() {
@@ -571,10 +546,10 @@ var Resource = (
   (function() {
     function Resource2(attributes, asyncAttributesPromise) {
       var _this = this;
-      var _a2;
+      var _a;
       this._attributes = attributes;
       this.asyncAttributesPending = asyncAttributesPromise != null;
-      this._syncAttributes = (_a2 = this._attributes) !== null && _a2 !== void 0 ? _a2 : {};
+      this._syncAttributes = (_a = this._attributes) !== null && _a !== void 0 ? _a : {};
       this._asyncAttributesPromise = asyncAttributesPromise === null || asyncAttributesPromise === void 0 ? void 0 : asyncAttributesPromise.then(function(asyncAttributes) {
         _this._attributes = Object.assign({}, _this._attributes, asyncAttributes);
         _this.asyncAttributesPending = false;
@@ -589,30 +564,30 @@ var Resource = (
       return Resource2.EMPTY;
     };
     Resource2.default = function() {
-      var _a2;
-      return new Resource2((_a2 = {}, _a2[SemanticResourceAttributes.SERVICE_NAME] = defaultServiceName(), _a2[SemanticResourceAttributes.TELEMETRY_SDK_LANGUAGE] = SDK_INFO[SemanticResourceAttributes.TELEMETRY_SDK_LANGUAGE], _a2[SemanticResourceAttributes.TELEMETRY_SDK_NAME] = SDK_INFO[SemanticResourceAttributes.TELEMETRY_SDK_NAME], _a2[SemanticResourceAttributes.TELEMETRY_SDK_VERSION] = SDK_INFO[SemanticResourceAttributes.TELEMETRY_SDK_VERSION], _a2));
+      var _a;
+      return new Resource2((_a = {}, _a[SemanticResourceAttributes.SERVICE_NAME] = defaultServiceName(), _a[SemanticResourceAttributes.TELEMETRY_SDK_LANGUAGE] = SDK_INFO[SemanticResourceAttributes.TELEMETRY_SDK_LANGUAGE], _a[SemanticResourceAttributes.TELEMETRY_SDK_NAME] = SDK_INFO[SemanticResourceAttributes.TELEMETRY_SDK_NAME], _a[SemanticResourceAttributes.TELEMETRY_SDK_VERSION] = SDK_INFO[SemanticResourceAttributes.TELEMETRY_SDK_VERSION], _a));
     };
     Object.defineProperty(Resource2.prototype, "attributes", {
       get: function() {
-        var _a2;
+        var _a;
         if (this.asyncAttributesPending) {
           diag.error("Accessing resource attributes before async attributes settled");
         }
-        return (_a2 = this._attributes) !== null && _a2 !== void 0 ? _a2 : {};
+        return (_a = this._attributes) !== null && _a !== void 0 ? _a : {};
       },
       enumerable: false,
       configurable: true
     });
     Resource2.prototype.waitForAsyncAttributes = function() {
       return __awaiter(this, void 0, void 0, function() {
-        return __generator(this, function(_a2) {
-          switch (_a2.label) {
+        return __generator(this, function(_a) {
+          switch (_a.label) {
             case 0:
               if (!this.asyncAttributesPending) return [3, 2];
               return [4, this._asyncAttributesPromise];
             case 1:
-              _a2.sent();
-              _a2.label = 2;
+              _a.sent();
+              _a.label = 2;
             case 2:
               return [
                 2
@@ -624,19 +599,19 @@ var Resource = (
     };
     Resource2.prototype.merge = function(other) {
       var _this = this;
-      var _a2;
+      var _a;
       if (!other)
         return this;
-      var mergedSyncAttributes = __assign(__assign({}, this._syncAttributes), (_a2 = other._syncAttributes) !== null && _a2 !== void 0 ? _a2 : other.attributes);
+      var mergedSyncAttributes = __assign(__assign({}, this._syncAttributes), (_a = other._syncAttributes) !== null && _a !== void 0 ? _a : other.attributes);
       if (!this._asyncAttributesPromise && !other._asyncAttributesPromise) {
         return new Resource2(mergedSyncAttributes);
       }
       var mergedAttributesPromise = Promise.all([
         this._asyncAttributesPromise,
         other._asyncAttributesPromise
-      ]).then(function(_a3) {
+      ]).then(function(_a2) {
         var _b;
-        var _c = __read(_a3, 2), thisAsyncAttributes = _c[0], otherAsyncAttributes = _c[1];
+        var _c = __read(_a2, 2), thisAsyncAttributes = _c[0], otherAsyncAttributes = _c[1];
         return __assign(__assign(__assign(__assign({}, _this._syncAttributes), thisAsyncAttributes), (_b = other._syncAttributes) !== null && _b !== void 0 ? _b : other.attributes), otherAsyncAttributes);
       });
       return new Resource2(mergedSyncAttributes, mergedAttributesPromise);
@@ -942,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,
@@ -960,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,
@@ -1033,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,
@@ -1054,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,
@@ -1147,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,
@@ -1164,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,
@@ -1212,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),
@@ -1242,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) {
@@ -1265,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,
@@ -1279,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,
@@ -1340,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;
@@ -1354,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,
@@ -1368,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,
@@ -1450,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) {
@@ -1462,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,
@@ -1476,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,
@@ -1538,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;
@@ -1548,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,
@@ -1562,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,
@@ -1620,6 +1763,127 @@ function createStreamObjectWrapper(aiModule) {
     return result;
   };
 }
+function wrapMastraAgent(agent) {
+  const originalGenerate = agent.generate.bind(agent);
+  const agentName = agent.name || "MastraAgent";
+  agent.generate = async function(...args) {
+    const ctx = sessionStorage.getStore() || fallbackSession;
+    if (!ctx || !initialized) {
+      return originalGenerate(...args);
+    }
+    let promptCtx = null;
+    try {
+      const { getPromptContext } = await import("./prompts-VAN5E3L4.mjs");
+      promptCtx = getPromptContext();
+    } catch {
+    }
+    const traceId = generateHexId(32);
+    const spanId = generateHexId(16);
+    const startTime = Date.now();
+    const messages = args[0] || [];
+    try {
+      const result = await originalGenerate(...args);
+      const endTime = Date.now();
+      const model = result?.model?.modelId || "unknown";
+      const toolCalls = [];
+      if (result?.steps?.length) {
+        for (const step of result.steps) {
+          if (step.toolCalls?.length) {
+            for (let i = 0; i < step.toolCalls.length; i++) {
+              const tc = step.toolCalls[i];
+              const tr = step.toolResults?.[i];
+              toolCalls.push({
+                name: tc.toolName,
+                arguments: tc.args,
+                result: tr?.result
+              });
+            }
+          }
+        }
+      }
+      const attributes = {
+        "gen_ai.system": "Mastra",
+        "gen_ai.request.model": model,
+        "gen_ai.response.model": model,
+        "fallom.source": "mastra-agent",
+        "llm.request.type": "chat"
+      };
+      if (Array.isArray(messages)) {
+        messages.forEach((msg, i) => {
+          attributes[`gen_ai.prompt.${i}.role`] = msg.role || "user";
+          attributes[`gen_ai.prompt.${i}.content`] = typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content);
+        });
+      }
+      if (result?.text) {
+        attributes["gen_ai.completion.0.role"] = "assistant";
+        attributes["gen_ai.completion.0.content"] = result.text;
+        attributes["gen_ai.completion.0.finish_reason"] = "stop";
+      }
+      if (toolCalls.length > 0) {
+        attributes["fallom.tool_calls"] = JSON.stringify(toolCalls);
+        toolCalls.forEach((tc, i) => {
+          attributes[`gen_ai.completion.0.tool_calls.${i}.name`] = tc.name;
+          attributes[`gen_ai.completion.0.tool_calls.${i}.type`] = "function";
+          attributes[`gen_ai.completion.0.tool_calls.${i}.arguments`] = JSON.stringify(tc.arguments);
+        });
+      }
+      if (result?.usage) {
+        attributes["gen_ai.usage.prompt_tokens"] = result.usage.promptTokens;
+        attributes["gen_ai.usage.completion_tokens"] = result.usage.completionTokens;
+        attributes["llm.usage.total_tokens"] = result.usage.totalTokens;
+      }
+      const traceData = {
+        config_key: ctx.configKey,
+        session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        name: `mastra.${agentName}.generate`,
+        kind: "client",
+        model,
+        start_time: new Date(startTime).toISOString(),
+        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,
+        attributes,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
+      };
+      sendTrace(traceData).catch(() => {
+      });
+      return result;
+    } catch (error) {
+      const endTime = Date.now();
+      const traceData = {
+        config_key: ctx.configKey,
+        session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        name: `mastra.${agentName}.generate`,
+        kind: "client",
+        start_time: new Date(startTime).toISOString(),
+        end_time: new Date(endTime).toISOString(),
+        duration_ms: endTime - startTime,
+        status: "ERROR",
+        error_message: error instanceof Error ? error.message : String(error),
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
+      };
+      sendTrace(traceData).catch(() => {
+      });
+      throw error;
+    }
+  };
+  return agent;
+}
 
 // src/models.ts
 var models_exports = {};
@@ -1885,6 +2149,241 @@ async function init4(options = {}) {
   });
 }
 
+// src/mastra.ts
+import { ExportResultCode } from "@opentelemetry/core";
+var promptContext = {};
+function setMastraPrompt(promptKey, version) {
+  promptContext = {
+    promptKey,
+    promptVersion: version,
+    promptAbTestKey: void 0,
+    promptVariantIndex: void 0
+  };
+}
+function setMastraPromptAB(abTestKey, variantIndex) {
+  promptContext = {
+    promptKey: void 0,
+    promptVersion: void 0,
+    promptAbTestKey: abTestKey,
+    promptVariantIndex: variantIndex
+  };
+}
+function clearMastraPrompt() {
+  promptContext = {};
+}
+var FallomExporter = class {
+  constructor(options = {}) {
+    this.pendingExports = [];
+    this.apiKey = options.apiKey ?? process.env.FALLOM_API_KEY ?? "";
+    this.baseUrl = options.baseUrl ?? "https://traces.fallom.com";
+    this.debug = options.debug ?? false;
+    console.log("[FallomExporter] Constructor called, debug:", this.debug);
+    console.log("[FallomExporter] API key present:", !!this.apiKey);
+    console.log("[FallomExporter] Base URL:", this.baseUrl);
+    if (!this.apiKey) {
+      console.warn(
+        "[FallomExporter] No API key provided. Set FALLOM_API_KEY env var or pass apiKey option."
+      );
+    }
+  }
+  log(...args) {
+    if (this.debug) {
+      console.log("[FallomExporter]", ...args);
+    }
+  }
+  /**
+   * Export spans to Fallom.
+   */
+  export(spans, resultCallback) {
+    if (spans.length === 0) {
+      resultCallback({ code: ExportResultCode.SUCCESS });
+      return;
+    }
+    this.log(`Exporting ${spans.length} spans...`);
+    if (this.debug) {
+      for (const span2 of spans) {
+        this.log(`  - ${span2.name}`, {
+          attributes: Object.fromEntries(
+            Object.entries(span2.attributes).filter(
+              ([k]) => k.startsWith("gen_ai") || k.startsWith("llm")
+            )
+          )
+        });
+      }
+    }
+    const exportPromise = this.sendSpans(spans).then(() => {
+      this.log("Export successful");
+      resultCallback({ code: ExportResultCode.SUCCESS });
+    }).catch((error) => {
+      console.error("[FallomExporter] Export failed:", error);
+      resultCallback({
+        code: ExportResultCode.FAILED,
+        error: error instanceof Error ? error : new Error(String(error))
+      });
+    });
+    this.pendingExports.push(exportPromise);
+  }
+  /**
+   * Shutdown the exporter, waiting for pending exports.
+   */
+  async shutdown() {
+    await Promise.all(this.pendingExports);
+    this.pendingExports = [];
+  }
+  /**
+   * Force flush pending exports.
+   */
+  async forceFlush() {
+    await Promise.all(this.pendingExports);
+  }
+  /**
+   * Send spans to Fallom's OTLP endpoint.
+   */
+  async sendSpans(spans) {
+    const session = getSession();
+    const resourceSpans = this.spansToOtlpJson(spans);
+    const headers = {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${this.apiKey}`
+    };
+    if (session?.configKey) {
+      headers["X-Fallom-Config-Key"] = session.configKey;
+    }
+    if (session?.sessionId) {
+      headers["X-Fallom-Session-Id"] = session.sessionId;
+    }
+    if (session?.customerId) {
+      headers["X-Fallom-Customer-Id"] = session.customerId;
+    }
+    if (promptContext.promptKey) {
+      headers["X-Fallom-Prompt-Key"] = promptContext.promptKey;
+    }
+    if (promptContext.promptVersion !== void 0) {
+      headers["X-Fallom-Prompt-Version"] = String(promptContext.promptVersion);
+    }
+    if (promptContext.promptAbTestKey) {
+      headers["X-Fallom-Prompt-AB-Test"] = promptContext.promptAbTestKey;
+    }
+    if (promptContext.promptVariantIndex !== void 0) {
+      headers["X-Fallom-Prompt-Variant"] = String(
+        promptContext.promptVariantIndex
+      );
+    }
+    const endpoint = `${this.baseUrl}/v1/traces`;
+    this.log("Sending to", endpoint);
+    this.log("Headers:", {
+      ...headers,
+      Authorization: "Bearer ***"
+    });
+    const response = await fetch(endpoint, {
+      method: "POST",
+      headers,
+      body: JSON.stringify({ resourceSpans })
+    });
+    if (!response.ok) {
+      const text = await response.text();
+      throw new Error(`Failed to export: ${response.status} ${text}`);
+    }
+  }
+  /**
+   * Convert OpenTelemetry spans to OTLP JSON format.
+   */
+  spansToOtlpJson(spans) {
+    const resourceMap = /* @__PURE__ */ new Map();
+    for (const span2 of spans) {
+      const resourceKey = JSON.stringify(span2.resource.attributes);
+      if (!resourceMap.has(resourceKey)) {
+        resourceMap.set(resourceKey, []);
+      }
+      resourceMap.get(resourceKey).push(span2);
+    }
+    const resourceSpans = [];
+    for (const [_resourceKey, resourceSpanList] of resourceMap) {
+      const firstSpan = resourceSpanList[0];
+      resourceSpans.push({
+        resource: {
+          attributes: this.attributesToOtlp(firstSpan.resource.attributes)
+        },
+        scopeSpans: [
+          {
+            scope: {
+              name: firstSpan.instrumentationLibrary.name,
+              version: firstSpan.instrumentationLibrary.version
+            },
+            spans: resourceSpanList.map((span2) => this.spanToOtlp(span2))
+          }
+        ]
+      });
+    }
+    return resourceSpans;
+  }
+  /**
+   * Convert a single span to OTLP format.
+   */
+  spanToOtlp(span2) {
+    return {
+      traceId: span2.spanContext().traceId,
+      spanId: span2.spanContext().spanId,
+      parentSpanId: span2.parentSpanId,
+      name: span2.name,
+      kind: span2.kind,
+      startTimeUnixNano: this.hrTimeToNanos(span2.startTime),
+      endTimeUnixNano: this.hrTimeToNanos(span2.endTime),
+      attributes: this.attributesToOtlp(span2.attributes),
+      status: {
+        code: span2.status.code,
+        message: span2.status.message
+      },
+      events: span2.events.map((event) => ({
+        timeUnixNano: this.hrTimeToNanos(event.time),
+        name: event.name,
+        attributes: this.attributesToOtlp(event.attributes || {})
+      }))
+    };
+  }
+  /**
+   * Convert attributes to OTLP format.
+   */
+  attributesToOtlp(attrs) {
+    return Object.entries(attrs).map(([key, value]) => ({
+      key,
+      value: this.valueToOtlp(value)
+    }));
+  }
+  /**
+   * Convert a value to OTLP AnyValue format.
+   */
+  valueToOtlp(value) {
+    if (typeof value === "string") {
+      return { stringValue: value };
+    }
+    if (typeof value === "number") {
+      if (Number.isInteger(value)) {
+        return { intValue: value };
+      }
+      return { doubleValue: value };
+    }
+    if (typeof value === "boolean") {
+      return { boolValue: value };
+    }
+    if (Array.isArray(value)) {
+      return {
+        arrayValue: {
+          values: value.map((v) => this.valueToOtlp(v))
+        }
+      };
+    }
+    return { stringValue: String(value) };
+  }
+  /**
+   * Convert HrTime to nanoseconds string.
+   */
+  hrTimeToNanos(hrTime) {
+    const [seconds, nanos] = hrTime;
+    return String(BigInt(seconds) * BigInt(1e9) + BigInt(nanos));
+  }
+};
+
 // src/index.ts
 var index_default = {
   init: init4,
@@ -1893,9 +2392,13 @@ var index_default = {
   prompts: prompts_exports
 };
 export {
+  FallomExporter,
+  clearMastraPrompt,
   index_default as default,
   init4 as init,
   models_exports as models,
   prompts_exports as prompts,
+  setMastraPrompt,
+  setMastraPromptAB,
   trace_exports as trace
 };