@mastra/langsmith 1.0.0-beta.9 → 1.0.1-alpha.0

package/dist/index.js

@@ -1,6 +1,6 @@
 import { SpanType } from '@mastra/core/observability';
 import { omitKeys } from '@mastra/core/utils';
-import { BaseExporter } from '@mastra/observability';
+import { TrackingExporter } from '@mastra/observability';
 import { Client, RunTree } from 'langsmith';
 
 // src/tracing.ts
@@ -65,117 +65,88 @@ function mapSpanType(spanType) {
 function isKVMap(value) {
   return value != null && typeof value === "object" && !Array.isArray(value) && !(value instanceof Date);
 }
-var LangSmithExporter = class extends BaseExporter {
+var LangSmithExporter = class extends TrackingExporter {
   name = "langsmith";
-  traceMap = /* @__PURE__ */ new Map();
-  config;
-  client;
-  constructor(config) {
-    super(config);
-    config.apiKey = config.apiKey ?? process.env.LANGSMITH_API_KEY;
-    if (!config.apiKey) {
-      this.setDisabled(`Missing required credentials (apiKey: ${!!config.apiKey})`);
-      this.config = null;
-      this.client = null;
+  #client;
+  constructor(config = {}) {
+    const apiKey = config.apiKey ?? process.env.LANGSMITH_API_KEY;
+    super({
+      ...config,
+      apiKey
+    });
+    if (!apiKey) {
+      this.setDisabled(`Missing required credentials (apiKey: ${!!apiKey})`);
       return;
     }
-    this.client = config.client ?? new Client(config);
-    this.config = config;
+    this.#client = config.client ?? new Client(this.config);
   }
-  async _exportTracingEvent(event) {
-    if (event.exportedSpan.isEvent) {
-      await this.handleEventSpan(event.exportedSpan);
-      return;
-    }
-    switch (event.type) {
-      case "span_started":
-        await this.handleSpanStarted(event.exportedSpan);
-        break;
-      case "span_updated":
-        await this.handleSpanUpdateOrEnd(event.exportedSpan, false);
-        break;
-      case "span_ended":
-        await this.handleSpanUpdateOrEnd(event.exportedSpan, true);
-        break;
-    }
+  skipBuildRootTask = true;
+  async _buildRoot(_args) {
+    throw new Error("Method not implemented.");
   }
-  initializeRootSpan(span) {
-    if (this.traceMap.has(span.traceId)) {
-      this.logger.debug("LangSmith exporter: Reusing existing trace from local map", {
-        traceId: span.traceId,
-        spanId: span.id,
-        spanName: span.name
-      });
+  async _buildSpan(args) {
+    const { span, traceData } = args;
+    const parent = span.isRootSpan ? void 0 : traceData.getParent(args);
+    if (!span.isRootSpan && !parent) {
       return;
     }
-    this.traceMap.set(span.traceId, { spans: /* @__PURE__ */ new Map(), activeIds: /* @__PURE__ */ new Set() });
-  }
-  async handleSpanStarted(span) {
-    this.logger.debug("LangSmith exporter: handleSpanStarted", span.id, span.name);
-    if (span.isRootSpan) {
-      this.initializeRootSpan(span);
-    }
-    const method = "handleSpanStarted";
-    const spanData = this.getSpanData({ span, method });
-    if (!spanData) {
-      return;
-    }
-    if (!span.isEvent) {
-      spanData.activeIds.add(span.id);
-    }
     const payload = {
       name: span.name,
-      run_type: mapSpanType(span.type),
-      ...this.buildRunTreePayload(span)
+      ...this.buildRunTreePayload(span, true)
     };
-    const langsmithParent = this.getLangSmithParent({ spanData, span, method });
-    let langsmithRunTree;
-    if (!langsmithParent) {
-      langsmithRunTree = new RunTree(payload);
-    } else {
-      langsmithRunTree = langsmithParent.createChild(payload);
-    }
-    spanData.spans.set(span.id, langsmithRunTree);
-    await langsmithRunTree.postRun();
+    const langSmithSpan = span.isRootSpan ? new RunTree(payload) : parent.createChild(payload);
+    await langSmithSpan.postRun();
+    return langSmithSpan;
   }
-  async handleSpanUpdateOrEnd(span, isEnd) {
-    this.logger.debug("LangSmith exporter: handleSpanUpdateOrEnd", span.id, span.name, "isEnd:", isEnd);
-    const method = isEnd ? "handleSpanEnd" : "handleSpanUpdate";
-    const spanData = this.getSpanData({ span, method });
-    if (!spanData) {
+  async _buildEvent(args) {
+    const langSmithSpan = await this._buildSpan(args);
+    if (!langSmithSpan) {
       return;
     }
-    const langsmithRunTree = spanData.spans.get(span.id);
-    if (!langsmithRunTree) {
-      this.logger.warn("LangSmith exporter: No LangSmith span found for span update/end", {
-        traceId: span.traceId,
-        spanId: span.id,
-        spanName: span.name,
-        spanType: span.type,
-        isRootSpan: span.isRootSpan,
-        parentSpanId: span.parentSpanId,
-        method
-      });
+    await langSmithSpan.end({ endTime: args.span.startTime.getTime() });
+    await langSmithSpan.patchRun();
+    return langSmithSpan;
+  }
+  async _updateSpan(args) {
+    await this.handleSpanUpdateOrEnd({ ...args, isEnd: false });
+  }
+  async _finishSpan(args) {
+    await this.handleSpanUpdateOrEnd({ ...args, isEnd: true });
+  }
+  async _abortSpan(args) {
+    const { span, reason } = args;
+    span.error = reason.message;
+    span.metadata = {
+      ...span.metadata,
+      errorDetails: reason
+    };
+    await span.end();
+    await span.patchRun();
+  }
+  async handleSpanUpdateOrEnd(args) {
+    const { span, traceData, isEnd } = args;
+    const langSmithSpan = traceData.getSpan({ spanId: span.id });
+    if (!langSmithSpan) {
       return;
     }
     const updatePayload = this.buildRunTreePayload(span);
-    langsmithRunTree.metadata = {
-      ...langsmithRunTree.metadata,
+    langSmithSpan.metadata = {
+      ...langSmithSpan.metadata,
       ...updatePayload.metadata
     };
     if (updatePayload.inputs != null) {
-      langsmithRunTree.inputs = updatePayload.inputs;
+      langSmithSpan.inputs = updatePayload.inputs;
     }
     if (updatePayload.outputs != null) {
-      langsmithRunTree.outputs = updatePayload.outputs;
+      langSmithSpan.outputs = updatePayload.outputs;
     }
     if (updatePayload.error != null) {
-      langsmithRunTree.error = updatePayload.error;
+      langSmithSpan.error = updatePayload.error;
     }
     if (span.type === SpanType.MODEL_GENERATION) {
       const modelAttr = span.attributes ?? {};
       if (modelAttr.completionStartTime !== void 0) {
-        langsmithRunTree.addEvent({
+        langSmithSpan.addEvent({
           name: "new_token",
           time: modelAttr.completionStartTime.toISOString()
         });
@@ -183,96 +154,25 @@ var LangSmithExporter = class extends BaseExporter {
     }
     if (isEnd) {
       if (span.endTime) {
-        await langsmithRunTree.end({ endTime: span.endTime.getTime() / 1e3 });
+        await langSmithSpan.end({ endTime: span.endTime.getTime() });
       } else {
-        await langsmithRunTree.end();
-      }
-      await langsmithRunTree.patchRun();
-      if (!span.isEvent) {
-        spanData.activeIds.delete(span.id);
-      }
-      if (spanData.activeIds.size === 0) {
-        this.traceMap.delete(span.traceId);
+        await langSmithSpan.end();
       }
     }
+    await langSmithSpan.patchRun();
   }
-  async handleEventSpan(span) {
-    if (span.isRootSpan) {
-      this.logger.debug("LangSmith exporter: Creating logger for event", {
-        traceId: span.traceId,
-        spanId: span.id,
-        spanName: span.name,
-        method: "handleEventSpan"
-      });
-      this.initializeRootSpan(span);
-    }
-    const method = "handleEventSpan";
-    const spanData = this.getSpanData({ span, method });
-    if (!spanData) {
-      return;
-    }
-    const langsmithParent = this.getLangSmithParent({ spanData, span, method });
+  buildRunTreePayload(span, isNew = false) {
     const payload = {
-      ...this.buildRunTreePayload(span),
-      name: span.name,
-      type: mapSpanType(span.type),
-      startTime: span.startTime.getTime() / 1e3
-    };
-    let langsmithRunTree;
-    if (!langsmithParent) {
-      langsmithRunTree = new RunTree(payload);
-    } else {
-      langsmithRunTree = langsmithParent.createChild(payload);
-    }
-    await langsmithRunTree.postRun();
-    await langsmithRunTree.end({ endTime: span.startTime.getTime() / 1e3 });
-    await langsmithRunTree.patchRun();
-  }
-  getSpanData(options) {
-    const { span, method } = options;
-    if (this.traceMap.has(span.traceId)) {
-      return this.traceMap.get(span.traceId);
-    }
-    this.logger.warn("LangSmith exporter: No span data found for span", {
-      traceId: span.traceId,
-      spanId: span.id,
-      spanName: span.name,
-      spanType: span.type,
-      isRootSpan: span.isRootSpan,
-      parentSpanId: span.parentSpanId,
-      method
-    });
-  }
-  getLangSmithParent(options) {
-    const { spanData, span, method } = options;
-    const parentId = span.parentSpanId;
-    if (!parentId) {
-      return void 0;
-    }
-    if (spanData.spans.has(parentId)) {
-      return spanData.spans.get(parentId);
-    }
-    if (parentId && !spanData.spans.has(parentId)) {
-      return void 0;
-    }
-    this.logger.warn("LangSmith exporter: No parent data found for span", {
-      traceId: span.traceId,
-      spanId: span.id,
-      spanName: span.name,
-      spanType: span.type,
-      isRootSpan: span.isRootSpan,
-      parentSpanId: span.parentSpanId,
-      method
-    });
-  }
-  buildRunTreePayload(span) {
-    const payload = {
-      client: this.client,
+      client: this.#client,
       metadata: {
         mastra_span_type: span.type,
         ...span.metadata
       }
     };
+    if (isNew) {
+      payload.run_type = mapSpanType(span.type);
+      payload.start_time = span.startTime.getTime();
+    }
     if (this.config.projectName) {
       payload.project_name = this.config.projectName;
     }
@@ -298,7 +198,7 @@ var LangSmithExporter = class extends BaseExporter {
       if (modelAttr.parameters !== void 0) {
         payload.metadata.modelParameters = modelAttr.parameters;
       }
-      const otherAttributes = omitKeys(attributes, ["model", "usage", "parameters", "completionStartTime"]);
+      const otherAttributes = omitKeys(attributes, ["model", "provider", "usage", "parameters", "completionStartTime"]);
       payload.metadata = {
         ...payload.metadata,
         ...otherAttributes
@@ -315,19 +215,6 @@ var LangSmithExporter = class extends BaseExporter {
     }
     return payload;
   }
-  async shutdown() {
-    if (!this.config) {
-      return;
-    }
-    for (const [_traceId, spanData] of this.traceMap) {
-      for (const [_spanId, runTree] of spanData.spans) {
-        await runTree.end();
-        await runTree.patchRun();
-      }
-    }
-    this.traceMap.clear();
-    await super.shutdown();
-  }
 };
 
 export { LangSmithExporter };

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/metrics.ts","../src/tracing.ts"],"names":[],"mappings":";;;;;;;;AAuBO,SAAS,mBAAmB,KAAA,EAA2C;AAC5E,EAAA,MAAM,UAAiC,EAAC;AAExC,EAAA,IAAI,KAAA,EAAO,gBAAgB,MAAA,EAAW;AACpC,IAAA,OAAA,CAAQ,eAAe,KAAA,CAAM,WAAA;AAAA,EAC/B;AAEA,EAAA,IAAI,KAAA,EAAO,iBAAiB,MAAA,EAAW;AACrC,IAAA,OAAA,CAAQ,gBAAgB,KAAA,CAAM,YAAA;AAAA,EAChC;AAGA,EAAA,IAAI,OAAA,CAAQ,YAAA,KAAiB,MAAA,IAAa,OAAA,CAAQ,kBAAkB,MAAA,EAAW;AAC7E,IAAA,OAAA,CAAQ,YAAA,GAAe,OAAA,CAAQ,YAAA,GAAe,OAAA,CAAQ,aAAA;AAAA,EACxD;AAEA,EAAA,IAAI,KAAA,EAAO,aAAA,EAAe,SAAA,KAAc,MAAA,EAAW;AACjD,IAAA,OAAA,CAAQ,oBAAA,GAAuB;AAAA,MAC7B,GAAI,OAAA,CAAQ,oBAAA,IAAwB,EAAC;AAAA,MACrC,gBAAA,EAAkB,MAAM,aAAA,CAAc;AAAA,KACxC;AAAA,EACF;AAEA,EAAA,IAAI,KAAA,EAAO,YAAA,EAAc,SAAA,KAAc,MAAA,EAAW;AAChD,IAAA,OAAA,CAAQ,mBAAA,GAAsB;AAAA,MAC5B,GAAI,OAAA,CAAQ,mBAAA,IAAuB,EAAC;AAAA,MACpC,UAAA,EAAY,MAAM,YAAA,CAAa;AAAA,KACjC;AAAA,EACF;AAEA,EAAA,IAAI,KAAA,EAAO,YAAA,EAAc,UAAA,KAAe,MAAA,EAAW;AACjD,IAAA,OAAA,CAAQ,mBAAA,GAAsB;AAAA,MAC5B,GAAI,OAAA,CAAQ,mBAAA,IAAuB,EAAC;AAAA,MACpC,WAAA,EAAa,MAAM,YAAA,CAAa;AAAA,KAClC;AAAA,EACF;AAEA,EAAA,IAAI,KAAA,EAAO,YAAA,EAAc,KAAA,KAAU,MAAA,EAAW;AAC5C,IAAA,OAAA,CAAQ,mBAAA,GAAsB;AAAA,MAC5B,GAAI,OAAA,CAAQ,mBAAA,IAAuB,EAAC;AAAA,MACpC,KAAA,EAAO,MAAM,YAAA,CAAa;AAAA,KAC5B;AAAA,EACF;AAEA,EAAA,IAAI,KAAA,EAAO,aAAA,EAAe,KAAA,KAAU,MAAA,EAAW;AAC7C,IAAA,OAAA,CAAQ,oBAAA,GAAuB;AAAA,MAC7B,GAAI,OAAA,CAAQ,oBAAA,IAAwB,EAAC;AAAA,MACrC,KAAA,EAAO,MAAM,aAAA,CAAc;AAAA,KAC7B;AAAA,EACF;AAEA,EAAA,OAAO,OAAA;AACT;;;ACxCA,IAAM,iBAAA,GAAoB,OAAA;AAG1B,IAAM,oBAAA,GAA4E;AAAA,EAChF,CAAC,QAAA,CAAS,gBAAgB,GAAG,KAAA;AAAA,EAC7B,CAAC,QAAA,CAAS,SAAS,GAAG,MAAA;AAAA,EACtB,CAAC,QAAA,CAAS,aAAa,GAAG,MAAA;AAAA,EAC1B,CAAC,QAAA,CAAS,yBAAyB,GAAG,OAAA;AAAA,EACtC,CAAC,QAAA,CAAS,mBAAmB,GAAG;AAClC,CAAA;AAGA,SAAS,YAAY,QAAA,EAA8C;AACjE,EAAA,OAAO,oBAAA,CAAqB,QAAQ,CAAA,IAAK,iBAAA;AAC3C;AAEA,SAAS,QAAQ,KAAA,EAAgC;AAC/C,EAAA,OAAO,KAAA,IAAS,IAAA,IAAQ,OAAO,KAAA,KAAU,QAAA,IAAY,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,IAAK,EAAE,KAAA,YAAiB,IAAA,CAAA;AACnG;AAEO,IAAM,iBAAA,GAAN,cAAgC,YAAA,CAAa;AAAA,EAClD,IAAA,GAAO,WAAA;AAAA,EACC,QAAA,uBAAe,GAAA,EAAsB;AAAA,EACrC,MAAA;AAAA,EACA,MAAA;AAAA,EAER,YAAY,MAAA,EAAiC;AAC3C,IAAA,KAAA,CAAM,MAAM,CAAA;AAEZ,IAAA,MAAA,CAAO,MAAA,GAAS,MAAA,CAAO,MAAA,IAAU,OAAA,CAAQ,GAAA,CAAI,iBAAA;AAE7C,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,YAAY,CAAA,sCAAA,EAAyC,CAAC,CAAC,MAAA,CAAO,MAAM,CAAA,CAAA,CAAG,CAAA;AAC5E,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA,CAAO,MAAA,IAAU,IAAI,OAAO,MAAM,CAAA;AAChD,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAgB,oBAAoB,KAAA,EAAoC;AACtE,IAAA,IAAI,KAAA,CAAM,aAAa,OAAA,EAAS;AAC9B,MAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,YAAY,CAAA;AAC7C,MAAA;AAAA,IACF;AAEA,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,YAAY,CAAA;AAC/C,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,KAAK,CAAA;AAC1D,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,YAAA,EAAc,IAAI,CAAA;AACzD,QAAA;AAAA;AACJ,EACF;AAAA,EAEQ,mBAAmB,IAAA,EAAuB;AAEhD,IAAA,IAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA,EAAG;AACnC,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,2DAAA,EAA6D;AAAA,QAC7E,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK;AAAA,OAChB,CAAA;AACD,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS,EAAE,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,SAAA,kBAAW,IAAI,GAAA,IAAO,CAAA;AAAA,EAC5E;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAsC;AACpE,IAAA,IAAA,CAAK,OAAO,KAAA,CAAM,uCAAA,EAAyC,IAAA,CAAK,EAAA,EAAI,KAAK,IAAI,CAAA;AAC7E,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,mBAAmB,IAAI,CAAA;AAAA,IAC9B;AAEA,IAAA,MAAM,MAAA,GAAS,mBAAA;AACf,IAAA,MAAM,WAAW,IAAA,CAAK,WAAA,CAAY,EAAE,IAAA,EAAM,QAAQ,CAAA;AAClD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA;AAAA,IACF;AAGA,IAAA,IAAI,CAAC,KAAK,OAAA,EAAS;AACjB,MAAA,QAAA,CAAS,SAAA,CAAU,GAAA,CAAI,IAAA,CAAK,EAAE,CAAA;AAAA,IAChC;AAEA,IAAA,MAAM,OAAA,GAAU;AAAA,MACd,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,QAAA,EAAU,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AAAA,MAC/B,GAAG,IAAA,CAAK,mBAAA,CAAoB,IAAI;AAAA,KAClC;AAEA,IAAA,MAAM,kBAAkB,IAAA,CAAK,kBAAA,CAAmB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC1E,IAAA,IAAI,gBAAA;AACJ,IAAA,IAAI,CAAC,eAAA,EAAiB;AACpB,MAAA,gBAAA,GAAmB,IAAI,QAAQ,OAAO,CAAA;AAAA,IACxC,CAAA,MAAO;AACL,MAAA,gBAAA,GAAmB,eAAA,CAAgB,YAAY,OAAO,CAAA;AAAA,IACxD;AAEA,IAAA,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,gBAAgB,CAAA;AAE5C,IAAA,MAAM,iBAAiB,OAAA,EAAQ;AAAA,EACjC;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAuB,KAAA,EAA+B;AACxF,IAAA,IAAA,CAAK,MAAA,CAAO,MAAM,2CAAA,EAA6C,IAAA,CAAK,IAAI,IAAA,CAAK,IAAA,EAAM,UAAU,KAAK,CAAA;AAClG,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,gBAAA,GAAmB,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AACnD,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA,IAAA,CAAK,MAAA,CAAO,KAAK,iEAAA,EAAmE;AAAA,QAClF,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,YAAY,IAAA,CAAK,UAAA;AAAA,QACjB,cAAc,IAAA,CAAK,YAAA;AAAA,QACnB;AAAA,OACD,CAAA;AACD,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,aAAA,GAAgB,IAAA,CAAK,mBAAA,CAAoB,IAAI,CAAA;AACnD,IAAA,gBAAA,CAAiB,QAAA,GAAW;AAAA,MAC1B,GAAG,gBAAA,CAAiB,QAAA;AAAA,MACpB,GAAG,aAAA,CAAc;AAAA,KACnB;AACA,IAAA,IAAI,aAAA,CAAc,UAAU,IAAA,EAAM;AAChC,MAAA,gBAAA,CAAiB,SAAS,aAAA,CAAc,MAAA;AAAA,IAC1C;AACA,IAAA,IAAI,aAAA,CAAc,WAAW,IAAA,EAAM;AACjC,MAAA,gBAAA,CAAiB,UAAU,aAAA,CAAc,OAAA;AAAA,IAC3C;AACA,IAAA,IAAI,aAAA,CAAc,SAAS,IAAA,EAAM;AAC/B,MAAA,gBAAA,CAAiB,QAAQ,aAAA,CAAc,KAAA;AAAA,IACzC;AAGA,IAAA,IAAI,IAAA,CAAK,IAAA,KAAS,QAAA,CAAS,gBAAA,EAAkB;AAC3C,MAAA,MAAM,SAAA,GAAa,IAAA,CAAK,UAAA,IAAc,EAAC;AACvC,MAAA,IAAI,SAAA,CAAU,wBAAwB,MAAA,EAAW;AAC/C,QAAA,gBAAA,CAAiB,QAAA,CAAS;AAAA,UACxB,IAAA,EAAM,WAAA;AAAA,UACN,IAAA,EAAM,SAAA,CAAU,mBAAA,CAAoB,WAAA;AAAY,SACjD,CAAA;AAAA,MACH;AAAA,IACF;AAEA,IAAA,IAAI,KAAA,EAAO;AAET,MAAA,IAAI,KAAK,OAAA,EAAS;AAChB,QAAA,MAAM,gBAAA,CAAiB,IAAI,EAAE,OAAA,EAAS,KAAK,OAAA,CAAQ,OAAA,EAAQ,GAAI,GAAA,EAAM,CAAA;AAAA,MACvE,CAAA,MAAO;AACL,QAAA,MAAM,iBAAiB,GAAA,EAAI;AAAA,MAC7B;AACA,MAAA,MAAM,iBAAiB,QAAA,EAAS;AAGhC,MAAA,IAAI,CAAC,KAAK,OAAA,EAAS;AACjB,QAAA,QAAA,CAAS,SAAA,CAAU,MAAA,CAAO,IAAA,CAAK,EAAE,CAAA;AAAA,MACnC;AAGA,MAAA,IAAI,QAAA,CAAS,SAAA,CAAU,IAAA,KAAS,CAAA,EAAG;AACjC,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,MACnC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAsC;AAClE,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,+CAAA,EAAiD;AAAA,QACjE,SAAS,IAAA,CAAK,OAAA;AAAA,QACd,QAAQ,IAAA,CAAK,EAAA;AAAA,QACb,UAAU,IAAA,CAAK,IAAA;AAAA,QACf,MAAA,EAAQ;AAAA,OACT,CAAA;AACD,MAAA,IAAA,CAAK,mBAAmB,IAAI,CAAA;AAAA,IAC9B;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,kBAAkB,IAAA,CAAK,kBAAA,CAAmB,EAAE,QAAA,EAAU,IAAA,EAAM,QAAQ,CAAA;AAC1E,IAAA,MAAM,OAAA,GAAU;AAAA,MACd,GAAG,IAAA,CAAK,mBAAA,CAAoB,IAAI,CAAA;AAAA,MAChC,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;AAAA,KACxC;AAEA,IAAA,IAAI,gBAAA;AACJ,IAAA,IAAI,CAAC,eAAA,EAAiB;AACpB,MAAA,gBAAA,GAAmB,IAAI,QAAQ,OAAO,CAAA;AAAA,IACxC,CAAA,MAAO;AACL,MAAA,gBAAA,GAAmB,eAAA,CAAgB,YAAY,OAAO,CAAA;AAAA,IACxD;AAEA,IAAA,MAAM,iBAAiB,OAAA,EAAQ;AAE/B,IAAA,MAAM,gBAAA,CAAiB,IAAI,EAAE,OAAA,EAAS,KAAK,SAAA,CAAU,OAAA,EAAQ,GAAI,GAAA,EAAM,CAAA;AACvE,IAAA,MAAM,iBAAiB,QAAA,EAAS;AAAA,EAClC;AAAA,EAEQ,YAAY,OAAA,EAA0E;AAC5F,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AACzB,IAAA,IAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA,EAAG;AACnC,MAAA,OAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACvC;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,iDAAA,EAAmD;AAAA,MAClE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,cAAc,IAAA,CAAK,YAAA;AAAA,MACnB;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,mBAAmB,OAAA,EAIH;AACtB,IAAA,MAAM,EAAE,QAAA,EAAU,IAAA,EAAM,MAAA,EAAO,GAAI,OAAA;AAEnC,IAAA,MAAM,WAAW,IAAA,CAAK,YAAA;AACtB,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,MAAA;AAAA,IACT;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,IAAI,YAAY,CAAC,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,QAAQ,CAAA,EAAG;AAG7C,MAAA,OAAO,MAAA;AAAA,IACT;AAEA,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,mDAAA,EAAqD;AAAA,MACpE,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,QAAQ,IAAA,CAAK,EAAA;AAAA,MACb,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,YAAY,IAAA,CAAK,UAAA;AAAA,MACjB,cAAc,IAAA,CAAK,YAAA;AAAA,MACnB;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEQ,oBAAoB,IAAA,EAA+C;AACzE,IAAA,MAAM,OAAA,GAAwD;AAAA,MAC5D,QAAQ,IAAA,CAAK,MAAA;AAAA,MACb,QAAA,EAAU;AAAA,QACR,kBAAkB,IAAA,CAAK,IAAA;AAAA,QACvB,GAAG,IAAA,CAAK;AAAA;AACV,KACF;AAGA,IAAA,IAAI,IAAA,CAAK,OAAO,WAAA,EAAa;AAC3B,MAAA,OAAA,CAAQ,YAAA,GAAe,KAAK,MAAA,CAAO,WAAA;AAAA,IACrC;AAGA,IAAA,IAAI,IAAA,CAAK,UAAA,IAAc,IAAA,CAAK,IAAA,EAAM,MAAA,EAAQ;AACxC,MAAA,OAAA,CAAQ,OAAO,IAAA,CAAK,IAAA;AAAA,IACtB;AAGA,IAAA,IAAI,IAAA,CAAK,UAAU,MAAA,EAAW;AAC5B,MAAA,OAAA,CAAQ,MAAA,GAAS,OAAA,CAAQ,IAAA,CAAK,KAAK,CAAA,GAAI,KAAK,KAAA,GAAQ,EAAE,KAAA,EAAO,IAAA,CAAK,KAAA,EAAM;AAAA,IAC1E;AAEA,IAAA,IAAI,IAAA,CAAK,WAAW,MAAA,EAAW;AAC7B,MAAA,OAAA,CAAQ,OAAA,GAAU,OAAA,CAAQ,IAAA,CAAK,MAAM,CAAA,GAAI,KAAK,MAAA,GAAS,EAAE,MAAA,EAAQ,IAAA,CAAK,MAAA,EAAO;AAAA,IAC/E;AAEA,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAExC,IAAA,IAAI,IAAA,CAAK,IAAA,KAAS,QAAA,CAAS,gBAAA,EAAkB;AAC3C,MAAA,MAAM,SAAA,GAAY,UAAA;AAGlB,MAAA,IAAI,SAAA,CAAU,UAAU,MAAA,EAAW;AAGjC,QAAA,OAAA,CAAQ,QAAA,CAAS,gBAAgB,SAAA,CAAU,KAAA;AAAA,MAC7C;AAGA,MAAA,IAAI,SAAA,CAAU,aAAa,MAAA,EAAW;AAGpC,QAAA,OAAA,CAAQ,QAAA,CAAS,cAAc,SAAA,CAAU,QAAA;AAAA,MAC3C;AAGA,MAAA,OAAA,CAAQ,QAAA,CAAS,cAAA,GAAiB,kBAAA,CAAmB,SAAA,CAAU,KAAK,CAAA;AAGpE,MAAA,IAAI,SAAA,CAAU,eAAe,MAAA,EAAW;AACtC,QAAA,OAAA,CAAQ,QAAA,CAAS,kBAAkB,SAAA,CAAU,UAAA;AAAA,MAC/C;AAGA,MAAA,MAAM,eAAA,GAAkB,SAAS,UAAA,EAAY,CAAC,SAAS,OAAA,EAAS,YAAA,EAAc,qBAAqB,CAAC,CAAA;AACpG,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;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,OAAO,CAAA,IAAK,SAAS,KAAA,EAAO;AAC/C,QAAA,MAAM,QAAQ,GAAA,EAAI;AAClB,QAAA,MAAM,QAAQ,QAAA,EAAS;AAAA,MACzB;AAAA,IACF;AACA,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AACpB,IAAA,MAAM,MAAM,QAAA,EAAS;AAAA,EACvB;AACF","file":"index.js","sourcesContent":["import type { UsageStats } from '@mastra/core/observability';\n\n/**\n * LangSmithUsageMetrics\n *\n * Canonical metric keys expected by LangSmith for LLM usage accounting.\n * See: https://docs.langchain.com/langsmith/log-llm-trace#provide-token-and-cost-information\n */\nexport interface LangSmithUsageMetrics {\n  input_tokens?: number;\n  output_tokens?: number;\n  total_tokens?: number;\n  input_token_details?: {\n    [key: string]: number;\n  };\n  output_token_details?: {\n    [key: string]: number;\n  };\n}\n\n/**\n * Formats UsageStats to LangSmith's expected metric format.\n */\nexport function formatUsageMetrics(usage?: UsageStats): LangSmithUsageMetrics {\n  const metrics: LangSmithUsageMetrics = {};\n\n  if (usage?.inputTokens !== undefined) {\n    metrics.input_tokens = usage.inputTokens;\n  }\n\n  if (usage?.outputTokens !== undefined) {\n    metrics.output_tokens = usage.outputTokens;\n  }\n\n  // Compute total if we have both\n  if (metrics.input_tokens !== undefined && metrics.output_tokens !== undefined) {\n    metrics.total_tokens = metrics.input_tokens + metrics.output_tokens;\n  }\n\n  if (usage?.outputDetails?.reasoning !== undefined) {\n    metrics.output_token_details = {\n      ...(metrics.output_token_details ?? {}),\n      reasoning_tokens: usage.outputDetails.reasoning,\n    };\n  }\n\n  if (usage?.inputDetails?.cacheRead !== undefined) {\n    metrics.input_token_details = {\n      ...(metrics.input_token_details ?? {}),\n      cache_read: usage.inputDetails.cacheRead,\n    };\n  }\n\n  if (usage?.inputDetails?.cacheWrite !== undefined) {\n    metrics.input_token_details = {\n      ...(metrics.input_token_details ?? {}),\n      cache_write: usage.inputDetails.cacheWrite,\n    };\n  }\n\n  if (usage?.inputDetails?.audio !== undefined) {\n    metrics.input_token_details = {\n      ...(metrics.input_token_details ?? {}),\n      audio: usage.inputDetails.audio,\n    };\n  }\n\n  if (usage?.outputDetails?.audio !== undefined) {\n    metrics.output_token_details = {\n      ...(metrics.output_token_details ?? {}),\n      audio: usage.outputDetails.audio,\n    };\n  }\n\n  return metrics;\n}\n","/**\n * LangSmith Exporter for Mastra Tracing\n *\n * This exporter sends observability data to LangSmith\n * Root spans become top-level LangSmith RunTrees (no trace wrapper).\n * Events are handled as zero-duration RunTrees with matching start/end times.\n */\n\nimport type { TracingEvent, AnyExportedSpan, ModelGenerationAttributes } from '@mastra/core/observability';\nimport { SpanType } from '@mastra/core/observability';\nimport { omitKeys } from '@mastra/core/utils';\nimport { BaseExporter } from '@mastra/observability';\nimport type { BaseExporterConfig } from '@mastra/observability';\nimport type { ClientConfig, RunTreeConfig } from 'langsmith';\nimport { Client, RunTree } from 'langsmith';\nimport type { KVMap } from 'langsmith/schemas';\nimport { formatUsageMetrics } from './metrics';\n\nexport interface LangSmithExporterConfig extends ClientConfig, BaseExporterConfig {\n  /** LangSmith client instance */\n  client?: Client;\n  /**\n   * The name of the LangSmith project to send traces to.\n   * Overrides the LANGCHAIN_PROJECT environment variable.\n   * If neither is set, traces are sent to the \"default\" project.\n   */\n  projectName?: string;\n}\n\ntype SpanData = {\n  spans: Map<string, RunTree>; // Maps span.id to LangSmith RunTrees\n  activeIds: Set<string>; // Tracks started (non-event) spans not yet ended, including root\n};\n\n// Default span type for all spans\nconst DEFAULT_SPAN_TYPE = 'chain';\n\n// Exceptions to the default mapping\nconst SPAN_TYPE_EXCEPTIONS: Partial<Record<SpanType, 'llm' | 'tool' | 'chain'>> = {\n  [SpanType.MODEL_GENERATION]: 'llm',\n  [SpanType.TOOL_CALL]: 'tool',\n  [SpanType.MCP_TOOL_CALL]: 'tool',\n  [SpanType.WORKFLOW_CONDITIONAL_EVAL]: 'chain',\n  [SpanType.WORKFLOW_WAIT_EVENT]: 'chain',\n};\n\n// Mapping function - returns valid LangSmith span types\nfunction mapSpanType(spanType: SpanType): 'llm' | 'tool' | 'chain' {\n  return SPAN_TYPE_EXCEPTIONS[spanType] ?? DEFAULT_SPAN_TYPE;\n}\n\nfunction isKVMap(value: unknown): value is KVMap {\n  return value != null && typeof value === 'object' && !Array.isArray(value) && !(value instanceof Date);\n}\n\nexport class LangSmithExporter extends BaseExporter {\n  name = 'langsmith';\n  private traceMap = new Map<string, SpanData>();\n  private config: LangSmithExporterConfig;\n  private client: Client;\n\n  constructor(config: LangSmithExporterConfig) {\n    super(config);\n\n    config.apiKey = config.apiKey ?? process.env.LANGSMITH_API_KEY;\n\n    if (!config.apiKey) {\n      this.setDisabled(`Missing required credentials (apiKey: ${!!config.apiKey})`);\n      this.config = null as any;\n      this.client = null as any;\n      return;\n    }\n\n    this.client = config.client ?? new Client(config);\n    this.config = config;\n  }\n\n  protected async _exportTracingEvent(event: TracingEvent): Promise<void> {\n    if (event.exportedSpan.isEvent) {\n      await this.handleEventSpan(event.exportedSpan);\n      return;\n    }\n\n    switch (event.type) {\n      case 'span_started':\n        await this.handleSpanStarted(event.exportedSpan);\n        break;\n      case 'span_updated':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, false);\n        break;\n      case 'span_ended':\n        await this.handleSpanUpdateOrEnd(event.exportedSpan, true);\n        break;\n    }\n  }\n\n  private initializeRootSpan(span: AnyExportedSpan) {\n    // Check if trace already exists - reuse existing trace data\n    if (this.traceMap.has(span.traceId)) {\n      this.logger.debug('LangSmith exporter: Reusing existing trace from local map', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n      });\n      return;\n    }\n\n    this.traceMap.set(span.traceId, { spans: new Map(), activeIds: new Set() });\n  }\n\n  private async handleSpanStarted(span: AnyExportedSpan): Promise<void> {\n    this.logger.debug('LangSmith exporter: handleSpanStarted', span.id, span.name);\n    if (span.isRootSpan) {\n      this.initializeRootSpan(span);\n    }\n\n    const method = 'handleSpanStarted';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    // Refcount: track active non-event spans (including root)\n    if (!span.isEvent) {\n      spanData.activeIds.add(span.id);\n    }\n\n    const payload = {\n      name: span.name,\n      run_type: mapSpanType(span.type),\n      ...this.buildRunTreePayload(span),\n    };\n\n    const langsmithParent = this.getLangSmithParent({ spanData, span, method });\n    let langsmithRunTree: RunTree;\n    if (!langsmithParent) {\n      langsmithRunTree = new RunTree(payload);\n    } else {\n      langsmithRunTree = langsmithParent.createChild(payload);\n    }\n\n    spanData.spans.set(span.id, langsmithRunTree);\n\n    await langsmithRunTree.postRun();\n  }\n\n  private async handleSpanUpdateOrEnd(span: AnyExportedSpan, isEnd: boolean): Promise<void> {\n    this.logger.debug('LangSmith exporter: handleSpanUpdateOrEnd', span.id, span.name, 'isEnd:', isEnd);\n    const method = isEnd ? 'handleSpanEnd' : 'handleSpanUpdate';\n\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const langsmithRunTree = spanData.spans.get(span.id);\n    if (!langsmithRunTree) {\n      this.logger.warn('LangSmith exporter: No LangSmith span found for span update/end', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        spanType: span.type,\n        isRootSpan: span.isRootSpan,\n        parentSpanId: span.parentSpanId,\n        method,\n      });\n      return;\n    }\n\n    const updatePayload = this.buildRunTreePayload(span);\n    langsmithRunTree.metadata = {\n      ...langsmithRunTree.metadata,\n      ...updatePayload.metadata,\n    };\n    if (updatePayload.inputs != null) {\n      langsmithRunTree.inputs = updatePayload.inputs;\n    }\n    if (updatePayload.outputs != null) {\n      langsmithRunTree.outputs = updatePayload.outputs;\n    }\n    if (updatePayload.error != null) {\n      langsmithRunTree.error = updatePayload.error;\n    }\n\n    // Add new_token event for TTFT tracking on MODEL_GENERATION spans\n    if (span.type === SpanType.MODEL_GENERATION) {\n      const modelAttr = (span.attributes ?? {}) as ModelGenerationAttributes;\n      if (modelAttr.completionStartTime !== undefined) {\n        langsmithRunTree.addEvent({\n          name: 'new_token',\n          time: modelAttr.completionStartTime.toISOString(),\n        });\n      }\n    }\n\n    if (isEnd) {\n      // End the span with the correct endTime (convert milliseconds to seconds)\n      if (span.endTime) {\n        await langsmithRunTree.end({ endTime: span.endTime.getTime() / 1000 });\n      } else {\n        await langsmithRunTree.end();\n      }\n      await langsmithRunTree.patchRun();\n\n      // Refcount: mark this span as ended\n      if (!span.isEvent) {\n        spanData.activeIds.delete(span.id);\n      }\n\n      // If no more active spans remain for this trace, clean up the trace entry\n      if (spanData.activeIds.size === 0) {\n        this.traceMap.delete(span.traceId);\n      }\n    }\n  }\n\n  private async handleEventSpan(span: AnyExportedSpan): Promise<void> {\n    if (span.isRootSpan) {\n      this.logger.debug('LangSmith exporter: Creating logger for event', {\n        traceId: span.traceId,\n        spanId: span.id,\n        spanName: span.name,\n        method: 'handleEventSpan',\n      });\n      this.initializeRootSpan(span);\n    }\n\n    const method = 'handleEventSpan';\n    const spanData = this.getSpanData({ span, method });\n    if (!spanData) {\n      return;\n    }\n\n    const langsmithParent = this.getLangSmithParent({ spanData, span, method });\n    const payload = {\n      ...this.buildRunTreePayload(span),\n      name: span.name,\n      type: mapSpanType(span.type),\n      startTime: span.startTime.getTime() / 1000,\n    };\n\n    let langsmithRunTree: RunTree;\n    if (!langsmithParent) {\n      langsmithRunTree = new RunTree(payload);\n    } else {\n      langsmithRunTree = langsmithParent.createChild(payload);\n    }\n\n    await langsmithRunTree.postRun();\n\n    await langsmithRunTree.end({ endTime: span.startTime.getTime() / 1000 });\n    await langsmithRunTree.patchRun();\n  }\n\n  private getSpanData(options: { span: AnyExportedSpan; method: string }): SpanData | undefined {\n    const { span, method } = options;\n    if (this.traceMap.has(span.traceId)) {\n      return this.traceMap.get(span.traceId);\n    }\n\n    this.logger.warn('LangSmith exporter: No span data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parentSpanId,\n      method,\n    });\n  }\n\n  private getLangSmithParent(options: {\n    spanData: SpanData;\n    span: AnyExportedSpan;\n    method: string;\n  }): RunTree | undefined {\n    const { spanData, span, method } = options;\n\n    const parentId = span.parentSpanId;\n    if (!parentId) {\n      return undefined;\n    }\n\n    if (spanData.spans.has(parentId)) {\n      return spanData.spans.get(parentId);\n    }\n\n    if (parentId && !spanData.spans.has(parentId)) {\n      // This means the parent exists but isn't tracked as a LangSmith span,\n      // which happens when the parent is the root span\n      return undefined;\n    }\n\n    this.logger.warn('LangSmith exporter: No parent data found for span', {\n      traceId: span.traceId,\n      spanId: span.id,\n      spanName: span.name,\n      spanType: span.type,\n      isRootSpan: span.isRootSpan,\n      parentSpanId: span.parentSpanId,\n      method,\n    });\n  }\n\n  private buildRunTreePayload(span: AnyExportedSpan): Partial<RunTreeConfig> {\n    const payload: Partial<RunTreeConfig> & { metadata: KVMap } = {\n      client: this.client,\n      metadata: {\n        mastra_span_type: span.type,\n        ...span.metadata,\n      },\n    };\n\n    // Add project name if configured\n    if (this.config.projectName) {\n      payload.project_name = this.config.projectName;\n    }\n\n    // Add tags for root spans\n    if (span.isRootSpan && span.tags?.length) {\n      payload.tags = span.tags;\n    }\n\n    // Core span data\n    if (span.input !== undefined) {\n      payload.inputs = isKVMap(span.input) ? span.input : { input: span.input };\n    }\n\n    if (span.output !== undefined) {\n      payload.outputs = isKVMap(span.output) ? span.output : { output: span.output };\n    }\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    if (span.type === SpanType.MODEL_GENERATION) {\n      const modelAttr = attributes as ModelGenerationAttributes;\n\n      // See: https://docs.langchain.com/langsmith/log-llm-trace\n      if (modelAttr.model !== undefined) {\n        // Note - this should map to a model name recognized by LangSmith\n        // eg “gpt-4o-mini”, “claude-3-opus-20240307”, etc.\n        payload.metadata.ls_model_name = modelAttr.model;\n      }\n\n      // Provider goes to metadata (if provided by attributes)\n      if (modelAttr.provider !== undefined) {\n        // Note - this should map to a provider name recognized by\n        // LangSmith eg “openai”, “anthropic”, etc.\n        payload.metadata.ls_provider = modelAttr.provider;\n      }\n\n      // Usage/token info goes to metrics\n      payload.metadata.usage_metadata = formatUsageMetrics(modelAttr.usage);\n\n      // Model parameters go to metadata\n      if (modelAttr.parameters !== undefined) {\n        payload.metadata.modelParameters = modelAttr.parameters;\n      }\n\n      // Other LLM attributes go to metadata\n      const otherAttributes = omitKeys(attributes, ['model', 'usage', 'parameters', 'completionStartTime']);\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    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, runTree] of spanData.spans) {\n        await runTree.end();\n        await runTree.patchRun();\n      }\n    }\n    this.traceMap.clear();\n    await super.shutdown();\n  }\n}\n"]}
+{"version":3,"sources":["../src/metrics.ts","../src/tracing.ts"],"names":[],"mappings":";;;;;;;;AAuBO,SAAS,mBAAmB,KAAA,EAA2C;AAC5E,EAAA,MAAM,UAAiC,EAAC;AAExC,EAAA,IAAI,KAAA,EAAO,gBAAgB,MAAA,EAAW;AACpC,IAAA,OAAA,CAAQ,eAAe,KAAA,CAAM,WAAA;AAAA,EAC/B;AAEA,EAAA,IAAI,KAAA,EAAO,iBAAiB,MAAA,EAAW;AACrC,IAAA,OAAA,CAAQ,gBAAgB,KAAA,CAAM,YAAA;AAAA,EAChC;AAGA,EAAA,IAAI,OAAA,CAAQ,YAAA,KAAiB,MAAA,IAAa,OAAA,CAAQ,kBAAkB,MAAA,EAAW;AAC7E,IAAA,OAAA,CAAQ,YAAA,GAAe,OAAA,CAAQ,YAAA,GAAe,OAAA,CAAQ,aAAA;AAAA,EACxD;AAEA,EAAA,IAAI,KAAA,EAAO,aAAA,EAAe,SAAA,KAAc,MAAA,EAAW;AACjD,IAAA,OAAA,CAAQ,oBAAA,GAAuB;AAAA,MAC7B,GAAI,OAAA,CAAQ,oBAAA,IAAwB,EAAC;AAAA,MACrC,gBAAA,EAAkB,MAAM,aAAA,CAAc;AAAA,KACxC;AAAA,EACF;AAEA,EAAA,IAAI,KAAA,EAAO,YAAA,EAAc,SAAA,KAAc,MAAA,EAAW;AAChD,IAAA,OAAA,CAAQ,mBAAA,GAAsB;AAAA,MAC5B,GAAI,OAAA,CAAQ,mBAAA,IAAuB,EAAC;AAAA,MACpC,UAAA,EAAY,MAAM,YAAA,CAAa;AAAA,KACjC;AAAA,EACF;AAEA,EAAA,IAAI,KAAA,EAAO,YAAA,EAAc,UAAA,KAAe,MAAA,EAAW;AACjD,IAAA,OAAA,CAAQ,mBAAA,GAAsB;AAAA,MAC5B,GAAI,OAAA,CAAQ,mBAAA,IAAuB,EAAC;AAAA,MACpC,WAAA,EAAa,MAAM,YAAA,CAAa;AAAA,KAClC;AAAA,EACF;AAEA,EAAA,IAAI,KAAA,EAAO,YAAA,EAAc,KAAA,KAAU,MAAA,EAAW;AAC5C,IAAA,OAAA,CAAQ,mBAAA,GAAsB;AAAA,MAC5B,GAAI,OAAA,CAAQ,mBAAA,IAAuB,EAAC;AAAA,MACpC,KAAA,EAAO,MAAM,YAAA,CAAa;AAAA,KAC5B;AAAA,EACF;AAEA,EAAA,IAAI,KAAA,EAAO,aAAA,EAAe,KAAA,KAAU,MAAA,EAAW;AAC7C,IAAA,OAAA,CAAQ,oBAAA,GAAuB;AAAA,MAC7B,GAAI,OAAA,CAAQ,oBAAA,IAAwB,EAAC;AAAA,MACrC,KAAA,EAAO,MAAM,aAAA,CAAc;AAAA,KAC7B;AAAA,EACF;AAEA,EAAA,OAAO,OAAA;AACT;;;ACvCA,IAAM,iBAAA,GAAoB,OAAA;AAG1B,IAAM,oBAAA,GAA4E;AAAA,EAChF,CAAC,QAAA,CAAS,gBAAgB,GAAG,KAAA;AAAA,EAC7B,CAAC,QAAA,CAAS,SAAS,GAAG,MAAA;AAAA,EACtB,CAAC,QAAA,CAAS,aAAa,GAAG,MAAA;AAAA,EAC1B,CAAC,QAAA,CAAS,yBAAyB,GAAG,OAAA;AAAA,EACtC,CAAC,QAAA,CAAS,mBAAmB,GAAG;AAClC,CAAA;AAGA,SAAS,YAAY,QAAA,EAA8C;AACjE,EAAA,OAAO,oBAAA,CAAqB,QAAQ,CAAA,IAAK,iBAAA;AAC3C;AAEA,SAAS,QAAQ,KAAA,EAAgC;AAC/C,EAAA,OAAO,KAAA,IAAS,IAAA,IAAQ,OAAO,KAAA,KAAU,QAAA,IAAY,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,IAAK,EAAE,KAAA,YAAiB,IAAA,CAAA;AACnG;AAEO,IAAM,iBAAA,GAAN,cAAgC,gBAAA,CAMrC;AAAA,EACS,IAAA,GAAO,WAAA;AAAA,EAChB,OAAA;AAAA,EAEA,WAAA,CAAY,MAAA,GAAkC,EAAC,EAAG;AAEhD,IAAA,MAAM,MAAA,GAAS,MAAA,CAAO,MAAA,IAAU,OAAA,CAAQ,GAAA,CAAI,iBAAA;AAE5C,IAAA,KAAA,CAAM;AAAA,MACJ,GAAG,MAAA;AAAA,MACH;AAAA,KACD,CAAA;AAED,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,IAAA,CAAK,WAAA,CAAY,CAAA,sCAAA,EAAyC,CAAC,CAAC,MAAM,CAAA,CAAA,CAAG,CAAA;AACrE,MAAA;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,UAAU,MAAA,CAAO,MAAA,IAAU,IAAI,MAAA,CAAO,KAAK,MAAM,CAAA;AAAA,EACxD;AAAA,EAEmB,iBAAA,GAAoB,IAAA;AAAA,EACvC,MAAyB,WAAW,KAAA,EAGG;AACrC,IAAA,MAAM,IAAI,MAAM,yBAAyB,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAyB,WAAW,IAAA,EAGG;AACrC,IAAA,MAAM,EAAE,IAAA,EAAM,SAAA,EAAU,GAAI,IAAA;AAE5B,IAAA,MAAM,SAAS,IAAA,CAAK,UAAA,GAAa,MAAA,GAAY,SAAA,CAAU,UAAU,IAAI,CAAA;AAErE,IAAA,IAAI,CAAC,IAAA,CAAK,UAAA,IAAc,CAAC,MAAA,EAAQ;AAE/B,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,GAAU;AAAA,MACd,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,GAAG,IAAA,CAAK,mBAAA,CAAoB,IAAA,EAAM,IAAI;AAAA,KACxC;AAEA,IAAA,MAAM,aAAA,GAAgB,KAAK,UAAA,GAAa,IAAI,QAAQ,OAAO,CAAA,GAAI,MAAA,CAAQ,WAAA,CAAY,OAAO,CAAA;AAE1F,IAAA,MAAM,cAAc,OAAA,EAAQ;AAC5B,IAAA,OAAO,aAAA;AAAA,EACT;AAAA,EAEA,MAAyB,YAAY,IAAA,EAGG;AACtC,IAAA,MAAM,aAAA,GAAgB,MAAM,IAAA,CAAK,UAAA,CAAW,IAAI,CAAA;AAEhD,IAAA,IAAI,CAAC,aAAA,EAAe;AAElB,MAAA;AAAA,IACF;AAGA,IAAA,MAAM,aAAA,CAAc,IAAI,EAAE,OAAA,EAAS,KAAK,IAAA,CAAK,SAAA,CAAU,OAAA,EAAQ,EAAG,CAAA;AAClE,IAAA,MAAM,cAAc,QAAA,EAAS;AAC7B,IAAA,OAAO,aAAA;AAAA,EACT;AAAA,EAEA,MAAyB,YAAY,IAAA,EAA+E;AAClH,IAAA,MAAM,KAAK,qBAAA,CAAsB,EAAE,GAAG,IAAA,EAAM,KAAA,EAAO,OAAO,CAAA;AAAA,EAC5D;AAAA,EAEA,MAAyB,YAAY,IAAA,EAA+E;AAClH,IAAA,MAAM,KAAK,qBAAA,CAAsB,EAAE,GAAG,IAAA,EAAM,KAAA,EAAO,MAAM,CAAA;AAAA,EAC3D;AAAA,EAEA,MAAyB,WAAW,IAAA,EAIlB;AAChB,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,IAAA;AACzB,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,OAAA;AACpB,IAAA,IAAA,CAAK,QAAA,GAAW;AAAA,MACd,GAAG,IAAA,CAAK,QAAA;AAAA,MACR,YAAA,EAAc;AAAA,KAChB;AACA,IAAA,MAAM,KAAK,GAAA,EAAI;AACf,IAAA,MAAM,KAAK,QAAA,EAAS;AAAA,EACtB;AAAA,EAEA,MAAc,sBAAsB,IAAA,EAIlB;AAChB,IAAA,MAAM,EAAE,IAAA,EAAM,SAAA,EAAW,KAAA,EAAM,GAAI,IAAA;AAEnC,IAAA,MAAM,gBAAgB,SAAA,CAAU,OAAA,CAAQ,EAAE,MAAA,EAAQ,IAAA,CAAK,IAAI,CAAA;AAC3D,IAAA,IAAI,CAAC,aAAA,EAAe;AAElB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,aAAA,GAAgB,IAAA,CAAK,mBAAA,CAAoB,IAAI,CAAA;AAEnD,IAAA,aAAA,CAAc,QAAA,GAAW;AAAA,MACvB,GAAG,aAAA,CAAc,QAAA;AAAA,MACjB,GAAG,aAAA,CAAc;AAAA,KACnB;AACA,IAAA,IAAI,aAAA,CAAc,UAAU,IAAA,EAAM;AAChC,MAAA,aAAA,CAAc,SAAS,aAAA,CAAc,MAAA;AAAA,IACvC;AACA,IAAA,IAAI,aAAA,CAAc,WAAW,IAAA,EAAM;AACjC,MAAA,aAAA,CAAc,UAAU,aAAA,CAAc,OAAA;AAAA,IACxC;AACA,IAAA,IAAI,aAAA,CAAc,SAAS,IAAA,EAAM;AAC/B,MAAA,aAAA,CAAc,QAAQ,aAAA,CAAc,KAAA;AAAA,IACtC;AAGA,IAAA,IAAI,IAAA,CAAK,IAAA,KAAS,QAAA,CAAS,gBAAA,EAAkB;AAC3C,MAAA,MAAM,SAAA,GAAa,IAAA,CAAK,UAAA,IAAc,EAAC;AACvC,MAAA,IAAI,SAAA,CAAU,wBAAwB,MAAA,EAAW;AAC/C,QAAA,aAAA,CAAc,QAAA,CAAS;AAAA,UACrB,IAAA,EAAM,WAAA;AAAA,UACN,IAAA,EAAM,SAAA,CAAU,mBAAA,CAAoB,WAAA;AAAY,SACjD,CAAA;AAAA,MACH;AAAA,IACF;AAEA,IAAA,IAAI,KAAA,EAAO;AAET,MAAA,IAAI,KAAK,OAAA,EAAS;AAChB,QAAA,MAAM,aAAA,CAAc,IAAI,EAAE,OAAA,EAAS,KAAK,OAAA,CAAQ,OAAA,IAAW,CAAA;AAAA,MAC7D,CAAA,MAAO;AACL,QAAA,MAAM,cAAc,GAAA,EAAI;AAAA,MAC1B;AAAA,IACF;AACA,IAAA,MAAM,cAAc,QAAA,EAAS;AAAA,EAC/B;AAAA,EAEQ,mBAAA,CAAoB,IAAA,EAAuB,KAAA,GAAQ,KAAA,EAA+B;AACxF,IAAA,MAAM,OAAA,GAAwD;AAAA,MAC5D,QAAQ,IAAA,CAAK,OAAA;AAAA,MACb,QAAA,EAAU;AAAA,QACR,kBAAkB,IAAA,CAAK,IAAA;AAAA,QACvB,GAAG,IAAA,CAAK;AAAA;AACV,KACF;AAEA,IAAA,IAAI,KAAA,EAAO;AACT,MAAA,OAAA,CAAQ,QAAA,GAAW,WAAA,CAAY,IAAA,CAAK,IAAI,CAAA;AACxC,MAAA,OAAA,CAAQ,UAAA,GAAa,IAAA,CAAK,SAAA,CAAU,OAAA,EAAQ;AAAA,IAC9C;AAGA,IAAA,IAAI,IAAA,CAAK,OAAO,WAAA,EAAa;AAC3B,MAAA,OAAA,CAAQ,YAAA,GAAe,KAAK,MAAA,CAAO,WAAA;AAAA,IACrC;AAGA,IAAA,IAAI,IAAA,CAAK,UAAA,IAAc,IAAA,CAAK,IAAA,EAAM,MAAA,EAAQ;AACxC,MAAA,OAAA,CAAQ,OAAO,IAAA,CAAK,IAAA;AAAA,IACtB;AAGA,IAAA,IAAI,IAAA,CAAK,UAAU,MAAA,EAAW;AAC5B,MAAA,OAAA,CAAQ,MAAA,GAAS,OAAA,CAAQ,IAAA,CAAK,KAAK,CAAA,GAAI,KAAK,KAAA,GAAQ,EAAE,KAAA,EAAO,IAAA,CAAK,KAAA,EAAM;AAAA,IAC1E;AAEA,IAAA,IAAI,IAAA,CAAK,WAAW,MAAA,EAAW;AAC7B,MAAA,OAAA,CAAQ,OAAA,GAAU,OAAA,CAAQ,IAAA,CAAK,MAAM,CAAA,GAAI,KAAK,MAAA,GAAS,EAAE,MAAA,EAAQ,IAAA,CAAK,MAAA,EAAO;AAAA,IAC/E;AAEA,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAExC,IAAA,IAAI,IAAA,CAAK,IAAA,KAAS,QAAA,CAAS,gBAAA,EAAkB;AAC3C,MAAA,MAAM,SAAA,GAAY,UAAA;AAGlB,MAAA,IAAI,SAAA,CAAU,UAAU,MAAA,EAAW;AAGjC,QAAA,OAAA,CAAQ,QAAA,CAAS,gBAAgB,SAAA,CAAU,KAAA;AAAA,MAC7C;AAGA,MAAA,IAAI,SAAA,CAAU,aAAa,MAAA,EAAW;AAGpC,QAAA,OAAA,CAAQ,QAAA,CAAS,cAAc,SAAA,CAAU,QAAA;AAAA,MAC3C;AAGA,MAAA,OAAA,CAAQ,QAAA,CAAS,cAAA,GAAiB,kBAAA,CAAmB,SAAA,CAAU,KAAK,CAAA;AAGpE,MAAA,IAAI,SAAA,CAAU,eAAe,MAAA,EAAW;AACtC,QAAA,OAAA,CAAQ,QAAA,CAAS,kBAAkB,SAAA,CAAU,UAAA;AAAA,MAC/C;AAGA,MAAA,MAAM,eAAA,GAAkB,SAAS,UAAA,EAAY,CAAC,SAAS,UAAA,EAAY,OAAA,EAAS,YAAA,EAAc,qBAAqB,CAAC,CAAA;AAChH,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;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AACF","file":"index.js","sourcesContent":["import type { UsageStats } from '@mastra/core/observability';\n\n/**\n * LangSmithUsageMetrics\n *\n * Canonical metric keys expected by LangSmith for LLM usage accounting.\n * See: https://docs.langchain.com/langsmith/log-llm-trace#provide-token-and-cost-information\n */\nexport interface LangSmithUsageMetrics {\n  input_tokens?: number;\n  output_tokens?: number;\n  total_tokens?: number;\n  input_token_details?: {\n    [key: string]: number;\n  };\n  output_token_details?: {\n    [key: string]: number;\n  };\n}\n\n/**\n * Formats UsageStats to LangSmith's expected metric format.\n */\nexport function formatUsageMetrics(usage?: UsageStats): LangSmithUsageMetrics {\n  const metrics: LangSmithUsageMetrics = {};\n\n  if (usage?.inputTokens !== undefined) {\n    metrics.input_tokens = usage.inputTokens;\n  }\n\n  if (usage?.outputTokens !== undefined) {\n    metrics.output_tokens = usage.outputTokens;\n  }\n\n  // Compute total if we have both\n  if (metrics.input_tokens !== undefined && metrics.output_tokens !== undefined) {\n    metrics.total_tokens = metrics.input_tokens + metrics.output_tokens;\n  }\n\n  if (usage?.outputDetails?.reasoning !== undefined) {\n    metrics.output_token_details = {\n      ...(metrics.output_token_details ?? {}),\n      reasoning_tokens: usage.outputDetails.reasoning,\n    };\n  }\n\n  if (usage?.inputDetails?.cacheRead !== undefined) {\n    metrics.input_token_details = {\n      ...(metrics.input_token_details ?? {}),\n      cache_read: usage.inputDetails.cacheRead,\n    };\n  }\n\n  if (usage?.inputDetails?.cacheWrite !== undefined) {\n    metrics.input_token_details = {\n      ...(metrics.input_token_details ?? {}),\n      cache_write: usage.inputDetails.cacheWrite,\n    };\n  }\n\n  if (usage?.inputDetails?.audio !== undefined) {\n    metrics.input_token_details = {\n      ...(metrics.input_token_details ?? {}),\n      audio: usage.inputDetails.audio,\n    };\n  }\n\n  if (usage?.outputDetails?.audio !== undefined) {\n    metrics.output_token_details = {\n      ...(metrics.output_token_details ?? {}),\n      audio: usage.outputDetails.audio,\n    };\n  }\n\n  return metrics;\n}\n","/**\n * LangSmith Exporter for Mastra Tracing\n *\n * This exporter sends observability data to LangSmith\n * Root spans become top-level LangSmith RunTrees (no trace wrapper).\n * Events are handled as zero-duration RunTrees with matching start/end times.\n */\n\nimport type { AnyExportedSpan, ModelGenerationAttributes, SpanErrorInfo } from '@mastra/core/observability';\nimport { SpanType } from '@mastra/core/observability';\nimport { omitKeys } from '@mastra/core/utils';\nimport { TrackingExporter } from '@mastra/observability';\nimport type { TraceData, TrackingExporterConfig } from '@mastra/observability';\nimport type { ClientConfig, RunTreeConfig } from 'langsmith';\nimport { Client, RunTree } from 'langsmith';\nimport type { KVMap } from 'langsmith/schemas';\nimport { formatUsageMetrics } from './metrics';\n\nexport interface LangSmithExporterConfig extends ClientConfig, TrackingExporterConfig {\n  /** LangSmith client instance */\n  client?: Client;\n  /**\n   * The name of the LangSmith project to send traces to.\n   * Overrides the LANGCHAIN_PROJECT environment variable.\n   * If neither is set, traces are sent to the \"default\" project.\n   */\n  projectName?: string;\n}\n\ntype LangSmithRoot = undefined;\ntype LangSmithSpan = RunTree;\ntype LangSmithEvent = RunTree;\ntype LangSmithMetadata = unknown;\ntype LangSmithTraceData = TraceData<LangSmithRoot, LangSmithSpan, LangSmithEvent, LangSmithMetadata>;\n\n// Default span type for all spans\nconst DEFAULT_SPAN_TYPE = 'chain';\n\n// Exceptions to the default mapping\nconst SPAN_TYPE_EXCEPTIONS: Partial<Record<SpanType, 'llm' | 'tool' | 'chain'>> = {\n  [SpanType.MODEL_GENERATION]: 'llm',\n  [SpanType.TOOL_CALL]: 'tool',\n  [SpanType.MCP_TOOL_CALL]: 'tool',\n  [SpanType.WORKFLOW_CONDITIONAL_EVAL]: 'chain',\n  [SpanType.WORKFLOW_WAIT_EVENT]: 'chain',\n};\n\n// Mapping function - returns valid LangSmith span types\nfunction mapSpanType(spanType: SpanType): 'llm' | 'tool' | 'chain' {\n  return SPAN_TYPE_EXCEPTIONS[spanType] ?? DEFAULT_SPAN_TYPE;\n}\n\nfunction isKVMap(value: unknown): value is KVMap {\n  return value != null && typeof value === 'object' && !Array.isArray(value) && !(value instanceof Date);\n}\n\nexport class LangSmithExporter extends TrackingExporter<\n  LangSmithRoot,\n  LangSmithSpan,\n  LangSmithEvent,\n  LangSmithMetadata,\n  LangSmithExporterConfig\n> {\n  override name = 'langsmith';\n  #client: Client | undefined;\n\n  constructor(config: LangSmithExporterConfig = {}) {\n    // Resolve env vars BEFORE calling super (config is readonly in base class)\n    const apiKey = config.apiKey ?? process.env.LANGSMITH_API_KEY;\n\n    super({\n      ...config,\n      apiKey,\n    });\n\n    if (!apiKey) {\n      this.setDisabled(`Missing required credentials (apiKey: ${!!apiKey})`);\n      return;\n    }\n\n    this.#client = config.client ?? new Client(this.config);\n  }\n\n  protected override skipBuildRootTask = true;\n  protected override async _buildRoot(_args: {\n    span: AnyExportedSpan;\n    traceData: LangSmithTraceData;\n  }): Promise<LangSmithRoot | undefined> {\n    throw new Error('Method not implemented.');\n  }\n\n  protected override async _buildSpan(args: {\n    span: AnyExportedSpan;\n    traceData: LangSmithTraceData;\n  }): Promise<LangSmithSpan | undefined> {\n    const { span, traceData } = args;\n\n    const parent = span.isRootSpan ? undefined : traceData.getParent(args);\n\n    if (!span.isRootSpan && !parent) {\n      // parent doesn't exist and not creating rootSpan, return early data\n      return;\n    }\n\n    const payload = {\n      name: span.name,\n      ...this.buildRunTreePayload(span, true),\n    };\n\n    const langSmithSpan = span.isRootSpan ? new RunTree(payload) : parent!.createChild(payload);\n\n    await langSmithSpan.postRun();\n    return langSmithSpan;\n  }\n\n  protected override async _buildEvent(args: {\n    span: AnyExportedSpan;\n    traceData: LangSmithTraceData;\n  }): Promise<LangSmithEvent | undefined> {\n    const langSmithSpan = await this._buildSpan(args);\n\n    if (!langSmithSpan) {\n      // parent doesn't exist and not creating rootSpan, return early data\n      return;\n    }\n\n    // use start-time as end-time to make an event span.\n    await langSmithSpan.end({ endTime: args.span.startTime.getTime() });\n    await langSmithSpan.patchRun();\n    return langSmithSpan;\n  }\n\n  protected override async _updateSpan(args: { span: AnyExportedSpan; traceData: LangSmithTraceData }): Promise<void> {\n    await this.handleSpanUpdateOrEnd({ ...args, isEnd: false });\n  }\n\n  protected override async _finishSpan(args: { span: AnyExportedSpan; traceData: LangSmithTraceData }): Promise<void> {\n    await this.handleSpanUpdateOrEnd({ ...args, isEnd: true });\n  }\n\n  protected override async _abortSpan(args: {\n    span: LangSmithSpan;\n    traceData: LangSmithTraceData;\n    reason: SpanErrorInfo;\n  }): Promise<void> {\n    const { span, reason } = args;\n    span.error = reason.message;\n    span.metadata = {\n      ...span.metadata,\n      errorDetails: reason,\n    };\n    await span.end();\n    await span.patchRun();\n  }\n\n  private async handleSpanUpdateOrEnd(args: {\n    span: AnyExportedSpan;\n    traceData: LangSmithTraceData;\n    isEnd: boolean;\n  }): Promise<void> {\n    const { span, traceData, isEnd } = args;\n\n    const langSmithSpan = traceData.getSpan({ spanId: span.id });\n    if (!langSmithSpan) {\n      //update occurred before span start, return early data\n      return;\n    }\n\n    const updatePayload = this.buildRunTreePayload(span);\n\n    langSmithSpan.metadata = {\n      ...langSmithSpan.metadata,\n      ...updatePayload.metadata,\n    };\n    if (updatePayload.inputs != null) {\n      langSmithSpan.inputs = updatePayload.inputs;\n    }\n    if (updatePayload.outputs != null) {\n      langSmithSpan.outputs = updatePayload.outputs;\n    }\n    if (updatePayload.error != null) {\n      langSmithSpan.error = updatePayload.error;\n    }\n\n    // Add new_token event for TTFT tracking on MODEL_GENERATION spans\n    if (span.type === SpanType.MODEL_GENERATION) {\n      const modelAttr = (span.attributes ?? {}) as ModelGenerationAttributes;\n      if (modelAttr.completionStartTime !== undefined) {\n        langSmithSpan.addEvent({\n          name: 'new_token',\n          time: modelAttr.completionStartTime.toISOString(),\n        });\n      }\n    }\n\n    if (isEnd) {\n      // End the span with the correct endTime\n      if (span.endTime) {\n        await langSmithSpan.end({ endTime: span.endTime.getTime() });\n      } else {\n        await langSmithSpan.end();\n      }\n    }\n    await langSmithSpan.patchRun();\n  }\n\n  private buildRunTreePayload(span: AnyExportedSpan, isNew = false): Partial<RunTreeConfig> {\n    const payload: Partial<RunTreeConfig> & { metadata: KVMap } = {\n      client: this.#client,\n      metadata: {\n        mastra_span_type: span.type,\n        ...span.metadata,\n      },\n    };\n\n    if (isNew) {\n      payload.run_type = mapSpanType(span.type);\n      payload.start_time = span.startTime.getTime();\n    }\n\n    // Add project name if configured\n    if (this.config.projectName) {\n      payload.project_name = this.config.projectName;\n    }\n\n    // Add tags for root spans\n    if (span.isRootSpan && span.tags?.length) {\n      payload.tags = span.tags;\n    }\n\n    // Core span data\n    if (span.input !== undefined) {\n      payload.inputs = isKVMap(span.input) ? span.input : { input: span.input };\n    }\n\n    if (span.output !== undefined) {\n      payload.outputs = isKVMap(span.output) ? span.output : { output: span.output };\n    }\n\n    const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n    if (span.type === SpanType.MODEL_GENERATION) {\n      const modelAttr = attributes as ModelGenerationAttributes;\n\n      // See: https://docs.langchain.com/langsmith/log-llm-trace\n      if (modelAttr.model !== undefined) {\n        // Note - this should map to a model name recognized by LangSmith\n        // eg “gpt-4o-mini”, “claude-3-opus-20240307”, etc.\n        payload.metadata.ls_model_name = modelAttr.model;\n      }\n\n      // Provider goes to metadata (if provided by attributes)\n      if (modelAttr.provider !== undefined) {\n        // Note - this should map to a provider name recognized by\n        // LangSmith eg “openai”, “anthropic”, etc.\n        payload.metadata.ls_provider = modelAttr.provider;\n      }\n\n      // Usage/token info goes to metrics\n      payload.metadata.usage_metadata = formatUsageMetrics(modelAttr.usage);\n\n      // Model parameters go to metadata\n      if (modelAttr.parameters !== undefined) {\n        payload.metadata.modelParameters = modelAttr.parameters;\n      }\n\n      // Other LLM attributes go to metadata\n      const otherAttributes = omitKeys(attributes, ['model', 'provider', 'usage', 'parameters', 'completionStartTime']);\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    return payload;\n  }\n}\n"]}

package/dist/tracing.d.ts

@@ -5,12 +5,12 @@
  * Root spans become top-level LangSmith RunTrees (no trace wrapper).
  * Events are handled as zero-duration RunTrees with matching start/end times.
  */
-import type { TracingEvent } from '@mastra/core/observability';
-import { BaseExporter } from '@mastra/observability';
-import type { BaseExporterConfig } from '@mastra/observability';
+import type { AnyExportedSpan, SpanErrorInfo } from '@mastra/core/observability';
+import { TrackingExporter } from '@mastra/observability';
+import type { TraceData, TrackingExporterConfig } from '@mastra/observability';
 import type { ClientConfig } from 'langsmith';
-import { Client } from 'langsmith';
-export interface LangSmithExporterConfig extends ClientConfig, BaseExporterConfig {
+import { Client, RunTree } from 'langsmith';
+export interface LangSmithExporterConfig extends ClientConfig, TrackingExporterConfig {
     /** LangSmith client instance */
     client?: Client;
     /**
@@ -20,20 +20,43 @@ export interface LangSmithExporterConfig extends ClientConfig, BaseExporterConfi
      */
     projectName?: string;
 }
-export declare class LangSmithExporter extends BaseExporter {
+type LangSmithRoot = undefined;
+type LangSmithSpan = RunTree;
+type LangSmithEvent = RunTree;
+type LangSmithMetadata = unknown;
+type LangSmithTraceData = TraceData<LangSmithRoot, LangSmithSpan, LangSmithEvent, LangSmithMetadata>;
+export declare class LangSmithExporter extends TrackingExporter<LangSmithRoot, LangSmithSpan, LangSmithEvent, LangSmithMetadata, LangSmithExporterConfig> {
+    #private;
     name: string;
-    private traceMap;
-    private config;
-    private client;
-    constructor(config: LangSmithExporterConfig);
-    protected _exportTracingEvent(event: TracingEvent): Promise<void>;
-    private initializeRootSpan;
-    private handleSpanStarted;
+    constructor(config?: LangSmithExporterConfig);
+    protected skipBuildRootTask: boolean;
+    protected _buildRoot(_args: {
+        span: AnyExportedSpan;
+        traceData: LangSmithTraceData;
+    }): Promise<LangSmithRoot | undefined>;
+    protected _buildSpan(args: {
+        span: AnyExportedSpan;
+        traceData: LangSmithTraceData;
+    }): Promise<LangSmithSpan | undefined>;
+    protected _buildEvent(args: {
+        span: AnyExportedSpan;
+        traceData: LangSmithTraceData;
+    }): Promise<LangSmithEvent | undefined>;
+    protected _updateSpan(args: {
+        span: AnyExportedSpan;
+        traceData: LangSmithTraceData;
+    }): Promise<void>;
+    protected _finishSpan(args: {
+        span: AnyExportedSpan;
+        traceData: LangSmithTraceData;
+    }): Promise<void>;
+    protected _abortSpan(args: {
+        span: LangSmithSpan;
+        traceData: LangSmithTraceData;
+        reason: SpanErrorInfo;
+    }): Promise<void>;
     private handleSpanUpdateOrEnd;
-    private handleEventSpan;
-    private getSpanData;
-    private getLangSmithParent;
     private buildRunTreePayload;
-    shutdown(): Promise<void>;
 }
+export {};
 //# sourceMappingURL=tracing.d.ts.map

package/dist/tracing.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tracing.d.ts","sourceRoot":"","sources":["../src/tracing.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,YAAY,EAA8C,MAAM,4BAA4B,CAAC;AAG3G,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,KAAK,EAAE,YAAY,EAAiB,MAAM,WAAW,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAW,MAAM,WAAW,CAAC;AAI5C,MAAM,WAAW,uBAAwB,SAAQ,YAAY,EAAE,kBAAkB;IAC/E,gCAAgC;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AA4BD,qBAAa,iBAAkB,SAAQ,YAAY;IACjD,IAAI,SAAe;IACnB,OAAO,CAAC,QAAQ,CAA+B;IAC/C,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,EAAE,uBAAuB;cAgB3B,mBAAmB,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBvE,OAAO,CAAC,kBAAkB;YAcZ,iBAAiB;YAoCjB,qBAAqB;YAsErB,eAAe;IAsC7B,OAAO,CAAC,WAAW;IAiBnB,OAAO,CAAC,kBAAkB;IAiC1B,OAAO,CAAC,mBAAmB;IA8ErB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAehC"}
+{"version":3,"file":"tracing.d.ts","sourceRoot":"","sources":["../src/tracing.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,eAAe,EAA6B,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAG5G,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,KAAK,EAAE,SAAS,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/E,OAAO,KAAK,EAAE,YAAY,EAAiB,MAAM,WAAW,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAI5C,MAAM,WAAW,uBAAwB,SAAQ,YAAY,EAAE,sBAAsB;IACnF,gCAAgC;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,KAAK,aAAa,GAAG,SAAS,CAAC;AAC/B,KAAK,aAAa,GAAG,OAAO,CAAC;AAC7B,KAAK,cAAc,GAAG,OAAO,CAAC;AAC9B,KAAK,iBAAiB,GAAG,OAAO,CAAC;AACjC,KAAK,kBAAkB,GAAG,SAAS,CAAC,aAAa,EAAE,aAAa,EAAE,cAAc,EAAE,iBAAiB,CAAC,CAAC;AAuBrG,qBAAa,iBAAkB,SAAQ,gBAAgB,CACrD,aAAa,EACb,aAAa,EACb,cAAc,EACd,iBAAiB,EACjB,uBAAuB,CACxB;;IACU,IAAI,SAAe;gBAGhB,MAAM,GAAE,uBAA4B;IAiBhD,UAAmB,iBAAiB,UAAQ;cACnB,UAAU,CAAC,KAAK,EAAE;QACzC,IAAI,EAAE,eAAe,CAAC;QACtB,SAAS,EAAE,kBAAkB,CAAC;KAC/B,GAAG,OAAO,CAAC,aAAa,GAAG,SAAS,CAAC;cAIb,UAAU,CAAC,IAAI,EAAE;QACxC,IAAI,EAAE,eAAe,CAAC;QACtB,SAAS,EAAE,kBAAkB,CAAC;KAC/B,GAAG,OAAO,CAAC,aAAa,GAAG,SAAS,CAAC;cAqBb,WAAW,CAAC,IAAI,EAAE;QACzC,IAAI,EAAE,eAAe,CAAC;QACtB,SAAS,EAAE,kBAAkB,CAAC;KAC/B,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC;cAcd,WAAW,CAAC,IAAI,EAAE;QAAE,IAAI,EAAE,eAAe,CAAC;QAAC,SAAS,EAAE,kBAAkB,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;cAI1F,WAAW,CAAC,IAAI,EAAE;QAAE,IAAI,EAAE,eAAe,CAAC;QAAC,SAAS,EAAE,kBAAkB,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;cAI1F,UAAU,CAAC,IAAI,EAAE;QACxC,IAAI,EAAE,aAAa,CAAC;QACpB,SAAS,EAAE,kBAAkB,CAAC;QAC9B,MAAM,EAAE,aAAa,CAAC;KACvB,GAAG,OAAO,CAAC,IAAI,CAAC;YAWH,qBAAqB;IAmDnC,OAAO,CAAC,mBAAmB;CAkF5B"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/langsmith",
-  "version": "1.0.0-beta.9",
+  "version": "1.0.1-alpha.0",
   "description": "Langsmith observability provider for Mastra - includes tracing and future observability features",
   "type": "module",
   "main": "dist/index.js",
@@ -25,22 +25,23 @@
   "license": "Apache-2.0",
   "dependencies": {
     "langsmith": ">=0.3.79",
-    "@mastra/observability": "1.0.0-beta.8"
+    "@mastra/observability": "1.1.0-alpha.0"
   },
   "devDependencies": {
     "@ai-sdk/openai": "^2.0.35",
-    "@types/node": "22.13.17",
+    "@types/node": "22.19.7",
     "@vitest/coverage-v8": "4.0.12",
     "@vitest/ui": "4.0.12",
-    "dotenv": "^17.2.2",
+    "dotenv": "^17.2.3",
     "eslint": "^9.37.0",
-    "tsup": "^8.5.0",
-    "typescript": "^5.8.3",
-    "vitest": "4.0.12",
+    "tsup": "^8.5.1",
+    "typescript": "^5.9.3",
+    "vitest": "4.0.16",
     "zod": "^3.25.76",
-    "@internal/types-builder": "0.0.28",
-    "@mastra/core": "1.0.0-beta.16",
-    "@internal/lint": "0.0.53"
+    "@internal/types-builder": "0.0.30",
+    "@mastra/core": "1.1.0-alpha.0",
+    "@observability/test-utils": "0.0.1",
+    "@internal/lint": "0.0.55"
   },
   "peerDependencies": {
     "@mastra/core": ">=1.0.0-0 <2.0.0-0"