@fallom/trace 0.1.6 → 0.1.11

package/dist/index.js

@@ -36,7 +36,7 @@ function log(msg) {
 }
 function init(options = {}) {
   apiKey = options.apiKey || process.env.FALLOM_API_KEY || null;
-  baseUrl = options.baseUrl || process.env.FALLOM_BASE_URL || "https://spans.fallom.com";
+  baseUrl = options.baseUrl || process.env.FALLOM_PROMPTS_URL || process.env.FALLOM_BASE_URL || "https://prompts.fallom.com";
   initialized = true;
   if (!apiKey) {
     return;
@@ -187,6 +187,13 @@ async function getAB(abTestKey, sessionId, options = {}) {
     throw new Error(`Prompt A/B test '${abTestKey}' has no current version.`);
   }
   const { variants } = versionData;
+  log(`A/B test '${abTestKey}' has ${variants?.length ?? 0} variants`);
+  log(`Version data: ${JSON.stringify(versionData, null, 2)}`);
+  if (!variants || variants.length === 0) {
+    throw new Error(
+      `Prompt A/B test '${abTestKey}' has no variants configured.`
+    );
+  }
   const hashBytes = (0, import_crypto.createHash)("md5").update(sessionId).digest();
   const hashVal = hashBytes.readUInt32BE(0) % 1e6;
   let cumulative = 0;
@@ -248,7 +255,7 @@ var init_prompts = __esm({
     "use strict";
     import_crypto = require("crypto");
     apiKey = null;
-    baseUrl = "https://spans.fallom.com";
+    baseUrl = "https://prompts.fallom.com";
     initialized = false;
     syncInterval = null;
     debugMode = false;
@@ -262,10 +269,14 @@ var init_prompts = __esm({
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  FallomExporter: () => FallomExporter,
+  clearMastraPrompt: () => clearMastraPrompt,
   default: () => index_default,
   init: () => init4,
   models: () => models_exports,
   prompts: () => prompts_exports,
+  setMastraPrompt: () => setMastraPrompt,
+  setMastraPromptAB: () => setMastraPromptAB,
   trace: () => trace_exports
 });
 module.exports = __toCommonJS(index_exports);
@@ -280,8 +291,10 @@ __export(trace_exports, {
   setSession: () => setSession,
   shutdown: () => shutdown,
   span: () => span,
+  wrapAISDK: () => wrapAISDK,
   wrapAnthropic: () => wrapAnthropic,
   wrapGoogleAI: () => wrapGoogleAI,
+  wrapMastraAgent: () => wrapMastraAgent,
   wrapOpenAI: () => wrapOpenAI
 });
 var import_async_hooks = require("async_hooks");
@@ -291,7 +304,7 @@ var import_exporter_trace_otlp_http = require("@opentelemetry/exporter-trace-otl
 // node_modules/@opentelemetry/resources/build/esm/Resource.js
 var import_api = require("@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.
@@ -671,35 +684,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
+var import_core = require("@opentelemetry/core");
 
 // node_modules/@opentelemetry/resources/build/esm/platform/node/default-service-name.js
 function defaultServiceName() {
@@ -836,10 +823,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;
@@ -854,30 +841,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] = import_core.SDK_INFO[SemanticResourceAttributes.TELEMETRY_SDK_LANGUAGE], _a[SemanticResourceAttributes.TELEMETRY_SDK_NAME] = import_core.SDK_INFO[SemanticResourceAttributes.TELEMETRY_SDK_NAME], _a[SemanticResourceAttributes.TELEMETRY_SDK_VERSION] = import_core.SDK_INFO[SemanticResourceAttributes.TELEMETRY_SDK_VERSION], _a));
     };
     Object.defineProperty(Resource2.prototype, "attributes", {
       get: function() {
-        var _a2;
+        var _a;
         if (this.asyncAttributesPending) {
           import_api.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
@@ -889,19 +876,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);
@@ -915,7 +902,7 @@ var Resource = (
 var sessionStorage = new import_async_hooks.AsyncLocalStorage();
 var fallbackSession = null;
 var apiKey2 = null;
-var baseUrl2 = "https://spans.fallom.com";
+var baseUrl2 = "https://traces.fallom.com";
 var initialized2 = false;
 var captureContent = true;
 var debugMode2 = false;
@@ -958,7 +945,7 @@ async function init2(options = {}) {
   debugMode2 = options.debug ?? false;
   log2("\u{1F680} Initializing Fallom tracing...");
   apiKey2 = options.apiKey || process.env.FALLOM_API_KEY || null;
-  baseUrl2 = options.baseUrl || process.env.FALLOM_BASE_URL || "https://spans.fallom.com";
+  baseUrl2 = options.baseUrl || process.env.FALLOM_TRACES_URL || process.env.FALLOM_BASE_URL || "https://traces.fallom.com";
   const envCapture = process.env.FALLOM_CAPTURE_CONTENT?.toLowerCase();
   if (envCapture === "false" || envCapture === "0" || envCapture === "no") {
     captureContent = false;
@@ -1130,12 +1117,12 @@ function messagesToOtelAttributes(messages, completion, model, responseId) {
   if (messages) {
     messages.forEach((msg, i) => {
       attrs[`gen_ai.prompt.${i}.role`] = msg.role;
-      attrs[`gen_ai.prompt.${i}.content`] = msg.content;
+      attrs[`gen_ai.prompt.${i}.content`] = typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content);
     });
   }
   if (completion) {
     attrs["gen_ai.completion.0.role"] = completion.role;
-    attrs["gen_ai.completion.0.content"] = completion.content;
+    attrs["gen_ai.completion.0.content"] = typeof completion.content === "string" ? completion.content : JSON.stringify(completion.content);
     if (completion.tool_calls) {
       attrs["gen_ai.completion.0.tool_calls"] = JSON.stringify(
         completion.tool_calls
@@ -1152,10 +1139,13 @@ function generateHexId(length) {
 var traceContextStorage = new import_async_hooks.AsyncLocalStorage();
 var fallbackTraceContext = null;
 async function sendTrace(trace) {
+  const url = `${baseUrl2}/v1/traces`;
+  log2("\u{1F4E4} Sending trace to:", url);
+  log2("   Session:", trace.session_id, "Config:", trace.config_key);
   try {
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), 5e3);
-    await fetch(`${baseUrl2}/v1/traces`, {
+    const response = await fetch(url, {
       method: "POST",
       headers: {
         Authorization: `Bearer ${apiKey2}`,
@@ -1165,8 +1155,14 @@ async function sendTrace(trace) {
       signal: controller.signal
     });
     clearTimeout(timeoutId);
-    log2("\u{1F4E4} Trace sent:", trace.name, trace.model);
-  } catch {
+    if (!response.ok) {
+      const text = await response.text();
+      log2("\u274C Trace send failed:", response.status, text);
+    } else {
+      log2("\u2705 Trace sent:", trace.name, trace.model);
+    }
+  } catch (err) {
+    log2("\u274C Trace send error:", err instanceof Error ? err.message : err);
   }
 }
 function wrapOpenAI(client) {
@@ -1468,6 +1464,535 @@ function wrapGoogleAI(model) {
   };
   return model;
 }
+function wrapAISDK(ai) {
+  const aiModule = ai;
+  return {
+    generateText: createGenerateTextWrapper(aiModule),
+    streamText: createStreamTextWrapper(aiModule),
+    generateObject: aiModule.generateObject ? createGenerateObjectWrapper(aiModule) : void 0,
+    streamObject: aiModule.streamObject ? createStreamObjectWrapper(aiModule) : void 0
+  };
+}
+function createGenerateTextWrapper(aiModule) {
+  return async (...args) => {
+    const ctx = sessionStorage.getStore() || fallbackSession;
+    if (!ctx || !initialized2) {
+      return aiModule.generateText(...args);
+    }
+    let promptCtx = null;
+    try {
+      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
+      promptCtx = getPromptContext2();
+    } catch {
+    }
+    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceId = traceCtx?.traceId || generateHexId(32);
+    const spanId = generateHexId(16);
+    const parentSpanId = traceCtx?.parentSpanId;
+    const params = args[0] || {};
+    const startTime = Date.now();
+    try {
+      const result = await aiModule.generateText(...args);
+      const endTime = Date.now();
+      const modelId = result?.response?.modelId || params?.model?.modelId || String(params?.model || "unknown");
+      const attributes = {};
+      if (captureContent) {
+        attributes["gen_ai.request.model"] = modelId;
+        attributes["gen_ai.response.model"] = modelId;
+        if (params?.prompt) {
+          attributes["gen_ai.prompt.0.role"] = "user";
+          attributes["gen_ai.prompt.0.content"] = params.prompt;
+        }
+        if (params?.messages) {
+          params.messages.forEach((msg, i) => {
+            attributes[`gen_ai.prompt.${i}.role`] = msg.role;
+            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;
+        }
+        if (result?.response?.id) {
+          attributes["gen_ai.response.id"] = result.response.id;
+        }
+      }
+      sendTrace({
+        config_key: ctx.configKey,
+        session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
+        name: "generateText",
+        kind: "llm",
+        model: modelId,
+        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: captureContent ? attributes : void 0,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
+      }).catch(() => {
+      });
+      return result;
+    } catch (error) {
+      const endTime = Date.now();
+      const modelId = params?.model?.modelId || String(params?.model || "unknown");
+      sendTrace({
+        config_key: ctx.configKey,
+        session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
+        name: "generateText",
+        kind: "llm",
+        model: modelId,
+        start_time: new Date(startTime).toISOString(),
+        end_time: new Date(endTime).toISOString(),
+        duration_ms: endTime - startTime,
+        status: "ERROR",
+        error_message: error?.message,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
+      }).catch(() => {
+      });
+      throw error;
+    }
+  };
+}
+function createStreamTextWrapper(aiModule) {
+  return async (...args) => {
+    const ctx = sessionStorage.getStore() || fallbackSession;
+    const params = args[0] || {};
+    const startTime = Date.now();
+    const result = await aiModule.streamText(...args);
+    if (!ctx || !initialized2) {
+      return result;
+    }
+    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceId = traceCtx?.traceId || generateHexId(32);
+    const spanId = generateHexId(16);
+    const parentSpanId = traceCtx?.parentSpanId;
+    let firstTokenTime = null;
+    const modelId = params?.model?.modelId || String(params?.model || "unknown");
+    let promptCtx = null;
+    try {
+      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
+      promptCtx = getPromptContext2();
+    } catch {
+    }
+    if (result?.usage) {
+      result.usage.then((usage) => {
+        const endTime = Date.now();
+        log2("\u{1F4CA} streamText usage:", JSON.stringify(usage, null, 2));
+        const attributes = {};
+        if (captureContent) {
+          attributes["gen_ai.request.model"] = modelId;
+          if (params?.prompt) {
+            attributes["gen_ai.prompt.0.role"] = "user";
+            attributes["gen_ai.prompt.0.content"] = params.prompt;
+          }
+        }
+        if (firstTokenTime) {
+          attributes["gen_ai.time_to_first_token_ms"] = firstTokenTime - startTime;
+        }
+        const tracePayload = {
+          config_key: ctx.configKey,
+          session_id: ctx.sessionId,
+          customer_id: ctx.customerId,
+          trace_id: traceId,
+          span_id: spanId,
+          parent_span_id: parentSpanId,
+          name: "streamText",
+          kind: "llm",
+          model: modelId,
+          start_time: new Date(startTime).toISOString(),
+          end_time: new Date(endTime).toISOString(),
+          duration_ms: endTime - startTime,
+          status: "OK",
+          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,
+          prompt_version: promptCtx?.promptVersion,
+          prompt_ab_test_key: promptCtx?.abTestKey,
+          prompt_variant_index: promptCtx?.variantIndex
+        };
+        sendTrace(tracePayload).catch(() => {
+        });
+      }).catch((error) => {
+        const endTime = Date.now();
+        log2("\u274C streamText error:", error?.message);
+        sendTrace({
+          config_key: ctx.configKey,
+          session_id: ctx.sessionId,
+          customer_id: ctx.customerId,
+          trace_id: traceId,
+          span_id: spanId,
+          parent_span_id: parentSpanId,
+          name: "streamText",
+          kind: "llm",
+          model: modelId,
+          start_time: new Date(startTime).toISOString(),
+          end_time: new Date(endTime).toISOString(),
+          duration_ms: endTime - startTime,
+          status: "ERROR",
+          error_message: error?.message,
+          prompt_key: promptCtx?.promptKey,
+          prompt_version: promptCtx?.promptVersion,
+          prompt_ab_test_key: promptCtx?.abTestKey,
+          prompt_variant_index: promptCtx?.variantIndex
+        }).catch(() => {
+        });
+      });
+    }
+    if (result?.textStream) {
+      const originalTextStream = result.textStream;
+      const wrappedTextStream = (async function* () {
+        for await (const chunk of originalTextStream) {
+          if (!firstTokenTime) {
+            firstTokenTime = Date.now();
+            log2("\u23F1\uFE0F Time to first token:", firstTokenTime - startTime, "ms");
+          }
+          yield chunk;
+        }
+      })();
+      return new Proxy(result, {
+        get(target, prop) {
+          if (prop === "textStream") {
+            return wrappedTextStream;
+          }
+          return target[prop];
+        }
+      });
+    }
+    return result;
+  };
+}
+function createGenerateObjectWrapper(aiModule) {
+  return async (...args) => {
+    const ctx = sessionStorage.getStore() || fallbackSession;
+    if (!ctx || !initialized2) {
+      return aiModule.generateObject(...args);
+    }
+    let promptCtx = null;
+    try {
+      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
+      promptCtx = getPromptContext2();
+    } catch {
+    }
+    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceId = traceCtx?.traceId || generateHexId(32);
+    const spanId = generateHexId(16);
+    const parentSpanId = traceCtx?.parentSpanId;
+    const params = args[0] || {};
+    const startTime = Date.now();
+    try {
+      const result = await aiModule.generateObject(...args);
+      const endTime = Date.now();
+      const modelId = result?.response?.modelId || params?.model?.modelId || String(params?.model || "unknown");
+      const attributes = {};
+      if (captureContent) {
+        attributes["gen_ai.request.model"] = modelId;
+        attributes["gen_ai.response.model"] = modelId;
+        if (result?.object) {
+          attributes["gen_ai.completion.0.role"] = "assistant";
+          attributes["gen_ai.completion.0.content"] = JSON.stringify(
+            result.object
+          );
+        }
+      }
+      sendTrace({
+        config_key: ctx.configKey,
+        session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
+        name: "generateObject",
+        kind: "llm",
+        model: modelId,
+        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: captureContent ? attributes : void 0,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
+      }).catch(() => {
+      });
+      return result;
+    } catch (error) {
+      const endTime = Date.now();
+      const modelId = params?.model?.modelId || String(params?.model || "unknown");
+      sendTrace({
+        config_key: ctx.configKey,
+        session_id: ctx.sessionId,
+        customer_id: ctx.customerId,
+        trace_id: traceId,
+        span_id: spanId,
+        parent_span_id: parentSpanId,
+        name: "generateObject",
+        kind: "llm",
+        model: modelId,
+        start_time: new Date(startTime).toISOString(),
+        end_time: new Date(endTime).toISOString(),
+        duration_ms: endTime - startTime,
+        status: "ERROR",
+        error_message: error?.message,
+        prompt_key: promptCtx?.promptKey,
+        prompt_version: promptCtx?.promptVersion,
+        prompt_ab_test_key: promptCtx?.abTestKey,
+        prompt_variant_index: promptCtx?.variantIndex
+      }).catch(() => {
+      });
+      throw error;
+    }
+  };
+}
+function createStreamObjectWrapper(aiModule) {
+  return async (...args) => {
+    const ctx = sessionStorage.getStore() || fallbackSession;
+    const params = args[0] || {};
+    const startTime = Date.now();
+    const result = await aiModule.streamObject(...args);
+    log2("\u{1F50D} streamObject result keys:", Object.keys(result || {}));
+    if (!ctx || !initialized2) {
+      return result;
+    }
+    const traceCtx = traceContextStorage.getStore() || fallbackTraceContext;
+    const traceId = traceCtx?.traceId || generateHexId(32);
+    const spanId = generateHexId(16);
+    const parentSpanId = traceCtx?.parentSpanId;
+    let firstTokenTime = null;
+    const modelId = params?.model?.modelId || String(params?.model || "unknown");
+    let promptCtx = null;
+    try {
+      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
+      promptCtx = getPromptContext2();
+    } catch {
+    }
+    if (result?.usage) {
+      result.usage.then((usage) => {
+        const endTime = Date.now();
+        log2("\u{1F4CA} streamObject usage:", JSON.stringify(usage, null, 2));
+        const attributes = {};
+        if (captureContent) {
+          attributes["gen_ai.request.model"] = modelId;
+        }
+        if (firstTokenTime) {
+          attributes["gen_ai.time_to_first_token_ms"] = firstTokenTime - startTime;
+        }
+        sendTrace({
+          config_key: ctx.configKey,
+          session_id: ctx.sessionId,
+          customer_id: ctx.customerId,
+          trace_id: traceId,
+          span_id: spanId,
+          parent_span_id: parentSpanId,
+          name: "streamObject",
+          kind: "llm",
+          model: modelId,
+          start_time: new Date(startTime).toISOString(),
+          end_time: new Date(endTime).toISOString(),
+          duration_ms: endTime - startTime,
+          status: "OK",
+          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,
+          prompt_ab_test_key: promptCtx?.abTestKey,
+          prompt_variant_index: promptCtx?.variantIndex
+        }).catch(() => {
+        });
+      }).catch((error) => {
+        const endTime = Date.now();
+        sendTrace({
+          config_key: ctx.configKey,
+          session_id: ctx.sessionId,
+          customer_id: ctx.customerId,
+          trace_id: traceId,
+          span_id: spanId,
+          parent_span_id: parentSpanId,
+          name: "streamObject",
+          kind: "llm",
+          model: modelId,
+          start_time: new Date(startTime).toISOString(),
+          end_time: new Date(endTime).toISOString(),
+          duration_ms: endTime - startTime,
+          status: "ERROR",
+          error_message: error?.message,
+          prompt_key: promptCtx?.promptKey,
+          prompt_version: promptCtx?.promptVersion,
+          prompt_ab_test_key: promptCtx?.abTestKey,
+          prompt_variant_index: promptCtx?.variantIndex
+        }).catch(() => {
+        });
+      });
+    }
+    if (result?.partialObjectStream) {
+      const originalStream = result.partialObjectStream;
+      const wrappedStream = (async function* () {
+        for await (const chunk of originalStream) {
+          if (!firstTokenTime) {
+            firstTokenTime = Date.now();
+            log2("\u23F1\uFE0F Time to first token:", firstTokenTime - startTime, "ms");
+          }
+          yield chunk;
+        }
+      })();
+      return new Proxy(result, {
+        get(target, prop) {
+          if (prop === "partialObjectStream") {
+            return wrappedStream;
+          }
+          return target[prop];
+        }
+      });
+    }
+    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 || !initialized2) {
+      return originalGenerate(...args);
+    }
+    let promptCtx = null;
+    try {
+      const { getPromptContext: getPromptContext2 } = await Promise.resolve().then(() => (init_prompts(), prompts_exports));
+      promptCtx = getPromptContext2();
+    } 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 = {};
@@ -1477,7 +2002,7 @@ __export(models_exports, {
 });
 var import_crypto2 = require("crypto");
 var apiKey3 = null;
-var baseUrl3 = "https://spans.fallom.com";
+var baseUrl3 = "https://configs.fallom.com";
 var initialized3 = false;
 var syncInterval2 = null;
 var debugMode3 = false;
@@ -1491,7 +2016,7 @@ function log3(msg) {
 }
 function init3(options = {}) {
   apiKey3 = options.apiKey || process.env.FALLOM_API_KEY || null;
-  baseUrl3 = options.baseUrl || process.env.FALLOM_BASE_URL || "https://spans.fallom.com";
+  baseUrl3 = options.baseUrl || process.env.FALLOM_CONFIGS_URL || process.env.FALLOM_BASE_URL || "https://configs.fallom.com";
   initialized3 = true;
   if (!apiKey3) {
     return;
@@ -1580,20 +2105,28 @@ async function get2(configKey, sessionId, options = {}) {
   const { version, fallback, debug = false } = options;
   debugMode3 = debug;
   ensureInit2();
-  log3(`get() called: configKey=${configKey}, sessionId=${sessionId}, fallback=${fallback}`);
+  log3(
+    `get() called: configKey=${configKey}, sessionId=${sessionId}, fallback=${fallback}`
+  );
   try {
     let configData = configCache.get(configKey);
-    log3(`Cache lookup for '${configKey}': ${configData ? "found" : "not found"}`);
+    log3(
+      `Cache lookup for '${configKey}': ${configData ? "found" : "not found"}`
+    );
     if (!configData) {
       log3("Not in cache, fetching...");
       await fetchConfigs(SYNC_TIMEOUT2);
       configData = configCache.get(configKey);
-      log3(`After fetch, cache lookup: ${configData ? "found" : "still not found"}`);
+      log3(
+        `After fetch, cache lookup: ${configData ? "found" : "still not found"}`
+      );
     }
     if (!configData) {
       log3(`Config not found, using fallback: ${fallback}`);
       if (fallback) {
-        console.warn(`[Fallom WARNING] Config '${configKey}' not found, using fallback model: ${fallback}`);
+        console.warn(
+          `[Fallom WARNING] Config '${configKey}' not found, using fallback model: ${fallback}`
+        );
         return returnWithTrace(configKey, sessionId, fallback, 0);
       }
       throw new Error(
@@ -1609,7 +2142,9 @@ async function get2(configKey, sessionId, options = {}) {
       }
       if (!config) {
         if (fallback) {
-          console.warn(`[Fallom WARNING] Config '${configKey}' version ${version} not found, using fallback: ${fallback}`);
+          console.warn(
+            `[Fallom WARNING] Config '${configKey}' version ${version} not found, using fallback: ${fallback}`
+          );
           return returnWithTrace(configKey, sessionId, fallback, 0);
         }
         throw new Error(`Config '${configKey}' version ${version} not found.`);
@@ -1620,7 +2155,9 @@ async function get2(configKey, sessionId, options = {}) {
       config = configData.versions.get(targetVersion);
       if (!config) {
         if (fallback) {
-          console.warn(`[Fallom WARNING] Config '${configKey}' has no cached version, using fallback: ${fallback}`);
+          console.warn(
+            `[Fallom WARNING] Config '${configKey}' has no cached version, using fallback: ${fallback}`
+          );
           return returnWithTrace(configKey, sessionId, fallback, 0);
         }
         throw new Error(`Config '${configKey}' has no cached version.`);
@@ -1629,7 +2166,11 @@ async function get2(configKey, sessionId, options = {}) {
     const variantsRaw = config.variants;
     const configVersion = config.version || targetVersion;
     const variants = Array.isArray(variantsRaw) ? variantsRaw : Object.values(variantsRaw);
-    log3(`Config found! Version: ${configVersion}, Variants: ${JSON.stringify(variants)}`);
+    log3(
+      `Config found! Version: ${configVersion}, Variants: ${JSON.stringify(
+        variants
+      )}`
+    );
     const hashBytes = (0, import_crypto2.createHash)("md5").update(sessionId).digest();
     const hashVal = hashBytes.readUInt32BE(0) % 1e6;
     log3(`Session hash: ${hashVal} (out of 1,000,000)`);
@@ -1638,7 +2179,9 @@ async function get2(configKey, sessionId, options = {}) {
     for (const v of variants) {
       const oldCumulative = cumulative;
       cumulative += v.weight * 1e4;
-      log3(`Variant ${v.model}: weight=${v.weight}%, range=${oldCumulative}-${cumulative}, hash=${hashVal}, match=${hashVal < cumulative}`);
+      log3(
+        `Variant ${v.model}: weight=${v.weight}%, range=${oldCumulative}-${cumulative}, hash=${hashVal}, match=${hashVal < cumulative}`
+      );
       if (hashVal < cumulative) {
         assignedModel = v.model;
         break;
@@ -1651,7 +2194,9 @@ async function get2(configKey, sessionId, options = {}) {
       throw e;
     }
     if (fallback) {
-      console.warn(`[Fallom WARNING] Error getting model for '${configKey}': ${e}. Using fallback: ${fallback}`);
+      console.warn(
+        `[Fallom WARNING] Error getting model for '${configKey}': ${e}. Using fallback: ${fallback}`
+      );
       return returnWithTrace(configKey, sessionId, fallback, 0);
     }
     throw e;
@@ -1698,23 +2243,260 @@ init_prompts();
 // src/init.ts
 init_prompts();
 async function init4(options = {}) {
-  const baseUrl4 = options.baseUrl || process.env.FALLOM_BASE_URL || "https://spans.fallom.com";
+  const tracesUrl = options.tracesUrl || process.env.FALLOM_TRACES_URL || "https://traces.fallom.com";
+  const configsUrl = options.configsUrl || process.env.FALLOM_CONFIGS_URL || "https://configs.fallom.com";
+  const promptsUrl = options.promptsUrl || process.env.FALLOM_PROMPTS_URL || "https://prompts.fallom.com";
   await init2({
     apiKey: options.apiKey,
-    baseUrl: baseUrl4,
+    baseUrl: tracesUrl,
     captureContent: options.captureContent,
     debug: options.debug
   });
   init3({
     apiKey: options.apiKey,
-    baseUrl: baseUrl4
+    baseUrl: configsUrl
   });
   init({
     apiKey: options.apiKey,
-    baseUrl: baseUrl4
+    baseUrl: promptsUrl
   });
 }
 
+// src/mastra.ts
+var import_core2 = require("@opentelemetry/core");
+var promptContext2 = {};
+function setMastraPrompt(promptKey, version) {
+  promptContext2 = {
+    promptKey,
+    promptVersion: version,
+    promptAbTestKey: void 0,
+    promptVariantIndex: void 0
+  };
+}
+function setMastraPromptAB(abTestKey, variantIndex) {
+  promptContext2 = {
+    promptKey: void 0,
+    promptVersion: void 0,
+    promptAbTestKey: abTestKey,
+    promptVariantIndex: variantIndex
+  };
+}
+function clearMastraPrompt() {
+  promptContext2 = {};
+}
+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: import_core2.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: import_core2.ExportResultCode.SUCCESS });
+    }).catch((error) => {
+      console.error("[FallomExporter] Export failed:", error);
+      resultCallback({
+        code: import_core2.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 (promptContext2.promptKey) {
+      headers["X-Fallom-Prompt-Key"] = promptContext2.promptKey;
+    }
+    if (promptContext2.promptVersion !== void 0) {
+      headers["X-Fallom-Prompt-Version"] = String(promptContext2.promptVersion);
+    }
+    if (promptContext2.promptAbTestKey) {
+      headers["X-Fallom-Prompt-AB-Test"] = promptContext2.promptAbTestKey;
+    }
+    if (promptContext2.promptVariantIndex !== void 0) {
+      headers["X-Fallom-Prompt-Variant"] = String(
+        promptContext2.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
 init_prompts();
 var index_default = {
@@ -1725,8 +2507,12 @@ var index_default = {
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  FallomExporter,
+  clearMastraPrompt,
   init,
   models,
   prompts,
+  setMastraPrompt,
+  setMastraPromptAB,
   trace
 });