braintrust 3.3.0 → 3.4.0

package/dist/auto-instrumentations/hook.mjs

@@ -5,14 +5,146 @@ import {
 // src/auto-instrumentations/hook.mts
 import { register } from "node:module";
 
+// src/isomorph.ts
+var DefaultAsyncLocalStorage = class {
+  constructor() {
+  }
+  enterWith(_) {
+  }
+  run(_, callback) {
+    return callback();
+  }
+  getStore() {
+    return void 0;
+  }
+};
+var DefaultTracingChannel = class {
+  hasSubscribers = false;
+  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(),
+  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 +153,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 +195,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 +207,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 +236,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 +248,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 +276,7 @@ var openaiConfigs = [
     }
   },
   {
-    channelName: "responses.stream",
+    channelName: openAIChannels.responsesStream.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -105,7 +289,7 @@ var openaiConfigs = [
     }
   },
   {
-    channelName: "responses.parse",
+    channelName: openAIChannels.responsesParse.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -119,15 +303,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 +333,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 +347,39 @@ 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"
+  }),
+  generateObject: channel({
+    channelName: "generateObject",
+    kind: "async"
+  }),
+  streamObject: channel({
+    channelName: "streamObject",
+    kind: "async"
+  }),
+  agentGenerate: channel({
+    channelName: "Agent.generate",
+    kind: "async"
+  }),
+  agentStream: channel({
+    channelName: "Agent.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",
@@ -168,7 +392,7 @@ var aiSDKConfigs = [
   },
   // streamText - async function
   {
-    channelName: "streamText",
+    channelName: aiSDKChannels.streamText.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -181,7 +405,7 @@ var aiSDKConfigs = [
   },
   // generateObject - async function
   {
-    channelName: "generateObject",
+    channelName: aiSDKChannels.generateObject.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -194,7 +418,7 @@ var aiSDKConfigs = [
   },
   // streamObject - async function
   {
-    channelName: "streamObject",
+    channelName: aiSDKChannels.streamObject.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -207,7 +431,7 @@ var aiSDKConfigs = [
   },
   // Agent.generate - async method (v3-v5 only, Agent structure changed in v6)
   {
-    channelName: "Agent.generate",
+    channelName: aiSDKChannels.agentGenerate.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0 <6.0.0",
@@ -221,7 +445,7 @@ var aiSDKConfigs = [
   },
   // Agent.stream - async method (v3-v5 only, Agent structure changed in v6)
   {
-    channelName: "Agent.stream",
+    channelName: aiSDKChannels.agentStream.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0 <6.0.0",
@@ -235,11 +459,22 @@ var aiSDKConfigs = [
   }
 ];
 
+// 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 +487,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 +519,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",
@@ -347,6 +594,42 @@ 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 && 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 promise = Reflect.apply(fn, thisArg, args);
+      return promise.then(resolve, reject);
+    } catch (err) {
+      context.error = err;
+      error?.publish(context);
+      throw err;
+    } finally {
+      end?.publish(context);
+    }
+  };
+}
 var allConfigs = [
   ...openaiConfigs,
   ...anthropicConfigs,