@mastra/braintrust 0.1.9 → 0.2.0

package/CHANGELOG.md

@@ -1,5 +1,59 @@
 # @mastra/braintrust
 
+## 0.2.0
+
+### Minor Changes
+
+- Rename LLM span types and attributes to use Model prefix ([#9105](https://github.com/mastra-ai/mastra/pull/9105))
+
+  BREAKING CHANGE: This release renames AI tracing span types and attribute interfaces to use the "Model" prefix instead of "LLM":
+  - `AISpanType.LLM_GENERATION` → `AISpanType.MODEL_GENERATION`
+  - `AISpanType.LLM_STEP` → `AISpanType.MODEL_STEP`
+  - `AISpanType.LLM_CHUNK` → `AISpanType.MODEL_CHUNK`
+  - `LLMGenerationAttributes` → `ModelGenerationAttributes`
+  - `LLMStepAttributes` → `ModelStepAttributes`
+  - `LLMChunkAttributes` → `ModelChunkAttributes`
+  - `InternalSpans.LLM` → `InternalSpans.MODEL`
+
+  This change better reflects that these span types apply to all AI models, not just Large Language Models.
+
+  Migration guide:
+  - Update all imports: `import { ModelGenerationAttributes } from '@mastra/core/ai-tracing'`
+  - Update span type references: `AISpanType.MODEL_GENERATION`
+  - Update InternalSpans usage: `InternalSpans.MODEL`
+
+### Patch Changes
+
+- Updated dependencies [[`f743dbb`](https://github.com/mastra-ai/mastra/commit/f743dbb8b40d1627b5c10c0e6fc154f4ebb6e394), [`5df9cce`](https://github.com/mastra-ai/mastra/commit/5df9cce1a753438413f64c11eeef8f845745c2a8), [`2060766`](https://github.com/mastra-ai/mastra/commit/20607667bf78ea104cca3e15dfb93ae0b62c9d18), [`2c4438b`](https://github.com/mastra-ai/mastra/commit/2c4438b87817ab7eed818c7990fef010475af1a3)]:
+  - @mastra/core@0.23.0
+
+## 0.2.0-alpha.0
+
+### Minor Changes
+
+- Rename LLM span types and attributes to use Model prefix ([#9105](https://github.com/mastra-ai/mastra/pull/9105))
+
+  BREAKING CHANGE: This release renames AI tracing span types and attribute interfaces to use the "Model" prefix instead of "LLM":
+  - `AISpanType.LLM_GENERATION` → `AISpanType.MODEL_GENERATION`
+  - `AISpanType.LLM_STEP` → `AISpanType.MODEL_STEP`
+  - `AISpanType.LLM_CHUNK` → `AISpanType.MODEL_CHUNK`
+  - `LLMGenerationAttributes` → `ModelGenerationAttributes`
+  - `LLMStepAttributes` → `ModelStepAttributes`
+  - `LLMChunkAttributes` → `ModelChunkAttributes`
+  - `InternalSpans.LLM` → `InternalSpans.MODEL`
+
+  This change better reflects that these span types apply to all AI models, not just Large Language Models.
+
+  Migration guide:
+  - Update all imports: `import { ModelGenerationAttributes } from '@mastra/core/ai-tracing'`
+  - Update span type references: `AISpanType.MODEL_GENERATION`
+  - Update InternalSpans usage: `InternalSpans.MODEL`
+
+### Patch Changes
+
+- Updated dependencies [[`f743dbb`](https://github.com/mastra-ai/mastra/commit/f743dbb8b40d1627b5c10c0e6fc154f4ebb6e394), [`5df9cce`](https://github.com/mastra-ai/mastra/commit/5df9cce1a753438413f64c11eeef8f845745c2a8), [`2060766`](https://github.com/mastra-ai/mastra/commit/20607667bf78ea104cca3e15dfb93ae0b62c9d18), [`2c4438b`](https://github.com/mastra-ai/mastra/commit/2c4438b87817ab7eed818c7990fef010475af1a3)]:
+  - @mastra/core@0.23.0-alpha.0
+
 ## 0.1.9
 
 ### Patch Changes

package/dist/ai-tracing.d.ts

@@ -5,8 +5,9 @@
  * Root spans become top-level Braintrust spans (no trace wrapper).
  * Events are handled as zero-duration spans with matching start/end times.
  */
-import { BaseExporter } from '@mastra/core/ai-tracing';
-import type { AITracingEvent, BaseExporterConfig } from '@mastra/core/ai-tracing';
+import type { AITracingEvent } from '@mastra/core/ai-tracing';
+import { BaseExporter } from '@mastra/core/ai-tracing/exporters';
+import type { BaseExporterConfig } from '@mastra/core/ai-tracing/exporters';
 export interface BraintrustExporterConfig extends BaseExporterConfig {
     /** Braintrust API key */
     apiKey?: string;

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

@@ -1 +1 @@
-{"version":3,"file":"ai-tracing.d.ts","sourceRoot":"","sources":["../src/ai-tracing.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAwB,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAC7E,OAAO,KAAK,EACV,cAAc,EAGd,kBAAkB,EACnB,MAAM,yBAAyB,CAAC;AAKjC,MAAM,WAAW,wBAAyB,SAAQ,kBAAkB;IAClE,yBAAyB;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,+BAA+B;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,0DAA0D;IAC1D,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,gCAAgC;IAChC,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACxC;AA0BD,qBAAa,kBAAmB,SAAQ,YAAY;IAClD,IAAI,SAAgB;IACpB,OAAO,CAAC,QAAQ,CAA+B;IAC/C,OAAO,CAAC,MAAM,CAA2B;gBAE7B,MAAM,EAAE,wBAAwB;cAY5B,YAAY,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;YAmBpD,iBAAiB;YAoCjB,qBAAqB;YA4CrB,eAAe;YAuCf,UAAU;IAWxB,OAAO,CAAC,WAAW;IAiBnB,OAAO,CAAC,mBAAmB;IAqC3B,OAAO,CAAC,gBAAgB;IAsElB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAehC"}
+{"version":3,"file":"ai-tracing.d.ts","sourceRoot":"","sources":["../src/ai-tracing.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAgD,MAAM,yBAAyB,CAAC;AAE5G,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAC;AACjE,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,mCAAmC,CAAC;AAK5E,MAAM,WAAW,wBAAyB,SAAQ,kBAAkB;IAClE,yBAAyB;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,+BAA+B;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,0DAA0D;IAC1D,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,gCAAgC;IAChC,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACxC;AA0BD,qBAAa,kBAAmB,SAAQ,YAAY;IAClD,IAAI,SAAgB;IACpB,OAAO,CAAC,QAAQ,CAA+B;IAC/C,OAAO,CAAC,MAAM,CAA2B;gBAE7B,MAAM,EAAE,wBAAwB;cAY5B,YAAY,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;YAmBpD,iBAAiB;YAoCjB,qBAAqB;YA4CrB,eAAe;YAuCf,UAAU;IAWxB,OAAO,CAAC,WAAW;IAiBnB,OAAO,CAAC,mBAAmB;IAqC3B,OAAO,CAAC,gBAAgB;IAsElB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAehC"}

package/dist/index.cjs

@@ -1,34 +1,35 @@
 'use strict';
 
 var aiTracing = require('@mastra/core/ai-tracing');
+var exporters = require('@mastra/core/ai-tracing/exporters');
 var braintrust = require('braintrust');
 
 // src/ai-tracing.ts
 
 // src/metrics.ts
-function normalizeUsageMetrics(llmAttr) {
+function normalizeUsageMetrics(modelAttr) {
   const metrics = {};
-  if (llmAttr.usage?.inputTokens !== void 0) {
-    metrics.prompt_tokens = llmAttr.usage?.inputTokens;
-  } else if (llmAttr.usage?.promptTokens !== void 0) {
-    metrics.prompt_tokens = llmAttr.usage?.promptTokens;
+  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 (llmAttr.usage?.outputTokens !== void 0) {
-    metrics.completion_tokens = llmAttr.usage?.outputTokens;
-  } else if (llmAttr.usage?.completionTokens !== void 0) {
-    metrics.completion_tokens = llmAttr.usage?.completionTokens;
+  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 (llmAttr.usage?.totalTokens !== void 0) {
-    metrics.tokens = llmAttr.usage?.totalTokens;
+  if (modelAttr.usage?.totalTokens !== void 0) {
+    metrics.tokens = modelAttr.usage?.totalTokens;
   }
-  if (llmAttr.usage?.reasoningTokens !== void 0) {
-    metrics.completion_reasoning_tokens = llmAttr.usage?.reasoningTokens;
+  if (modelAttr.usage?.reasoningTokens !== void 0) {
+    metrics.completion_reasoning_tokens = modelAttr.usage?.reasoningTokens;
   }
-  if (llmAttr.usage?.promptCacheHitTokens !== void 0) {
-    metrics.prompt_cached_tokens = llmAttr.usage?.promptCacheHitTokens;
+  if (modelAttr.usage?.promptCacheHitTokens !== void 0) {
+    metrics.prompt_cached_tokens = modelAttr.usage?.promptCacheHitTokens;
   }
-  if (llmAttr.usage?.promptCacheMissTokens !== void 0) {
-    metrics.prompt_cache_creation_tokens = llmAttr.usage?.promptCacheMissTokens;
+  if (modelAttr.usage?.promptCacheMissTokens !== void 0) {
+    metrics.prompt_cache_creation_tokens = modelAttr.usage?.promptCacheMissTokens;
   }
   return metrics;
 }
@@ -36,8 +37,8 @@ function normalizeUsageMetrics(llmAttr) {
 // src/ai-tracing.ts
 var DEFAULT_SPAN_TYPE = "task";
 var SPAN_TYPE_EXCEPTIONS = {
-  [aiTracing.AISpanType.LLM_GENERATION]: "llm",
-  [aiTracing.AISpanType.LLM_CHUNK]: "llm",
+  [aiTracing.AISpanType.MODEL_GENERATION]: "llm",
+  [aiTracing.AISpanType.MODEL_CHUNK]: "llm",
   [aiTracing.AISpanType.TOOL_CALL]: "tool",
   [aiTracing.AISpanType.MCP_TOOL_CALL]: "tool",
   [aiTracing.AISpanType.WORKFLOW_CONDITIONAL_EVAL]: "function",
@@ -46,7 +47,7 @@ var SPAN_TYPE_EXCEPTIONS = {
 function mapSpanType(spanType) {
   return SPAN_TYPE_EXCEPTIONS[spanType] ?? DEFAULT_SPAN_TYPE;
 }
-var BraintrustExporter = class extends aiTracing.BaseExporter {
+var BraintrustExporter = class extends exporters.BaseExporter {
   name = "braintrust";
   traceMap = /* @__PURE__ */ new Map();
   config;
@@ -226,17 +227,17 @@ var BraintrustExporter = class extends aiTracing.BaseExporter {
       ...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 === aiTracing.AISpanType.MODEL_GENERATION) {
+      const modelAttr = attributes;
+      if (modelAttr.model !== void 0) {
+        payload.metadata.model = modelAttr.model;
       }
-      if (llmAttr.provider !== void 0) {
-        payload.metadata.provider = llmAttr.provider;
+      if (modelAttr.provider !== void 0) {
+        payload.metadata.provider = modelAttr.provider;
       }
-      payload.metrics = normalizeUsageMetrics(llmAttr);
-      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"]);
       payload.metadata = {

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/metrics.ts","../src/ai-tracing.ts"],"names":["AISpanType","BaseExporter","initLogger","omitKeys"],"mappings":";;;;;;;;AAuBO,SAAS,sBAAsB,OAAA,EAA0D;AAC9F,EAAA,MAAM,UAAkC,EAAC;AAEzC,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,WAAA,KAAgB,MAAA,EAAW;AAC5C,IAAA,OAAA,CAAQ,aAAA,GAAgB,QAAQ,KAAA,EAAO,WAAA;AAAA,EACzC,CAAA,MAAA,IAAW,OAAA,CAAQ,KAAA,EAAO,YAAA,KAAiB,MAAA,EAAW;AACpD,IAAA,OAAA,CAAQ,aAAA,GAAgB,QAAQ,KAAA,EAAO,YAAA;AAAA,EACzC;AAEA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,YAAA,KAAiB,MAAA,EAAW;AAC7C,IAAA,OAAA,CAAQ,iBAAA,GAAoB,QAAQ,KAAA,EAAO,YAAA;AAAA,EAC7C,CAAA,MAAA,IAAW,OAAA,CAAQ,KAAA,EAAO,gBAAA,KAAqB,MAAA,EAAW;AACxD,IAAA,OAAA,CAAQ,iBAAA,GAAoB,QAAQ,KAAA,EAAO,gBAAA;AAAA,EAC7C;AAEA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,WAAA,KAAgB,MAAA,EAAW;AAC5C,IAAA,OAAA,CAAQ,MAAA,GAAS,QAAQ,KAAA,EAAO,WAAA;AAAA,EAClC;AACA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,eAAA,KAAoB,MAAA,EAAW;AAChD,IAAA,OAAA,CAAQ,2BAAA,GAA8B,QAAQ,KAAA,EAAO,eAAA;AAAA,EACvD;AACA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,oBAAA,KAAyB,MAAA,EAAW;AACrD,IAAA,OAAA,CAAQ,oBAAA,GAAuB,QAAQ,KAAA,EAAO,oBAAA;AAAA,EAChD;AACA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,qBAAA,KAA0B,MAAA,EAAW;AACtD,IAAA,OAAA,CAAQ,4BAAA,GAA+B,QAAQ,KAAA,EAAO,qBAAA;AAAA,EACxD;AAEA,EAAA,OAAO,OAAA;AACT;;;ACfA,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,kBAAA,GAAN,cAAiCC,sBAAA,CAAa;AAAA,EACnD,IAAA,GAAO,YAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA,EAER,YAAY,MAAA,EAAkC;AAC5C,IAAA,KAAA,CAAM,MAAM,CAAA;AAEZ,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,YAAY,CAAA,sCAAA,EAAyC,CAAC,CAAC,MAAA,CAAO,MAAM,CAAA,CAAA,CAAG,CAAA;AAC5E,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAgB,aAAa,KAAA,EAAsC;AACjE,IAAA,IAAI,KAAA,CAAM,aAAa,OAAA,EAAS;AAC9B,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,YAAY,CAAA;AAC7C,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,YAAY,CAAA;AAC/C,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,KAAK,CAAA;AAC1D,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,IAAI,CAAA;AACzD,QAAA;AAAA;AACJ,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAwC;AACtE,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;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;AAE1C,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,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,EAAyB,KAAA,EAA+B;AAC1F,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;AAGA,MAAA,IAAI,QAAA,CAAS,SAAA,CAAU,IAAA,KAAS,CAAA,EAAG;AACjC,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,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,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,EAEA,MAAc,WAAW,IAAA,EAAwC;AAC/D,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,EAAE,MAAA,EAAQ,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,SAAA,kBAAW,IAAI,GAAA,IAAO,CAAA;AAAA,EACpF;AAAA,EAEQ,YAAY,OAAA,EAA4E;AAC9F,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,EAA8C;AACrE,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,aAAa,MAAA,EAAW;AAClC,QAAA,OAAA,CAAQ,QAAA,CAAS,WAAW,OAAA,CAAQ,QAAA;AAAA,MACtC;AAGA,MAAA,OAAA,CAAQ,OAAA,GAAU,sBAAsB,OAAO,CAAA;AAG/C,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;AACpB,IAAA,MAAM,MAAM,QAAA,EAAS;AAAA,EACvB;AACF","file":"index.cjs","sourcesContent":["import type { LLMGenerationAttributes } from '@mastra/core/ai-tracing';\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(llmAttr: LLMGenerationAttributes): BraintrustUsageMetrics {\n  const metrics: BraintrustUsageMetrics = {};\n\n  if (llmAttr.usage?.inputTokens !== undefined) {\n    metrics.prompt_tokens = llmAttr.usage?.inputTokens;\n  } else if (llmAttr.usage?.promptTokens !== undefined) {\n    metrics.prompt_tokens = llmAttr.usage?.promptTokens;\n  }\n\n  if (llmAttr.usage?.outputTokens !== undefined) {\n    metrics.completion_tokens = llmAttr.usage?.outputTokens;\n  } else if (llmAttr.usage?.completionTokens !== undefined) {\n    metrics.completion_tokens = llmAttr.usage?.completionTokens;\n  }\n\n  if (llmAttr.usage?.totalTokens !== undefined) {\n    metrics.tokens = llmAttr.usage?.totalTokens;\n  }\n  if (llmAttr.usage?.reasoningTokens !== undefined) {\n    metrics.completion_reasoning_tokens = llmAttr.usage?.reasoningTokens;\n  }\n  if (llmAttr.usage?.promptCacheHitTokens !== undefined) {\n    metrics.prompt_cached_tokens = llmAttr.usage?.promptCacheHitTokens;\n  }\n  if (llmAttr.usage?.promptCacheMissTokens !== undefined) {\n    metrics.prompt_cache_creation_tokens = llmAttr.usage?.promptCacheMissTokens;\n  }\n\n  return metrics;\n}\n","/**\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 { AISpanType, omitKeys, BaseExporter } from '@mastra/core/ai-tracing';\nimport type {\n  AITracingEvent,\n  AnyExportedAISpan,\n  LLMGenerationAttributes,\n  BaseExporterConfig,\n} from '@mastra/core/ai-tracing';\nimport { initLogger } from 'braintrust';\nimport type { Span, Logger } from 'braintrust';\nimport { normalizeUsageMetrics } from './metrics';\n\nexport interface BraintrustExporterConfig extends BaseExporterConfig {\n  /** Braintrust API key */\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>; // Braintrust logger (for root spans)\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};\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 extends BaseExporter {\n  name = 'braintrust';\n  private traceMap = new Map<string, SpanData>();\n  private config: BraintrustExporterConfig;\n\n  constructor(config: BraintrustExporterConfig) {\n    super(config);\n\n    if (!config.apiKey) {\n      this.setDisabled(`Missing required credentials (apiKey: ${!!config.apiKey})`);\n      this.config = null as any;\n      return;\n    }\n\n    this.config = config;\n  }\n\n  protected async _exportEvent(event: AITracingEvent): Promise<void> {\n    if (event.exportedSpan.isEvent) {\n      await this.handleEventSpan(event.exportedSpan);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.exportedSpan);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, true);\n        break;\n    }\n  }\n\n  private async handleSpanStarted(span: AnyExportedAISpan): 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    // 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    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      ...payload,\n    });\n\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyExportedAISpan, 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      if (spanData.activeIds.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('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      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  private async initLogger(span: AnyExportedAISpan): 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, { logger, spans: new Map(), activeIds: new Set() });\n  }\n\n  private getSpanData(options: { span: AnyExportedAISpan; 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: AnyExportedAISpan;\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: AnyExportedAISpan): 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      // Provider goes to metadata (if provided by attributes)\n      if (llmAttr.provider !== undefined) {\n        payload.metadata.provider = llmAttr.provider;\n      }\n\n      // Usage/token info goes to metrics\n      payload.metrics = normalizeUsageMetrics(llmAttr);\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    await super.shutdown();\n  }\n}\n"]}
+{"version":3,"sources":["../src/metrics.ts","../src/ai-tracing.ts"],"names":["AISpanType","BaseExporter","initLogger","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;;;AClBA,IAAM,iBAAA,GAAoB,MAAA;AAG1B,IAAM,oBAAA,GAA4D;AAAA,EAChE,CAACA,oBAAA,CAAW,gBAAgB,GAAG,KAAA;AAAA,EAC/B,CAACA,oBAAA,CAAW,WAAW,GAAG,KAAA;AAAA,EAC1B,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,kBAAA,GAAN,cAAiCC,sBAAA,CAAa;AAAA,EACnD,IAAA,GAAO,YAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA,EAER,YAAY,MAAA,EAAkC;AAC5C,IAAA,KAAA,CAAM,MAAM,CAAA;AAEZ,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,YAAY,CAAA,sCAAA,EAAyC,CAAC,CAAC,MAAA,CAAO,MAAM,CAAA,CAAA,CAAG,CAAA;AAC5E,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAgB,aAAa,KAAA,EAAsC;AACjE,IAAA,IAAI,KAAA,CAAM,aAAa,OAAA,EAAS;AAC9B,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,YAAY,CAAA;AAC7C,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,YAAY,CAAA;AAC/C,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,KAAK,CAAA;AAC1D,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,IAAI,CAAA;AACzD,QAAA;AAAA;AACJ,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAwC;AACtE,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;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;AAE1C,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,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,EAAyB,KAAA,EAA+B;AAC1F,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;AAGA,MAAA,IAAI,QAAA,CAAS,SAAA,CAAU,IAAA,KAAS,CAAA,EAAG;AACjC,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,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,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,EAEA,MAAc,WAAW,IAAA,EAAwC;AAC/D,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,EAAE,MAAA,EAAQ,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,SAAA,kBAAW,IAAI,GAAA,IAAO,CAAA;AAAA,EACpF;AAAA,EAEQ,YAAY,OAAA,EAA4E;AAC9F,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,EAA8C;AACrE,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,gBAAA,EAAkB;AAC7C,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,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;AACpB,IAAA,MAAM,MAAM,QAAA,EAAS;AAAA,EACvB;AACF","file":"index.cjs","sourcesContent":["import type { ModelGenerationAttributes } from '@mastra/core/ai-tracing';\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 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 { AITracingEvent, AnyExportedAISpan, ModelGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, omitKeys } from '@mastra/core/ai-tracing';\nimport { BaseExporter } from '@mastra/core/ai-tracing/exporters';\nimport type { BaseExporterConfig } from '@mastra/core/ai-tracing/exporters';\nimport { initLogger } from 'braintrust';\nimport type { Span, Logger } from 'braintrust';\nimport { normalizeUsageMetrics } from './metrics';\n\nexport interface BraintrustExporterConfig extends BaseExporterConfig {\n  /** Braintrust API key */\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>; // Braintrust logger (for root spans)\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};\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.MODEL_GENERATION]: 'llm',\n  [AISpanType.MODEL_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 extends BaseExporter {\n  name = 'braintrust';\n  private traceMap = new Map<string, SpanData>();\n  private config: BraintrustExporterConfig;\n\n  constructor(config: BraintrustExporterConfig) {\n    super(config);\n\n    if (!config.apiKey) {\n      this.setDisabled(`Missing required credentials (apiKey: ${!!config.apiKey})`);\n      this.config = null as any;\n      return;\n    }\n\n    this.config = config;\n  }\n\n  protected async _exportEvent(event: AITracingEvent): Promise<void> {\n    if (event.exportedSpan.isEvent) {\n      await this.handleEventSpan(event.exportedSpan);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.exportedSpan);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, true);\n        break;\n    }\n  }\n\n  private async handleSpanStarted(span: AnyExportedAISpan): 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    // 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    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      ...payload,\n    });\n\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyExportedAISpan, 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      if (spanData.activeIds.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('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      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  private async initLogger(span: AnyExportedAISpan): 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, { logger, spans: new Map(), activeIds: new Set() });\n  }\n\n  private getSpanData(options: { span: AnyExportedAISpan; 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: AnyExportedAISpan;\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: AnyExportedAISpan): 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.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.js

@@ -1,32 +1,33 @@
-import { BaseExporter, AISpanType, omitKeys } from '@mastra/core/ai-tracing';
+import { AISpanType, omitKeys } from '@mastra/core/ai-tracing';
+import { BaseExporter } from '@mastra/core/ai-tracing/exporters';
 import { initLogger } from 'braintrust';
 
 // src/ai-tracing.ts
 
 // src/metrics.ts
-function normalizeUsageMetrics(llmAttr) {
+function normalizeUsageMetrics(modelAttr) {
   const metrics = {};
-  if (llmAttr.usage?.inputTokens !== void 0) {
-    metrics.prompt_tokens = llmAttr.usage?.inputTokens;
-  } else if (llmAttr.usage?.promptTokens !== void 0) {
-    metrics.prompt_tokens = llmAttr.usage?.promptTokens;
+  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 (llmAttr.usage?.outputTokens !== void 0) {
-    metrics.completion_tokens = llmAttr.usage?.outputTokens;
-  } else if (llmAttr.usage?.completionTokens !== void 0) {
-    metrics.completion_tokens = llmAttr.usage?.completionTokens;
+  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 (llmAttr.usage?.totalTokens !== void 0) {
-    metrics.tokens = llmAttr.usage?.totalTokens;
+  if (modelAttr.usage?.totalTokens !== void 0) {
+    metrics.tokens = modelAttr.usage?.totalTokens;
   }
-  if (llmAttr.usage?.reasoningTokens !== void 0) {
-    metrics.completion_reasoning_tokens = llmAttr.usage?.reasoningTokens;
+  if (modelAttr.usage?.reasoningTokens !== void 0) {
+    metrics.completion_reasoning_tokens = modelAttr.usage?.reasoningTokens;
   }
-  if (llmAttr.usage?.promptCacheHitTokens !== void 0) {
-    metrics.prompt_cached_tokens = llmAttr.usage?.promptCacheHitTokens;
+  if (modelAttr.usage?.promptCacheHitTokens !== void 0) {
+    metrics.prompt_cached_tokens = modelAttr.usage?.promptCacheHitTokens;
   }
-  if (llmAttr.usage?.promptCacheMissTokens !== void 0) {
-    metrics.prompt_cache_creation_tokens = llmAttr.usage?.promptCacheMissTokens;
+  if (modelAttr.usage?.promptCacheMissTokens !== void 0) {
+    metrics.prompt_cache_creation_tokens = modelAttr.usage?.promptCacheMissTokens;
   }
   return metrics;
 }
@@ -34,8 +35,8 @@ function normalizeUsageMetrics(llmAttr) {
 // src/ai-tracing.ts
 var DEFAULT_SPAN_TYPE = "task";
 var SPAN_TYPE_EXCEPTIONS = {
-  [AISpanType.LLM_GENERATION]: "llm",
-  [AISpanType.LLM_CHUNK]: "llm",
+  [AISpanType.MODEL_GENERATION]: "llm",
+  [AISpanType.MODEL_CHUNK]: "llm",
   [AISpanType.TOOL_CALL]: "tool",
   [AISpanType.MCP_TOOL_CALL]: "tool",
   [AISpanType.WORKFLOW_CONDITIONAL_EVAL]: "function",
@@ -224,17 +225,17 @@ var BraintrustExporter = class extends BaseExporter {
       ...span.metadata
     };
     const attributes = span.attributes ?? {};
-    if (span.type === AISpanType.LLM_GENERATION) {
-      const llmAttr = attributes;
-      if (llmAttr.model !== void 0) {
-        payload.metadata.model = llmAttr.model;
+    if (span.type === AISpanType.MODEL_GENERATION) {
+      const modelAttr = attributes;
+      if (modelAttr.model !== void 0) {
+        payload.metadata.model = modelAttr.model;
       }
-      if (llmAttr.provider !== void 0) {
-        payload.metadata.provider = llmAttr.provider;
+      if (modelAttr.provider !== void 0) {
+        payload.metadata.provider = modelAttr.provider;
       }
-      payload.metrics = normalizeUsageMetrics(llmAttr);
-      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 = omitKeys(attributes, ["model", "usage", "parameters"]);
       payload.metadata = {

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/metrics.ts","../src/ai-tracing.ts"],"names":[],"mappings":";;;;;;AAuBO,SAAS,sBAAsB,OAAA,EAA0D;AAC9F,EAAA,MAAM,UAAkC,EAAC;AAEzC,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,WAAA,KAAgB,MAAA,EAAW;AAC5C,IAAA,OAAA,CAAQ,aAAA,GAAgB,QAAQ,KAAA,EAAO,WAAA;AAAA,EACzC,CAAA,MAAA,IAAW,OAAA,CAAQ,KAAA,EAAO,YAAA,KAAiB,MAAA,EAAW;AACpD,IAAA,OAAA,CAAQ,aAAA,GAAgB,QAAQ,KAAA,EAAO,YAAA;AAAA,EACzC;AAEA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,YAAA,KAAiB,MAAA,EAAW;AAC7C,IAAA,OAAA,CAAQ,iBAAA,GAAoB,QAAQ,KAAA,EAAO,YAAA;AAAA,EAC7C,CAAA,MAAA,IAAW,OAAA,CAAQ,KAAA,EAAO,gBAAA,KAAqB,MAAA,EAAW;AACxD,IAAA,OAAA,CAAQ,iBAAA,GAAoB,QAAQ,KAAA,EAAO,gBAAA;AAAA,EAC7C;AAEA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,WAAA,KAAgB,MAAA,EAAW;AAC5C,IAAA,OAAA,CAAQ,MAAA,GAAS,QAAQ,KAAA,EAAO,WAAA;AAAA,EAClC;AACA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,eAAA,KAAoB,MAAA,EAAW;AAChD,IAAA,OAAA,CAAQ,2BAAA,GAA8B,QAAQ,KAAA,EAAO,eAAA;AAAA,EACvD;AACA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,oBAAA,KAAyB,MAAA,EAAW;AACrD,IAAA,OAAA,CAAQ,oBAAA,GAAuB,QAAQ,KAAA,EAAO,oBAAA;AAAA,EAChD;AACA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,qBAAA,KAA0B,MAAA,EAAW;AACtD,IAAA,OAAA,CAAQ,4BAAA,GAA+B,QAAQ,KAAA,EAAO,qBAAA;AAAA,EACxD;AAEA,EAAA,OAAO,OAAA;AACT;;;ACfA,IAAM,iBAAA,GAAoB,MAAA;AAG1B,IAAM,oBAAA,GAA4D;AAAA,EAChE,CAAC,UAAA,CAAW,cAAc,GAAG,KAAA;AAAA,EAC7B,CAAC,UAAA,CAAW,SAAS,GAAG,KAAA;AAAA,EACxB,CAAC,UAAA,CAAW,SAAS,GAAG,MAAA;AAAA,EACxB,CAAC,UAAA,CAAW,aAAa,GAAG,MAAA;AAAA,EAC5B,CAAC,UAAA,CAAW,yBAAyB,GAAG,UAAA;AAAA,EACxC,CAAC,UAAA,CAAW,mBAAmB,GAAG;AACpC,CAAA;AAGA,SAAS,YAAY,QAAA,EAA+E;AAClG,EAAA,OAAQ,oBAAA,CAAqB,QAAQ,CAAA,IAAa,iBAAA;AACpD;AAEO,IAAM,kBAAA,GAAN,cAAiC,YAAA,CAAa;AAAA,EACnD,IAAA,GAAO,YAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA,EAER,YAAY,MAAA,EAAkC;AAC5C,IAAA,KAAA,CAAM,MAAM,CAAA;AAEZ,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,YAAY,CAAA,sCAAA,EAAyC,CAAC,CAAC,MAAA,CAAO,MAAM,CAAA,CAAA,CAAG,CAAA;AAC5E,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAgB,aAAa,KAAA,EAAsC;AACjE,IAAA,IAAI,KAAA,CAAM,aAAa,OAAA,EAAS;AAC9B,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,YAAY,CAAA;AAC7C,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,YAAY,CAAA;AAC/C,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,KAAK,CAAA;AAC1D,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,IAAI,CAAA;AACzD,QAAA;AAAA;AACJ,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAwC;AACtE,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;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;AAE1C,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,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,EAAyB,KAAA,EAA+B;AAC1F,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;AAGA,MAAA,IAAI,QAAA,CAAS,SAAA,CAAU,IAAA,KAAS,CAAA,EAAG;AACjC,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,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,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,EAEA,MAAc,WAAW,IAAA,EAAwC;AAC/D,IAAA,MAAM,MAAA,GAAS,MAAM,UAAA,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,EAAE,MAAA,EAAQ,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,SAAA,kBAAW,IAAI,GAAA,IAAO,CAAA;AAAA,EACpF;AAAA,EAEQ,YAAY,OAAA,EAA4E;AAC9F,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,EAA8C;AACrE,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,KAAS,UAAA,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,aAAa,MAAA,EAAW;AAClC,QAAA,OAAA,CAAQ,QAAA,CAAS,WAAW,OAAA,CAAQ,QAAA;AAAA,MACtC;AAGA,MAAA,OAAA,CAAQ,OAAA,GAAU,sBAAsB,OAAO,CAAA;AAG/C,MAAA,IAAI,OAAA,CAAQ,eAAe,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,QAAA,CAAS,kBAAkB,OAAA,CAAQ,UAAA;AAAA,MAC7C;AAGA,MAAA,MAAM,kBAAkB,QAAA,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.js","sourcesContent":["import type { LLMGenerationAttributes } from '@mastra/core/ai-tracing';\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(llmAttr: LLMGenerationAttributes): BraintrustUsageMetrics {\n  const metrics: BraintrustUsageMetrics = {};\n\n  if (llmAttr.usage?.inputTokens !== undefined) {\n    metrics.prompt_tokens = llmAttr.usage?.inputTokens;\n  } else if (llmAttr.usage?.promptTokens !== undefined) {\n    metrics.prompt_tokens = llmAttr.usage?.promptTokens;\n  }\n\n  if (llmAttr.usage?.outputTokens !== undefined) {\n    metrics.completion_tokens = llmAttr.usage?.outputTokens;\n  } else if (llmAttr.usage?.completionTokens !== undefined) {\n    metrics.completion_tokens = llmAttr.usage?.completionTokens;\n  }\n\n  if (llmAttr.usage?.totalTokens !== undefined) {\n    metrics.tokens = llmAttr.usage?.totalTokens;\n  }\n  if (llmAttr.usage?.reasoningTokens !== undefined) {\n    metrics.completion_reasoning_tokens = llmAttr.usage?.reasoningTokens;\n  }\n  if (llmAttr.usage?.promptCacheHitTokens !== undefined) {\n    metrics.prompt_cached_tokens = llmAttr.usage?.promptCacheHitTokens;\n  }\n  if (llmAttr.usage?.promptCacheMissTokens !== undefined) {\n    metrics.prompt_cache_creation_tokens = llmAttr.usage?.promptCacheMissTokens;\n  }\n\n  return metrics;\n}\n","/**\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 { AISpanType, omitKeys, BaseExporter } from '@mastra/core/ai-tracing';\nimport type {\n  AITracingEvent,\n  AnyExportedAISpan,\n  LLMGenerationAttributes,\n  BaseExporterConfig,\n} from '@mastra/core/ai-tracing';\nimport { initLogger } from 'braintrust';\nimport type { Span, Logger } from 'braintrust';\nimport { normalizeUsageMetrics } from './metrics';\n\nexport interface BraintrustExporterConfig extends BaseExporterConfig {\n  /** Braintrust API key */\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>; // Braintrust logger (for root spans)\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};\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 extends BaseExporter {\n  name = 'braintrust';\n  private traceMap = new Map<string, SpanData>();\n  private config: BraintrustExporterConfig;\n\n  constructor(config: BraintrustExporterConfig) {\n    super(config);\n\n    if (!config.apiKey) {\n      this.setDisabled(`Missing required credentials (apiKey: ${!!config.apiKey})`);\n      this.config = null as any;\n      return;\n    }\n\n    this.config = config;\n  }\n\n  protected async _exportEvent(event: AITracingEvent): Promise<void> {\n    if (event.exportedSpan.isEvent) {\n      await this.handleEventSpan(event.exportedSpan);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.exportedSpan);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, true);\n        break;\n    }\n  }\n\n  private async handleSpanStarted(span: AnyExportedAISpan): 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    // 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    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      ...payload,\n    });\n\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyExportedAISpan, 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      if (spanData.activeIds.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('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      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  private async initLogger(span: AnyExportedAISpan): 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, { logger, spans: new Map(), activeIds: new Set() });\n  }\n\n  private getSpanData(options: { span: AnyExportedAISpan; 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: AnyExportedAISpan;\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: AnyExportedAISpan): 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      // Provider goes to metadata (if provided by attributes)\n      if (llmAttr.provider !== undefined) {\n        payload.metadata.provider = llmAttr.provider;\n      }\n\n      // Usage/token info goes to metrics\n      payload.metrics = normalizeUsageMetrics(llmAttr);\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    await super.shutdown();\n  }\n}\n"]}
+{"version":3,"sources":["../src/metrics.ts","../src/ai-tracing.ts"],"names":[],"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;;;AClBA,IAAM,iBAAA,GAAoB,MAAA;AAG1B,IAAM,oBAAA,GAA4D;AAAA,EAChE,CAAC,UAAA,CAAW,gBAAgB,GAAG,KAAA;AAAA,EAC/B,CAAC,UAAA,CAAW,WAAW,GAAG,KAAA;AAAA,EAC1B,CAAC,UAAA,CAAW,SAAS,GAAG,MAAA;AAAA,EACxB,CAAC,UAAA,CAAW,aAAa,GAAG,MAAA;AAAA,EAC5B,CAAC,UAAA,CAAW,yBAAyB,GAAG,UAAA;AAAA,EACxC,CAAC,UAAA,CAAW,mBAAmB,GAAG;AACpC,CAAA;AAGA,SAAS,YAAY,QAAA,EAA+E;AAClG,EAAA,OAAQ,oBAAA,CAAqB,QAAQ,CAAA,IAAa,iBAAA;AACpD;AAEO,IAAM,kBAAA,GAAN,cAAiC,YAAA,CAAa;AAAA,EACnD,IAAA,GAAO,YAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA,EAER,YAAY,MAAA,EAAkC;AAC5C,IAAA,KAAA,CAAM,MAAM,CAAA;AAEZ,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,YAAY,CAAA,sCAAA,EAAyC,CAAC,CAAC,MAAA,CAAO,MAAM,CAAA,CAAA,CAAG,CAAA;AAC5E,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAgB,aAAa,KAAA,EAAsC;AACjE,IAAA,IAAI,KAAA,CAAM,aAAa,OAAA,EAAS;AAC9B,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,YAAY,CAAA;AAC7C,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,YAAY,CAAA;AAC/C,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,KAAK,CAAA;AAC1D,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,IAAI,CAAA;AACzD,QAAA;AAAA;AACJ,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAwC;AACtE,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;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;AAE1C,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,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,EAAyB,KAAA,EAA+B;AAC1F,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;AAGA,MAAA,IAAI,QAAA,CAAS,SAAA,CAAU,IAAA,KAAS,CAAA,EAAG;AACjC,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,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,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,EAEA,MAAc,WAAW,IAAA,EAAwC;AAC/D,IAAA,MAAM,MAAA,GAAS,MAAM,UAAA,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,EAAE,MAAA,EAAQ,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,SAAA,kBAAW,IAAI,GAAA,IAAO,CAAA;AAAA,EACpF;AAAA,EAEQ,YAAY,OAAA,EAA4E;AAC9F,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,EAA8C;AACrE,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,KAAS,UAAA,CAAW,gBAAA,EAAkB;AAC7C,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,kBAAkB,QAAA,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.js","sourcesContent":["import type { ModelGenerationAttributes } from '@mastra/core/ai-tracing';\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 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 { AITracingEvent, AnyExportedAISpan, ModelGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, omitKeys } from '@mastra/core/ai-tracing';\nimport { BaseExporter } from '@mastra/core/ai-tracing/exporters';\nimport type { BaseExporterConfig } from '@mastra/core/ai-tracing/exporters';\nimport { initLogger } from 'braintrust';\nimport type { Span, Logger } from 'braintrust';\nimport { normalizeUsageMetrics } from './metrics';\n\nexport interface BraintrustExporterConfig extends BaseExporterConfig {\n  /** Braintrust API key */\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>; // Braintrust logger (for root spans)\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};\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.MODEL_GENERATION]: 'llm',\n  [AISpanType.MODEL_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 extends BaseExporter {\n  name = 'braintrust';\n  private traceMap = new Map<string, SpanData>();\n  private config: BraintrustExporterConfig;\n\n  constructor(config: BraintrustExporterConfig) {\n    super(config);\n\n    if (!config.apiKey) {\n      this.setDisabled(`Missing required credentials (apiKey: ${!!config.apiKey})`);\n      this.config = null as any;\n      return;\n    }\n\n    this.config = config;\n  }\n\n  protected async _exportEvent(event: AITracingEvent): Promise<void> {\n    if (event.exportedSpan.isEvent) {\n      await this.handleEventSpan(event.exportedSpan);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.exportedSpan);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, true);\n        break;\n    }\n  }\n\n  private async handleSpanStarted(span: AnyExportedAISpan): 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    // 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    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      ...payload,\n    });\n\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyExportedAISpan, 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      if (spanData.activeIds.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('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      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  private async initLogger(span: AnyExportedAISpan): 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, { logger, spans: new Map(), activeIds: new Set() });\n  }\n\n  private getSpanData(options: { span: AnyExportedAISpan; 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: AnyExportedAISpan;\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: AnyExportedAISpan): 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.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/metrics.d.ts

@@ -1,4 +1,4 @@
-import type { LLMGenerationAttributes } from '@mastra/core/ai-tracing';
+import type { ModelGenerationAttributes } from '@mastra/core/ai-tracing';
 /**
  * BraintrustUsageMetrics
  *
@@ -20,5 +20,5 @@ export interface BraintrustUsageMetrics {
     prompt_cache_creation_tokens?: number;
     [key: string]: number | undefined;
 }
-export declare function normalizeUsageMetrics(llmAttr: LLMGenerationAttributes): BraintrustUsageMetrics;
+export declare function normalizeUsageMetrics(modelAttr: ModelGenerationAttributes): BraintrustUsageMetrics;
 //# sourceMappingURL=metrics.d.ts.map

package/dist/metrics.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"metrics.d.ts","sourceRoot":"","sources":["../src/metrics.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AACvE;;;;;;;;;;;GAWG;AACH,MAAM,WAAW,sBAAsB;IACrC,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,2BAA2B,CAAC,EAAE,MAAM,CAAC;IACrC,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,4BAA4B,CAAC,EAAE,MAAM,CAAC;IACtC,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;CACnC;AAED,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,uBAAuB,GAAG,sBAAsB,CA6B9F"}
+{"version":3,"file":"metrics.d.ts","sourceRoot":"","sources":["../src/metrics.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACzE;;;;;;;;;;;GAWG;AACH,MAAM,WAAW,sBAAsB;IACrC,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,2BAA2B,CAAC,EAAE,MAAM,CAAC;IACrC,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,4BAA4B,CAAC,EAAE,MAAM,CAAC;IACtC,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;CACnC;AAED,wBAAgB,qBAAqB,CAAC,SAAS,EAAE,yBAAyB,GAAG,sBAAsB,CA6BlG"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/braintrust",
-  "version": "0.1.9",
+  "version": "0.2.0",
   "description": "Braintrust observability provider for Mastra - includes AI tracing and future observability features",
   "type": "module",
   "main": "dist/index.js",
@@ -33,12 +33,12 @@
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
     "vitest": "^3.2.4",
-    "@internal/lint": "0.0.52",
-    "@internal/types-builder": "0.0.27",
-    "@mastra/core": "0.22.1"
+    "@internal/lint": "0.0.54",
+    "@mastra/core": "0.23.0",
+    "@internal/types-builder": "0.0.29"
   },
   "peerDependencies": {
-    "@mastra/core": ">=0.18.1-0 <0.23.0-0"
+    "@mastra/core": ">=0.18.1-0 <0.24.0-0"
   },
   "homepage": "https://mastra.ai",
   "repository": {
@@ -49,6 +49,13 @@
   "bugs": {
     "url": "https://github.com/mastra-ai/mastra/issues"
   },
+  "publishConfig": {
+    "access": "public",
+    "publish-branch": [
+      "main",
+      "0.x"
+    ]
+  },
   "scripts": {
     "build": "tsup --silent --config tsup.config.ts",
     "build:watch": "pnpm build --watch",