braintrust 3.3.0-rc.45 → 3.4.0

package/dist/auto-instrumentations/index.cjs

@@ -28,14 +28,146 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 
+// 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: {
@@ -44,9 +176,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",
@@ -60,10 +218,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: {
@@ -72,9 +230,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",
@@ -88,10 +259,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: {
@@ -100,9 +271,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",
@@ -115,7 +299,7 @@ var openaiConfigs = [
     }
   },
   {
-    channelName: "responses.stream",
+    channelName: openAIChannels.responsesStream.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -128,7 +312,7 @@ var openaiConfigs = [
     }
   },
   {
-    channelName: "responses.parse",
+    channelName: openAIChannels.responsesParse.channelName,
     module: {
       name: "openai",
       versionRange: ">=4.87.0",
@@ -142,15 +326,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",
@@ -160,7 +356,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",
@@ -174,11 +370,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",
@@ -191,7 +415,7 @@ var aiSDKConfigs = [
   },
   // streamText - async function
   {
-    channelName: "streamText",
+    channelName: aiSDKChannels.streamText.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -204,7 +428,7 @@ var aiSDKConfigs = [
   },
   // generateObject - async function
   {
-    channelName: "generateObject",
+    channelName: aiSDKChannels.generateObject.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -217,7 +441,7 @@ var aiSDKConfigs = [
   },
   // streamObject - async function
   {
-    channelName: "streamObject",
+    channelName: aiSDKChannels.streamObject.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
@@ -230,7 +454,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",
@@ -244,7 +468,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",
@@ -258,11 +482,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",
@@ -275,12 +510,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",
@@ -295,7 +542,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/index.d.mts

@@ -8,8 +8,8 @@ export { InstrumentationConfig } from '@apm-js-collab/code-transformer';
  * to emit events on. They are used by orchestrion-js to perform AST
  * transformation at build-time or load-time.
  *
- * NOTE: Channel names should NOT include the braintrust: prefix. The code-transformer
- * will prepend "orchestrion:openai:" to these names, resulting in final channel names like:
+ * NOTE: Channel names should NOT include the orchestrion: prefix. The code-transformer
+ * will prepend "orchestrion:" + module.name + ":" to these names, resulting in final channel names like:
  * "orchestrion:openai:chat.completions.create"
  */
 declare const openaiConfigs: InstrumentationConfig[];
@@ -21,9 +21,9 @@ declare const openaiConfigs: InstrumentationConfig[];
  * to emit events on. They are used by orchestrion-js to perform AST
  * transformation at build-time or load-time.
  *
- * NOTE: Channel names should NOT include the braintrust: prefix. The code-transformer
- * will prepend "orchestrion:anthropic:" to these names, resulting in final channel names like:
- * "orchestrion:anthropic:messages.create"
+ * NOTE: Channel names should NOT include the orchestrion: prefix. The code-transformer
+ * will prepend "orchestrion:" + module.name + ":" to these names, resulting in final channel names like:
+ * "orchestrion:@anthropic-ai/sdk:messages.create"
  */
 declare const anthropicConfigs: InstrumentationConfig[];
 

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

@@ -8,8 +8,8 @@ export { InstrumentationConfig } from '@apm-js-collab/code-transformer';
  * to emit events on. They are used by orchestrion-js to perform AST
  * transformation at build-time or load-time.
  *
- * NOTE: Channel names should NOT include the braintrust: prefix. The code-transformer
- * will prepend "orchestrion:openai:" to these names, resulting in final channel names like:
+ * NOTE: Channel names should NOT include the orchestrion: prefix. The code-transformer
+ * will prepend "orchestrion:" + module.name + ":" to these names, resulting in final channel names like:
  * "orchestrion:openai:chat.completions.create"
  */
 declare const openaiConfigs: InstrumentationConfig[];
@@ -21,9 +21,9 @@ declare const openaiConfigs: InstrumentationConfig[];
  * to emit events on. They are used by orchestrion-js to perform AST
  * transformation at build-time or load-time.
  *
- * NOTE: Channel names should NOT include the braintrust: prefix. The code-transformer
- * will prepend "orchestrion:anthropic:" to these names, resulting in final channel names like:
- * "orchestrion:anthropic:messages.create"
+ * NOTE: Channel names should NOT include the orchestrion: prefix. The code-transformer
+ * will prepend "orchestrion:" + module.name + ":" to these names, resulting in final channel names like:
+ * "orchestrion:@anthropic-ai/sdk:messages.create"
  */
 declare const anthropicConfigs: InstrumentationConfig[];
 

package/dist/auto-instrumentations/index.mjs

@@ -4,7 +4,7 @@ import {
   claudeAgentSDKConfigs,
   googleGenAIConfigs,
   openaiConfigs
-} from "./chunk-KVX7OFPD.mjs";
+} from "./chunk-LVWWLUMN.mjs";
 export {
   aiSDKConfigs,
   anthropicConfigs,

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

@@ -10,21 +10,20 @@ import {
   create
 } from "@apm-js-collab/code-transformer";
 import moduleDetailsFromPath from "module-details-from-path";
-var transformers = null;
-var packages = null;
-var instrumentator = null;
+var instrumentator;
+var packages;
+var transformers = /* @__PURE__ */ new Map();
 async function initialize(data = {}) {
-  const instrumentations = data?.instrumentations || [];
-  instrumentator = create(instrumentations);
-  packages = new Set(instrumentations.map((i) => i.module.name));
-  transformers = /* @__PURE__ */ new Map();
+  const configs = data.instrumentations || [];
+  instrumentator = create(configs);
+  packages = new Set(configs.map((i) => i.module.name));
 }
 async function resolve(specifier, context, nextResolve) {
   const url = await nextResolve(specifier, context);
   const filePath = url.url.startsWith("file:") ? fileURLToPath(url.url) : url.url;
   const normalizedForPlatform = filePath.split("/").join(sep);
   const resolvedModule = moduleDetailsFromPath(normalizedForPlatform);
-  if (resolvedModule && packages.has(resolvedModule.name)) {
+  if (resolvedModule && packages?.has(resolvedModule.name)) {
     const version = getPackageVersion(resolvedModule.basedir);
     const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
     const transformer = instrumentator.getTransformer(