braintrust 3.7.1 → 3.9.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";
@@ -182,6 +182,10 @@ var openAIChannels = defineChannels("openai", {
   responsesParse: channel({
     channelName: "responses.parse",
     kind: "async"
+  }),
+  responsesCompact: channel({
+    channelName: "responses.compact",
+    kind: "async"
   })
 });
 
@@ -348,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"
+    }
   }
 ];
 
@@ -360,6 +377,10 @@ var anthropicChannels = defineChannels("@anthropic-ai/sdk", {
   betaMessagesCreate: channel({
     channelName: "beta.messages.create",
     kind: "async"
+  }),
+  betaMessagesToolRunner: channel({
+    channelName: "beta.messages.toolRunner",
+    kind: "sync-stream"
   })
 });
 
@@ -406,6 +427,20 @@ var anthropicConfigs = [
       methodName: "create",
       kind: "Async"
     }
+  },
+  // Beta Messages API - toolRunner (sync helper returning async iterable/thenable)
+  {
+    channelName: anthropicChannels.betaMessagesToolRunner.channelName,
+    module: {
+      name: "@anthropic-ai/sdk",
+      versionRange: ">=0.39.0",
+      filePath: "resources/beta/messages/messages.mjs"
+    },
+    functionQuery: {
+      className: "Messages",
+      methodName: "toolRunner",
+      kind: "Sync"
+    }
   }
 ];
 
@@ -419,6 +454,10 @@ var aiSDKChannels = defineChannels("ai", {
     channelName: "streamText",
     kind: "async"
   }),
+  streamTextSync: channel({
+    channelName: "streamText.sync",
+    kind: "sync-stream"
+  }),
   generateObject: channel({
     channelName: "generateObject",
     kind: "async"
@@ -427,6 +466,24 @@ var aiSDKChannels = defineChannels("ai", {
     channelName: "streamObject",
     kind: "async"
   }),
+  streamObjectSync: channel({
+    channelName: "streamObject.sync",
+    kind: "sync-stream"
+  }),
+  embed: channel(
+    {
+      channelName: "embed",
+      kind: "async"
+    }
+  ),
+  embedMany: channel({
+    channelName: "embedMany",
+    kind: "async"
+  }),
+  rerank: channel({
+    channelName: "rerank",
+    kind: "async"
+  }),
   agentGenerate: channel({
     channelName: "Agent.generate",
     kind: "async"
@@ -435,6 +492,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"
@@ -472,12 +533,12 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
-  // streamText - function returning stream
+  // 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: {
@@ -485,11 +546,24 @@ 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",
+      versionRange: ">=3.0.0 <4.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -497,6 +571,18 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: aiSDKChannels.streamTextSync.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=4.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "streamText",
+      kind: "Sync"
+    }
+  },
   // generateObject - async function
   {
     channelName: aiSDKChannels.generateObject.channelName,
@@ -522,12 +608,87 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
-  // streamObject - function returning stream
+  // 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"
+    }
+  },
+  // rerank - async function
+  {
+    channelName: aiSDKChannels.rerank.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=5.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "rerank",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: aiSDKChannels.rerank.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=5.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "rerank",
+      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: {
@@ -535,11 +696,24 @@ 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",
+      versionRange: ">=3.0.0 <4.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -547,6 +721,18 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: aiSDKChannels.streamObjectSync.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=4.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "streamObject",
+      kind: "Sync"
+    }
+  },
   // Agent.generate - async method (v5 only)
   // The compiled AI SDK bundle emits this as an anonymous class method, so we
   // target the first async `generate` method in the file instead of a class name.
@@ -576,11 +762,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",
@@ -588,12 +774,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",
@@ -601,7 +787,7 @@ var aiSDKConfigs = [
     },
     functionQuery: {
       methodName: "stream",
-      kind: "Async",
+      kind: "Sync",
       index: 0
     }
   },
@@ -716,6 +902,10 @@ var googleGenAIChannels = defineChannels("@google/genai", {
   generateContentStream: channel({
     channelName: "models.generateContentStream",
     kind: "async"
+  }),
+  embedContent: channel({
+    channelName: "models.embedContent",
+    kind: "async"
   })
 });
 
@@ -750,110 +940,1019 @@ var googleGenAIConfigs = [
       methodName: "generateContentStreamInternal",
       kind: "Async"
     }
+  },
+  // Models.embedContent - class method in older SDK versions
+  {
+    channelName: googleGenAIChannels.embedContent.channelName,
+    module: {
+      name: "@google/genai",
+      versionRange: ">=1.0.0 <1.44.0",
+      filePath: "dist/node/index.mjs"
+    },
+    functionQuery: {
+      className: "Models",
+      methodName: "embedContent",
+      kind: "Async"
+    }
+  },
+  // Models.embedContentInternal - class method in newer SDK versions
+  // Note: embedContent is an arrow function property that calls this method
+  {
+    channelName: googleGenAIChannels.embedContent.channelName,
+    module: {
+      name: "@google/genai",
+      versionRange: ">=1.44.0",
+      filePath: "dist/node/index.mjs"
+    },
+    functionQuery: {
+      className: "Models",
+      methodName: "embedContentInternal",
+      kind: "Async"
+    }
   }
 ];
 
-// src/instrumentation/plugins/openrouter-channels.ts
-var openRouterChannels = defineChannels("@openrouter/sdk", {
-  chatSend: channel({
-    channelName: "chat.send",
+// src/instrumentation/plugins/huggingface-channels.ts
+var huggingFaceChannels = defineChannels("@huggingface/inference", {
+  chatCompletion: channel({
+    channelName: "chatCompletion",
     kind: "async"
   }),
-  embeddingsGenerate: channel({
-    channelName: "embeddings.generate",
-    kind: "async"
+  chatCompletionStream: channel({
+    channelName: "chatCompletionStream",
+    kind: "sync-stream"
   }),
-  betaResponsesSend: channel({
-    channelName: "beta.responses.send",
+  textGeneration: channel({
+    channelName: "textGeneration",
     kind: "async"
   }),
-  callModel: channel({
-    channelName: "callModel",
+  textGenerationStream: channel({
+    channelName: "textGenerationStream",
     kind: "sync-stream"
   }),
-  callModelTurn: channel({
-    channelName: "callModel.turn",
-    kind: "async"
-  }),
-  toolExecute: channel({
-    channelName: "tool.execute",
+  featureExtraction: channel({
+    channelName: "featureExtraction",
     kind: "async"
   })
 });
 
-// src/auto-instrumentations/configs/openrouter.ts
-var openRouterConfigs = [
+// src/auto-instrumentations/configs/huggingface.ts
+var huggingFaceConfigs = [
   {
-    channelName: openRouterChannels.chatSend.channelName,
+    channelName: huggingFaceChannels.chatCompletion.channelName,
     module: {
-      name: "@openrouter/sdk",
-      versionRange: ">=0.9.11 <1.0.0",
-      filePath: "esm/sdk/chat.js"
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.js"
     },
     functionQuery: {
-      className: "Chat",
-      methodName: "send",
+      functionName: "chatCompletion",
       kind: "Async"
     }
   },
   {
-    channelName: openRouterChannels.embeddingsGenerate.channelName,
+    channelName: huggingFaceChannels.chatCompletion.channelName,
     module: {
-      name: "@openrouter/sdk",
-      versionRange: ">=0.9.11 <1.0.0",
-      filePath: "esm/sdk/embeddings.js"
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.cjs"
     },
     functionQuery: {
-      className: "Embeddings",
-      methodName: "generate",
+      functionName: "chatCompletion",
       kind: "Async"
     }
   },
   {
-    channelName: openRouterChannels.betaResponsesSend.channelName,
+    channelName: huggingFaceChannels.chatCompletionStream.channelName,
     module: {
-      name: "@openrouter/sdk",
-      versionRange: ">=0.9.11 <1.0.0",
-      filePath: "esm/sdk/responses.js"
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.js"
     },
     functionQuery: {
-      className: "Responses",
-      methodName: "send",
+      functionName: "chatCompletionStream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.chatCompletionStream.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.cjs"
+    },
+    functionQuery: {
+      functionName: "chatCompletionStream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.textGeneration.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "textGeneration",
       kind: "Async"
     }
   },
   {
-    channelName: openRouterChannels.callModel.channelName,
+    channelName: huggingFaceChannels.textGeneration.channelName,
     module: {
-      name: "@openrouter/sdk",
-      versionRange: ">=0.9.11 <1.0.0",
-      filePath: "esm/sdk/sdk.js"
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.cjs"
     },
     functionQuery: {
-      className: "OpenRouter",
-      methodName: "callModel",
+      functionName: "textGeneration",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.textGenerationStream.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "textGenerationStream",
       kind: "Sync"
     }
-  }
-];
-
-// src/auto-instrumentations/loader/cjs-patch.ts
-import {
-  create
-} from "@apm-js-collab/code-transformer";
-import * as Module from "module";
-import { sep } from "path";
-import moduleDetailsFromPath from "module-details-from-path";
-var ModulePatch = class {
-  packages;
-  instrumentator;
-  originalCompile;
-  constructor({
-    instrumentations = []
-  } = {}) {
+  },
+  {
+    channelName: huggingFaceChannels.textGenerationStream.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.cjs"
+    },
+    functionQuery: {
+      functionName: "textGenerationStream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.featureExtraction.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "featureExtraction",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.featureExtraction.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.cjs"
+    },
+    functionQuery: {
+      functionName: "featureExtraction",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.chatCompletion.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist/esm/tasks/nlp/chatCompletion.js"
+    },
+    functionQuery: {
+      functionName: "chatCompletion",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.chatCompletion.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist/commonjs/tasks/nlp/chatCompletion.js"
+    },
+    functionQuery: {
+      functionName: "chatCompletion",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.chatCompletionStream.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist/esm/tasks/nlp/chatCompletionStream.js"
+    },
+    functionQuery: {
+      functionName: "chatCompletionStream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.chatCompletionStream.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist/commonjs/tasks/nlp/chatCompletionStream.js"
+    },
+    functionQuery: {
+      functionName: "chatCompletionStream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.textGeneration.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist/esm/tasks/nlp/textGeneration.js"
+    },
+    functionQuery: {
+      functionName: "textGeneration",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.textGeneration.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist/commonjs/tasks/nlp/textGeneration.js"
+    },
+    functionQuery: {
+      functionName: "textGeneration",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.textGenerationStream.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist/esm/tasks/nlp/textGenerationStream.js"
+    },
+    functionQuery: {
+      functionName: "textGenerationStream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.textGenerationStream.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist/commonjs/tasks/nlp/textGenerationStream.js"
+    },
+    functionQuery: {
+      functionName: "textGenerationStream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.featureExtraction.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist/esm/tasks/nlp/featureExtraction.js"
+    },
+    functionQuery: {
+      functionName: "featureExtraction",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.featureExtraction.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist/commonjs/tasks/nlp/featureExtraction.js"
+    },
+    functionQuery: {
+      functionName: "featureExtraction",
+      kind: "Async"
+    }
+  }
+];
+
+// 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({
+    channelName: "chat.send",
+    kind: "async"
+  }),
+  embeddingsGenerate: channel({
+    channelName: "embeddings.generate",
+    kind: "async"
+  }),
+  rerankRerank: channel(
+    {
+      channelName: "rerank.rerank",
+      kind: "async"
+    }
+  ),
+  betaResponsesSend: channel({
+    channelName: "beta.responses.send",
+    kind: "async"
+  }),
+  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.ts
+var openRouterConfigs = [
+  {
+    channelName: openRouterChannels.chatSend.channelName,
+    module: {
+      name: "@openrouter/sdk",
+      versionRange: ">=0.9.11 <1.0.0",
+      filePath: "esm/sdk/chat.js"
+    },
+    functionQuery: {
+      className: "Chat",
+      methodName: "send",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: openRouterChannels.embeddingsGenerate.channelName,
+    module: {
+      name: "@openrouter/sdk",
+      versionRange: ">=0.9.11 <1.0.0",
+      filePath: "esm/sdk/embeddings.js"
+    },
+    functionQuery: {
+      className: "Embeddings",
+      methodName: "generate",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: openRouterChannels.rerankRerank.channelName,
+    module: {
+      name: "@openrouter/sdk",
+      versionRange: ">=0.12.0 <1.0.0",
+      filePath: "esm/sdk/rerank.js"
+    },
+    functionQuery: {
+      className: "Rerank",
+      methodName: "rerank",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: openRouterChannels.betaResponsesSend.channelName,
+    module: {
+      name: "@openrouter/sdk",
+      versionRange: ">=0.9.11 <1.0.0",
+      filePath: "esm/sdk/responses.js"
+    },
+    functionQuery: {
+      className: "Responses",
+      methodName: "send",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: openRouterChannels.callModel.channelName,
+    module: {
+      name: "@openrouter/sdk",
+      versionRange: ">=0.9.11 <1.0.0",
+      filePath: "esm/sdk/sdk.js"
+    },
+    functionQuery: {
+      className: "OpenRouter",
+      methodName: "callModel",
+      kind: "Sync"
+    }
+  }
+];
+
+// 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/instrumentation/plugins/google-adk-channels.ts
+var googleADKChannels = defineChannels("@google/adk", {
+  runnerRunAsync: channel({
+    channelName: "runner.runAsync",
+    kind: "sync-stream"
+  }),
+  agentRunAsync: channel({
+    channelName: "agent.runAsync",
+    kind: "sync-stream"
+  }),
+  toolRunAsync: channel({
+    channelName: "tool.runAsync",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/google-adk.ts
+var googleADKVersionRange = ">=0.1.0";
+var googleADKBundledIndexVersionRange = ">=0.6.1 <0.7.0";
+var googleADKConfigs = [
+  // --- Runner.runAsync --- async generator, kind "Sync" + sync-stream channel
+  // Runner.runAsync — ESM individual module file
+  {
+    channelName: googleADKChannels.runnerRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKVersionRange,
+      filePath: "dist/esm/runner/runner.js"
+    },
+    functionQuery: {
+      className: "Runner",
+      methodName: "runAsync",
+      kind: "Sync"
+    }
+  },
+  // Runner.runAsync — bundled CJS/ESM indexes
+  // The bundled entrypoints minify class names, so target the 12th sync
+  // `runAsync` method in file order rather than a class name. This mapping is
+  // only validated against the current 0.6.x bundle layout, so keep the range
+  // tight until we verify newer bundled outputs.
+  {
+    channelName: googleADKChannels.runnerRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexVersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 11
+    }
+  },
+  {
+    channelName: googleADKChannels.runnerRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexVersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 11
+    }
+  },
+  // --- BaseAgent.runAsync --- async generator, kind "Sync" + sync-stream channel
+  // BaseAgent.runAsync — ESM individual module file
+  {
+    channelName: googleADKChannels.agentRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKVersionRange,
+      filePath: "dist/esm/agents/base_agent.js"
+    },
+    functionQuery: {
+      className: "BaseAgent",
+      methodName: "runAsync",
+      kind: "Sync"
+    }
+  },
+  // BaseAgent.runAsync — bundled CJS/ESM indexes
+  // The bundled entrypoints minify class names, so target the first sync
+  // `runAsync` method in file order rather than a class name. This mapping is
+  // only validated against the current 0.6.x bundle layout, so keep the range
+  // tight until we verify newer bundled outputs.
+  {
+    channelName: googleADKChannels.agentRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexVersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 0
+    }
+  },
+  {
+    channelName: googleADKChannels.agentRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexVersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 0
+    }
+  },
+  // --- FunctionTool.runAsync --- regular async, kind "Async"
+  // FunctionTool.runAsync — ESM individual module file
+  {
+    channelName: googleADKChannels.toolRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKVersionRange,
+      filePath: "dist/esm/tools/function_tool.js"
+    },
+    functionQuery: {
+      className: "FunctionTool",
+      methodName: "runAsync",
+      kind: "Async"
+    }
+  },
+  // FunctionTool.runAsync — bundled CJS/ESM indexes
+  // The bundled entrypoints minify class names, so target the first async
+  // `runAsync` method in file order rather than a class name. This mapping is
+  // only validated against the current 0.6.x bundle layout, so keep the range
+  // tight until we verify newer bundled outputs.
+  {
+    channelName: googleADKChannels.toolRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexVersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Async",
+      index: 1
+    }
+  },
+  {
+    channelName: googleADKChannels.toolRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexVersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Async",
+      index: 1
+    }
+  }
+];
+
+// src/instrumentation/plugins/cohere-channels.ts
+var cohereChannels = defineChannels("cohere-ai", {
+  chat: channel({
+    channelName: "chat",
+    kind: "async"
+  }),
+  chatStream: channel({
+    channelName: "chatStream",
+    kind: "async"
+  }),
+  embed: channel({
+    channelName: "embed",
+    kind: "async"
+  }),
+  rerank: channel({
+    channelName: "rerank",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/cohere.ts
+var cohereConfigs = [
+  {
+    channelName: cohereChannels.chat.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=7.0.0 <8.0.0",
+      filePath: "Client.js"
+    },
+    functionQuery: {
+      className: "CohereClient",
+      methodName: "chat",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.chat.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=8.0.0 <9.0.0",
+      filePath: "Client.js"
+    },
+    functionQuery: {
+      className: "CohereClient",
+      methodName: "chat",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.chat.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=8.0.0 <9.0.0",
+      filePath: "api/resources/v2/client/Client.js"
+    },
+    functionQuery: {
+      className: "V2Client",
+      methodName: "chat",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.chatStream.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=7.0.0 <8.0.0",
+      filePath: "Client.js"
+    },
+    functionQuery: {
+      className: "CohereClient",
+      methodName: "chatStream",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.chatStream.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=8.0.0 <9.0.0",
+      filePath: "Client.js"
+    },
+    functionQuery: {
+      className: "CohereClient",
+      methodName: "chatStream",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.chatStream.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=8.0.0 <9.0.0",
+      filePath: "api/resources/v2/client/Client.js"
+    },
+    functionQuery: {
+      className: "V2Client",
+      methodName: "chatStream",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.embed.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=7.0.0 <8.0.0",
+      filePath: "Client.js"
+    },
+    functionQuery: {
+      className: "CohereClient",
+      methodName: "embed",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.embed.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=8.0.0 <9.0.0",
+      filePath: "Client.js"
+    },
+    functionQuery: {
+      className: "CohereClient",
+      methodName: "embed",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.embed.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=8.0.0 <9.0.0",
+      filePath: "api/resources/v2/client/Client.js"
+    },
+    functionQuery: {
+      className: "V2Client",
+      methodName: "embed",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.rerank.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=7.0.0 <8.0.0",
+      filePath: "Client.js"
+    },
+    functionQuery: {
+      className: "CohereClient",
+      methodName: "rerank",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.rerank.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=8.0.0 <9.0.0",
+      filePath: "Client.js"
+    },
+    functionQuery: {
+      className: "CohereClient",
+      methodName: "rerank",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.rerank.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=8.0.0 <9.0.0",
+      filePath: "api/resources/v2/client/Client.js"
+    },
+    functionQuery: {
+      className: "V2Client",
+      methodName: "rerank",
+      kind: "Async"
+    }
+  }
+];
+
+// src/auto-instrumentations/loader/cjs-patch.ts
+import {
+  create
+} from "@apm-js-collab/code-transformer";
+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.
@@ -862,7 +1961,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);
@@ -897,9 +1996,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) {
@@ -990,13 +2096,31 @@ var dc = await import(
   dcPath
 );
 patchTracingChannel(dc.tracingChannel);
+function readDisabledIntegrations() {
+  const raw = process.env.BRAINTRUST_DISABLE_INSTRUMENTATION;
+  if (!raw) {
+    return /* @__PURE__ */ new Set();
+  }
+  return new Set(
+    raw.split(",").map((value) => value.trim().toLowerCase()).filter((value) => value.length > 0)
+  );
+}
+function isDisabled(disabled, ...names) {
+  return names.some((name) => disabled.has(name));
+}
+var disabledIntegrations = readDisabledIntegrations();
 var allConfigs = [
-  ...openaiConfigs,
-  ...anthropicConfigs,
-  ...aiSDKConfigs,
-  ...claudeAgentSDKConfigs,
-  ...googleGenAIConfigs,
-  ...openRouterConfigs
+  ...isDisabled(disabledIntegrations, "openai") ? [] : openaiConfigs,
+  ...isDisabled(disabledIntegrations, "anthropic") ? [] : anthropicConfigs,
+  ...isDisabled(disabledIntegrations, "aisdk", "ai-sdk", "vercel-ai") ? [] : aiSDKConfigs,
+  ...isDisabled(disabledIntegrations, "claudeagentsdk", "claude-agent-sdk") ? [] : claudeAgentSDKConfigs,
+  ...isDisabled(disabledIntegrations, "google", "google-genai") ? [] : googleGenAIConfigs,
+  ...isDisabled(disabledIntegrations, "huggingface") ? [] : huggingFaceConfigs,
+  ...isDisabled(disabledIntegrations, "openrouter") ? [] : openRouterConfigs,
+  ...isDisabled(disabledIntegrations, "openrouteragent", "openrouter-agent") ? [] : openRouterAgentConfigs,
+  ...isDisabled(disabledIntegrations, "mistral") ? [] : mistralConfigs,
+  ...isDisabled(disabledIntegrations, "googleadk", "google-adk") ? [] : googleADKConfigs,
+  ...isDisabled(disabledIntegrations, "cohere") ? [] : cohereConfigs
 ];
 register("./loader/esm-hook.mjs", {
   parentURL: import.meta.url,