braintrust 3.12.0 → 3.14.0

package/dist/auto-instrumentations/index.cjs

@@ -126,6 +126,7 @@ var iso = {
   getRepoInfo: async (_settings) => void 0,
   getPastNAncestors: async () => [],
   getEnv: (_name) => void 0,
+  getBraintrustApiKey: async () => void 0,
   getCallerLocation: () => void 0,
   newAsyncLocalStorage: () => new DefaultAsyncLocalStorage(),
   // eslint-disable-next-line @typescript-eslint/no-explicit-any

package/dist/auto-instrumentations/index.mjs

@@ -17,7 +17,7 @@ import {
   openRouterAgentConfigs,
   openRouterConfigs,
   openaiConfigs
-} from "./chunk-2DPA74KK.mjs";
+} from "./chunk-E5DUYJWK.mjs";
 export {
   aiSDKConfigs,
   anthropicConfigs,

package/dist/auto-instrumentations/loader/cjs-patch.cjs

@@ -93,6 +93,175 @@ function getPackageName(baseDir) {
   return void 0;
 }
 
+// src/auto-instrumentations/loader/mastra-observability-patch.ts
+var MASTRA_EXPORTER_FACTORY_GLOBAL = "__braintrustMastraExporterFactory";
+var MASTRA_CORE_PACKAGE = "@mastra/core";
+var MASTRA_OBSERVABILITY_PACKAGE = "@mastra/observability";
+var MASTRA_CORE_ENTRY_PATHS = /* @__PURE__ */ new Set([
+  "dist/index.js",
+  "dist/index.cjs",
+  "dist/mastra/index.js",
+  "dist/mastra/index.cjs"
+]);
+var MASTRA_OBSERVABILITY_ENTRY_PATHS = /* @__PURE__ */ new Set([
+  "dist/index.js",
+  "dist/index.cjs"
+]);
+function classifyMastraTarget(packageName, modulePath) {
+  if (packageName === MASTRA_CORE_PACKAGE && MASTRA_CORE_ENTRY_PATHS.has(modulePath)) {
+    return "core";
+  }
+  if (packageName === MASTRA_OBSERVABILITY_PACKAGE && MASTRA_OBSERVABILITY_ENTRY_PATHS.has(modulePath)) {
+    return "observability";
+  }
+  return null;
+}
+function extractChunkPath(source) {
+  const esmMatch = source.match(
+    /export\s*\{\s*Mastra(?:\s+as\s+\w+)?\s*\}\s*from\s*['"]([^'"]+)['"]/
+  );
+  if (esmMatch) return esmMatch[1];
+  const cjsMatch = source.match(
+    /require\s*\(\s*['"]([^'"]+chunk-[^'"]+)['"]\s*\)/
+  );
+  if (cjsMatch) return cjsMatch[1];
+  return null;
+}
+var EXPORTER_FACTORY_KEY = JSON.stringify(MASTRA_EXPORTER_FACTORY_GLOBAL);
+var MASTRA_PROXY_HANDLER_BODY = `
+{
+  construct(target, args, newTarget) {
+    var firstArg = args[0];
+    if (
+      (!firstArg || typeof firstArg !== "object" || !firstArg.observability) &&
+      __braintrustObservabilityClass
+    ) {
+      try {
+        // serviceName is required by Mastra's Observability validator; pass
+        // something sensible by default. Users who want a different name
+        // should construct Observability themselves.
+        var observability = new __braintrustObservabilityClass({
+          configs: { default: { serviceName: "mastra" } },
+        });
+        args = args.slice();
+        args[0] = Object.assign({}, firstArg, { observability: observability });
+      } catch (e) {
+        // Fall through. Mastra will use its own NoOp; user code still works,
+        // just without auto-instrumentation.
+      }
+    }
+    return Reflect.construct(target, args, newTarget);
+  },
+}`;
+function buildMastraEsmWrapper(chunkPath) {
+  const chunk = JSON.stringify(chunkPath);
+  return `import { Mastra as __braintrustOrigMastra } from ${chunk};
+import { createRequire as __braintrustCreateRequire } from "node:module";
+
+let __braintrustObservabilityClass = null;
+try {
+  // Resolve @mastra/observability relative to this module (the Mastra entry),
+  // so it's looked up from the user's node_modules tree.
+  const __braintrustRequire = __braintrustCreateRequire(import.meta.url);
+  __braintrustObservabilityClass =
+    __braintrustRequire("@mastra/observability").Observability;
+} catch (e) {
+  // @mastra/observability isn't installed; the construct trap will skip the
+  // auto-construct branch and Mastra falls back to its own NoOp.
+}
+const Mastra = new Proxy(__braintrustOrigMastra, ${MASTRA_PROXY_HANDLER_BODY});
+export { Mastra };
+`;
+}
+function buildMastraCjsWrapper(chunkPath) {
+  const chunk = JSON.stringify(chunkPath);
+  return `'use strict';
+const __braintrustChunk = require(${chunk});
+
+let __braintrustObservabilityClass = null;
+try {
+  __braintrustObservabilityClass = require("@mastra/observability").Observability;
+} catch (e) {
+  // @mastra/observability isn't installed; same fallback as the ESM wrapper.
+}
+
+const __braintrustWrappedMastra = new Proxy(
+  __braintrustChunk.Mastra,
+  ${MASTRA_PROXY_HANDLER_BODY},
+);
+Object.defineProperty(exports, "Mastra", {
+  enumerable: true,
+  configurable: true,
+  get: function () { return __braintrustWrappedMastra; }
+});
+`;
+}
+var OBSERVABILITY_APPEND_BODY = `
+;(function __braintrustWrapObservability() {
+  // Top-level so we can both read and reassign the var binding the original
+  // entry declared.
+  if (typeof Observability === "undefined") return;
+  if (Observability.__braintrustWrapped) return;
+  function __braintrustEnsureExporter(rawConfig) {
+    try {
+      var factory = globalThis[${EXPORTER_FACTORY_KEY}];
+      if (typeof factory !== "function") return rawConfig;
+      var config = rawConfig && typeof rawConfig === "object" ? rawConfig : {};
+      var configsIn = config.configs && typeof config.configs === "object" ? config.configs : null;
+      var configsOut = {};
+      var hadEntries = false;
+      if (configsIn) {
+        for (var name in configsIn) {
+          if (!Object.prototype.hasOwnProperty.call(configsIn, name)) continue;
+          hadEntries = true;
+          var inst = configsIn[name] || {};
+          var existing = Array.isArray(inst.exporters) ? inst.exporters : [];
+          var hasOurs = existing.some(function (e) { return e && e.name === "braintrust"; });
+          configsOut[name] = Object.assign({}, inst, {
+            exporters: hasOurs ? existing : existing.concat([factory()]),
+          });
+        }
+      }
+      if (!hadEntries) {
+        configsOut.default = {
+          serviceName: "mastra",
+          exporters: [factory()],
+        };
+      }
+      return Object.assign({}, config, { configs: configsOut });
+    } catch (e) {
+      return rawConfig;
+    }
+  }
+  var __OriginalObservability = Observability;
+  Observability = new Proxy(__OriginalObservability, {
+    construct: function (target, args, newTarget) {
+      var nextArgs = args.slice();
+      nextArgs[0] = __braintrustEnsureExporter(nextArgs[0]);
+      return Reflect.construct(target, nextArgs, newTarget);
+    },
+  });
+  Observability.__braintrustWrapped = true;
+  if (typeof exports !== "undefined" && exports && typeof exports === "object") {
+    try {
+      Object.defineProperty(exports, "Observability", {
+        enumerable: true,
+        configurable: true,
+        get: function () { return Observability; },
+      });
+    } catch (e) {}
+  }
+})();
+`;
+function patchMastraSource(source, target, format) {
+  if (target === "core") {
+    const chunkPath = extractChunkPath(source);
+    if (!chunkPath) return source;
+    return format === "esm" ? buildMastraEsmWrapper(chunkPath) : buildMastraCjsWrapper(chunkPath);
+  }
+  return source + OBSERVABILITY_APPEND_BODY;
+}
+
 // src/auto-instrumentations/loader/openai-api-promise-patch.ts
 var OPENAI_API_PROMISE_PATCH = `
 ;(function __btPatchAPIPromise() {
@@ -114,6 +283,21 @@ var OPENAI_API_PROMISE_PATCH = `
 })();
 `;
 
+// src/auto-instrumentations/loader/special-case-patches.ts
+function applySpecialCasePatch(input) {
+  if (input.packageName === "openai" && input.modulePath.includes("api-promise")) {
+    return input.source + OPENAI_API_PROMISE_PATCH;
+  }
+  const mastraTarget = classifyMastraTarget(
+    input.packageName,
+    input.modulePath
+  );
+  if (mastraTarget) {
+    return patchMastraSource(input.source, mastraTarget, input.format);
+  }
+  return null;
+}
+
 // src/auto-instrumentations/loader/cjs-patch.ts
 var ModulePatch = class {
   packages;
@@ -142,12 +326,18 @@ var ModulePatch = class {
       const resolvedModule = (0, import_module_details_from_path.default)(normalizedForPlatform);
       if (resolvedModule) {
         const packageName = getPackageName(resolvedModule.basedir) ?? resolvedModule.name;
-        if (!self.packages.has(packageName)) {
+        const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+        const patched = applySpecialCasePatch({
+          packageName,
+          modulePath: normalizedModulePath,
+          source: String(content),
+          format: "cjs"
+        });
+        if (patched !== null) {
+          args[0] = patched;
           return self.originalCompile.apply(this, args);
         }
-        const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
-        if (packageName === "openai" && normalizedModulePath.includes("api-promise")) {
-          args[0] = content + OPENAI_API_PROMISE_PATCH;
+        if (!self.packages.has(packageName)) {
           return self.originalCompile.apply(this, args);
         }
         const version = getPackageVersion(resolvedModule.basedir);

package/dist/auto-instrumentations/loader/cjs-patch.mjs

@@ -1,3 +1,6 @@
+import {
+  applySpecialCasePatch
+} from "../chunk-J57YF7WS.mjs";
 import {
   getPackageName,
   getPackageVersion
@@ -10,29 +13,6 @@ import {
 import * as NodeModule from "module";
 import { sep } from "path";
 import moduleDetailsFromPath from "module-details-from-path";
-
-// src/auto-instrumentations/loader/openai-api-promise-patch.ts
-var OPENAI_API_PROMISE_PATCH = `
-;(function __btPatchAPIPromise() {
-  if (typeof APIPromise === "undefined" || APIPromise.prototype.__btParsePatched) return;
-  APIPromise.prototype.__btParsePatched = true;
-  var _origThen = APIPromise.prototype.then;
-  APIPromise.prototype.then = function __btThen(onfulfilled, onrejected) {
-    if (!this.__btParseWrapped && Object.prototype.hasOwnProperty.call(this, "parseResponse")) {
-      this.__btParseWrapped = true;
-      var _origParse = this.parseResponse;
-      var _cached;
-      this.parseResponse = function() {
-        if (!_cached) _cached = _origParse.apply(this, arguments);
-        return _cached;
-      };
-    }
-    return _origThen.call(this, onfulfilled, onrejected);
-  };
-})();
-`;
-
-// src/auto-instrumentations/loader/cjs-patch.ts
 var ModulePatch = class {
   packages;
   instrumentator;
@@ -60,12 +40,18 @@ var ModulePatch = class {
       const resolvedModule = moduleDetailsFromPath(normalizedForPlatform);
       if (resolvedModule) {
         const packageName = getPackageName(resolvedModule.basedir) ?? resolvedModule.name;
-        if (!self.packages.has(packageName)) {
+        const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+        const patched = applySpecialCasePatch({
+          packageName,
+          modulePath: normalizedModulePath,
+          source: String(content),
+          format: "cjs"
+        });
+        if (patched !== null) {
+          args[0] = patched;
           return self.originalCompile.apply(this, args);
         }
-        const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
-        if (packageName === "openai" && normalizedModulePath.includes("api-promise")) {
-          args[0] = content + OPENAI_API_PROMISE_PATCH;
+        if (!self.packages.has(packageName)) {
           return self.originalCompile.apply(this, args);
         }
         const version = getPackageVersion(resolvedModule.basedir);

package/dist/auto-instrumentations/loader/esm-hook.mjs

@@ -1,8 +1,9 @@
 import {
-  OPENAI_API_PROMISE_PATCH,
+  applySpecialCasePatch,
   getPackageName,
-  getPackageVersion
-} from "../chunk-MWZXZQUO.mjs";
+  getPackageVersion,
+  isSpecialCaseTarget
+} from "../chunk-QFMACSOL.mjs";
 
 // src/auto-instrumentations/loader/esm-hook.mts
 import { readFile } from "fs/promises";
@@ -15,6 +16,7 @@ import moduleDetailsFromPath from "module-details-from-path";
 var instrumentator;
 var packages;
 var transformers = /* @__PURE__ */ new Map();
+var specialCaseUrls = /* @__PURE__ */ new Map();
 function getModuleType(url, format) {
   if (format === "module") {
     return "esm";
@@ -44,11 +46,17 @@ async function resolve(specifier, context, nextResolve) {
   const resolvedModule = moduleDetailsFromPath(normalizedForPlatform);
   if (resolvedModule) {
     const packageName = getPackageName(resolvedModule.basedir) ?? resolvedModule.name;
+    const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+    if (isSpecialCaseTarget(packageName, normalizedModulePath)) {
+      specialCaseUrls.set(url.url, {
+        packageName,
+        modulePath: normalizedModulePath
+      });
+    }
     if (!packages?.has(packageName)) {
       return url;
     }
     const version = getPackageVersion(resolvedModule.basedir);
-    const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
     const transformer = instrumentator.getTransformer(
       packageName,
       version,
@@ -60,19 +68,25 @@ async function resolve(specifier, context, nextResolve) {
   }
   return url;
 }
-function isOpenAIApiPromise(url) {
-  return url.includes("/openai") && url.includes("api-promise");
-}
 async function load(url, context, nextLoad) {
   const result = await nextLoad(url, context);
-  if (isOpenAIApiPromise(url)) {
+  const specialCase = specialCaseUrls.get(url);
+  if (specialCase) {
     if (result.format === "commonjs") {
       const parsedUrl = new URL(result.responseURL ?? url);
       result.source ??= await readFile(parsedUrl);
     }
     if (result.source) {
-      result.source = result.source.toString("utf8") + OPENAI_API_PROMISE_PATCH;
-      result.shortCircuit = true;
+      const patched = applySpecialCasePatch({
+        packageName: specialCase.packageName,
+        modulePath: specialCase.modulePath,
+        source: result.source.toString("utf8"),
+        format: result.format === "commonjs" ? "cjs" : "esm"
+      });
+      if (patched !== null) {
+        result.source = patched;
+        result.shortCircuit = true;
+      }
     }
     return result;
   }