braintrust 3.10.0 → 3.12.0

package/dist/auto-instrumentations/index.d.mts

@@ -14,6 +14,8 @@ export { InstrumentationConfig } from '@apm-js-collab/code-transformer';
  */
 declare const openaiConfigs: InstrumentationConfig[];
 
+declare const openAICodexConfigs: InstrumentationConfig[];
+
 /**
  * Instrumentation configurations for the Anthropic SDK.
  *
@@ -55,6 +57,8 @@ declare const claudeAgentSDKConfigs: InstrumentationConfig[];
 
 declare const cursorSDKConfigs: InstrumentationConfig[];
 
+declare const openAIAgentsCoreConfigs: InstrumentationConfig[];
+
 /**
  * Instrumentation configurations for the Google GenAI SDK.
  *
@@ -93,4 +97,17 @@ declare const cohereConfigs: InstrumentationConfig[];
 
 declare const groqConfigs: InstrumentationConfig[];
 
-export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, cursorSDKConfigs, googleADKConfigs, googleGenAIConfigs, groqConfigs, huggingFaceConfigs, mistralConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };
+/**
+ * Instrumentation configurations for Genkit's JavaScript SDK.
+ *
+ * Genkit's public instance methods live on the GenkitAI base class in
+ * @genkit-ai/ai. The top-level `genkit` package subclasses that class, so
+ * targeting these methods instruments regular `genkit({ ... })` instances.
+ */
+declare const genkitConfigs: InstrumentationConfig[];
+
+declare const gitHubCopilotConfigs: InstrumentationConfig[];
+
+declare const langchainConfigs: InstrumentationConfig[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, cursorSDKConfigs, genkitConfigs, gitHubCopilotConfigs, googleADKConfigs, googleGenAIConfigs, groqConfigs, huggingFaceConfigs, langchainConfigs, mistralConfigs, openAIAgentsCoreConfigs, openAICodexConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };

package/dist/auto-instrumentations/index.d.ts

@@ -14,6 +14,8 @@ export { InstrumentationConfig } from '@apm-js-collab/code-transformer';
  */
 declare const openaiConfigs: InstrumentationConfig[];
 
+declare const openAICodexConfigs: InstrumentationConfig[];
+
 /**
  * Instrumentation configurations for the Anthropic SDK.
  *
@@ -55,6 +57,8 @@ declare const claudeAgentSDKConfigs: InstrumentationConfig[];
 
 declare const cursorSDKConfigs: InstrumentationConfig[];
 
+declare const openAIAgentsCoreConfigs: InstrumentationConfig[];
+
 /**
  * Instrumentation configurations for the Google GenAI SDK.
  *
@@ -93,4 +97,17 @@ declare const cohereConfigs: InstrumentationConfig[];
 
 declare const groqConfigs: InstrumentationConfig[];
 
-export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, cursorSDKConfigs, googleADKConfigs, googleGenAIConfigs, groqConfigs, huggingFaceConfigs, mistralConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };
+/**
+ * Instrumentation configurations for Genkit's JavaScript SDK.
+ *
+ * Genkit's public instance methods live on the GenkitAI base class in
+ * @genkit-ai/ai. The top-level `genkit` package subclasses that class, so
+ * targeting these methods instruments regular `genkit({ ... })` instances.
+ */
+declare const genkitConfigs: InstrumentationConfig[];
+
+declare const gitHubCopilotConfigs: InstrumentationConfig[];
+
+declare const langchainConfigs: InstrumentationConfig[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, cursorSDKConfigs, genkitConfigs, gitHubCopilotConfigs, googleADKConfigs, googleGenAIConfigs, groqConfigs, huggingFaceConfigs, langchainConfigs, mistralConfigs, openAIAgentsCoreConfigs, openAICodexConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };

package/dist/auto-instrumentations/index.mjs

@@ -1,274 +1,39 @@
 import {
   aiSDKConfigs,
   anthropicConfigs,
-  channel,
   claudeAgentSDKConfigs,
   cohereConfigs,
   cursorSDKConfigs,
-  defineChannels,
+  genkitConfigs,
+  gitHubCopilotConfigs,
+  googleADKConfigs,
   googleGenAIConfigs,
   groqConfigs,
   huggingFaceConfigs,
+  langchainConfigs,
   mistralConfigs,
+  openAIAgentsCoreConfigs,
+  openAICodexConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,
   openaiConfigs
-} from "./chunk-GZNXBBPU.mjs";
-
-// 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 googleADKBundledIndexV06VersionRange = ">=0.6.1 <0.7.0";
-var googleADKBundledIndexV1VersionRange = ">=1.0.0 <2.0.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: googleADKBundledIndexV06VersionRange,
-      filePath: "dist/cjs/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Sync",
-      index: 11
-    }
-  },
-  {
-    channelName: googleADKChannels.runnerRunAsync.channelName,
-    module: {
-      name: "@google/adk",
-      versionRange: googleADKBundledIndexV06VersionRange,
-      filePath: "dist/esm/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Sync",
-      index: 11
-    }
-  },
-  // The 1.x bundled entrypoints still inline the runtime into index.js, but
-  // the minified method order changed. These indices are verified against 1.0.0.
-  {
-    channelName: googleADKChannels.runnerRunAsync.channelName,
-    module: {
-      name: "@google/adk",
-      versionRange: googleADKBundledIndexV1VersionRange,
-      filePath: "dist/cjs/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Sync",
-      index: 12
-    }
-  },
-  {
-    channelName: googleADKChannels.runnerRunAsync.channelName,
-    module: {
-      name: "@google/adk",
-      versionRange: googleADKBundledIndexV1VersionRange,
-      filePath: "dist/esm/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Sync",
-      index: 12
-    }
-  },
-  // --- 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: googleADKBundledIndexV06VersionRange,
-      filePath: "dist/cjs/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Sync",
-      index: 0
-    }
-  },
-  {
-    channelName: googleADKChannels.agentRunAsync.channelName,
-    module: {
-      name: "@google/adk",
-      versionRange: googleADKBundledIndexV06VersionRange,
-      filePath: "dist/esm/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Sync",
-      index: 0
-    }
-  },
-  // The 1.x bundled entrypoints keep BaseAgent.runAsync as the first bundled
-  // async-generator runAsync method in file order.
-  {
-    channelName: googleADKChannels.agentRunAsync.channelName,
-    module: {
-      name: "@google/adk",
-      versionRange: googleADKBundledIndexV1VersionRange,
-      filePath: "dist/cjs/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Sync",
-      index: 0
-    }
-  },
-  {
-    channelName: googleADKChannels.agentRunAsync.channelName,
-    module: {
-      name: "@google/adk",
-      versionRange: googleADKBundledIndexV1VersionRange,
-      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: googleADKBundledIndexV06VersionRange,
-      filePath: "dist/cjs/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Async",
-      index: 1
-    }
-  },
-  {
-    channelName: googleADKChannels.toolRunAsync.channelName,
-    module: {
-      name: "@google/adk",
-      versionRange: googleADKBundledIndexV06VersionRange,
-      filePath: "dist/esm/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Async",
-      index: 1
-    }
-  },
-  // The 1.x bundle moves FunctionTool.runAsync behind one helper method.
-  {
-    channelName: googleADKChannels.toolRunAsync.channelName,
-    module: {
-      name: "@google/adk",
-      versionRange: googleADKBundledIndexV1VersionRange,
-      filePath: "dist/cjs/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Async",
-      index: 1
-    }
-  },
-  {
-    channelName: googleADKChannels.toolRunAsync.channelName,
-    module: {
-      name: "@google/adk",
-      versionRange: googleADKBundledIndexV1VersionRange,
-      filePath: "dist/esm/index.js"
-    },
-    functionQuery: {
-      methodName: "runAsync",
-      kind: "Async",
-      index: 1
-    }
-  }
-];
+} from "./chunk-2DPA74KK.mjs";
 export {
   aiSDKConfigs,
   anthropicConfigs,
   claudeAgentSDKConfigs,
   cohereConfigs,
   cursorSDKConfigs,
+  genkitConfigs,
+  gitHubCopilotConfigs,
   googleADKConfigs,
   googleGenAIConfigs,
   groqConfigs,
   huggingFaceConfigs,
+  langchainConfigs,
   mistralConfigs,
+  openAIAgentsCoreConfigs,
+  openAICodexConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,
   openaiConfigs

package/dist/auto-instrumentations/loader/cjs-patch.cjs

@@ -93,6 +93,27 @@ function getPackageName(baseDir) {
   return void 0;
 }
 
+// src/auto-instrumentations/loader/openai-api-promise-patch.ts
+var OPENAI_API_PROMISE_PATCH = `
+;(function __btPatchAPIPromise() {
+  if (typeof APIPromise === "undefined" || APIPromise.prototype.__btParsePatched) return;
+  APIPromise.prototype.__btParsePatched = true;
+  var _origThen = APIPromise.prototype.then;
+  APIPromise.prototype.then = function __btThen(onfulfilled, onrejected) {
+    if (!this.__btParseWrapped && Object.prototype.hasOwnProperty.call(this, "parseResponse")) {
+      this.__btParseWrapped = true;
+      var _origParse = this.parseResponse;
+      var _cached;
+      this.parseResponse = function() {
+        if (!_cached) _cached = _origParse.apply(this, arguments);
+        return _cached;
+      };
+    }
+    return _origThen.call(this, onfulfilled, onrejected);
+  };
+})();
+`;
+
 // src/auto-instrumentations/loader/cjs-patch.ts
 var ModulePatch = class {
   packages;
@@ -124,8 +145,12 @@ var ModulePatch = class {
         if (!self.packages.has(packageName)) {
           return self.originalCompile.apply(this, args);
         }
-        const version = getPackageVersion(resolvedModule.basedir);
         const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+        if (packageName === "openai" && normalizedModulePath.includes("api-promise")) {
+          args[0] = content + OPENAI_API_PROMISE_PATCH;
+          return self.originalCompile.apply(this, args);
+        }
+        const version = getPackageVersion(resolvedModule.basedir);
         const transformer = self.instrumentator.getTransformer(
           packageName,
           version,

package/dist/auto-instrumentations/loader/cjs-patch.mjs

@@ -10,6 +10,29 @@ import {
 import * as NodeModule from "module";
 import { sep } from "path";
 import moduleDetailsFromPath from "module-details-from-path";
+
+// src/auto-instrumentations/loader/openai-api-promise-patch.ts
+var OPENAI_API_PROMISE_PATCH = `
+;(function __btPatchAPIPromise() {
+  if (typeof APIPromise === "undefined" || APIPromise.prototype.__btParsePatched) return;
+  APIPromise.prototype.__btParsePatched = true;
+  var _origThen = APIPromise.prototype.then;
+  APIPromise.prototype.then = function __btThen(onfulfilled, onrejected) {
+    if (!this.__btParseWrapped && Object.prototype.hasOwnProperty.call(this, "parseResponse")) {
+      this.__btParseWrapped = true;
+      var _origParse = this.parseResponse;
+      var _cached;
+      this.parseResponse = function() {
+        if (!_cached) _cached = _origParse.apply(this, arguments);
+        return _cached;
+      };
+    }
+    return _origThen.call(this, onfulfilled, onrejected);
+  };
+})();
+`;
+
+// src/auto-instrumentations/loader/cjs-patch.ts
 var ModulePatch = class {
   packages;
   instrumentator;
@@ -40,8 +63,12 @@ var ModulePatch = class {
         if (!self.packages.has(packageName)) {
           return self.originalCompile.apply(this, args);
         }
-        const version = getPackageVersion(resolvedModule.basedir);
         const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+        if (packageName === "openai" && normalizedModulePath.includes("api-promise")) {
+          args[0] = content + OPENAI_API_PROMISE_PATCH;
+          return self.originalCompile.apply(this, args);
+        }
+        const version = getPackageVersion(resolvedModule.basedir);
         const transformer = self.instrumentator.getTransformer(
           packageName,
           version,

package/dist/auto-instrumentations/loader/esm-hook.mjs

@@ -1,12 +1,13 @@
 import {
+  OPENAI_API_PROMISE_PATCH,
   getPackageName,
   getPackageVersion
-} from "../chunk-ITP7RAUY.mjs";
+} from "../chunk-MWZXZQUO.mjs";
 
 // src/auto-instrumentations/loader/esm-hook.mts
 import { readFile } from "fs/promises";
 import { fileURLToPath } from "url";
-import { sep } from "path";
+import { extname, sep } from "path";
 import {
   create
 } from "@apm-js-collab/code-transformer";
@@ -14,6 +15,23 @@ import moduleDetailsFromPath from "module-details-from-path";
 var instrumentator;
 var packages;
 var transformers = /* @__PURE__ */ new Map();
+function getModuleType(url, format) {
+  if (format === "module") {
+    return "esm";
+  }
+  if (format === "commonjs") {
+    return "cjs";
+  }
+  const pathname = url.startsWith("file:") ? fileURLToPath(url) : url;
+  const ext = extname(pathname);
+  if (ext === ".mjs") {
+    return "esm";
+  }
+  if (ext === ".cjs") {
+    return "cjs";
+  }
+  return "unknown";
+}
 async function initialize(data = {}) {
   const configs = data.instrumentations || [];
   instrumentator = create(configs);
@@ -42,8 +60,22 @@ async function resolve(specifier, context, nextResolve) {
   }
   return url;
 }
+function isOpenAIApiPromise(url) {
+  return url.includes("/openai") && url.includes("api-promise");
+}
 async function load(url, context, nextLoad) {
   const result = await nextLoad(url, context);
+  if (isOpenAIApiPromise(url)) {
+    if (result.format === "commonjs") {
+      const parsedUrl = new URL(result.responseURL ?? url);
+      result.source ??= await readFile(parsedUrl);
+    }
+    if (result.source) {
+      result.source = result.source.toString("utf8") + OPENAI_API_PROMISE_PATCH;
+      result.shortCircuit = true;
+    }
+    return result;
+  }
   if (!transformers.has(url)) {
     return result;
   }
@@ -55,7 +87,7 @@ async function load(url, context, nextLoad) {
   if (code) {
     const transformer = transformers.get(url);
     try {
-      const moduleType = result.format === "module" ? "esm" : result.format === "commonjs" ? "cjs" : "unknown";
+      const moduleType = getModuleType(url, result.format);
       const transformedCode = transformer.transform(
         code.toString("utf8"),
         moduleType

package/dist/auto-instrumentations/plugin-D7nDswtC.d.mts

@@ -0,0 +1,44 @@
+import { InstrumentationConfig } from '@apm-js-collab/code-transformer';
+
+interface LegacyBundlerPluginOptions {
+    /**
+     * Enable debug logging
+     */
+    debug?: boolean;
+    /**
+     * Additional instrumentation configs to apply
+     */
+    instrumentations?: InstrumentationConfig[];
+    /**
+     * Whether to bundle for browser environments.
+     *
+     * When true, uses 'dc-browser' for browser-compatible diagnostics_channel polyfill.
+     * When false, uses Node.js built-in 'diagnostics_channel' and 'async_hooks'.
+     * Defaults to true (assumes browser build).
+     */
+    browser?: boolean;
+}
+interface BundlerPluginOptions {
+    /**
+     * Enable debug logging
+     */
+    debug?: boolean;
+    /**
+     * Additional instrumentation configs to apply
+     */
+    instrumentations?: InstrumentationConfig[];
+    /**
+     * Use the `diagnostics_channel` compatibility shim in patched code instead
+     * of Node.js's built-in `diagnostics_channel` module.
+     *
+     * Enable this for browser, edge, or worker bundles where Node's
+     * `diagnostics_channel` module is unavailable. Leave it disabled for Node.js
+     * bundles so transformed SDK code publishes on the native `diagnostics_channel`
+     * registry.
+     *
+     * @default false
+     */
+    useDiagnosticChannelCompatShim?: boolean;
+}
+
+export type { BundlerPluginOptions as B, LegacyBundlerPluginOptions as L };

package/dist/auto-instrumentations/plugin-D7nDswtC.d.ts

@@ -0,0 +1,44 @@
+import { InstrumentationConfig } from '@apm-js-collab/code-transformer';
+
+interface LegacyBundlerPluginOptions {
+    /**
+     * Enable debug logging
+     */
+    debug?: boolean;
+    /**
+     * Additional instrumentation configs to apply
+     */
+    instrumentations?: InstrumentationConfig[];
+    /**
+     * Whether to bundle for browser environments.
+     *
+     * When true, uses 'dc-browser' for browser-compatible diagnostics_channel polyfill.
+     * When false, uses Node.js built-in 'diagnostics_channel' and 'async_hooks'.
+     * Defaults to true (assumes browser build).
+     */
+    browser?: boolean;
+}
+interface BundlerPluginOptions {
+    /**
+     * Enable debug logging
+     */
+    debug?: boolean;
+    /**
+     * Additional instrumentation configs to apply
+     */
+    instrumentations?: InstrumentationConfig[];
+    /**
+     * Use the `diagnostics_channel` compatibility shim in patched code instead
+     * of Node.js's built-in `diagnostics_channel` module.
+     *
+     * Enable this for browser, edge, or worker bundles where Node's
+     * `diagnostics_channel` module is unavailable. Leave it disabled for Node.js
+     * bundles so transformed SDK code publishes on the native `diagnostics_channel`
+     * registry.
+     *
+     * @default false
+     */
+    useDiagnosticChannelCompatShim?: boolean;
+}
+
+export type { BundlerPluginOptions as B, LegacyBundlerPluginOptions as L };