braintrust 3.8.0 → 3.9.0

package/dist/auto-instrumentations/{chunk-OLBMPZXE.mjs → chunk-KIMMUFAK.mjs}

@@ -369,6 +369,10 @@ var anthropicChannels = defineChannels("@anthropic-ai/sdk", {
   betaMessagesCreate: channel({
     channelName: "beta.messages.create",
     kind: "async"
+  }),
+  betaMessagesToolRunner: channel({
+    channelName: "beta.messages.toolRunner",
+    kind: "sync-stream"
   })
 });
 
@@ -415,6 +419,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"
+    }
   }
 ];
 
@@ -454,6 +472,10 @@ var aiSDKChannels = defineChannels("ai", {
     channelName: "embedMany",
     kind: "async"
   }),
+  rerank: channel({
+    channelName: "rerank",
+    kind: "async"
+  }),
   agentGenerate: channel({
     channelName: "Agent.generate",
     kind: "async"
@@ -533,7 +555,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: {
@@ -541,6 +563,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,
@@ -616,6 +650,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,
@@ -646,7 +705,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: {
@@ -654,6 +713,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.
@@ -823,6 +894,10 @@ var googleGenAIChannels = defineChannels("@google/genai", {
   generateContentStream: channel({
     channelName: "models.generateContentStream",
     kind: "async"
+  }),
+  embedContent: channel({
+    channelName: "models.embedContent",
+    kind: "async"
   })
 });
 
@@ -857,6 +932,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"
+    }
   }
 ];
 
@@ -902,6 +1274,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"
@@ -948,6 +1326,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: {
@@ -1194,13 +1585,197 @@ 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"
+    }
+  }
+];
+
 export {
+  channel,
+  defineChannels,
   openaiConfigs,
   anthropicConfigs,
   aiSDKConfigs,
   claudeAgentSDKConfigs,
   googleGenAIConfigs,
+  huggingFaceConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,
-  mistralConfigs
+  mistralConfigs,
+  cohereConfigs
 };