braintrust 3.26.0 → 3.28.0

package/dist/apply-auto-instrumentation.mjs

@@ -9,6 +9,7 @@ import {
   claudeAgentSDKChannels,
   cloudflareAIChatChannels,
   cloudflareAgentsChannels,
+  cloudflareThinkChannels,
   cohereChannels,
   cursorSDKChannels,
   flueChannels,
@@ -25,6 +26,7 @@ import {
   langChainChannels,
   langSmithChannels,
   mistralChannels,
+  ollamaChannels,
   openAIAgentsCoreChannels,
   openAIChannels,
   openAICodexChannels,
@@ -35,8 +37,9 @@ import {
   readDisabledInstrumentationEnvConfig,
   smithyClientChannels,
   smithyCoreChannels,
-  strandsAgentSDKChannels
-} from "./chunk-PN7EWK66.mjs";
+  strandsAgentSDKChannels,
+  voyageAIChannels
+} from "./chunk-ZHUHZWFY.mjs";
 
 // src/node/apply-auto-instrumentation-entry.ts
 import { register } from "node:module";
@@ -549,7 +552,40 @@ var anthropicConfigs = [
       methodName: "toolRunner",
       kind: "Sync"
     }
-  }
+  },
+  // Managed Agents Sessions event streams are discovered automatically, but
+  // remain passive until the user calls collectAnthropicSession() on the
+  // returned stream.
+  ...["mjs", "js"].map(
+    (extension) => ({
+      channelName: anthropicChannels.betaSessionsEventsStream.channelName,
+      module: {
+        name: "@anthropic-ai/sdk",
+        versionRange: ">=0.86.0",
+        filePath: `resources/beta/sessions/events.${extension}`
+      },
+      functionQuery: {
+        className: "Events",
+        methodName: "stream",
+        kind: "Async"
+      }
+    })
+  ),
+  ...["mjs", "js"].map(
+    (extension) => ({
+      channelName: anthropicChannels.betaSessionsThreadsEventsStream.channelName,
+      module: {
+        name: "@anthropic-ai/sdk",
+        versionRange: ">=0.86.0",
+        filePath: `resources/beta/sessions/threads/events.${extension}`
+      },
+      functionQuery: {
+        className: "Events",
+        methodName: "stream",
+        kind: "Async"
+      }
+    })
+  )
 ];
 
 // src/auto-instrumentations/configs/bedrock-runtime.ts
@@ -679,6 +715,24 @@ var cloudflareAgentsConfigs = [
   }
 ];
 
+// src/auto-instrumentations/configs/cloudflare-think.ts
+var cloudflareThinkVersionRange = ">=0.13.0 <0.14.0";
+var cloudflareThinkConfigs = [
+  {
+    channelName: cloudflareThinkChannels.runInferenceLoop.channelName,
+    module: {
+      name: "@cloudflare/think",
+      versionRange: cloudflareThinkVersionRange,
+      filePath: "dist/think.js"
+    },
+    functionQuery: {
+      className: "Think",
+      methodName: "_runInferenceLoop",
+      kind: "Async"
+    }
+  }
+];
+
 // src/auto-instrumentations/configs/cohere.ts
 var cohereConfigs = [
   {
@@ -2248,6 +2302,31 @@ var mistralConfigs = [
   }
 ];
 
+// src/auto-instrumentations/configs/ollama.ts
+var methods = [
+  ["chat", ollamaChannels.chat.channelName],
+  ["generate", ollamaChannels.generate.channelName],
+  ["embed", ollamaChannels.embed.channelName]
+];
+var implementationFiles = ["dist/browser.mjs", "dist/browser.cjs"];
+var ollamaConfigs = [
+  ...implementationFiles.flatMap(
+    (filePath) => methods.map(([methodName, channelName]) => ({
+      channelName,
+      module: {
+        name: "ollama",
+        versionRange: ">=0.6.0 <0.7.0",
+        filePath
+      },
+      functionQuery: {
+        className: "Ollama",
+        methodName,
+        kind: "Async"
+      }
+    }))
+  )
+];
+
 // src/auto-instrumentations/configs/openai-agents.ts
 var lifecycleMethods = [
   ["onTraceStart", openAIAgentsCoreChannels.onTraceStart.channelName],
@@ -2537,6 +2616,19 @@ var openRouterConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: openRouterChannels.betaResponsesSend.channelName,
+    module: {
+      name: "@openrouter/sdk",
+      versionRange: ">=1.1.2 <2.0.0",
+      filePath: "esm/sdk/betaresponses.js"
+    },
+    functionQuery: {
+      className: "BetaResponses",
+      methodName: "send",
+      kind: "Async"
+    }
+  },
   {
     channelName: openRouterChannels.callModel.channelName,
     module: {
@@ -2630,6 +2722,53 @@ var strandsAgentSDKConfigs = [
   }
 ];
 
+// src/auto-instrumentations/configs/voyageai.ts
+var GENERATED_CLIENT_MODULES = [
+  "dist/cjs/Client.js",
+  "dist/esm/Client.mjs"
+];
+var generatedClientOperations = [
+  {
+    channelName: voyageAIChannels.embed.channelName,
+    methodName: "embed",
+    versionRange: ">=0.2.0 <1.0.0"
+  },
+  {
+    channelName: voyageAIChannels.multimodalEmbed.channelName,
+    methodName: "multimodalEmbed",
+    versionRange: ">=0.2.0 <1.0.0"
+  },
+  {
+    channelName: voyageAIChannels.rerank.channelName,
+    methodName: "rerank",
+    versionRange: ">=0.2.0 <1.0.0"
+  },
+  {
+    channelName: voyageAIChannels.contextualizedEmbed.channelName,
+    methodName: "contextualizedEmbed",
+    versionRange: ">=0.2.0 <1.0.0"
+  }
+];
+var voyageAIConfigs = [
+  ...GENERATED_CLIENT_MODULES.flatMap(
+    (filePath) => generatedClientOperations.map(
+      ({ channelName, methodName, versionRange: versionRange2 }) => ({
+        channelName,
+        module: {
+          name: "voyageai",
+          versionRange: versionRange2,
+          filePath
+        },
+        functionQuery: {
+          className: "VoyageAIClient",
+          methodName,
+          kind: "Async"
+        }
+      })
+    )
+  )
+];
+
 // src/auto-instrumentations/configs/all.ts
 var defaultInstrumentationConfigGroups = [
   { integrations: ["openai"], configs: openaiConfigs },
@@ -2658,6 +2797,10 @@ var defaultInstrumentationConfigGroups = [
     integrations: ["cloudflareAgents"],
     configs: cloudflareAgentsConfigs
   },
+  {
+    integrations: ["cloudflareThink"],
+    configs: cloudflareThinkConfigs
+  },
   { integrations: ["cursor", "cursorSDK"], configs: cursorSDKConfigs },
   {
     integrations: ["openAIAgents"],
@@ -2682,6 +2825,7 @@ var defaultInstrumentationConfigGroups = [
     configs: openRouterAgentConfigs
   },
   { integrations: ["mistral"], configs: mistralConfigs },
+  { integrations: ["ollama"], configs: ollamaConfigs },
   { integrations: ["googleADK"], configs: googleADKConfigs },
   { integrations: ["cohere"], configs: cohereConfigs },
   { integrations: ["groq"], configs: groqConfigs },
@@ -2704,6 +2848,10 @@ var defaultInstrumentationConfigGroups = [
   {
     integrations: ["flue"],
     configs: flueConfigs
+  },
+  {
+    integrations: ["voyageai"],
+    configs: voyageAIConfigs
   }
   // Note: `@mastra/core` is not listed here because its instrumentation
   // doesn't go through the AST `code-transformer` matcher — Mastra's
@@ -2760,8 +2908,7 @@ var transforms = {
   tracingHookDeclaration(state2, node) {
     const {
       channelName,
-      module: { name },
-      operator
+      module: { name }
     } = state2;
     const channelVariable = formatChannelVariable(channelName);
     const channelGetter = formatChannelGetter(channelName);
@@ -2774,7 +2921,7 @@ var transforms = {
       return;
     }
     const sharedHookLookup = hasSharedHookLookup ? "" : `
-      const ${SHARED_HOOK_LOOKUP} = (__bt$hookName, __bt$operator) => {
+      const ${SHARED_HOOK_LOOKUP} = (__bt$hookName) => {
         try {
           const __bt$hooks = globalThis[${JSON.stringify(
       GLOBAL_INSTRUMENTATION_HOOKS_KEY
@@ -2796,8 +2943,7 @@ var transforms = {
             __bt$hook[Symbol.for(${JSON.stringify(
       GLOBAL_INSTRUMENTATION_HOOK_BRAND
     )})] !== ${GLOBAL_INSTRUMENTATION_HOOKS_PROTOCOL_VERSION} ||
-            typeof __bt$hook.hasSubscribers !== "boolean" ||
-            typeof __bt$hook[__bt$operator] !== "function"
+            typeof __bt$hook.traceInvocation !== "function"
           ) return undefined;
           return __bt$hook;
         } catch {
@@ -2810,8 +2956,7 @@ var transforms = {
       let ${channelVariable};
       const ${channelGetter} = () =>
         ${channelVariable} ??= ${SHARED_HOOK_LOOKUP}(
-          ${JSON.stringify(`orchestrion:${name}:${channelName}`)},
-          ${JSON.stringify(operator)}
+          ${JSON.stringify(`orchestrion:${name}:${channelName}`)}
         );
     `;
     const sharedHookLookupIndex = node.body.findIndex(
@@ -2886,28 +3031,18 @@ function traceInstanceMethod(state2, node, program) {
   ctor.value.body.body.push(...ctorBody);
 }
 function wrap(state2, node) {
-  const { operator, moduleVersion } = state2;
-  const wrapper = operator === "traceCallback" ? wrapCallback(state2) : operator === "tracePromise" ? wrapPromise(state2) : wrapSync(state2);
+  const wrapper = wrapInvocation(state2);
   const block = wrapper.body[0].body;
   const common = parse(
     node.type === "ArrowFunctionExpression" ? `
-    const __bt$ctx = {
-      arguments,
-      moduleVersion: ${JSON.stringify(moduleVersion)}
-    };
-    const __bt$traced = () => {
+    const __bt$target = (...__bt$args) => {
       const __bt$wrapped = () => {};
-      return __bt$wrapped(...arguments);
+      return __bt$wrapped(...__bt$args);
     };
   ` : `
-    const __bt$ctx = {
-      arguments,
-      self: this,
-      moduleVersion: ${JSON.stringify(moduleVersion)}
-    };
-    const __bt$traced = () => {
+    const __bt$target = function (...__bt$args) {
       const __bt$wrapped = () => {};
-      return __bt$wrapped.apply(this, arguments);
+      return __bt$wrapped.apply(this, __bt$args);
     };
   `
   ).body;
@@ -2949,49 +3084,25 @@ function wrapSuper(node) {
     node.body.body.unshift(parse("const __bt$super = {}").body[0]);
   }
 }
-function wrapCallback(state2) {
-  const {
-    channelName,
-    functionQuery: { callbackIndex = -1 }
-  } = state2;
+function wrapInvocation(state2) {
+  const { channelName, moduleVersion, operator, functionQuery } = state2;
   const channelGetter = formatChannelGetter(channelName);
+  const callbackIndex = functionQuery.callbackIndex ?? -1;
   return parse(`
     function wrapper () {
       const __bt$hook = ${channelGetter}();
-      if (!__bt$hook?.hasSubscribers) return __bt$traced();
-      __bt$ctx.self ??= this;
-      return __bt$hook.traceCallback(
-        __bt$traced,
-        ${callbackIndex},
-        __bt$ctx
+      if (!__bt$hook) return __bt$target.apply(this, arguments);
+      return __bt$hook.traceInvocation(
+        ${JSON.stringify(operator)},
+        __bt$target,
+        this,
+        arguments,
+        { moduleVersion: ${JSON.stringify(moduleVersion)} },
+        ${callbackIndex}
       );
     }
   `);
 }
-function wrapPromise(state2) {
-  const { channelName } = state2;
-  const channelGetter = formatChannelGetter(channelName);
-  return parse(`
-    function wrapper () {
-      const __bt$hook = ${channelGetter}();
-      if (!__bt$hook?.hasSubscribers) return __bt$traced();
-      __bt$ctx.self ??= this;
-      return __bt$hook.tracePromise(__bt$traced, __bt$ctx);
-    }
-  `);
-}
-function wrapSync(state2) {
-  const { channelName } = state2;
-  const channelGetter = formatChannelGetter(channelName);
-  return parse(`
-    function wrapper () {
-      const __bt$hook = ${channelGetter}();
-      if (!__bt$hook?.hasSubscribers) return __bt$traced();
-      __bt$ctx.self ??= this;
-      return __bt$hook.traceSync(__bt$traced, __bt$ctx);
-    }
-  `);
-}
 
 // src/auto-instrumentations/orchestrion-js/transformer.ts
 var Transformer = class {