@fallom/trace 0.1.10 → 0.1.11

package/dist/index.d.mts

@@ -1,3 +1,6 @@
+import { SpanExporter, ReadableSpan } from '@opentelemetry/sdk-trace-base';
+import { ExportResult } from '@opentelemetry/core';
+
 /**
  * Fallom tracing module.
  *
@@ -222,6 +225,31 @@ declare function wrapAISDK<T extends {
     generateObject: T["generateObject"];
     streamObject: T["streamObject"];
 };
+/**
+ * Wrap a Mastra agent to automatically trace all generate() calls.
+ *
+ * @param agent - The Mastra Agent instance
+ * @returns The same agent with tracing enabled
+ *
+ * @example
+ * ```typescript
+ * import { trace } from "@fallom/trace";
+ * import { Agent } from "@mastra/core";
+ *
+ * await trace.init({ apiKey: "your-key" });
+ *
+ * const agent = new Agent({ ... });
+ * const tracedAgent = trace.wrapMastraAgent(agent);
+ *
+ * trace.setSession("my-app", "session-123", "user-456");
+ * const result = await tracedAgent.generate([{ role: "user", content: "Hello" }]);
+ * // ^ Automatically traced!
+ * ```
+ */
+declare function wrapMastraAgent<T extends {
+    generate: (...args: any[]) => Promise<any>;
+    name?: string;
+}>(agent: T): T;
 
 declare const trace_clearSession: typeof clearSession;
 declare const trace_getSession: typeof getSession;
@@ -232,9 +260,10 @@ declare const trace_span: typeof span;
 declare const trace_wrapAISDK: typeof wrapAISDK;
 declare const trace_wrapAnthropic: typeof wrapAnthropic;
 declare const trace_wrapGoogleAI: typeof wrapGoogleAI;
+declare const trace_wrapMastraAgent: typeof wrapMastraAgent;
 declare const trace_wrapOpenAI: typeof wrapOpenAI;
 declare namespace trace {
-  export { trace_clearSession as clearSession, trace_getSession as getSession, init$3 as init, trace_runWithSession as runWithSession, trace_setSession as setSession, trace_shutdown as shutdown, trace_span as span, trace_wrapAISDK as wrapAISDK, trace_wrapAnthropic as wrapAnthropic, trace_wrapGoogleAI as wrapGoogleAI, trace_wrapOpenAI as wrapOpenAI };
+  export { trace_clearSession as clearSession, trace_getSession as getSession, init$3 as init, trace_runWithSession as runWithSession, trace_setSession as setSession, trace_shutdown as shutdown, trace_span as span, trace_wrapAISDK as wrapAISDK, trace_wrapAnthropic as wrapAnthropic, trace_wrapGoogleAI as wrapGoogleAI, trace_wrapMastraAgent as wrapMastraAgent, trace_wrapOpenAI as wrapOpenAI };
 }
 
 /**
@@ -444,6 +473,112 @@ interface InitOptions {
  */
 declare function init(options?: InitOptions): Promise<void>;
 
+/**
+ * Fallom Exporter for Mastra
+ *
+ * Custom OpenTelemetry exporter that sends traces from Mastra agents to Fallom.
+ * Reads session context from the shared trace module (set via trace.setSession()).
+ *
+ * Usage with Mastra:
+ * ```typescript
+ * import { trace, FallomExporter } from "@fallom/trace";
+ * import { Mastra } from "@mastra/core/mastra";
+ *
+ * // Initialize trace module
+ * await trace.init({ apiKey: process.env.FALLOM_API_KEY });
+ *
+ * // Create Mastra with Fallom exporter
+ * const mastra = new Mastra({
+ *   agents: { myAgent },
+ *   telemetry: {
+ *     serviceName: "my-agent",
+ *     enabled: true,
+ *     export: {
+ *       type: "custom",
+ *       exporter: new FallomExporter(),
+ *     },
+ *   },
+ * });
+ *
+ * // In your request handler:
+ * trace.setSession("my-app", "session-123", "user-456");
+ * const result = await mastra.getAgent("myAgent").generate("Hello!");
+ * ```
+ */
+
+interface FallomExporterOptions {
+    /** Fallom API key. Defaults to FALLOM_API_KEY env var. */
+    apiKey?: string;
+    /** Base URL for traces endpoint (defaults to https://traces.fallom.com) */
+    baseUrl?: string;
+    /** Enable debug logging */
+    debug?: boolean;
+}
+/**
+ * Set prompt tracking info.
+ * Call this after prompts.get() to track which prompt was used.
+ */
+declare function setMastraPrompt(promptKey: string, version?: number): void;
+/**
+ * Set A/B test prompt tracking info.
+ * Call this after prompts.getAB() to track which variant was used.
+ */
+declare function setMastraPromptAB(abTestKey: string, variantIndex: number): void;
+/**
+ * Clear prompt tracking info.
+ */
+declare function clearMastraPrompt(): void;
+/**
+ * OpenTelemetry SpanExporter that sends traces to Fallom.
+ *
+ * Reads session context from trace.setSession() automatically.
+ * Compatible with Mastra's custom exporter interface.
+ */
+declare class FallomExporter implements SpanExporter {
+    private apiKey;
+    private baseUrl;
+    private debug;
+    private pendingExports;
+    constructor(options?: FallomExporterOptions);
+    private log;
+    /**
+     * Export spans to Fallom.
+     */
+    export(spans: ReadableSpan[], resultCallback: (result: ExportResult) => void): void;
+    /**
+     * Shutdown the exporter, waiting for pending exports.
+     */
+    shutdown(): Promise<void>;
+    /**
+     * Force flush pending exports.
+     */
+    forceFlush(): Promise<void>;
+    /**
+     * Send spans to Fallom's OTLP endpoint.
+     */
+    private sendSpans;
+    /**
+     * Convert OpenTelemetry spans to OTLP JSON format.
+     */
+    private spansToOtlpJson;
+    /**
+     * Convert a single span to OTLP format.
+     */
+    private spanToOtlp;
+    /**
+     * Convert attributes to OTLP format.
+     */
+    private attributesToOtlp;
+    /**
+     * Convert a value to OTLP AnyValue format.
+     */
+    private valueToOtlp;
+    /**
+     * Convert HrTime to nanoseconds string.
+     */
+    private hrTimeToNanos;
+}
+
 /**
  * Fallom - Model A/B testing, prompt management, and tracing for LLM applications.
  *
@@ -488,4 +623,4 @@ declare const _default: {
     prompts: typeof prompts;
 };
 
-export { type InitOptions, type PromptResult, _default as default, init, models, prompts, trace };
+export { FallomExporter, type FallomExporterOptions, type InitOptions, type PromptResult, clearMastraPrompt, _default as default, init, models, prompts, setMastraPrompt, setMastraPromptAB, trace };

package/dist/index.d.ts

@@ -1,3 +1,6 @@
+import { SpanExporter, ReadableSpan } from '@opentelemetry/sdk-trace-base';
+import { ExportResult } from '@opentelemetry/core';
+
 /**
  * Fallom tracing module.
  *
@@ -222,6 +225,31 @@ declare function wrapAISDK<T extends {
     generateObject: T["generateObject"];
     streamObject: T["streamObject"];
 };
+/**
+ * Wrap a Mastra agent to automatically trace all generate() calls.
+ *
+ * @param agent - The Mastra Agent instance
+ * @returns The same agent with tracing enabled
+ *
+ * @example
+ * ```typescript
+ * import { trace } from "@fallom/trace";
+ * import { Agent } from "@mastra/core";
+ *
+ * await trace.init({ apiKey: "your-key" });
+ *
+ * const agent = new Agent({ ... });
+ * const tracedAgent = trace.wrapMastraAgent(agent);
+ *
+ * trace.setSession("my-app", "session-123", "user-456");
+ * const result = await tracedAgent.generate([{ role: "user", content: "Hello" }]);
+ * // ^ Automatically traced!
+ * ```
+ */
+declare function wrapMastraAgent<T extends {
+    generate: (...args: any[]) => Promise<any>;
+    name?: string;
+}>(agent: T): T;
 
 declare const trace_clearSession: typeof clearSession;
 declare const trace_getSession: typeof getSession;
@@ -232,9 +260,10 @@ declare const trace_span: typeof span;
 declare const trace_wrapAISDK: typeof wrapAISDK;
 declare const trace_wrapAnthropic: typeof wrapAnthropic;
 declare const trace_wrapGoogleAI: typeof wrapGoogleAI;
+declare const trace_wrapMastraAgent: typeof wrapMastraAgent;
 declare const trace_wrapOpenAI: typeof wrapOpenAI;
 declare namespace trace {
-  export { trace_clearSession as clearSession, trace_getSession as getSession, init$3 as init, trace_runWithSession as runWithSession, trace_setSession as setSession, trace_shutdown as shutdown, trace_span as span, trace_wrapAISDK as wrapAISDK, trace_wrapAnthropic as wrapAnthropic, trace_wrapGoogleAI as wrapGoogleAI, trace_wrapOpenAI as wrapOpenAI };
+  export { trace_clearSession as clearSession, trace_getSession as getSession, init$3 as init, trace_runWithSession as runWithSession, trace_setSession as setSession, trace_shutdown as shutdown, trace_span as span, trace_wrapAISDK as wrapAISDK, trace_wrapAnthropic as wrapAnthropic, trace_wrapGoogleAI as wrapGoogleAI, trace_wrapMastraAgent as wrapMastraAgent, trace_wrapOpenAI as wrapOpenAI };
 }
 
 /**
@@ -444,6 +473,112 @@ interface InitOptions {
  */
 declare function init(options?: InitOptions): Promise<void>;
 
+/**
+ * Fallom Exporter for Mastra
+ *
+ * Custom OpenTelemetry exporter that sends traces from Mastra agents to Fallom.
+ * Reads session context from the shared trace module (set via trace.setSession()).
+ *
+ * Usage with Mastra:
+ * ```typescript
+ * import { trace, FallomExporter } from "@fallom/trace";
+ * import { Mastra } from "@mastra/core/mastra";
+ *
+ * // Initialize trace module
+ * await trace.init({ apiKey: process.env.FALLOM_API_KEY });
+ *
+ * // Create Mastra with Fallom exporter
+ * const mastra = new Mastra({
+ *   agents: { myAgent },
+ *   telemetry: {
+ *     serviceName: "my-agent",
+ *     enabled: true,
+ *     export: {
+ *       type: "custom",
+ *       exporter: new FallomExporter(),
+ *     },
+ *   },
+ * });
+ *
+ * // In your request handler:
+ * trace.setSession("my-app", "session-123", "user-456");
+ * const result = await mastra.getAgent("myAgent").generate("Hello!");
+ * ```
+ */
+
+interface FallomExporterOptions {
+    /** Fallom API key. Defaults to FALLOM_API_KEY env var. */
+    apiKey?: string;
+    /** Base URL for traces endpoint (defaults to https://traces.fallom.com) */
+    baseUrl?: string;
+    /** Enable debug logging */
+    debug?: boolean;
+}
+/**
+ * Set prompt tracking info.
+ * Call this after prompts.get() to track which prompt was used.
+ */
+declare function setMastraPrompt(promptKey: string, version?: number): void;
+/**
+ * Set A/B test prompt tracking info.
+ * Call this after prompts.getAB() to track which variant was used.
+ */
+declare function setMastraPromptAB(abTestKey: string, variantIndex: number): void;
+/**
+ * Clear prompt tracking info.
+ */
+declare function clearMastraPrompt(): void;
+/**
+ * OpenTelemetry SpanExporter that sends traces to Fallom.
+ *
+ * Reads session context from trace.setSession() automatically.
+ * Compatible with Mastra's custom exporter interface.
+ */
+declare class FallomExporter implements SpanExporter {
+    private apiKey;
+    private baseUrl;
+    private debug;
+    private pendingExports;
+    constructor(options?: FallomExporterOptions);
+    private log;
+    /**
+     * Export spans to Fallom.
+     */
+    export(spans: ReadableSpan[], resultCallback: (result: ExportResult) => void): void;
+    /**
+     * Shutdown the exporter, waiting for pending exports.
+     */
+    shutdown(): Promise<void>;
+    /**
+     * Force flush pending exports.
+     */
+    forceFlush(): Promise<void>;
+    /**
+     * Send spans to Fallom's OTLP endpoint.
+     */
+    private sendSpans;
+    /**
+     * Convert OpenTelemetry spans to OTLP JSON format.
+     */
+    private spansToOtlpJson;
+    /**
+     * Convert a single span to OTLP format.
+     */
+    private spanToOtlp;
+    /**
+     * Convert attributes to OTLP format.
+     */
+    private attributesToOtlp;
+    /**
+     * Convert a value to OTLP AnyValue format.
+     */
+    private valueToOtlp;
+    /**
+     * Convert HrTime to nanoseconds string.
+     */
+    private hrTimeToNanos;
+}
+
 /**
  * Fallom - Model A/B testing, prompt management, and tracing for LLM applications.
  *
@@ -488,4 +623,4 @@ declare const _default: {
     prompts: typeof prompts;
 };
 
-export { type InitOptions, type PromptResult, _default as default, init, models, prompts, trace };
+export { FallomExporter, type FallomExporterOptions, type InitOptions, type PromptResult, clearMastraPrompt, _default as default, init, models, prompts, setMastraPrompt, setMastraPromptAB, trace };

package/dist/index.js

@@ -269,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);
@@ -290,6 +294,7 @@ __export(trace_exports, {
   wrapAISDK: () => wrapAISDK,
   wrapAnthropic: () => wrapAnthropic,
   wrapGoogleAI: () => wrapGoogleAI,
+  wrapMastraAgent: () => wrapMastraAgent,
   wrapOpenAI: () => wrapOpenAI
 });
 var import_async_hooks = require("async_hooks");
@@ -299,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.
@@ -679,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() {
@@ -844,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;
@@ -862,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
@@ -897,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);
@@ -1893,6 +1872,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 || !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 = {};
@@ -2162,6 +2262,241 @@ async function init4(options = {}) {
   });
 }
 
+// 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 = {
@@ -2172,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
 });

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);
@@ -1620,6 +1595,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 +1981,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 +2224,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
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fallom/trace",
-  "version": "0.1.10",
+  "version": "0.1.11",
   "description": "Model A/B testing and tracing for LLM applications. Zero latency, production-ready.",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -44,7 +44,9 @@
   },
   "dependencies": {
     "@opentelemetry/api": "^1.7.0",
+    "@opentelemetry/core": "^1.19.0",
     "@opentelemetry/sdk-node": "^0.46.0",
+    "@opentelemetry/sdk-trace-base": "^1.19.0",
     "@opentelemetry/sdk-trace-node": "^1.19.0",
     "@opentelemetry/exporter-trace-otlp-http": "^0.46.0",
     "@traceloop/instrumentation-openai": "^0.11.0",