braintrust 3.3.0 → 3.5.0

package/dist/auto-instrumentations/hook.mjs

@@ -1,18 +1,197 @@
 import {
+  getPackageName,
   getPackageVersion
-} from "./chunk-ZEC7BCL4.mjs";
+} from "./chunk-EVUKFMHG.mjs";
 
 // src/auto-instrumentations/hook.mts
-import { register } from "node:module";
+import { register } from "module";
+
+// src/isomorph.ts
+var DefaultAsyncLocalStorage = class {
+  constructor() {
+  }
+  enterWith(_) {
+  }
+  run(_, callback) {
+    return callback();
+  }
+  getStore() {
+    return void 0;
+  }
+};
+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) {
+    return false;
+  }
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  traceSync(fn, _message, thisArg, ...args) {
+    return fn.apply(thisArg, args);
+  }
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  tracePromise(fn, _message, thisArg, ...args) {
+    return Promise.resolve(fn.apply(thisArg, args));
+  }
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  traceCallback(fn, _position, _message, thisArg, ...args) {
+    return fn.apply(thisArg, args);
+  }
+};
+var iso = {
+  buildType: "unknown",
+  // Will be set by configureBrowser() or configureNode()
+  getRepoInfo: async (_settings) => void 0,
+  getPastNAncestors: async () => [],
+  getEnv: (_name) => void 0,
+  getCallerLocation: () => void 0,
+  newAsyncLocalStorage: () => new DefaultAsyncLocalStorage(),
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  newTracingChannel: (nameOrChannels) => new DefaultTracingChannel(nameOrChannels),
+  processOn: (_0, _1) => {
+  },
+  basename: (filepath) => filepath.split(/[\\/]/).pop() || filepath,
+  writeln: (text) => console.log(text)
+};
+var isomorph_default = iso;
+
+// src/instrumentation/core/channel-definitions.ts
+function channel(spec) {
+  return spec;
+}
+function defineChannels(pkg, channels) {
+  return Object.fromEntries(
+    Object.entries(channels).map(([key, spec]) => {
+      const fullChannelName = `orchestrion:${pkg}:${spec.channelName}`;
+      if (spec.kind === "async") {
+        const asyncSpec = spec;
+        const tracingChannel2 = () => isomorph_default.newTracingChannel(
+          fullChannelName
+        );
+        return [
+          key,
+          {
+            ...asyncSpec,
+            tracingChannel: tracingChannel2,
+            tracePromise: (fn, context) => tracingChannel2().tracePromise(
+              fn,
+              // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+              context
+            )
+          }
+        ];
+      }
+      const syncSpec = spec;
+      const tracingChannel = () => isomorph_default.newTracingChannel(
+        fullChannelName
+      );
+      return [
+        key,
+        {
+          ...syncSpec,
+          tracingChannel,
+          traceSync: (fn, context) => tracingChannel().traceSync(
+            fn,
+            // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+            context
+          )
+        }
+      ];
+    })
+  );
+}
+
+// src/instrumentation/plugins/openai-channels.ts
+var openAIChannels = defineChannels("openai", {
+  chatCompletionsCreate: channel({
+    channelName: "chat.completions.create",
+    kind: "async"
+  }),
+  embeddingsCreate: channel({
+    channelName: "embeddings.create",
+    kind: "async"
+  }),
+  betaChatCompletionsParse: channel({
+    channelName: "beta.chat.completions.parse",
+    kind: "async"
+  }),
+  betaChatCompletionsStream: channel({
+    channelName: "beta.chat.completions.stream",
+    kind: "sync-stream"
+  }),
+  moderationsCreate: channel({
+    channelName: "moderations.create",
+    kind: "async"
+  }),
+  responsesCreate: channel({
+    channelName: "responses.create",
+    kind: "async"
+  }),
+  responsesStream: channel({
+    channelName: "responses.stream",
+    kind: "sync-stream"
+  }),
+  responsesParse: channel({
+    channelName: "responses.parse",
+    kind: "async"
+  })
+});
 
 // src/auto-instrumentations/configs/openai.ts
 var openaiConfigs = [
   // Chat Completions
   {
-    channelName: "chat.completions.create",
+    channelName: openAIChannels.chatCompletionsCreate.channelName,
     module: {
       name: "openai",
-      versionRange: ">=4.0.0",
+      versionRange: ">=4.0.0 <5.0.0",
       filePath: "resources/chat/completions.mjs"
     },
     functionQuery: {
@@ -21,9 +200,35 @@ var openaiConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: openAIChannels.chatCompletionsCreate.channelName,
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0 <5.0.0",
+      filePath: "resources/chat/completions/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: openAIChannels.chatCompletionsCreate.channelName,
+    module: {
+      name: "openai",
+      versionRange: ">=5.0.0",
+      filePath: "resources/chat/completions/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
   // Embeddings
   {
-    channelName: "embeddings.create",
+    channelName: openAIChannels.embeddingsCreate.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.0.0",
@@ -37,10 +242,10 @@ var openaiConfigs = [
   },
   // Beta Chat Completions Parse
   {
-    channelName: "beta.chat.completions.parse",
+    channelName: openAIChannels.betaChatCompletionsParse.channelName,
     module: {
       name: "openai",
-      versionRange: ">=4.0.0",
+      versionRange: ">=4.0.0 <5.0.0",
       filePath: "resources/beta/chat/completions.mjs"
     },
     functionQuery: {
@@ -49,9 +254,22 @@ var openaiConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: openAIChannels.betaChatCompletionsParse.channelName,
+    module: {
+      name: "openai",
+      versionRange: ">=5.0.0",
+      filePath: "resources/chat/completions/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "parse",
+      kind: "Async"
+    }
+  },
   // Moderations
   {
-    channelName: "moderations.create",
+    channelName: openAIChannels.moderationsCreate.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.0.0",
@@ -65,10 +283,10 @@ var openaiConfigs = [
   },
   // Beta Chat Completions Stream
   {
-    channelName: "beta.chat.completions.stream",
+    channelName: openAIChannels.betaChatCompletionsStream.channelName,
     module: {
       name: "openai",
-      versionRange: ">=4.0.0",
+      versionRange: ">=4.0.0 <5.0.0",
       filePath: "resources/beta/chat/completions.mjs"
     },
     functionQuery: {
@@ -77,9 +295,22 @@ var openaiConfigs = [
       kind: "Sync"
     }
   },
+  {
+    channelName: openAIChannels.betaChatCompletionsStream.channelName,
+    module: {
+      name: "openai",
+      versionRange: ">=5.0.0",
+      filePath: "resources/chat/completions/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "stream",
+      kind: "Sync"
+    }
+  },
   // Responses API (v4.87.0+)
   {
-    channelName: "responses.create",
+    channelName: openAIChannels.responsesCreate.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -92,7 +323,7 @@ var openaiConfigs = [
     }
   },
   {
-    channelName: "responses.stream",
+    channelName: openAIChannels.responsesStream.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -105,7 +336,7 @@ var openaiConfigs = [
     }
   },
   {
-    channelName: "responses.parse",
+    channelName: openAIChannels.responsesParse.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -119,15 +350,27 @@ var openaiConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/anthropic-channels.ts
+var anthropicChannels = defineChannels("@anthropic-ai/sdk", {
+  messagesCreate: channel({
+    channelName: "messages.create",
+    kind: "async"
+  }),
+  betaMessagesCreate: channel({
+    channelName: "beta.messages.create",
+    kind: "async"
+  })
+});
+
 // src/auto-instrumentations/configs/anthropic.ts
 var anthropicConfigs = [
   // Messages API - create (supports streaming via stream=true parameter)
   {
-    channelName: "messages.create",
+    channelName: anthropicChannels.messagesCreate.channelName,
     module: {
       name: "@anthropic-ai/sdk",
       versionRange: ">=0.60.0",
-      filePath: "resources/messages.mjs"
+      filePath: "resources/messages/messages.mjs"
     },
     functionQuery: {
       className: "Messages",
@@ -137,7 +380,7 @@ var anthropicConfigs = [
   },
   // Beta Messages API - create (supports streaming via stream=true parameter)
   {
-    channelName: "beta.messages.create",
+    channelName: anthropicChannels.betaMessagesCreate.channelName,
     module: {
       name: "@anthropic-ai/sdk",
       versionRange: ">=0.60.0",
@@ -151,11 +394,55 @@ var anthropicConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/ai-sdk-channels.ts
+var aiSDKChannels = defineChannels("ai", {
+  generateText: channel({
+    channelName: "generateText",
+    kind: "async"
+  }),
+  streamText: channel({
+    channelName: "streamText",
+    kind: "async"
+  }),
+  streamTextSync: channel({
+    channelName: "streamText.sync",
+    kind: "sync-stream"
+  }),
+  generateObject: channel({
+    channelName: "generateObject",
+    kind: "async"
+  }),
+  streamObject: channel({
+    channelName: "streamObject",
+    kind: "async"
+  }),
+  streamObjectSync: channel({
+    channelName: "streamObject.sync",
+    kind: "sync-stream"
+  }),
+  agentGenerate: channel({
+    channelName: "Agent.generate",
+    kind: "async"
+  }),
+  agentStream: channel({
+    channelName: "Agent.stream",
+    kind: "async"
+  }),
+  toolLoopAgentGenerate: channel({
+    channelName: "ToolLoopAgent.generate",
+    kind: "async"
+  }),
+  toolLoopAgentStream: channel({
+    channelName: "ToolLoopAgent.stream",
+    kind: "async"
+  })
+});
+
 // src/auto-instrumentations/configs/ai-sdk.ts
 var aiSDKConfigs = [
   // generateText - async function
   {
-    channelName: "generateText",
+    channelName: aiSDKChannels.generateText.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -166,9 +453,21 @@ 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: "streamText",
+    channelName: aiSDKChannels.streamText.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -179,9 +478,21 @@ 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: "generateObject",
+    channelName: aiSDKChannels.generateObject.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -192,9 +503,21 @@ 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: "streamObject",
+    channelName: aiSDKChannels.streamObject.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -205,41 +528,152 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
-  // Agent.generate - async method (v3-v5 only, Agent structure changed in v6)
   {
-    channelName: "Agent.generate",
+    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: "Agent.stream",
+    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
     }
   }
 ];
 
+// src/instrumentation/plugins/claude-agent-sdk-channels.ts
+var claudeAgentSDKChannels = defineChannels(
+  "@anthropic-ai/claude-agent-sdk",
+  {
+    query: channel({
+      channelName: "query",
+      kind: "async"
+    })
+  }
+);
+
 // src/auto-instrumentations/configs/claude-agent-sdk.ts
 var claudeAgentSDKConfigs = [
   // query - Main entry point for agent interactions (top-level exported async generator function)
   {
-    channelName: "query",
+    channelName: claudeAgentSDKChannels.query.channelName,
     module: {
       name: "@anthropic-ai/claude-agent-sdk",
       versionRange: ">=0.1.0",
@@ -252,12 +686,24 @@ var claudeAgentSDKConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/google-genai-channels.ts
+var googleGenAIChannels = defineChannels("@google/genai", {
+  generateContent: channel({
+    channelName: "models.generateContent",
+    kind: "async"
+  }),
+  generateContentStream: channel({
+    channelName: "models.generateContentStream",
+    kind: "async"
+  })
+});
+
 // src/auto-instrumentations/configs/google-genai.ts
 var googleGenAIConfigs = [
   // Models.generateContentInternal - The actual class method (Node.js entry point)
   // Note: generateContent is an arrow function property that calls this internal method
   {
-    channelName: "models.generateContent",
+    channelName: googleGenAIChannels.generateContent.channelName,
     module: {
       name: "@google/genai",
       versionRange: ">=1.0.0",
@@ -272,7 +718,7 @@ var googleGenAIConfigs = [
   // Models.generateContentStreamInternal - The actual class method (Node.js entry point)
   // Note: generateContentStream is an arrow function property that calls this internal method
   {
-    channelName: "models.generateContentStream",
+    channelName: googleGenAIChannels.generateContentStream.channelName,
     module: {
       name: "@google/genai",
       versionRange: ">=1.0.0",
@@ -290,8 +736,8 @@ var googleGenAIConfigs = [
 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;
@@ -315,11 +761,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
         );
@@ -347,6 +797,59 @@ var ModulePatch = class {
 };
 
 // src/auto-instrumentations/hook.mts
+var dcPath = ["node", "diagnostics_channel"].join(":");
+var dc = await import(
+  /* @vite-ignore */
+  dcPath
+);
+var dummyChannel = dc.tracingChannel("dummy");
+var TracingChannel = dummyChannel.constructor;
+if (TracingChannel && !Object.getOwnPropertyDescriptor(TracingChannel.prototype, "hasSubscribers")) {
+  Object.defineProperty(TracingChannel.prototype, "hasSubscribers", {
+    configurable: true,
+    enumerable: false,
+    get() {
+      return Boolean(
+        this.start?.hasSubscribers || this.end?.hasSubscribers || this.asyncStart?.hasSubscribers || this.asyncEnd?.hasSubscribers || this.error?.hasSubscribers
+      );
+    }
+  });
+}
+if (TracingChannel && TracingChannel.prototype.tracePromise) {
+  TracingChannel.prototype.tracePromise = function(fn, context = {}, thisArg, ...args) {
+    const { start, end, asyncStart, asyncEnd, error } = this;
+    function reject(err) {
+      context.error = err;
+      error?.publish(context);
+      asyncStart?.publish(context);
+      asyncEnd?.publish(context);
+      return Promise.reject(err);
+    }
+    function resolve(result) {
+      context.result = result;
+      asyncStart?.publish(context);
+      asyncEnd?.publish(context);
+      return result;
+    }
+    start?.publish(context);
+    try {
+      const result = Reflect.apply(fn, thisArg, args);
+      if (result && (typeof result === "object" || typeof result === "function") && typeof result.then === "function") {
+        return result.then(resolve, reject);
+      }
+      context.result = result;
+      asyncStart?.publish(context);
+      asyncEnd?.publish(context);
+      return result;
+    } catch (err) {
+      context.error = err;
+      error?.publish(context);
+      throw err;
+    } finally {
+      end?.publish(context);
+    }
+  };
+}
 var allConfigs = [
   ...openaiConfigs,
   ...anthropicConfigs,