braintrust 3.7.0 → 3.8.0

package/dist/auto-instrumentations/index.cjs

@@ -24,6 +24,8 @@ __export(index_exports, {
   anthropicConfigs: () => anthropicConfigs,
   claudeAgentSDKConfigs: () => claudeAgentSDKConfigs,
   googleGenAIConfigs: () => googleGenAIConfigs,
+  mistralConfigs: () => mistralConfigs,
+  openRouterAgentConfigs: () => openRouterAgentConfigs,
   openRouterConfigs: () => openRouterConfigs,
   openaiConfigs: () => openaiConfigs
 });
@@ -101,7 +103,7 @@ var DefaultTracingChannel = class {
   }
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   tracePromise(fn, _message, thisArg, ...args) {
-    return Promise.resolve(fn.apply(thisArg, args));
+    return fn.apply(thisArg, args);
   }
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   traceCallback(fn, _position, _message, thisArg, ...args) {
@@ -121,6 +123,7 @@ var iso = {
   processOn: (_0, _1) => {
   },
   basename: (filepath) => filepath.split(/[\\/]/).pop() || filepath,
+  // eslint-disable-next-line no-restricted-properties -- preserving intentional console usage.
   writeln: (text) => console.log(text)
 };
 var isomorph_default = iso;
@@ -204,6 +207,10 @@ var openAIChannels = defineChannels("openai", {
   responsesParse: channel({
     channelName: "responses.parse",
     kind: "async"
+  }),
+  responsesCompact: channel({
+    channelName: "responses.compact",
+    kind: "async"
   })
 });
 
@@ -370,6 +377,19 @@ var openaiConfigs = [
       methodName: "parse",
       kind: "Async"
     }
+  },
+  {
+    channelName: openAIChannels.responsesCompact.channelName,
+    module: {
+      name: "openai",
+      versionRange: ">=6.10.0",
+      filePath: "resources/responses/responses.mjs"
+    },
+    functionQuery: {
+      className: "Responses",
+      methodName: "compact",
+      kind: "Async"
+    }
   }
 ];
 
@@ -457,6 +477,16 @@ var aiSDKChannels = defineChannels("ai", {
     channelName: "streamObject.sync",
     kind: "sync-stream"
   }),
+  embed: channel(
+    {
+      channelName: "embed",
+      kind: "async"
+    }
+  ),
+  embedMany: channel({
+    channelName: "embedMany",
+    kind: "async"
+  }),
   agentGenerate: channel({
     channelName: "Agent.generate",
     kind: "async"
@@ -465,6 +495,10 @@ var aiSDKChannels = defineChannels("ai", {
     channelName: "Agent.stream",
     kind: "async"
   }),
+  agentStreamSync: channel({
+    channelName: "Agent.stream.sync",
+    kind: "sync-stream"
+  }),
   toolLoopAgentGenerate: channel({
     channelName: "ToolLoopAgent.generate",
     kind: "async"
@@ -502,12 +536,12 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
-  // streamText - async function
+  // streamText - async function (v3 only, before the sync refactor in v4)
   {
     channelName: aiSDKChannels.streamText.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <4.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
@@ -515,8 +549,21 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
+  // streamText - sync function returning stream (v4+)
   {
     channelName: aiSDKChannels.streamTextSync.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=4.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "streamText",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: aiSDKChannels.streamText.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -524,7 +571,7 @@ var aiSDKConfigs = [
     },
     functionQuery: {
       functionName: "streamText",
-      kind: "Sync"
+      kind: "Async"
     }
   },
   // generateObject - async function
@@ -552,12 +599,62 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
-  // streamObject - async function
+  // embed - async function
   {
-    channelName: aiSDKChannels.streamObject.channelName,
+    channelName: aiSDKChannels.embed.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "embed",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: aiSDKChannels.embed.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "embed",
+      kind: "Async"
+    }
+  },
+  // embedMany - async function
+  {
+    channelName: aiSDKChannels.embedMany.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "embedMany",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: aiSDKChannels.embedMany.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "embedMany",
+      kind: "Async"
+    }
+  },
+  // streamObject - async function (v3 only, before the sync refactor in v4)
+  {
+    channelName: aiSDKChannels.streamObject.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0 <4.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
@@ -565,8 +662,21 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
+  // streamObject - sync function returning stream (v4+)
   {
     channelName: aiSDKChannels.streamObjectSync.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=4.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "streamObject",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: aiSDKChannels.streamObject.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -574,7 +684,7 @@ var aiSDKConfigs = [
     },
     functionQuery: {
       functionName: "streamObject",
-      kind: "Sync"
+      kind: "Async"
     }
   },
   // Agent.generate - async method (v5 only)
@@ -606,11 +716,11 @@ var aiSDKConfigs = [
       index: 0
     }
   },
-  // Agent.stream - async method (v5 only)
+  // Agent.stream - sync method (v5 only)
   // The compiled AI SDK bundle emits this as an anonymous class method, so we
-  // target the first async `stream` method in the file instead of a class name.
+  // target the first sync `stream` method in the file instead of a class name.
   {
-    channelName: aiSDKChannels.agentStream.channelName,
+    channelName: aiSDKChannels.agentStreamSync.channelName,
     module: {
       name: "ai",
       versionRange: ">=5.0.0 <6.0.0",
@@ -618,12 +728,12 @@ var aiSDKConfigs = [
     },
     functionQuery: {
       methodName: "stream",
-      kind: "Async",
+      kind: "Sync",
       index: 0
     }
   },
   {
-    channelName: aiSDKChannels.agentStream.channelName,
+    channelName: aiSDKChannels.agentStreamSync.channelName,
     module: {
       name: "ai",
       versionRange: ">=5.0.0 <6.0.0",
@@ -631,7 +741,7 @@ var aiSDKConfigs = [
     },
     functionQuery: {
       methodName: "stream",
-      kind: "Async",
+      kind: "Sync",
       index: 0
     }
   },
@@ -783,6 +893,38 @@ var googleGenAIConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/openrouter-agent-channels.ts
+var openRouterAgentChannels = defineChannels("@openrouter/agent", {
+  callModel: channel({
+    channelName: "callModel",
+    kind: "sync-stream"
+  }),
+  callModelTurn: channel({
+    channelName: "callModel.turn",
+    kind: "async"
+  }),
+  toolExecute: channel({
+    channelName: "tool.execute",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/openrouter-agent.ts
+var openRouterAgentConfigs = [
+  {
+    channelName: openRouterAgentChannels.callModel.channelName,
+    module: {
+      name: "@openrouter/agent",
+      versionRange: ">=0.1.2",
+      filePath: "esm/inner-loop/call-model.js"
+    },
+    functionQuery: {
+      functionName: "callModel",
+      kind: "Sync"
+    }
+  }
+];
+
 // src/instrumentation/plugins/openrouter-channels.ts
 var openRouterChannels = defineChannels("@openrouter/sdk", {
   chatSend: channel({
@@ -801,6 +943,10 @@ var openRouterChannels = defineChannels("@openrouter/sdk", {
     channelName: "callModel",
     kind: "sync-stream"
   }),
+  callModelTurn: channel({
+    channelName: "callModel.turn",
+    kind: "async"
+  }),
   toolExecute: channel({
     channelName: "tool.execute",
     kind: "async"
@@ -862,12 +1008,232 @@ var openRouterConfigs = [
     }
   }
 ];
+
+// src/instrumentation/plugins/mistral-channels.ts
+var mistralChannels = defineChannels("@mistralai/mistralai", {
+  chatComplete: channel({
+    channelName: "chat.complete",
+    kind: "async"
+  }),
+  chatStream: channel({
+    channelName: "chat.stream",
+    kind: "async"
+  }),
+  embeddingsCreate: channel({
+    channelName: "embeddings.create",
+    kind: "async"
+  }),
+  fimComplete: channel({
+    channelName: "fim.complete",
+    kind: "async"
+  }),
+  fimStream: channel({
+    channelName: "fim.stream",
+    kind: "async"
+  }),
+  agentsComplete: channel({
+    channelName: "agents.complete",
+    kind: "async"
+  }),
+  agentsStream: channel({
+    channelName: "agents.stream",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/mistral.ts
+var mistralConfigs = [
+  {
+    channelName: mistralChannels.chatComplete.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.0.0 <2.0.0",
+      filePath: "sdk/chat.js"
+    },
+    functionQuery: {
+      className: "Chat",
+      methodName: "complete",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.chatComplete.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/chat.js"
+    },
+    functionQuery: {
+      className: "Chat",
+      methodName: "complete",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.chatStream.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.0.0 <2.0.0",
+      filePath: "sdk/chat.js"
+    },
+    functionQuery: {
+      className: "Chat",
+      methodName: "stream",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.chatStream.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/chat.js"
+    },
+    functionQuery: {
+      className: "Chat",
+      methodName: "stream",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.embeddingsCreate.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.0.0 <2.0.0",
+      filePath: "sdk/embeddings.js"
+    },
+    functionQuery: {
+      className: "Embeddings",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.embeddingsCreate.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/embeddings.js"
+    },
+    functionQuery: {
+      className: "Embeddings",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.fimComplete.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.0.0 <2.0.0",
+      filePath: "sdk/fim.js"
+    },
+    functionQuery: {
+      className: "Fim",
+      methodName: "complete",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.fimComplete.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/fim.js"
+    },
+    functionQuery: {
+      className: "Fim",
+      methodName: "complete",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.fimStream.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.0.0 <2.0.0",
+      filePath: "sdk/fim.js"
+    },
+    functionQuery: {
+      className: "Fim",
+      methodName: "stream",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.fimStream.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/fim.js"
+    },
+    functionQuery: {
+      className: "Fim",
+      methodName: "stream",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.agentsComplete.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.0.0 <2.0.0",
+      filePath: "sdk/agents.js"
+    },
+    functionQuery: {
+      className: "Agents",
+      methodName: "complete",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.agentsComplete.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/agents.js"
+    },
+    functionQuery: {
+      className: "Agents",
+      methodName: "complete",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.agentsStream.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.0.0 <2.0.0",
+      filePath: "sdk/agents.js"
+    },
+    functionQuery: {
+      className: "Agents",
+      methodName: "stream",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.agentsStream.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/agents.js"
+    },
+    functionQuery: {
+      className: "Agents",
+      methodName: "stream",
+      kind: "Async"
+    }
+  }
+];
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   aiSDKConfigs,
   anthropicConfigs,
   claudeAgentSDKConfigs,
   googleGenAIConfigs,
+  mistralConfigs,
+  openRouterAgentConfigs,
   openRouterConfigs,
   openaiConfigs
 });

package/dist/auto-instrumentations/index.d.mts

@@ -66,6 +66,10 @@ declare const claudeAgentSDKConfigs: InstrumentationConfig[];
  */
 declare const googleGenAIConfigs: InstrumentationConfig[];
 
+declare const openRouterAgentConfigs: InstrumentationConfig[];
+
 declare const openRouterConfigs: InstrumentationConfig[];
 
-export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, googleGenAIConfigs, openRouterConfigs, openaiConfigs };
+declare const mistralConfigs: InstrumentationConfig[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, googleGenAIConfigs, mistralConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };

package/dist/auto-instrumentations/index.d.ts

@@ -66,6 +66,10 @@ declare const claudeAgentSDKConfigs: InstrumentationConfig[];
  */
 declare const googleGenAIConfigs: InstrumentationConfig[];
 
+declare const openRouterAgentConfigs: InstrumentationConfig[];
+
 declare const openRouterConfigs: InstrumentationConfig[];
 
-export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, googleGenAIConfigs, openRouterConfigs, openaiConfigs };
+declare const mistralConfigs: InstrumentationConfig[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, googleGenAIConfigs, mistralConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };

package/dist/auto-instrumentations/index.mjs

@@ -3,14 +3,18 @@ import {
   anthropicConfigs,
   claudeAgentSDKConfigs,
   googleGenAIConfigs,
+  mistralConfigs,
+  openRouterAgentConfigs,
   openRouterConfigs,
   openaiConfigs
-} from "./chunk-AKEXR4AL.mjs";
+} from "./chunk-OLBMPZXE.mjs";
 export {
   aiSDKConfigs,
   anthropicConfigs,
   claudeAgentSDKConfigs,
   googleGenAIConfigs,
+  mistralConfigs,
+  openRouterAgentConfigs,
   openRouterConfigs,
   openaiConfigs
 };

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

@@ -34,7 +34,7 @@ __export(cjs_patch_exports, {
 });
 module.exports = __toCommonJS(cjs_patch_exports);
 var import_code_transformer = require("@apm-js-collab/code-transformer");
-var Module = __toESM(require("module"));
+var NodeModule = __toESM(require("module"));
 var import_node_path2 = require("path");
 var import_module_details_from_path = __toESM(require("module-details-from-path"));
 
@@ -52,11 +52,29 @@ function readPackageJson(baseDir) {
     return void 0;
   }
 }
+function resolvePackageBaseDir(baseDir) {
+  try {
+    return (0, import_node_fs.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 = readPackageJson(baseDir);
+  const packageJson = readPackageJsonWithFallback(baseDir);
   if (typeof packageJson?.version === "string") {
     packageVersions.set(baseDir, packageJson.version);
     return packageJson.version;
@@ -67,7 +85,7 @@ function getPackageName(baseDir) {
   if (packageNames.has(baseDir)) {
     return packageNames.get(baseDir);
   }
-  const packageJson = readPackageJson(baseDir);
+  const packageJson = readPackageJsonWithFallback(baseDir);
   if (typeof packageJson?.name === "string") {
     packageNames.set(baseDir, packageJson.name);
     return packageJson.name;
@@ -79,13 +97,16 @@ function getPackageName(baseDir) {
 var ModulePatch = class {
   packages;
   instrumentator;
+  modulePrototype;
   originalCompile;
   constructor({
     instrumentations = []
   } = {}) {
+    const modulePrototype = resolveModulePrototype();
     this.packages = new Set(instrumentations.map((i) => i.module.name));
     this.instrumentator = (0, import_code_transformer.create)(instrumentations);
-    this.originalCompile = Module.prototype._compile;
+    this.modulePrototype = modulePrototype;
+    this.originalCompile = modulePrototype._compile;
   }
   /**
    * Patches the Node.js module class method that is responsible for compiling code.
@@ -94,7 +115,7 @@ var ModulePatch = class {
    */
   patch() {
     const self = this;
-    Module.prototype._compile = function wrappedCompile(...args) {
+    this.modulePrototype._compile = function wrappedCompile(...args) {
       const [content, filename] = args;
       const normalizedForPlatform = filename.split("/").join(import_node_path2.sep);
       const resolvedModule = (0, import_module_details_from_path.default)(normalizedForPlatform);
@@ -129,9 +150,16 @@ var ModulePatch = class {
    * **Note**: This is intended to be used in testing only.
    */
   unpatch() {
-    Module.prototype._compile = this.originalCompile;
+    this.modulePrototype._compile = this.originalCompile;
   }
 };
+function resolveModulePrototype() {
+  const moduleCtor = NodeModule.Module;
+  if (moduleCtor && typeof moduleCtor === "function") {
+    return moduleCtor.prototype;
+  }
+  return NodeModule.prototype;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ModulePatch

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

@@ -8,6 +8,7 @@ import { InstrumentationConfig } from '@apm-js-collab/code-transformer';
 declare class ModulePatch {
     private packages;
     private instrumentator;
+    private modulePrototype;
     private originalCompile;
     constructor({ instrumentations, }?: {
         instrumentations?: InstrumentationConfig[];

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

@@ -8,6 +8,7 @@ import { InstrumentationConfig } from '@apm-js-collab/code-transformer';
 declare class ModulePatch {
     private packages;
     private instrumentator;
+    private modulePrototype;
     private originalCompile;
     constructor({ instrumentations, }?: {
         instrumentations?: InstrumentationConfig[];