braintrust 3.13.0 → 3.15.0

package/dist/auto-instrumentations/chunk-QFMACSOL.mjs

@@ -0,0 +1,280 @@
+// src/auto-instrumentations/loader/mastra-observability-patch.ts
+var MASTRA_EXPORTER_FACTORY_GLOBAL = "__braintrustMastraExporterFactory";
+function installMastraExporterFactory(factory) {
+  const globals = globalThis;
+  globals[MASTRA_EXPORTER_FACTORY_GLOBAL] ??= factory;
+}
+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/get-package-version.ts
+import { readFileSync, realpathSync } from "fs";
+import { join } from "path";
+var packageVersions = /* @__PURE__ */ new Map();
+var packageNames = /* @__PURE__ */ new Map();
+function readPackageJson(baseDir) {
+  try {
+    const packageJsonPath = join(baseDir, "package.json");
+    const jsonFile = readFileSync(packageJsonPath, "utf8");
+    return JSON.parse(jsonFile);
+  } catch {
+    return void 0;
+  }
+}
+function resolvePackageBaseDir(baseDir) {
+  try {
+    return realpathSync(baseDir);
+  } catch {
+    return baseDir;
+  }
+}
+function readPackageJsonWithFallback(baseDir) {
+  const packageJson = readPackageJson(baseDir);
+  if (packageJson) {
+    return packageJson;
+  }
+  const resolvedBaseDir = resolvePackageBaseDir(baseDir);
+  if (resolvedBaseDir === baseDir) {
+    return void 0;
+  }
+  return readPackageJson(resolvedBaseDir);
+}
+function getPackageVersion(baseDir) {
+  if (packageVersions.has(baseDir)) {
+    return packageVersions.get(baseDir);
+  }
+  const packageJson = readPackageJsonWithFallback(baseDir);
+  if (typeof packageJson?.version === "string") {
+    packageVersions.set(baseDir, packageJson.version);
+    return packageJson.version;
+  }
+  return process.version.slice(1);
+}
+function getPackageName(baseDir) {
+  if (packageNames.has(baseDir)) {
+    return packageNames.get(baseDir);
+  }
+  const packageJson = readPackageJsonWithFallback(baseDir);
+  if (typeof packageJson?.name === "string") {
+    packageNames.set(baseDir, packageJson.name);
+    return packageJson.name;
+  }
+  return void 0;
+}
+
+// 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/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;
+}
+function isSpecialCaseTarget(packageName, modulePath) {
+  if (packageName === "openai" && modulePath.includes("api-promise")) {
+    return true;
+  }
+  if (classifyMastraTarget(packageName, modulePath) !== null) {
+    return true;
+  }
+  return false;
+}
+
+export {
+  installMastraExporterFactory,
+  getPackageVersion,
+  getPackageName,
+  applySpecialCasePatch,
+  isSpecialCaseTarget
+};

package/dist/auto-instrumentations/{chunk-GJOO4ESL.mjs → chunk-VXJONZVX.mjs}

@@ -20,6 +20,9 @@ import {
   openRouterConfigs,
   openaiConfigs
 } from "./chunk-E5DUYJWK.mjs";
+import {
+  applySpecialCasePatch
+} from "./chunk-J57YF7WS.mjs";
 
 // src/auto-instrumentations/bundler/plugin.ts
 import { createUnplugin } from "unplugin";
@@ -51,6 +54,7 @@ var envIntegrationAliases = {
   cursorsdk: "cursorSDK",
   flue: "flue",
   "flue-runtime": "flue",
+  mastra: "mastra",
   "openai-agents": "openAIAgents",
   openaiagents: "openAIAgents",
   "openai-agents-core": "openAIAgents",
@@ -99,35 +103,11 @@ var flueChannels = defineChannels("@flue/runtime", {
   createContext: channel({
     channelName: "createFlueContext",
     kind: "sync-stream"
-  }),
-  openSession: channel({
-    channelName: "Harness.openSession",
-    kind: "async"
-  }),
-  contextEvent: channel({
-    channelName: "context.event",
-    kind: "sync-stream"
-  }),
-  prompt: channel({
-    channelName: "session.prompt",
-    kind: "async"
-  }),
-  skill: channel({
-    channelName: "session.skill",
-    kind: "async"
-  }),
-  task: channel({
-    channelName: "session.task",
-    kind: "async"
-  }),
-  compact: channel({
-    channelName: "session.compact",
-    kind: "async"
   })
 });
 
 // src/auto-instrumentations/configs/flue.ts
-var flueVersionRange = ">=0.7.0 <0.8.0";
+var flueVersionRange = ">=0.8.0";
 var flueConfigs = [
   {
     channelName: flueChannels.createContext.channelName,
@@ -140,19 +120,6 @@ var flueConfigs = [
       functionName: "createFlueContext",
       kind: "Sync"
     }
-  },
-  {
-    channelName: flueChannels.openSession.channelName,
-    module: {
-      name: "@flue/runtime",
-      versionRange: flueVersionRange,
-      filePath: "dist/internal.mjs"
-    },
-    functionQuery: {
-      className: "Harness",
-      methodName: "openSession",
-      kind: "Async"
-    }
   }
 ];
 
@@ -173,7 +140,6 @@ var defaultInstrumentationConfigGroups = [
     configs: claudeAgentSDKConfigs
   },
   { integrations: ["cursor", "cursorSDK"], configs: cursorSDKConfigs },
-  { integrations: ["flue"], configs: flueConfigs },
   {
     integrations: ["openAIAgents"],
     configs: openAIAgentsCoreConfigs
@@ -203,7 +169,18 @@ var defaultInstrumentationConfigGroups = [
   {
     integrations: ["gitHubCopilot"],
     configs: gitHubCopilotConfigs
+  },
+  {
+    integrations: ["flue"],
+    configs: flueConfigs
   }
+  // Note: `@mastra/core` is not listed here because its instrumentation
+  // doesn't go through the AST `code-transformer` matcher — Mastra's
+  // content-hashed chunks make `filePath`-based matching too brittle.
+  // Instead it's handled by the source-replacement entry in
+  // `loader/special-case-patches.ts`, which both the runtime loader
+  // (`hook.mjs` → `cjs-patch.ts`/`esm-hook.mts`) and the bundler plugin
+  // (`bundler/plugin.ts`) call. The `mastra` env-var disable still works.
 ];
 function getDefaultInstrumentationConfigs({
   additionalInstrumentations,
@@ -259,6 +236,18 @@ var unplugin = createUnplugin(
           return null;
         }
         const moduleName = moduleDetails.name;
+        const normalizedModulePath = moduleDetails.path.replace(/\\/g, "/");
+        if (options.browser !== true) {
+          const patched = applySpecialCasePatch({
+            packageName: moduleName,
+            modulePath: normalizedModulePath,
+            source: code,
+            format: isModule ? "esm" : "cjs"
+          });
+          if (patched !== null) {
+            return { code: patched, map: null };
+          }
+        }
         const moduleVersion = getModuleVersion(moduleDetails.basedir);
         if (!moduleVersion) {
           console.warn(
@@ -266,7 +255,6 @@ var unplugin = createUnplugin(
           );
           return null;
         }
-        const normalizedModulePath = moduleDetails.path.replace(/\\/g, "/");
         const transformer = instrumentationMatcher.getTransformer(
           moduleName,
           moduleVersion,