braintrust 3.3.0 → 3.5.0

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

@@ -42,14 +42,192 @@ 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 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: {
@@ -58,9 +236,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 +278,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 +290,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 +319,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 +331,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 +359,7 @@ var openaiConfigs = [
     }
   },
   {
-    channelName: "responses.stream",
+    channelName: openAIChannels.responsesStream.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -142,7 +372,7 @@ var openaiConfigs = [
     }
   },
   {
-    channelName: "responses.parse",
+    channelName: openAIChannels.responsesParse.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -156,15 +386,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 +416,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 +430,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",
@@ -203,9 +489,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",
@@ -216,9 +514,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",
@@ -229,9 +539,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",
@@ -242,41 +564,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 <6.0.0",
+      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: ">=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: ">=3.0.0 <6.0.0",
+      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: ">=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",
@@ -289,12 +722,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 +754,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.d.mts

@@ -1,8 +1,8 @@
-import * as vite from 'vite';
+import * as unplugin from 'unplugin';
 import { B as BundlerPluginOptions } from '../plugin-Df3qKIl2.mjs';
 import '@apm-js-collab/code-transformer';
 
 type VitePluginOptions = BundlerPluginOptions;
-declare const vitePlugin: (options: BundlerPluginOptions) => vite.Plugin<any> | vite.Plugin<any>[];
+declare const vitePlugin: (options: BundlerPluginOptions) => unplugin.VitePlugin<any> | unplugin.VitePlugin<any>[];
 
 export { type VitePluginOptions, vitePlugin };

package/dist/auto-instrumentations/bundler/vite.d.ts

@@ -1,8 +1,8 @@
-import * as vite from 'vite';
+import * as unplugin from 'unplugin';
 import { B as BundlerPluginOptions } from '../plugin-Df3qKIl2.js';
 import '@apm-js-collab/code-transformer';
 
 type VitePluginOptions = BundlerPluginOptions;
-declare const vitePlugin: (options: BundlerPluginOptions) => vite.Plugin<any> | vite.Plugin<any>[];
+declare const vitePlugin: (options: BundlerPluginOptions) => unplugin.VitePlugin<any> | unplugin.VitePlugin<any>[];
 
 export { type VitePluginOptions, vitePlugin };