braintrust 3.27.0 → 3.29.0

package/dist/apply-auto-instrumentation.mjs

@@ -12,6 +12,7 @@ import {
   cloudflareThinkChannels,
   cohereChannels,
   cursorSDKChannels,
+  debugLogger,
   flueChannels,
   genkitChannels,
   genkitCoreChannels,
@@ -37,10 +38,11 @@ import {
   readDisabledInstrumentationEnvConfig,
   smithyClientChannels,
   smithyCoreChannels,
-  strandsAgentSDKChannels
-} from "./chunk-CZM5JIQL.mjs";
+  strandsAgentSDKChannels,
+  voyageAIChannels
+} from "./chunk-7FA6VP2S.mjs";
 
-// src/node/apply-auto-instrumentation-entry.ts
+// src/node/apply-auto-instrumentation.ts
 import { register } from "node:module";
 import { pathToFileURL } from "node:url";
 
@@ -93,6 +95,33 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
+  // generateImage - stable in v6+, experimental alias in v5
+  ...[
+    {
+      functionName: "generateImage",
+      versionRange: ">=6.0.0 <8.0.0",
+      isExportAlias: false
+    },
+    {
+      functionName: "experimental_generateImage",
+      versionRange: ">=5.0.0 <6.0.0",
+      isExportAlias: true
+    }
+  ].flatMap(
+    ({ functionName, versionRange: versionRange2, isExportAlias }) => ["dist/index.mjs", "dist/index.js"].map((filePath) => ({
+      channelName: aiSDKChannels.generateImage.channelName,
+      module: {
+        name: "ai",
+        versionRange: versionRange2,
+        filePath
+      },
+      functionQuery: {
+        functionName,
+        kind: "Async",
+        ...isExportAlias ? { isExportAlias } : {}
+      }
+    }))
+  ),
   // streamText - async function (v3 only, before the sync refactor in v4)
   {
     channelName: aiSDKChannels.streamText.channelName,
@@ -2721,6 +2750,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 },
@@ -2800,6 +2876,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
@@ -2824,8 +2904,9 @@ function getDefaultInstrumentationConfigs({
     ...additionalInstrumentations ?? []
   ];
 }
-function getDefaultAutoInstrumentationConfigs() {
+function getDefaultAutoInstrumentationConfigs(additionalInstrumentations) {
   return getDefaultInstrumentationConfigs({
+    additionalInstrumentations,
     disabledIntegrationConfig: readDisabledInstrumentationEnvConfig(
       process.env.BRAINTRUST_DISABLE_INSTRUMENTATION
     ).integrations
@@ -2853,11 +2934,11 @@ function formatChannelGetter(channelName) {
   return `tr_ch_bt$get_${channelName.replace(CHANNEL_REGEX, "_")}`;
 }
 var transforms = {
-  tracingHookDeclaration(state2, node) {
+  tracingHookDeclaration(state, node) {
     const {
       channelName,
       module: { name }
-    } = state2;
+    } = state;
     const channelVariable = formatChannelVariable(channelName);
     const channelGetter = formatChannelGetter(channelName);
     const hasSharedHookLookup = node.body.some(
@@ -2919,23 +3000,23 @@ var transforms = {
   tracePromise: traceAny,
   traceSync: traceAny
 };
-function traceAny(state2, node, _parent, ancestry) {
+function traceAny(state, node, _parent, ancestry) {
   const program = ancestry[ancestry.length - 1];
   if (node.type === "ClassDeclaration" || node.type === "ClassExpression") {
-    traceInstanceMethod(state2, node, program);
+    traceInstanceMethod(state, node, program);
   } else {
-    traceFunction(state2, node, program);
+    traceFunction(state, node, program);
   }
 }
-function traceFunction(state2, node, program) {
-  transforms.tracingHookDeclaration(state2, program, null, []);
-  const { functionQuery } = state2;
+function traceFunction(state, node, program) {
+  transforms.tracingHookDeclaration(state, program, null, []);
+  const { functionQuery } = state;
   const methodName = "methodName" in functionQuery ? functionQuery.methodName : void 0;
   const privateMethodName = "privateMethodName" in functionQuery ? functionQuery.privateMethodName : void 0;
   const functionName = "functionName" in functionQuery ? functionQuery.functionName : void 0;
   const isConstructor = methodName === "constructor" || !methodName && !privateMethodName && !functionName;
   const type = isConstructor ? "ArrowFunctionExpression" : "FunctionExpression";
-  node.body = wrap(state2, {
+  node.body = wrap(state, {
     type,
     params: node.params,
     body: node.body,
@@ -2947,8 +3028,8 @@ function traceFunction(state2, node, program) {
   node.async = false;
   wrapSuper(node);
 }
-function traceInstanceMethod(state2, node, program) {
-  const { functionQuery, operator } = state2;
+function traceInstanceMethod(state, node, program) {
+  const { functionQuery, operator } = state;
   const { methodName } = functionQuery;
   if (!methodName) {
     return;
@@ -2958,7 +3039,7 @@ function traceInstanceMethod(state2, node, program) {
     return;
   }
   let ctor = classBody.body.find(({ kind }) => kind === "constructor");
-  transforms.tracingHookDeclaration(state2, program, null, []);
+  transforms.tracingHookDeclaration(state, program, null, []);
   if (!ctor) {
     ctor = parse(
       node.superClass ? "class A extends Object { constructor (...args) { super(...args) } }" : "class A { constructor () {} }"
@@ -2971,15 +3052,15 @@ function traceInstanceMethod(state2, node, program) {
   `).body;
   const fn = ctorBody[1].expression.right;
   fn.async = operator === "tracePromise";
-  fn.body = wrap(state2, {
+  fn.body = wrap(state, {
     type: "Identifier",
     name: `__bt$${methodName}`
   });
   wrapSuper(fn);
   ctor.value.body.body.push(...ctorBody);
 }
-function wrap(state2, node) {
-  const wrapper = wrapInvocation(state2);
+function wrap(state, node) {
+  const wrapper = wrapInvocation(state);
   const block = wrapper.body[0].body;
   const common = parse(
     node.type === "ArrowFunctionExpression" ? `
@@ -3032,8 +3113,8 @@ function wrapSuper(node) {
     node.body.body.unshift(parse("const __bt$super = {}").body[0]);
   }
 }
-function wrapInvocation(state2) {
-  const { channelName, moduleVersion, operator, functionQuery } = state2;
+function wrapInvocation(state) {
+  const { channelName, moduleVersion, operator, functionQuery } = state;
   const channelGetter = formatChannelGetter(channelName);
   const callbackIndex = functionQuery.callbackIndex ?? -1;
   return parse(`
@@ -3101,7 +3182,7 @@ var Transformer = class {
         aliases
       );
       const query = astQuery || this.fromFunctionQuery(resolvedFunctionQuery);
-      const state2 = {
+      const state = {
         ...config,
         moduleVersion: this.version,
         functionQuery: resolvedFunctionQuery,
@@ -3109,7 +3190,7 @@ var Transformer = class {
       };
       esquery2.traverse(ast, esquery2.parse(query), (...args) => {
         injectionCount++;
-        this.visit(state2, ...args);
+        this.visit(state, ...args);
       });
     }
     if (injectionCount === 0 && this.configs.length > 0) {
@@ -3139,21 +3220,21 @@ var Transformer = class {
   }
   free() {
   }
-  visit(state2, ...args) {
-    const transform = transforms[state2.operator];
-    const { index = 0 } = state2.functionQuery;
+  visit(state, ...args) {
+    const transform = transforms[state.operator];
+    const { index = 0 } = state.functionQuery;
     const [node] = args;
     const type = node.init?.type || node.type;
     if (type !== "ClassDeclaration" && type !== "ClassExpression") {
       if (node.type === "VariableDeclarator") {
         return;
       }
-      state2.functionIndex = state2.functionIndex === void 0 ? 0 : state2.functionIndex + 1;
-      if (index !== null && index !== state2.functionIndex) {
+      state.functionIndex = state.functionIndex === void 0 ? 0 : state.functionIndex + 1;
+      if (index !== null && index !== state.functionIndex) {
         return;
       }
     }
-    transform(state2, ...args);
+    transform(state, ...args);
   }
   getOperator(kind) {
     switch (kind) {
@@ -3465,26 +3546,27 @@ function resolveModulePrototype() {
   return NodeModule.prototype;
 }
 
-// src/node/apply-auto-instrumentation-entry.ts
+// src/node/apply-auto-instrumentation.ts
 var stateKey = /* @__PURE__ */ Symbol.for(
   `braintrust.applyAutoInstrumentation.global-hooks.v${GLOBAL_INSTRUMENTATION_HOOKS_PROTOCOL_VERSION}`
 );
-var existingState = Object.getOwnPropertyDescriptor(
-  globalThis,
-  stateKey
-)?.value;
-var state = isApplyAutoInstrumentationState(
-  existingState
-) ? existingState : {};
-if (state !== existingState) {
-  Object.defineProperty(globalThis, stateKey, {
-    configurable: false,
-    enumerable: false,
-    value: state,
-    writable: false
-  });
-}
-if (!state.applied) {
+function applyAutoInstrumentation() {
+  const existingState = Object.getOwnPropertyDescriptor(
+    globalThis,
+    stateKey
+  )?.value;
+  const state = typeof existingState === "object" && existingState !== null ? existingState : {};
+  if (state !== existingState) {
+    Object.defineProperty(globalThis, stateKey, {
+      configurable: false,
+      enumerable: false,
+      value: state,
+      writable: false
+    });
+  }
+  if (state.applied) {
+    return;
+  }
   const allConfigs = getDefaultAutoInstrumentationConfigs();
   const currentModuleUrl = getCurrentModuleUrl();
   register("./auto-instrumentations/loader/esm-hook.mjs", {
@@ -3493,14 +3575,14 @@ if (!state.applied) {
   });
   state.applied = true;
   try {
-    const patch = new ModulePatch({ instrumentations: allConfigs });
-    patch.patch();
-  } catch {
+    new ModulePatch({ instrumentations: allConfigs }).patch();
+  } catch (error) {
+    debugLogger.warn(
+      "Failed to enable CommonJS auto-instrumentation; ESM instrumentation remains active:",
+      error
+    );
   }
 }
-function isApplyAutoInstrumentationState(value) {
-  return typeof value === "object" && value !== null;
-}
 function getCurrentModuleUrl() {
   if (typeof __filename !== "undefined") {
     return pathToFileURL(__filename).href;
@@ -3512,3 +3594,6 @@ function getCurrentModuleUrl() {
   }
   return pathToFileURL(process.argv[1] ?? process.cwd()).href;
 }
+
+// src/node/apply-auto-instrumentation-entry.ts
+applyAutoInstrumentation();

package/dist/auto-instrumentations/bundler/esbuild.cjs

@@ -1218,7 +1218,10 @@ var envIntegrationAliases = {
   "langchain-js": "langchain",
   "@langchain": "langchain",
   langgraph: "langgraph",
-  langsmith: "langsmith"
+  langsmith: "langsmith",
+  voyage: "voyageai",
+  "voyage-ai": "voyageai",
+  voyageai: "voyageai"
 };
 function readDisabledInstrumentationEnvConfig(disabledList) {
   const integrations = {};
@@ -1241,8 +1244,6 @@ function isInstrumentationIntegrationDisabled(integrations, ...names) {
 var DefaultAsyncLocalStorage = class {
   constructor() {
   }
-  enterWith(_) {
-  }
   run(_, callback) {
     return callback();
   }
@@ -1350,6 +1351,7 @@ var INSTRUMENTATION_NAMES = {
   CLOUDFLARE_THINK: "cloudflare-think",
   COHERE: "cohere",
   CURSOR_SDK: "cursor-sdk",
+  DEEPSEEK_HARNESS: "deepseek-harness",
   EVE: "eve",
   FLUE: "flue",
   GENKIT: "genkit",
@@ -1369,7 +1371,8 @@ var INSTRUMENTATION_NAMES = {
   OPENROUTER: "openrouter",
   OPENROUTER_AGENT: "openrouter-agent",
   PI_CODING_AGENT: "pi-coding-agent",
-  STRANDS_AGENT_SDK: "strands-agent-sdk"
+  STRANDS_AGENT_SDK: "strands-agent-sdk",
+  VOYAGEAI: "voyageai"
 };
 
 // src/instrumentation/plugins/ai-sdk-channels.ts
@@ -1380,6 +1383,10 @@ var aiSDKChannels = defineChannels(
       channelName: "generateText",
       kind: "async"
     }),
+    generateImage: channel({
+      channelName: "generateImage",
+      kind: "async"
+    }),
     streamText: channel({
       channelName: "streamText",
       kind: "async"
@@ -1519,6 +1526,33 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
+  // generateImage - stable in v6+, experimental alias in v5
+  ...[
+    {
+      functionName: "generateImage",
+      versionRange: ">=6.0.0 <8.0.0",
+      isExportAlias: false
+    },
+    {
+      functionName: "experimental_generateImage",
+      versionRange: ">=5.0.0 <6.0.0",
+      isExportAlias: true
+    }
+  ].flatMap(
+    ({ functionName, versionRange: versionRange2, isExportAlias }) => ["dist/index.mjs", "dist/index.js"].map((filePath) => ({
+      channelName: aiSDKChannels.generateImage.channelName,
+      module: {
+        name: "ai",
+        versionRange: versionRange2,
+        filePath
+      },
+      functionQuery: {
+        functionName,
+        kind: "Async",
+        ...isExportAlias ? { isExportAlias } : {}
+      }
+    }))
+  ),
   // streamText - async function (v3 only, before the sync refactor in v4)
   {
     channelName: aiSDKChannels.streamText.channelName,
@@ -4760,6 +4794,77 @@ var strandsAgentSDKConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/voyageai-channels.ts
+var voyageAIChannels = defineChannels(
+  "voyageai",
+  {
+    embed: channel({
+      channelName: "embed",
+      kind: "async"
+    }),
+    multimodalEmbed: channel({
+      channelName: "multimodalEmbed",
+      kind: "async"
+    }),
+    rerank: channel({
+      channelName: "rerank",
+      kind: "async"
+    }),
+    contextualizedEmbed: channel({
+      channelName: "contextualizedEmbed",
+      kind: "async"
+    })
+  },
+  { instrumentationName: INSTRUMENTATION_NAMES.VOYAGEAI }
+);
+
+// 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 },
@@ -4839,6 +4944,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
@@ -4863,6 +4972,14 @@ function getDefaultInstrumentationConfigs({
     ...additionalInstrumentations ?? []
   ];
 }
+function getDefaultAutoInstrumentationConfigs(additionalInstrumentations) {
+  return getDefaultInstrumentationConfigs({
+    additionalInstrumentations,
+    disabledIntegrationConfig: readDisabledInstrumentationEnvConfig(
+      process.env.BRAINTRUST_DISABLE_INSTRUMENTATION
+    ).integrations
+  });
+}
 
 // src/auto-instrumentations/loader/mastra-observability-patch.ts
 var MASTRA_EXPORTER_FACTORY_GLOBAL = "__braintrustMastraExporterFactory";
@@ -5129,9 +5246,9 @@ function getModuleVersion(basedir) {
 }
 var unplugin = (0, import_unplugin.createUnplugin)((options = {}) => {
   const browser = options.browser ?? options.useDiagnosticChannelCompatShim ?? false;
-  const allInstrumentations = getDefaultInstrumentationConfigs({
-    additionalInstrumentations: options.instrumentations
-  });
+  const allInstrumentations = getDefaultAutoInstrumentationConfigs(
+    options.instrumentations
+  );
   const instrumentationMatcher = create(allInstrumentations);
   return {
     name: "code-transformer",

package/dist/auto-instrumentations/bundler/esbuild.mjs

@@ -1,7 +1,7 @@
 import {
   unplugin
-} from "../chunk-XEYKUBLY.mjs";
-import "../chunk-ZNHTSSGI.mjs";
+} from "../chunk-DKTGDNA7.mjs";
+import "../chunk-AOIYCVEL.mjs";
 import "../chunk-I55G56ZL.mjs";
 import "../chunk-JPUFNW7X.mjs";
 import "../chunk-P5YLNB2A.mjs";