braintrust 3.17.0 → 3.18.0

package/dist/auto-instrumentations/hook.mjs

@@ -16,6 +16,11 @@ 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",
   anthropic: "anthropic",
   aisdk: "aisdk",
   "ai-sdk": "aisdk",
@@ -6940,6 +6945,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 +7089,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 +8248,10 @@ var aiSDKChannels = defineChannels("ai", {
   toolLoopAgentStream: channel({
     channelName: "ToolLoopAgent.stream",
     kind: "async"
+  }),
+  v7CreateTelemetryDispatcher: channel({
+    channelName: "createTelemetryDispatcher",
+    kind: "sync-stream"
   })
 });
 
@@ -8241,7 +8262,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 +8274,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 +8300,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 +8324,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 +8337,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 +8349,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 +8362,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 +8374,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 +8387,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 +8399,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 +8412,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 +8424,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 +8432,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 +8465,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 +8489,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: {
@@ -9625,6 +9661,10 @@ var googleGenAIChannels = defineChannels("@google/genai", {
   embedContent: channel({
     channelName: "models.embedContent",
     kind: "async"
+  }),
+  interactionsCreate: channel({
+    channelName: "interactions.create",
+    kind: "async"
   })
 });
 
@@ -9688,6 +9728,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 +10881,35 @@ 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/auto-instrumentations/configs/all.ts
 var defaultInstrumentationConfigGroups = [
   { integrations: ["openai"], configs: openaiConfigs },
@@ -10874,6 +10957,10 @@ var defaultInstrumentationConfigGroups = [
     integrations: ["gitHubCopilot"],
     configs: gitHubCopilotConfigs
   },
+  {
+    integrations: ["piCodingAgent"],
+    configs: piCodingAgentConfigs
+  },
   {
     integrations: ["flue"],
     configs: flueConfigs
@@ -10944,6 +11031,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 +11045,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,

package/dist/auto-instrumentations/index.cjs

@@ -37,7 +37,8 @@ __export(index_exports, {
   openAICodexConfigs: () => openAICodexConfigs,
   openRouterAgentConfigs: () => openRouterAgentConfigs,
   openRouterConfigs: () => openRouterConfigs,
-  openaiConfigs: () => openaiConfigs
+  openaiConfigs: () => openaiConfigs,
+  piCodingAgentConfigs: () => piCodingAgentConfigs
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -582,6 +583,10 @@ var aiSDKChannels = defineChannels("ai", {
   toolLoopAgentStream: channel({
     channelName: "ToolLoopAgent.stream",
     kind: "async"
+  }),
+  v7CreateTelemetryDispatcher: channel({
+    channelName: "createTelemetryDispatcher",
+    kind: "sync-stream"
   })
 });
 
@@ -592,7 +597,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: {
@@ -604,7 +609,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: {
@@ -630,7 +635,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: {
@@ -654,7 +659,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: {
@@ -667,7 +672,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: {
@@ -679,7 +684,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: {
@@ -692,7 +697,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: {
@@ -704,7 +709,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: {
@@ -717,7 +722,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: {
@@ -729,7 +734,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: {
@@ -742,7 +747,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: {
@@ -754,7 +759,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: {
@@ -762,6 +767,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,
@@ -780,7 +800,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: {
@@ -804,7 +824,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: {
@@ -1095,6 +1115,10 @@ var googleGenAIChannels = defineChannels("@google/genai", {
   embedContent: channel({
     channelName: "models.embedContent",
     kind: "async"
+  }),
+  interactionsCreate: channel({
+    channelName: "interactions.create",
+    kind: "async"
   })
 });
 
@@ -1158,6 +1182,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"
+    }
   }
 ];
 
@@ -2814,6 +2852,35 @@ var langchainConfigs = [
     }
   }
 ];
+
+// 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"
+    }
+  }
+];
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   aiSDKConfigs,
@@ -2833,5 +2900,6 @@ var langchainConfigs = [
   openAICodexConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,
-  openaiConfigs
+  openaiConfigs,
+  piCodingAgentConfigs
 });

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

@@ -110,4 +110,6 @@ declare const gitHubCopilotConfigs: InstrumentationConfig[];
 
 declare const langchainConfigs: InstrumentationConfig[];
 
-export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, cursorSDKConfigs, genkitConfigs, gitHubCopilotConfigs, googleADKConfigs, googleGenAIConfigs, groqConfigs, huggingFaceConfigs, langchainConfigs, mistralConfigs, openAIAgentsCoreConfigs, openAICodexConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };
+declare const piCodingAgentConfigs: InstrumentationConfig[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, cursorSDKConfigs, genkitConfigs, gitHubCopilotConfigs, googleADKConfigs, googleGenAIConfigs, groqConfigs, huggingFaceConfigs, langchainConfigs, mistralConfigs, openAIAgentsCoreConfigs, openAICodexConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs, piCodingAgentConfigs };

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

@@ -110,4 +110,6 @@ declare const gitHubCopilotConfigs: InstrumentationConfig[];
 
 declare const langchainConfigs: InstrumentationConfig[];
 
-export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, cursorSDKConfigs, genkitConfigs, gitHubCopilotConfigs, googleADKConfigs, googleGenAIConfigs, groqConfigs, huggingFaceConfigs, langchainConfigs, mistralConfigs, openAIAgentsCoreConfigs, openAICodexConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };
+declare const piCodingAgentConfigs: InstrumentationConfig[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, cursorSDKConfigs, genkitConfigs, gitHubCopilotConfigs, googleADKConfigs, googleGenAIConfigs, groqConfigs, huggingFaceConfigs, langchainConfigs, mistralConfigs, openAIAgentsCoreConfigs, openAICodexConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs, piCodingAgentConfigs };

package/dist/auto-instrumentations/index.mjs

@@ -16,8 +16,9 @@ import {
   openAICodexConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,
-  openaiConfigs
-} from "./chunk-E5DUYJWK.mjs";
+  openaiConfigs,
+  piCodingAgentConfigs
+} from "./chunk-ZYKZEMRT.mjs";
 export {
   aiSDKConfigs,
   anthropicConfigs,
@@ -36,5 +37,6 @@ export {
   openAICodexConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,
-  openaiConfigs
+  openaiConfigs,
+  piCodingAgentConfigs
 };

package/dist/auto-instrumentations/loader/cjs-patch.cjs

@@ -327,6 +327,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,
@@ -340,7 +341,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,

package/dist/auto-instrumentations/loader/cjs-patch.mjs

@@ -41,6 +41,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,
@@ -54,7 +55,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,

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

@@ -47,6 +47,7 @@ async function resolve(specifier, context, nextResolve) {
   if (resolvedModule) {
     const packageName = getPackageName(resolvedModule.basedir) ?? resolvedModule.name;
     const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+    const version = getPackageVersion(resolvedModule.basedir);
     if (isSpecialCaseTarget(packageName, normalizedModulePath)) {
       specialCaseUrls.set(url.url, {
         packageName,
@@ -56,7 +57,6 @@ async function resolve(specifier, context, nextResolve) {
     if (!packages?.has(packageName)) {
       return url;
     }
-    const version = getPackageVersion(resolvedModule.basedir);
     const transformer = instrumentator.getTransformer(
       packageName,
       version,