@mastra/braintrust 0.0.0-issue-7498-20250905233741

package/CHANGELOG.md

@@ -0,0 +1,62 @@
+# @mastra/braintrust
+
+## 0.0.0-issue-7498-20250905233741
+
+### Patch Changes
+
+- Updated dependencies [0662d02]
+- Updated dependencies [6189844]
+- Updated dependencies [d7a8f59]
+- Updated dependencies [4dda259]
+- Updated dependencies [94edeff]
+- Updated dependencies [defed1c]
+- Updated dependencies [6991ced]
+- Updated dependencies [9cb9c42]
+- Updated dependencies [8334859]
+  - @mastra/core@0.0.0-issue-7498-20250905233741
+
+## 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
+
+- 376913a: Update peerdeps
+- Updated dependencies [8fbf79e]
+  - @mastra/core@0.16.0-alpha.1
+
+## 0.1.0-alpha.0
+
+### Minor Changes
+
+- 3fc9261: "Initial release of BraintrustExporter for ai-observability"
+
+### Patch Changes
+
+- 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-alpha.0

package/LICENSE.md

@@ -0,0 +1,15 @@
+# Apache License 2.0
+
+Copyright (c) 2025 Kepler Software, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

package/README.md

@@ -0,0 +1,47 @@
+# @mastra/braintrust
+
+Braintrust AI Observability exporter for Mastra applications.
+
+## Installation
+
+```bash
+npm install @mastra/braintrust
+```
+
+## Usage
+
+```typescript
+import { BraintrustExporter } from '@mastra/braintrust';
+
+// Use with Mastra
+const mastra = new Mastra({
+  ...,
+  observability: {
+    instances: {
+      braintrust: {
+        serviceName: 'service',
+        exporters: [
+          new BraintrustExporter({
+            apiKey: process.env.BRAINTRUST_API_KEY,
+            endpoint: process.env.BRAINTRUST_ENDPOINT, // optional
+          }),
+        ],
+      },
+    },
+  },
+});
+```
+
+## Features
+
+### AI Tracing
+
+- **Automatic span mapping**: Root spans become Braintrust traces
+- **Type-specific metadata**: Extracts relevant metadata for each span type (agents, tools, workflows)
+- **Error tracking**: Automatic error status and message tracking
+- **Hierarchical traces**: Maintains parent-child relationships
+- **Event span support**: Zero-duration spans for event-type traces
+
+## License
+
+Apache 2.0

package/dist/ai-tracing.d.ts

@@ -0,0 +1,35 @@
+/**
+ * Braintrust Exporter for Mastra AI Tracing
+ *
+ * This exporter sends tracing data to Braintrust for AI observability.
+ * Root spans become top-level Braintrust spans (no trace wrapper).
+ * Events are handled as zero-duration spans with matching start/end times.
+ */
+import type { AITracingExporter, AITracingEvent } from '@mastra/core/ai-tracing';
+export interface BraintrustExporterConfig {
+    /** Braintrust API key */
+    apiKey?: string;
+    /** Optional custom endpoint */
+    endpoint?: string;
+    /** Logger level for diagnostic messages (default: 'warn') */
+    logLevel?: 'debug' | 'info' | 'warn' | 'error';
+    /** Support tuning parameters */
+    tuningParameters?: Record<string, any>;
+}
+export declare class BraintrustExporter implements AITracingExporter {
+    name: string;
+    private traceMap;
+    private logger;
+    private config;
+    constructor(config: BraintrustExporterConfig);
+    exportEvent(event: AITracingEvent): Promise<void>;
+    private handleSpanStarted;
+    private handleSpanUpdateOrEnd;
+    private handleEventSpan;
+    private initLogger;
+    private getSpanData;
+    private getBraintrustParent;
+    private buildSpanPayload;
+    shutdown(): Promise<void>;
+}
+//# sourceMappingURL=ai-tracing.d.ts.map

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

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

package/dist/index.cjs

@@ -0,0 +1,244 @@
+'use strict';
+
+var aiTracing = require('@mastra/core/ai-tracing');
+var logger = require('@mastra/core/logger');
+var braintrust = require('braintrust');
+
+// 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.TOOL_CALL]: "tool",
+  [aiTracing.AISpanType.MCP_TOOL_CALL]: "tool",
+  [aiTracing.AISpanType.WORKFLOW_CONDITIONAL_EVAL]: "function",
+  [aiTracing.AISpanType.WORKFLOW_WAIT_EVENT]: "function"
+};
+function mapSpanType(spanType) {
+  return SPAN_TYPE_EXCEPTIONS[spanType] ?? DEFAULT_SPAN_TYPE;
+}
+var BraintrustExporter = class {
+  name = "braintrust";
+  traceMap = /* @__PURE__ */ new Map();
+  logger;
+  config;
+  constructor(config) {
+    this.logger = new logger.ConsoleLogger({ level: config.logLevel ?? "warn" });
+    if (!config.apiKey) {
+      this.logger.error("BraintrustExporter: Missing required credentials, exporter will be disabled", {
+        hasApiKey: !!config.apiKey
+      });
+      this.config = null;
+      return;
+    }
+    this.config = config;
+  }
+  async exportEvent(event) {
+    if (!this.config) {
+      return;
+    }
+    if (event.span.isEvent) {
+      await this.handleEventSpan(event.span);
+      return;
+    }
+    switch (event.type) {
+      case "span_started":
+        await this.handleSpanStarted(event.span);
+        break;
+      case "span_updated":
+        await this.handleSpanUpdateOrEnd(event.span, false);
+        break;
+      case "span_ended":
+        await this.handleSpanUpdateOrEnd(event.span, true);
+        break;
+    }
+  }
+  async handleSpanStarted(span) {
+    if (span.isRootSpan) {
+      await this.initLogger(span);
+    }
+    const method = "handleSpanStarted";
+    const spanData = this.getSpanData({ span, method });
+    if (!spanData) {
+      return;
+    }
+    const braintrustParent = this.getBraintrustParent({ spanData, span, method });
+    if (!braintrustParent) {
+      return;
+    }
+    const payload = this.buildSpanPayload(span);
+    const braintrustSpan = braintrustParent.startSpan({
+      name: span.name,
+      type: mapSpanType(span.type),
+      ...payload
+    });
+    spanData.spans.set(span.id, braintrustSpan);
+  }
+  async handleSpanUpdateOrEnd(span, isEnd) {
+    const method = isEnd ? "handleSpanEnd" : "handleSpanUpdate";
+    const spanData = this.getSpanData({ span, method });
+    if (!spanData) {
+      return;
+    }
+    const braintrustSpan = spanData.spans.get(span.id);
+    if (!braintrustSpan) {
+      this.logger.warn("Braintrust exporter: No Braintrust span found for span update/end", {
+        traceId: span.traceId,
+        spanId: span.id,
+        spanName: span.name,
+        spanType: span.type,
+        isRootSpan: span.isRootSpan,
+        parentSpanId: span.parent?.id,
+        method
+      });
+      return;
+    }
+    braintrustSpan.log(this.buildSpanPayload(span));
+    if (isEnd) {
+      if (span.endTime) {
+        braintrustSpan.end({ endTime: span.endTime.getTime() / 1e3 });
+      } else {
+        braintrustSpan.end();
+      }
+      if (span.isRootSpan) {
+        this.traceMap.delete(span.traceId);
+      }
+    }
+  }
+  async handleEventSpan(span) {
+    if (span.isRootSpan) {
+      this.logger.debug("Braintrust exporter: Creating logger for event", {
+        traceId: span.traceId,
+        spanId: span.id,
+        spanName: span.name,
+        method: "handleEventSpan"
+      });
+      await this.initLogger(span);
+    }
+    const method = "handleEventSpan";
+    const spanData = this.getSpanData({ span, method });
+    if (!spanData) {
+      return;
+    }
+    const braintrustParent = this.getBraintrustParent({ spanData, span, method });
+    if (!braintrustParent) {
+      return;
+    }
+    const payload = this.buildSpanPayload(span);
+    const braintrustSpan = braintrustParent.startSpan({
+      name: span.name,
+      type: mapSpanType(span.type),
+      startTime: span.startTime.getTime() / 1e3,
+      ...payload
+    });
+    braintrustSpan.end({ endTime: span.startTime.getTime() / 1e3 });
+    spanData.spans.set(span.id, braintrustSpan);
+  }
+  async initLogger(span) {
+    const logger = await braintrust.initLogger({
+      projectName: "mastra-tracing",
+      // TODO: Make this configurable
+      apiKey: this.config.apiKey,
+      appUrl: this.config.endpoint,
+      ...this.config.tuningParameters
+    });
+    this.traceMap.set(span.traceId, { logger, spans: /* @__PURE__ */ new Map() });
+  }
+  getSpanData(options) {
+    const { span, method } = options;
+    if (this.traceMap.has(span.traceId)) {
+      return this.traceMap.get(span.traceId);
+    }
+    this.logger.warn("Braintrust exporter: No span data found for span", {
+      traceId: span.traceId,
+      spanId: span.id,
+      spanName: span.name,
+      spanType: span.type,
+      isRootSpan: span.isRootSpan,
+      parentSpanId: span.parent?.id,
+      method
+    });
+  }
+  getBraintrustParent(options) {
+    const { spanData, span, method } = options;
+    const parentId = span.parent?.id;
+    if (!parentId) {
+      return spanData.logger;
+    }
+    if (spanData.spans.has(parentId)) {
+      return spanData.spans.get(parentId);
+    }
+    this.logger.warn("Braintrust exporter: No parent data found for span", {
+      traceId: span.traceId,
+      spanId: span.id,
+      spanName: span.name,
+      spanType: span.type,
+      isRootSpan: span.isRootSpan,
+      parentSpanId: span.parent?.id,
+      method
+    });
+  }
+  buildSpanPayload(span) {
+    const payload = {};
+    if (span.input !== void 0) {
+      payload.input = span.input;
+    }
+    if (span.output !== void 0) {
+      payload.output = span.output;
+    }
+    payload.metrics = {};
+    payload.metadata = {
+      spanType: span.type,
+      ...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 (llmAttr.usage !== void 0) {
+        payload.metrics = {
+          ...payload.metrics,
+          ...llmAttr.usage
+        };
+      }
+      if (llmAttr.parameters !== void 0) {
+        payload.metadata.modelParameters = llmAttr.parameters;
+      }
+      const otherAttributes = aiTracing.omitKeys(attributes, ["model", "usage", "parameters"]);
+      payload.metadata = {
+        ...payload.metadata,
+        ...otherAttributes
+      };
+    } else {
+      payload.metadata = {
+        ...payload.metadata,
+        ...attributes
+      };
+    }
+    if (span.errorInfo) {
+      payload.error = span.errorInfo.message;
+      payload.metadata.errorDetails = span.errorInfo;
+    }
+    if (Object.keys(payload.metrics).length === 0) {
+      delete payload.metrics;
+    }
+    return payload;
+  }
+  async shutdown() {
+    if (!this.config) {
+      return;
+    }
+    for (const [_traceId, spanData] of this.traceMap) {
+      for (const [_spanId, span] of spanData.spans) {
+        span.end();
+      }
+    }
+    this.traceMap.clear();
+  }
+};
+
+exports.BraintrustExporter = BraintrustExporter;
+//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

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

package/dist/index.d.ts

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

package/dist/index.d.ts.map

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

package/dist/index.js

@@ -0,0 +1,242 @@
+import { AISpanType, omitKeys } from '@mastra/core/ai-tracing';
+import { ConsoleLogger } from '@mastra/core/logger';
+import { initLogger } from 'braintrust';
+
+// src/ai-tracing.ts
+var DEFAULT_SPAN_TYPE = "task";
+var SPAN_TYPE_EXCEPTIONS = {
+  [AISpanType.LLM_GENERATION]: "llm",
+  [AISpanType.LLM_CHUNK]: "llm",
+  [AISpanType.TOOL_CALL]: "tool",
+  [AISpanType.MCP_TOOL_CALL]: "tool",
+  [AISpanType.WORKFLOW_CONDITIONAL_EVAL]: "function",
+  [AISpanType.WORKFLOW_WAIT_EVENT]: "function"
+};
+function mapSpanType(spanType) {
+  return SPAN_TYPE_EXCEPTIONS[spanType] ?? DEFAULT_SPAN_TYPE;
+}
+var BraintrustExporter = class {
+  name = "braintrust";
+  traceMap = /* @__PURE__ */ new Map();
+  logger;
+  config;
+  constructor(config) {
+    this.logger = new ConsoleLogger({ level: config.logLevel ?? "warn" });
+    if (!config.apiKey) {
+      this.logger.error("BraintrustExporter: Missing required credentials, exporter will be disabled", {
+        hasApiKey: !!config.apiKey
+      });
+      this.config = null;
+      return;
+    }
+    this.config = config;
+  }
+  async exportEvent(event) {
+    if (!this.config) {
+      return;
+    }
+    if (event.span.isEvent) {
+      await this.handleEventSpan(event.span);
+      return;
+    }
+    switch (event.type) {
+      case "span_started":
+        await this.handleSpanStarted(event.span);
+        break;
+      case "span_updated":
+        await this.handleSpanUpdateOrEnd(event.span, false);
+        break;
+      case "span_ended":
+        await this.handleSpanUpdateOrEnd(event.span, true);
+        break;
+    }
+  }
+  async handleSpanStarted(span) {
+    if (span.isRootSpan) {
+      await this.initLogger(span);
+    }
+    const method = "handleSpanStarted";
+    const spanData = this.getSpanData({ span, method });
+    if (!spanData) {
+      return;
+    }
+    const braintrustParent = this.getBraintrustParent({ spanData, span, method });
+    if (!braintrustParent) {
+      return;
+    }
+    const payload = this.buildSpanPayload(span);
+    const braintrustSpan = braintrustParent.startSpan({
+      name: span.name,
+      type: mapSpanType(span.type),
+      ...payload
+    });
+    spanData.spans.set(span.id, braintrustSpan);
+  }
+  async handleSpanUpdateOrEnd(span, isEnd) {
+    const method = isEnd ? "handleSpanEnd" : "handleSpanUpdate";
+    const spanData = this.getSpanData({ span, method });
+    if (!spanData) {
+      return;
+    }
+    const braintrustSpan = spanData.spans.get(span.id);
+    if (!braintrustSpan) {
+      this.logger.warn("Braintrust exporter: No Braintrust span found for span update/end", {
+        traceId: span.traceId,
+        spanId: span.id,
+        spanName: span.name,
+        spanType: span.type,
+        isRootSpan: span.isRootSpan,
+        parentSpanId: span.parent?.id,
+        method
+      });
+      return;
+    }
+    braintrustSpan.log(this.buildSpanPayload(span));
+    if (isEnd) {
+      if (span.endTime) {
+        braintrustSpan.end({ endTime: span.endTime.getTime() / 1e3 });
+      } else {
+        braintrustSpan.end();
+      }
+      if (span.isRootSpan) {
+        this.traceMap.delete(span.traceId);
+      }
+    }
+  }
+  async handleEventSpan(span) {
+    if (span.isRootSpan) {
+      this.logger.debug("Braintrust exporter: Creating logger for event", {
+        traceId: span.traceId,
+        spanId: span.id,
+        spanName: span.name,
+        method: "handleEventSpan"
+      });
+      await this.initLogger(span);
+    }
+    const method = "handleEventSpan";
+    const spanData = this.getSpanData({ span, method });
+    if (!spanData) {
+      return;
+    }
+    const braintrustParent = this.getBraintrustParent({ spanData, span, method });
+    if (!braintrustParent) {
+      return;
+    }
+    const payload = this.buildSpanPayload(span);
+    const braintrustSpan = braintrustParent.startSpan({
+      name: span.name,
+      type: mapSpanType(span.type),
+      startTime: span.startTime.getTime() / 1e3,
+      ...payload
+    });
+    braintrustSpan.end({ endTime: span.startTime.getTime() / 1e3 });
+    spanData.spans.set(span.id, braintrustSpan);
+  }
+  async initLogger(span) {
+    const logger = await initLogger({
+      projectName: "mastra-tracing",
+      // TODO: Make this configurable
+      apiKey: this.config.apiKey,
+      appUrl: this.config.endpoint,
+      ...this.config.tuningParameters
+    });
+    this.traceMap.set(span.traceId, { logger, spans: /* @__PURE__ */ new Map() });
+  }
+  getSpanData(options) {
+    const { span, method } = options;
+    if (this.traceMap.has(span.traceId)) {
+      return this.traceMap.get(span.traceId);
+    }
+    this.logger.warn("Braintrust exporter: No span data found for span", {
+      traceId: span.traceId,
+      spanId: span.id,
+      spanName: span.name,
+      spanType: span.type,
+      isRootSpan: span.isRootSpan,
+      parentSpanId: span.parent?.id,
+      method
+    });
+  }
+  getBraintrustParent(options) {
+    const { spanData, span, method } = options;
+    const parentId = span.parent?.id;
+    if (!parentId) {
+      return spanData.logger;
+    }
+    if (spanData.spans.has(parentId)) {
+      return spanData.spans.get(parentId);
+    }
+    this.logger.warn("Braintrust exporter: No parent data found for span", {
+      traceId: span.traceId,
+      spanId: span.id,
+      spanName: span.name,
+      spanType: span.type,
+      isRootSpan: span.isRootSpan,
+      parentSpanId: span.parent?.id,
+      method
+    });
+  }
+  buildSpanPayload(span) {
+    const payload = {};
+    if (span.input !== void 0) {
+      payload.input = span.input;
+    }
+    if (span.output !== void 0) {
+      payload.output = span.output;
+    }
+    payload.metrics = {};
+    payload.metadata = {
+      spanType: span.type,
+      ...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 (llmAttr.usage !== void 0) {
+        payload.metrics = {
+          ...payload.metrics,
+          ...llmAttr.usage
+        };
+      }
+      if (llmAttr.parameters !== void 0) {
+        payload.metadata.modelParameters = llmAttr.parameters;
+      }
+      const otherAttributes = omitKeys(attributes, ["model", "usage", "parameters"]);
+      payload.metadata = {
+        ...payload.metadata,
+        ...otherAttributes
+      };
+    } else {
+      payload.metadata = {
+        ...payload.metadata,
+        ...attributes
+      };
+    }
+    if (span.errorInfo) {
+      payload.error = span.errorInfo.message;
+      payload.metadata.errorDetails = span.errorInfo;
+    }
+    if (Object.keys(payload.metrics).length === 0) {
+      delete payload.metrics;
+    }
+    return payload;
+  }
+  async shutdown() {
+    if (!this.config) {
+      return;
+    }
+    for (const [_traceId, spanData] of this.traceMap) {
+      for (const [_spanId, span] of spanData.spans) {
+        span.end();
+      }
+    }
+    this.traceMap.clear();
+  }
+};
+
+export { BraintrustExporter };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

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

package/package.json

@@ -0,0 +1,59 @@
+{
+  "name": "@mastra/braintrust",
+  "version": "0.0.0-issue-7498-20250905233741",
+  "description": "Braintrust observability provider for Mastra - includes AI tracing and future observability features",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "CHANGELOG.md"
+  ],
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "require": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.cjs"
+      }
+    },
+    "./package.json": "./package.json"
+  },
+  "license": "Apache-2.0",
+  "dependencies": {
+    "braintrust": "^0.3.6"
+  },
+  "devDependencies": {
+    "@microsoft/api-extractor": "^7.52.8",
+    "@types/node": "^20.19.0",
+    "eslint": "^9.30.1",
+    "tsup": "^8.5.0",
+    "typescript": "^5.8.3",
+    "vitest": "^3.2.4",
+    "@mastra/core": "0.0.0-issue-7498-20250905233741",
+    "@internal/types-builder": "0.0.0-issue-7498-20250905233741",
+    "@internal/lint": "0.0.0-issue-7498-20250905233741"
+  },
+  "peerDependencies": {
+    "@mastra/core": "0.0.0-issue-7498-20250905233741"
+  },
+  "homepage": "https://mastra.ai",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/mastra-ai/mastra.git",
+    "directory": "observability/braintrust"
+  },
+  "bugs": {
+    "url": "https://github.com/mastra-ai/mastra/issues"
+  },
+  "scripts": {
+    "build": "tsup --silent --config tsup.config.ts",
+    "build:watch": "pnpm build --watch",
+    "test": "vitest run",
+    "test:watch": "vitest watch",
+    "lint": "eslint ."
+  }
+}