braintrust 3.17.0 → 3.18.0

package/dist/apply-auto-instrumentation.mjs

@@ -23,8 +23,9 @@ import {
   openRouterChannels,
   patchMastraSource,
   patchTracingChannel,
+  piCodingAgentChannels,
   readDisabledInstrumentationEnvConfig
-} from "./chunk-O4ZIWXO3.mjs";
+} from "./chunk-BYFADLEZ.mjs";
 
 // src/node/apply-auto-instrumentation-entry.ts
 import * as diagnostics_channel from "node:diagnostics_channel";
@@ -38,7 +39,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: {
@@ -50,7 +51,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: {
@@ -76,7 +77,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: {
@@ -100,7 +101,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: {
@@ -113,7 +114,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: {
@@ -125,7 +126,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: {
@@ -138,7 +139,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: {
@@ -150,7 +151,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: {
@@ -163,7 +164,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: {
@@ -175,7 +176,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: {
@@ -188,7 +189,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: {
@@ -200,7 +201,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: {
@@ -208,6 +209,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,
@@ -226,7 +242,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: {
@@ -250,7 +266,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: {
@@ -1328,6 +1344,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"
+    }
   }
 ];
 
@@ -2247,6 +2277,24 @@ var openRouterAgentConfigs = [
   }
 ];
 
+// 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 },
@@ -2294,6 +2342,10 @@ var defaultInstrumentationConfigGroups = [
     integrations: ["gitHubCopilot"],
     configs: gitHubCopilotConfigs
   },
+  {
+    integrations: ["piCodingAgent"],
+    configs: piCodingAgentConfigs
+  },
   {
     integrations: ["flue"],
     configs: flueConfigs
@@ -2457,6 +2509,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,
@@ -2470,7 +2523,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/bundler/esbuild.cjs

@@ -51,6 +51,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",
@@ -310,6 +315,10 @@ var aiSDKChannels = defineChannels("ai", {
   toolLoopAgentStream: channel({
     channelName: "ToolLoopAgent.stream",
     kind: "async"
+  }),
+  v7CreateTelemetryDispatcher: channel({
+    channelName: "createTelemetryDispatcher",
+    kind: "sync-stream"
   })
 });
 
@@ -320,7 +329,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: {
@@ -332,7 +341,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: {
@@ -358,7 +367,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: {
@@ -382,7 +391,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: {
@@ -395,7 +404,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: {
@@ -407,7 +416,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: {
@@ -420,7 +429,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: {
@@ -432,7 +441,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: {
@@ -445,7 +454,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: {
@@ -457,7 +466,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: {
@@ -470,7 +479,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: {
@@ -482,7 +491,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: {
@@ -490,6 +499,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,
@@ -508,7 +532,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: {
@@ -532,7 +556,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: {
@@ -1704,6 +1728,10 @@ var googleGenAIChannels = defineChannels("@google/genai", {
   embedContent: channel({
     channelName: "models.embedContent",
     kind: "async"
+  }),
+  interactionsCreate: channel({
+    channelName: "interactions.create",
+    kind: "async"
   })
 });
 
@@ -1767,6 +1795,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"
+    }
   }
 ];
 
@@ -2906,6 +2948,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 },
@@ -2953,6 +3024,10 @@ var defaultInstrumentationConfigGroups = [
     integrations: ["gitHubCopilot"],
     configs: gitHubCopilotConfigs
   },
+  {
+    integrations: ["piCodingAgent"],
+    configs: piCodingAgentConfigs
+  },
   {
     integrations: ["flue"],
     configs: flueConfigs
@@ -3225,6 +3300,7 @@ var unplugin = (0, import_unplugin.createUnplugin)(
         }
         const moduleName = moduleDetails.name;
         const normalizedModulePath = moduleDetails.path.replace(/\\/g, "/");
+        const moduleVersion = getModuleVersion(moduleDetails.basedir);
         if (options.browser !== true) {
           const patched = applySpecialCasePatch({
             packageName: moduleName,
@@ -3236,7 +3312,6 @@ var unplugin = (0, import_unplugin.createUnplugin)(
             return { code: patched, map: null };
           }
         }
-        const moduleVersion = getModuleVersion(moduleDetails.basedir);
         if (!moduleVersion) {
           console.warn(
             `No 'package.json' version found for module ${moduleName} at ${moduleDetails.basedir}. Skipping transformation.`

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

@@ -1,7 +1,7 @@
 import {
   unplugin
-} from "../chunk-VXJONZVX.mjs";
-import "../chunk-E5DUYJWK.mjs";
+} from "../chunk-GNUEZ2PE.mjs";
+import "../chunk-ZYKZEMRT.mjs";
 import "../chunk-J57YF7WS.mjs";
 
 // src/auto-instrumentations/bundler/esbuild.ts