@mastra/langfuse 0.0.0-vector-query-tool-provider-options-20250828222356 → 0.0.0-vnext-20251104230439

package/dist/index.cjs

@@ -1,16 +1,26 @@
 'use strict';
 
-var aiTracing = require('@mastra/core/ai-tracing');
+var observability$1 = require('@mastra/core/observability');
+var utils = require('@mastra/core/utils');
+var observability = require('@mastra/observability');
 var langfuse = require('langfuse');
 
 // src/ai-tracing.ts
-var LangfuseExporter = class {
+var LangfuseExporter = class extends observability.BaseExporter {
   name = "langfuse";
   client;
   realtime;
   traceMap = /* @__PURE__ */ new Map();
   constructor(config) {
+    super(config);
     this.realtime = config.realtime ?? false;
+    if (!config.publicKey || !config.secretKey) {
+      this.setDisabled(
+        `Missing required credentials (publicKey: ${!!config.publicKey}, secretKey: ${!!config.secretKey})`
+      );
+      this.client = null;
+      return;
+    }
     this.client = new langfuse.Langfuse({
       publicKey: config.publicKey,
       secretKey: config.secretKey,
@@ -18,16 +28,20 @@ var LangfuseExporter = class {
       ...config.options
     });
   }
-  async exportEvent(event) {
+  async _exportEvent(event) {
+    if (event.exportedSpan.isEvent) {
+      await this.handleEventSpan(event.exportedSpan);
+      return;
+    }
     switch (event.type) {
       case "span_started":
-        await this.handleSpanStarted(event.span);
+        await this.handleSpanStarted(event.exportedSpan);
         break;
       case "span_updated":
-        await this.handleSpanUpdateOrEnd(event.span, true);
+        await this.handleSpanUpdateOrEnd(event.exportedSpan, false);
         break;
       case "span_ended":
-        await this.handleSpanUpdateOrEnd(event.span, false);
+        await this.handleSpanUpdateOrEnd(event.exportedSpan, true);
         break;
     }
     if (this.realtime) {
@@ -36,43 +50,135 @@ 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);
+    const langfuseSpan = span.type === observability$1.AISpanType.MODEL_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);
     traceData.spans.set(span.id, langfuseSpan);
+    traceData.activeSpans.add(span.id);
   }
-  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");
+      if (isEnd && span.isEvent) {
+        traceData.activeSpans.delete(span.id);
+        if (traceData.activeSpans.size === 0) {
+          this.traceMap.delete(span.traceId);
+        }
+        return;
+      }
+      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.parentSpanId,
+        method
+      });
       return;
     }
-    if (isUpdate) {
-      langfuseSpan.update(this.buildSpanPayload(span, false));
-    } else {
-      langfuseSpan.end(this.buildSpanPayload(span, false));
+    langfuseSpan.update(this.buildSpanPayload(span, false));
+    if (isEnd) {
+      traceData.activeSpans.delete(span.id);
       if (span.isRootSpan) {
         traceData.trace.update({ output: span.output });
-        this.traceMap.delete(span.trace.id);
+      }
+      if (traceData.activeSpans.size === 0) {
+        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);
+    if (!span.endTime) {
+      traceData.activeSpans.add(span.id);
+    }
+  }
+  initTrace(span) {
+    const trace = this.client.trace(this.buildTracePayload(span));
+    this.traceMap.set(span.traceId, {
+      trace,
+      spans: /* @__PURE__ */ new Map(),
+      events: /* @__PURE__ */ new Map(),
+      activeSpans: /* @__PURE__ */ new Set(),
+      rootSpanId: span.id
+    });
+  }
+  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.parentSpanId,
+      method
+    });
+  }
+  getLangfuseParent(options) {
+    const { traceData, span, method } = options;
+    const parentId = span.parentSpanId;
+    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.parentSpanId,
+      method
+    });
+  }
   buildTracePayload(span) {
     const payload = {
-      id: span.trace.id,
+      id: span.traceId,
       name: span.name
     };
     const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};
@@ -81,11 +187,53 @@ 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;
   }
+  /**
+   * Normalize usage data to handle both AI SDK v4 and v5 formats.
+   *
+   * AI SDK v4 uses: promptTokens, completionTokens
+   * AI SDK v5 uses: inputTokens, outputTokens
+   *
+   * This function normalizes to a unified format that Langfuse can consume,
+   * prioritizing v5 format while maintaining backward compatibility.
+   *
+   * @param usage - Token usage data from AI SDK (v4 or v5 format)
+   * @returns Normalized usage object, or undefined if no usage data available
+   */
+  normalizeUsage(usage) {
+    if (!usage) return void 0;
+    const normalized = {};
+    const inputTokens = usage.inputTokens ?? usage.promptTokens;
+    if (inputTokens !== void 0) {
+      normalized.input = inputTokens;
+    }
+    const outputTokens = usage.outputTokens ?? usage.completionTokens;
+    if (outputTokens !== void 0) {
+      normalized.output = outputTokens;
+    }
+    if (usage.totalTokens !== void 0) {
+      normalized.total = usage.totalTokens;
+    } else if (normalized.input !== void 0 && normalized.output !== void 0) {
+      normalized.total = normalized.input + normalized.output;
+    }
+    if (usage.reasoningTokens !== void 0) {
+      normalized.reasoning = usage.reasoningTokens;
+    }
+    if (usage.cachedInputTokens !== void 0) {
+      normalized.cachedInput = usage.cachedInputTokens;
+    }
+    if (usage.promptCacheHitTokens !== void 0) {
+      normalized.promptCacheHit = usage.promptCacheHitTokens;
+    }
+    if (usage.promptCacheMissTokens !== void 0) {
+      normalized.promptCacheMiss = usage.promptCacheMissTokens;
+    }
+    return Object.keys(normalized).length > 0 ? normalized : void 0;
+  }
   buildSpanPayload(span, isCreate) {
     const payload = {};
     if (isCreate) {
@@ -98,25 +246,28 @@ var LangfuseExporter = class {
     if (span.endTime !== void 0) payload.endTime = span.endTime;
     const attributes = span.attributes ?? {};
     const attributesToOmit = [];
-    if (span.type === aiTracing.AISpanType.LLM_GENERATION) {
-      const llmAttr = attributes;
-      if (llmAttr.model !== void 0) {
-        payload.model = llmAttr.model;
+    if (span.type === observability$1.AISpanType.MODEL_GENERATION) {
+      const modelAttr = attributes;
+      if (modelAttr.model !== void 0) {
+        payload.model = modelAttr.model;
         attributesToOmit.push("model");
       }
-      if (llmAttr.usage !== void 0) {
-        payload.usage = llmAttr.usage;
+      if (modelAttr.usage !== void 0) {
+        const normalizedUsage = this.normalizeUsage(modelAttr.usage);
+        if (normalizedUsage) {
+          payload.usage = normalizedUsage;
+        }
         attributesToOmit.push("usage");
       }
-      if (llmAttr.parameters !== void 0) {
-        payload.modelParameters = llmAttr.parameters;
+      if (modelAttr.parameters !== void 0) {
+        payload.modelParameters = modelAttr.parameters;
         attributesToOmit.push("parameters");
       }
     }
     payload.metadata = {
       spanType: span.type,
-      ...aiTracing.sanitizeMetadata(aiTracing.omitKeys(attributes, attributesToOmit)),
-      ...aiTracing.sanitizeMetadata(span.metadata)
+      ...utils.omitKeys(attributes, attributesToOmit),
+      ...span.metadata
     };
     if (span.errorInfo) {
       payload.level = "ERROR";
@@ -124,9 +275,41 @@ var LangfuseExporter = class {
     }
     return payload;
   }
+  async addScoreToTrace({
+    traceId,
+    spanId,
+    score,
+    reason,
+    scorerName,
+    metadata
+  }) {
+    if (!this.client) return;
+    try {
+      await this.client.score({
+        id: `${traceId}-${scorerName}`,
+        traceId,
+        observationId: spanId,
+        name: scorerName,
+        value: score,
+        ...metadata?.sessionId ? { sessionId: metadata.sessionId } : {},
+        metadata: { ...reason ? { reason } : {} },
+        dataType: "NUMERIC"
+      });
+    } catch (error) {
+      this.logger.error("Langfuse exporter: Error adding score to trace", {
+        error,
+        traceId,
+        spanId,
+        scorerName
+      });
+    }
+  }
   async shutdown() {
-    await this.client.shutdownAsync();
+    if (this.client) {
+      await this.client.shutdownAsync();
+    }
     this.traceMap.clear();
+    await super.shutdown();
   }
 };
 

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":["BaseExporter","Langfuse","AISpanType","omitKeys"],"mappings":";;;;;;;;AA4GO,IAAM,gBAAA,GAAN,cAA+BA,0BAAA,CAAa;AAAA,EACjD,IAAA,GAAO,UAAA;AAAA,EACC,MAAA;AAAA,EACA,QAAA;AAAA,EACA,QAAA,uBAAe,GAAA,EAAuB;AAAA,EAE9C,YAAY,MAAA,EAAgC;AAC1C,IAAA,KAAA,CAAM,MAAM,CAAA;AAEZ,IAAA,IAAA,CAAK,QAAA,GAAW,OAAO,QAAA,IAAY,KAAA;AAEnC,IAAA,IAAI,CAAC,MAAA,CAAO,SAAA,IAAa,CAAC,OAAO,SAAA,EAAW;AAC1C,MAAA,IAAA,CAAK,WAAA;AAAA,QACH,CAAA,yCAAA,EAA4C,CAAC,CAAC,MAAA,CAAO,SAAS,CAAA,aAAA,EAAgB,CAAC,CAAC,MAAA,CAAO,SAAS,CAAA,CAAA;AAAA,OAClG;AAEA,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,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,MAAgB,aAAa,KAAA,EAAsC;AACjE,IAAA,IAAI,KAAA,CAAM,aAAa,OAAA,EAAS;AAC9B,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,YAAY,CAAA;AAC7C,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,YAAY,CAAA;AAC/C,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,KAAK,CAAA;AAC1D,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,IAAI,CAAA;AACzD,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,EAAwC;AACtE,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,0BAAA,CAAW,gBAAA,GAAmB,cAAA,CAAe,UAAA,CAAW,OAAO,CAAA,GAAI,cAAA,CAAe,IAAA,CAAK,OAAO,CAAA;AAE9G,IAAA,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,YAAY,CAAA;AACzC,IAAA,SAAA,CAAU,WAAA,CAAY,GAAA,CAAI,IAAA,CAAK,EAAE,CAAA;AAAA,EACnC;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAyB,KAAA,EAA+B;AAC1F,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;AAEjB,MAAA,IAAI,KAAA,IAAS,KAAK,OAAA,EAAS;AAEzB,QAAA,SAAA,CAAU,WAAA,CAAY,MAAA,CAAO,IAAA,CAAK,EAAE,CAAA;AACpC,QAAA,IAAI,SAAA,CAAU,WAAA,CAAY,IAAA,KAAS,CAAA,EAAG;AACpC,UAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,QACnC;AACA,QAAA;AAAA,MACF;AAEA,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,cAAc,IAAA,CAAK,YAAA;AAAA,QACnB;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,EAAO;AAET,MAAA,SAAA,CAAU,WAAA,CAAY,MAAA,CAAO,IAAA,CAAK,EAAE,CAAA;AAEpC,MAAA,IAAI,KAAK,UAAA,EAAY;AACnB,QAAA,SAAA,CAAU,MAAM,MAAA,CAAO,EAAE,MAAA,EAAQ,IAAA,CAAK,QAAQ,CAAA;AAAA,MAChD;AAGA,MAAA,IAAI,SAAA,CAAU,WAAA,CAAY,IAAA,KAAS,CAAA,EAAG;AACpC,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,MACnC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAwC;AACpE,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;AAG3C,IAAA,IAAI,CAAC,KAAK,OAAA,EAAS;AACjB,MAAA,SAAA,CAAU,WAAA,CAAY,GAAA,CAAI,IAAA,CAAK,EAAE,CAAA;AAAA,IACnC;AAAA,EACF;AAAA,EAEQ,UAAU,IAAA,EAA+B;AAC/C,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;AAAA,MAC9B,KAAA;AAAA,MACA,KAAA,sBAAW,GAAA,EAAI;AAAA,MACf,MAAA,sBAAY,GAAA,EAAI;AAAA,MAChB,WAAA,sBAAiB,GAAA,EAAI;AAAA,MACrB,YAAY,IAAA,CAAK;AAAA,KAClB,CAAA;AAAA,EACH;AAAA,EAEQ,aAAa,OAAA,EAA6E;AAChG,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AAEzB,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;AAEA,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,cAAc,IAAA,CAAK,YAAA;AAAA,MACnB;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,WAAW,IAAA,CAAK,YAAA;AACtB,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,cAAc,IAAA,CAAK,YAAA;AAAA,MACnB;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,kBAAkB,IAAA,EAA8C;AACtE,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;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAcQ,eAAe,KAAA,EAAwE;AAC7F,IAAA,IAAI,CAAC,OAAO,OAAO,MAAA;AAEnB,IAAA,MAAM,aAA8B,EAAC;AAIrC,IAAA,MAAM,WAAA,GAAc,KAAA,CAAM,WAAA,IAAe,KAAA,CAAM,YAAA;AAC/C,IAAA,IAAI,gBAAgB,MAAA,EAAW;AAC7B,MAAA,UAAA,CAAW,KAAA,GAAQ,WAAA;AAAA,IACrB;AAGA,IAAA,MAAM,YAAA,GAAe,KAAA,CAAM,YAAA,IAAgB,KAAA,CAAM,gBAAA;AACjD,IAAA,IAAI,iBAAiB,MAAA,EAAW;AAC9B,MAAA,UAAA,CAAW,MAAA,GAAS,YAAA;AAAA,IACtB;AAGA,IAAA,IAAI,KAAA,CAAM,gBAAgB,MAAA,EAAW;AACnC,MAAA,UAAA,CAAW,QAAQ,KAAA,CAAM,WAAA;AAAA,IAC3B,WAAW,UAAA,CAAW,KAAA,KAAU,MAAA,IAAa,UAAA,CAAW,WAAW,MAAA,EAAW;AAC5E,MAAA,UAAA,CAAW,KAAA,GAAQ,UAAA,CAAW,KAAA,GAAQ,UAAA,CAAW,MAAA;AAAA,IACnD;AAGA,IAAA,IAAI,KAAA,CAAM,oBAAoB,MAAA,EAAW;AACvC,MAAA,UAAA,CAAW,YAAY,KAAA,CAAM,eAAA;AAAA,IAC/B;AAGA,IAAA,IAAI,KAAA,CAAM,sBAAsB,MAAA,EAAW;AACzC,MAAA,UAAA,CAAW,cAAc,KAAA,CAAM,iBAAA;AAAA,IACjC;AAGA,IAAA,IAAI,KAAA,CAAM,yBAAyB,MAAA,EAAW;AAC5C,MAAA,UAAA,CAAW,iBAAiB,KAAA,CAAM,oBAAA;AAAA,IACpC;AACA,IAAA,IAAI,KAAA,CAAM,0BAA0B,MAAA,EAAW;AAC7C,MAAA,UAAA,CAAW,kBAAkB,KAAA,CAAM,qBAAA;AAAA,IACrC;AAEA,IAAA,OAAO,OAAO,IAAA,CAAK,UAAU,CAAA,CAAE,MAAA,GAAS,IAAI,UAAA,GAAa,MAAA;AAAA,EAC3D;AAAA,EAEQ,gBAAA,CAAiB,MAAyB,QAAA,EAAwC;AACxF,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,0BAAA,CAAW,gBAAA,EAAkB;AAC7C,MAAA,MAAM,SAAA,GAAY,UAAA;AAElB,MAAA,IAAI,SAAA,CAAU,UAAU,MAAA,EAAW;AACjC,QAAA,OAAA,CAAQ,QAAQ,SAAA,CAAU,KAAA;AAC1B,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,SAAA,CAAU,UAAU,MAAA,EAAW;AAEjC,QAAA,MAAM,eAAA,GAAkB,IAAA,CAAK,cAAA,CAAe,SAAA,CAAU,KAAK,CAAA;AAC3D,QAAA,IAAI,eAAA,EAAiB;AACnB,UAAA,OAAA,CAAQ,KAAA,GAAQ,eAAA;AAAA,QAClB;AACA,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,SAAA,CAAU,eAAe,MAAA,EAAW;AACtC,QAAA,OAAA,CAAQ,kBAAkB,SAAA,CAAU,UAAA;AACpC,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,cAAA,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,eAAA,CAAgB;AAAA,IACpB,OAAA;AAAA,IACA,MAAA;AAAA,IACA,KAAA;AAAA,IACA,MAAA;AAAA,IACA,UAAA;AAAA,IACA;AAAA,GACF,EAOkB;AAChB,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAElB,IAAA,IAAI;AACF,MAAA,MAAM,IAAA,CAAK,OAAO,KAAA,CAAM;AAAA,QACtB,EAAA,EAAI,CAAA,EAAG,OAAO,CAAA,CAAA,EAAI,UAAU,CAAA,CAAA;AAAA,QAC5B,OAAA;AAAA,QACA,aAAA,EAAe,MAAA;AAAA,QACf,IAAA,EAAM,UAAA;AAAA,QACN,KAAA,EAAO,KAAA;AAAA,QACP,GAAI,UAAU,SAAA,GAAY,EAAE,WAAW,QAAA,CAAS,SAAA,KAAc,EAAC;AAAA,QAC/D,QAAA,EAAU,EAAE,GAAI,MAAA,GAAS,EAAE,MAAA,EAAO,GAAI,EAAC,EAAG;AAAA,QAC1C,QAAA,EAAU;AAAA,OACX,CAAA;AAAA,IACH,SAAS,KAAA,EAAO;AACd,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,gDAAA,EAAkD;AAAA,QAClE,KAAA;AAAA,QACA,OAAA;AAAA,QACA,MAAA;AAAA,QACA;AAAA,OACD,CAAA;AAAA,IACH;AAAA,EACF;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,IAAI,KAAK,MAAA,EAAQ;AACf,MAAA,MAAM,IAAA,CAAK,OAAO,aAAA,EAAc;AAAA,IAClC;AACA,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AACpB,IAAA,MAAM,MAAM,QAAA,EAAS;AAAA,EACvB;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 * MODEL_GENERATION spans become Langfuse generations, all others become spans.\n *\n * Compatible with both AI SDK v4 and v5:\n * - Handles both legacy token usage format (promptTokens/completionTokens)\n *   and v5 format (inputTokens/outputTokens)\n * - Supports v5 reasoning tokens and cache-related metrics\n * - Adapts to v5 streaming protocol changes\n */\n\nimport type { AITracingEvent, AnyExportedAISpan, ModelGenerationAttributes } from '@mastra/core/observability';\nimport { AISpanType } from '@mastra/core/observability';\nimport { omitKeys } from '@mastra/core/utils';\nimport { BaseExporter } from '@mastra/observability';\nimport type { BaseExporterConfig } from '@mastra/observability';\nimport { Langfuse } from 'langfuse';\nimport type { LangfuseTraceClient, LangfuseSpanClient, LangfuseGenerationClient, LangfuseEventClient } from 'langfuse';\n\nexport interface LangfuseExporterConfig extends BaseExporterConfig {\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  events: Map<string, LangfuseEventClient>; // Maps span.id to Langfuse event\n  activeSpans: Set<string>; // Tracks which spans haven't ended yet\n  rootSpanId?: string; // Track the root span ID\n};\n\ntype LangfuseParent = LangfuseTraceClient | LangfuseSpanClient | LangfuseGenerationClient | LangfuseEventClient;\n\n/**\n * Normalized token usage format compatible with Langfuse.\n * This unified format supports both AI SDK v4 and v5 token structures.\n *\n * @example\n * ```typescript\n * // AI SDK v4 format normalizes to:\n * { input: 100, output: 50, total: 150 }\n *\n * // AI SDK v5 format normalizes to:\n * { input: 120, output: 60, total: 180, reasoning: 1000, cachedInput: 50 }\n * ```\n */\ninterface NormalizedUsage {\n  /**\n   * Input tokens sent to the model\n   * @source AI SDK v5: `inputTokens` | AI SDK v4: `promptTokens`\n   */\n  input?: number;\n\n  /**\n   * Output tokens received from the model\n   * @source AI SDK v5: `outputTokens` | AI SDK v4: `completionTokens`\n   */\n  output?: number;\n\n  /**\n   * Total tokens (input + output + reasoning if applicable)\n   * @source AI SDK v4 & v5: `totalTokens`\n   */\n  total?: number;\n\n  /**\n   * Reasoning tokens used by reasoning models\n   * @source AI SDK v5: `reasoningTokens`\n   * @since AI SDK v5.0.0\n   * @example Models like o1-preview, o1-mini\n   */\n  reasoning?: number;\n\n  /**\n   * Cached input tokens (prompt cache hit)\n   * @source AI SDK v5: `cachedInputTokens`\n   * @since AI SDK v5.0.0\n   * @example Anthropic's prompt caching, OpenAI prompt caching\n   */\n  cachedInput?: number;\n\n  /**\n   * Prompt cache hit tokens (legacy format)\n   * @source AI SDK v4: `promptCacheHitTokens`\n   * @deprecated Prefer `cachedInput` from v5 format\n   */\n  promptCacheHit?: number;\n\n  /**\n   * Prompt cache miss tokens (legacy format)\n   * @source AI SDK v4: `promptCacheMissTokens`\n   * @deprecated Prefer v5 format which uses `cachedInputTokens`\n   */\n  promptCacheMiss?: number;\n}\n\nexport class LangfuseExporter extends BaseExporter {\n  name = 'langfuse';\n  private client: Langfuse;\n  private realtime: boolean;\n  private traceMap = new Map<string, TraceData>();\n\n  constructor(config: LangfuseExporterConfig) {\n    super(config);\n\n    this.realtime = config.realtime ?? false;\n\n    if (!config.publicKey || !config.secretKey) {\n      this.setDisabled(\n        `Missing required credentials (publicKey: ${!!config.publicKey}, secretKey: ${!!config.secretKey})`,\n      );\n      // Create a no-op client to prevent runtime errors\n      this.client = null as any;\n      return;\n    }\n\n    this.client = new Langfuse({\n      publicKey: config.publicKey,\n      secretKey: config.secretKey,\n      baseUrl: config.baseUrl,\n      ...config.options,\n    });\n  }\n\n  protected async _exportEvent(event: AITracingEvent): Promise<void> {\n    if (event.exportedSpan.isEvent) {\n      await this.handleEventSpan(event.exportedSpan);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.exportedSpan);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, 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: AnyExportedAISpan): 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.MODEL_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);\n\n    traceData.spans.set(span.id, langfuseSpan);\n    traceData.activeSpans.add(span.id); // Track as active\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyExportedAISpan, 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      // For event spans that only send SPAN_ENDED, we might not have the span yet\n      if (isEnd && span.isEvent) {\n        // Just make sure it's not in active spans\n        traceData.activeSpans.delete(span.id);\n        if (traceData.activeSpans.size === 0) {\n          this.traceMap.delete(span.traceId);\n        }\n        return;\n      }\n\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.parentSpanId,\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) {\n      // Remove from active spans\n      traceData.activeSpans.delete(span.id);\n\n      if (span.isRootSpan) {\n        traceData.trace.update({ output: span.output });\n      }\n\n      // Only clean up the trace when ALL spans have ended\n      if (traceData.activeSpans.size === 0) {\n        this.traceMap.delete(span.traceId);\n      }\n    }\n  }\n\n  private async handleEventSpan(span: AnyExportedAISpan): 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    // Event spans are typically immediately ended, but let's track them properly\n    if (!span.endTime) {\n      traceData.activeSpans.add(span.id);\n    }\n  }\n\n  private initTrace(span: AnyExportedAISpan): void {\n    const trace = this.client.trace(this.buildTracePayload(span));\n    this.traceMap.set(span.traceId, {\n      trace,\n      spans: new Map(),\n      events: new Map(),\n      activeSpans: new Set(),\n      rootSpanId: span.id,\n    });\n  }\n\n  private getTraceData(options: { span: AnyExportedAISpan; method: string }): TraceData | undefined {\n    const { span, method } = options;\n\n    if (this.traceMap.has(span.traceId)) {\n      return this.traceMap.get(span.traceId);\n    }\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.parentSpanId,\n      method,\n    });\n  }\n\n  private getLangfuseParent(options: {\n    traceData: TraceData;\n    span: AnyExportedAISpan;\n    method: string;\n  }): LangfuseParent | undefined {\n    const { traceData, span, method } = options;\n\n    const parentId = span.parentSpanId;\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.parentSpanId,\n      method,\n    });\n  }\n\n  private buildTracePayload(span: AnyExportedAISpan): 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  /**\n   * Normalize usage data to handle both AI SDK v4 and v5 formats.\n   *\n   * AI SDK v4 uses: promptTokens, completionTokens\n   * AI SDK v5 uses: inputTokens, outputTokens\n   *\n   * This function normalizes to a unified format that Langfuse can consume,\n   * prioritizing v5 format while maintaining backward compatibility.\n   *\n   * @param usage - Token usage data from AI SDK (v4 or v5 format)\n   * @returns Normalized usage object, or undefined if no usage data available\n   */\n  private normalizeUsage(usage: ModelGenerationAttributes['usage']): NormalizedUsage | undefined {\n    if (!usage) return undefined;\n\n    const normalized: NormalizedUsage = {};\n\n    // Handle input tokens (v5 'inputTokens' or v4 'promptTokens')\n    // Using ?? to prioritize v5 format while falling back to v4\n    const inputTokens = usage.inputTokens ?? usage.promptTokens;\n    if (inputTokens !== undefined) {\n      normalized.input = inputTokens;\n    }\n\n    // Handle output tokens (v5 'outputTokens' or v4 'completionTokens')\n    const outputTokens = usage.outputTokens ?? usage.completionTokens;\n    if (outputTokens !== undefined) {\n      normalized.output = outputTokens;\n    }\n\n    // Total tokens - calculate if not provided\n    if (usage.totalTokens !== undefined) {\n      normalized.total = usage.totalTokens;\n    } else if (normalized.input !== undefined && normalized.output !== undefined) {\n      normalized.total = normalized.input + normalized.output;\n    }\n\n    // AI SDK v5-specific: reasoning tokens\n    if (usage.reasoningTokens !== undefined) {\n      normalized.reasoning = usage.reasoningTokens;\n    }\n\n    // AI SDK v5-specific: cached tokens (cache hit)\n    if (usage.cachedInputTokens !== undefined) {\n      normalized.cachedInput = usage.cachedInputTokens;\n    }\n\n    // Legacy cache metrics (promptCacheHitTokens/promptCacheMissTokens)\n    if (usage.promptCacheHitTokens !== undefined) {\n      normalized.promptCacheHit = usage.promptCacheHitTokens;\n    }\n    if (usage.promptCacheMissTokens !== undefined) {\n      normalized.promptCacheMiss = usage.promptCacheMissTokens;\n    }\n\n    return Object.keys(normalized).length > 0 ? normalized : undefined;\n  }\n\n  private buildSpanPayload(span: AnyExportedAISpan, 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.MODEL_GENERATION) {\n      const modelAttr = attributes as ModelGenerationAttributes;\n\n      if (modelAttr.model !== undefined) {\n        payload.model = modelAttr.model;\n        attributesToOmit.push('model');\n      }\n\n      if (modelAttr.usage !== undefined) {\n        // Normalize usage to handle both v4 and v5 formats\n        const normalizedUsage = this.normalizeUsage(modelAttr.usage);\n        if (normalizedUsage) {\n          payload.usage = normalizedUsage;\n        }\n        attributesToOmit.push('usage');\n      }\n\n      if (modelAttr.parameters !== undefined) {\n        payload.modelParameters = modelAttr.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 addScoreToTrace({\n    traceId,\n    spanId,\n    score,\n    reason,\n    scorerName,\n    metadata,\n  }: {\n    traceId: string;\n    spanId?: string;\n    score: number;\n    reason?: string;\n    scorerName: string;\n    metadata?: Record<string, any>;\n  }): Promise<void> {\n    if (!this.client) return;\n\n    try {\n      await this.client.score({\n        id: `${traceId}-${scorerName}`,\n        traceId,\n        observationId: spanId,\n        name: scorerName,\n        value: score,\n        ...(metadata?.sessionId ? { sessionId: metadata.sessionId } : {}),\n        metadata: { ...(reason ? { reason } : {}) },\n        dataType: 'NUMERIC',\n      });\n    } catch (error) {\n      this.logger.error('Langfuse exporter: Error adding score to trace', {\n        error,\n        traceId,\n        spanId,\n        scorerName,\n      });\n    }\n  }\n\n  async shutdown(): Promise<void> {\n    if (this.client) {\n      await this.client.shutdownAsync();\n    }\n    this.traceMap.clear();\n    await super.shutdown();\n  }\n}\n"]}