braintrust 3.8.0 → 3.9.0

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

@@ -66,10 +66,27 @@ 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[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, googleADKConfigs, googleGenAIConfigs, huggingFaceConfigs, mistralConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };

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

@@ -66,10 +66,27 @@ 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[];
+
+export { aiSDKConfigs, anthropicConfigs, claudeAgentSDKConfigs, cohereConfigs, googleADKConfigs, googleGenAIConfigs, huggingFaceConfigs, mistralConfigs, openRouterAgentConfigs, openRouterConfigs, openaiConfigs };

package/dist/auto-instrumentations/index.mjs

@@ -1,18 +1,185 @@
 import {
   aiSDKConfigs,
   anthropicConfigs,
+  channel,
   claudeAgentSDKConfigs,
+  cohereConfigs,
+  defineChannels,
   googleGenAIConfigs,
+  huggingFaceConfigs,
   mistralConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,
   openaiConfigs
-} from "./chunk-OLBMPZXE.mjs";
+} from "./chunk-KIMMUFAK.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 googleADKBundledIndexVersionRange = ">=0.6.1 <0.7.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: googleADKBundledIndexVersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 11
+    }
+  },
+  {
+    channelName: googleADKChannels.runnerRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexVersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 11
+    }
+  },
+  // --- 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: googleADKBundledIndexVersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Sync",
+      index: 0
+    }
+  },
+  {
+    channelName: googleADKChannels.agentRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexVersionRange,
+      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: googleADKBundledIndexVersionRange,
+      filePath: "dist/cjs/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Async",
+      index: 1
+    }
+  },
+  {
+    channelName: googleADKChannels.toolRunAsync.channelName,
+    module: {
+      name: "@google/adk",
+      versionRange: googleADKBundledIndexVersionRange,
+      filePath: "dist/esm/index.js"
+    },
+    functionQuery: {
+      methodName: "runAsync",
+      kind: "Async",
+      index: 1
+    }
+  }
+];
 export {
   aiSDKConfigs,
   anthropicConfigs,
   claudeAgentSDKConfigs,
+  cohereConfigs,
+  googleADKConfigs,
   googleGenAIConfigs,
+  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;