braintrust 3.4.0 → 3.6.0

package/dist/auto-instrumentations/index.cjs

@@ -24,6 +24,7 @@ __export(index_exports, {
   anthropicConfigs: () => anthropicConfigs,
   claudeAgentSDKConfigs: () => claudeAgentSDKConfigs,
   googleGenAIConfigs: () => googleGenAIConfigs,
+  openRouterConfigs: () => openRouterConfigs,
   openaiConfigs: () => openaiConfigs
 });
 module.exports = __toCommonJS(index_exports);
@@ -41,8 +42,54 @@ var DefaultAsyncLocalStorage = class {
     return void 0;
   }
 };
-var DefaultTracingChannel = class {
+var DefaultChannel = class {
+  constructor(name) {
+    this.name = name;
+  }
   hasSubscribers = false;
+  subscribe(_subscription) {
+  }
+  unsubscribe(_subscription) {
+    return false;
+  }
+  bindStore(_store, _transform) {
+  }
+  unbindStore(_store) {
+    return false;
+  }
+  publish(_message) {
+  }
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  runStores(_message, fn, thisArg, ...args) {
+    return fn.apply(thisArg, args);
+  }
+};
+var DefaultTracingChannel = class {
+  start;
+  end;
+  asyncStart;
+  asyncEnd;
+  error;
+  constructor(nameOrChannels) {
+    if (typeof nameOrChannels === "string") {
+      this.start = new DefaultChannel(`tracing:${nameOrChannels}:start`);
+      this.end = new DefaultChannel(`tracing:${nameOrChannels}:end`);
+      this.asyncStart = new DefaultChannel(
+        `tracing:${nameOrChannels}:asyncStart`
+      );
+      this.asyncEnd = new DefaultChannel(`tracing:${nameOrChannels}:asyncEnd`);
+      this.error = new DefaultChannel(`tracing:${nameOrChannels}:error`);
+      return;
+    }
+    this.start = nameOrChannels.start ?? new DefaultChannel("tracing:start");
+    this.end = nameOrChannels.end ?? new DefaultChannel("tracing:end");
+    this.asyncStart = nameOrChannels.asyncStart ?? new DefaultChannel("tracing:asyncStart");
+    this.asyncEnd = nameOrChannels.asyncEnd ?? new DefaultChannel("tracing:asyncEnd");
+    this.error = nameOrChannels.error ?? new DefaultChannel("tracing:error");
+  }
+  get hasSubscribers() {
+    return this.start.hasSubscribers || this.end.hasSubscribers || this.asyncStart.hasSubscribers || this.asyncEnd.hasSubscribers || this.error.hasSubscribers;
+  }
   subscribe(_handlers) {
   }
   unsubscribe(_handlers) {
@@ -70,7 +117,7 @@ var iso = {
   getCallerLocation: () => void 0,
   newAsyncLocalStorage: () => new DefaultAsyncLocalStorage(),
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  newTracingChannel: (_nameOrChannels) => new DefaultTracingChannel(),
+  newTracingChannel: (nameOrChannels) => new DefaultTracingChannel(nameOrChannels),
   processOn: (_0, _1) => {
   },
   basename: (filepath) => filepath.split(/[\\/]/).pop() || filepath,
@@ -380,6 +427,10 @@ var aiSDKChannels = defineChannels("ai", {
     channelName: "streamText",
     kind: "async"
   }),
+  streamTextSync: channel({
+    channelName: "streamText.sync",
+    kind: "sync-stream"
+  }),
   generateObject: channel({
     channelName: "generateObject",
     kind: "async"
@@ -388,6 +439,10 @@ var aiSDKChannels = defineChannels("ai", {
     channelName: "streamObject",
     kind: "async"
   }),
+  streamObjectSync: channel({
+    channelName: "streamObject.sync",
+    kind: "sync-stream"
+  }),
   agentGenerate: channel({
     channelName: "Agent.generate",
     kind: "async"
@@ -395,6 +450,14 @@ var aiSDKChannels = defineChannels("ai", {
   agentStream: channel({
     channelName: "Agent.stream",
     kind: "async"
+  }),
+  toolLoopAgentGenerate: channel({
+    channelName: "ToolLoopAgent.generate",
+    kind: "async"
+  }),
+  toolLoopAgentStream: channel({
+    channelName: "ToolLoopAgent.stream",
+    kind: "async"
   })
 });
 
@@ -413,6 +476,18 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: aiSDKChannels.generateText.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "generateText",
+      kind: "Async"
+    }
+  },
   // streamText - async function
   {
     channelName: aiSDKChannels.streamText.channelName,
@@ -426,6 +501,18 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: aiSDKChannels.streamTextSync.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "streamText",
+      kind: "Sync"
+    }
+  },
   // generateObject - async function
   {
     channelName: aiSDKChannels.generateObject.channelName,
@@ -439,6 +526,18 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: aiSDKChannels.generateObject.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "generateObject",
+      kind: "Async"
+    }
+  },
   // streamObject - async function
   {
     channelName: aiSDKChannels.streamObject.channelName,
@@ -452,32 +551,132 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
-  // Agent.generate - async method (v3-v5 only, Agent structure changed in v6)
+  {
+    channelName: aiSDKChannels.streamObjectSync.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=3.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.
   {
     channelName: aiSDKChannels.agentGenerate.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0 <6.0.0",
+      versionRange: ">=5.0.0 <6.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
-      className: "Agent",
       methodName: "generate",
-      kind: "Async"
+      kind: "Async",
+      index: 0
     }
   },
-  // Agent.stream - async method (v3-v5 only, Agent structure changed in v6)
+  {
+    channelName: aiSDKChannels.agentGenerate.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=5.0.0 <6.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      methodName: "generate",
+      kind: "Async",
+      index: 0
+    }
+  },
+  // Agent.stream - async method (v5 only)
+  // The compiled AI SDK bundle emits this as an anonymous class method, so we
+  // target the first async `stream` method in the file instead of a class name.
   {
     channelName: aiSDKChannels.agentStream.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0 <6.0.0",
+      versionRange: ">=5.0.0 <6.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
-      className: "Agent",
       methodName: "stream",
-      kind: "Async"
+      kind: "Async",
+      index: 0
+    }
+  },
+  {
+    channelName: aiSDKChannels.agentStream.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=5.0.0 <6.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      methodName: "stream",
+      kind: "Async",
+      index: 0
+    }
+  },
+  // ToolLoopAgent.generate - async method (v6 only, Experimental_Agent is an alias)
+  // 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.
+  {
+    channelName: aiSDKChannels.toolLoopAgentGenerate.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=6.0.0 <7.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      methodName: "generate",
+      kind: "Async",
+      index: 0
+    }
+  },
+  {
+    channelName: aiSDKChannels.toolLoopAgentGenerate.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=6.0.0 <7.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      methodName: "generate",
+      kind: "Async",
+      index: 0
+    }
+  },
+  // ToolLoopAgent.stream - async method (v6 only, Experimental_Agent is an alias)
+  // The compiled AI SDK bundle emits this as an anonymous class method, so we
+  // target the first async `stream` method in the file instead of a class name.
+  {
+    channelName: aiSDKChannels.toolLoopAgentStream.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=6.0.0 <7.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      methodName: "stream",
+      kind: "Async",
+      index: 0
+    }
+  },
+  {
+    channelName: aiSDKChannels.toolLoopAgentStream.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=6.0.0 <7.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      methodName: "stream",
+      kind: "Async",
+      index: 0
     }
   }
 ];
@@ -555,11 +754,92 @@ var googleGenAIConfigs = [
     }
   }
 ];
+
+// src/instrumentation/plugins/openrouter-channels.ts
+var openRouterChannels = defineChannels("@openrouter/sdk", {
+  chatSend: channel({
+    channelName: "chat.send",
+    kind: "async"
+  }),
+  embeddingsGenerate: channel({
+    channelName: "embeddings.generate",
+    kind: "async"
+  }),
+  betaResponsesSend: channel({
+    channelName: "beta.responses.send",
+    kind: "async"
+  }),
+  callModel: channel({
+    channelName: "callModel",
+    kind: "sync-stream"
+  }),
+  toolExecute: channel({
+    channelName: "tool.execute",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/openrouter.ts
+var openRouterConfigs = [
+  {
+    channelName: openRouterChannels.chatSend.channelName,
+    module: {
+      name: "@openrouter/sdk",
+      versionRange: ">=0.9.11 <1.0.0",
+      filePath: "esm/sdk/chat.js"
+    },
+    functionQuery: {
+      className: "Chat",
+      methodName: "send",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: openRouterChannels.embeddingsGenerate.channelName,
+    module: {
+      name: "@openrouter/sdk",
+      versionRange: ">=0.9.11 <1.0.0",
+      filePath: "esm/sdk/embeddings.js"
+    },
+    functionQuery: {
+      className: "Embeddings",
+      methodName: "generate",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: openRouterChannels.betaResponsesSend.channelName,
+    module: {
+      name: "@openrouter/sdk",
+      versionRange: ">=0.9.11 <1.0.0",
+      filePath: "esm/sdk/responses.js"
+    },
+    functionQuery: {
+      className: "Responses",
+      methodName: "send",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: openRouterChannels.callModel.channelName,
+    module: {
+      name: "@openrouter/sdk",
+      versionRange: ">=0.9.11 <1.0.0",
+      filePath: "esm/sdk/sdk.js"
+    },
+    functionQuery: {
+      className: "OpenRouter",
+      methodName: "callModel",
+      kind: "Sync"
+    }
+  }
+];
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   aiSDKConfigs,
   anthropicConfigs,
   claudeAgentSDKConfigs,
   googleGenAIConfigs,
+  openRouterConfigs,
   openaiConfigs
 });

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

@@ -66,4 +66,6 @@ declare const claudeAgentSDKConfigs: InstrumentationConfig[];
  */
 declare const googleGenAIConfigs: InstrumentationConfig[];
 
-export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, googleGenAIConfigs, openaiConfigs };
+declare const openRouterConfigs: InstrumentationConfig[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, googleGenAIConfigs, openRouterConfigs, openaiConfigs };

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

@@ -66,4 +66,6 @@ declare const claudeAgentSDKConfigs: InstrumentationConfig[];
  */
 declare const googleGenAIConfigs: InstrumentationConfig[];
 
-export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, googleGenAIConfigs, openaiConfigs };
+declare const openRouterConfigs: InstrumentationConfig[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, googleGenAIConfigs, openRouterConfigs, openaiConfigs };

package/dist/auto-instrumentations/index.mjs

@@ -3,12 +3,14 @@ import {
   anthropicConfigs,
   claudeAgentSDKConfigs,
   googleGenAIConfigs,
+  openRouterConfigs,
   openaiConfigs
-} from "./chunk-LVWWLUMN.mjs";
+} from "./chunk-F7WAXFNM.mjs";
 export {
   aiSDKConfigs,
   anthropicConfigs,
   claudeAgentSDKConfigs,
   googleGenAIConfigs,
+  openRouterConfigs,
   openaiConfigs
 };

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

@@ -42,19 +42,37 @@ var import_module_details_from_path = __toESM(require("module-details-from-path"
 var import_node_fs = require("fs");
 var import_node_path = require("path");
 var packageVersions = /* @__PURE__ */ new Map();
-function getPackageVersion(baseDir) {
-  if (packageVersions.has(baseDir)) {
-    return packageVersions.get(baseDir);
-  }
+var packageNames = /* @__PURE__ */ new Map();
+function readPackageJson(baseDir) {
   try {
     const packageJsonPath = (0, import_node_path.join)(baseDir, "package.json");
     const jsonFile = (0, import_node_fs.readFileSync)(packageJsonPath, "utf8");
-    const { version } = JSON.parse(jsonFile);
-    packageVersions.set(baseDir, version);
-    return version;
+    return JSON.parse(jsonFile);
   } catch {
-    return process.version.slice(1);
+    return void 0;
+  }
+}
+function getPackageVersion(baseDir) {
+  if (packageVersions.has(baseDir)) {
+    return packageVersions.get(baseDir);
   }
+  const packageJson = readPackageJson(baseDir);
+  if (typeof packageJson?.version === "string") {
+    packageVersions.set(baseDir, packageJson.version);
+    return packageJson.version;
+  }
+  return process.version.slice(1);
+}
+function getPackageName(baseDir) {
+  if (packageNames.has(baseDir)) {
+    return packageNames.get(baseDir);
+  }
+  const packageJson = readPackageJson(baseDir);
+  if (typeof packageJson?.name === "string") {
+    packageNames.set(baseDir, packageJson.name);
+    return packageJson.name;
+  }
+  return void 0;
 }
 
 // src/auto-instrumentations/loader/cjs-patch.ts
@@ -80,11 +98,15 @@ var ModulePatch = class {
       const [content, filename] = args;
       const normalizedForPlatform = filename.split("/").join(import_node_path2.sep);
       const resolvedModule = (0, import_module_details_from_path.default)(normalizedForPlatform);
-      if (resolvedModule && self.packages.has(resolvedModule.name)) {
+      if (resolvedModule) {
+        const packageName = getPackageName(resolvedModule.basedir) ?? resolvedModule.name;
+        if (!self.packages.has(packageName)) {
+          return self.originalCompile.apply(this, args);
+        }
         const version = getPackageVersion(resolvedModule.basedir);
         const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
         const transformer = self.instrumentator.getTransformer(
-          resolvedModule.name,
+          packageName,
           version,
           normalizedModulePath
         );

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

@@ -1,13 +1,14 @@
 import {
+  getPackageName,
   getPackageVersion
-} from "../chunk-XDBPUTVE.mjs";
+} from "../chunk-VLEJ5AEK.mjs";
 
 // src/auto-instrumentations/loader/cjs-patch.ts
 import {
   create
 } from "@apm-js-collab/code-transformer";
-import * as Module from "node:module";
-import { sep } from "node:path";
+import * as Module from "module";
+import { sep } from "path";
 import moduleDetailsFromPath from "module-details-from-path";
 var ModulePatch = class {
   packages;
@@ -31,11 +32,15 @@ var ModulePatch = class {
       const [content, filename] = args;
       const normalizedForPlatform = filename.split("/").join(sep);
       const resolvedModule = moduleDetailsFromPath(normalizedForPlatform);
-      if (resolvedModule && self.packages.has(resolvedModule.name)) {
+      if (resolvedModule) {
+        const packageName = getPackageName(resolvedModule.basedir) ?? resolvedModule.name;
+        if (!self.packages.has(packageName)) {
+          return self.originalCompile.apply(this, args);
+        }
         const version = getPackageVersion(resolvedModule.basedir);
         const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
         const transformer = self.instrumentator.getTransformer(
-          resolvedModule.name,
+          packageName,
           version,
           normalizedModulePath
         );

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

@@ -1,11 +1,11 @@
 import {
   getPackageVersion
-} from "../chunk-ZEC7BCL4.mjs";
+} from "../chunk-EVUKFMHG.mjs";
 
 // src/auto-instrumentations/loader/esm-hook.mts
-import { readFile } from "node:fs/promises";
-import { fileURLToPath } from "node:url";
-import { sep } from "node:path";
+import { readFile } from "fs/promises";
+import { fileURLToPath } from "url";
+import { sep } from "path";
 import {
   create
 } from "@apm-js-collab/code-transformer";

package/dist/auto-instrumentations/loader/get-package-version.cjs

@@ -20,27 +20,47 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/auto-instrumentations/loader/get-package-version.ts
 var get_package_version_exports = {};
 __export(get_package_version_exports, {
+  getPackageName: () => getPackageName,
   getPackageVersion: () => getPackageVersion
 });
 module.exports = __toCommonJS(get_package_version_exports);
 var import_node_fs = require("fs");
 var import_node_path = require("path");
 var packageVersions = /* @__PURE__ */ new Map();
-function getPackageVersion(baseDir) {
-  if (packageVersions.has(baseDir)) {
-    return packageVersions.get(baseDir);
-  }
+var packageNames = /* @__PURE__ */ new Map();
+function readPackageJson(baseDir) {
   try {
     const packageJsonPath = (0, import_node_path.join)(baseDir, "package.json");
     const jsonFile = (0, import_node_fs.readFileSync)(packageJsonPath, "utf8");
-    const { version } = JSON.parse(jsonFile);
-    packageVersions.set(baseDir, version);
-    return version;
+    return JSON.parse(jsonFile);
   } catch {
-    return process.version.slice(1);
+    return void 0;
+  }
+}
+function getPackageVersion(baseDir) {
+  if (packageVersions.has(baseDir)) {
+    return packageVersions.get(baseDir);
+  }
+  const packageJson = readPackageJson(baseDir);
+  if (typeof packageJson?.version === "string") {
+    packageVersions.set(baseDir, packageJson.version);
+    return packageJson.version;
+  }
+  return process.version.slice(1);
+}
+function getPackageName(baseDir) {
+  if (packageNames.has(baseDir)) {
+    return packageNames.get(baseDir);
+  }
+  const packageJson = readPackageJson(baseDir);
+  if (typeof packageJson?.name === "string") {
+    packageNames.set(baseDir, packageJson.name);
+    return packageJson.name;
   }
+  return void 0;
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  getPackageName,
   getPackageVersion
 });

package/dist/auto-instrumentations/loader/get-package-version.d.mts

@@ -3,5 +3,6 @@
  * If the package.json file cannot be read, it defaults to the Node.js version.
  */
 declare function getPackageVersion(baseDir: string): string;
+declare function getPackageName(baseDir: string): string | undefined;
 
-export { getPackageVersion };
+export { getPackageName, getPackageVersion };

package/dist/auto-instrumentations/loader/get-package-version.d.ts

@@ -3,5 +3,6 @@
  * If the package.json file cannot be read, it defaults to the Node.js version.
  */
 declare function getPackageVersion(baseDir: string): string;
+declare function getPackageName(baseDir: string): string | undefined;
 
-export { getPackageVersion };
+export { getPackageName, getPackageVersion };

package/dist/auto-instrumentations/loader/get-package-version.mjs

@@ -1,6 +1,8 @@
 import {
+  getPackageName,
   getPackageVersion
-} from "../chunk-XDBPUTVE.mjs";
+} from "../chunk-VLEJ5AEK.mjs";
 export {
+  getPackageName,
   getPackageVersion
 };