@mastra/langfuse 0.0.4 → 0.0.5-alpha.0

package/.turbo/turbo-build.log

@@ -1,4 +1,4 @@
 
-> @mastra/langfuse@0.0.3 build /home/runner/work/mastra/mastra/observability/langfuse
+> @mastra/langfuse@0.0.5-alpha.0 build /home/runner/work/mastra/mastra/observability/langfuse
 > tsup --silent --config tsup.config.ts
 

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/langfuse
 
+## 0.0.5-alpha.0
+
+### Patch Changes
+
+- [#7266](https://github.com/mastra-ai/mastra/pull/7266) [`979912c`](https://github.com/mastra-ai/mastra/commit/979912cfd180aad53287cda08af771df26454e2c) Thanks [@epinzur](https://github.com/epinzur)! - "Updated langfuse exporter to handle Event spans"
+
+- Updated dependencies [[`ab48c97`](https://github.com/mastra-ai/mastra/commit/ab48c979098ea571faf998a55d3a00e7acd7a715), [`ff89505`](https://github.com/mastra-ai/mastra/commit/ff895057c8c7e91a5535faef46c5e5391085ddfa), [`183dc95`](https://github.com/mastra-ai/mastra/commit/183dc95596f391b977bd1a2c050b8498dac74891), [`a1111e2`](https://github.com/mastra-ai/mastra/commit/a1111e24e705488adfe5e0a6f20c53bddf26cb22), [`61debef`](https://github.com/mastra-ai/mastra/commit/61debefd80ad3a7ed5737e19df6a23d40091689a), [`9beaeff`](https://github.com/mastra-ai/mastra/commit/9beaeffa4a97b1d5fd01a7f8af8708b16067f67c), [`9eee594`](https://github.com/mastra-ai/mastra/commit/9eee594e35e0ca2a650fcc33fa82009a142b9ed0), [`979912c`](https://github.com/mastra-ai/mastra/commit/979912cfd180aad53287cda08af771df26454e2c), [`7dcf4c0`](https://github.com/mastra-ai/mastra/commit/7dcf4c04f44d9345b1f8bc5d41eae3f11ac61611), [`ad78bfc`](https://github.com/mastra-ai/mastra/commit/ad78bfc4ea6a1fff140432bf4f638e01af7af668), [`0ce418a`](https://github.com/mastra-ai/mastra/commit/0ce418a1ccaa5e125d4483a9651b635046152569), [`8387952`](https://github.com/mastra-ai/mastra/commit/838795227b4edf758c84a2adf6f7fba206c27719), [`5eca5d2`](https://github.com/mastra-ai/mastra/commit/5eca5d2655788863ea0442a46c9ef5d3c6dbe0a8)]:
+  - @mastra/core@0.15.3-alpha.4
+
 ## 0.0.4
 
 ### Patch Changes

package/dist/ai-tracing.d.ts

@@ -15,6 +15,8 @@ export interface LangfuseExporterConfig {
     baseUrl: string;
     /** Enable realtime mode - flushes after each event for immediate visibility */
     realtime?: boolean;
+    /** Logger level for diagnostic messages (default: 'warn') */
+    logLevel?: 'debug' | 'info' | 'warn' | 'error';
     /** Additional options to pass to the Langfuse client */
     options?: any;
 }
@@ -23,10 +25,15 @@ export declare class LangfuseExporter implements AITracingExporter {
     private client;
     private realtime;
     private traceMap;
+    private logger;
     constructor(config: LangfuseExporterConfig);
     exportEvent(event: AITracingEvent): Promise<void>;
     private handleSpanStarted;
     private handleSpanUpdateOrEnd;
+    private handleEventSpan;
+    private initTrace;
+    private getTraceData;
+    private getLangfuseParent;
     private buildTracePayload;
     private buildSpanPayload;
     shutdown(): Promise<void>;

package/dist/ai-tracing.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai-tracing.d.ts","sourceRoot":"","sources":["../src/ai-tracing.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAsC,MAAM,yBAAyB,CAAC;AAKrH,MAAM,WAAW,sBAAsB;IACrC,uBAAuB;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,0BAA0B;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,wBAAwB;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,+EAA+E;IAC/E,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,wDAAwD;IACxD,OAAO,CAAC,EAAE,GAAG,CAAC;CACf;AASD,qBAAa,gBAAiB,YAAW,iBAAiB;IACxD,IAAI,SAAc;IAClB,OAAO,CAAC,MAAM,CAAW;IACzB,OAAO,CAAC,QAAQ,CAAU;IAC1B,OAAO,CAAC,QAAQ,CAAgC;gBAEpC,MAAM,EAAE,sBAAsB;IAUpC,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;YAmBzC,iBAAiB;YA0BjB,qBAAqB;IA2BnC,OAAO,CAAC,iBAAiB;IAqBzB,OAAO,CAAC,gBAAgB;IAmDlB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAIhC"}
+{"version":3,"file":"ai-tracing.d.ts","sourceRoot":"","sources":["../src/ai-tracing.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAsC,MAAM,yBAAyB,CAAC;AAMrH,MAAM,WAAW,sBAAsB;IACrC,uBAAuB;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,0BAA0B;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,wBAAwB;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,+EAA+E;IAC/E,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,6DAA6D;IAC7D,QAAQ,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;IAC/C,wDAAwD;IACxD,OAAO,CAAC,EAAE,GAAG,CAAC;CACf;AAUD,qBAAa,gBAAiB,YAAW,iBAAiB;IACxD,IAAI,SAAc;IAClB,OAAO,CAAC,MAAM,CAAW;IACzB,OAAO,CAAC,QAAQ,CAAU;IAC1B,OAAO,CAAC,QAAQ,CAAgC;IAChD,OAAO,CAAC,MAAM,CAAgB;gBAElB,MAAM,EAAE,sBAAsB;IAWpC,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;YAwBzC,iBAAiB;YAwBjB,qBAAqB;YAiCrB,eAAe;IA6B7B,OAAO,CAAC,SAAS;IAKjB,OAAO,CAAC,YAAY;IAgBpB,OAAO,CAAC,iBAAiB;IA4BzB,OAAO,CAAC,iBAAiB;IAqBzB,OAAO,CAAC,gBAAgB;IAmDlB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAIhC"}

package/dist/index.cjs

@@ -1,6 +1,7 @@
 'use strict';
 
 var aiTracing = require('@mastra/core/ai-tracing');
+var logger = require('@mastra/core/logger');
 var langfuse = require('langfuse');
 
 // src/ai-tracing.ts
@@ -9,8 +10,10 @@ var LangfuseExporter = class {
   client;
   realtime;
   traceMap = /* @__PURE__ */ new Map();
+  logger;
   constructor(config) {
     this.realtime = config.realtime ?? false;
+    this.logger = new logger.ConsoleLogger({ level: config.logLevel ?? "warn" });
     this.client = new langfuse.Langfuse({
       publicKey: config.publicKey,
       secretKey: config.secretKey,
@@ -19,15 +22,19 @@ var LangfuseExporter = class {
     });
   }
   async exportEvent(event) {
+    if (event.span.isEvent) {
+      await this.handleEventSpan(event.span);
+      return;
+    }
     switch (event.type) {
       case "span_started":
         await this.handleSpanStarted(event.span);
         break;
       case "span_updated":
-        await this.handleSpanUpdateOrEnd(event.span, true);
+        await this.handleSpanUpdateOrEnd(event.span, false);
         break;
       case "span_ended":
-        await this.handleSpanUpdateOrEnd(event.span, false);
+        await this.handleSpanUpdateOrEnd(event.span, true);
         break;
     }
     if (this.realtime) {
@@ -36,43 +43,114 @@ var LangfuseExporter = class {
   }
   async handleSpanStarted(span) {
     if (span.isRootSpan) {
-      const trace = this.client.trace(this.buildTracePayload(span));
-      this.traceMap.set(span.trace.id, { trace, spans: /* @__PURE__ */ new Map() });
+      this.initTrace(span);
     }
-    const traceData = this.traceMap.get(span.trace.id);
+    const method = "handleSpanStarted";
+    const traceData = this.getTraceData({ span, method });
     if (!traceData) {
-      console.log("NO TRACE");
       return;
     }
-    const langfuseParent = span.parent && traceData.spans.has(span.parent.id) ? traceData.spans.get(span.parent.id) : traceData.trace;
+    const langfuseParent = this.getLangfuseParent({ traceData, span, method });
+    if (!langfuseParent) {
+      return;
+    }
     const payload = this.buildSpanPayload(span, true);
     const langfuseSpan = span.type === aiTracing.AISpanType.LLM_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);
     traceData.spans.set(span.id, langfuseSpan);
   }
-  async handleSpanUpdateOrEnd(span, isUpdate) {
-    const traceData = this.traceMap.get(span.trace.id);
+  async handleSpanUpdateOrEnd(span, isEnd) {
+    const method = isEnd ? "handleSpanEnd" : "handleSpanUpdate";
+    const traceData = this.getTraceData({ span, method });
     if (!traceData) {
-      console.log("NO TRACE");
       return;
     }
     const langfuseSpan = traceData.spans.get(span.id);
     if (!langfuseSpan) {
-      console.log("NO SPAN");
+      this.logger.warn("Langfuse exporter: No Langfuse span found for span update/end", {
+        traceId: span.traceId,
+        spanId: span.id,
+        spanName: span.name,
+        spanType: span.type,
+        isRootSpan: span.isRootSpan,
+        parentSpanId: span.parent?.id,
+        availableSpanIds: Array.from(traceData.spans.keys()),
+        method
+      });
       return;
     }
-    if (isUpdate) {
-      langfuseSpan.update(this.buildSpanPayload(span, false));
-    } else {
-      langfuseSpan.end(this.buildSpanPayload(span, false));
-      if (span.isRootSpan) {
-        traceData.trace.update({ output: span.output });
-        this.traceMap.delete(span.trace.id);
-      }
+    langfuseSpan.update(this.buildSpanPayload(span, false));
+    if (isEnd && span.isRootSpan) {
+      traceData.trace.update({ output: span.output });
+      this.traceMap.delete(span.traceId);
+    }
+  }
+  async handleEventSpan(span) {
+    if (span.isRootSpan) {
+      this.logger.debug("Langfuse exporter: Creating trace", {
+        traceId: span.traceId,
+        spanId: span.id,
+        spanName: span.name,
+        method: "handleEventSpan"
+      });
+      this.initTrace(span);
+    }
+    const method = "handleEventSpan";
+    const traceData = this.getTraceData({ span, method });
+    if (!traceData) {
+      return;
     }
+    const langfuseParent = this.getLangfuseParent({ traceData, span, method });
+    if (!langfuseParent) {
+      return;
+    }
+    const payload = this.buildSpanPayload(span, true);
+    const langfuseEvent = langfuseParent.event(payload);
+    traceData.events.set(span.id, langfuseEvent);
+  }
+  initTrace(span) {
+    const trace = this.client.trace(this.buildTracePayload(span));
+    this.traceMap.set(span.traceId, { trace, spans: /* @__PURE__ */ new Map(), events: /* @__PURE__ */ new Map() });
+  }
+  getTraceData(options) {
+    const { span, method } = options;
+    if (this.traceMap.has(span.traceId)) {
+      return this.traceMap.get(span.traceId);
+    }
+    this.logger.warn("Langfuse exporter: No trace data found for span", {
+      traceId: span.traceId,
+      spanId: span.id,
+      spanName: span.name,
+      spanType: span.type,
+      isRootSpan: span.isRootSpan,
+      parentSpanId: span.parent?.id,
+      method
+    });
+  }
+  getLangfuseParent(options) {
+    const { traceData, span, method } = options;
+    const parentId = span.parent?.id;
+    if (!parentId) {
+      return traceData.trace;
+    }
+    if (traceData.spans.has(parentId)) {
+      return traceData.spans.get(parentId);
+    }
+    if (traceData.events.has(parentId)) {
+      return traceData.events.get(parentId);
+    }
+    this.logger.warn("Langfuse exporter: No parent data found for span", {
+      traceId: span.traceId,
+      spanId: span.id,
+      spanName: span.name,
+      spanType: span.type,
+      isRootSpan: span.isRootSpan,
+      parentSpanId: span.parent?.id,
+      method
+    });
   }
   buildTracePayload(span) {
     const payload = {
-      id: span.trace.id,
+      id: span.traceId,
       name: span.name
     };
     const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};
@@ -81,8 +159,8 @@ var LangfuseExporter = class {
     if (span.input) payload.input = span.input;
     payload.metadata = {
       spanType: span.type,
-      ...aiTracing.sanitizeMetadata(span.attributes),
-      ...aiTracing.sanitizeMetadata(remainingMetadata)
+      ...span.attributes,
+      ...remainingMetadata
     };
     return payload;
   }
@@ -115,8 +193,8 @@ var LangfuseExporter = class {
     }
     payload.metadata = {
       spanType: span.type,
-      ...aiTracing.sanitizeMetadata(aiTracing.omitKeys(attributes, attributesToOmit)),
-      ...aiTracing.sanitizeMetadata(span.metadata)
+      ...aiTracing.omitKeys(attributes, attributesToOmit),
+      ...span.metadata
     };
     if (span.errorInfo) {
       payload.level = "ERROR";

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/ai-tracing.ts"],"names":["Langfuse","AISpanType","sanitizeMetadata","omitKeys"],"mappings":";;;;;;AAiCO,IAAM,mBAAN,MAAoD;AAAA,EACzD,IAAA,GAAO,UAAA;AAAA,EACC,MAAA;AAAA,EACA,QAAA;AAAA,EACA,QAAA,uBAAe,GAAA,EAAuB;AAAA,EAE9C,YAAY,MAAA,EAAgC;AAC1C,IAAA,IAAA,CAAK,QAAA,GAAW,OAAO,QAAA,IAAY,KAAA;AACnC,IAAA,IAAA,CAAK,MAAA,GAAS,IAAIA,iBAAA,CAAS;AAAA,MACzB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,SAAS,MAAA,CAAO,OAAA;AAAA,MAChB,GAAG,MAAA,CAAO;AAAA,KACX,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA;AAIJ,IAAA,IAAI,KAAK,QAAA,EAAU;AACjB,MAAA,MAAM,IAAA,CAAK,OAAO,UAAA,EAAW;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAM,QAAQ,IAAA,CAAK,MAAA,CAAO,MAAM,IAAA,CAAK,iBAAA,CAAkB,IAAI,CAAC,CAAA;AAC5D,MAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,KAAA,CAAM,EAAA,EAAI,EAAE,KAAA,EAAO,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,YAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,MAAM,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,OAAA,CAAQ,IAAI,UAAU,CAAA;AAEtB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,iBACJ,IAAA,CAAK,MAAA,IAAU,SAAA,CAAU,KAAA,CAAM,IAAI,IAAA,CAAK,MAAA,CAAO,EAAE,CAAA,GAC5C,UAAU,KAAA,CAAM,GAAA,CAAI,KAAK,MAAA,CAAO,EAAE,IACnC,SAAA,CAAU,KAAA;AAEhB,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,IAAI,CAAA;AAEhD,IAAA,MAAM,YAAA,GACJ,IAAA,CAAK,IAAA,KAASC,oBAAA,CAAW,cAAA,GAAiB,cAAA,CAAe,UAAA,CAAW,OAAO,CAAA,GAAI,cAAA,CAAe,IAAA,CAAK,OAAO,CAAA;AAE5G,IAAA,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,YAAY,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,QAAA,EAAkC;AACrF,IAAA,MAAM,YAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,MAAM,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,OAAA,CAAQ,IAAI,UAAU,CAAA;AAEtB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,YAAA,GAAe,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AAChD,IAAA,IAAI,CAAC,YAAA,EAAc;AACjB,MAAA,OAAA,CAAQ,IAAI,SAAS,CAAA;AAErB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,YAAA,CAAa,MAAA,CAAO,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAK,CAAC,CAAA;AAAA,IACxD,CAAA,MAAO;AACL,MAAA,YAAA,CAAa,GAAA,CAAI,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAK,CAAC,CAAA;AAEnD,MAAA,IAAI,KAAK,UAAA,EAAY;AACnB,QAAA,SAAA,CAAU,MAAM,MAAA,CAAO,EAAE,MAAA,EAAQ,IAAA,CAAK,QAAQ,CAAA;AAC9C,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,KAAA,CAAM,EAAE,CAAA;AAAA,MACpC;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,kBAAkB,IAAA,EAAsC;AAC9D,IAAA,MAAM,OAAA,GAA+B;AAAA,MACnC,EAAA,EAAI,KAAK,KAAA,CAAM,EAAA;AAAA,MACf,MAAM,IAAA,CAAK;AAAA,KACb;AAEA,IAAA,MAAM,EAAE,QAAQ,SAAA,EAAW,GAAG,mBAAkB,GAAI,IAAA,CAAK,YAAY,EAAC;AAEtE,IAAA,IAAI,MAAA,UAAgB,MAAA,GAAS,MAAA;AAC7B,IAAA,IAAI,SAAA,UAAmB,SAAA,GAAY,SAAA;AACnC,IAAA,IAAI,IAAA,CAAK,KAAA,EAAO,OAAA,CAAQ,KAAA,GAAQ,IAAA,CAAK,KAAA;AAErC,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAGC,0BAAA,CAAiB,IAAA,CAAK,UAAU,CAAA;AAAA,MACnC,GAAGA,2BAAiB,iBAAiB;AAAA,KACvC;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEQ,gBAAA,CAAiB,MAAiB,QAAA,EAAwC;AAChF,IAAA,MAAM,UAA+B,EAAC;AAEtC,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,OAAA,CAAQ,KAAK,IAAA,CAAK,EAAA;AAClB,MAAA,OAAA,CAAQ,OAAO,IAAA,CAAK,IAAA;AACpB,MAAA,OAAA,CAAQ,YAAY,IAAA,CAAK,SAAA;AACzB,MAAA,IAAI,IAAA,CAAK,KAAA,KAAU,MAAA,EAAW,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACrD;AAEA,IAAA,IAAI,IAAA,CAAK,MAAA,KAAW,MAAA,EAAW,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AACrD,IAAA,IAAI,IAAA,CAAK,OAAA,KAAY,MAAA,EAAW,OAAA,CAAQ,UAAU,IAAA,CAAK,OAAA;AAEvD,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAGxC,IAAA,MAAM,mBAA6B,EAAC;AAEpC,IAAA,IAAI,IAAA,CAAK,IAAA,KAASD,oBAAA,CAAW,cAAA,EAAgB;AAC3C,MAAA,MAAM,OAAA,GAAU,UAAA;AAEhB,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,eAAe,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,kBAAkB,OAAA,CAAQ,UAAA;AAClC,QAAA,gBAAA,CAAiB,KAAK,YAAY,CAAA;AAAA,MACpC;AAAA,IACF;AAEA,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAGC,0BAAA,CAAiBC,kBAAA,CAAS,UAAA,EAAY,gBAAgB,CAAC,CAAA;AAAA,MAC1D,GAAGD,0BAAA,CAAiB,IAAA,CAAK,QAAQ;AAAA,KACnC;AAEA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,OAAA,CAAQ,KAAA,GAAQ,OAAA;AAChB,MAAA,OAAA,CAAQ,aAAA,GAAgB,KAAK,SAAA,CAAU,OAAA;AAAA,IACzC;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,MAAM,IAAA,CAAK,OAAO,aAAA,EAAc;AAChC,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AAAA,EACtB;AACF","file":"index.cjs","sourcesContent":["/**\n * Langfuse Exporter for Mastra AI Tracing\n *\n * This exporter sends tracing data to Langfuse for AI observability.\n * Root spans start traces in Langfuse.\n * LLM_GENERATION spans become Langfuse generations, all others become spans.\n */\n\nimport type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, sanitizeMetadata, omitKeys } from '@mastra/core/ai-tracing';\nimport { Langfuse } from 'langfuse';\nimport type { LangfuseTraceClient, LangfuseSpanClient, LangfuseGenerationClient } from 'langfuse';\n\nexport interface LangfuseExporterConfig {\n  /** Langfuse API key */\n  publicKey: string;\n  /** Langfuse secret key */\n  secretKey: string;\n  /** Langfuse host URL */\n  baseUrl: string;\n  /** Enable realtime mode - flushes after each event for immediate visibility */\n  realtime?: boolean;\n  /** Additional options to pass to the Langfuse client */\n  options?: any;\n}\n\ntype TraceData = {\n  trace: LangfuseTraceClient; // Langfuse trace object\n  spans: Map<string, LangfuseSpanClient | LangfuseGenerationClient>; // Maps span.id to Langfuse span/generation\n};\n\ntype LangfuseSpan = LangfuseTraceClient | LangfuseSpanClient | LangfuseGenerationClient;\n\nexport class LangfuseExporter implements AITracingExporter {\n  name = 'langfuse';\n  private client: Langfuse;\n  private realtime: boolean;\n  private traceMap = new Map<string, TraceData>();\n\n  constructor(config: LangfuseExporterConfig) {\n    this.realtime = config.realtime ?? false;\n    this.client = new Langfuse({\n      publicKey: config.publicKey,\n      secretKey: config.secretKey,\n      baseUrl: config.baseUrl,\n      ...config.options,\n    });\n  }\n\n  async exportEvent(event: AITracingEvent): Promise<void> {\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.span);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.span, true);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.span, false);\n        break;\n    }\n\n    // Flush immediately in realtime mode for instant visibility\n    if (this.realtime) {\n      await this.client.flushAsync();\n    }\n  }\n\n  private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      const trace = this.client.trace(this.buildTracePayload(span));\n      this.traceMap.set(span.trace.id, { trace, spans: new Map() });\n    }\n\n    const traceData = this.traceMap.get(span.trace.id);\n    if (!traceData) {\n      console.log('NO TRACE');\n      // TODO: log warning\n      return;\n    }\n\n    const langfuseParent =\n      span.parent && traceData.spans.has(span.parent.id)\n        ? (traceData.spans.get(span.parent.id) as LangfuseSpan)\n        : traceData.trace;\n\n    const payload = this.buildSpanPayload(span, true);\n\n    const langfuseSpan =\n      span.type === AISpanType.LLM_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);\n\n    traceData.spans.set(span.id, langfuseSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyAISpan, isUpdate: boolean): Promise<void> {\n    const traceData = this.traceMap.get(span.trace.id);\n    if (!traceData) {\n      console.log('NO TRACE');\n      // TODO: log warning\n      return;\n    }\n\n    const langfuseSpan = traceData.spans.get(span.id);\n    if (!langfuseSpan) {\n      console.log('NO SPAN');\n      // TODO: log warning\n      return;\n    }\n\n    if (isUpdate) {\n      langfuseSpan.update(this.buildSpanPayload(span, false));\n    } else {\n      langfuseSpan.end(this.buildSpanPayload(span, false));\n\n      if (span.isRootSpan) {\n        traceData.trace.update({ output: span.output });\n        this.traceMap.delete(span.trace.id);\n      }\n    }\n  }\n\n  private buildTracePayload(span: AnyAISpan): Record<string, any> {\n    const payload: Record<string, any> = {\n      id: span.trace.id,\n      name: span.name,\n    };\n\n    const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};\n\n    if (userId) payload.userId = userId;\n    if (sessionId) payload.sessionId = sessionId;\n    if (span.input) payload.input = span.input;\n\n    payload.metadata = {\n      spanType: span.type,\n      ...sanitizeMetadata(span.attributes),\n      ...sanitizeMetadata(remainingMetadata),\n    };\n\n    return payload;\n  }\n\n  private buildSpanPayload(span: AnyAISpan, isCreate: boolean): Record<string, any> {\n    const payload: Record<string, any> = {};\n\n    if (isCreate) {\n      payload.id = span.id;\n      payload.name = span.name;\n      payload.startTime = span.startTime;\n      if (span.input !== undefined) payload.input = span.input;\n    }\n\n    if (span.output !== undefined) payload.output = span.output;\n    if (span.endTime !== undefined) payload.endTime = span.endTime;\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    // Strip special fields from metadata if used in top-level keys\n    const attributesToOmit: string[] = [];\n\n    if (span.type === AISpanType.LLM_GENERATION) {\n      const llmAttr = attributes as LLMGenerationAttributes;\n\n      if (llmAttr.model !== undefined) {\n        payload.model = llmAttr.model;\n        attributesToOmit.push('model');\n      }\n\n      if (llmAttr.usage !== undefined) {\n        payload.usage = llmAttr.usage;\n        attributesToOmit.push('usage');\n      }\n\n      if (llmAttr.parameters !== undefined) {\n        payload.modelParameters = llmAttr.parameters;\n        attributesToOmit.push('parameters');\n      }\n    }\n\n    payload.metadata = {\n      spanType: span.type,\n      ...sanitizeMetadata(omitKeys(attributes, attributesToOmit)),\n      ...sanitizeMetadata(span.metadata),\n    };\n\n    if (span.errorInfo) {\n      payload.level = 'ERROR';\n      payload.statusMessage = span.errorInfo.message;\n    }\n\n    return payload;\n  }\n\n  async shutdown(): Promise<void> {\n    await this.client.shutdownAsync();\n    this.traceMap.clear();\n  }\n}\n"]}
+{"version":3,"sources":["../src/ai-tracing.ts"],"names":["ConsoleLogger","Langfuse","AISpanType","omitKeys"],"mappings":";;;;;;;AAqCO,IAAM,mBAAN,MAAoD;AAAA,EACzD,IAAA,GAAO,UAAA;AAAA,EACC,MAAA;AAAA,EACA,QAAA;AAAA,EACA,QAAA,uBAAe,GAAA,EAAuB;AAAA,EACtC,MAAA;AAAA,EAER,YAAY,MAAA,EAAgC;AAC1C,IAAA,IAAA,CAAK,QAAA,GAAW,OAAO,QAAA,IAAY,KAAA;AACnC,IAAA,IAAA,CAAK,MAAA,GAAS,IAAIA,oBAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AACpE,IAAA,IAAA,CAAK,MAAA,GAAS,IAAIC,iBAAA,CAAS;AAAA,MACzB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,SAAS,MAAA,CAAO,OAAA;AAAA,MAChB,GAAG,MAAA,CAAO;AAAA,KACX,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,IAAI,KAAA,CAAM,KAAK,OAAA,EAAS;AACtB,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,IAAI,CAAA;AACrC,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA;AAIJ,IAAA,IAAI,KAAK,QAAA,EAAU;AACjB,MAAA,MAAM,IAAA,CAAK,OAAO,UAAA,EAAW;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,UAAU,IAAI,CAAA;AAAA,IACrB;AACA,IAAA,MAAM,MAAA,GAAS,mBAAA;AAEf,IAAA,MAAM,YAAY,IAAA,CAAK,YAAA,CAAa,EAAE,IAAA,EAAM,QAAQ,CAAA;AACpD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,iBAAiB,IAAA,CAAK,iBAAA,CAAkB,EAAE,SAAA,EAAW,IAAA,EAAM,QAAQ,CAAA;AACzE,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,IAAI,CAAA;AAEhD,IAAA,MAAM,YAAA,GACJ,IAAA,CAAK,IAAA,KAASC,oBAAA,CAAW,cAAA,GAAiB,cAAA,CAAe,UAAA,CAAW,OAAO,CAAA,GAAI,cAAA,CAAe,IAAA,CAAK,OAAO,CAAA;AAE5G,IAAA,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,YAAY,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,KAAA,EAA+B;AAClF,IAAA,MAAM,MAAA,GAAS,QAAQ,eAAA,GAAkB,kBAAA;AAEzC,IAAA,MAAM,YAAY,IAAA,CAAK,YAAA,CAAa,EAAE,IAAA,EAAM,QAAQ,CAAA;AACpD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,YAAA,GAAe,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AAChD,IAAA,IAAI,CAAC,YAAA,EAAc;AACjB,MAAA,IAAA,CAAK,MAAA,CAAO,KAAK,+DAAA,EAAiE;AAAA,QAChF,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,YAAY,IAAA,CAAK,UAAA;AAAA,QACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,QAC3B,kBAAkB,KAAA,CAAM,IAAA,CAAK,SAAA,CAAU,KAAA,CAAM,MAAM,CAAA;AAAA,QACnD;AAAA,OACD,CAAA;AACD,MAAA;AAAA,IACF;AAIA,IAAA,YAAA,CAAa,MAAA,CAAO,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAK,CAAC,CAAA;AAEtD,IAAA,IAAI,KAAA,IAAS,KAAK,UAAA,EAAY;AAC5B,MAAA,SAAA,CAAU,MAAM,MAAA,CAAO,EAAE,MAAA,EAAQ,IAAA,CAAK,QAAQ,CAAA;AAC9C,MAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,IACnC;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAgC;AAC5D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,mCAAA,EAAqC;AAAA,QACrD,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,MAAA,EAAQ;AAAA,OACT,CAAA;AACD,MAAA,IAAA,CAAK,UAAU,IAAI,CAAA;AAAA,IACrB;AACA,IAAA,MAAM,MAAA,GAAS,iBAAA;AAEf,IAAA,MAAM,YAAY,IAAA,CAAK,YAAA,CAAa,EAAE,IAAA,EAAM,QAAQ,CAAA;AACpD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,iBAAiB,IAAA,CAAK,iBAAA,CAAkB,EAAE,SAAA,EAAW,IAAA,EAAM,QAAQ,CAAA;AACzE,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,IAAI,CAAA;AAEhD,IAAA,MAAM,aAAA,GAAgB,cAAA,CAAe,KAAA,CAAM,OAAO,CAAA;AAElD,IAAA,SAAA,CAAU,MAAA,CAAO,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,aAAa,CAAA;AAAA,EAC7C;AAAA,EAEQ,UAAU,IAAA,EAAuB;AACvC,IAAA,MAAM,QAAQ,IAAA,CAAK,MAAA,CAAO,MAAM,IAAA,CAAK,iBAAA,CAAkB,IAAI,CAAC,CAAA;AAC5D,IAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS,EAAE,KAAA,EAAO,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,MAAA,kBAAQ,IAAI,GAAA,IAAO,CAAA;AAAA,EAChF;AAAA,EAEQ,aAAa,OAAA,EAAqE;AACxF,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AACzB,IAAA,IAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA,EAAG;AACnC,MAAA,OAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACvC;AACA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,iDAAA,EAAmD;AAAA,MAClE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,kBAAkB,OAAA,EAIK;AAC7B,IAAA,MAAM,EAAE,SAAA,EAAW,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AAEpC,IAAA,MAAM,QAAA,GAAW,KAAK,MAAA,EAAQ,EAAA;AAC9B,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,SAAA,CAAU,KAAA;AAAA,IACnB;AACA,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA,EAAG;AACjC,MAAA,OAAO,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA;AAAA,IACrC;AACA,IAAA,IAAI,SAAA,CAAU,MAAA,CAAO,GAAA,CAAI,QAAQ,CAAA,EAAG;AAClC,MAAA,OAAO,SAAA,CAAU,MAAA,CAAO,GAAA,CAAI,QAAQ,CAAA;AAAA,IACtC;AACA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,kDAAA,EAAoD;AAAA,MACnE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,kBAAkB,IAAA,EAAsC;AAC9D,IAAA,MAAM,OAAA,GAA+B;AAAA,MACnC,IAAI,IAAA,CAAK,OAAA;AAAA,MACT,MAAM,IAAA,CAAK;AAAA,KACb;AAEA,IAAA,MAAM,EAAE,QAAQ,SAAA,EAAW,GAAG,mBAAkB,GAAI,IAAA,CAAK,YAAY,EAAC;AAEtE,IAAA,IAAI,MAAA,UAAgB,MAAA,GAAS,MAAA;AAC7B,IAAA,IAAI,SAAA,UAAmB,SAAA,GAAY,SAAA;AACnC,IAAA,IAAI,IAAA,CAAK,KAAA,EAAO,OAAA,CAAQ,KAAA,GAAQ,IAAA,CAAK,KAAA;AAErC,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,IAAA,CAAK,UAAA;AAAA,MACR,GAAG;AAAA,KACL;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEQ,gBAAA,CAAiB,MAAiB,QAAA,EAAwC;AAChF,IAAA,MAAM,UAA+B,EAAC;AAEtC,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,OAAA,CAAQ,KAAK,IAAA,CAAK,EAAA;AAClB,MAAA,OAAA,CAAQ,OAAO,IAAA,CAAK,IAAA;AACpB,MAAA,OAAA,CAAQ,YAAY,IAAA,CAAK,SAAA;AACzB,MAAA,IAAI,IAAA,CAAK,KAAA,KAAU,MAAA,EAAW,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACrD;AAEA,IAAA,IAAI,IAAA,CAAK,MAAA,KAAW,MAAA,EAAW,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AACrD,IAAA,IAAI,IAAA,CAAK,OAAA,KAAY,MAAA,EAAW,OAAA,CAAQ,UAAU,IAAA,CAAK,OAAA;AAEvD,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAGxC,IAAA,MAAM,mBAA6B,EAAC;AAEpC,IAAA,IAAI,IAAA,CAAK,IAAA,KAASA,oBAAA,CAAW,cAAA,EAAgB;AAC3C,MAAA,MAAM,OAAA,GAAU,UAAA;AAEhB,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,eAAe,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,kBAAkB,OAAA,CAAQ,UAAA;AAClC,QAAA,gBAAA,CAAiB,KAAK,YAAY,CAAA;AAAA,MACpC;AAAA,IACF;AAEA,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAGC,kBAAA,CAAS,UAAA,EAAY,gBAAgB,CAAA;AAAA,MACxC,GAAG,IAAA,CAAK;AAAA,KACV;AAEA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,OAAA,CAAQ,KAAA,GAAQ,OAAA;AAChB,MAAA,OAAA,CAAQ,aAAA,GAAgB,KAAK,SAAA,CAAU,OAAA;AAAA,IACzC;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,MAAM,IAAA,CAAK,OAAO,aAAA,EAAc;AAChC,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AAAA,EACtB;AACF","file":"index.cjs","sourcesContent":["/**\n * Langfuse Exporter for Mastra AI Tracing\n *\n * This exporter sends tracing data to Langfuse for AI observability.\n * Root spans start traces in Langfuse.\n * LLM_GENERATION spans become Langfuse generations, all others become spans.\n */\n\nimport type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, omitKeys } from '@mastra/core/ai-tracing';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { Langfuse } from 'langfuse';\nimport type { LangfuseTraceClient, LangfuseSpanClient, LangfuseGenerationClient, LangfuseEventClient } from 'langfuse';\n\nexport interface LangfuseExporterConfig {\n  /** Langfuse API key */\n  publicKey: string;\n  /** Langfuse secret key */\n  secretKey: string;\n  /** Langfuse host URL */\n  baseUrl: string;\n  /** Enable realtime mode - flushes after each event for immediate visibility */\n  realtime?: boolean;\n  /** Logger level for diagnostic messages (default: 'warn') */\n  logLevel?: 'debug' | 'info' | 'warn' | 'error';\n  /** Additional options to pass to the Langfuse client */\n  options?: any;\n}\n\ntype TraceData = {\n  trace: LangfuseTraceClient; // Langfuse trace object\n  spans: Map<string, LangfuseSpanClient | LangfuseGenerationClient>; // Maps span.id to Langfuse span/generation\n  events: Map<string, LangfuseEventClient>; // Maps span.id to Langfuse event\n};\n\ntype LangfuseParent = LangfuseTraceClient | LangfuseSpanClient | LangfuseGenerationClient | LangfuseEventClient;\n\nexport class LangfuseExporter implements AITracingExporter {\n  name = 'langfuse';\n  private client: Langfuse;\n  private realtime: boolean;\n  private traceMap = new Map<string, TraceData>();\n  private logger: ConsoleLogger;\n\n  constructor(config: LangfuseExporterConfig) {\n    this.realtime = config.realtime ?? false;\n    this.logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n    this.client = new Langfuse({\n      publicKey: config.publicKey,\n      secretKey: config.secretKey,\n      baseUrl: config.baseUrl,\n      ...config.options,\n    });\n  }\n\n  async exportEvent(event: AITracingEvent): Promise<void> {\n    if (event.span.isEvent) {\n      await this.handleEventSpan(event.span);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.span);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.span, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.span, true);\n        break;\n    }\n\n    // Flush immediately in realtime mode for instant visibility\n    if (this.realtime) {\n      await this.client.flushAsync();\n    }\n  }\n\n  private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.initTrace(span);\n    }\n    const method = 'handleSpanStarted';\n\n    const traceData = this.getTraceData({ span, method });\n    if (!traceData) {\n      return;\n    }\n\n    const langfuseParent = this.getLangfuseParent({ traceData, span, method });\n    if (!langfuseParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span, true);\n\n    const langfuseSpan =\n      span.type === AISpanType.LLM_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);\n\n    traceData.spans.set(span.id, langfuseSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyAISpan, isEnd: boolean): Promise<void> {\n    const method = isEnd ? 'handleSpanEnd' : 'handleSpanUpdate';\n\n    const traceData = this.getTraceData({ span, method });\n    if (!traceData) {\n      return;\n    }\n\n    const langfuseSpan = traceData.spans.get(span.id);\n    if (!langfuseSpan) {\n      this.logger.warn('Langfuse exporter: No Langfuse span found for span update/end', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        spanType: span.type,\n        isRootSpan: span.isRootSpan,\n        parentSpanId: span.parent?.id,\n        availableSpanIds: Array.from(traceData.spans.keys()),\n        method,\n      });\n      return;\n    }\n\n    // use update for both update & end, so that we can use the\n    // end time we set when ending the span.\n    langfuseSpan.update(this.buildSpanPayload(span, false));\n\n    if (isEnd && span.isRootSpan) {\n      traceData.trace.update({ output: span.output });\n      this.traceMap.delete(span.traceId);\n    }\n  }\n\n  private async handleEventSpan(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.logger.debug('Langfuse exporter: Creating trace', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        method: 'handleEventSpan',\n      });\n      this.initTrace(span);\n    }\n    const method = 'handleEventSpan';\n\n    const traceData = this.getTraceData({ span, method });\n    if (!traceData) {\n      return;\n    }\n\n    const langfuseParent = this.getLangfuseParent({ traceData, span, method });\n    if (!langfuseParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span, true);\n\n    const langfuseEvent = langfuseParent.event(payload);\n\n    traceData.events.set(span.id, langfuseEvent);\n  }\n\n  private initTrace(span: AnyAISpan): void {\n    const trace = this.client.trace(this.buildTracePayload(span));\n    this.traceMap.set(span.traceId, { trace, spans: new Map(), events: new Map() });\n  }\n\n  private getTraceData(options: { span: AnyAISpan; method: string }): TraceData | undefined {\n    const { span, method } = options;\n    if (this.traceMap.has(span.traceId)) {\n      return this.traceMap.get(span.traceId);\n    }\n    this.logger.warn('Langfuse exporter: No trace data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private getLangfuseParent(options: {\n    traceData: TraceData;\n    span: AnyAISpan;\n    method: string;\n  }): LangfuseParent | undefined {\n    const { traceData, span, method } = options;\n\n    const parentId = span.parent?.id;\n    if (!parentId) {\n      return traceData.trace;\n    }\n    if (traceData.spans.has(parentId)) {\n      return traceData.spans.get(parentId);\n    }\n    if (traceData.events.has(parentId)) {\n      return traceData.events.get(parentId);\n    }\n    this.logger.warn('Langfuse exporter: No parent data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private buildTracePayload(span: AnyAISpan): Record<string, any> {\n    const payload: Record<string, any> = {\n      id: span.traceId,\n      name: span.name,\n    };\n\n    const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};\n\n    if (userId) payload.userId = userId;\n    if (sessionId) payload.sessionId = sessionId;\n    if (span.input) payload.input = span.input;\n\n    payload.metadata = {\n      spanType: span.type,\n      ...span.attributes,\n      ...remainingMetadata,\n    };\n\n    return payload;\n  }\n\n  private buildSpanPayload(span: AnyAISpan, isCreate: boolean): Record<string, any> {\n    const payload: Record<string, any> = {};\n\n    if (isCreate) {\n      payload.id = span.id;\n      payload.name = span.name;\n      payload.startTime = span.startTime;\n      if (span.input !== undefined) payload.input = span.input;\n    }\n\n    if (span.output !== undefined) payload.output = span.output;\n    if (span.endTime !== undefined) payload.endTime = span.endTime;\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    // Strip special fields from metadata if used in top-level keys\n    const attributesToOmit: string[] = [];\n\n    if (span.type === AISpanType.LLM_GENERATION) {\n      const llmAttr = attributes as LLMGenerationAttributes;\n\n      if (llmAttr.model !== undefined) {\n        payload.model = llmAttr.model;\n        attributesToOmit.push('model');\n      }\n\n      if (llmAttr.usage !== undefined) {\n        payload.usage = llmAttr.usage;\n        attributesToOmit.push('usage');\n      }\n\n      if (llmAttr.parameters !== undefined) {\n        payload.modelParameters = llmAttr.parameters;\n        attributesToOmit.push('parameters');\n      }\n    }\n\n    payload.metadata = {\n      spanType: span.type,\n      ...omitKeys(attributes, attributesToOmit),\n      ...span.metadata,\n    };\n\n    if (span.errorInfo) {\n      payload.level = 'ERROR';\n      payload.statusMessage = span.errorInfo.message;\n    }\n\n    return payload;\n  }\n\n  async shutdown(): Promise<void> {\n    await this.client.shutdownAsync();\n    this.traceMap.clear();\n  }\n}\n"]}

package/dist/index.js

@@ -1,4 +1,5 @@
-import { AISpanType, sanitizeMetadata, omitKeys } from '@mastra/core/ai-tracing';
+import { AISpanType, omitKeys } from '@mastra/core/ai-tracing';
+import { ConsoleLogger } from '@mastra/core/logger';
 import { Langfuse } from 'langfuse';
 
 // src/ai-tracing.ts
@@ -7,8 +8,10 @@ var LangfuseExporter = class {
   client;
   realtime;
   traceMap = /* @__PURE__ */ new Map();
+  logger;
   constructor(config) {
     this.realtime = config.realtime ?? false;
+    this.logger = new ConsoleLogger({ level: config.logLevel ?? "warn" });
     this.client = new Langfuse({
       publicKey: config.publicKey,
       secretKey: config.secretKey,
@@ -17,15 +20,19 @@ var LangfuseExporter = class {
     });
   }
   async exportEvent(event) {
+    if (event.span.isEvent) {
+      await this.handleEventSpan(event.span);
+      return;
+    }
     switch (event.type) {
       case "span_started":
         await this.handleSpanStarted(event.span);
         break;
       case "span_updated":
-        await this.handleSpanUpdateOrEnd(event.span, true);
+        await this.handleSpanUpdateOrEnd(event.span, false);
         break;
       case "span_ended":
-        await this.handleSpanUpdateOrEnd(event.span, false);
+        await this.handleSpanUpdateOrEnd(event.span, true);
         break;
     }
     if (this.realtime) {
@@ -34,43 +41,114 @@ var LangfuseExporter = class {
   }
   async handleSpanStarted(span) {
     if (span.isRootSpan) {
-      const trace = this.client.trace(this.buildTracePayload(span));
-      this.traceMap.set(span.trace.id, { trace, spans: /* @__PURE__ */ new Map() });
+      this.initTrace(span);
     }
-    const traceData = this.traceMap.get(span.trace.id);
+    const method = "handleSpanStarted";
+    const traceData = this.getTraceData({ span, method });
     if (!traceData) {
-      console.log("NO TRACE");
       return;
     }
-    const langfuseParent = span.parent && traceData.spans.has(span.parent.id) ? traceData.spans.get(span.parent.id) : traceData.trace;
+    const langfuseParent = this.getLangfuseParent({ traceData, span, method });
+    if (!langfuseParent) {
+      return;
+    }
     const payload = this.buildSpanPayload(span, true);
     const langfuseSpan = span.type === AISpanType.LLM_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);
     traceData.spans.set(span.id, langfuseSpan);
   }
-  async handleSpanUpdateOrEnd(span, isUpdate) {
-    const traceData = this.traceMap.get(span.trace.id);
+  async handleSpanUpdateOrEnd(span, isEnd) {
+    const method = isEnd ? "handleSpanEnd" : "handleSpanUpdate";
+    const traceData = this.getTraceData({ span, method });
     if (!traceData) {
-      console.log("NO TRACE");
       return;
     }
     const langfuseSpan = traceData.spans.get(span.id);
     if (!langfuseSpan) {
-      console.log("NO SPAN");
+      this.logger.warn("Langfuse exporter: No Langfuse span found for span update/end", {
+        traceId: span.traceId,
+        spanId: span.id,
+        spanName: span.name,
+        spanType: span.type,
+        isRootSpan: span.isRootSpan,
+        parentSpanId: span.parent?.id,
+        availableSpanIds: Array.from(traceData.spans.keys()),
+        method
+      });
       return;
     }
-    if (isUpdate) {
-      langfuseSpan.update(this.buildSpanPayload(span, false));
-    } else {
-      langfuseSpan.end(this.buildSpanPayload(span, false));
-      if (span.isRootSpan) {
-        traceData.trace.update({ output: span.output });
-        this.traceMap.delete(span.trace.id);
-      }
+    langfuseSpan.update(this.buildSpanPayload(span, false));
+    if (isEnd && span.isRootSpan) {
+      traceData.trace.update({ output: span.output });
+      this.traceMap.delete(span.traceId);
+    }
+  }
+  async handleEventSpan(span) {
+    if (span.isRootSpan) {
+      this.logger.debug("Langfuse exporter: Creating trace", {
+        traceId: span.traceId,
+        spanId: span.id,
+        spanName: span.name,
+        method: "handleEventSpan"
+      });
+      this.initTrace(span);
+    }
+    const method = "handleEventSpan";
+    const traceData = this.getTraceData({ span, method });
+    if (!traceData) {
+      return;
     }
+    const langfuseParent = this.getLangfuseParent({ traceData, span, method });
+    if (!langfuseParent) {
+      return;
+    }
+    const payload = this.buildSpanPayload(span, true);
+    const langfuseEvent = langfuseParent.event(payload);
+    traceData.events.set(span.id, langfuseEvent);
+  }
+  initTrace(span) {
+    const trace = this.client.trace(this.buildTracePayload(span));
+    this.traceMap.set(span.traceId, { trace, spans: /* @__PURE__ */ new Map(), events: /* @__PURE__ */ new Map() });
+  }
+  getTraceData(options) {
+    const { span, method } = options;
+    if (this.traceMap.has(span.traceId)) {
+      return this.traceMap.get(span.traceId);
+    }
+    this.logger.warn("Langfuse exporter: No trace data found for span", {
+      traceId: span.traceId,
+      spanId: span.id,
+      spanName: span.name,
+      spanType: span.type,
+      isRootSpan: span.isRootSpan,
+      parentSpanId: span.parent?.id,
+      method
+    });
+  }
+  getLangfuseParent(options) {
+    const { traceData, span, method } = options;
+    const parentId = span.parent?.id;
+    if (!parentId) {
+      return traceData.trace;
+    }
+    if (traceData.spans.has(parentId)) {
+      return traceData.spans.get(parentId);
+    }
+    if (traceData.events.has(parentId)) {
+      return traceData.events.get(parentId);
+    }
+    this.logger.warn("Langfuse exporter: No parent data found for span", {
+      traceId: span.traceId,
+      spanId: span.id,
+      spanName: span.name,
+      spanType: span.type,
+      isRootSpan: span.isRootSpan,
+      parentSpanId: span.parent?.id,
+      method
+    });
   }
   buildTracePayload(span) {
     const payload = {
-      id: span.trace.id,
+      id: span.traceId,
       name: span.name
     };
     const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};
@@ -79,8 +157,8 @@ var LangfuseExporter = class {
     if (span.input) payload.input = span.input;
     payload.metadata = {
       spanType: span.type,
-      ...sanitizeMetadata(span.attributes),
-      ...sanitizeMetadata(remainingMetadata)
+      ...span.attributes,
+      ...remainingMetadata
     };
     return payload;
   }
@@ -113,8 +191,8 @@ var LangfuseExporter = class {
     }
     payload.metadata = {
       spanType: span.type,
-      ...sanitizeMetadata(omitKeys(attributes, attributesToOmit)),
-      ...sanitizeMetadata(span.metadata)
+      ...omitKeys(attributes, attributesToOmit),
+      ...span.metadata
     };
     if (span.errorInfo) {
       payload.level = "ERROR";

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/ai-tracing.ts"],"names":[],"mappings":";;;;AAiCO,IAAM,mBAAN,MAAoD;AAAA,EACzD,IAAA,GAAO,UAAA;AAAA,EACC,MAAA;AAAA,EACA,QAAA;AAAA,EACA,QAAA,uBAAe,GAAA,EAAuB;AAAA,EAE9C,YAAY,MAAA,EAAgC;AAC1C,IAAA,IAAA,CAAK,QAAA,GAAW,OAAO,QAAA,IAAY,KAAA;AACnC,IAAA,IAAA,CAAK,MAAA,GAAS,IAAI,QAAA,CAAS;AAAA,MACzB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,SAAS,MAAA,CAAO,OAAA;AAAA,MAChB,GAAG,MAAA,CAAO;AAAA,KACX,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA;AAIJ,IAAA,IAAI,KAAK,QAAA,EAAU;AACjB,MAAA,MAAM,IAAA,CAAK,OAAO,UAAA,EAAW;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAM,QAAQ,IAAA,CAAK,MAAA,CAAO,MAAM,IAAA,CAAK,iBAAA,CAAkB,IAAI,CAAC,CAAA;AAC5D,MAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,KAAA,CAAM,EAAA,EAAI,EAAE,KAAA,EAAO,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,YAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,MAAM,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,OAAA,CAAQ,IAAI,UAAU,CAAA;AAEtB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,iBACJ,IAAA,CAAK,MAAA,IAAU,SAAA,CAAU,KAAA,CAAM,IAAI,IAAA,CAAK,MAAA,CAAO,EAAE,CAAA,GAC5C,UAAU,KAAA,CAAM,GAAA,CAAI,KAAK,MAAA,CAAO,EAAE,IACnC,SAAA,CAAU,KAAA;AAEhB,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,IAAI,CAAA;AAEhD,IAAA,MAAM,YAAA,GACJ,IAAA,CAAK,IAAA,KAAS,UAAA,CAAW,cAAA,GAAiB,cAAA,CAAe,UAAA,CAAW,OAAO,CAAA,GAAI,cAAA,CAAe,IAAA,CAAK,OAAO,CAAA;AAE5G,IAAA,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,YAAY,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,QAAA,EAAkC;AACrF,IAAA,MAAM,YAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,MAAM,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,OAAA,CAAQ,IAAI,UAAU,CAAA;AAEtB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,YAAA,GAAe,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AAChD,IAAA,IAAI,CAAC,YAAA,EAAc;AACjB,MAAA,OAAA,CAAQ,IAAI,SAAS,CAAA;AAErB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,YAAA,CAAa,MAAA,CAAO,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAK,CAAC,CAAA;AAAA,IACxD,CAAA,MAAO;AACL,MAAA,YAAA,CAAa,GAAA,CAAI,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAK,CAAC,CAAA;AAEnD,MAAA,IAAI,KAAK,UAAA,EAAY;AACnB,QAAA,SAAA,CAAU,MAAM,MAAA,CAAO,EAAE,MAAA,EAAQ,IAAA,CAAK,QAAQ,CAAA;AAC9C,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,KAAA,CAAM,EAAE,CAAA;AAAA,MACpC;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,kBAAkB,IAAA,EAAsC;AAC9D,IAAA,MAAM,OAAA,GAA+B;AAAA,MACnC,EAAA,EAAI,KAAK,KAAA,CAAM,EAAA;AAAA,MACf,MAAM,IAAA,CAAK;AAAA,KACb;AAEA,IAAA,MAAM,EAAE,QAAQ,SAAA,EAAW,GAAG,mBAAkB,GAAI,IAAA,CAAK,YAAY,EAAC;AAEtE,IAAA,IAAI,MAAA,UAAgB,MAAA,GAAS,MAAA;AAC7B,IAAA,IAAI,SAAA,UAAmB,SAAA,GAAY,SAAA;AACnC,IAAA,IAAI,IAAA,CAAK,KAAA,EAAO,OAAA,CAAQ,KAAA,GAAQ,IAAA,CAAK,KAAA;AAErC,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,gBAAA,CAAiB,IAAA,CAAK,UAAU,CAAA;AAAA,MACnC,GAAG,iBAAiB,iBAAiB;AAAA,KACvC;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEQ,gBAAA,CAAiB,MAAiB,QAAA,EAAwC;AAChF,IAAA,MAAM,UAA+B,EAAC;AAEtC,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,OAAA,CAAQ,KAAK,IAAA,CAAK,EAAA;AAClB,MAAA,OAAA,CAAQ,OAAO,IAAA,CAAK,IAAA;AACpB,MAAA,OAAA,CAAQ,YAAY,IAAA,CAAK,SAAA;AACzB,MAAA,IAAI,IAAA,CAAK,KAAA,KAAU,MAAA,EAAW,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACrD;AAEA,IAAA,IAAI,IAAA,CAAK,MAAA,KAAW,MAAA,EAAW,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AACrD,IAAA,IAAI,IAAA,CAAK,OAAA,KAAY,MAAA,EAAW,OAAA,CAAQ,UAAU,IAAA,CAAK,OAAA;AAEvD,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAGxC,IAAA,MAAM,mBAA6B,EAAC;AAEpC,IAAA,IAAI,IAAA,CAAK,IAAA,KAAS,UAAA,CAAW,cAAA,EAAgB;AAC3C,MAAA,MAAM,OAAA,GAAU,UAAA;AAEhB,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,eAAe,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,kBAAkB,OAAA,CAAQ,UAAA;AAClC,QAAA,gBAAA,CAAiB,KAAK,YAAY,CAAA;AAAA,MACpC;AAAA,IACF;AAEA,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,gBAAA,CAAiB,QAAA,CAAS,UAAA,EAAY,gBAAgB,CAAC,CAAA;AAAA,MAC1D,GAAG,gBAAA,CAAiB,IAAA,CAAK,QAAQ;AAAA,KACnC;AAEA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,OAAA,CAAQ,KAAA,GAAQ,OAAA;AAChB,MAAA,OAAA,CAAQ,aAAA,GAAgB,KAAK,SAAA,CAAU,OAAA;AAAA,IACzC;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,MAAM,IAAA,CAAK,OAAO,aAAA,EAAc;AAChC,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AAAA,EACtB;AACF","file":"index.js","sourcesContent":["/**\n * Langfuse Exporter for Mastra AI Tracing\n *\n * This exporter sends tracing data to Langfuse for AI observability.\n * Root spans start traces in Langfuse.\n * LLM_GENERATION spans become Langfuse generations, all others become spans.\n */\n\nimport type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, sanitizeMetadata, omitKeys } from '@mastra/core/ai-tracing';\nimport { Langfuse } from 'langfuse';\nimport type { LangfuseTraceClient, LangfuseSpanClient, LangfuseGenerationClient } from 'langfuse';\n\nexport interface LangfuseExporterConfig {\n  /** Langfuse API key */\n  publicKey: string;\n  /** Langfuse secret key */\n  secretKey: string;\n  /** Langfuse host URL */\n  baseUrl: string;\n  /** Enable realtime mode - flushes after each event for immediate visibility */\n  realtime?: boolean;\n  /** Additional options to pass to the Langfuse client */\n  options?: any;\n}\n\ntype TraceData = {\n  trace: LangfuseTraceClient; // Langfuse trace object\n  spans: Map<string, LangfuseSpanClient | LangfuseGenerationClient>; // Maps span.id to Langfuse span/generation\n};\n\ntype LangfuseSpan = LangfuseTraceClient | LangfuseSpanClient | LangfuseGenerationClient;\n\nexport class LangfuseExporter implements AITracingExporter {\n  name = 'langfuse';\n  private client: Langfuse;\n  private realtime: boolean;\n  private traceMap = new Map<string, TraceData>();\n\n  constructor(config: LangfuseExporterConfig) {\n    this.realtime = config.realtime ?? false;\n    this.client = new Langfuse({\n      publicKey: config.publicKey,\n      secretKey: config.secretKey,\n      baseUrl: config.baseUrl,\n      ...config.options,\n    });\n  }\n\n  async exportEvent(event: AITracingEvent): Promise<void> {\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.span);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.span, true);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.span, false);\n        break;\n    }\n\n    // Flush immediately in realtime mode for instant visibility\n    if (this.realtime) {\n      await this.client.flushAsync();\n    }\n  }\n\n  private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      const trace = this.client.trace(this.buildTracePayload(span));\n      this.traceMap.set(span.trace.id, { trace, spans: new Map() });\n    }\n\n    const traceData = this.traceMap.get(span.trace.id);\n    if (!traceData) {\n      console.log('NO TRACE');\n      // TODO: log warning\n      return;\n    }\n\n    const langfuseParent =\n      span.parent && traceData.spans.has(span.parent.id)\n        ? (traceData.spans.get(span.parent.id) as LangfuseSpan)\n        : traceData.trace;\n\n    const payload = this.buildSpanPayload(span, true);\n\n    const langfuseSpan =\n      span.type === AISpanType.LLM_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);\n\n    traceData.spans.set(span.id, langfuseSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyAISpan, isUpdate: boolean): Promise<void> {\n    const traceData = this.traceMap.get(span.trace.id);\n    if (!traceData) {\n      console.log('NO TRACE');\n      // TODO: log warning\n      return;\n    }\n\n    const langfuseSpan = traceData.spans.get(span.id);\n    if (!langfuseSpan) {\n      console.log('NO SPAN');\n      // TODO: log warning\n      return;\n    }\n\n    if (isUpdate) {\n      langfuseSpan.update(this.buildSpanPayload(span, false));\n    } else {\n      langfuseSpan.end(this.buildSpanPayload(span, false));\n\n      if (span.isRootSpan) {\n        traceData.trace.update({ output: span.output });\n        this.traceMap.delete(span.trace.id);\n      }\n    }\n  }\n\n  private buildTracePayload(span: AnyAISpan): Record<string, any> {\n    const payload: Record<string, any> = {\n      id: span.trace.id,\n      name: span.name,\n    };\n\n    const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};\n\n    if (userId) payload.userId = userId;\n    if (sessionId) payload.sessionId = sessionId;\n    if (span.input) payload.input = span.input;\n\n    payload.metadata = {\n      spanType: span.type,\n      ...sanitizeMetadata(span.attributes),\n      ...sanitizeMetadata(remainingMetadata),\n    };\n\n    return payload;\n  }\n\n  private buildSpanPayload(span: AnyAISpan, isCreate: boolean): Record<string, any> {\n    const payload: Record<string, any> = {};\n\n    if (isCreate) {\n      payload.id = span.id;\n      payload.name = span.name;\n      payload.startTime = span.startTime;\n      if (span.input !== undefined) payload.input = span.input;\n    }\n\n    if (span.output !== undefined) payload.output = span.output;\n    if (span.endTime !== undefined) payload.endTime = span.endTime;\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    // Strip special fields from metadata if used in top-level keys\n    const attributesToOmit: string[] = [];\n\n    if (span.type === AISpanType.LLM_GENERATION) {\n      const llmAttr = attributes as LLMGenerationAttributes;\n\n      if (llmAttr.model !== undefined) {\n        payload.model = llmAttr.model;\n        attributesToOmit.push('model');\n      }\n\n      if (llmAttr.usage !== undefined) {\n        payload.usage = llmAttr.usage;\n        attributesToOmit.push('usage');\n      }\n\n      if (llmAttr.parameters !== undefined) {\n        payload.modelParameters = llmAttr.parameters;\n        attributesToOmit.push('parameters');\n      }\n    }\n\n    payload.metadata = {\n      spanType: span.type,\n      ...sanitizeMetadata(omitKeys(attributes, attributesToOmit)),\n      ...sanitizeMetadata(span.metadata),\n    };\n\n    if (span.errorInfo) {\n      payload.level = 'ERROR';\n      payload.statusMessage = span.errorInfo.message;\n    }\n\n    return payload;\n  }\n\n  async shutdown(): Promise<void> {\n    await this.client.shutdownAsync();\n    this.traceMap.clear();\n  }\n}\n"]}
+{"version":3,"sources":["../src/ai-tracing.ts"],"names":[],"mappings":";;;;;AAqCO,IAAM,mBAAN,MAAoD;AAAA,EACzD,IAAA,GAAO,UAAA;AAAA,EACC,MAAA;AAAA,EACA,QAAA;AAAA,EACA,QAAA,uBAAe,GAAA,EAAuB;AAAA,EACtC,MAAA;AAAA,EAER,YAAY,MAAA,EAAgC;AAC1C,IAAA,IAAA,CAAK,QAAA,GAAW,OAAO,QAAA,IAAY,KAAA;AACnC,IAAA,IAAA,CAAK,MAAA,GAAS,IAAI,aAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AACpE,IAAA,IAAA,CAAK,MAAA,GAAS,IAAI,QAAA,CAAS;AAAA,MACzB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,SAAS,MAAA,CAAO,OAAA;AAAA,MAChB,GAAG,MAAA,CAAO;AAAA,KACX,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,IAAI,KAAA,CAAM,KAAK,OAAA,EAAS;AACtB,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,IAAI,CAAA;AACrC,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA;AAIJ,IAAA,IAAI,KAAK,QAAA,EAAU;AACjB,MAAA,MAAM,IAAA,CAAK,OAAO,UAAA,EAAW;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,UAAU,IAAI,CAAA;AAAA,IACrB;AACA,IAAA,MAAM,MAAA,GAAS,mBAAA;AAEf,IAAA,MAAM,YAAY,IAAA,CAAK,YAAA,CAAa,EAAE,IAAA,EAAM,QAAQ,CAAA;AACpD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,iBAAiB,IAAA,CAAK,iBAAA,CAAkB,EAAE,SAAA,EAAW,IAAA,EAAM,QAAQ,CAAA;AACzE,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,IAAI,CAAA;AAEhD,IAAA,MAAM,YAAA,GACJ,IAAA,CAAK,IAAA,KAAS,UAAA,CAAW,cAAA,GAAiB,cAAA,CAAe,UAAA,CAAW,OAAO,CAAA,GAAI,cAAA,CAAe,IAAA,CAAK,OAAO,CAAA;AAE5G,IAAA,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,YAAY,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,KAAA,EAA+B;AAClF,IAAA,MAAM,MAAA,GAAS,QAAQ,eAAA,GAAkB,kBAAA;AAEzC,IAAA,MAAM,YAAY,IAAA,CAAK,YAAA,CAAa,EAAE,IAAA,EAAM,QAAQ,CAAA;AACpD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,YAAA,GAAe,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AAChD,IAAA,IAAI,CAAC,YAAA,EAAc;AACjB,MAAA,IAAA,CAAK,MAAA,CAAO,KAAK,+DAAA,EAAiE;AAAA,QAChF,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,YAAY,IAAA,CAAK,UAAA;AAAA,QACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,QAC3B,kBAAkB,KAAA,CAAM,IAAA,CAAK,SAAA,CAAU,KAAA,CAAM,MAAM,CAAA;AAAA,QACnD;AAAA,OACD,CAAA;AACD,MAAA;AAAA,IACF;AAIA,IAAA,YAAA,CAAa,MAAA,CAAO,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAK,CAAC,CAAA;AAEtD,IAAA,IAAI,KAAA,IAAS,KAAK,UAAA,EAAY;AAC5B,MAAA,SAAA,CAAU,MAAM,MAAA,CAAO,EAAE,MAAA,EAAQ,IAAA,CAAK,QAAQ,CAAA;AAC9C,MAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,IACnC;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAgC;AAC5D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,mCAAA,EAAqC;AAAA,QACrD,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,MAAA,EAAQ;AAAA,OACT,CAAA;AACD,MAAA,IAAA,CAAK,UAAU,IAAI,CAAA;AAAA,IACrB;AACA,IAAA,MAAM,MAAA,GAAS,iBAAA;AAEf,IAAA,MAAM,YAAY,IAAA,CAAK,YAAA,CAAa,EAAE,IAAA,EAAM,QAAQ,CAAA;AACpD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,iBAAiB,IAAA,CAAK,iBAAA,CAAkB,EAAE,SAAA,EAAW,IAAA,EAAM,QAAQ,CAAA;AACzE,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,IAAI,CAAA;AAEhD,IAAA,MAAM,aAAA,GAAgB,cAAA,CAAe,KAAA,CAAM,OAAO,CAAA;AAElD,IAAA,SAAA,CAAU,MAAA,CAAO,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,aAAa,CAAA;AAAA,EAC7C;AAAA,EAEQ,UAAU,IAAA,EAAuB;AACvC,IAAA,MAAM,QAAQ,IAAA,CAAK,MAAA,CAAO,MAAM,IAAA,CAAK,iBAAA,CAAkB,IAAI,CAAC,CAAA;AAC5D,IAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS,EAAE,KAAA,EAAO,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,MAAA,kBAAQ,IAAI,GAAA,IAAO,CAAA;AAAA,EAChF;AAAA,EAEQ,aAAa,OAAA,EAAqE;AACxF,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AACzB,IAAA,IAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA,EAAG;AACnC,MAAA,OAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACvC;AACA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,iDAAA,EAAmD;AAAA,MAClE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,kBAAkB,OAAA,EAIK;AAC7B,IAAA,MAAM,EAAE,SAAA,EAAW,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AAEpC,IAAA,MAAM,QAAA,GAAW,KAAK,MAAA,EAAQ,EAAA;AAC9B,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,SAAA,CAAU,KAAA;AAAA,IACnB;AACA,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA,EAAG;AACjC,MAAA,OAAO,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA;AAAA,IACrC;AACA,IAAA,IAAI,SAAA,CAAU,MAAA,CAAO,GAAA,CAAI,QAAQ,CAAA,EAAG;AAClC,MAAA,OAAO,SAAA,CAAU,MAAA,CAAO,GAAA,CAAI,QAAQ,CAAA;AAAA,IACtC;AACA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,kDAAA,EAAoD;AAAA,MACnE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,kBAAkB,IAAA,EAAsC;AAC9D,IAAA,MAAM,OAAA,GAA+B;AAAA,MACnC,IAAI,IAAA,CAAK,OAAA;AAAA,MACT,MAAM,IAAA,CAAK;AAAA,KACb;AAEA,IAAA,MAAM,EAAE,QAAQ,SAAA,EAAW,GAAG,mBAAkB,GAAI,IAAA,CAAK,YAAY,EAAC;AAEtE,IAAA,IAAI,MAAA,UAAgB,MAAA,GAAS,MAAA;AAC7B,IAAA,IAAI,SAAA,UAAmB,SAAA,GAAY,SAAA;AACnC,IAAA,IAAI,IAAA,CAAK,KAAA,EAAO,OAAA,CAAQ,KAAA,GAAQ,IAAA,CAAK,KAAA;AAErC,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,IAAA,CAAK,UAAA;AAAA,MACR,GAAG;AAAA,KACL;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEQ,gBAAA,CAAiB,MAAiB,QAAA,EAAwC;AAChF,IAAA,MAAM,UAA+B,EAAC;AAEtC,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,OAAA,CAAQ,KAAK,IAAA,CAAK,EAAA;AAClB,MAAA,OAAA,CAAQ,OAAO,IAAA,CAAK,IAAA;AACpB,MAAA,OAAA,CAAQ,YAAY,IAAA,CAAK,SAAA;AACzB,MAAA,IAAI,IAAA,CAAK,KAAA,KAAU,MAAA,EAAW,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACrD;AAEA,IAAA,IAAI,IAAA,CAAK,MAAA,KAAW,MAAA,EAAW,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AACrD,IAAA,IAAI,IAAA,CAAK,OAAA,KAAY,MAAA,EAAW,OAAA,CAAQ,UAAU,IAAA,CAAK,OAAA;AAEvD,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAGxC,IAAA,MAAM,mBAA6B,EAAC;AAEpC,IAAA,IAAI,IAAA,CAAK,IAAA,KAAS,UAAA,CAAW,cAAA,EAAgB;AAC3C,MAAA,MAAM,OAAA,GAAU,UAAA;AAEhB,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,eAAe,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,kBAAkB,OAAA,CAAQ,UAAA;AAClC,QAAA,gBAAA,CAAiB,KAAK,YAAY,CAAA;AAAA,MACpC;AAAA,IACF;AAEA,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,QAAA,CAAS,UAAA,EAAY,gBAAgB,CAAA;AAAA,MACxC,GAAG,IAAA,CAAK;AAAA,KACV;AAEA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,OAAA,CAAQ,KAAA,GAAQ,OAAA;AAChB,MAAA,OAAA,CAAQ,aAAA,GAAgB,KAAK,SAAA,CAAU,OAAA;AAAA,IACzC;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,MAAM,IAAA,CAAK,OAAO,aAAA,EAAc;AAChC,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AAAA,EACtB;AACF","file":"index.js","sourcesContent":["/**\n * Langfuse Exporter for Mastra AI Tracing\n *\n * This exporter sends tracing data to Langfuse for AI observability.\n * Root spans start traces in Langfuse.\n * LLM_GENERATION spans become Langfuse generations, all others become spans.\n */\n\nimport type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, omitKeys } from '@mastra/core/ai-tracing';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { Langfuse } from 'langfuse';\nimport type { LangfuseTraceClient, LangfuseSpanClient, LangfuseGenerationClient, LangfuseEventClient } from 'langfuse';\n\nexport interface LangfuseExporterConfig {\n  /** Langfuse API key */\n  publicKey: string;\n  /** Langfuse secret key */\n  secretKey: string;\n  /** Langfuse host URL */\n  baseUrl: string;\n  /** Enable realtime mode - flushes after each event for immediate visibility */\n  realtime?: boolean;\n  /** Logger level for diagnostic messages (default: 'warn') */\n  logLevel?: 'debug' | 'info' | 'warn' | 'error';\n  /** Additional options to pass to the Langfuse client */\n  options?: any;\n}\n\ntype TraceData = {\n  trace: LangfuseTraceClient; // Langfuse trace object\n  spans: Map<string, LangfuseSpanClient | LangfuseGenerationClient>; // Maps span.id to Langfuse span/generation\n  events: Map<string, LangfuseEventClient>; // Maps span.id to Langfuse event\n};\n\ntype LangfuseParent = LangfuseTraceClient | LangfuseSpanClient | LangfuseGenerationClient | LangfuseEventClient;\n\nexport class LangfuseExporter implements AITracingExporter {\n  name = 'langfuse';\n  private client: Langfuse;\n  private realtime: boolean;\n  private traceMap = new Map<string, TraceData>();\n  private logger: ConsoleLogger;\n\n  constructor(config: LangfuseExporterConfig) {\n    this.realtime = config.realtime ?? false;\n    this.logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n    this.client = new Langfuse({\n      publicKey: config.publicKey,\n      secretKey: config.secretKey,\n      baseUrl: config.baseUrl,\n      ...config.options,\n    });\n  }\n\n  async exportEvent(event: AITracingEvent): Promise<void> {\n    if (event.span.isEvent) {\n      await this.handleEventSpan(event.span);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.span);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.span, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.span, true);\n        break;\n    }\n\n    // Flush immediately in realtime mode for instant visibility\n    if (this.realtime) {\n      await this.client.flushAsync();\n    }\n  }\n\n  private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.initTrace(span);\n    }\n    const method = 'handleSpanStarted';\n\n    const traceData = this.getTraceData({ span, method });\n    if (!traceData) {\n      return;\n    }\n\n    const langfuseParent = this.getLangfuseParent({ traceData, span, method });\n    if (!langfuseParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span, true);\n\n    const langfuseSpan =\n      span.type === AISpanType.LLM_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);\n\n    traceData.spans.set(span.id, langfuseSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyAISpan, isEnd: boolean): Promise<void> {\n    const method = isEnd ? 'handleSpanEnd' : 'handleSpanUpdate';\n\n    const traceData = this.getTraceData({ span, method });\n    if (!traceData) {\n      return;\n    }\n\n    const langfuseSpan = traceData.spans.get(span.id);\n    if (!langfuseSpan) {\n      this.logger.warn('Langfuse exporter: No Langfuse span found for span update/end', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        spanType: span.type,\n        isRootSpan: span.isRootSpan,\n        parentSpanId: span.parent?.id,\n        availableSpanIds: Array.from(traceData.spans.keys()),\n        method,\n      });\n      return;\n    }\n\n    // use update for both update & end, so that we can use the\n    // end time we set when ending the span.\n    langfuseSpan.update(this.buildSpanPayload(span, false));\n\n    if (isEnd && span.isRootSpan) {\n      traceData.trace.update({ output: span.output });\n      this.traceMap.delete(span.traceId);\n    }\n  }\n\n  private async handleEventSpan(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.logger.debug('Langfuse exporter: Creating trace', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        method: 'handleEventSpan',\n      });\n      this.initTrace(span);\n    }\n    const method = 'handleEventSpan';\n\n    const traceData = this.getTraceData({ span, method });\n    if (!traceData) {\n      return;\n    }\n\n    const langfuseParent = this.getLangfuseParent({ traceData, span, method });\n    if (!langfuseParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span, true);\n\n    const langfuseEvent = langfuseParent.event(payload);\n\n    traceData.events.set(span.id, langfuseEvent);\n  }\n\n  private initTrace(span: AnyAISpan): void {\n    const trace = this.client.trace(this.buildTracePayload(span));\n    this.traceMap.set(span.traceId, { trace, spans: new Map(), events: new Map() });\n  }\n\n  private getTraceData(options: { span: AnyAISpan; method: string }): TraceData | undefined {\n    const { span, method } = options;\n    if (this.traceMap.has(span.traceId)) {\n      return this.traceMap.get(span.traceId);\n    }\n    this.logger.warn('Langfuse exporter: No trace data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private getLangfuseParent(options: {\n    traceData: TraceData;\n    span: AnyAISpan;\n    method: string;\n  }): LangfuseParent | undefined {\n    const { traceData, span, method } = options;\n\n    const parentId = span.parent?.id;\n    if (!parentId) {\n      return traceData.trace;\n    }\n    if (traceData.spans.has(parentId)) {\n      return traceData.spans.get(parentId);\n    }\n    if (traceData.events.has(parentId)) {\n      return traceData.events.get(parentId);\n    }\n    this.logger.warn('Langfuse exporter: No parent data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private buildTracePayload(span: AnyAISpan): Record<string, any> {\n    const payload: Record<string, any> = {\n      id: span.traceId,\n      name: span.name,\n    };\n\n    const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};\n\n    if (userId) payload.userId = userId;\n    if (sessionId) payload.sessionId = sessionId;\n    if (span.input) payload.input = span.input;\n\n    payload.metadata = {\n      spanType: span.type,\n      ...span.attributes,\n      ...remainingMetadata,\n    };\n\n    return payload;\n  }\n\n  private buildSpanPayload(span: AnyAISpan, isCreate: boolean): Record<string, any> {\n    const payload: Record<string, any> = {};\n\n    if (isCreate) {\n      payload.id = span.id;\n      payload.name = span.name;\n      payload.startTime = span.startTime;\n      if (span.input !== undefined) payload.input = span.input;\n    }\n\n    if (span.output !== undefined) payload.output = span.output;\n    if (span.endTime !== undefined) payload.endTime = span.endTime;\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    // Strip special fields from metadata if used in top-level keys\n    const attributesToOmit: string[] = [];\n\n    if (span.type === AISpanType.LLM_GENERATION) {\n      const llmAttr = attributes as LLMGenerationAttributes;\n\n      if (llmAttr.model !== undefined) {\n        payload.model = llmAttr.model;\n        attributesToOmit.push('model');\n      }\n\n      if (llmAttr.usage !== undefined) {\n        payload.usage = llmAttr.usage;\n        attributesToOmit.push('usage');\n      }\n\n      if (llmAttr.parameters !== undefined) {\n        payload.modelParameters = llmAttr.parameters;\n        attributesToOmit.push('parameters');\n      }\n    }\n\n    payload.metadata = {\n      spanType: span.type,\n      ...omitKeys(attributes, attributesToOmit),\n      ...span.metadata,\n    };\n\n    if (span.errorInfo) {\n      payload.level = 'ERROR';\n      payload.statusMessage = span.errorInfo.message;\n    }\n\n    return payload;\n  }\n\n  async shutdown(): Promise<void> {\n    await this.client.shutdownAsync();\n    this.traceMap.clear();\n  }\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/langfuse",
-  "version": "0.0.4",
+  "version": "0.0.5-alpha.0",
   "description": "Langfuse observability provider for Mastra - includes AI tracing and future observability features",
   "type": "module",
   "main": "dist/index.js",
@@ -29,12 +29,12 @@
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
     "vitest": "^3.2.4",
-    "@mastra/core": "0.15.2",
+    "@internal/lint": "0.0.34",
     "@internal/types-builder": "0.0.9",
-    "@internal/lint": "0.0.34"
+    "@mastra/core": "0.15.3-alpha.4"
   },
   "peerDependencies": {
-    "@mastra/core": ">=0.13.0-0 <0.16.0-0"
+    "@mastra/core": ">=0.15.3-0 <0.16.0-0"
   },
   "scripts": {
     "build": "tsup --silent --config tsup.config.ts",

package/src/ai-tracing.test.ts

@@ -35,20 +35,21 @@ describe('LangfuseExporter', () => {
     // Set up mocks
     mockGeneration = {
       update: vi.fn(),
-      end: vi.fn(),
+      event: vi.fn(),
     };
 
     mockSpan = {
       update: vi.fn(),
-      end: vi.fn(),
       generation: vi.fn().mockReturnValue(mockGeneration),
       span: vi.fn(),
+      event: vi.fn(),
     };
 
     mockTrace = {
       generation: vi.fn().mockReturnValue(mockGeneration),
       span: vi.fn().mockReturnValue(mockSpan),
       update: vi.fn(),
+      event: vi.fn(),
     };
 
     // Set up circular reference
@@ -456,7 +457,7 @@ describe('LangfuseExporter', () => {
   });
 
   describe('Span Ending', () => {
-    it('should end span with success status', async () => {
+    it('should update span with endTime on span end', async () => {
       const span = createMockSpan({
         id: 'test-span',
         name: 'test',
@@ -477,10 +478,7 @@ describe('LangfuseExporter', () => {
         span,
       });
 
-      console.log('BOOP');
-      console.log(span.metadata);
-
-      expect(mockSpan.end).toHaveBeenCalledWith({
+      expect(mockSpan.update).toHaveBeenCalledWith({
         endTime: span.endTime,
         metadata: expect.objectContaining({
           spanType: 'generic',
@@ -488,7 +486,7 @@ describe('LangfuseExporter', () => {
       });
     });
 
-    it('should end span with error status', async () => {
+    it('should update span with error information on span end', async () => {
       const errorSpan = createMockSpan({
         id: 'error-span',
         name: 'failing-operation',
@@ -516,7 +514,7 @@ describe('LangfuseExporter', () => {
         span: errorSpan,
       });
 
-      expect(mockSpan.end).toHaveBeenCalledWith({
+      expect(mockSpan.update).toHaveBeenCalledWith({
         endTime: errorSpan.endTime,
         metadata: expect.objectContaining({
           spanType: 'tool_call',
@@ -526,6 +524,40 @@ describe('LangfuseExporter', () => {
         statusMessage: 'Tool execution failed',
       });
     });
+
+    it('should update root trace and delete from traceMap when root span ends', async () => {
+      const rootSpan = createMockSpan({
+        id: 'root-span-id',
+        name: 'root-span',
+        type: AISpanType.AGENT_RUN,
+        isRoot: true,
+        attributes: {},
+      });
+
+      rootSpan.output = { result: 'success' };
+      rootSpan.endTime = new Date();
+
+      await exporter.exportEvent({
+        type: AITracingEventType.SPAN_STARTED,
+        span: rootSpan,
+      });
+
+      // Verify trace was created
+      expect((exporter as any).traceMap.has('root-span-id')).toBe(true);
+
+      await exporter.exportEvent({
+        type: AITracingEventType.SPAN_ENDED,
+        span: rootSpan,
+      });
+
+      // Should update trace with output
+      expect(mockTrace.update).toHaveBeenCalledWith({
+        output: { result: 'success' },
+      });
+
+      // Should remove trace from traceMap
+      expect((exporter as any).traceMap.has('root-span-id')).toBe(false);
+    });
   });
 
   describe('Error Handling', () => {
@@ -578,6 +610,138 @@ describe('LangfuseExporter', () => {
     });
   });
 
+  describe('Event Span Handling', () => {
+    let mockEvent: any;
+
+    beforeEach(() => {
+      mockEvent = {
+        update: vi.fn(),
+      };
+      mockTrace.event.mockReturnValue(mockEvent);
+      mockSpan.event.mockReturnValue(mockEvent);
+      mockGeneration.event.mockReturnValue(mockEvent);
+    });
+
+    it('should create Langfuse event for root event spans', async () => {
+      const eventSpan = createMockSpan({
+        id: 'event-span-id',
+        name: 'user-feedback',
+        type: AISpanType.GENERIC,
+        isRoot: true,
+        attributes: {
+          eventType: 'user_feedback',
+          rating: 5,
+        },
+        input: { message: 'Great response!' },
+      });
+      eventSpan.isEvent = true;
+
+      await exporter.exportEvent({
+        type: AITracingEventType.SPAN_STARTED,
+        span: eventSpan,
+      });
+
+      // Should create trace for root event span
+      expect(mockLangfuseClient.trace).toHaveBeenCalledWith({
+        id: 'event-span-id',
+        name: 'user-feedback',
+        input: { message: 'Great response!' },
+        metadata: {
+          spanType: 'generic',
+          eventType: 'user_feedback',
+          rating: 5,
+        },
+      });
+
+      // Should create Langfuse event
+      expect(mockTrace.event).toHaveBeenCalledWith({
+        id: 'event-span-id',
+        name: 'user-feedback',
+        startTime: eventSpan.startTime,
+        input: { message: 'Great response!' },
+        metadata: {
+          spanType: 'generic',
+          eventType: 'user_feedback',
+          rating: 5,
+        },
+      });
+    });
+
+    it('should create Langfuse event for child event spans', async () => {
+      // First create a root span
+      const rootSpan = createMockSpan({
+        id: 'root-span-id',
+        name: 'root-agent',
+        type: AISpanType.AGENT_RUN,
+        isRoot: true,
+        attributes: {},
+      });
+
+      await exporter.exportEvent({
+        type: AITracingEventType.SPAN_STARTED,
+        span: rootSpan,
+      });
+
+      // Then create a child event span
+      const childEventSpan = createMockSpan({
+        id: 'child-event-id',
+        name: 'tool-result',
+        type: AISpanType.GENERIC,
+        isRoot: false,
+        attributes: {
+          toolName: 'calculator',
+          success: true,
+        },
+        output: { result: 42 },
+      });
+      childEventSpan.isEvent = true;
+      childEventSpan.traceId = 'root-span-id';
+      childEventSpan.parent = { id: 'root-span-id' };
+
+      await exporter.exportEvent({
+        type: AITracingEventType.SPAN_STARTED,
+        span: childEventSpan,
+      });
+
+      // Should create event under the parent span
+      expect(mockSpan.event).toHaveBeenCalledWith({
+        id: 'child-event-id',
+        name: 'tool-result',
+        startTime: childEventSpan.startTime,
+        output: { result: 42 },
+        metadata: {
+          spanType: 'generic',
+          toolName: 'calculator',
+          success: true,
+        },
+      });
+    });
+
+    it('should handle event spans with missing parent gracefully', async () => {
+      const orphanEventSpan = createMockSpan({
+        id: 'orphan-event-id',
+        name: 'orphan-event',
+        type: AISpanType.GENERIC,
+        isRoot: false,
+        attributes: {},
+      });
+      orphanEventSpan.isEvent = true;
+      orphanEventSpan.traceId = 'missing-trace-id';
+
+      // Should not throw
+      await expect(
+        exporter.exportEvent({
+          type: AITracingEventType.SPAN_STARTED,
+          span: orphanEventSpan,
+        }),
+      ).resolves.not.toThrow();
+
+      // Should not create any Langfuse objects
+      expect(mockTrace.event).not.toHaveBeenCalled();
+      expect(mockSpan.event).not.toHaveBeenCalled();
+    });
+  });
+
   describe('Shutdown', () => {
     it('should shutdown Langfuse client and clear maps', async () => {
       // Add some data to internal maps
@@ -657,6 +821,8 @@ function createMockSpan({
     error: vi.fn(),
     update: vi.fn(),
     createChildSpan: vi.fn(),
+    createEventSpan: vi.fn(),
+    isEvent: false,
   } as AnyAISpan;
 
   return mockSpan;

package/src/ai-tracing.ts

@@ -7,9 +7,10 @@
  */
 
 import type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';
-import { AISpanType, sanitizeMetadata, omitKeys } from '@mastra/core/ai-tracing';
+import { AISpanType, omitKeys } from '@mastra/core/ai-tracing';
+import { ConsoleLogger } from '@mastra/core/logger';
 import { Langfuse } from 'langfuse';
-import type { LangfuseTraceClient, LangfuseSpanClient, LangfuseGenerationClient } from 'langfuse';
+import type { LangfuseTraceClient, LangfuseSpanClient, LangfuseGenerationClient, LangfuseEventClient } from 'langfuse';
 
 export interface LangfuseExporterConfig {
   /** Langfuse API key */
@@ -20,6 +21,8 @@ export interface LangfuseExporterConfig {
   baseUrl: string;
   /** Enable realtime mode - flushes after each event for immediate visibility */
   realtime?: boolean;
+  /** Logger level for diagnostic messages (default: 'warn') */
+  logLevel?: 'debug' | 'info' | 'warn' | 'error';
   /** Additional options to pass to the Langfuse client */
   options?: any;
 }
@@ -27,18 +30,21 @@ export interface LangfuseExporterConfig {
 type TraceData = {
   trace: LangfuseTraceClient; // Langfuse trace object
   spans: Map<string, LangfuseSpanClient | LangfuseGenerationClient>; // Maps span.id to Langfuse span/generation
+  events: Map<string, LangfuseEventClient>; // Maps span.id to Langfuse event
 };
 
-type LangfuseSpan = LangfuseTraceClient | LangfuseSpanClient | LangfuseGenerationClient;
+type LangfuseParent = LangfuseTraceClient | LangfuseSpanClient | LangfuseGenerationClient | LangfuseEventClient;
 
 export class LangfuseExporter implements AITracingExporter {
   name = 'langfuse';
   private client: Langfuse;
   private realtime: boolean;
   private traceMap = new Map<string, TraceData>();
+  private logger: ConsoleLogger;
 
   constructor(config: LangfuseExporterConfig) {
     this.realtime = config.realtime ?? false;
+    this.logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });
     this.client = new Langfuse({
       publicKey: config.publicKey,
       secretKey: config.secretKey,
@@ -48,15 +54,20 @@ export class LangfuseExporter implements AITracingExporter {
   }
 
   async exportEvent(event: AITracingEvent): Promise<void> {
+    if (event.span.isEvent) {
+      await this.handleEventSpan(event.span);
+      return;
+    }
+
     switch (event.type) {
       case 'span_started':
         await this.handleSpanStarted(event.span);
         break;
       case 'span_updated':
-        await this.handleSpanUpdateOrEnd(event.span, true);
+        await this.handleSpanUpdateOrEnd(event.span, false);
         break;
       case 'span_ended':
-        await this.handleSpanUpdateOrEnd(event.span, false);
+        await this.handleSpanUpdateOrEnd(event.span, true);
         break;
     }
 
@@ -68,21 +79,19 @@ export class LangfuseExporter implements AITracingExporter {
 
   private async handleSpanStarted(span: AnyAISpan): Promise<void> {
     if (span.isRootSpan) {
-      const trace = this.client.trace(this.buildTracePayload(span));
-      this.traceMap.set(span.trace.id, { trace, spans: new Map() });
+      this.initTrace(span);
     }
+    const method = 'handleSpanStarted';
 
-    const traceData = this.traceMap.get(span.trace.id);
+    const traceData = this.getTraceData({ span, method });
     if (!traceData) {
-      console.log('NO TRACE');
-      // TODO: log warning
       return;
     }
 
-    const langfuseParent =
-      span.parent && traceData.spans.has(span.parent.id)
-        ? (traceData.spans.get(span.parent.id) as LangfuseSpan)
-        : traceData.trace;
+    const langfuseParent = this.getLangfuseParent({ traceData, span, method });
+    if (!langfuseParent) {
+      return;
+    }
 
     const payload = this.buildSpanPayload(span, true);
 
@@ -92,36 +101,120 @@ export class LangfuseExporter implements AITracingExporter {
     traceData.spans.set(span.id, langfuseSpan);
   }
 
-  private async handleSpanUpdateOrEnd(span: AnyAISpan, isUpdate: boolean): Promise<void> {
-    const traceData = this.traceMap.get(span.trace.id);
+  private async handleSpanUpdateOrEnd(span: AnyAISpan, isEnd: boolean): Promise<void> {
+    const method = isEnd ? 'handleSpanEnd' : 'handleSpanUpdate';
+
+    const traceData = this.getTraceData({ span, method });
     if (!traceData) {
-      console.log('NO TRACE');
-      // TODO: log warning
       return;
     }
 
     const langfuseSpan = traceData.spans.get(span.id);
     if (!langfuseSpan) {
-      console.log('NO SPAN');
-      // TODO: log warning
+      this.logger.warn('Langfuse exporter: No Langfuse span found for span update/end', {
+        traceId: span.traceId,
+        spanId: span.id,
+        spanName: span.name,
+        spanType: span.type,
+        isRootSpan: span.isRootSpan,
+        parentSpanId: span.parent?.id,
+        availableSpanIds: Array.from(traceData.spans.keys()),
+        method,
+      });
       return;
     }
 
-    if (isUpdate) {
-      langfuseSpan.update(this.buildSpanPayload(span, false));
-    } else {
-      langfuseSpan.end(this.buildSpanPayload(span, false));
+    // use update for both update & end, so that we can use the
+    // end time we set when ending the span.
+    langfuseSpan.update(this.buildSpanPayload(span, false));
 
-      if (span.isRootSpan) {
-        traceData.trace.update({ output: span.output });
-        this.traceMap.delete(span.trace.id);
-      }
+    if (isEnd && span.isRootSpan) {
+      traceData.trace.update({ output: span.output });
+      this.traceMap.delete(span.traceId);
     }
   }
 
+  private async handleEventSpan(span: AnyAISpan): Promise<void> {
+    if (span.isRootSpan) {
+      this.logger.debug('Langfuse exporter: Creating trace', {
+        traceId: span.traceId,
+        spanId: span.id,
+        spanName: span.name,
+        method: 'handleEventSpan',
+      });
+      this.initTrace(span);
+    }
+    const method = 'handleEventSpan';
+
+    const traceData = this.getTraceData({ span, method });
+    if (!traceData) {
+      return;
+    }
+
+    const langfuseParent = this.getLangfuseParent({ traceData, span, method });
+    if (!langfuseParent) {
+      return;
+    }
+
+    const payload = this.buildSpanPayload(span, true);
+
+    const langfuseEvent = langfuseParent.event(payload);
+
+    traceData.events.set(span.id, langfuseEvent);
+  }
+
+  private initTrace(span: AnyAISpan): void {
+    const trace = this.client.trace(this.buildTracePayload(span));
+    this.traceMap.set(span.traceId, { trace, spans: new Map(), events: new Map() });
+  }
+
+  private getTraceData(options: { span: AnyAISpan; method: string }): TraceData | undefined {
+    const { span, method } = options;
+    if (this.traceMap.has(span.traceId)) {
+      return this.traceMap.get(span.traceId);
+    }
+    this.logger.warn('Langfuse exporter: No trace data found for span', {
+      traceId: span.traceId,
+      spanId: span.id,
+      spanName: span.name,
+      spanType: span.type,
+      isRootSpan: span.isRootSpan,
+      parentSpanId: span.parent?.id,
+      method,
+    });
+  }
+
+  private getLangfuseParent(options: {
+    traceData: TraceData;
+    span: AnyAISpan;
+    method: string;
+  }): LangfuseParent | undefined {
+    const { traceData, span, method } = options;
+
+    const parentId = span.parent?.id;
+    if (!parentId) {
+      return traceData.trace;
+    }
+    if (traceData.spans.has(parentId)) {
+      return traceData.spans.get(parentId);
+    }
+    if (traceData.events.has(parentId)) {
+      return traceData.events.get(parentId);
+    }
+    this.logger.warn('Langfuse exporter: No parent data found for span', {
+      traceId: span.traceId,
+      spanId: span.id,
+      spanName: span.name,
+      spanType: span.type,
+      isRootSpan: span.isRootSpan,
+      parentSpanId: span.parent?.id,
+      method,
+    });
+  }
+
   private buildTracePayload(span: AnyAISpan): Record<string, any> {
     const payload: Record<string, any> = {
-      id: span.trace.id,
+      id: span.traceId,
       name: span.name,
     };
 
@@ -133,8 +226,8 @@ export class LangfuseExporter implements AITracingExporter {
 
     payload.metadata = {
       spanType: span.type,
-      ...sanitizeMetadata(span.attributes),
-      ...sanitizeMetadata(remainingMetadata),
+      ...span.attributes,
+      ...remainingMetadata,
     };
 
     return payload;
@@ -179,8 +272,8 @@ export class LangfuseExporter implements AITracingExporter {
 
     payload.metadata = {
       spanType: span.type,
-      ...sanitizeMetadata(omitKeys(attributes, attributesToOmit)),
-      ...sanitizeMetadata(span.metadata),
+      ...omitKeys(attributes, attributesToOmit),
+      ...span.metadata,
     };
 
     if (span.errorInfo) {