braintrust 3.8.0 → 3.10.0

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

@@ -53,6 +53,8 @@ declare const aiSDKConfigs: InstrumentationConfig[];
  */
 declare const claudeAgentSDKConfigs: InstrumentationConfig[];
 
+declare const cursorSDKConfigs: InstrumentationConfig[];
+
 /**
  * Instrumentation configurations for the Google GenAI SDK.
  *
@@ -66,10 +68,29 @@ declare const claudeAgentSDKConfigs: InstrumentationConfig[];
  */
 declare const googleGenAIConfigs: InstrumentationConfig[];
 
+declare const huggingFaceConfigs: InstrumentationConfig[];
+
 declare const openRouterAgentConfigs: InstrumentationConfig[];
 
 declare const openRouterConfigs: InstrumentationConfig[];
 
 declare const mistralConfigs: InstrumentationConfig[];
 
-export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, googleGenAIConfigs, mistralConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };
+/**
+ * Instrumentation configurations for the Google ADK (@google/adk).
+ *
+ * Runner.runAsync and BaseAgent.runAsync are async generators (`async *`).
+ * They synchronously return an AsyncGenerator object, so we use kind "Sync"
+ * paired with "sync-stream" channels — the same pattern used by the
+ * Claude Agent SDK's `query` function.
+ *
+ * FunctionTool.runAsync is a regular async method (returns Promise<unknown>)
+ * and uses kind "Async".
+ */
+declare const googleADKConfigs: InstrumentationConfig[];
+
+declare const cohereConfigs: InstrumentationConfig[];
+
+declare const groqConfigs: InstrumentationConfig[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, cursorSDKConfigs, googleADKConfigs, googleGenAIConfigs, groqConfigs, huggingFaceConfigs, mistralConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };

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

@@ -53,6 +53,8 @@ declare const aiSDKConfigs: InstrumentationConfig[];
  */
 declare const claudeAgentSDKConfigs: InstrumentationConfig[];
 
+declare const cursorSDKConfigs: InstrumentationConfig[];
+
 /**
  * Instrumentation configurations for the Google GenAI SDK.
  *
@@ -66,10 +68,29 @@ declare const claudeAgentSDKConfigs: InstrumentationConfig[];
  */
 declare const googleGenAIConfigs: InstrumentationConfig[];
 
+declare const huggingFaceConfigs: InstrumentationConfig[];
+
 declare const openRouterAgentConfigs: InstrumentationConfig[];
 
 declare const openRouterConfigs: InstrumentationConfig[];
 
 declare const mistralConfigs: InstrumentationConfig[];
 
-export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, googleGenAIConfigs, mistralConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };
+/**
+ * Instrumentation configurations for the Google ADK (@google/adk).
+ *
+ * Runner.runAsync and BaseAgent.runAsync are async generators (`async *`).
+ * They synchronously return an AsyncGenerator object, so we use kind "Sync"
+ * paired with "sync-stream" channels — the same pattern used by the
+ * Claude Agent SDK's `query` function.
+ *
+ * FunctionTool.runAsync is a regular async method (returns Promise<unknown>)
+ * and uses kind "Async".
+ */
+declare const googleADKConfigs: InstrumentationConfig[];
+
+declare const cohereConfigs: InstrumentationConfig[];
+
+declare const groqConfigs: InstrumentationConfig[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, cursorSDKConfigs, googleADKConfigs, googleGenAIConfigs, groqConfigs, huggingFaceConfigs, mistralConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };

package/dist/auto-instrumentations/index.mjs

@@ -1,18 +1,273 @@
 import {
   aiSDKConfigs,
   anthropicConfigs,
+  channel,
   claudeAgentSDKConfigs,
+  cohereConfigs,
+  cursorSDKConfigs,
+  defineChannels,
   googleGenAIConfigs,
+  groqConfigs,
+  huggingFaceConfigs,
   mistralConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,
   openaiConfigs
-} from "./chunk-OLBMPZXE.mjs";
+} from "./chunk-GZNXBBPU.mjs";
+
+// src/instrumentation/plugins/google-adk-channels.ts
+var googleADKChannels = defineChannels("@google/adk", {
+  runnerRunAsync: channel({
+    channelName: "runner.runAsync",
+    kind: "sync-stream"
+  }),
+  agentRunAsync: channel({
+    channelName: "agent.runAsync",
+    kind: "sync-stream"
+  }),
+  toolRunAsync: channel({
+    channelName: "tool.runAsync",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/google-adk.ts
+var googleADKVersionRange = ">=0.1.0";
+var googleADKBundledIndexV06VersionRange = ">=0.6.1 <0.7.0";
+var googleADKBundledIndexV1VersionRange = ">=1.0.0 <2.0.0";
+var googleADKConfigs = [
+  // --- Runner.runAsync --- async generator, kind "Sync" + sync-stream channel
+  // Runner.runAsync — ESM individual module file
+  {
+    channelName: googleADKChannels.runnerRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKVersionRange,
+      filePath: "dist/esm/runner/runner.js"
+    },
+    functionQuery: {
+      className: "Runner",
+      methodName: "runAsync",
+      kind: "Sync"
+    }
+  },
+  // Runner.runAsync — bundled CJS/ESM indexes
+  // The bundled entrypoints minify class names, so target the 12th sync
+  // `runAsync` method in file order rather than a class name. This mapping is
+  // only validated against the current 0.6.x bundle layout, so keep the range
+  // tight until we verify newer bundled outputs.
+  {
+    channelName: googleADKChannels.runnerRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV06VersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 11
+    }
+  },
+  {
+    channelName: googleADKChannels.runnerRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV06VersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 11
+    }
+  },
+  // The 1.x bundled entrypoints still inline the runtime into index.js, but
+  // the minified method order changed. These indices are verified against 1.0.0.
+  {
+    channelName: googleADKChannels.runnerRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV1VersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 12
+    }
+  },
+  {
+    channelName: googleADKChannels.runnerRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV1VersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 12
+    }
+  },
+  // --- BaseAgent.runAsync --- async generator, kind "Sync" + sync-stream channel
+  // BaseAgent.runAsync — ESM individual module file
+  {
+    channelName: googleADKChannels.agentRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKVersionRange,
+      filePath: "dist/esm/agents/base_agent.js"
+    },
+    functionQuery: {
+      className: "BaseAgent",
+      methodName: "runAsync",
+      kind: "Sync"
+    }
+  },
+  // BaseAgent.runAsync — bundled CJS/ESM indexes
+  // The bundled entrypoints minify class names, so target the first sync
+  // `runAsync` method in file order rather than a class name. This mapping is
+  // only validated against the current 0.6.x bundle layout, so keep the range
+  // tight until we verify newer bundled outputs.
+  {
+    channelName: googleADKChannels.agentRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV06VersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 0
+    }
+  },
+  {
+    channelName: googleADKChannels.agentRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV06VersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 0
+    }
+  },
+  // The 1.x bundled entrypoints keep BaseAgent.runAsync as the first bundled
+  // async-generator runAsync method in file order.
+  {
+    channelName: googleADKChannels.agentRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV1VersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 0
+    }
+  },
+  {
+    channelName: googleADKChannels.agentRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV1VersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 0
+    }
+  },
+  // --- FunctionTool.runAsync --- regular async, kind "Async"
+  // FunctionTool.runAsync — ESM individual module file
+  {
+    channelName: googleADKChannels.toolRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKVersionRange,
+      filePath: "dist/esm/tools/function_tool.js"
+    },
+    functionQuery: {
+      className: "FunctionTool",
+      methodName: "runAsync",
+      kind: "Async"
+    }
+  },
+  // FunctionTool.runAsync — bundled CJS/ESM indexes
+  // The bundled entrypoints minify class names, so target the first async
+  // `runAsync` method in file order rather than a class name. This mapping is
+  // only validated against the current 0.6.x bundle layout, so keep the range
+  // tight until we verify newer bundled outputs.
+  {
+    channelName: googleADKChannels.toolRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV06VersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Async",
+      index: 1
+    }
+  },
+  {
+    channelName: googleADKChannels.toolRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV06VersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Async",
+      index: 1
+    }
+  },
+  // The 1.x bundle moves FunctionTool.runAsync behind one helper method.
+  {
+    channelName: googleADKChannels.toolRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV1VersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Async",
+      index: 1
+    }
+  },
+  {
+    channelName: googleADKChannels.toolRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexV1VersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Async",
+      index: 1
+    }
+  }
+];
 export {
   aiSDKConfigs,
   anthropicConfigs,
   claudeAgentSDKConfigs,
+  cohereConfigs,
+  cursorSDKConfigs,
+  googleADKConfigs,
   googleGenAIConfigs,
+  groqConfigs,
+  huggingFaceConfigs,
   mistralConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,

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

@@ -55,9 +55,10 @@ async function load(url, context, nextLoad) {
   if (code) {
     const transformer = transformers.get(url);
     try {
+      const moduleType = result.format === "module" ? "esm" : result.format === "commonjs" ? "cjs" : "unknown";
       const transformedCode = transformer.transform(
         code.toString("utf8"),
-        "unknown"
+        moduleType
       );
       result.source = transformedCode?.code;
       result.shortCircuit = true;