@mastra/braintrust 0.1.0-alpha.1 → 0.1.1-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,34 @@
 # @mastra/braintrust
 
+## 0.1.1-alpha.0
+
+### Patch Changes
+
+- add projectName as a top-level field and transform usage metrics for Braintrust ([#7596](https://github.com/mastra-ai/mastra/pull/7596))
+
+- Updated dependencies [[`47b6dc9`](https://github.com/mastra-ai/mastra/commit/47b6dc94f4976d4f3d3882e8f19eb365bbc5976c), [`565d65f`](https://github.com/mastra-ai/mastra/commit/565d65fc16314a99f081975ec92f2636dff0c86d), [`4da3d68`](https://github.com/mastra-ai/mastra/commit/4da3d68a778e5c4d5a17351ef223289fe2f45a45), [`0b0bbb2`](https://github.com/mastra-ai/mastra/commit/0b0bbb24f4198ead69792e92b68a350f52b45cf3), [`d951f41`](https://github.com/mastra-ai/mastra/commit/d951f41771e4e5da8da4b9f870949f9509e38756), [`8049e2e`](https://github.com/mastra-ai/mastra/commit/8049e2e8cce80a00353c64894c62b695ac34e35e)]:
+  - @mastra/core@0.16.1-alpha.1
+
+## 0.1.0
+
+### Minor Changes
+
+- 3fc9261: "Initial release of BraintrustExporter for ai-observability"
+
+### Patch Changes
+
+- 376913a: Update peerdeps
+- Updated dependencies [8fbf79e]
+- Updated dependencies [fd83526]
+- Updated dependencies [d0b90ab]
+- Updated dependencies [6f5eb7a]
+- Updated dependencies [a01cf14]
+- Updated dependencies [a9e50ee]
+- Updated dependencies [5397eb4]
+- Updated dependencies [c9f4e4a]
+- Updated dependencies [0acbc80]
+  - @mastra/core@0.16.0
+
 ## 0.1.0-alpha.1
 
 ### Patch Changes

package/README.md

@@ -23,6 +23,7 @@ const mastra = new Mastra({
         exporters: [
           new BraintrustExporter({
             apiKey: process.env.BRAINTRUST_API_KEY,
+            projectName: "mastra-tracing", // optional
             endpoint: process.env.BRAINTRUST_ENDPOINT, // optional
           }),
         ],

package/dist/ai-tracing.d.ts

@@ -11,6 +11,8 @@ export interface BraintrustExporterConfig {
     apiKey?: string;
     /** Optional custom endpoint */
     endpoint?: string;
+    /** Braintrust project name (default: 'mastra-tracing') */
+    projectName?: string;
     /** Logger level for diagnostic messages (default: 'warn') */
     logLevel?: 'debug' | 'info' | 'warn' | 'error';
     /** Support tuning parameters */

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

@@ -1 +1 @@
-{"version":3,"file":"ai-tracing.d.ts","sourceRoot":"","sources":["../src/ai-tracing.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAsC,MAAM,yBAAyB,CAAC;AAMrH,MAAM,WAAW,wBAAwB;IACvC,yBAAyB;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,+BAA+B;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,6DAA6D;IAC7D,QAAQ,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;IAC/C,gCAAgC;IAChC,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACxC;AAyBD,qBAAa,kBAAmB,YAAW,iBAAiB;IAC1D,IAAI,SAAgB;IACpB,OAAO,CAAC,QAAQ,CAA+B;IAC/C,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,MAAM,CAA2B;gBAE7B,MAAM,EAAE,wBAAwB;IActC,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;YAuBzC,iBAAiB;YA2BjB,qBAAqB;YAsCrB,eAAe;YAoCf,UAAU;IAWxB,OAAO,CAAC,WAAW;IAiBnB,OAAO,CAAC,mBAAmB;IA2B3B,OAAO,CAAC,gBAAgB;IAsElB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAchC"}
+{"version":3,"file":"ai-tracing.d.ts","sourceRoot":"","sources":["../src/ai-tracing.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAsC,MAAM,yBAAyB,CAAC;AAOrH,MAAM,WAAW,wBAAwB;IACvC,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,6DAA6D;IAC7D,QAAQ,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;IAC/C,gCAAgC;IAChC,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACxC;AAyBD,qBAAa,kBAAmB,YAAW,iBAAiB;IAC1D,IAAI,SAAgB;IACpB,OAAO,CAAC,QAAQ,CAA+B;IAC/C,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,MAAM,CAA2B;gBAE7B,MAAM,EAAE,wBAAwB;IActC,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;YAuBzC,iBAAiB;YA2BjB,qBAAqB;YAsCrB,eAAe;YAoCf,UAAU;IAWxB,OAAO,CAAC,WAAW;IAiBnB,OAAO,CAAC,mBAAmB;IA2B3B,OAAO,CAAC,gBAAgB;IAsElB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAchC"}

package/dist/index.cjs

@@ -4,6 +4,29 @@ var aiTracing = require('@mastra/core/ai-tracing');
 var logger = require('@mastra/core/logger');
 var braintrust = require('braintrust');
 
+// src/ai-tracing.ts
+
+// src/metrics.ts
+function normalizeUsageMetrics(llmAttr) {
+  const metrics = {};
+  if (llmAttr.usage?.promptTokens !== void 0) {
+    metrics.prompt_tokens = llmAttr.usage?.promptTokens;
+  }
+  if (llmAttr.usage?.completionTokens !== void 0) {
+    metrics.completion_tokens = llmAttr.usage?.completionTokens;
+  }
+  if (llmAttr.usage?.totalTokens !== void 0) {
+    metrics.tokens = llmAttr.usage?.totalTokens;
+  }
+  if (llmAttr.usage?.promptCacheHitTokens !== void 0) {
+    metrics.prompt_cached_tokens = llmAttr.usage?.promptCacheHitTokens;
+  }
+  if (llmAttr.usage?.promptCacheMissTokens !== void 0) {
+    metrics.prompt_cache_creation_tokens = llmAttr.usage?.promptCacheMissTokens;
+  }
+  return metrics;
+}
+
 // src/ai-tracing.ts
 var DEFAULT_SPAN_TYPE = "task";
 var SPAN_TYPE_EXCEPTIONS = {
@@ -136,8 +159,7 @@ var BraintrustExporter = class {
   }
   async initLogger(span) {
     const logger = await braintrust.initLogger({
-      projectName: "mastra-tracing",
-      // TODO: Make this configurable
+      projectName: this.config.projectName ?? "mastra-tracing",
       apiKey: this.config.apiKey,
       appUrl: this.config.endpoint,
       ...this.config.tuningParameters
@@ -197,12 +219,10 @@ var BraintrustExporter = class {
       if (llmAttr.model !== void 0) {
         payload.metadata.model = llmAttr.model;
       }
-      if (llmAttr.usage !== void 0) {
-        payload.metrics = {
-          ...payload.metrics,
-          ...llmAttr.usage
-        };
+      if (llmAttr.provider !== void 0) {
+        payload.metadata.provider = llmAttr.provider;
       }
+      payload.metrics = normalizeUsageMetrics(llmAttr);
       if (llmAttr.parameters !== void 0) {
         payload.metadata.modelParameters = llmAttr.parameters;
       }

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/ai-tracing.ts"],"names":["AISpanType","ConsoleLogger","initLogger","omitKeys"],"mappings":";;;;;;;AA+BA,IAAM,iBAAA,GAAoB,MAAA;AAG1B,IAAM,oBAAA,GAA4D;AAAA,EAChE,CAACA,oBAAA,CAAW,cAAc,GAAG,KAAA;AAAA,EAC7B,CAACA,oBAAA,CAAW,SAAS,GAAG,KAAA;AAAA,EACxB,CAACA,oBAAA,CAAW,SAAS,GAAG,MAAA;AAAA,EACxB,CAACA,oBAAA,CAAW,aAAa,GAAG,MAAA;AAAA,EAC5B,CAACA,oBAAA,CAAW,yBAAyB,GAAG,UAAA;AAAA,EACxC,CAACA,oBAAA,CAAW,mBAAmB,GAAG;AACpC,CAAA;AAGA,SAAS,YAAY,QAAA,EAA+E;AAClG,EAAA,OAAQ,oBAAA,CAAqB,QAAQ,CAAA,IAAa,iBAAA;AACpD;AAEO,IAAM,qBAAN,MAAsD;AAAA,EAC3D,IAAA,GAAO,YAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA,EACA,MAAA;AAAA,EAER,YAAY,MAAA,EAAkC;AAC5C,IAAA,IAAA,CAAK,MAAA,GAAS,IAAIC,oBAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AAEpE,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,6EAAA,EAA+E;AAAA,QAC/F,SAAA,EAAW,CAAC,CAAC,MAAA,CAAO;AAAA,OACrB,CAAA;AACD,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAChB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,KAAA,CAAM,KAAK,OAAA,EAAS;AACtB,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,IAAI,CAAA;AACrC,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA;AACJ,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAM,IAAA,CAAK,WAAW,IAAI,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,MAAA,GAAS,mBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAE1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,KAAA,EAA+B;AAClF,IAAA,MAAM,MAAA,GAAS,QAAQ,eAAA,GAAkB,kBAAA;AAEzC,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,cAAA,GAAiB,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,KAAK,mEAAA,EAAqE;AAAA,QACpF,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,YAAY,IAAA,CAAK,UAAA;AAAA,QACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,QAC3B;AAAA,OACD,CAAA;AACD,MAAA;AAAA,IACF;AAEA,IAAA,cAAA,CAAe,GAAA,CAAI,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAC,CAAA;AAE9C,IAAA,IAAI,KAAA,EAAO;AAET,MAAA,IAAI,KAAK,OAAA,EAAS;AAChB,QAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,QAAQ,OAAA,EAAQ,GAAI,KAAM,CAAA;AAAA,MAC/D,CAAA,MAAO;AACL,QAAA,cAAA,CAAe,GAAA,EAAI;AAAA,MACrB;AAEA,MAAA,IAAI,KAAK,UAAA,EAAY;AACnB,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,MACnC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAgC;AAC5D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,gDAAA,EAAkD;AAAA,QAClE,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,MAAA,EAAQ;AAAA,OACT,CAAA;AACD,MAAA,MAAM,IAAA,CAAK,WAAW,IAAI,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,MAAA,GAAS,iBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAG1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,SAAA,EAAW,IAAA,CAAK,SAAA,CAAU,OAAA,EAAQ,GAAI,GAAA;AAAA,MACtC,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,UAAU,OAAA,EAAQ,GAAI,KAAM,CAAA;AAC/D,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,WAAW,IAAA,EAAgC;AACvD,IAAA,MAAM,MAAA,GAAS,MAAMC,qBAAA,CAAW;AAAA,MAC9B,WAAA,EAAa,gBAAA;AAAA;AAAA,MACb,MAAA,EAAQ,KAAK,MAAA,CAAO,MAAA;AAAA,MACpB,MAAA,EAAQ,KAAK,MAAA,CAAO,QAAA;AAAA,MACpB,GAAG,KAAK,MAAA,CAAO;AAAA,KAChB,CAAA;AAED,IAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS,EAAE,QAAQ,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,CAAA;AAAA,EAC9D;AAAA,EAEQ,YAAY,OAAA,EAAoE;AACtF,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AACzB,IAAA,IAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA,EAAG;AACnC,MAAA,OAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACvC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,kDAAA,EAAoD;AAAA,MACnE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,oBAAoB,OAAA,EAIQ;AAClC,IAAA,MAAM,EAAE,QAAA,EAAU,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AAEnC,IAAA,MAAM,QAAA,GAAW,KAAK,MAAA,EAAQ,EAAA;AAC9B,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,QAAA,CAAS,MAAA;AAAA,IAClB;AAEA,IAAA,IAAI,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA,EAAG;AAChC,MAAA,OAAO,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA;AAAA,IACpC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,oDAAA,EAAsD;AAAA,MACrE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,iBAAiB,IAAA,EAAsC;AAC7D,IAAA,MAAM,UAA+B,EAAC;AAGtC,IAAA,IAAI,IAAA,CAAK,UAAU,MAAA,EAAW;AAC5B,MAAA,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACvB;AAEA,IAAA,IAAI,IAAA,CAAK,WAAW,MAAA,EAAW;AAC7B,MAAA,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AAAA,IACxB;AAGA,IAAA,OAAA,CAAQ,UAAU,EAAC;AACnB,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,IAAA,CAAK;AAAA,KACV;AAEA,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAExC,IAAA,IAAI,IAAA,CAAK,IAAA,KAASF,oBAAA,CAAW,cAAA,EAAgB;AAC3C,MAAA,MAAM,OAAA,GAAU,UAAA;AAGhB,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAA,CAAS,QAAQ,OAAA,CAAQ,KAAA;AAAA,MACnC;AAGA,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,OAAA,GAAU;AAAA,UAChB,GAAG,OAAA,CAAQ,OAAA;AAAA,UACX,GAAG,OAAA,CAAQ;AAAA,SACb;AAAA,MACF;AAGA,MAAA,IAAI,OAAA,CAAQ,eAAe,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,QAAA,CAAS,kBAAkB,OAAA,CAAQ,UAAA;AAAA,MAC7C;AAGA,MAAA,MAAM,kBAAkBG,kBAAA,CAAS,UAAA,EAAY,CAAC,OAAA,EAAS,OAAA,EAAS,YAAY,CAAC,CAAA;AAC7E,MAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,QACjB,GAAG,OAAA,CAAQ,QAAA;AAAA,QACX,GAAG;AAAA,OACL;AAAA,IACF,CAAA,MAAO;AAEL,MAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,QACjB,GAAG,OAAA,CAAQ,QAAA;AAAA,QACX,GAAG;AAAA,OACL;AAAA,IACF;AAGA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,OAAA,CAAQ,KAAA,GAAQ,KAAK,SAAA,CAAU,OAAA;AAC/B,MAAA,OAAA,CAAQ,QAAA,CAAS,eAAe,IAAA,CAAK,SAAA;AAAA,IACvC;AAGA,IAAA,IAAI,OAAO,IAAA,CAAK,OAAA,CAAQ,OAAO,CAAA,CAAE,WAAW,CAAA,EAAG;AAC7C,MAAA,OAAO,OAAA,CAAQ,OAAA;AAAA,IACjB;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAChB,MAAA;AAAA,IACF;AAGA,IAAA,KAAA,MAAW,CAAC,QAAA,EAAU,QAAQ,CAAA,IAAK,KAAK,QAAA,EAAU;AAChD,MAAA,KAAA,MAAW,CAAC,OAAA,EAAS,IAAI,CAAA,IAAK,SAAS,KAAA,EAAO;AAC5C,QAAA,IAAA,CAAK,GAAA,EAAI;AAAA,MACX;AAAA,IAEF;AACA,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AAAA,EACtB;AACF","file":"index.cjs","sourcesContent":["/**\n * Braintrust Exporter for Mastra AI Tracing\n *\n * This exporter sends tracing data to Braintrust for AI observability.\n * Root spans become top-level Braintrust spans (no trace wrapper).\n * Events are handled as zero-duration spans with matching start/end times.\n */\n\nimport type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, omitKeys } from '@mastra/core/ai-tracing';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { initLogger } from 'braintrust';\nimport type { Span, Logger } from 'braintrust';\n\nexport interface BraintrustExporterConfig {\n  /** Braintrust API key */\n  apiKey?: string;\n  /** Optional custom endpoint */\n  endpoint?: string;\n  /** Logger level for diagnostic messages (default: 'warn') */\n  logLevel?: 'debug' | 'info' | 'warn' | 'error';\n  /** Support tuning parameters */\n  tuningParameters?: Record<string, any>;\n}\n\ntype SpanData = {\n  logger: Logger<true>; // Braintrust logger (for root spans)\n  spans: Map<string, Span>; // Maps span.id to Braintrust span\n};\n\n// Default span type for all spans\nconst DEFAULT_SPAN_TYPE = 'task';\n\n// Exceptions to the default mapping\nconst SPAN_TYPE_EXCEPTIONS: Partial<Record<AISpanType, string>> = {\n  [AISpanType.LLM_GENERATION]: 'llm',\n  [AISpanType.LLM_CHUNK]: 'llm',\n  [AISpanType.TOOL_CALL]: 'tool',\n  [AISpanType.MCP_TOOL_CALL]: 'tool',\n  [AISpanType.WORKFLOW_CONDITIONAL_EVAL]: 'function',\n  [AISpanType.WORKFLOW_WAIT_EVENT]: 'function',\n};\n\n// Mapping function - returns valid Braintrust span types\nfunction mapSpanType(spanType: AISpanType): 'llm' | 'score' | 'function' | 'eval' | 'task' | 'tool' {\n  return (SPAN_TYPE_EXCEPTIONS[spanType] as any) ?? DEFAULT_SPAN_TYPE;\n}\n\nexport class BraintrustExporter implements AITracingExporter {\n  name = 'braintrust';\n  private traceMap = new Map<string, SpanData>();\n  private logger: ConsoleLogger;\n  private config: BraintrustExporterConfig;\n\n  constructor(config: BraintrustExporterConfig) {\n    this.logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n\n    if (!config.apiKey) {\n      this.logger.error('BraintrustExporter: Missing required credentials, exporter will be disabled', {\n        hasApiKey: !!config.apiKey,\n      });\n      this.config = null as any;\n      return;\n    }\n\n    this.config = config;\n  }\n\n  async exportEvent(event: AITracingEvent): Promise<void> {\n    if (!this.config) {\n      return;\n    }\n\n    if (event.span.isEvent) {\n      await this.handleEventSpan(event.span);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.span);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.span, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.span, true);\n        break;\n    }\n  }\n\n  private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      await this.initLogger(span);\n    }\n\n    const method = 'handleSpanStarted';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    const braintrustSpan = braintrustParent.startSpan({\n      name: span.name,\n      type: mapSpanType(span.type),\n      ...payload,\n    });\n\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyAISpan, isEnd: boolean): Promise<void> {\n    const method = isEnd ? 'handleSpanEnd' : 'handleSpanUpdate';\n\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustSpan = spanData.spans.get(span.id);\n    if (!braintrustSpan) {\n      this.logger.warn('Braintrust exporter: No Braintrust span found for span update/end', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        spanType: span.type,\n        isRootSpan: span.isRootSpan,\n        parentSpanId: span.parent?.id,\n        method,\n      });\n      return;\n    }\n\n    braintrustSpan.log(this.buildSpanPayload(span));\n\n    if (isEnd) {\n      // End the span with the correct endTime (convert milliseconds to seconds)\n      if (span.endTime) {\n        braintrustSpan.end({ endTime: span.endTime.getTime() / 1000 });\n      } else {\n        braintrustSpan.end();\n      }\n\n      if (span.isRootSpan) {\n        this.traceMap.delete(span.traceId);\n      }\n    }\n  }\n\n  private async handleEventSpan(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.logger.debug('Braintrust exporter: Creating logger for event', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        method: 'handleEventSpan',\n      });\n      await this.initLogger(span);\n    }\n\n    const method = 'handleEventSpan';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    // Create zero-duration span for event (convert milliseconds to seconds)\n    const braintrustSpan = braintrustParent.startSpan({\n      name: span.name,\n      type: mapSpanType(span.type),\n      startTime: span.startTime.getTime() / 1000,\n      ...payload,\n    });\n\n    braintrustSpan.end({ endTime: span.startTime.getTime() / 1000 });\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async initLogger(span: AnyAISpan): Promise<void> {\n    const logger = await initLogger({\n      projectName: 'mastra-tracing', // TODO: Make this configurable\n      apiKey: this.config.apiKey,\n      appUrl: this.config.endpoint,\n      ...this.config.tuningParameters,\n    });\n\n    this.traceMap.set(span.traceId, { logger, spans: new Map() });\n  }\n\n  private getSpanData(options: { span: AnyAISpan; method: string }): SpanData | undefined {\n    const { span, method } = options;\n    if (this.traceMap.has(span.traceId)) {\n      return this.traceMap.get(span.traceId);\n    }\n\n    this.logger.warn('Braintrust exporter: No span data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private getBraintrustParent(options: {\n    spanData: SpanData;\n    span: AnyAISpan;\n    method: string;\n  }): Logger<true> | Span | undefined {\n    const { spanData, span, method } = options;\n\n    const parentId = span.parent?.id;\n    if (!parentId) {\n      return spanData.logger;\n    }\n\n    if (spanData.spans.has(parentId)) {\n      return spanData.spans.get(parentId);\n    }\n\n    this.logger.warn('Braintrust exporter: No parent data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private buildSpanPayload(span: AnyAISpan): Record<string, any> {\n    const payload: Record<string, any> = {};\n\n    // Core span data\n    if (span.input !== undefined) {\n      payload.input = span.input;\n    }\n\n    if (span.output !== undefined) {\n      payload.output = span.output;\n    }\n\n    // Initialize metrics and metadata objects\n    payload.metrics = {};\n    payload.metadata = {\n      spanType: span.type,\n      ...span.metadata,\n    };\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    if (span.type === AISpanType.LLM_GENERATION) {\n      const llmAttr = attributes as LLMGenerationAttributes;\n\n      // Model goes to metadata\n      if (llmAttr.model !== undefined) {\n        payload.metadata.model = llmAttr.model;\n      }\n\n      // Usage/token info goes to metrics\n      if (llmAttr.usage !== undefined) {\n        payload.metrics = {\n          ...payload.metrics,\n          ...llmAttr.usage,\n        };\n      }\n\n      // Model parameters go to metadata\n      if (llmAttr.parameters !== undefined) {\n        payload.metadata.modelParameters = llmAttr.parameters;\n      }\n\n      // Other LLM attributes go to metadata\n      const otherAttributes = omitKeys(attributes, ['model', 'usage', 'parameters']);\n      payload.metadata = {\n        ...payload.metadata,\n        ...otherAttributes,\n      };\n    } else {\n      // For non-LLM spans, put all attributes in metadata\n      payload.metadata = {\n        ...payload.metadata,\n        ...attributes,\n      };\n    }\n\n    // Handle errors\n    if (span.errorInfo) {\n      payload.error = span.errorInfo.message;\n      payload.metadata.errorDetails = span.errorInfo;\n    }\n\n    // Clean up empty metrics object\n    if (Object.keys(payload.metrics).length === 0) {\n      delete payload.metrics;\n    }\n\n    return payload;\n  }\n\n  async shutdown(): Promise<void> {\n    if (!this.config) {\n      return;\n    }\n\n    // End all active spans\n    for (const [_traceId, spanData] of this.traceMap) {\n      for (const [_spanId, span] of spanData.spans) {\n        span.end();\n      }\n      // Loggers don't have an explicit shutdown method\n    }\n    this.traceMap.clear();\n  }\n}\n"]}
+{"version":3,"sources":["../src/metrics.ts","../src/ai-tracing.ts"],"names":["AISpanType","ConsoleLogger","initLogger","omitKeys"],"mappings":";;;;;;;;;AAuBO,SAAS,sBAAsB,OAAA,EAA0D;AAC9F,EAAA,MAAM,UAAkC,EAAC;AAEzC,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,YAAA,KAAiB,MAAA,EAAW;AAC7C,IAAA,OAAA,CAAQ,aAAA,GAAgB,QAAQ,KAAA,EAAO,YAAA;AAAA,EACzC;AACA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,gBAAA,KAAqB,MAAA,EAAW;AACjD,IAAA,OAAA,CAAQ,iBAAA,GAAoB,QAAQ,KAAA,EAAO,gBAAA;AAAA,EAC7C;AACA,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,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;;;ACTA,IAAM,iBAAA,GAAoB,MAAA;AAG1B,IAAM,oBAAA,GAA4D;AAAA,EAChE,CAACA,oBAAA,CAAW,cAAc,GAAG,KAAA;AAAA,EAC7B,CAACA,oBAAA,CAAW,SAAS,GAAG,KAAA;AAAA,EACxB,CAACA,oBAAA,CAAW,SAAS,GAAG,MAAA;AAAA,EACxB,CAACA,oBAAA,CAAW,aAAa,GAAG,MAAA;AAAA,EAC5B,CAACA,oBAAA,CAAW,yBAAyB,GAAG,UAAA;AAAA,EACxC,CAACA,oBAAA,CAAW,mBAAmB,GAAG;AACpC,CAAA;AAGA,SAAS,YAAY,QAAA,EAA+E;AAClG,EAAA,OAAQ,oBAAA,CAAqB,QAAQ,CAAA,IAAa,iBAAA;AACpD;AAEO,IAAM,qBAAN,MAAsD;AAAA,EAC3D,IAAA,GAAO,YAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA,EACA,MAAA;AAAA,EAER,YAAY,MAAA,EAAkC;AAC5C,IAAA,IAAA,CAAK,MAAA,GAAS,IAAIC,oBAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AAEpE,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,6EAAA,EAA+E;AAAA,QAC/F,SAAA,EAAW,CAAC,CAAC,MAAA,CAAO;AAAA,OACrB,CAAA;AACD,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAChB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,KAAA,CAAM,KAAK,OAAA,EAAS;AACtB,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,IAAI,CAAA;AACrC,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA;AACJ,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAM,IAAA,CAAK,WAAW,IAAI,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,MAAA,GAAS,mBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAE1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,KAAA,EAA+B;AAClF,IAAA,MAAM,MAAA,GAAS,QAAQ,eAAA,GAAkB,kBAAA;AAEzC,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,cAAA,GAAiB,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,KAAK,mEAAA,EAAqE;AAAA,QACpF,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,YAAY,IAAA,CAAK,UAAA;AAAA,QACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,QAC3B;AAAA,OACD,CAAA;AACD,MAAA;AAAA,IACF;AAEA,IAAA,cAAA,CAAe,GAAA,CAAI,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAC,CAAA;AAE9C,IAAA,IAAI,KAAA,EAAO;AAET,MAAA,IAAI,KAAK,OAAA,EAAS;AAChB,QAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,QAAQ,OAAA,EAAQ,GAAI,KAAM,CAAA;AAAA,MAC/D,CAAA,MAAO;AACL,QAAA,cAAA,CAAe,GAAA,EAAI;AAAA,MACrB;AAEA,MAAA,IAAI,KAAK,UAAA,EAAY;AACnB,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,MACnC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAgC;AAC5D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,gDAAA,EAAkD;AAAA,QAClE,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,MAAA,EAAQ;AAAA,OACT,CAAA;AACD,MAAA,MAAM,IAAA,CAAK,WAAW,IAAI,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,MAAA,GAAS,iBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAG1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,SAAA,EAAW,IAAA,CAAK,SAAA,CAAU,OAAA,EAAQ,GAAI,GAAA;AAAA,MACtC,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,UAAU,OAAA,EAAQ,GAAI,KAAM,CAAA;AAC/D,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,WAAW,IAAA,EAAgC;AACvD,IAAA,MAAM,MAAA,GAAS,MAAMC,qBAAA,CAAW;AAAA,MAC9B,WAAA,EAAa,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,QAAQ,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,CAAA;AAAA,EAC9D;AAAA,EAEQ,YAAY,OAAA,EAAoE;AACtF,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AACzB,IAAA,IAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA,EAAG;AACnC,MAAA,OAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACvC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,kDAAA,EAAoD;AAAA,MACnE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,oBAAoB,OAAA,EAIQ;AAClC,IAAA,MAAM,EAAE,QAAA,EAAU,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AAEnC,IAAA,MAAM,QAAA,GAAW,KAAK,MAAA,EAAQ,EAAA;AAC9B,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,QAAA,CAAS,MAAA;AAAA,IAClB;AAEA,IAAA,IAAI,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA,EAAG;AAChC,MAAA,OAAO,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA;AAAA,IACpC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,oDAAA,EAAsD;AAAA,MACrE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,iBAAiB,IAAA,EAAsC;AAC7D,IAAA,MAAM,UAA+B,EAAC;AAGtC,IAAA,IAAI,IAAA,CAAK,UAAU,MAAA,EAAW;AAC5B,MAAA,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACvB;AAEA,IAAA,IAAI,IAAA,CAAK,WAAW,MAAA,EAAW;AAC7B,MAAA,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AAAA,IACxB;AAGA,IAAA,OAAA,CAAQ,UAAU,EAAC;AACnB,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,IAAA,CAAK;AAAA,KACV;AAEA,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAExC,IAAA,IAAI,IAAA,CAAK,IAAA,KAASF,oBAAA,CAAW,cAAA,EAAgB;AAC3C,MAAA,MAAM,OAAA,GAAU,UAAA;AAGhB,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAA,CAAS,QAAQ,OAAA,CAAQ,KAAA;AAAA,MACnC;AAGA,MAAA,IAAI,OAAA,CAAQ,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;AAAA,EACtB;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?.promptTokens !== undefined) {\n    metrics.prompt_tokens = llmAttr.usage?.promptTokens;\n  }\n  if (llmAttr.usage?.completionTokens !== undefined) {\n    metrics.completion_tokens = llmAttr.usage?.completionTokens;\n  }\n  if (llmAttr.usage?.totalTokens !== undefined) {\n    metrics.tokens = llmAttr.usage?.totalTokens;\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 type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, omitKeys } from '@mastra/core/ai-tracing';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { initLogger } from 'braintrust';\nimport type { Span, Logger } from 'braintrust';\nimport { normalizeUsageMetrics } from './metrics';\n\nexport interface BraintrustExporterConfig {\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  /** Logger level for diagnostic messages (default: 'warn') */\n  logLevel?: 'debug' | 'info' | 'warn' | 'error';\n  /** Support tuning parameters */\n  tuningParameters?: Record<string, any>;\n}\n\ntype SpanData = {\n  logger: Logger<true>; // Braintrust logger (for root spans)\n  spans: Map<string, Span>; // Maps span.id to Braintrust span\n};\n\n// Default span type for all spans\nconst DEFAULT_SPAN_TYPE = 'task';\n\n// Exceptions to the default mapping\nconst SPAN_TYPE_EXCEPTIONS: Partial<Record<AISpanType, string>> = {\n  [AISpanType.LLM_GENERATION]: 'llm',\n  [AISpanType.LLM_CHUNK]: 'llm',\n  [AISpanType.TOOL_CALL]: 'tool',\n  [AISpanType.MCP_TOOL_CALL]: 'tool',\n  [AISpanType.WORKFLOW_CONDITIONAL_EVAL]: 'function',\n  [AISpanType.WORKFLOW_WAIT_EVENT]: 'function',\n};\n\n// Mapping function - returns valid Braintrust span types\nfunction mapSpanType(spanType: AISpanType): 'llm' | 'score' | 'function' | 'eval' | 'task' | 'tool' {\n  return (SPAN_TYPE_EXCEPTIONS[spanType] as any) ?? DEFAULT_SPAN_TYPE;\n}\n\nexport class BraintrustExporter implements AITracingExporter {\n  name = 'braintrust';\n  private traceMap = new Map<string, SpanData>();\n  private logger: ConsoleLogger;\n  private config: BraintrustExporterConfig;\n\n  constructor(config: BraintrustExporterConfig) {\n    this.logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n\n    if (!config.apiKey) {\n      this.logger.error('BraintrustExporter: Missing required credentials, exporter will be disabled', {\n        hasApiKey: !!config.apiKey,\n      });\n      this.config = null as any;\n      return;\n    }\n\n    this.config = config;\n  }\n\n  async exportEvent(event: AITracingEvent): Promise<void> {\n    if (!this.config) {\n      return;\n    }\n\n    if (event.span.isEvent) {\n      await this.handleEventSpan(event.span);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.span);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.span, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.span, true);\n        break;\n    }\n  }\n\n  private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      await this.initLogger(span);\n    }\n\n    const method = 'handleSpanStarted';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    const braintrustSpan = braintrustParent.startSpan({\n      name: span.name,\n      type: mapSpanType(span.type),\n      ...payload,\n    });\n\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyAISpan, isEnd: boolean): Promise<void> {\n    const method = isEnd ? 'handleSpanEnd' : 'handleSpanUpdate';\n\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustSpan = spanData.spans.get(span.id);\n    if (!braintrustSpan) {\n      this.logger.warn('Braintrust exporter: No Braintrust span found for span update/end', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        spanType: span.type,\n        isRootSpan: span.isRootSpan,\n        parentSpanId: span.parent?.id,\n        method,\n      });\n      return;\n    }\n\n    braintrustSpan.log(this.buildSpanPayload(span));\n\n    if (isEnd) {\n      // End the span with the correct endTime (convert milliseconds to seconds)\n      if (span.endTime) {\n        braintrustSpan.end({ endTime: span.endTime.getTime() / 1000 });\n      } else {\n        braintrustSpan.end();\n      }\n\n      if (span.isRootSpan) {\n        this.traceMap.delete(span.traceId);\n      }\n    }\n  }\n\n  private async handleEventSpan(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.logger.debug('Braintrust exporter: Creating logger for event', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        method: 'handleEventSpan',\n      });\n      await this.initLogger(span);\n    }\n\n    const method = 'handleEventSpan';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    // Create zero-duration span for event (convert milliseconds to seconds)\n    const braintrustSpan = braintrustParent.startSpan({\n      name: span.name,\n      type: mapSpanType(span.type),\n      startTime: span.startTime.getTime() / 1000,\n      ...payload,\n    });\n\n    braintrustSpan.end({ endTime: span.startTime.getTime() / 1000 });\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async initLogger(span: AnyAISpan): Promise<void> {\n    const logger = await initLogger({\n      projectName: 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() });\n  }\n\n  private getSpanData(options: { span: AnyAISpan; method: string }): SpanData | undefined {\n    const { span, method } = options;\n    if (this.traceMap.has(span.traceId)) {\n      return this.traceMap.get(span.traceId);\n    }\n\n    this.logger.warn('Braintrust exporter: No span data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private getBraintrustParent(options: {\n    spanData: SpanData;\n    span: AnyAISpan;\n    method: string;\n  }): Logger<true> | Span | undefined {\n    const { spanData, span, method } = options;\n\n    const parentId = span.parent?.id;\n    if (!parentId) {\n      return spanData.logger;\n    }\n\n    if (spanData.spans.has(parentId)) {\n      return spanData.spans.get(parentId);\n    }\n\n    this.logger.warn('Braintrust exporter: No parent data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private buildSpanPayload(span: AnyAISpan): Record<string, any> {\n    const payload: Record<string, any> = {};\n\n    // Core span data\n    if (span.input !== undefined) {\n      payload.input = span.input;\n    }\n\n    if (span.output !== undefined) {\n      payload.output = span.output;\n    }\n\n    // Initialize metrics and metadata objects\n    payload.metrics = {};\n    payload.metadata = {\n      spanType: span.type,\n      ...span.metadata,\n    };\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    if (span.type === AISpanType.LLM_GENERATION) {\n      const llmAttr = attributes as LLMGenerationAttributes;\n\n      // Model goes to metadata\n      if (llmAttr.model !== undefined) {\n        payload.metadata.model = llmAttr.model;\n      }\n\n      // 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  }\n}\n"]}

package/dist/index.js

@@ -2,6 +2,29 @@ import { AISpanType, omitKeys } from '@mastra/core/ai-tracing';
 import { ConsoleLogger } from '@mastra/core/logger';
 import { initLogger } from 'braintrust';
 
+// src/ai-tracing.ts
+
+// src/metrics.ts
+function normalizeUsageMetrics(llmAttr) {
+  const metrics = {};
+  if (llmAttr.usage?.promptTokens !== void 0) {
+    metrics.prompt_tokens = llmAttr.usage?.promptTokens;
+  }
+  if (llmAttr.usage?.completionTokens !== void 0) {
+    metrics.completion_tokens = llmAttr.usage?.completionTokens;
+  }
+  if (llmAttr.usage?.totalTokens !== void 0) {
+    metrics.tokens = llmAttr.usage?.totalTokens;
+  }
+  if (llmAttr.usage?.promptCacheHitTokens !== void 0) {
+    metrics.prompt_cached_tokens = llmAttr.usage?.promptCacheHitTokens;
+  }
+  if (llmAttr.usage?.promptCacheMissTokens !== void 0) {
+    metrics.prompt_cache_creation_tokens = llmAttr.usage?.promptCacheMissTokens;
+  }
+  return metrics;
+}
+
 // src/ai-tracing.ts
 var DEFAULT_SPAN_TYPE = "task";
 var SPAN_TYPE_EXCEPTIONS = {
@@ -134,8 +157,7 @@ var BraintrustExporter = class {
   }
   async initLogger(span) {
     const logger = await initLogger({
-      projectName: "mastra-tracing",
-      // TODO: Make this configurable
+      projectName: this.config.projectName ?? "mastra-tracing",
       apiKey: this.config.apiKey,
       appUrl: this.config.endpoint,
       ...this.config.tuningParameters
@@ -195,12 +217,10 @@ var BraintrustExporter = class {
       if (llmAttr.model !== void 0) {
         payload.metadata.model = llmAttr.model;
       }
-      if (llmAttr.usage !== void 0) {
-        payload.metrics = {
-          ...payload.metrics,
-          ...llmAttr.usage
-        };
+      if (llmAttr.provider !== void 0) {
+        payload.metadata.provider = llmAttr.provider;
       }
+      payload.metrics = normalizeUsageMetrics(llmAttr);
       if (llmAttr.parameters !== void 0) {
         payload.metadata.modelParameters = llmAttr.parameters;
       }

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/ai-tracing.ts"],"names":[],"mappings":";;;;;AA+BA,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,qBAAN,MAAsD;AAAA,EAC3D,IAAA,GAAO,YAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA,EACA,MAAA;AAAA,EAER,YAAY,MAAA,EAAkC;AAC5C,IAAA,IAAA,CAAK,MAAA,GAAS,IAAI,aAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AAEpE,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,6EAAA,EAA+E;AAAA,QAC/F,SAAA,EAAW,CAAC,CAAC,MAAA,CAAO;AAAA,OACrB,CAAA;AACD,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAChB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,KAAA,CAAM,KAAK,OAAA,EAAS;AACtB,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,IAAI,CAAA;AACrC,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA;AACJ,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAM,IAAA,CAAK,WAAW,IAAI,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,MAAA,GAAS,mBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAE1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,KAAA,EAA+B;AAClF,IAAA,MAAM,MAAA,GAAS,QAAQ,eAAA,GAAkB,kBAAA;AAEzC,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,cAAA,GAAiB,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,KAAK,mEAAA,EAAqE;AAAA,QACpF,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,YAAY,IAAA,CAAK,UAAA;AAAA,QACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,QAC3B;AAAA,OACD,CAAA;AACD,MAAA;AAAA,IACF;AAEA,IAAA,cAAA,CAAe,GAAA,CAAI,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAC,CAAA;AAE9C,IAAA,IAAI,KAAA,EAAO;AAET,MAAA,IAAI,KAAK,OAAA,EAAS;AAChB,QAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,QAAQ,OAAA,EAAQ,GAAI,KAAM,CAAA;AAAA,MAC/D,CAAA,MAAO;AACL,QAAA,cAAA,CAAe,GAAA,EAAI;AAAA,MACrB;AAEA,MAAA,IAAI,KAAK,UAAA,EAAY;AACnB,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,MACnC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAgC;AAC5D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,gDAAA,EAAkD;AAAA,QAClE,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,MAAA,EAAQ;AAAA,OACT,CAAA;AACD,MAAA,MAAM,IAAA,CAAK,WAAW,IAAI,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,MAAA,GAAS,iBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAG1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,SAAA,EAAW,IAAA,CAAK,SAAA,CAAU,OAAA,EAAQ,GAAI,GAAA;AAAA,MACtC,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,UAAU,OAAA,EAAQ,GAAI,KAAM,CAAA;AAC/D,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,WAAW,IAAA,EAAgC;AACvD,IAAA,MAAM,MAAA,GAAS,MAAM,UAAA,CAAW;AAAA,MAC9B,WAAA,EAAa,gBAAA;AAAA;AAAA,MACb,MAAA,EAAQ,KAAK,MAAA,CAAO,MAAA;AAAA,MACpB,MAAA,EAAQ,KAAK,MAAA,CAAO,QAAA;AAAA,MACpB,GAAG,KAAK,MAAA,CAAO;AAAA,KAChB,CAAA;AAED,IAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS,EAAE,QAAQ,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,CAAA;AAAA,EAC9D;AAAA,EAEQ,YAAY,OAAA,EAAoE;AACtF,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AACzB,IAAA,IAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA,EAAG;AACnC,MAAA,OAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACvC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,kDAAA,EAAoD;AAAA,MACnE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,oBAAoB,OAAA,EAIQ;AAClC,IAAA,MAAM,EAAE,QAAA,EAAU,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AAEnC,IAAA,MAAM,QAAA,GAAW,KAAK,MAAA,EAAQ,EAAA;AAC9B,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,QAAA,CAAS,MAAA;AAAA,IAClB;AAEA,IAAA,IAAI,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA,EAAG;AAChC,MAAA,OAAO,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA;AAAA,IACpC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,oDAAA,EAAsD;AAAA,MACrE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,iBAAiB,IAAA,EAAsC;AAC7D,IAAA,MAAM,UAA+B,EAAC;AAGtC,IAAA,IAAI,IAAA,CAAK,UAAU,MAAA,EAAW;AAC5B,MAAA,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACvB;AAEA,IAAA,IAAI,IAAA,CAAK,WAAW,MAAA,EAAW;AAC7B,MAAA,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AAAA,IACxB;AAGA,IAAA,OAAA,CAAQ,UAAU,EAAC;AACnB,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,IAAA,CAAK;AAAA,KACV;AAEA,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAExC,IAAA,IAAI,IAAA,CAAK,IAAA,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,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,OAAA,GAAU;AAAA,UAChB,GAAG,OAAA,CAAQ,OAAA;AAAA,UACX,GAAG,OAAA,CAAQ;AAAA,SACb;AAAA,MACF;AAGA,MAAA,IAAI,OAAA,CAAQ,eAAe,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,QAAA,CAAS,kBAAkB,OAAA,CAAQ,UAAA;AAAA,MAC7C;AAGA,MAAA,MAAM,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;AAAA,EACtB;AACF","file":"index.js","sourcesContent":["/**\n * Braintrust Exporter for Mastra AI Tracing\n *\n * This exporter sends tracing data to Braintrust for AI observability.\n * Root spans become top-level Braintrust spans (no trace wrapper).\n * Events are handled as zero-duration spans with matching start/end times.\n */\n\nimport type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, omitKeys } from '@mastra/core/ai-tracing';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { initLogger } from 'braintrust';\nimport type { Span, Logger } from 'braintrust';\n\nexport interface BraintrustExporterConfig {\n  /** Braintrust API key */\n  apiKey?: string;\n  /** Optional custom endpoint */\n  endpoint?: string;\n  /** Logger level for diagnostic messages (default: 'warn') */\n  logLevel?: 'debug' | 'info' | 'warn' | 'error';\n  /** Support tuning parameters */\n  tuningParameters?: Record<string, any>;\n}\n\ntype SpanData = {\n  logger: Logger<true>; // Braintrust logger (for root spans)\n  spans: Map<string, Span>; // Maps span.id to Braintrust span\n};\n\n// Default span type for all spans\nconst DEFAULT_SPAN_TYPE = 'task';\n\n// Exceptions to the default mapping\nconst SPAN_TYPE_EXCEPTIONS: Partial<Record<AISpanType, string>> = {\n  [AISpanType.LLM_GENERATION]: 'llm',\n  [AISpanType.LLM_CHUNK]: 'llm',\n  [AISpanType.TOOL_CALL]: 'tool',\n  [AISpanType.MCP_TOOL_CALL]: 'tool',\n  [AISpanType.WORKFLOW_CONDITIONAL_EVAL]: 'function',\n  [AISpanType.WORKFLOW_WAIT_EVENT]: 'function',\n};\n\n// Mapping function - returns valid Braintrust span types\nfunction mapSpanType(spanType: AISpanType): 'llm' | 'score' | 'function' | 'eval' | 'task' | 'tool' {\n  return (SPAN_TYPE_EXCEPTIONS[spanType] as any) ?? DEFAULT_SPAN_TYPE;\n}\n\nexport class BraintrustExporter implements AITracingExporter {\n  name = 'braintrust';\n  private traceMap = new Map<string, SpanData>();\n  private logger: ConsoleLogger;\n  private config: BraintrustExporterConfig;\n\n  constructor(config: BraintrustExporterConfig) {\n    this.logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n\n    if (!config.apiKey) {\n      this.logger.error('BraintrustExporter: Missing required credentials, exporter will be disabled', {\n        hasApiKey: !!config.apiKey,\n      });\n      this.config = null as any;\n      return;\n    }\n\n    this.config = config;\n  }\n\n  async exportEvent(event: AITracingEvent): Promise<void> {\n    if (!this.config) {\n      return;\n    }\n\n    if (event.span.isEvent) {\n      await this.handleEventSpan(event.span);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.span);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.span, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.span, true);\n        break;\n    }\n  }\n\n  private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      await this.initLogger(span);\n    }\n\n    const method = 'handleSpanStarted';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    const braintrustSpan = braintrustParent.startSpan({\n      name: span.name,\n      type: mapSpanType(span.type),\n      ...payload,\n    });\n\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyAISpan, isEnd: boolean): Promise<void> {\n    const method = isEnd ? 'handleSpanEnd' : 'handleSpanUpdate';\n\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustSpan = spanData.spans.get(span.id);\n    if (!braintrustSpan) {\n      this.logger.warn('Braintrust exporter: No Braintrust span found for span update/end', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        spanType: span.type,\n        isRootSpan: span.isRootSpan,\n        parentSpanId: span.parent?.id,\n        method,\n      });\n      return;\n    }\n\n    braintrustSpan.log(this.buildSpanPayload(span));\n\n    if (isEnd) {\n      // End the span with the correct endTime (convert milliseconds to seconds)\n      if (span.endTime) {\n        braintrustSpan.end({ endTime: span.endTime.getTime() / 1000 });\n      } else {\n        braintrustSpan.end();\n      }\n\n      if (span.isRootSpan) {\n        this.traceMap.delete(span.traceId);\n      }\n    }\n  }\n\n  private async handleEventSpan(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.logger.debug('Braintrust exporter: Creating logger for event', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        method: 'handleEventSpan',\n      });\n      await this.initLogger(span);\n    }\n\n    const method = 'handleEventSpan';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    // Create zero-duration span for event (convert milliseconds to seconds)\n    const braintrustSpan = braintrustParent.startSpan({\n      name: span.name,\n      type: mapSpanType(span.type),\n      startTime: span.startTime.getTime() / 1000,\n      ...payload,\n    });\n\n    braintrustSpan.end({ endTime: span.startTime.getTime() / 1000 });\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async initLogger(span: AnyAISpan): Promise<void> {\n    const logger = await initLogger({\n      projectName: 'mastra-tracing', // TODO: Make this configurable\n      apiKey: this.config.apiKey,\n      appUrl: this.config.endpoint,\n      ...this.config.tuningParameters,\n    });\n\n    this.traceMap.set(span.traceId, { logger, spans: new Map() });\n  }\n\n  private getSpanData(options: { span: AnyAISpan; method: string }): SpanData | undefined {\n    const { span, method } = options;\n    if (this.traceMap.has(span.traceId)) {\n      return this.traceMap.get(span.traceId);\n    }\n\n    this.logger.warn('Braintrust exporter: No span data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private getBraintrustParent(options: {\n    spanData: SpanData;\n    span: AnyAISpan;\n    method: string;\n  }): Logger<true> | Span | undefined {\n    const { spanData, span, method } = options;\n\n    const parentId = span.parent?.id;\n    if (!parentId) {\n      return spanData.logger;\n    }\n\n    if (spanData.spans.has(parentId)) {\n      return spanData.spans.get(parentId);\n    }\n\n    this.logger.warn('Braintrust exporter: No parent data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private buildSpanPayload(span: AnyAISpan): Record<string, any> {\n    const payload: Record<string, any> = {};\n\n    // Core span data\n    if (span.input !== undefined) {\n      payload.input = span.input;\n    }\n\n    if (span.output !== undefined) {\n      payload.output = span.output;\n    }\n\n    // Initialize metrics and metadata objects\n    payload.metrics = {};\n    payload.metadata = {\n      spanType: span.type,\n      ...span.metadata,\n    };\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    if (span.type === AISpanType.LLM_GENERATION) {\n      const llmAttr = attributes as LLMGenerationAttributes;\n\n      // Model goes to metadata\n      if (llmAttr.model !== undefined) {\n        payload.metadata.model = llmAttr.model;\n      }\n\n      // Usage/token info goes to metrics\n      if (llmAttr.usage !== undefined) {\n        payload.metrics = {\n          ...payload.metrics,\n          ...llmAttr.usage,\n        };\n      }\n\n      // Model parameters go to metadata\n      if (llmAttr.parameters !== undefined) {\n        payload.metadata.modelParameters = llmAttr.parameters;\n      }\n\n      // Other LLM attributes go to metadata\n      const otherAttributes = omitKeys(attributes, ['model', 'usage', 'parameters']);\n      payload.metadata = {\n        ...payload.metadata,\n        ...otherAttributes,\n      };\n    } else {\n      // For non-LLM spans, put all attributes in metadata\n      payload.metadata = {\n        ...payload.metadata,\n        ...attributes,\n      };\n    }\n\n    // Handle errors\n    if (span.errorInfo) {\n      payload.error = span.errorInfo.message;\n      payload.metadata.errorDetails = span.errorInfo;\n    }\n\n    // Clean up empty metrics object\n    if (Object.keys(payload.metrics).length === 0) {\n      delete payload.metrics;\n    }\n\n    return payload;\n  }\n\n  async shutdown(): Promise<void> {\n    if (!this.config) {\n      return;\n    }\n\n    // End all active spans\n    for (const [_traceId, spanData] of this.traceMap) {\n      for (const [_spanId, span] of spanData.spans) {\n        span.end();\n      }\n      // Loggers don't have an explicit shutdown method\n    }\n    this.traceMap.clear();\n  }\n}\n"]}
+{"version":3,"sources":["../src/metrics.ts","../src/ai-tracing.ts"],"names":[],"mappings":";;;;;;;AAuBO,SAAS,sBAAsB,OAAA,EAA0D;AAC9F,EAAA,MAAM,UAAkC,EAAC;AAEzC,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,YAAA,KAAiB,MAAA,EAAW;AAC7C,IAAA,OAAA,CAAQ,aAAA,GAAgB,QAAQ,KAAA,EAAO,YAAA;AAAA,EACzC;AACA,EAAA,IAAI,OAAA,CAAQ,KAAA,EAAO,gBAAA,KAAqB,MAAA,EAAW;AACjD,IAAA,OAAA,CAAQ,iBAAA,GAAoB,QAAQ,KAAA,EAAO,gBAAA;AAAA,EAC7C;AACA,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,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;;;ACTA,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,qBAAN,MAAsD;AAAA,EAC3D,IAAA,GAAO,YAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA,EACA,MAAA;AAAA,EAER,YAAY,MAAA,EAAkC;AAC5C,IAAA,IAAA,CAAK,MAAA,GAAS,IAAI,aAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AAEpE,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,6EAAA,EAA+E;AAAA,QAC/F,SAAA,EAAW,CAAC,CAAC,MAAA,CAAO;AAAA,OACrB,CAAA;AACD,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAChB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,KAAA,CAAM,KAAK,OAAA,EAAS;AACtB,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,IAAI,CAAA;AACrC,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA;AACJ,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAM,IAAA,CAAK,WAAW,IAAI,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,MAAA,GAAS,mBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAE1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,KAAA,EAA+B;AAClF,IAAA,MAAM,MAAA,GAAS,QAAQ,eAAA,GAAkB,kBAAA;AAEzC,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,cAAA,GAAiB,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,KAAK,mEAAA,EAAqE;AAAA,QACpF,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,YAAY,IAAA,CAAK,UAAA;AAAA,QACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,QAC3B;AAAA,OACD,CAAA;AACD,MAAA;AAAA,IACF;AAEA,IAAA,cAAA,CAAe,GAAA,CAAI,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAC,CAAA;AAE9C,IAAA,IAAI,KAAA,EAAO;AAET,MAAA,IAAI,KAAK,OAAA,EAAS;AAChB,QAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,QAAQ,OAAA,EAAQ,GAAI,KAAM,CAAA;AAAA,MAC/D,CAAA,MAAO;AACL,QAAA,cAAA,CAAe,GAAA,EAAI;AAAA,MACrB;AAEA,MAAA,IAAI,KAAK,UAAA,EAAY;AACnB,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,MACnC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAgC;AAC5D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,gDAAA,EAAkD;AAAA,QAClE,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,MAAA,EAAQ;AAAA,OACT,CAAA;AACD,MAAA,MAAM,IAAA,CAAK,WAAW,IAAI,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,MAAA,GAAS,iBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,mBAAmB,IAAA,CAAK,mBAAA,CAAoB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC5E,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAI,CAAA;AAG1C,IAAA,MAAM,cAAA,GAAiB,iBAAiB,SAAA,CAAU;AAAA,MAChD,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,IAAA,EAAM,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC3B,SAAA,EAAW,IAAA,CAAK,SAAA,CAAU,OAAA,EAAQ,GAAI,GAAA;AAAA,MACtC,GAAG;AAAA,KACJ,CAAA;AAED,IAAA,cAAA,CAAe,GAAA,CAAI,EAAE,OAAA,EAAS,IAAA,CAAK,UAAU,OAAA,EAAQ,GAAI,KAAM,CAAA;AAC/D,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,cAAc,CAAA;AAAA,EAC5C;AAAA,EAEA,MAAc,WAAW,IAAA,EAAgC;AACvD,IAAA,MAAM,MAAA,GAAS,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,QAAQ,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,CAAA;AAAA,EAC9D;AAAA,EAEQ,YAAY,OAAA,EAAoE;AACtF,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AACzB,IAAA,IAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA,EAAG;AACnC,MAAA,OAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACvC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,kDAAA,EAAoD;AAAA,MACnE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,oBAAoB,OAAA,EAIQ;AAClC,IAAA,MAAM,EAAE,QAAA,EAAU,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AAEnC,IAAA,MAAM,QAAA,GAAW,KAAK,MAAA,EAAQ,EAAA;AAC9B,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,QAAA,CAAS,MAAA;AAAA,IAClB;AAEA,IAAA,IAAI,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA,EAAG;AAChC,MAAA,OAAO,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA;AAAA,IACpC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,oDAAA,EAAsD;AAAA,MACrE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,YAAA,EAAc,KAAK,MAAA,EAAQ,EAAA;AAAA,MAC3B;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,iBAAiB,IAAA,EAAsC;AAC7D,IAAA,MAAM,UAA+B,EAAC;AAGtC,IAAA,IAAI,IAAA,CAAK,UAAU,MAAA,EAAW;AAC5B,MAAA,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACvB;AAEA,IAAA,IAAI,IAAA,CAAK,WAAW,MAAA,EAAW;AAC7B,MAAA,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AAAA,IACxB;AAGA,IAAA,OAAA,CAAQ,UAAU,EAAC;AACnB,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,IAAA,CAAK;AAAA,KACV;AAEA,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAExC,IAAA,IAAI,IAAA,CAAK,IAAA,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;AAAA,EACtB;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?.promptTokens !== undefined) {\n    metrics.prompt_tokens = llmAttr.usage?.promptTokens;\n  }\n  if (llmAttr.usage?.completionTokens !== undefined) {\n    metrics.completion_tokens = llmAttr.usage?.completionTokens;\n  }\n  if (llmAttr.usage?.totalTokens !== undefined) {\n    metrics.tokens = llmAttr.usage?.totalTokens;\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 type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, omitKeys } from '@mastra/core/ai-tracing';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { initLogger } from 'braintrust';\nimport type { Span, Logger } from 'braintrust';\nimport { normalizeUsageMetrics } from './metrics';\n\nexport interface BraintrustExporterConfig {\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  /** Logger level for diagnostic messages (default: 'warn') */\n  logLevel?: 'debug' | 'info' | 'warn' | 'error';\n  /** Support tuning parameters */\n  tuningParameters?: Record<string, any>;\n}\n\ntype SpanData = {\n  logger: Logger<true>; // Braintrust logger (for root spans)\n  spans: Map<string, Span>; // Maps span.id to Braintrust span\n};\n\n// Default span type for all spans\nconst DEFAULT_SPAN_TYPE = 'task';\n\n// Exceptions to the default mapping\nconst SPAN_TYPE_EXCEPTIONS: Partial<Record<AISpanType, string>> = {\n  [AISpanType.LLM_GENERATION]: 'llm',\n  [AISpanType.LLM_CHUNK]: 'llm',\n  [AISpanType.TOOL_CALL]: 'tool',\n  [AISpanType.MCP_TOOL_CALL]: 'tool',\n  [AISpanType.WORKFLOW_CONDITIONAL_EVAL]: 'function',\n  [AISpanType.WORKFLOW_WAIT_EVENT]: 'function',\n};\n\n// Mapping function - returns valid Braintrust span types\nfunction mapSpanType(spanType: AISpanType): 'llm' | 'score' | 'function' | 'eval' | 'task' | 'tool' {\n  return (SPAN_TYPE_EXCEPTIONS[spanType] as any) ?? DEFAULT_SPAN_TYPE;\n}\n\nexport class BraintrustExporter implements AITracingExporter {\n  name = 'braintrust';\n  private traceMap = new Map<string, SpanData>();\n  private logger: ConsoleLogger;\n  private config: BraintrustExporterConfig;\n\n  constructor(config: BraintrustExporterConfig) {\n    this.logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n\n    if (!config.apiKey) {\n      this.logger.error('BraintrustExporter: Missing required credentials, exporter will be disabled', {\n        hasApiKey: !!config.apiKey,\n      });\n      this.config = null as any;\n      return;\n    }\n\n    this.config = config;\n  }\n\n  async exportEvent(event: AITracingEvent): Promise<void> {\n    if (!this.config) {\n      return;\n    }\n\n    if (event.span.isEvent) {\n      await this.handleEventSpan(event.span);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.span);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.span, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.span, true);\n        break;\n    }\n  }\n\n  private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      await this.initLogger(span);\n    }\n\n    const method = 'handleSpanStarted';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    const braintrustSpan = braintrustParent.startSpan({\n      name: span.name,\n      type: mapSpanType(span.type),\n      ...payload,\n    });\n\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyAISpan, isEnd: boolean): Promise<void> {\n    const method = isEnd ? 'handleSpanEnd' : 'handleSpanUpdate';\n\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustSpan = spanData.spans.get(span.id);\n    if (!braintrustSpan) {\n      this.logger.warn('Braintrust exporter: No Braintrust span found for span update/end', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        spanType: span.type,\n        isRootSpan: span.isRootSpan,\n        parentSpanId: span.parent?.id,\n        method,\n      });\n      return;\n    }\n\n    braintrustSpan.log(this.buildSpanPayload(span));\n\n    if (isEnd) {\n      // End the span with the correct endTime (convert milliseconds to seconds)\n      if (span.endTime) {\n        braintrustSpan.end({ endTime: span.endTime.getTime() / 1000 });\n      } else {\n        braintrustSpan.end();\n      }\n\n      if (span.isRootSpan) {\n        this.traceMap.delete(span.traceId);\n      }\n    }\n  }\n\n  private async handleEventSpan(span: AnyAISpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.logger.debug('Braintrust exporter: Creating logger for event', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        method: 'handleEventSpan',\n      });\n      await this.initLogger(span);\n    }\n\n    const method = 'handleEventSpan';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const braintrustParent = this.getBraintrustParent({ spanData, span, method });\n    if (!braintrustParent) {\n      return;\n    }\n\n    const payload = this.buildSpanPayload(span);\n\n    // Create zero-duration span for event (convert milliseconds to seconds)\n    const braintrustSpan = braintrustParent.startSpan({\n      name: span.name,\n      type: mapSpanType(span.type),\n      startTime: span.startTime.getTime() / 1000,\n      ...payload,\n    });\n\n    braintrustSpan.end({ endTime: span.startTime.getTime() / 1000 });\n    spanData.spans.set(span.id, braintrustSpan);\n  }\n\n  private async initLogger(span: AnyAISpan): Promise<void> {\n    const logger = await initLogger({\n      projectName: 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() });\n  }\n\n  private getSpanData(options: { span: AnyAISpan; method: string }): SpanData | undefined {\n    const { span, method } = options;\n    if (this.traceMap.has(span.traceId)) {\n      return this.traceMap.get(span.traceId);\n    }\n\n    this.logger.warn('Braintrust exporter: No span data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private getBraintrustParent(options: {\n    spanData: SpanData;\n    span: AnyAISpan;\n    method: string;\n  }): Logger<true> | Span | undefined {\n    const { spanData, span, method } = options;\n\n    const parentId = span.parent?.id;\n    if (!parentId) {\n      return spanData.logger;\n    }\n\n    if (spanData.spans.has(parentId)) {\n      return spanData.spans.get(parentId);\n    }\n\n    this.logger.warn('Braintrust exporter: No parent data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parent?.id,\n      method,\n    });\n  }\n\n  private buildSpanPayload(span: AnyAISpan): Record<string, any> {\n    const payload: Record<string, any> = {};\n\n    // Core span data\n    if (span.input !== undefined) {\n      payload.input = span.input;\n    }\n\n    if (span.output !== undefined) {\n      payload.output = span.output;\n    }\n\n    // Initialize metrics and metadata objects\n    payload.metrics = {};\n    payload.metadata = {\n      spanType: span.type,\n      ...span.metadata,\n    };\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    if (span.type === AISpanType.LLM_GENERATION) {\n      const llmAttr = attributes as LLMGenerationAttributes;\n\n      // Model goes to metadata\n      if (llmAttr.model !== undefined) {\n        payload.metadata.model = llmAttr.model;\n      }\n\n      // 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  }\n}\n"]}

package/dist/metrics.d.ts

@@ -0,0 +1,24 @@
+import type { LLMGenerationAttributes } from '@mastra/core/ai-tracing';
+/**
+ * BraintrustUsageMetrics
+ *
+ * Canonical metric keys expected by Braintrust for LLM usage accounting.
+ * These map various provider/SDK-specific usage fields to a common schema.
+ * - prompt_tokens: input-side tokens (aka inputTokens/promptTokens)
+ * - completion_tokens: output-side tokens (aka outputTokens/completionTokens)
+ * - tokens: total tokens (provided or derived)
+ * - completion_reasoning_tokens: reasoning tokens, when available
+ * - prompt_cached_tokens: tokens served from cache (provider-specific)
+ * - prompt_cache_creation_tokens: tokens used to create cache (provider-specific)
+ */
+export interface BraintrustUsageMetrics {
+    prompt_tokens?: number;
+    completion_tokens?: number;
+    tokens?: number;
+    completion_reasoning_tokens?: number;
+    prompt_cached_tokens?: number;
+    prompt_cache_creation_tokens?: number;
+    [key: string]: number | undefined;
+}
+export declare function normalizeUsageMetrics(llmAttr: LLMGenerationAttributes): BraintrustUsageMetrics;
+//# sourceMappingURL=metrics.d.ts.map

package/dist/metrics.d.ts.map

@@ -0,0 +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,CAoB9F"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/braintrust",
-  "version": "0.1.0-alpha.1",
+  "version": "0.1.1-alpha.0",
   "description": "Braintrust observability provider for Mastra - includes AI tracing and future observability features",
   "type": "module",
   "main": "dist/index.js",
@@ -33,9 +33,9 @@
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
     "vitest": "^3.2.4",
-    "@internal/types-builder": "0.0.10",
-    "@mastra/core": "0.16.0-alpha.1",
-    "@internal/lint": "0.0.35"
+    "@internal/lint": "0.0.36",
+    "@mastra/core": "0.16.1-alpha.1",
+    "@internal/types-builder": "0.0.11"
   },
   "peerDependencies": {
     "@mastra/core": ">=0.15.3-0 <0.17.0-0"