braintrust 3.17.0 → 3.19.0

package/dist/auto-instrumentations/hook.mjs

@@ -16,6 +16,15 @@ var envIntegrationAliases = {
   openaicodexsdk: "openaiCodexSDK",
   codex: "openaiCodexSDK",
   "codex-sdk": "openaiCodexSDK",
+  "pi-coding-agent": "piCodingAgent",
+  "pi-coding-agent-sdk": "piCodingAgent",
+  picodingagent: "piCodingAgent",
+  picodingagentsdk: "piCodingAgent",
+  "@earendil-works/pi-coding-agent": "piCodingAgent",
+  strandsAgentSDK: "strandsAgentSDK",
+  strandsagentsdk: "strandsAgentSDK",
+  "strands-agent-sdk": "strandsAgentSDK",
+  "@strands-agents/sdk": "strandsAgentSDK",
   anthropic: "anthropic",
   aisdk: "aisdk",
   "ai-sdk": "aisdk",
@@ -46,6 +55,12 @@ var envIntegrationAliases = {
   cohere: "cohere",
   groq: "groq",
   "groq-sdk": "groq",
+  bedrock: "bedrock",
+  "aws-bedrock": "awsBedrock",
+  awsbedrock: "awsBedrock",
+  "aws-bedrock-runtime": "awsBedrockRuntime",
+  awsbedrockruntime: "awsBedrockRuntime",
+  "@aws-sdk/client-bedrock-runtime": "awsBedrockRuntime",
   genkit: "genkit",
   "firebase-genkit": "genkit",
   githubcopilot: "gitHubCopilot",
@@ -60,9 +75,10 @@ function readDisabledInstrumentationEnvConfig(disabledList) {
   const integrations = {};
   if (disabledList) {
     for (const value of disabledList.split(",")) {
-      const sdk = value.trim().toLowerCase();
+      const rawSdk = value.trim();
+      const sdk = rawSdk.toLowerCase();
       if (sdk.length > 0) {
-        integrations[envIntegrationAliases[sdk] ?? sdk] = false;
+        integrations[envIntegrationAliases[rawSdk] ?? envIntegrationAliases[sdk] ?? sdk] = false;
       }
     }
   }
@@ -6940,6 +6956,10 @@ var Experiment2 = class extends ObjectFetcher {
       return (await this.lazyMetadata.get()).project;
     })();
   }
+  async _getBaseExperimentId() {
+    const baseExperimentId = (await this.lazyMetadata.get()).experiment.fullInfo["base_exp_id"];
+    return typeof baseExperimentId === "string" && baseExperimentId ? baseExperimentId : void 0;
+  }
   parentObjectType() {
     return 1 /* EXPERIMENT */;
   }
@@ -7080,6 +7100,14 @@ var Experiment2 = class extends ObjectFetcher {
           comparisonExperimentId = baseExperiment.id;
           comparisonExperimentName = baseExperiment.name;
         }
+      } else {
+        try {
+          const comparisonExperiment = await state2.apiConn().get_json(`v1/experiment/${comparisonExperimentId}`);
+          if (typeof comparisonExperiment["name"] === "string") {
+            comparisonExperimentName = comparisonExperiment["name"];
+          }
+        } catch {
+        }
       }
       try {
         const results = await state2.apiConn().get_json(
@@ -8231,6 +8259,10 @@ var aiSDKChannels = defineChannels("ai", {
   toolLoopAgentStream: channel({
     channelName: "ToolLoopAgent.stream",
     kind: "async"
+  }),
+  v7CreateTelemetryDispatcher: channel({
+    channelName: "createTelemetryDispatcher",
+    kind: "sync-stream"
   })
 });
 
@@ -8241,7 +8273,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.generateText.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <7.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
@@ -8253,7 +8285,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.generateText.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <7.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -8279,7 +8311,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.streamTextSync.channelName,
     module: {
       name: "ai",
-      versionRange: ">=4.0.0",
+      versionRange: ">=4.0.0 <7.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
@@ -8303,7 +8335,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.streamTextSync.channelName,
     module: {
       name: "ai",
-      versionRange: ">=4.0.0",
+      versionRange: ">=4.0.0 <7.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -8316,7 +8348,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.generateObject.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <7.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
@@ -8328,7 +8360,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.generateObject.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <7.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -8341,7 +8373,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.embed.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <7.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
@@ -8353,7 +8385,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.embed.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <7.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -8366,7 +8398,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.embedMany.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <7.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
@@ -8378,7 +8410,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.embedMany.channelName,
     module: {
       name: "ai",
-      versionRange: ">=3.0.0",
+      versionRange: ">=3.0.0 <7.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -8391,7 +8423,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.rerank.channelName,
     module: {
       name: "ai",
-      versionRange: ">=5.0.0",
+      versionRange: ">=5.0.0 <7.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
@@ -8403,7 +8435,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.rerank.channelName,
     module: {
       name: "ai",
-      versionRange: ">=5.0.0",
+      versionRange: ">=5.0.0 <7.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -8411,6 +8443,21 @@ var aiSDKConfigs = [
       kind: "Async"
     }
   },
+  // AI SDK v7 exposes its telemetry lifecycle through a dispatcher created for
+  // each operation. We patch that dispatcher in the plugin instead of rewriting
+  // the module to call registerTelemetry().
+  {
+    channelName: aiSDKChannels.v7CreateTelemetryDispatcher.channelName,
+    module: {
+      name: "ai",
+      versionRange: ">=7.0.0-0 <8.0.0",
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      functionName: "createTelemetryDispatcher",
+      kind: "Sync"
+    }
+  },
   // streamObject - async function (v3 only, before the sync refactor in v4)
   {
     channelName: aiSDKChannels.streamObject.channelName,
@@ -8429,7 +8476,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.streamObjectSync.channelName,
     module: {
       name: "ai",
-      versionRange: ">=4.0.0",
+      versionRange: ">=4.0.0 <7.0.0",
       filePath: "dist/index.mjs"
     },
     functionQuery: {
@@ -8453,7 +8500,7 @@ var aiSDKConfigs = [
     channelName: aiSDKChannels.streamObjectSync.channelName,
     module: {
       name: "ai",
-      versionRange: ">=4.0.0",
+      versionRange: ">=4.0.0 <7.0.0",
       filePath: "dist/index.js"
     },
     functionQuery: {
@@ -8597,6 +8644,10 @@ var anthropicChannels = defineChannels("@anthropic-ai/sdk", {
 
 // src/auto-instrumentations/configs/anthropic.ts
 var anthropicConfigs = [
+  // Each logical target is listed for both published module formats:
+  // `.mjs` covers ESM imports, while `.js` covers CJS requires. The Bedrock
+  // SDK delegates CJS `messages.create` calls through these Anthropic SDK
+  // `.js` resource files.
   // Messages API - create in older SDK layouts (supports streaming via stream=true parameter)
   {
     channelName: anthropicChannels.messagesCreate.channelName,
@@ -8611,6 +8662,19 @@ var anthropicConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: anthropicChannels.messagesCreate.channelName,
+    module: {
+      name: "@anthropic-ai/sdk",
+      versionRange: ">=0.27.0 <0.39.0",
+      filePath: "resources/messages.js"
+    },
+    functionQuery: {
+      className: "Messages",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
   // Messages API - create (supports streaming via stream=true parameter)
   {
     channelName: anthropicChannels.messagesCreate.channelName,
@@ -8625,6 +8689,19 @@ var anthropicConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: anthropicChannels.messagesCreate.channelName,
+    module: {
+      name: "@anthropic-ai/sdk",
+      versionRange: ">=0.39.0",
+      filePath: "resources/messages/messages.js"
+    },
+    functionQuery: {
+      className: "Messages",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
   // Beta Messages API - create (supports streaming via stream=true parameter)
   {
     channelName: anthropicChannels.betaMessagesCreate.channelName,
@@ -8639,6 +8716,19 @@ var anthropicConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: anthropicChannels.betaMessagesCreate.channelName,
+    module: {
+      name: "@anthropic-ai/sdk",
+      versionRange: ">=0.39.0",
+      filePath: "resources/beta/messages/messages.js"
+    },
+    functionQuery: {
+      className: "Messages",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
   // Beta Messages API - toolRunner (sync helper returning async iterable/thenable)
   {
     channelName: anthropicChannels.betaMessagesToolRunner.channelName,
@@ -8652,6 +8742,90 @@ var anthropicConfigs = [
       methodName: "toolRunner",
       kind: "Sync"
     }
+  },
+  {
+    channelName: anthropicChannels.betaMessagesToolRunner.channelName,
+    module: {
+      name: "@anthropic-ai/sdk",
+      versionRange: ">=0.39.0",
+      filePath: "resources/beta/messages/messages.js"
+    },
+    functionQuery: {
+      className: "Messages",
+      methodName: "toolRunner",
+      kind: "Sync"
+    }
+  }
+];
+
+// src/instrumentation/plugins/bedrock-runtime-channels.ts
+var clientSendChannel = channel({
+  channelName: "client.send",
+  kind: "async"
+});
+var bedrockRuntimeChannels = defineChannels("aws-bedrock-runtime", {
+  clientSend: clientSendChannel
+});
+var smithyCoreChannels = defineChannels("@smithy/core", {
+  clientSend: clientSendChannel
+});
+var smithyClientChannels = defineChannels("@smithy/smithy-client", {
+  clientSend: clientSendChannel
+});
+
+// src/auto-instrumentations/configs/bedrock-runtime.ts
+var bedrockRuntimeConfigs = [
+  {
+    channelName: smithyCoreChannels.clientSend.channelName,
+    module: {
+      name: "@smithy/core",
+      versionRange: ">=3.0.0 <4.0.0",
+      filePath: "dist-cjs/submodules/client/index.js"
+    },
+    functionQuery: {
+      className: "Client",
+      methodName: "send",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: smithyCoreChannels.clientSend.channelName,
+    module: {
+      name: "@smithy/core",
+      versionRange: ">=3.0.0 <4.0.0",
+      filePath: "dist-es/submodules/client/smithy-client/client.js"
+    },
+    functionQuery: {
+      className: "Client",
+      methodName: "send",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: smithyClientChannels.clientSend.channelName,
+    module: {
+      name: "@smithy/smithy-client",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist-cjs/index.js"
+    },
+    functionQuery: {
+      className: "Client",
+      methodName: "send",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: smithyClientChannels.clientSend.channelName,
+    module: {
+      name: "@smithy/smithy-client",
+      versionRange: ">=3.0.0 <5.0.0",
+      filePath: "dist-es/client.js"
+    },
+    functionQuery: {
+      className: "Client",
+      methodName: "send",
+      kind: "Async"
+    }
   }
 ];
 
@@ -9055,7 +9229,7 @@ var flueChannels = defineChannels("@flue/runtime", {
 });
 
 // src/auto-instrumentations/configs/flue.ts
-var flueVersionRange = ">=0.8.0";
+var flueVersionRange = ">=0.8.0 <1.0.0";
 var flueConfigs = [
   {
     channelName: flueChannels.createContext.channelName,
@@ -9625,6 +9799,10 @@ var googleGenAIChannels = defineChannels("@google/genai", {
   embedContent: channel({
     channelName: "models.embedContent",
     kind: "async"
+  }),
+  interactionsCreate: channel({
+    channelName: "interactions.create",
+    kind: "async"
   })
 });
 
@@ -9688,6 +9866,20 @@ var googleGenAIConfigs = [
       methodName: "embedContentInternal",
       kind: "Async"
     }
+  },
+  // BaseInteractions.create - Interactions API entry point
+  {
+    channelName: googleGenAIChannels.interactionsCreate.channelName,
+    module: {
+      name: "@google/genai",
+      versionRange: ">=1.33.0",
+      filePath: "dist/node/index.mjs"
+    },
+    functionQuery: {
+      className: "BaseInteractions",
+      methodName: "create",
+      kind: "Async"
+    }
   }
 ];
 
@@ -10827,6 +11019,95 @@ var openRouterAgentConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/pi-coding-agent-channels.ts
+var piCodingAgentChannels = defineChannels(
+  "@earendil-works/pi-coding-agent",
+  {
+    prompt: channel({
+      channelName: "AgentSession.prompt",
+      kind: "async"
+    })
+  }
+);
+
+// src/auto-instrumentations/configs/pi-coding-agent.ts
+var piCodingAgentVersionRange = ">=0.79.0 <0.80.0";
+var piCodingAgentConfigs = [
+  {
+    channelName: piCodingAgentChannels.prompt.channelName,
+    module: {
+      name: "@earendil-works/pi-coding-agent",
+      versionRange: piCodingAgentVersionRange,
+      filePath: "dist/core/agent-session.js"
+    },
+    functionQuery: {
+      className: "AgentSession",
+      methodName: "prompt",
+      kind: "Async"
+    }
+  }
+];
+
+// src/instrumentation/plugins/strands-agent-sdk-channels.ts
+var strandsAgentSDKChannels = defineChannels("@strands-agents/sdk", {
+  agentStream: channel({
+    channelName: "Agent.stream",
+    kind: "sync-stream"
+  }),
+  graphStream: channel({
+    channelName: "Graph.stream",
+    kind: "sync-stream"
+  }),
+  swarmStream: channel({
+    channelName: "Swarm.stream",
+    kind: "sync-stream"
+  })
+});
+
+// src/auto-instrumentations/configs/strands-agent-sdk.ts
+var strandsAgentSDKVersionRange = ">=1.0.0 <2.0.0";
+var strandsAgentSDKConfigs = [
+  {
+    channelName: strandsAgentSDKChannels.agentStream.channelName,
+    module: {
+      name: "@strands-agents/sdk",
+      versionRange: strandsAgentSDKVersionRange,
+      filePath: "dist/src/agent/agent.js"
+    },
+    functionQuery: {
+      className: "Agent",
+      methodName: "stream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: strandsAgentSDKChannels.graphStream.channelName,
+    module: {
+      name: "@strands-agents/sdk",
+      versionRange: strandsAgentSDKVersionRange,
+      filePath: "dist/src/multiagent/graph.js"
+    },
+    functionQuery: {
+      className: "Graph",
+      methodName: "stream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: strandsAgentSDKChannels.swarmStream.channelName,
+    module: {
+      name: "@strands-agents/sdk",
+      versionRange: strandsAgentSDKVersionRange,
+      filePath: "dist/src/multiagent/swarm.js"
+    },
+    functionQuery: {
+      className: "Swarm",
+      methodName: "stream",
+      kind: "Sync"
+    }
+  }
+];
+
 // src/auto-instrumentations/configs/all.ts
 var defaultInstrumentationConfigGroups = [
   { integrations: ["openai"], configs: openaiConfigs },
@@ -10835,6 +11116,10 @@ var defaultInstrumentationConfigGroups = [
     configs: openAICodexConfigs
   },
   { integrations: ["anthropic"], configs: anthropicConfigs },
+  {
+    integrations: ["bedrock", "awsBedrock", "awsBedrockRuntime"],
+    configs: bedrockRuntimeConfigs
+  },
   {
     integrations: ["aisdk", "vercel"],
     configs: aiSDKConfigs
@@ -10874,6 +11159,14 @@ var defaultInstrumentationConfigGroups = [
     integrations: ["gitHubCopilot"],
     configs: gitHubCopilotConfigs
   },
+  {
+    integrations: ["piCodingAgent"],
+    configs: piCodingAgentConfigs
+  },
+  {
+    integrations: ["strandsAgentSDK"],
+    configs: strandsAgentSDKConfigs
+  },
   {
     integrations: ["flue"],
     configs: flueConfigs
@@ -10944,6 +11237,7 @@ var ModulePatch = class {
       if (resolvedModule) {
         const packageName = getPackageName(resolvedModule.basedir) ?? resolvedModule.name;
         const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+        const version = getPackageVersion(resolvedModule.basedir);
         const patched = applySpecialCasePatch({
           packageName,
           modulePath: normalizedModulePath,
@@ -10957,7 +11251,6 @@ var ModulePatch = class {
         if (!self.packages.has(packageName)) {
           return self.originalCompile.apply(this, args);
         }
-        const version = getPackageVersion(resolvedModule.basedir);
         const transformer = self.instrumentator.getTransformer(
           packageName,
           version,