braintrust 3.8.0 → 3.9.0

package/dist/auto-instrumentations/bundler/webpack-loader.cjs

@@ -399,6 +399,10 @@ var anthropicChannels = defineChannels("@anthropic-ai/sdk", {
   betaMessagesCreate: channel({
     channelName: "beta.messages.create",
     kind: "async"
+  }),
+  betaMessagesToolRunner: channel({
+    channelName: "beta.messages.toolRunner",
+    kind: "sync-stream"
   })
 });
 
@@ -445,6 +449,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"
+    }
   }
 ];
 
@@ -484,6 +502,10 @@ var aiSDKChannels = defineChannels("ai", {
     channelName: "embedMany",
     kind: "async"
   }),
+  rerank: channel({
+    channelName: "rerank",
+    kind: "async"
+  }),
   agentGenerate: channel({
     channelName: "Agent.generate",
     kind: "async"
@@ -563,7 +585,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.streamText.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <4.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -571,6 +593,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,
@@ -646,6 +680,31 @@ var aiSDKConfigs = [
       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,
@@ -676,7 +735,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.streamObject.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <4.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -684,6 +743,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.
@@ -853,6 +924,10 @@ var googleGenAIChannels = defineChannels("@google/genai", {
   generateContentStream: channel({
     channelName: "models.generateContentStream",
     kind: "async"
+  }),
+  embedContent: channel({
+    channelName: "models.embedContent",
+    kind: "async"
   })
 });
 
@@ -887,6 +962,303 @@ 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/huggingface-channels.ts
+var huggingFaceChannels = defineChannels("@huggingface/inference", {
+  chatCompletion: channel({
+    channelName: "chatCompletion",
+    kind: "async"
+  }),
+  chatCompletionStream: channel({
+    channelName: "chatCompletionStream",
+    kind: "sync-stream"
+  }),
+  textGeneration: channel({
+    channelName: "textGeneration",
+    kind: "async"
+  }),
+  textGenerationStream: channel({
+    channelName: "textGenerationStream",
+    kind: "sync-stream"
+  }),
+  featureExtraction: channel({
+    channelName: "featureExtraction",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/huggingface.ts
+var huggingFaceConfigs = [
+  {
+    channelName: huggingFaceChannels.chatCompletion.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "chatCompletion",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.chatCompletion.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.cjs"
+    },
+    functionQuery: {
+      functionName: "chatCompletion",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: huggingFaceChannels.chatCompletionStream.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      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: huggingFaceChannels.textGeneration.channelName,
+    module: {
+      name: "@huggingface/inference",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "dist/index.cjs"
+    },
+    functionQuery: {
+      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"
+    }
+  },
+  {
+    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"
+    }
   }
 ];
 
@@ -932,6 +1304,12 @@ var openRouterChannels = defineChannels("@openrouter/sdk", {
     channelName: "embeddings.generate",
     kind: "async"
   }),
+  rerankRerank: channel(
+    {
+      channelName: "rerank.rerank",
+      kind: "async"
+    }
+  ),
   betaResponsesSend: channel({
     channelName: "beta.responses.send",
     kind: "async"
@@ -978,6 +1356,19 @@ var openRouterConfigs = [
       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: {
@@ -1224,6 +1615,186 @@ var mistralConfigs = [
   }
 ];
 
+// 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/bundler/webpack-loader.ts
 function getModuleVersion(basedir) {
   try {
@@ -1244,9 +1815,11 @@ function getMatcher(options) {
     ...aiSDKConfigs,
     ...claudeAgentSDKConfigs,
     ...googleGenAIConfigs,
+    ...huggingFaceConfigs,
     ...openRouterConfigs,
     ...openRouterAgentConfigs,
     ...mistralConfigs,
+    ...cohereConfigs,
     ...options.instrumentations ?? []
   ];
   const dcModule = options.browser ? "dc-browser" : void 0;