braintrust 3.3.0 → 3.4.0

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

@@ -42,14 +42,146 @@ var import_fs = require("fs");
 var import_url = require("url");
 var import_module_details_from_path = __toESM(require("module-details-from-path"));
 
+// 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: {
@@ -58,9 +190,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",
@@ -74,10 +232,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: {
@@ -86,9 +244,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",
@@ -102,10 +273,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: {
@@ -114,9 +285,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",
@@ -129,7 +313,7 @@ var openaiConfigs = [
     }
   },
   {
-    channelName: "responses.stream",
+    channelName: openAIChannels.responsesStream.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -142,7 +326,7 @@ var openaiConfigs = [
     }
   },
   {
-    channelName: "responses.parse",
+    channelName: openAIChannels.responsesParse.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -156,15 +340,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",
@@ -174,7 +370,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",
@@ -188,11 +384,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",
@@ -205,7 +429,7 @@ var aiSDKConfigs = [
   },
   // streamText - async function
   {
-    channelName: "streamText",
+    channelName: aiSDKChannels.streamText.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -218,7 +442,7 @@ var aiSDKConfigs = [
   },
   // generateObject - async function
   {
-    channelName: "generateObject",
+    channelName: aiSDKChannels.generateObject.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -231,7 +455,7 @@ var aiSDKConfigs = [
   },
   // streamObject - async function
   {
-    channelName: "streamObject",
+    channelName: aiSDKChannels.streamObject.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -244,7 +468,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",
@@ -258,7 +482,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",
@@ -272,11 +496,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",
@@ -289,12 +524,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",
@@ -309,7 +556,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",

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

@@ -1,7 +1,7 @@
 import {
   unplugin
-} from "../chunk-OLOPGWTJ.mjs";
-import "../chunk-KVX7OFPD.mjs";
+} from "../chunk-D5ZPIUEL.mjs";
+import "../chunk-LVWWLUMN.mjs";
 
 // src/auto-instrumentations/bundler/vite.ts
 var vitePlugin = unplugin.vite;