@mastra/langfuse 0.0.0-unified-sidebar-20251010130811 → 0.0.0-unified-workspace-snapshot-20260128233410

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/ai-tracing.ts"],"names":["ConsoleLogger","Langfuse","AISpanType","omitKeys"],"mappings":";;;;;;;AA4CO,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;AAEpE,IAAA,IAAI,CAAC,MAAA,CAAO,SAAA,IAAa,CAAC,OAAO,SAAA,EAAW;AAC1C,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,2EAAA,EAA6E;AAAA,QAC7F,YAAA,EAAc,CAAC,CAAC,MAAA,CAAO,SAAA;AAAA,QACvB,YAAA,EAAc,CAAC,CAAC,MAAA,CAAO;AAAA,OACxB,CAAA;AAED,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,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAEhB,MAAA;AAAA,IACF;AAEA,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,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;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,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,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,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;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 {\n  AITracingExporter,\n  AITracingEvent,\n  AnyExportedAISpan,\n  LLMGenerationAttributes,\n} 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  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\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\n    if (!config.publicKey || !config.secretKey) {\n      this.logger.error('LangfuseExporter: Missing required credentials, exporter will be disabled', {\n        hasPublicKey: !!config.publicKey,\n        hasSecretKey: !!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  async exportEvent(event: AITracingEvent): Promise<void> {\n    if (!this.client) {\n      // Exporter is disabled due to missing credentials\n      return;\n    }\n\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.LLM_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  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.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 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  }\n}\n"]}
+{"version":3,"sources":["../src/metrics.ts","../src/tracing.ts","../src/helpers.ts"],"names":["TrackingExporter","Langfuse","SpanType","omitKeys"],"mappings":";;;;;;;;;;AAiBO,SAAS,mBAAmB,KAAA,EAA0C;AAC3E,EAAA,IAAI,CAAC,KAAA,EAAO,OAAO,EAAC;AAEpB,EAAA,MAAM,UAAgC,EAAC;AAEvC,EAAA,IAAI,KAAA,CAAM,gBAAgB,MAAA,EAAW;AACnC,IAAA,OAAA,CAAQ,QAAQ,KAAA,CAAM,WAAA;AAEtB,IAAA,IAAI,KAAA,CAAM,YAAA,EAAc,UAAA,KAAe,MAAA,EAAW;AAChD,MAAA,OAAA,CAAQ,wBAAA,GAA2B,MAAM,YAAA,CAAa,UAAA;AACtD,MAAA,OAAA,CAAQ,SAAS,OAAA,CAAQ,wBAAA;AAAA,IAC3B;AAAA,EACF;AAEA,EAAA,IAAI,KAAA,CAAM,YAAA,EAAc,SAAA,KAAc,MAAA,EAAW;AAC/C,IAAA,OAAA,CAAQ,uBAAA,GAA0B,MAAM,YAAA,CAAa,SAAA;AAAA,EACvD;AAEA,EAAA,IAAI,KAAA,CAAM,iBAAiB,MAAA,EAAW;AACpC,IAAA,OAAA,CAAQ,SAAS,KAAA,CAAM,YAAA;AAAA,EACzB;AAEA,EAAA,IAAI,KAAA,CAAM,aAAA,EAAe,SAAA,KAAc,MAAA,EAAW;AAChD,IAAA,OAAA,CAAQ,SAAA,GAAY,MAAM,aAAA,CAAc,SAAA;AAAA,EAC1C;AAGA,EAAA,IAAI,OAAA,CAAQ,KAAA,IAAS,IAAA,IAAQ,OAAA,CAAQ,UAAU,IAAA,EAAM;AACnD,IAAA,OAAA,CAAQ,KAAA,GAAQ,OAAA,CAAQ,KAAA,GAAQ,OAAA,CAAQ,MAAA;AACxC,IAAA,IAAI,OAAA,CAAQ,4BAA4B,IAAA,EAAM;AAC5C,MAAA,OAAA,CAAQ,SAAS,OAAA,CAAQ,wBAAA;AAAA,IAC3B;AAAA,EACF;AAEA,EAAA,OAAO,OAAA;AACT;;;ACTO,IAAM,gBAAA,GAAN,cAA+BA,8BAAA,CAMpC;AAAA,EACA,IAAA,GAAO,UAAA;AAAA,EACP,OAAA;AAAA,EACA,SAAA;AAAA,EAEA,WAAA,CAAY,MAAA,GAAiC,EAAC,EAAG;AAE/C,IAAA,MAAM,SAAA,GAAY,MAAA,CAAO,SAAA,IAAa,OAAA,CAAQ,GAAA,CAAI,mBAAA;AAClD,IAAA,MAAM,SAAA,GAAY,MAAA,CAAO,SAAA,IAAa,OAAA,CAAQ,GAAA,CAAI,mBAAA;AAClD,IAAA,MAAM,OAAA,GAAU,MAAA,CAAO,OAAA,IAAW,OAAA,CAAQ,GAAA,CAAI,iBAAA;AAE9C,IAAA,KAAA,CAAM;AAAA,MACJ,GAAG,MAAA;AAAA,MACH,SAAA;AAAA,MACA,SAAA;AAAA,MACA;AAAA,KACD,CAAA;AAED,IAAA,IAAA,CAAK,SAAA,GAAY,OAAO,QAAA,IAAY,KAAA;AAEpC,IAAA,IAAI,CAAC,SAAA,IAAa,CAAC,SAAA,EAAW;AAC5B,MAAA,MAAM,kBAAkB,MAAA,CAAO,SAAA,GAC3B,gBACA,OAAA,CAAQ,GAAA,CAAI,sBACV,UAAA,GACA,SAAA;AACN,MAAA,MAAM,kBAAkB,MAAA,CAAO,SAAA,GAC3B,gBACA,OAAA,CAAQ,GAAA,CAAI,sBACV,UAAA,GACA,SAAA;AACN,MAAA,IAAA,CAAK,WAAA;AAAA,QACH,CAAA,yCAAA,EAA4C,eAAe,CAAA,aAAA,EAAgB,eAAe,CAAA,gGAAA;AAAA,OAE5F;AACA,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,OAAA,GAAU,IAAIC,iBAAA,CAAS;AAAA,MAC1B,SAAA;AAAA,MACA,SAAA;AAAA,MACA,OAAA;AAAA,MACA,GAAG,MAAA,CAAO;AAAA,KACX,CAAA;AAAA,EACH;AAAA,EAEA,MAAyB,uBAAA,GAAyC;AAEhE,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,MAAM,IAAA,CAAK,SAAS,UAAA,EAAW;AAAA,IACjC;AAAA,EACF;AAAA,EAEA,MAAyB,WAAW,IAAA,EAGS;AAC3C,IAAA,MAAM,EAAE,MAAK,GAAI,IAAA;AAMjB,IAAA,OAAO,KAAK,OAAA,EAAS,KAAA,CAAM,IAAA,CAAK,iBAAA,CAAkB,IAAI,CAAC,CAAA;AAAA,EACzD;AAAA,EAEA,MAAyB,YAAY,IAAA,EAGE;AACrC,IAAA,MAAM,EAAE,IAAA,EAAM,SAAA,EAAU,GAAI,IAAA;AAC5B,IAAA,MAAM,cAAA,GAAiB,SAAA,CAAU,eAAA,CAAgB,EAAE,MAAM,CAAA;AACzD,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,MAAM,SAAS,CAAA;AAC3D,IAAA,OAAO,cAAA,CAAe,MAAM,OAAO,CAAA;AAAA,EACrC;AAAA,EAEA,MAAyB,WAAW,IAAA,EAGE;AACpC,IAAA,MAAM,EAAE,IAAA,EAAM,SAAA,EAAU,GAAI,IAAA;AAC5B,IAAA,MAAM,cAAA,GAAiB,SAAA,CAAU,eAAA,CAAgB,EAAE,MAAM,CAAA;AACzD,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,MAAM,SAAS,CAAA;AAE3D,IAAA,MAAM,YAAA,GACJ,IAAA,CAAK,IAAA,KAASC,wBAAA,CAAS,gBAAA,GAAmB,cAAA,CAAe,UAAA,CAAW,OAAO,CAAA,GAAI,cAAA,CAAe,IAAA,CAAK,OAAO,CAAA;AAE5G,IAAA,IAAA,CAAK,MAAA,CAAO,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,IAAI,CAAA,YAAA,CAAA,EAAgB;AAAA,MAC5C,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,OAAA,CAAQ,EAAA;AAAA,MAChB,MAAA,EAAQ;AAAA,KACT,CAAA;AAED,IAAA,OAAO,YAAA;AAAA,EACT;AAAA,EAEA,MAAyB,YAAY,IAAA,EAA8E;AACjH,IAAA,MAAM,EAAE,IAAA,EAAM,SAAA,EAAU,GAAI,IAAA;AAC5B,IAAA,MAAM,eAAe,SAAA,CAAU,OAAA,CAAQ,EAAE,MAAA,EAAQ,IAAA,CAAK,IAAI,CAAA;AAC1D,IAAA,IAAI,YAAA,EAAc;AAChB,MAAA,IAAA,CAAK,MAAA,CAAO,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,IAAI,CAAA,uBAAA,CAAA,EAA2B;AAAA,QACvD,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,YAAA,CAAa,EAAA;AAAA,QACrB,MAAA,EAAQ;AAAA,OACT,CAAA;AAED,MAAA,MAAM,aAAA,GAAgB,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,OAAO,SAAS,CAAA;AAIlE,MAAA,YAAA,CAAa,OAAO,aAAa,CAAA;AAAA,IACnC;AAAA,EACF;AAAA,EAEA,MAAyB,YAAY,IAAA,EAA8E;AACjH,IAAA,MAAM,EAAE,IAAA,EAAM,SAAA,EAAU,GAAI,IAAA;AAC5B,IAAA,MAAM,eAAe,SAAA,CAAU,OAAA,CAAQ,EAAE,MAAA,EAAQ,IAAA,CAAK,IAAI,CAAA;AAG1D,IAAA,YAAA,EAAc,OAAO,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAA,EAAO,SAAS,CAAC,CAAA;AAElE,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAM,YAAA,GAAe,UAAU,OAAA,EAAQ;AACvC,MAAA,YAAA,EAAc,MAAA,CAAO,EAAE,MAAA,EAAQ,IAAA,CAAK,QAAQ,CAAA;AAAA,IAC9C;AAAA,EACF;AAAA,EAEA,MAAyB,WAAW,IAAA,EAAoE;AACtG,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,IAAA;AACzB,IAAA,IAAA,CAAK,GAAA,CAAI;AAAA,MACP,KAAA,EAAO,OAAA;AAAA,MACP,eAAe,MAAA,CAAO;AAAA,KACvB,CAAA;AAAA,EACH;AAAA,EAEQ,kBAAkB,IAAA,EAA4C;AACpE,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,IAAI,IAAA,CAAK,IAAA,EAAM,MAAA,EAAQ,OAAA,CAAQ,OAAO,IAAA,CAAK,IAAA;AAE3C,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,EAWQ,kBAAA,CAAmB,WAA8B,IAAA,EAAuD;AAC9G,IAAA,IAAI,gBAAoC,IAAA,CAAK,EAAA;AAE7C,IAAA,OAAO,aAAA,EAAe;AACpB,MAAA,MAAM,mBAAmB,SAAA,CAAU,WAAA,CAAY,EAAE,MAAA,EAAQ,eAAe,CAAA;AAExE,MAAA,IAAI,kBAAkB,MAAA,EAAQ;AAC5B,QAAA,IAAA,CAAK,MAAA,CAAO,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,IAAI,CAAA,mCAAA,CAAA,EAAuC;AAAA,UACnE,SAAS,IAAA,CAAK,OAAA;AAAA,UACd,QAAQ,IAAA,CAAK,EAAA;AAAA,UACb,QAAQ,gBAAA,EAAkB;AAAA,SAC3B,CAAA;AACD,QAAA,OAAO,gBAAA,CAAiB,MAAA;AAAA,MAC1B;AACA,MAAA,aAAA,GAAgB,SAAA,CAAU,WAAA,CAAY,EAAE,MAAA,EAAQ,eAAe,CAAA;AAAA,IACjE;AAEA,IAAA,OAAO,MAAA;AAAA,EACT;AAAA,EAEQ,gBAAA,CACN,IAAA,EACA,QAAA,EACA,SAAA,EACqB;AACrB,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;AAAA,IAC3B;AAEA,IAAA,IAAI,IAAA,CAAK,KAAA,KAAU,MAAA,EAAW,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AACnD,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;AAExC,IAAA,MAAM,QAAA,GAAgC;AAAA,MACpC,GAAG,IAAA,CAAK;AAAA,KACV;AAGA,IAAA,MAAM,mBAA6B,EAAC;AACpC,IAAA,MAAM,iBAA2B,EAAC;AAElC,IAAA,IAAI,IAAA,CAAK,IAAA,KAASA,wBAAA,CAAS,gBAAA,EAAkB;AAC3C,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;AACjC,QAAA,OAAA,CAAQ,YAAA,GAAe,kBAAA,CAAmB,SAAA,CAAU,KAAK,CAAA;AACzD,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;AAOA,MAAA,MAAM,UAAA,GAAa,IAAA,CAAK,kBAAA,CAAmB,SAAA,EAAW,IAAI,CAAA;AAC1D,MAAA,MAAM,iBAAA,GAAoB,UAAA,EAAY,IAAA,KAAS,MAAA,IAAa,YAAY,OAAA,KAAY,MAAA;AACpF,MAAA,MAAM,KAAA,GAAQ,YAAY,EAAA,KAAO,MAAA;AAEjC,MAAA,IAAI,qBAAqB,KAAA,EAAO;AAC9B,QAAA,OAAA,CAAQ,SAAS,EAAC;AAElB,QAAA,IAAI,YAAY,IAAA,KAAS,MAAA,EAAW,OAAA,CAAQ,MAAA,CAAO,OAAO,UAAA,CAAW,IAAA;AACrE,QAAA,IAAI,YAAY,OAAA,KAAY,MAAA,EAAW,OAAA,CAAQ,MAAA,CAAO,UAAU,UAAA,CAAW,OAAA;AAC3E,QAAA,IAAI,YAAY,EAAA,KAAO,MAAA,EAAW,OAAA,CAAQ,MAAA,CAAO,KAAK,UAAA,CAAW,EAAA;AAEjE,QAAA,cAAA,CAAe,KAAK,UAAU,CAAA;AAAA,MAChC;AAGA,MAAA,IAAI,SAAA,CAAU,wBAAwB,MAAA,EAAW;AAC/C,QAAA,OAAA,CAAQ,sBAAsB,SAAA,CAAU,mBAAA;AACxC,QAAA,gBAAA,CAAiB,KAAK,qBAAqB,CAAA;AAAA,MAC7C;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,GAAGA,cAAA,CAAS,QAAA,EAAU,cAAc;AAAA,KACtC;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,OAAA,EAAS;AAEnB,IAAA,IAAI;AACF,MAAA,MAAM,IAAA,CAAK,QAAQ,KAAA,CAAM;AAAA,QACvB,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;AAAA;AAAA;AAAA,EAKA,MAAyB,MAAA,GAAwB;AAC/C,IAAA,IAAI,KAAK,OAAA,EAAS;AAChB,MAAA,MAAM,IAAA,CAAK,QAAQ,UAAA,EAAW;AAAA,IAChC;AAAA,EACF;AAAA,EAEA,MAAe,aAAA,GAA+B;AAC5C,IAAA,IAAI,KAAK,OAAA,EAAS;AAChB,MAAA,MAAM,IAAA,CAAK,QAAQ,aAAA,EAAc;AAAA,IACnC;AAAA,EACF;AACF;;;AC7SO,SAAS,mBAAmB,MAAA,EAAoD;AACrF,EAAA,OAAO,CAAA,IAAA,MAAS;AAAA,IACd,GAAG,IAAA;AAAA,IACH,QAAA,EAAU;AAAA,MACR,GAAG,IAAA,CAAK,QAAA;AAAA,MACR,QAAA,EAAU;AAAA,QACR,GAAI,KAAK,QAAA,EAAU,QAAA;AAAA,QACnB,MAAA,EAAQ;AAAA,UACN,GAAI,MAAA,CAAO,IAAA,KAAS,UAAa,EAAE,IAAA,EAAM,OAAO,IAAA,EAAK;AAAA,UACrD,GAAI,MAAA,CAAO,OAAA,KAAY,UAAa,EAAE,OAAA,EAAS,OAAO,OAAA,EAAQ;AAAA,UAC9D,GAAI,MAAA,CAAO,EAAA,KAAO,UAAa,EAAE,EAAA,EAAI,OAAO,EAAA;AAAG;AACjD;AACF;AACF,GACF,CAAA;AACF","file":"index.cjs","sourcesContent":["import type { UsageStats } from '@mastra/core/observability';\n\n/**\n * Token usage format compatible with Langfuse.\n */\nexport interface LangfuseUsageMetrics {\n  input?: number;\n  output?: number;\n  total?: number;\n  reasoning?: number;\n  cache_read_input_tokens?: number;\n  cache_write_input_tokens?: number;\n}\n\n/**\n * Formats UsageStats to Langfuse's expected format.\n */\nexport function formatUsageMetrics(usage?: UsageStats): LangfuseUsageMetrics {\n  if (!usage) return {};\n\n  const metrics: LangfuseUsageMetrics = {};\n\n  if (usage.inputTokens !== undefined) {\n    metrics.input = usage.inputTokens;\n\n    if (usage.inputDetails?.cacheWrite !== undefined) {\n      metrics.cache_write_input_tokens = usage.inputDetails.cacheWrite;\n      metrics.input -= metrics.cache_write_input_tokens;\n    }\n  }\n\n  if (usage.inputDetails?.cacheRead !== undefined) {\n    metrics.cache_read_input_tokens = usage.inputDetails.cacheRead;\n  }\n\n  if (usage.outputTokens !== undefined) {\n    metrics.output = usage.outputTokens;\n  }\n\n  if (usage.outputDetails?.reasoning !== undefined) {\n    metrics.reasoning = usage.outputDetails.reasoning;\n  }\n\n  // Use explicit null checks to handle zero values correctly\n  if (metrics.input != null && metrics.output != null) {\n    metrics.total = metrics.input + metrics.output;\n    if (metrics.cache_write_input_tokens != null) {\n      metrics.total += metrics.cache_write_input_tokens;\n    }\n  }\n\n  return metrics;\n}\n","/**\n * Langfuse Exporter for Mastra Observability\n *\n * This exporter sends observability data to Langfuse.\n * Root spans start traces in Langfuse.\n * MODEL_GENERATION spans become Langfuse generations, all others become spans.\n */\n\nimport type { AnyExportedSpan, ModelGenerationAttributes, SpanErrorInfo } from '@mastra/core/observability';\nimport { SpanType } from '@mastra/core/observability';\nimport { omitKeys } from '@mastra/core/utils';\nimport { TrackingExporter } from '@mastra/observability';\nimport type { TrackingExporterConfig, TraceData } from '@mastra/observability';\nimport { Langfuse } from 'langfuse';\nimport type { LangfuseTraceClient, LangfuseSpanClient, LangfuseGenerationClient, LangfuseEventClient } from 'langfuse';\nimport { formatUsageMetrics } from './metrics';\n\nexport interface LangfuseExporterConfig extends TrackingExporterConfig {\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 LangfusePromptData = { name?: string; version?: number; id?: string };\n\n/**\n * With Langfuse, data from the root span is stored in both the Root and the\n * first span.\n */\n\ntype LangfuseRoot = LangfuseTraceClient;\ntype LangfuseSpan = LangfuseSpanClient | LangfuseGenerationClient;\ntype LangfuseEvent = LangfuseEventClient;\ntype LangfuseMetadata = { prompt?: LangfusePromptData };\ntype LangfuseTraceData = TraceData<LangfuseRoot, LangfuseSpan, LangfuseEvent, LangfuseMetadata>;\n\nexport class LangfuseExporter extends TrackingExporter<\n  LangfuseRoot,\n  LangfuseSpan,\n  LangfuseEvent,\n  LangfuseMetadata,\n  LangfuseExporterConfig\n> {\n  name = 'langfuse';\n  #client: Langfuse | undefined;\n  #realtime: boolean;\n\n  constructor(config: LangfuseExporterConfig = {}) {\n    // Resolve env vars BEFORE calling super (config is readonly in base class)\n    const publicKey = config.publicKey ?? process.env.LANGFUSE_PUBLIC_KEY;\n    const secretKey = config.secretKey ?? process.env.LANGFUSE_SECRET_KEY;\n    const baseUrl = config.baseUrl ?? process.env.LANGFUSE_BASE_URL;\n\n    super({\n      ...config,\n      publicKey,\n      secretKey,\n      baseUrl,\n    });\n\n    this.#realtime = config.realtime ?? false;\n\n    if (!publicKey || !secretKey) {\n      const publicKeySource = config.publicKey\n        ? 'from config'\n        : process.env.LANGFUSE_PUBLIC_KEY\n          ? 'from env'\n          : 'missing';\n      const secretKeySource = config.secretKey\n        ? 'from config'\n        : process.env.LANGFUSE_SECRET_KEY\n          ? 'from env'\n          : 'missing';\n      this.setDisabled(\n        `Missing required credentials (publicKey: ${publicKeySource}, secretKey: ${secretKeySource}). ` +\n          `Set LANGFUSE_PUBLIC_KEY and LANGFUSE_SECRET_KEY environment variables or pass them in config.`,\n      );\n      return;\n    }\n\n    this.#client = new Langfuse({\n      publicKey,\n      secretKey,\n      baseUrl,\n      ...config.options,\n    });\n  }\n\n  protected override async _postExportTracingEvent(): Promise<void> {\n    // Flush immediately in realtime mode for instant visibility\n    if (this.#realtime) {\n      await this.#client?.flushAsync();\n    }\n  }\n\n  protected override async _buildRoot(args: {\n    span: AnyExportedSpan;\n    traceData: LangfuseTraceData;\n  }): Promise<LangfuseTraceClient | undefined> {\n    const { span } = args;\n    // Note: If the traceId already exists in Langfuse (e.g., from a previous server instance\n    // or session), the Langfuse SDK handles this gracefully. Calling client.trace() with\n    // an existing ID is idempotent - it will update/continue the existing trace rather than\n    // failing or creating a duplicate. This is by design for distributed tracing scenarios.\n    // See: https://langfuse.com/docs/tracing-features/trace-ids\n    return this.#client?.trace(this.buildTracePayload(span));\n  }\n\n  protected override async _buildEvent(args: {\n    span: AnyExportedSpan;\n    traceData: LangfuseTraceData;\n  }): Promise<LangfuseEvent | undefined> {\n    const { span, traceData } = args;\n    const langfuseParent = traceData.getParentOrRoot({ span });\n    if (!langfuseParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span, true, traceData);\n    return langfuseParent.event(payload);\n  }\n\n  protected override async _buildSpan(args: {\n    span: AnyExportedSpan;\n    traceData: LangfuseTraceData;\n  }): Promise<LangfuseSpan | undefined> {\n    const { span, traceData } = args;\n    const langfuseParent = traceData.getParentOrRoot({ span });\n    if (!langfuseParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span, true, traceData);\n\n    const langfuseSpan =\n      span.type === SpanType.MODEL_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);\n\n    this.logger.debug(`${this.name}: built span`, {\n      traceId: span.traceId,\n      spanId: payload.id,\n      method: '_buildSpan',\n    });\n\n    return langfuseSpan;\n  }\n\n  protected override async _updateSpan(args: { span: AnyExportedSpan; traceData: LangfuseTraceData }): Promise<void> {\n    const { span, traceData } = args;\n    const langfuseSpan = traceData.getSpan({ spanId: span.id });\n    if (langfuseSpan) {\n      this.logger.debug(`${this.name}: found span for update`, {\n        traceId: span.traceId,\n        spanId: langfuseSpan.id,\n        method: '_updateSpan',\n      });\n\n      const updatePayload = this.buildSpanPayload(span, false, traceData);\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(updatePayload);\n    }\n  }\n\n  protected override async _finishSpan(args: { span: AnyExportedSpan; traceData: LangfuseTraceData }): Promise<void> {\n    const { span, traceData } = args;\n    const langfuseSpan = traceData.getSpan({ spanId: span.id });\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, traceData));\n\n    if (span.isRootSpan) {\n      const langfuseRoot = traceData.getRoot();\n      langfuseRoot?.update({ output: span.output });\n    }\n  }\n\n  protected override async _abortSpan(args: { span: LangfuseSpan; reason: SpanErrorInfo }): Promise<void> {\n    const { span, reason } = args;\n    span.end({\n      level: 'ERROR',\n      statusMessage: reason.message,\n    });\n  }\n\n  private buildTracePayload(span: AnyExportedSpan): 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    // Include tags if present (only for root spans, which is always the case here)\n    if (span.tags?.length) payload.tags = span.tags;\n\n    payload.metadata = {\n      spanType: span.type,\n      ...span.attributes,\n      ...remainingMetadata,\n    };\n\n    return payload;\n  }\n\n  /**\n   * Look up the Langfuse prompt from the closest span that has one.\n   * This enables prompt inheritance for MODEL_GENERATION spans when the prompt\n   * is set on a parent span (e.g., AGENT_RUN) rather than directly on the generation.\n   * This enables prompt linking when:\n   * - A workflow calls multiple agents, each with different prompts\n   * - Nested agents have different prompts\n   * - The prompt is set on AGENT_RUN but MODEL_GENERATION inherits it\n   */\n  private findLangfusePrompt(traceData: LangfuseTraceData, span: AnyExportedSpan): LangfusePromptData | undefined {\n    let currentSpanId: string | undefined = span.id;\n\n    while (currentSpanId) {\n      const providerMetadata = traceData.getMetadata({ spanId: currentSpanId });\n\n      if (providerMetadata?.prompt) {\n        this.logger.debug(`${this.name}: found prompt in provider metadata`, {\n          traceId: span.traceId,\n          spanId: span.id,\n          prompt: providerMetadata?.prompt,\n        });\n        return providerMetadata.prompt;\n      }\n      currentSpanId = traceData.getParentId({ spanId: currentSpanId });\n    }\n\n    return undefined;\n  }\n\n  private buildSpanPayload(\n    span: AnyExportedSpan,\n    isCreate: boolean,\n    traceData: LangfuseTraceData,\n  ): 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    }\n\n    if (span.input !== undefined) payload.input = span.input;\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    const metadata: Record<string, any> = {\n      ...span.metadata,\n    };\n\n    // Strip special fields from metadata if used in top-level keys\n    const attributesToOmit: string[] = [];\n    const metadataToOmit: string[] = [];\n\n    if (span.type === SpanType.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        payload.usageDetails = formatUsageMetrics(modelAttr.usage);\n        attributesToOmit.push('usage');\n      }\n\n      if (modelAttr.parameters !== undefined) {\n        payload.modelParameters = modelAttr.parameters;\n        attributesToOmit.push('parameters');\n      }\n\n      // Users can set metadata.langfuse.prompt to link generations to Langfuse Prompt Management\n      // Supported formats:\n      // - { id } - link by prompt UUID alone\n      // - { name, version } - link by name and version\n      // - { name, version, id } - link with all fields\n      const promptData = this.findLangfusePrompt(traceData, span);\n      const hasNameAndVersion = promptData?.name !== undefined && promptData?.version !== undefined;\n      const hasId = promptData?.id !== undefined;\n\n      if (hasNameAndVersion || hasId) {\n        payload.prompt = {};\n\n        if (promptData?.name !== undefined) payload.prompt.name = promptData.name;\n        if (promptData?.version !== undefined) payload.prompt.version = promptData.version;\n        if (promptData?.id !== undefined) payload.prompt.id = promptData.id;\n\n        metadataToOmit.push('langfuse');\n      }\n\n      // completionStartTime is used by Langfuse to calculate time-to-first-token (TTFT)\n      if (modelAttr.completionStartTime !== undefined) {\n        payload.completionStartTime = modelAttr.completionStartTime;\n        attributesToOmit.push('completionStartTime');\n      }\n    }\n\n    payload.metadata = {\n      spanType: span.type,\n      ...omitKeys(attributes, attributesToOmit),\n      ...omitKeys(metadata, metadataToOmit),\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  /**\n   * Force flush any buffered data to Langfuse without shutting down.\n   */\n  protected override async _flush(): Promise<void> {\n    if (this.#client) {\n      await this.#client.flushAsync();\n    }\n  }\n\n  override async _postShutdown(): Promise<void> {\n    if (this.#client) {\n      await this.#client.shutdownAsync();\n    }\n  }\n}\n","/**\n * Langfuse Tracing Options Helpers\n *\n * These helpers integrate with the `buildTracingOptions` pattern from\n * `@mastra/observability` to add Langfuse-specific tracing features.\n *\n * @example\n * ```typescript\n * import { buildTracingOptions } from '@mastra/observability';\n * import { withLangfusePrompt } from '@mastra/langfuse';\n *\n * const prompt = await langfuse.getPrompt('my-prompt');\n *\n * const agent = new Agent({\n *   defaultGenerateOptions: {\n *     tracingOptions: buildTracingOptions(withLangfusePrompt(prompt)),\n *   },\n * });\n * ```\n */\n\nimport type { TracingOptionsUpdater } from '@mastra/observability';\n\n/**\n * Langfuse prompt input - accepts either a Langfuse SDK prompt object\n * or manual fields.\n */\nexport interface LangfusePromptInput {\n  /** Prompt name */\n  name?: string;\n  /** Prompt version */\n  version?: number;\n  /** Prompt UUID */\n  id?: string;\n}\n\n/**\n * Adds Langfuse prompt metadata to the tracing options\n * to enable Langfuse Prompt Tracing.\n *\n * The metadata is added under `metadata.langfuse.prompt` and includes:\n * - `name` - Prompt name\n * - `version` - Prompt version\n * - `id` - Prompt UUID\n *\n * All fields are deeply merged with any existing metadata.\n *\n * @param prompt - A Langfuse prompt object (from `langfuse.getPrompt()`) or manual fields\n * @returns A TracingOptionsUpdater function for use with `buildTracingOptions`\n *\n * @example\n * ```typescript\n * import { buildTracingOptions } from '@mastra/observability';\n * import { withLangfusePrompt } from '@mastra/langfuse';\n * import { Langfuse } from 'langfuse';\n *\n * const langfuse = new Langfuse();\n * const prompt = await langfuse.getPrompt('customer-support');\n *\n * // Use with buildTracingOptions\n * const tracingOptions = buildTracingOptions(\n *   withLangfusePrompt(prompt),\n * );\n *\n * // Or directly in agent config\n * const agent = new Agent({\n *   name: 'support-agent',\n *   instructions: prompt.prompt,\n *   model: openai('gpt-4o'),\n *   defaultGenerateOptions: {\n *     tracingOptions: buildTracingOptions(withLangfusePrompt(prompt)),\n *   },\n * });\n *\n * // Manual fields also work\n * const tracingOptions = buildTracingOptions(\n *   withLangfusePrompt({ name: 'my-prompt', version: 1 }),\n * );\n * ```\n */\nexport function withLangfusePrompt(prompt: LangfusePromptInput): TracingOptionsUpdater {\n  return opts => ({\n    ...opts,\n    metadata: {\n      ...opts.metadata,\n      langfuse: {\n        ...(opts.metadata?.langfuse as Record<string, unknown>),\n        prompt: {\n          ...(prompt.name !== undefined && { name: prompt.name }),\n          ...(prompt.version !== undefined && { version: prompt.version }),\n          ...(prompt.id !== undefined && { id: prompt.id }),\n        },\n      },\n    },\n  });\n}\n"]}

package/dist/index.d.ts

@@ -2,7 +2,8 @@
  * Langfuse Observability Provider for Mastra
  *
  * This package provides Langfuse-specific observability features for Mastra applications.
- * Currently includes AI tracing support with plans for additional observability features.
+ * Currently includes tracing support with plans for additional observability features.
  */
-export * from './ai-tracing.js';
+export * from './tracing.js';
+export * from './helpers.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,cAAc,cAAc,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,cAAc,WAAW,CAAC;AAG1B,cAAc,WAAW,CAAC"}

package/dist/index.js

@@ -1,181 +1,130 @@
-import { AISpanType, omitKeys } from '@mastra/core/ai-tracing';
-import { ConsoleLogger } from '@mastra/core/logger';
+import { SpanType } from '@mastra/core/observability';
+import { omitKeys } from '@mastra/core/utils';
+import { TrackingExporter } from '@mastra/observability';
 import { Langfuse } from 'langfuse';
 
-// src/ai-tracing.ts
-var LangfuseExporter = class {
+// src/tracing.ts
+
+// src/metrics.ts
+function formatUsageMetrics(usage) {
+  if (!usage) return {};
+  const metrics = {};
+  if (usage.inputTokens !== void 0) {
+    metrics.input = usage.inputTokens;
+    if (usage.inputDetails?.cacheWrite !== void 0) {
+      metrics.cache_write_input_tokens = usage.inputDetails.cacheWrite;
+      metrics.input -= metrics.cache_write_input_tokens;
+    }
+  }
+  if (usage.inputDetails?.cacheRead !== void 0) {
+    metrics.cache_read_input_tokens = usage.inputDetails.cacheRead;
+  }
+  if (usage.outputTokens !== void 0) {
+    metrics.output = usage.outputTokens;
+  }
+  if (usage.outputDetails?.reasoning !== void 0) {
+    metrics.reasoning = usage.outputDetails.reasoning;
+  }
+  if (metrics.input != null && metrics.output != null) {
+    metrics.total = metrics.input + metrics.output;
+    if (metrics.cache_write_input_tokens != null) {
+      metrics.total += metrics.cache_write_input_tokens;
+    }
+  }
+  return metrics;
+}
+
+// src/tracing.ts
+var LangfuseExporter = class extends TrackingExporter {
   name = "langfuse";
-  client;
-  realtime;
-  traceMap = /* @__PURE__ */ new Map();
-  logger;
-  constructor(config) {
-    this.realtime = config.realtime ?? false;
-    this.logger = new ConsoleLogger({ level: config.logLevel ?? "warn" });
-    if (!config.publicKey || !config.secretKey) {
-      this.logger.error("LangfuseExporter: Missing required credentials, exporter will be disabled", {
-        hasPublicKey: !!config.publicKey,
-        hasSecretKey: !!config.secretKey
-      });
-      this.client = null;
+  #client;
+  #realtime;
+  constructor(config = {}) {
+    const publicKey = config.publicKey ?? process.env.LANGFUSE_PUBLIC_KEY;
+    const secretKey = config.secretKey ?? process.env.LANGFUSE_SECRET_KEY;
+    const baseUrl = config.baseUrl ?? process.env.LANGFUSE_BASE_URL;
+    super({
+      ...config,
+      publicKey,
+      secretKey,
+      baseUrl
+    });
+    this.#realtime = config.realtime ?? false;
+    if (!publicKey || !secretKey) {
+      const publicKeySource = config.publicKey ? "from config" : process.env.LANGFUSE_PUBLIC_KEY ? "from env" : "missing";
+      const secretKeySource = config.secretKey ? "from config" : process.env.LANGFUSE_SECRET_KEY ? "from env" : "missing";
+      this.setDisabled(
+        `Missing required credentials (publicKey: ${publicKeySource}, secretKey: ${secretKeySource}). Set LANGFUSE_PUBLIC_KEY and LANGFUSE_SECRET_KEY environment variables or pass them in config.`
+      );
       return;
     }
-    this.client = new Langfuse({
-      publicKey: config.publicKey,
-      secretKey: config.secretKey,
-      baseUrl: config.baseUrl,
+    this.#client = new Langfuse({
+      publicKey,
+      secretKey,
+      baseUrl,
       ...config.options
     });
   }
-  async exportEvent(event) {
-    if (!this.client) {
-      return;
-    }
-    if (event.exportedSpan.isEvent) {
-      await this.handleEventSpan(event.exportedSpan);
-      return;
-    }
-    switch (event.type) {
-      case "span_started":
-        await this.handleSpanStarted(event.exportedSpan);
-        break;
-      case "span_updated":
-        await this.handleSpanUpdateOrEnd(event.exportedSpan, false);
-        break;
-      case "span_ended":
-        await this.handleSpanUpdateOrEnd(event.exportedSpan, true);
-        break;
-    }
-    if (this.realtime) {
-      await this.client.flushAsync();
+  async _postExportTracingEvent() {
+    if (this.#realtime) {
+      await this.#client?.flushAsync();
     }
   }
-  async handleSpanStarted(span) {
-    if (span.isRootSpan) {
-      this.initTrace(span);
-    }
-    const method = "handleSpanStarted";
-    const traceData = this.getTraceData({ span, method });
-    if (!traceData) {
-      return;
-    }
-    const langfuseParent = this.getLangfuseParent({ traceData, span, method });
+  async _buildRoot(args) {
+    const { span } = args;
+    return this.#client?.trace(this.buildTracePayload(span));
+  }
+  async _buildEvent(args) {
+    const { span, traceData } = args;
+    const langfuseParent = traceData.getParentOrRoot({ span });
     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);
-    traceData.activeSpans.add(span.id);
+    const payload = this.buildSpanPayload(span, true, traceData);
+    return langfuseParent.event(payload);
   }
-  async handleSpanUpdateOrEnd(span, isEnd) {
-    const method = isEnd ? "handleSpanEnd" : "handleSpanUpdate";
-    const traceData = this.getTraceData({ span, method });
-    if (!traceData) {
-      return;
-    }
-    const langfuseSpan = traceData.spans.get(span.id);
-    if (!langfuseSpan) {
-      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
-      });
+  async _buildSpan(args) {
+    const { span, traceData } = args;
+    const langfuseParent = traceData.getParentOrRoot({ span });
+    if (!langfuseParent) {
       return;
     }
-    langfuseSpan.update(this.buildSpanPayload(span, false));
-    if (isEnd) {
-      traceData.activeSpans.delete(span.id);
-      if (span.isRootSpan) {
-        traceData.trace.update({ output: span.output });
-      }
-      if (traceData.activeSpans.size === 0) {
-        this.traceMap.delete(span.traceId);
-      }
-    }
+    const payload = this.buildSpanPayload(span, true, traceData);
+    const langfuseSpan = span.type === SpanType.MODEL_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);
+    this.logger.debug(`${this.name}: built span`, {
+      traceId: span.traceId,
+      spanId: payload.id,
+      method: "_buildSpan"
+    });
+    return langfuseSpan;
   }
-  async handleEventSpan(span) {
-    if (span.isRootSpan) {
-      this.logger.debug("Langfuse exporter: Creating trace", {
+  async _updateSpan(args) {
+    const { span, traceData } = args;
+    const langfuseSpan = traceData.getSpan({ spanId: span.id });
+    if (langfuseSpan) {
+      this.logger.debug(`${this.name}: found span for update`, {
         traceId: span.traceId,
-        spanId: span.id,
-        spanName: span.name,
-        method: "handleEventSpan"
+        spanId: langfuseSpan.id,
+        method: "_updateSpan"
       });
-      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 updatePayload = this.buildSpanPayload(span, false, traceData);
+      langfuseSpan.update(updatePayload);
     }
-    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);
+  async _finishSpan(args) {
+    const { span, traceData } = args;
+    const langfuseSpan = traceData.getSpan({ spanId: span.id });
+    langfuseSpan?.update(this.buildSpanPayload(span, false, traceData));
+    if (span.isRootSpan) {
+      const langfuseRoot = traceData.getRoot();
+      langfuseRoot?.update({ output: span.output });
     }
-    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
+  async _abortSpan(args) {
+    const { span, reason } = args;
+    span.end({
+      level: "ERROR",
+      statusMessage: reason.message
     });
   }
   buildTracePayload(span) {
@@ -187,6 +136,7 @@ var LangfuseExporter = class {
     if (userId) payload.userId = userId;
     if (sessionId) payload.sessionId = sessionId;
     if (span.input) payload.input = span.input;
+    if (span.tags?.length) payload.tags = span.tags;
     payload.metadata = {
       spanType: span.type,
       ...span.attributes,
@@ -194,37 +144,80 @@ var LangfuseExporter = class {
     };
     return payload;
   }
-  buildSpanPayload(span, isCreate) {
+  /**
+   * Look up the Langfuse prompt from the closest span that has one.
+   * This enables prompt inheritance for MODEL_GENERATION spans when the prompt
+   * is set on a parent span (e.g., AGENT_RUN) rather than directly on the generation.
+   * This enables prompt linking when:
+   * - A workflow calls multiple agents, each with different prompts
+   * - Nested agents have different prompts
+   * - The prompt is set on AGENT_RUN but MODEL_GENERATION inherits it
+   */
+  findLangfusePrompt(traceData, span) {
+    let currentSpanId = span.id;
+    while (currentSpanId) {
+      const providerMetadata = traceData.getMetadata({ spanId: currentSpanId });
+      if (providerMetadata?.prompt) {
+        this.logger.debug(`${this.name}: found prompt in provider metadata`, {
+          traceId: span.traceId,
+          spanId: span.id,
+          prompt: providerMetadata?.prompt
+        });
+        return providerMetadata.prompt;
+      }
+      currentSpanId = traceData.getParentId({ spanId: currentSpanId });
+    }
+    return void 0;
+  }
+  buildSpanPayload(span, isCreate, traceData) {
     const payload = {};
     if (isCreate) {
       payload.id = span.id;
       payload.name = span.name;
       payload.startTime = span.startTime;
-      if (span.input !== void 0) payload.input = span.input;
     }
+    if (span.input !== void 0) payload.input = span.input;
     if (span.output !== void 0) payload.output = span.output;
     if (span.endTime !== void 0) payload.endTime = span.endTime;
     const attributes = span.attributes ?? {};
+    const metadata = {
+      ...span.metadata
+    };
     const attributesToOmit = [];
-    if (span.type === AISpanType.LLM_GENERATION) {
-      const llmAttr = attributes;
-      if (llmAttr.model !== void 0) {
-        payload.model = llmAttr.model;
+    const metadataToOmit = [];
+    if (span.type === SpanType.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) {
+        payload.usageDetails = formatUsageMetrics(modelAttr.usage);
         attributesToOmit.push("usage");
       }
-      if (llmAttr.parameters !== void 0) {
-        payload.modelParameters = llmAttr.parameters;
+      if (modelAttr.parameters !== void 0) {
+        payload.modelParameters = modelAttr.parameters;
         attributesToOmit.push("parameters");
       }
+      const promptData = this.findLangfusePrompt(traceData, span);
+      const hasNameAndVersion = promptData?.name !== void 0 && promptData?.version !== void 0;
+      const hasId = promptData?.id !== void 0;
+      if (hasNameAndVersion || hasId) {
+        payload.prompt = {};
+        if (promptData?.name !== void 0) payload.prompt.name = promptData.name;
+        if (promptData?.version !== void 0) payload.prompt.version = promptData.version;
+        if (promptData?.id !== void 0) payload.prompt.id = promptData.id;
+        metadataToOmit.push("langfuse");
+      }
+      if (modelAttr.completionStartTime !== void 0) {
+        payload.completionStartTime = modelAttr.completionStartTime;
+        attributesToOmit.push("completionStartTime");
+      }
     }
     payload.metadata = {
       spanType: span.type,
       ...omitKeys(attributes, attributesToOmit),
-      ...span.metadata
+      ...omitKeys(metadata, metadataToOmit)
     };
     if (span.errorInfo) {
       payload.level = "ERROR";
@@ -240,9 +233,9 @@ var LangfuseExporter = class {
     scorerName,
     metadata
   }) {
-    if (!this.client) return;
+    if (!this.#client) return;
     try {
-      await this.client.score({
+      await this.#client.score({
         id: `${traceId}-${scorerName}`,
         traceId,
         observationId: spanId,
@@ -261,14 +254,39 @@ var LangfuseExporter = class {
       });
     }
   }
-  async shutdown() {
-    if (this.client) {
-      await this.client.shutdownAsync();
+  /**
+   * Force flush any buffered data to Langfuse without shutting down.
+   */
+  async _flush() {
+    if (this.#client) {
+      await this.#client.flushAsync();
+    }
+  }
+  async _postShutdown() {
+    if (this.#client) {
+      await this.#client.shutdownAsync();
     }
-    this.traceMap.clear();
   }
 };
 
-export { LangfuseExporter };
+// src/helpers.ts
+function withLangfusePrompt(prompt) {
+  return (opts) => ({
+    ...opts,
+    metadata: {
+      ...opts.metadata,
+      langfuse: {
+        ...opts.metadata?.langfuse,
+        prompt: {
+          ...prompt.name !== void 0 && { name: prompt.name },
+          ...prompt.version !== void 0 && { version: prompt.version },
+          ...prompt.id !== void 0 && { id: prompt.id }
+        }
+      }
+    }
+  });
+}
+
+export { LangfuseExporter, withLangfusePrompt };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map