@mastra/braintrust 0.0.0-pgvector-index-fix-20250905222058 → 0.0.0-playground-studio-again-20251114100107

package/dist/index.cjs

@@ -1,77 +1,131 @@
 'use strict';
 
-var aiTracing = require('@mastra/core/ai-tracing');
-var logger = require('@mastra/core/logger');
+var observability$1 = require('@mastra/core/observability');
+var utils = require('@mastra/core/utils');
+var observability = require('@mastra/observability');
 var braintrust = require('braintrust');
 
-// src/ai-tracing.ts
+// src/tracing.ts
+
+// src/metrics.ts
+function normalizeUsageMetrics(modelAttr) {
+  const metrics = {};
+  if (modelAttr.usage?.inputTokens !== void 0) {
+    metrics.prompt_tokens = modelAttr.usage?.inputTokens;
+  } else if (modelAttr.usage?.promptTokens !== void 0) {
+    metrics.prompt_tokens = modelAttr.usage?.promptTokens;
+  }
+  if (modelAttr.usage?.outputTokens !== void 0) {
+    metrics.completion_tokens = modelAttr.usage?.outputTokens;
+  } else if (modelAttr.usage?.completionTokens !== void 0) {
+    metrics.completion_tokens = modelAttr.usage?.completionTokens;
+  }
+  if (modelAttr.usage?.totalTokens !== void 0) {
+    metrics.tokens = modelAttr.usage?.totalTokens;
+  }
+  if (modelAttr.usage?.reasoningTokens !== void 0) {
+    metrics.completion_reasoning_tokens = modelAttr.usage?.reasoningTokens;
+  }
+  if (modelAttr.usage?.promptCacheHitTokens !== void 0) {
+    metrics.prompt_cached_tokens = modelAttr.usage?.promptCacheHitTokens;
+  }
+  if (modelAttr.usage?.promptCacheMissTokens !== void 0) {
+    metrics.prompt_cache_creation_tokens = modelAttr.usage?.promptCacheMissTokens;
+  }
+  return metrics;
+}
+
+// src/tracing.ts
+var MASTRA_TRACE_ID_METADATA_KEY = "mastra-trace-id";
 var DEFAULT_SPAN_TYPE = "task";
 var SPAN_TYPE_EXCEPTIONS = {
-  [aiTracing.AISpanType.LLM_GENERATION]: "llm",
-  [aiTracing.AISpanType.LLM_CHUNK]: "llm",
-  [aiTracing.AISpanType.TOOL_CALL]: "tool",
-  [aiTracing.AISpanType.MCP_TOOL_CALL]: "tool",
-  [aiTracing.AISpanType.WORKFLOW_CONDITIONAL_EVAL]: "function",
-  [aiTracing.AISpanType.WORKFLOW_WAIT_EVENT]: "function"
+  [observability$1.SpanType.MODEL_GENERATION]: "llm",
+  [observability$1.SpanType.MODEL_CHUNK]: "llm",
+  [observability$1.SpanType.TOOL_CALL]: "tool",
+  [observability$1.SpanType.MCP_TOOL_CALL]: "tool",
+  [observability$1.SpanType.WORKFLOW_CONDITIONAL_EVAL]: "function",
+  [observability$1.SpanType.WORKFLOW_WAIT_EVENT]: "function"
 };
 function mapSpanType(spanType) {
   return SPAN_TYPE_EXCEPTIONS[spanType] ?? DEFAULT_SPAN_TYPE;
 }
-var BraintrustExporter = class {
+var BraintrustExporter = class extends observability.BaseExporter {
   name = "braintrust";
   traceMap = /* @__PURE__ */ new Map();
-  logger;
   config;
+  // Flags and logger for context-aware mode
+  useProvidedLogger;
+  providedLogger;
   constructor(config) {
-    this.logger = new logger.ConsoleLogger({ level: config.logLevel ?? "warn" });
-    if (!config.apiKey) {
-      this.logger.error("BraintrustExporter: Missing required credentials, exporter will be disabled", {
-        hasApiKey: !!config.apiKey
-      });
-      this.config = null;
-      return;
+    super(config);
+    if (config.braintrustLogger) {
+      this.useProvidedLogger = true;
+      this.providedLogger = config.braintrustLogger;
+      this.config = config;
+    } else {
+      if (!config.apiKey) {
+        this.setDisabled(`Missing required credentials (apiKey: ${!!config.apiKey})`);
+        this.config = null;
+        this.useProvidedLogger = false;
+        return;
+      }
+      this.useProvidedLogger = false;
+      this.config = config;
     }
-    this.config = config;
   }
-  async exportEvent(event) {
-    if (!this.config) {
-      return;
-    }
-    if (event.span.isEvent) {
-      await this.handleEventSpan(event.span);
+  async _exportTracingEvent(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, false);
+        await this.handleSpanUpdateOrEnd(event.exportedSpan, false);
         break;
       case "span_ended":
-        await this.handleSpanUpdateOrEnd(event.span, true);
+        await this.handleSpanUpdateOrEnd(event.exportedSpan, true);
         break;
     }
   }
   async handleSpanStarted(span) {
     if (span.isRootSpan) {
-      await this.initLogger(span);
+      if (this.useProvidedLogger) {
+        await this.initLoggerOrUseContext(span);
+      } else {
+        await this.initLoggerPerTrace(span);
+      }
     }
     const method = "handleSpanStarted";
     const spanData = this.getSpanData({ span, method });
     if (!spanData) {
       return;
     }
+    if (!span.isEvent) {
+      spanData.activeIds.add(span.id);
+    }
     const braintrustParent = this.getBraintrustParent({ spanData, span, method });
     if (!braintrustParent) {
       return;
     }
     const payload = this.buildSpanPayload(span);
+    const shouldOmitParentIds = spanData.isExternal && !span.parentSpanId;
     const braintrustSpan = braintrustParent.startSpan({
+      spanId: span.id,
       name: span.name,
       type: mapSpanType(span.type),
+      ...shouldOmitParentIds ? {} : {
+        parentSpanIds: span.parentSpanId ? { spanId: span.parentSpanId, rootSpanId: span.traceId } : { spanId: span.traceId, rootSpanId: span.traceId }
+      },
       ...payload
     });
+    braintrustSpan.log({
+      metadata: {
+        [MASTRA_TRACE_ID_METADATA_KEY]: span.traceId
+      }
+    });
     spanData.spans.set(span.id, braintrustSpan);
   }
   async handleSpanUpdateOrEnd(span, isEnd) {
@@ -88,7 +142,7 @@ var BraintrustExporter = class {
         spanName: span.name,
         spanType: span.type,
         isRootSpan: span.isRootSpan,
-        parentSpanId: span.parent?.id,
+        parentSpanId: span.parentSpanId,
         method
       });
       return;
@@ -100,7 +154,10 @@ var BraintrustExporter = class {
       } else {
         braintrustSpan.end();
       }
-      if (span.isRootSpan) {
+      if (!span.isEvent) {
+        spanData.activeIds.delete(span.id);
+      }
+      if (spanData.activeIds.size === 0 && !spanData.isExternal) {
         this.traceMap.delete(span.traceId);
       }
     }
@@ -113,7 +170,11 @@ var BraintrustExporter = class {
         spanName: span.name,
         method: "handleEventSpan"
       });
-      await this.initLogger(span);
+      if (this.useProvidedLogger) {
+        await this.initLoggerOrUseContext(span);
+      } else {
+        await this.initLoggerPerTrace(span);
+      }
     }
     const method = "handleEventSpan";
     const spanData = this.getSpanData({ span, method });
@@ -126,23 +187,54 @@ var BraintrustExporter = class {
     }
     const payload = this.buildSpanPayload(span);
     const braintrustSpan = braintrustParent.startSpan({
+      spanId: span.id,
       name: span.name,
       type: mapSpanType(span.type),
+      parentSpanIds: span.parentSpanId ? { spanId: span.parentSpanId, rootSpanId: span.traceId } : { spanId: span.traceId, rootSpanId: span.traceId },
       startTime: span.startTime.getTime() / 1e3,
       ...payload
     });
     braintrustSpan.end({ endTime: span.startTime.getTime() / 1e3 });
-    spanData.spans.set(span.id, braintrustSpan);
   }
-  async initLogger(span) {
+  /**
+   * Creates a new logger per trace using config credentials
+   */
+  async initLoggerPerTrace(span) {
     const logger = await braintrust.initLogger({
-      projectName: "mastra-tracing",
-      // TODO: Make this configurable
+      projectName: this.config.projectName ?? "mastra-tracing",
       apiKey: this.config.apiKey,
       appUrl: this.config.endpoint,
       ...this.config.tuningParameters
     });
-    this.traceMap.set(span.traceId, { logger, spans: /* @__PURE__ */ new Map() });
+    this.traceMap.set(span.traceId, {
+      logger,
+      spans: /* @__PURE__ */ new Map(),
+      activeIds: /* @__PURE__ */ new Set(),
+      isExternal: false
+    });
+  }
+  /**
+   * Uses provided logger and detects external Braintrust spans.
+   * If a Braintrust span is detected (from logger.traced() or Eval()), attaches to it.
+   * Otherwise, uses the provided logger instance.
+   */
+  async initLoggerOrUseContext(span) {
+    const braintrustSpan = braintrust.currentSpan();
+    if (braintrustSpan && braintrustSpan.id) {
+      this.traceMap.set(span.traceId, {
+        logger: braintrustSpan,
+        spans: /* @__PURE__ */ new Map(),
+        activeIds: /* @__PURE__ */ new Set(),
+        isExternal: true
+      });
+    } else {
+      this.traceMap.set(span.traceId, {
+        logger: this.providedLogger,
+        spans: /* @__PURE__ */ new Map(),
+        activeIds: /* @__PURE__ */ new Set(),
+        isExternal: false
+      });
+    }
   }
   getSpanData(options) {
     const { span, method } = options;
@@ -155,26 +247,29 @@ var BraintrustExporter = class {
       spanName: span.name,
       spanType: span.type,
       isRootSpan: span.isRootSpan,
-      parentSpanId: span.parent?.id,
+      parentSpanId: span.parentSpanId,
       method
     });
   }
   getBraintrustParent(options) {
     const { spanData, span, method } = options;
-    const parentId = span.parent?.id;
+    const parentId = span.parentSpanId;
     if (!parentId) {
       return spanData.logger;
     }
     if (spanData.spans.has(parentId)) {
       return spanData.spans.get(parentId);
     }
+    if (parentId && !spanData.spans.has(parentId)) {
+      return spanData.logger;
+    }
     this.logger.warn("Braintrust exporter: No parent data found for span", {
       traceId: span.traceId,
       spanId: span.id,
       spanName: span.name,
       spanType: span.type,
       isRootSpan: span.isRootSpan,
-      parentSpanId: span.parent?.id,
+      parentSpanId: span.parentSpanId,
       method
     });
   }
@@ -192,21 +287,19 @@ var BraintrustExporter = class {
       ...span.metadata
     };
     const attributes = span.attributes ?? {};
-    if (span.type === aiTracing.AISpanType.LLM_GENERATION) {
-      const llmAttr = attributes;
-      if (llmAttr.model !== void 0) {
-        payload.metadata.model = llmAttr.model;
+    if (span.type === observability$1.SpanType.MODEL_GENERATION) {
+      const modelAttr = attributes;
+      if (modelAttr.model !== void 0) {
+        payload.metadata.model = modelAttr.model;
       }
-      if (llmAttr.usage !== void 0) {
-        payload.metrics = {
-          ...payload.metrics,
-          ...llmAttr.usage
-        };
+      if (modelAttr.provider !== void 0) {
+        payload.metadata.provider = modelAttr.provider;
       }
-      if (llmAttr.parameters !== void 0) {
-        payload.metadata.modelParameters = llmAttr.parameters;
+      payload.metrics = normalizeUsageMetrics(modelAttr);
+      if (modelAttr.parameters !== void 0) {
+        payload.metadata.modelParameters = modelAttr.parameters;
       }
-      const otherAttributes = aiTracing.omitKeys(attributes, ["model", "usage", "parameters"]);
+      const otherAttributes = utils.omitKeys(attributes, ["model", "usage", "parameters"]);
       payload.metadata = {
         ...payload.metadata,
         ...otherAttributes
@@ -236,6 +329,7 @@ var BraintrustExporter = class {
       }
     }
     this.traceMap.clear();
+    await super.shutdown();
   }
 };
 

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/ai-tracing.ts"],"names":["AISpanType","ConsoleLogger","initLogger","omitKeys"],"mappings":";;;;;;;AA+BA,IAAM,iBAAA,GAAoB,MAAA;AAG1B,IAAM,oBAAA,GAA4D;AAAA,EAChE,CAACA,oBAAA,CAAW,cAAc,GAAG,KAAA;AAAA,EAC7B,CAACA,oBAAA,CAAW,SAAS,GAAG,KAAA;AAAA,EACxB,CAACA,oBAAA,CAAW,SAAS,GAAG,MAAA;AAAA,EACxB,CAACA,oBAAA,CAAW,aAAa,GAAG,MAAA;AAAA,EAC5B,CAACA,oBAAA,CAAW,yBAAyB,GAAG,UAAA;AAAA,EACxC,CAACA,oBAAA,CAAW,mBAAmB,GAAG;AACpC,CAAA;AAGA,SAAS,YAAY,QAAA,EAA+E;AAClG,EAAA,OAAQ,oBAAA,CAAqB,QAAQ,CAAA,IAAa,iBAAA;AACpD;AAEO,IAAM,qBAAN,MAAsD;AAAA,EAC3D,IAAA,GAAO,YAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA,EACA,MAAA;AAAA,EAER,YAAY,MAAA,EAAkC;AAC5C,IAAA,IAAA,CAAK,MAAA,GAAS,IAAIC,oBAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AAEpE,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,6EAAA,EAA+E;AAAA,QAC/F,SAAA,EAAW,CAAC,CAAC,MAAA,CAAO;AAAA,OACrB,CAAA;AACD,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAChB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,KAAA,CAAM,KAAK,OAAA,EAAS;AACtB,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,IAAI,CAAA;AACrC,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA;AACJ,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAM,IAAA,CAAK,WAAW,IAAI,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,MAAA,GAAS,mBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAE1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,KAAA,EAA+B;AAClF,IAAA,MAAM,MAAA,GAAS,QAAQ,eAAA,GAAkB,kBAAA;AAEzC,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,cAAA,GAAiB,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,KAAK,mEAAA,EAAqE;AAAA,QACpF,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,YAAY,IAAA,CAAK,UAAA;AAAA,QACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,QAC3B;AAAA,OACD,CAAA;AACD,MAAA;AAAA,IACF;AAEA,IAAA,cAAA,CAAe,GAAA,CAAI,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAC,CAAA;AAE9C,IAAA,IAAI,KAAA,EAAO;AAET,MAAA,IAAI,KAAK,OAAA,EAAS;AAChB,QAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,QAAQ,OAAA,EAAQ,GAAI,KAAM,CAAA;AAAA,MAC/D,CAAA,MAAO;AACL,QAAA,cAAA,CAAe,GAAA,EAAI;AAAA,MACrB;AAEA,MAAA,IAAI,KAAK,UAAA,EAAY;AACnB,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,MACnC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAgC;AAC5D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,gDAAA,EAAkD;AAAA,QAClE,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,MAAM,IAAA,CAAK,WAAW,IAAI,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,MAAA,GAAS,iBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAG1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,SAAA,EAAW,IAAA,CAAK,SAAA,CAAU,OAAA,EAAQ,GAAI,GAAA;AAAA,MACtC,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,UAAU,OAAA,EAAQ,GAAI,KAAM,CAAA;AAC/D,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,WAAW,IAAA,EAAgC;AACvD,IAAA,MAAM,MAAA,GAAS,MAAMC,qBAAA,CAAW;AAAA,MAC9B,WAAA,EAAa,gBAAA;AAAA;AAAA,MACb,MAAA,EAAQ,KAAK,MAAA,CAAO,MAAA;AAAA,MACpB,MAAA,EAAQ,KAAK,MAAA,CAAO,QAAA;AAAA,MACpB,GAAG,KAAK,MAAA,CAAO;AAAA,KAChB,CAAA;AAED,IAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS,EAAE,QAAQ,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,CAAA;AAAA,EAC9D;AAAA,EAEQ,YAAY,OAAA,EAAoE;AACtF,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AACzB,IAAA,IAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA,EAAG;AACnC,MAAA,OAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACvC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,kDAAA,EAAoD;AAAA,MACnE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,oBAAoB,OAAA,EAIQ;AAClC,IAAA,MAAM,EAAE,QAAA,EAAU,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AAEnC,IAAA,MAAM,QAAA,GAAW,KAAK,MAAA,EAAQ,EAAA;AAC9B,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,QAAA,CAAS,MAAA;AAAA,IAClB;AAEA,IAAA,IAAI,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA,EAAG;AAChC,MAAA,OAAO,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA;AAAA,IACpC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,oDAAA,EAAsD;AAAA,MACrE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,iBAAiB,IAAA,EAAsC;AAC7D,IAAA,MAAM,UAA+B,EAAC;AAGtC,IAAA,IAAI,IAAA,CAAK,UAAU,MAAA,EAAW;AAC5B,MAAA,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACvB;AAEA,IAAA,IAAI,IAAA,CAAK,WAAW,MAAA,EAAW;AAC7B,MAAA,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AAAA,IACxB;AAGA,IAAA,OAAA,CAAQ,UAAU,EAAC;AACnB,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,IAAA,CAAK;AAAA,KACV;AAEA,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAExC,IAAA,IAAI,IAAA,CAAK,IAAA,KAASF,oBAAA,CAAW,cAAA,EAAgB;AAC3C,MAAA,MAAM,OAAA,GAAU,UAAA;AAGhB,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAA,CAAS,QAAQ,OAAA,CAAQ,KAAA;AAAA,MACnC;AAGA,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,OAAA,GAAU;AAAA,UAChB,GAAG,OAAA,CAAQ,OAAA;AAAA,UACX,GAAG,OAAA,CAAQ;AAAA,SACb;AAAA,MACF;AAGA,MAAA,IAAI,OAAA,CAAQ,eAAe,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,QAAA,CAAS,kBAAkB,OAAA,CAAQ,UAAA;AAAA,MAC7C;AAGA,MAAA,MAAM,kBAAkBG,kBAAA,CAAS,UAAA,EAAY,CAAC,OAAA,EAAS,OAAA,EAAS,YAAY,CAAC,CAAA;AAC7E,MAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,QACjB,GAAG,OAAA,CAAQ,QAAA;AAAA,QACX,GAAG;AAAA,OACL;AAAA,IACF,CAAA,MAAO;AAEL,MAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,QACjB,GAAG,OAAA,CAAQ,QAAA;AAAA,QACX,GAAG;AAAA,OACL;AAAA,IACF;AAGA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,OAAA,CAAQ,KAAA,GAAQ,KAAK,SAAA,CAAU,OAAA;AAC/B,MAAA,OAAA,CAAQ,QAAA,CAAS,eAAe,IAAA,CAAK,SAAA;AAAA,IACvC;AAGA,IAAA,IAAI,OAAO,IAAA,CAAK,OAAA,CAAQ,OAAO,CAAA,CAAE,WAAW,CAAA,EAAG;AAC7C,MAAA,OAAO,OAAA,CAAQ,OAAA;AAAA,IACjB;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAChB,MAAA;AAAA,IACF;AAGA,IAAA,KAAA,MAAW,CAAC,QAAA,EAAU,QAAQ,CAAA,IAAK,KAAK,QAAA,EAAU;AAChD,MAAA,KAAA,MAAW,CAAC,OAAA,EAAS,IAAI,CAAA,IAAK,SAAS,KAAA,EAAO;AAC5C,QAAA,IAAA,CAAK,GAAA,EAAI;AAAA,MACX;AAAA,IAEF;AACA,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AAAA,EACtB;AACF","file":"index.cjs","sourcesContent":["/**\n * Braintrust Exporter for Mastra AI Tracing\n *\n * This exporter sends tracing data to Braintrust for AI observability.\n * Root spans become top-level Braintrust spans (no trace wrapper).\n * Events are handled as zero-duration spans with matching start/end times.\n */\n\nimport type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, omitKeys } from '@mastra/core/ai-tracing';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { initLogger } from 'braintrust';\nimport type { Span, Logger } from 'braintrust';\n\nexport interface BraintrustExporterConfig {\n  /** Braintrust API key */\n  apiKey?: string;\n  /** Optional custom endpoint */\n  endpoint?: string;\n  /** Logger level for diagnostic messages (default: 'warn') */\n  logLevel?: 'debug' | 'info' | 'warn' | 'error';\n  /** Support tuning parameters */\n  tuningParameters?: Record<string, any>;\n}\n\ntype SpanData = {\n  logger: Logger<true>; // Braintrust logger (for root spans)\n  spans: Map<string, Span>; // Maps span.id to Braintrust span\n};\n\n// Default span type for all spans\nconst DEFAULT_SPAN_TYPE = 'task';\n\n// Exceptions to the default mapping\nconst SPAN_TYPE_EXCEPTIONS: Partial<Record<AISpanType, string>> = {\n  [AISpanType.LLM_GENERATION]: 'llm',\n  [AISpanType.LLM_CHUNK]: 'llm',\n  [AISpanType.TOOL_CALL]: 'tool',\n  [AISpanType.MCP_TOOL_CALL]: 'tool',\n  [AISpanType.WORKFLOW_CONDITIONAL_EVAL]: 'function',\n  [AISpanType.WORKFLOW_WAIT_EVENT]: 'function',\n};\n\n// Mapping function - returns valid Braintrust span types\nfunction mapSpanType(spanType: AISpanType): 'llm' | 'score' | 'function' | 'eval' | 'task' | 'tool' {\n  return (SPAN_TYPE_EXCEPTIONS[spanType] as any) ?? DEFAULT_SPAN_TYPE;\n}\n\nexport class BraintrustExporter implements AITracingExporter {\n  name = 'braintrust';\n  private traceMap = new Map<string, SpanData>();\n  private logger: ConsoleLogger;\n  private config: BraintrustExporterConfig;\n\n  constructor(config: BraintrustExporterConfig) {\n    this.logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n\n    if (!config.apiKey) {\n      this.logger.error('BraintrustExporter: Missing required credentials, exporter will be disabled', {\n        hasApiKey: !!config.apiKey,\n      });\n      this.config = null as any;\n      return;\n    }\n\n    this.config = config;\n  }\n\n  async exportEvent(event: AITracingEvent): Promise<void> {\n    if (!this.config) {\n      return;\n    }\n\n    if (event.span.isEvent) {\n      await this.handleEventSpan(event.span);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.span);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.span, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.span, true);\n        break;\n    }\n  }\n\n  private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      await this.initLogger(span);\n    }\n\n    const method = 'handleSpanStarted';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    const braintrustSpan = braintrustParent.startSpan({\n      name: span.name,\n      type: mapSpanType(span.type),\n      ...payload,\n    });\n\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyAISpan, isEnd: boolean): Promise<void> {\n    const method = isEnd ? 'handleSpanEnd' : 'handleSpanUpdate';\n\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustSpan = spanData.spans.get(span.id);\n    if (!braintrustSpan) {\n      this.logger.warn('Braintrust exporter: No Braintrust span found for span update/end', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        spanType: span.type,\n        isRootSpan: span.isRootSpan,\n        parentSpanId: span.parent?.id,\n        method,\n      });\n      return;\n    }\n\n    braintrustSpan.log(this.buildSpanPayload(span));\n\n    if (isEnd) {\n      // End the span with the correct endTime (convert milliseconds to seconds)\n      if (span.endTime) {\n        braintrustSpan.end({ endTime: span.endTime.getTime() / 1000 });\n      } else {\n        braintrustSpan.end();\n      }\n\n      if (span.isRootSpan) {\n        this.traceMap.delete(span.traceId);\n      }\n    }\n  }\n\n  private async handleEventSpan(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.logger.debug('Braintrust exporter: Creating logger for event', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        method: 'handleEventSpan',\n      });\n      await this.initLogger(span);\n    }\n\n    const method = 'handleEventSpan';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    // Create zero-duration span for event (convert milliseconds to seconds)\n    const braintrustSpan = braintrustParent.startSpan({\n      name: span.name,\n      type: mapSpanType(span.type),\n      startTime: span.startTime.getTime() / 1000,\n      ...payload,\n    });\n\n    braintrustSpan.end({ endTime: span.startTime.getTime() / 1000 });\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async initLogger(span: AnyAISpan): Promise<void> {\n    const logger = await initLogger({\n      projectName: 'mastra-tracing', // TODO: Make this configurable\n      apiKey: this.config.apiKey,\n      appUrl: this.config.endpoint,\n      ...this.config.tuningParameters,\n    });\n\n    this.traceMap.set(span.traceId, { logger, spans: new Map() });\n  }\n\n  private getSpanData(options: { span: AnyAISpan; method: string }): SpanData | undefined {\n    const { span, method } = options;\n    if (this.traceMap.has(span.traceId)) {\n      return this.traceMap.get(span.traceId);\n    }\n\n    this.logger.warn('Braintrust exporter: No span data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private getBraintrustParent(options: {\n    spanData: SpanData;\n    span: AnyAISpan;\n    method: string;\n  }): Logger<true> | Span | undefined {\n    const { spanData, span, method } = options;\n\n    const parentId = span.parent?.id;\n    if (!parentId) {\n      return spanData.logger;\n    }\n\n    if (spanData.spans.has(parentId)) {\n      return spanData.spans.get(parentId);\n    }\n\n    this.logger.warn('Braintrust exporter: No parent data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private buildSpanPayload(span: AnyAISpan): Record<string, any> {\n    const payload: Record<string, any> = {};\n\n    // Core span data\n    if (span.input !== undefined) {\n      payload.input = span.input;\n    }\n\n    if (span.output !== undefined) {\n      payload.output = span.output;\n    }\n\n    // Initialize metrics and metadata objects\n    payload.metrics = {};\n    payload.metadata = {\n      spanType: span.type,\n      ...span.metadata,\n    };\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    if (span.type === AISpanType.LLM_GENERATION) {\n      const llmAttr = attributes as LLMGenerationAttributes;\n\n      // Model goes to metadata\n      if (llmAttr.model !== undefined) {\n        payload.metadata.model = llmAttr.model;\n      }\n\n      // Usage/token info goes to metrics\n      if (llmAttr.usage !== undefined) {\n        payload.metrics = {\n          ...payload.metrics,\n          ...llmAttr.usage,\n        };\n      }\n\n      // Model parameters go to metadata\n      if (llmAttr.parameters !== undefined) {\n        payload.metadata.modelParameters = llmAttr.parameters;\n      }\n\n      // Other LLM attributes go to metadata\n      const otherAttributes = omitKeys(attributes, ['model', 'usage', 'parameters']);\n      payload.metadata = {\n        ...payload.metadata,\n        ...otherAttributes,\n      };\n    } else {\n      // For non-LLM spans, put all attributes in metadata\n      payload.metadata = {\n        ...payload.metadata,\n        ...attributes,\n      };\n    }\n\n    // Handle errors\n    if (span.errorInfo) {\n      payload.error = span.errorInfo.message;\n      payload.metadata.errorDetails = span.errorInfo;\n    }\n\n    // Clean up empty metrics object\n    if (Object.keys(payload.metrics).length === 0) {\n      delete payload.metrics;\n    }\n\n    return payload;\n  }\n\n  async shutdown(): Promise<void> {\n    if (!this.config) {\n      return;\n    }\n\n    // End all active spans\n    for (const [_traceId, spanData] of this.traceMap) {\n      for (const [_spanId, span] of spanData.spans) {\n        span.end();\n      }\n      // Loggers don't have an explicit shutdown method\n    }\n    this.traceMap.clear();\n  }\n}\n"]}
+{"version":3,"sources":["../src/metrics.ts","../src/tracing.ts"],"names":["SpanType","BaseExporter","initLogger","currentSpan","omitKeys"],"mappings":";;;;;;;;;;AAuBO,SAAS,sBAAsB,SAAA,EAA8D;AAClG,EAAA,MAAM,UAAkC,EAAC;AAEzC,EAAA,IAAI,SAAA,CAAU,KAAA,EAAO,WAAA,KAAgB,MAAA,EAAW;AAC9C,IAAA,OAAA,CAAQ,aAAA,GAAgB,UAAU,KAAA,EAAO,WAAA;AAAA,EAC3C,CAAA,MAAA,IAAW,SAAA,CAAU,KAAA,EAAO,YAAA,KAAiB,MAAA,EAAW;AACtD,IAAA,OAAA,CAAQ,aAAA,GAAgB,UAAU,KAAA,EAAO,YAAA;AAAA,EAC3C;AAEA,EAAA,IAAI,SAAA,CAAU,KAAA,EAAO,YAAA,KAAiB,MAAA,EAAW;AAC/C,IAAA,OAAA,CAAQ,iBAAA,GAAoB,UAAU,KAAA,EAAO,YAAA;AAAA,EAC/C,CAAA,MAAA,IAAW,SAAA,CAAU,KAAA,EAAO,gBAAA,KAAqB,MAAA,EAAW;AAC1D,IAAA,OAAA,CAAQ,iBAAA,GAAoB,UAAU,KAAA,EAAO,gBAAA;AAAA,EAC/C;AAEA,EAAA,IAAI,SAAA,CAAU,KAAA,EAAO,WAAA,KAAgB,MAAA,EAAW;AAC9C,IAAA,OAAA,CAAQ,MAAA,GAAS,UAAU,KAAA,EAAO,WAAA;AAAA,EACpC;AACA,EAAA,IAAI,SAAA,CAAU,KAAA,EAAO,eAAA,KAAoB,MAAA,EAAW;AAClD,IAAA,OAAA,CAAQ,2BAAA,GAA8B,UAAU,KAAA,EAAO,eAAA;AAAA,EACzD;AACA,EAAA,IAAI,SAAA,CAAU,KAAA,EAAO,oBAAA,KAAyB,MAAA,EAAW;AACvD,IAAA,OAAA,CAAQ,oBAAA,GAAuB,UAAU,KAAA,EAAO,oBAAA;AAAA,EAClD;AACA,EAAA,IAAI,SAAA,CAAU,KAAA,EAAO,qBAAA,KAA0B,MAAA,EAAW;AACxD,IAAA,OAAA,CAAQ,4BAAA,GAA+B,UAAU,KAAA,EAAO,qBAAA;AAAA,EAC1D;AAEA,EAAA,OAAO,OAAA;AACT;;;ACnCA,IAAM,4BAAA,GAA+B,iBAAA;AA8BrC,IAAM,iBAAA,GAAoB,MAAA;AAG1B,IAAM,oBAAA,GAA0D;AAAA,EAC9D,CAACA,wBAAA,CAAS,gBAAgB,GAAG,KAAA;AAAA,EAC7B,CAACA,wBAAA,CAAS,WAAW,GAAG,KAAA;AAAA,EACxB,CAACA,wBAAA,CAAS,SAAS,GAAG,MAAA;AAAA,EACtB,CAACA,wBAAA,CAAS,aAAa,GAAG,MAAA;AAAA,EAC1B,CAACA,wBAAA,CAAS,yBAAyB,GAAG,UAAA;AAAA,EACtC,CAACA,wBAAA,CAAS,mBAAmB,GAAG;AAClC,CAAA;AAGA,SAAS,YAAY,QAAA,EAA6E;AAChG,EAAA,OAAQ,oBAAA,CAAqB,QAAQ,CAAA,IAAa,iBAAA;AACpD;AAEO,IAAM,kBAAA,GAAN,cAAiCC,0BAAA,CAAa;AAAA,EACnD,IAAA,GAAO,YAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA;AAAA,EAGA,iBAAA;AAAA,EACA,cAAA;AAAA,EAER,YAAY,MAAA,EAAkC;AAC5C,IAAA,KAAA,CAAM,MAAM,CAAA;AAEZ,IAAA,IAAI,OAAO,gBAAA,EAAkB;AAE3B,MAAA,IAAA,CAAK,iBAAA,GAAoB,IAAA;AACzB,MAAA,IAAA,CAAK,iBAAiB,MAAA,CAAO,gBAAA;AAC7B,MAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,IAChB,CAAA,MAAO;AAEL,MAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,QAAA,IAAA,CAAK,YAAY,CAAA,sCAAA,EAAyC,CAAC,CAAC,MAAA,CAAO,MAAM,CAAA,CAAA,CAAG,CAAA;AAC5E,QAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,QAAA,IAAA,CAAK,iBAAA,GAAoB,KAAA;AACzB,QAAA;AAAA,MACF;AACA,MAAA,IAAA,CAAK,iBAAA,GAAoB,KAAA;AACzB,MAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,IAChB;AAAA,EACF;AAAA,EAEA,MAAgB,oBAAoB,KAAA,EAAoC;AACtE,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;AACJ,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAsC;AACpE,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAI,KAAK,iBAAA,EAAmB;AAE1B,QAAA,MAAM,IAAA,CAAK,uBAAuB,IAAI,CAAA;AAAA,MACxC,CAAA,MAAO;AAEL,QAAA,MAAM,IAAA,CAAK,mBAAmB,IAAI,CAAA;AAAA,MACpC;AAAA,IACF;AAEA,IAAA,MAAM,MAAA,GAAS,mBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAGA,IAAA,IAAI,CAAC,KAAK,OAAA,EAAS;AACjB,MAAA,QAAA,CAAS,SAAA,CAAU,GAAA,CAAI,IAAA,CAAK,EAAE,CAAA;AAAA,IAChC;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAI1C,IAAA,MAAM,mBAAA,GAAsB,QAAA,CAAS,UAAA,IAAc,CAAC,IAAA,CAAK,YAAA;AAEzD,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,GAAI,mBAAA,GACA,EAAC,GACD;AAAA,QACE,eAAe,IAAA,CAAK,YAAA,GAChB,EAAE,MAAA,EAAQ,KAAK,YAAA,EAAc,UAAA,EAAY,IAAA,CAAK,OAAA,KAC9C,EAAE,MAAA,EAAQ,KAAK,OAAA,EAAS,UAAA,EAAY,KAAK,OAAA;AAAQ,OACvD;AAAA,MACJ,GAAG;AAAA,KACJ,CAAA;AAGD,IAAA,cAAA,CAAe,GAAA,CAAI;AAAA,MACjB,QAAA,EAAU;AAAA,QACR,CAAC,4BAA4B,GAAG,IAAA,CAAK;AAAA;AACvC,KACD,CAAA;AAED,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAuB,KAAA,EAA+B;AACxF,IAAA,MAAM,MAAA,GAAS,QAAQ,eAAA,GAAkB,kBAAA;AAEzC,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,cAAA,GAAiB,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,KAAK,mEAAA,EAAqE;AAAA,QACpF,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;AAEA,IAAA,cAAA,CAAe,GAAA,CAAI,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAC,CAAA;AAE9C,IAAA,IAAI,KAAA,EAAO;AAET,MAAA,IAAI,KAAK,OAAA,EAAS;AAChB,QAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,QAAQ,OAAA,EAAQ,GAAI,KAAM,CAAA;AAAA,MAC/D,CAAA,MAAO;AACL,QAAA,cAAA,CAAe,GAAA,EAAI;AAAA,MACrB;AAGA,MAAA,IAAI,CAAC,KAAK,OAAA,EAAS;AACjB,QAAA,QAAA,CAAS,SAAA,CAAU,MAAA,CAAO,IAAA,CAAK,EAAE,CAAA;AAAA,MACnC;AAIA,MAAA,IAAI,SAAS,SAAA,CAAU,IAAA,KAAS,CAAA,IAAK,CAAC,SAAS,UAAA,EAAY;AACzD,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,MACnC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAsC;AAClE,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,gDAAA,EAAkD;AAAA,QAClE,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,MAAA,EAAQ;AAAA,OACT,CAAA;AAED,MAAA,IAAI,KAAK,iBAAA,EAAmB;AAE1B,QAAA,MAAM,IAAA,CAAK,uBAAuB,IAAI,CAAA;AAAA,MACxC,CAAA,MAAO;AAEL,QAAA,MAAM,IAAA,CAAK,mBAAmB,IAAI,CAAA;AAAA,MACpC;AAAA,IACF;AAEA,IAAA,MAAM,MAAA,GAAS,iBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAG1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,eAAe,IAAA,CAAK,YAAA,GAChB,EAAE,MAAA,EAAQ,KAAK,YAAA,EAAc,UAAA,EAAY,IAAA,CAAK,OAAA,KAC9C,EAAE,MAAA,EAAQ,KAAK,OAAA,EAAS,UAAA,EAAY,KAAK,OAAA,EAAQ;AAAA,MACrD,SAAA,EAAW,IAAA,CAAK,SAAA,CAAU,OAAA,EAAQ,GAAI,GAAA;AAAA,MACtC,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,UAAU,OAAA,EAAQ,GAAI,KAAM,CAAA;AAAA,EACjE;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,mBAAmB,IAAA,EAAsC;AACrE,IAAA,MAAM,MAAA,GAAS,MAAMC,qBAAA,CAAW;AAAA,MAC9B,WAAA,EAAa,IAAA,CAAK,MAAA,CAAO,WAAA,IAAe,gBAAA;AAAA,MACxC,MAAA,EAAQ,KAAK,MAAA,CAAO,MAAA;AAAA,MACpB,MAAA,EAAQ,KAAK,MAAA,CAAO,QAAA;AAAA,MACpB,GAAG,KAAK,MAAA,CAAO;AAAA,KAChB,CAAA;AAED,IAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS;AAAA,MAC9B,MAAA;AAAA,MACA,KAAA,sBAAW,GAAA,EAAI;AAAA,MACf,SAAA,sBAAe,GAAA,EAAI;AAAA,MACnB,UAAA,EAAY;AAAA,KACb,CAAA;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAc,uBAAuB,IAAA,EAAsC;AAIzE,IAAA,MAAM,iBAAiBC,sBAAA,EAAY;AAGnC,IAAA,IAAI,cAAA,IAAkB,eAAe,EAAA,EAAI;AAEvC,MAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS;AAAA,QAC9B,MAAA,EAAQ,cAAA;AAAA,QACR,KAAA,sBAAW,GAAA,EAAI;AAAA,QACf,SAAA,sBAAe,GAAA,EAAI;AAAA,QACnB,UAAA,EAAY;AAAA,OACb,CAAA;AAAA,IACH,CAAA,MAAO;AAEL,MAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS;AAAA,QAC9B,QAAQ,IAAA,CAAK,cAAA;AAAA,QACb,KAAA,sBAAW,GAAA,EAAI;AAAA,QACf,SAAA,sBAAe,GAAA,EAAI;AAAA,QACnB,UAAA,EAAY;AAAA,OACb,CAAA;AAAA,IACH;AAAA,EACF;AAAA,EAEQ,YAAY,OAAA,EAA0E;AAC5F,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AACzB,IAAA,IAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA,EAAG;AACnC,MAAA,OAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACvC;AAEA,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,oBAAoB,OAAA,EAIQ;AAClC,IAAA,MAAM,EAAE,QAAA,EAAU,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AAEnC,IAAA,MAAM,WAAW,IAAA,CAAK,YAAA;AACtB,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,QAAA,CAAS,MAAA;AAAA,IAClB;AAEA,IAAA,IAAI,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA,EAAG;AAChC,MAAA,OAAO,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA;AAAA,IACpC;AAMA,IAAA,IAAI,YAAY,CAAC,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA,EAAG;AAG7C,MAAA,OAAO,QAAA,CAAS,MAAA;AAAA,IAClB;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,oDAAA,EAAsD;AAAA,MACrE,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,iBAAiB,IAAA,EAA4C;AACnE,IAAA,MAAM,UAA+B,EAAC;AAGtC,IAAA,IAAI,IAAA,CAAK,UAAU,MAAA,EAAW;AAC5B,MAAA,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACvB;AAEA,IAAA,IAAI,IAAA,CAAK,WAAW,MAAA,EAAW;AAC7B,MAAA,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AAAA,IACxB;AAGA,IAAA,OAAA,CAAQ,UAAU,EAAC;AACnB,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,IAAA,CAAK;AAAA,KACV;AAEA,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAExC,IAAA,IAAI,IAAA,CAAK,IAAA,KAASH,wBAAA,CAAS,gBAAA,EAAkB;AAC3C,MAAA,MAAM,SAAA,GAAY,UAAA;AAGlB,MAAA,IAAI,SAAA,CAAU,UAAU,MAAA,EAAW;AACjC,QAAA,OAAA,CAAQ,QAAA,CAAS,QAAQ,SAAA,CAAU,KAAA;AAAA,MACrC;AAGA,MAAA,IAAI,SAAA,CAAU,aAAa,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,QAAA,CAAS,WAAW,SAAA,CAAU,QAAA;AAAA,MACxC;AAGA,MAAA,OAAA,CAAQ,OAAA,GAAU,sBAAsB,SAAS,CAAA;AAGjD,MAAA,IAAI,SAAA,CAAU,eAAe,MAAA,EAAW;AACtC,QAAA,OAAA,CAAQ,QAAA,CAAS,kBAAkB,SAAA,CAAU,UAAA;AAAA,MAC/C;AAGA,MAAA,MAAM,kBAAkBI,cAAA,CAAS,UAAA,EAAY,CAAC,OAAA,EAAS,OAAA,EAAS,YAAY,CAAC,CAAA;AAC7E,MAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,QACjB,GAAG,OAAA,CAAQ,QAAA;AAAA,QACX,GAAG;AAAA,OACL;AAAA,IACF,CAAA,MAAO;AAEL,MAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,QACjB,GAAG,OAAA,CAAQ,QAAA;AAAA,QACX,GAAG;AAAA,OACL;AAAA,IACF;AAGA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,OAAA,CAAQ,KAAA,GAAQ,KAAK,SAAA,CAAU,OAAA;AAC/B,MAAA,OAAA,CAAQ,QAAA,CAAS,eAAe,IAAA,CAAK,SAAA;AAAA,IACvC;AAGA,IAAA,IAAI,OAAO,IAAA,CAAK,OAAA,CAAQ,OAAO,CAAA,CAAE,WAAW,CAAA,EAAG;AAC7C,MAAA,OAAO,OAAA,CAAQ,OAAA;AAAA,IACjB;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAChB,MAAA;AAAA,IACF;AAGA,IAAA,KAAA,MAAW,CAAC,QAAA,EAAU,QAAQ,CAAA,IAAK,KAAK,QAAA,EAAU;AAChD,MAAA,KAAA,MAAW,CAAC,OAAA,EAAS,IAAI,CAAA,IAAK,SAAS,KAAA,EAAO;AAC5C,QAAA,IAAA,CAAK,GAAA,EAAI;AAAA,MACX;AAAA,IAEF;AACA,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AACpB,IAAA,MAAM,MAAM,QAAA,EAAS;AAAA,EACvB;AACF","file":"index.cjs","sourcesContent":["import type { ModelGenerationAttributes } from '@mastra/core/observability';\n/**\n * BraintrustUsageMetrics\n *\n * Canonical metric keys expected by Braintrust for LLM usage accounting.\n * These map various provider/SDK-specific usage fields to a common schema.\n * - prompt_tokens: input-side tokens (aka inputTokens/promptTokens)\n * - completion_tokens: output-side tokens (aka outputTokens/completionTokens)\n * - tokens: total tokens (provided or derived)\n * - completion_reasoning_tokens: reasoning tokens, when available\n * - prompt_cached_tokens: tokens served from cache (provider-specific)\n * - prompt_cache_creation_tokens: tokens used to create cache (provider-specific)\n */\nexport interface BraintrustUsageMetrics {\n  prompt_tokens?: number;\n  completion_tokens?: number;\n  tokens?: number;\n  completion_reasoning_tokens?: number;\n  prompt_cached_tokens?: number;\n  prompt_cache_creation_tokens?: number;\n  [key: string]: number | undefined;\n}\n\nexport function normalizeUsageMetrics(modelAttr: ModelGenerationAttributes): BraintrustUsageMetrics {\n  const metrics: BraintrustUsageMetrics = {};\n\n  if (modelAttr.usage?.inputTokens !== undefined) {\n    metrics.prompt_tokens = modelAttr.usage?.inputTokens;\n  } else if (modelAttr.usage?.promptTokens !== undefined) {\n    metrics.prompt_tokens = modelAttr.usage?.promptTokens;\n  }\n\n  if (modelAttr.usage?.outputTokens !== undefined) {\n    metrics.completion_tokens = modelAttr.usage?.outputTokens;\n  } else if (modelAttr.usage?.completionTokens !== undefined) {\n    metrics.completion_tokens = modelAttr.usage?.completionTokens;\n  }\n\n  if (modelAttr.usage?.totalTokens !== undefined) {\n    metrics.tokens = modelAttr.usage?.totalTokens;\n  }\n  if (modelAttr.usage?.reasoningTokens !== undefined) {\n    metrics.completion_reasoning_tokens = modelAttr.usage?.reasoningTokens;\n  }\n  if (modelAttr.usage?.promptCacheHitTokens !== undefined) {\n    metrics.prompt_cached_tokens = modelAttr.usage?.promptCacheHitTokens;\n  }\n  if (modelAttr.usage?.promptCacheMissTokens !== undefined) {\n    metrics.prompt_cache_creation_tokens = modelAttr.usage?.promptCacheMissTokens;\n  }\n\n  return metrics;\n}\n","/**\n * Braintrust Exporter for Mastra Observability\n *\n * This exporter sends observability data to Braintrust.\n * Root spans become top-level Braintrust spans (no trace wrapper).\n * Events are handled as zero-duration spans with matching start/end times.\n */\n\nimport type { TracingEvent, AnyExportedSpan, ModelGenerationAttributes } from '@mastra/core/observability';\nimport { SpanType } from '@mastra/core/observability';\nimport { omitKeys } from '@mastra/core/utils';\nimport { BaseExporter } from '@mastra/observability';\nimport type { BaseExporterConfig } from '@mastra/observability';\nimport { initLogger, currentSpan } from 'braintrust';\nimport type { Span, Logger } from 'braintrust';\nimport { normalizeUsageMetrics } from './metrics';\n\nconst MASTRA_TRACE_ID_METADATA_KEY = 'mastra-trace-id';\n\nexport interface BraintrustExporterConfig extends BaseExporterConfig {\n  /**\n   * Optional Braintrust logger instance.\n   * When provided, enables integration with Braintrust contexts such as:\n   * - Evals: Agent traces nest inside eval task spans\n   * - logger.traced(): Agent traces nest inside traced spans\n   * - Parent spans: Auto-detects and attaches to external Braintrust spans\n   */\n  braintrustLogger?: Logger<true>;\n\n  /** Braintrust API key. Required if logger is not provided. */\n  apiKey?: string;\n  /** Optional custom endpoint */\n  endpoint?: string;\n  /** Braintrust project name (default: 'mastra-tracing') */\n  projectName?: string;\n  /** Support tuning parameters */\n  tuningParameters?: Record<string, any>;\n}\n\ntype SpanData = {\n  logger: Logger<true> | Span; // Braintrust logger (for root spans) or external span\n  spans: Map<string, Span>; // Maps span.id to Braintrust span\n  activeIds: Set<string>; // Tracks started (non-event) spans not yet ended, including root\n  isExternal: boolean; // True if logger is an external span from logger.traced() or Eval()\n};\n\n// Default span type for all spans\nconst DEFAULT_SPAN_TYPE = 'task';\n\n// Exceptions to the default mapping\nconst SPAN_TYPE_EXCEPTIONS: Partial<Record<SpanType, string>> = {\n  [SpanType.MODEL_GENERATION]: 'llm',\n  [SpanType.MODEL_CHUNK]: 'llm',\n  [SpanType.TOOL_CALL]: 'tool',\n  [SpanType.MCP_TOOL_CALL]: 'tool',\n  [SpanType.WORKFLOW_CONDITIONAL_EVAL]: 'function',\n  [SpanType.WORKFLOW_WAIT_EVENT]: 'function',\n};\n\n// Mapping function - returns valid Braintrust span types\nfunction mapSpanType(spanType: SpanType): 'llm' | 'score' | 'function' | 'eval' | 'task' | 'tool' {\n  return (SPAN_TYPE_EXCEPTIONS[spanType] as any) ?? DEFAULT_SPAN_TYPE;\n}\n\nexport class BraintrustExporter extends BaseExporter {\n  name = 'braintrust';\n  private traceMap = new Map<string, SpanData>();\n  private config: BraintrustExporterConfig;\n\n  // Flags and logger for context-aware mode\n  private useProvidedLogger: boolean;\n  private providedLogger?: Logger<true>;\n\n  constructor(config: BraintrustExporterConfig) {\n    super(config);\n\n    if (config.braintrustLogger) {\n      // Use provided logger - enables Braintrust context integration\n      this.useProvidedLogger = true;\n      this.providedLogger = config.braintrustLogger;\n      this.config = config;\n    } else {\n      // Validate apiKey for creating loggers per trace\n      if (!config.apiKey) {\n        this.setDisabled(`Missing required credentials (apiKey: ${!!config.apiKey})`);\n        this.config = null as any;\n        this.useProvidedLogger = false;\n        return;\n      }\n      this.useProvidedLogger = false;\n      this.config = config;\n    }\n  }\n\n  protected async _exportTracingEvent(event: TracingEvent): 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\n  private async handleSpanStarted(span: AnyExportedSpan): Promise<void> {\n    if (span.isRootSpan) {\n      if (this.useProvidedLogger) {\n        // Use provided logger, detect external Braintrust spans\n        await this.initLoggerOrUseContext(span);\n      } else {\n        // Create new logger per trace\n        await this.initLoggerPerTrace(span);\n      }\n    }\n\n    const method = 'handleSpanStarted';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    // Refcount: track active non-event spans (including root)\n    if (!span.isEvent) {\n      spanData.activeIds.add(span.id);\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    // When attaching to an external parent (eval/logger span), don't pass Mastra's internal\n    // parentSpanIds. Let Braintrust auto-handle the parent-child relationship.\n    const shouldOmitParentIds = spanData.isExternal && !span.parentSpanId;\n\n    const braintrustSpan = braintrustParent.startSpan({\n      spanId: span.id,\n      name: span.name,\n      type: mapSpanType(span.type),\n      ...(shouldOmitParentIds\n        ? {}\n        : {\n            parentSpanIds: span.parentSpanId\n              ? { spanId: span.parentSpanId, rootSpanId: span.traceId }\n              : { spanId: span.traceId, rootSpanId: span.traceId },\n          }),\n      ...payload,\n    });\n\n    // Include the Mastra trace ID in the span metadata for correlation\n    braintrustSpan.log({\n      metadata: {\n        [MASTRA_TRACE_ID_METADATA_KEY]: span.traceId,\n      },\n    });\n\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyExportedSpan, isEnd: boolean): Promise<void> {\n    const method = isEnd ? 'handleSpanEnd' : 'handleSpanUpdate';\n\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustSpan = spanData.spans.get(span.id);\n    if (!braintrustSpan) {\n      this.logger.warn('Braintrust exporter: No Braintrust 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    braintrustSpan.log(this.buildSpanPayload(span));\n\n    if (isEnd) {\n      // End the span with the correct endTime (convert milliseconds to seconds)\n      if (span.endTime) {\n        braintrustSpan.end({ endTime: span.endTime.getTime() / 1000 });\n      } else {\n        braintrustSpan.end();\n      }\n\n      // Refcount: mark this span as ended\n      if (!span.isEvent) {\n        spanData.activeIds.delete(span.id);\n      }\n\n      // If no more active spans remain for this trace, clean up the trace entry\n      // Don't clean up if using external spans (they're managed by Braintrust)\n      if (spanData.activeIds.size === 0 && !spanData.isExternal) {\n        this.traceMap.delete(span.traceId);\n      }\n    }\n  }\n\n  private async handleEventSpan(span: AnyExportedSpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.logger.debug('Braintrust exporter: Creating logger for event', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        method: 'handleEventSpan',\n      });\n\n      if (this.useProvidedLogger) {\n        // Use provided logger, detect external Braintrust spans\n        await this.initLoggerOrUseContext(span);\n      } else {\n        // Create new logger per trace\n        await this.initLoggerPerTrace(span);\n      }\n    }\n\n    const method = 'handleEventSpan';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    // Create zero-duration span for event (convert milliseconds to seconds)\n    const braintrustSpan = braintrustParent.startSpan({\n      spanId: span.id,\n      name: span.name,\n      type: mapSpanType(span.type),\n      parentSpanIds: span.parentSpanId\n        ? { spanId: span.parentSpanId, rootSpanId: span.traceId }\n        : { spanId: span.traceId, rootSpanId: span.traceId },\n      startTime: span.startTime.getTime() / 1000,\n      ...payload,\n    });\n\n    braintrustSpan.end({ endTime: span.startTime.getTime() / 1000 });\n  }\n\n  /**\n   * Creates a new logger per trace using config credentials\n   */\n  private async initLoggerPerTrace(span: AnyExportedSpan): Promise<void> {\n    const logger = await initLogger({\n      projectName: this.config.projectName ?? 'mastra-tracing',\n      apiKey: this.config.apiKey,\n      appUrl: this.config.endpoint,\n      ...this.config.tuningParameters,\n    });\n\n    this.traceMap.set(span.traceId, {\n      logger,\n      spans: new Map(),\n      activeIds: new Set(),\n      isExternal: false,\n    });\n  }\n\n  /**\n   * Uses provided logger and detects external Braintrust spans.\n   * If a Braintrust span is detected (from logger.traced() or Eval()), attaches to it.\n   * Otherwise, uses the provided logger instance.\n   */\n  private async initLoggerOrUseContext(span: AnyExportedSpan): Promise<void> {\n    // Try to find a Braintrust span to attach to:\n    // 1. Auto-detect from Braintrust's current span (logger.traced(), Eval(), etc.)\n    // 2. Fall back to the configured logger\n    const braintrustSpan = currentSpan();\n\n    // Check if it's a valid span (not the NOOP_SPAN)\n    if (braintrustSpan && braintrustSpan.id) {\n      // External span detected - attach Mastra traces to it\n      this.traceMap.set(span.traceId, {\n        logger: braintrustSpan,\n        spans: new Map(),\n        activeIds: new Set(),\n        isExternal: true,\n      });\n    } else {\n      // No external span - use provided logger\n      this.traceMap.set(span.traceId, {\n        logger: this.providedLogger!,\n        spans: new Map(),\n        activeIds: new Set(),\n        isExternal: false,\n      });\n    }\n  }\n\n  private getSpanData(options: { span: AnyExportedSpan; method: string }): SpanData | undefined {\n    const { span, method } = options;\n    if (this.traceMap.has(span.traceId)) {\n      return this.traceMap.get(span.traceId);\n    }\n\n    this.logger.warn('Braintrust exporter: No span 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 getBraintrustParent(options: {\n    spanData: SpanData;\n    span: AnyExportedSpan;\n    method: string;\n  }): Logger<true> | Span | undefined {\n    const { spanData, span, method } = options;\n\n    const parentId = span.parentSpanId;\n    if (!parentId) {\n      return spanData.logger;\n    }\n\n    if (spanData.spans.has(parentId)) {\n      return spanData.spans.get(parentId);\n    }\n\n    // If the parent exists but is the root span (not represented as a Braintrust\n    // span because we use the logger as the root), attach to the logger so the\n    // span is not orphaned. We need to check if parentSpanId exists but the\n    // parent span is not in our spans map (indicating it's the root span).\n    if (parentId && !spanData.spans.has(parentId)) {\n      // This means the parent exists but isn't tracked as a Braintrust span,\n      // which happens when the parent is the root span (we use logger as root)\n      return spanData.logger;\n    }\n\n    this.logger.warn('Braintrust 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 buildSpanPayload(span: AnyExportedSpan): Record<string, any> {\n    const payload: Record<string, any> = {};\n\n    // Core span data\n    if (span.input !== undefined) {\n      payload.input = span.input;\n    }\n\n    if (span.output !== undefined) {\n      payload.output = span.output;\n    }\n\n    // Initialize metrics and metadata objects\n    payload.metrics = {};\n    payload.metadata = {\n      spanType: span.type,\n      ...span.metadata,\n    };\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    if (span.type === SpanType.MODEL_GENERATION) {\n      const modelAttr = attributes as ModelGenerationAttributes;\n\n      // Model goes to metadata\n      if (modelAttr.model !== undefined) {\n        payload.metadata.model = modelAttr.model;\n      }\n\n      // Provider goes to metadata (if provided by attributes)\n      if (modelAttr.provider !== undefined) {\n        payload.metadata.provider = modelAttr.provider;\n      }\n\n      // Usage/token info goes to metrics\n      payload.metrics = normalizeUsageMetrics(modelAttr);\n\n      // Model parameters go to metadata\n      if (modelAttr.parameters !== undefined) {\n        payload.metadata.modelParameters = modelAttr.parameters;\n      }\n\n      // Other LLM attributes go to metadata\n      const otherAttributes = omitKeys(attributes, ['model', 'usage', 'parameters']);\n      payload.metadata = {\n        ...payload.metadata,\n        ...otherAttributes,\n      };\n    } else {\n      // For non-LLM spans, put all attributes in metadata\n      payload.metadata = {\n        ...payload.metadata,\n        ...attributes,\n      };\n    }\n\n    // Handle errors\n    if (span.errorInfo) {\n      payload.error = span.errorInfo.message;\n      payload.metadata.errorDetails = span.errorInfo;\n    }\n\n    // Clean up empty metrics object\n    if (Object.keys(payload.metrics).length === 0) {\n      delete payload.metrics;\n    }\n\n    return payload;\n  }\n\n  async shutdown(): Promise<void> {\n    if (!this.config) {\n      return;\n    }\n\n    // End all active spans\n    for (const [_traceId, spanData] of this.traceMap) {\n      for (const [_spanId, span] of spanData.spans) {\n        span.end();\n      }\n      // Loggers don't have an explicit shutdown method\n    }\n    this.traceMap.clear();\n    await super.shutdown();\n  }\n}\n"]}

package/dist/index.d.ts

@@ -2,7 +2,7 @@
  * Braintrust Observability Provider for Mastra
  *
  * This package provides Braintrust-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';
 //# 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"}