braintrust 3.7.0 → 3.8.0

package/dist/auto-instrumentations/hook.mjs

@@ -1,7 +1,7 @@
 import {
   getPackageName,
   getPackageVersion
-} from "./chunk-EVUKFMHG.mjs";
+} from "./chunk-ITP7RAUY.mjs";
 
 // src/auto-instrumentations/hook.mts
 import { register } from "module";
@@ -78,7 +78,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) {
@@ -98,6 +98,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;
@@ -181,6 +182,10 @@ var openAIChannels = defineChannels("openai", {
   responsesParse: channel({
     channelName: "responses.parse",
     kind: "async"
+  }),
+  responsesCompact: channel({
+    channelName: "responses.compact",
+    kind: "async"
   })
 });
 
@@ -347,6 +352,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"
+    }
   }
 ];
 
@@ -434,6 +452,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"
@@ -442,6 +470,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"
@@ -479,12 +511,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: {
@@ -492,8 +524,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",
@@ -501,7 +546,7 @@ var aiSDKConfigs = [
     },
     functionQuery: {
       functionName: "streamText",
-      kind: "Sync"
+      kind: "Async"
     }
   },
   // generateObject - async function
@@ -529,21 +574,84 @@ 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: {
       functionName: "streamObject",
       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",
@@ -551,7 +659,7 @@ var aiSDKConfigs = [
     },
     functionQuery: {
       functionName: "streamObject",
-      kind: "Sync"
+      kind: "Async"
     }
   },
   // Agent.generate - async method (v5 only)
@@ -583,11 +691,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",
@@ -595,12 +703,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",
@@ -608,7 +716,7 @@ var aiSDKConfigs = [
     },
     functionQuery: {
       methodName: "stream",
-      kind: "Async",
+      kind: "Sync",
       index: 0
     }
   },
@@ -760,6 +868,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({
@@ -778,6 +918,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"
@@ -840,23 +984,244 @@ 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"
+    }
+  }
+];
+
 // src/auto-instrumentations/loader/cjs-patch.ts
 import {
   create
 } from "@apm-js-collab/code-transformer";
-import * as Module from "module";
+import * as NodeModule from "module";
 import { sep } from "path";
 import moduleDetailsFromPath from "module-details-from-path";
 var ModulePatch = class {
   packages;
   instrumentator;
+  modulePrototype;
   originalCompile;
   constructor({
     instrumentations = []
   } = {}) {
+    const modulePrototype = resolveModulePrototype();
     this.packages = new Set(instrumentations.map((i) => i.module.name));
     this.instrumentator = 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.
@@ -865,7 +1230,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(sep);
       const resolvedModule = moduleDetailsFromPath(normalizedForPlatform);
@@ -900,9 +1265,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;
+}
 
 // src/auto-instrumentations/patch-tracing-channel.ts
 function patchTracingChannel(tracingChannelFn) {
@@ -999,7 +1371,9 @@ var allConfigs = [
   ...aiSDKConfigs,
   ...claudeAgentSDKConfigs,
   ...googleGenAIConfigs,
-  ...openRouterConfigs
+  ...openRouterConfigs,
+  ...openRouterAgentConfigs,
+  ...mistralConfigs
 ];
 register("./loader/esm-hook.mjs", {
   parentURL: import.meta.url,