braintrust 3.25.0 → 3.26.0

package/dist/{chunk-2SANLSWX.mjs → chunk-XIZOM2HO.mjs}

@@ -1,9 +1,12 @@
 import {
+  GLOBAL_INSTRUMENTATION_HOOKS_PROTOCOL_VERSION,
   INSTRUMENTATION_NAMES,
   aiSDKChannels,
   anthropicChannels,
   bedrockRuntimeChannels,
   claudeAgentSDKChannels,
+  cloudflareAIChatChannels,
+  cloudflareAgentsChannels,
   cohereChannels,
   cursorSDKChannels,
   detectSpanOriginEnvironment,
@@ -11,6 +14,7 @@ import {
   genkitChannels,
   genkitCoreChannels,
   getDefaultInstrumentationIntegrations,
+  getHuggingFaceTransformersPipelineInfo,
   getSpanInstrumentationName,
   gitHubCopilotChannels,
   googleADKChannels,
@@ -18,8 +22,10 @@ import {
   groqChannels,
   harnessAgentChannels,
   huggingFaceChannels,
+  huggingFaceTransformersChannels,
   installMastraExporterFactory,
   isInstrumentationIntegrationDisabled,
+  isSupportedHuggingFaceTransformersTask,
   isomorph_default,
   langChainChannels,
   langSmithChannels,
@@ -30,14 +36,15 @@ import {
   openAICodexChannels,
   openRouterAgentChannels,
   openRouterChannels,
-  patchTracingChannel,
   piCodingAgentChannels,
   readDisabledInstrumentationEnvConfig,
+  registerHuggingFaceTransformersPipeline,
+  setGlobalHookErrorReporter,
   smithyClientChannels,
   smithyCoreChannels,
   strandsAgentSDKChannels,
   withSpanInstrumentationName
-} from "./chunk-ABR2QWDP.mjs";
+} from "./chunk-PN7EWK66.mjs";
 
 // src/id-gen.ts
 import { v4 as uuidv4 } from "uuid";
@@ -150,6 +157,9 @@ var debugLogger = {
     return createDebugLogger(state);
   }
 };
+setGlobalHookErrorReporter((error) => {
+  debugLogger.error("Global instrumentation hook error:", error);
+});
 
 // src/id-gen.ts
 var IDGenerator = class {
@@ -1877,7 +1887,8 @@ var ApiKey = z6.object({
   user_email: z6.union([z6.string(), z6.null()]).optional(),
   user_given_name: z6.union([z6.string(), z6.null()]).optional(),
   user_family_name: z6.union([z6.string(), z6.null()]).optional(),
-  org_id: z6.union([z6.string(), z6.null()]).optional()
+  org_id: z6.union([z6.string(), z6.null()]).optional(),
+  expires_at: z6.union([z6.string(), z6.null()]).optional()
 });
 var TriggeredFunctionState = z6.object({
   triggered_xact_id: z6.string(),
@@ -1997,6 +2008,7 @@ var TopicMapData = z6.object({
   topic_names: z6.record(z6.string()).optional(),
   generation_settings: TopicMapGenerationSettings.optional(),
   disable_reconciliation: z6.boolean().optional(),
+  reconcile_mode: z6.enum(["evolve", "names_only"]).optional(),
   distance_threshold: z6.number().optional(),
   btql_filter: z6.string().optional(),
   automation_btql_filter: z6.string().optional()
@@ -2577,7 +2589,12 @@ var ModelParams = z6.union([
   }).partial().passthrough()
 ]);
 var PromptOptionsNullish = z6.union([
-  z6.object({ model: z6.string(), params: ModelParams, position: z6.string() }).partial(),
+  z6.object({
+    model: z6.string(),
+    params: ModelParams,
+    position: z6.string(),
+    endpoint_name: z6.union([z6.string(), z6.null()])
+  }).partial(),
   z6.null()
 ]);
 var PromptParserNullish = z6.union([
@@ -2591,11 +2608,25 @@ var PromptParserNullish = z6.union([
   }),
   z6.null()
 ]);
+var PreprocessorSavedFunctionId = z6.union([
+  z6.object({
+    type: z6.literal("function"),
+    id: z6.string(),
+    version: z6.string().optional()
+  }),
+  z6.object({
+    type: z6.literal("global"),
+    name: z6.string(),
+    function_type: z6.literal("preprocessor").optional().default("preprocessor")
+  }),
+  z6.null()
+]);
 var PromptDataNullish = z6.union([
   z6.object({
     prompt: PromptBlockDataNullish,
     options: PromptOptionsNullish,
     parser: PromptParserNullish,
+    preprocessor: PreprocessorSavedFunctionId,
     tool_functions: z6.union([z6.array(SavedFunctionId), z6.null()]),
     template_format: z6.union([
       z6.enum(["mustache", "nunjucks", "none"]),
@@ -2797,6 +2828,7 @@ var PromptData = z6.object({
   prompt: PromptBlockDataNullish,
   options: PromptOptionsNullish,
   parser: PromptParserNullish,
+  preprocessor: PreprocessorSavedFunctionId,
   tool_functions: z6.union([z6.array(SavedFunctionId), z6.null()]),
   template_format: z6.union([
     z6.enum(["mustache", "nunjucks", "none"]),
@@ -2959,7 +2991,8 @@ var InvokeFunction = FunctionId.and(
     mode: StreamingMode,
     strict: z6.union([z6.boolean(), z6.null()]),
     mcp_auth: z6.record(z6.object({ oauth_token: z6.string() }).partial()),
-    overrides: z6.union([z6.object({}).partial().passthrough(), z6.null()])
+    overrides: z6.union([z6.object({}).partial().passthrough(), z6.null()]),
+    endpoint_name: z6.union([z6.string(), z6.null()])
   }).partial()
 );
 var MCPServer = z6.object({
@@ -3023,6 +3056,24 @@ var Organization = z6.object({
   created: z6.union([z6.string(), z6.null()]).optional(),
   image_rendering_mode: ImageRenderingMode.optional()
 });
+var RetentionObjectType = z6.enum([
+  "project_logs",
+  "experiment",
+  "dataset"
+]);
+var OrgAutomation = z6.object({
+  id: z6.string().uuid(),
+  org_id: z6.string().uuid(),
+  user_id: z6.union([z6.string(), z6.null()]).optional(),
+  created: z6.union([z6.string(), z6.null()]).optional(),
+  name: z6.string(),
+  description: z6.union([z6.string(), z6.null()]).optional(),
+  config: z6.object({
+    event_type: z6.literal("retention"),
+    object_type: RetentionObjectType,
+    retention_days: z6.number().int().gte(0)
+  })
+});
 var ProjectSettings = z6.union([
   z6.object({
     comparison_key: z6.union([z6.string(), z6.null()]),
@@ -3049,6 +3100,7 @@ var ProjectSettings = z6.union([
       z6.null()
     ]),
     disable_realtime_queries: z6.union([z6.boolean(), z6.null()]),
+    monitor_charts_use_metrics_start: z6.union([z6.boolean(), z6.null()]),
     default_preprocessor: NullableSavedFunctionId
   }).partial(),
   z6.null()
@@ -3063,11 +3115,6 @@ var Project = z6.object({
   user_id: z6.union([z6.string(), z6.null()]).optional(),
   settings: ProjectSettings.optional()
 });
-var RetentionObjectType = z6.enum([
-  "project_logs",
-  "experiment",
-  "dataset"
-]);
 var TopicAutomationFacetModel = z6.union([
   z6.enum(["brain-facet-latest", "brain-facet-1", "brain-facet-2"]),
   z6.null()
@@ -3178,7 +3225,7 @@ var ProjectAutomation = z6.object({
     z6.object({
       event_type: z6.literal("retention"),
       object_type: RetentionObjectType,
-      retention_days: z6.number().gte(0)
+      retention_days: z6.number().int().gte(0)
     }),
     z6.object({
       event_type: z6.literal("environment_update"),
@@ -3350,7 +3397,12 @@ var Prompt = z6.object({
   metadata: z6.union([z6.object({}).partial().passthrough(), z6.null()]).optional(),
   function_type: FunctionTypeEnumNullish.optional()
 });
-var PromptOptions = z6.object({ model: z6.string(), params: ModelParams, position: z6.string() }).partial();
+var PromptOptions = z6.object({
+  model: z6.string(),
+  params: ModelParams,
+  position: z6.string(),
+  endpoint_name: z6.union([z6.string(), z6.null()])
+}).partial();
 var PromptSessionEvent = z6.object({
   id: z6.string(),
   _xact_id: z6.string(),
@@ -3433,7 +3485,8 @@ var RunEval = z6.object({
   stop_token: z6.union([z6.string(), z6.null()]).optional(),
   extra_messages: z6.string().optional(),
   tags: z6.array(z6.string()).optional(),
-  mcp_auth: z6.record(z6.object({ oauth_token: z6.string() }).partial()).optional()
+  mcp_auth: z6.record(z6.object({ oauth_token: z6.string() }).partial()).optional(),
+  endpoint_name: z6.union([z6.string(), z6.null()]).optional()
 });
 var ServiceToken = z6.object({
   id: z6.string().uuid(),
@@ -3443,7 +3496,8 @@ var ServiceToken = z6.object({
   service_account_id: z6.union([z6.string(), z6.null()]).optional(),
   service_account_email: z6.union([z6.string(), z6.null()]).optional(),
   service_account_name: z6.union([z6.string(), z6.null()]).optional(),
-  org_id: z6.union([z6.string(), z6.null()]).optional()
+  org_id: z6.union([z6.string(), z6.null()]).optional(),
+  expires_at: z6.union([z6.string(), z6.null()]).optional()
 });
 var SpanIFrame = z6.object({
   id: z6.string().uuid(),
@@ -3565,6 +3619,13 @@ var ViewOptions = z6.union([
       }),
       z6.null()
     ]),
+    pointSizeMetric: z6.union([
+      z6.object({
+        type: z6.enum(["none", "score", "metric", "metadata"]),
+        value: z6.string()
+      }),
+      z6.null()
+    ]),
     xAxisAggregation: z6.union([z6.string(), z6.null()]),
     chartAnnotations: z6.union([
       z6.array(z6.object({ id: z6.string(), text: z6.string() })),
@@ -4847,12 +4908,13 @@ function applyMaskingToField(maskingFunction, data, fieldName) {
 var INITIAL_SPAN_WRITE_AS_MERGE = /* @__PURE__ */ Symbol(
   "braintrust.initial-span-write-as-merge"
 );
+var INTERNAL_SPAN_CONTEXT = /* @__PURE__ */ Symbol("braintrust.internal-span-context");
 var BRAINTRUST_CURRENT_SPAN_STORE = /* @__PURE__ */ Symbol.for(
   "braintrust.currentSpanStore"
 );
 var ContextManager = class {
   /**
-   * Returns the value to store in the ALS bound to a TracingChannel's start event.
+   * Returns the value to store in the ALS bound to a global hook's start event.
    * In default mode this is the Span itself; in OTEL mode it is the OTEL Context
    * containing the span so that OTEL's own ALS stores a proper Context object.
    */
@@ -8585,6 +8647,12 @@ function _internalStartSpanWithInitialMerge(args) {
     [INITIAL_SPAN_WRITE_AS_MERGE]: true
   }).span;
 }
+function _internalStartSpanWithContext(args, context) {
+  return startSpanAndIsLogger({
+    ...args,
+    [INTERNAL_SPAN_CONTEXT]: context
+  }).span;
+}
 async function flush(options) {
   const state = options?.state ?? _globalState;
   return await state.bgLogger().flush();
@@ -9486,7 +9554,7 @@ var SpanImpl = class _SpanImpl {
         start: args.startTime ?? getCurrentUnixTimestamp()
       },
       context: mergeSpanOriginContext(
-        { ...callerLocation },
+        { ...callerLocation, ...args[INTERNAL_SPAN_CONTEXT] },
         instrumentationName,
         this._state.spanOriginEnvironment
       ),
@@ -11481,8 +11549,8 @@ function runWithAutoInstrumentationSuppressed(callback) {
     callback
   );
 }
-function bindAutoInstrumentationSuppressionToStart(tracingChannel2) {
-  const startChannel = tracingChannel2.start;
+function bindAutoInstrumentationSuppressionToStart(tracingChannel) {
+  const startChannel = tracingChannel.start;
   if (!startChannel) {
     return void 0;
   }
@@ -11593,9 +11661,9 @@ function ensureSpanStateForEvent(states, config, event, channelName, instrumenta
   states.set(key, created);
   return created;
 }
-function bindCurrentSpanStoreToStart(tracingChannel2, states, config, channelName, instrumentationName) {
+function bindCurrentSpanStoreToStart(tracingChannel, states, config, channelName, instrumentationName) {
   const state = _internalGetGlobalState();
-  const startChannel = tracingChannel2.start;
+  const startChannel = tracingChannel.start;
   const contextManager = state?.contextManager;
   const currentSpanStore = contextManager ? contextManager[BRAINTRUST_CURRENT_SPAN_STORE] : void 0;
   if (!currentSpanStore || !startChannel) {
@@ -11678,11 +11746,11 @@ function runStreamingErrorHook(args) {
   }
 }
 function traceAsyncChannel(channel, config) {
-  const tracingChannel2 = channel.tracingChannel();
+  const tracingChannel = channel.tracingChannel();
   const states = /* @__PURE__ */ new WeakMap();
   const channelName = channel.channelName;
   const unbindCurrentSpanStore = bindCurrentSpanStoreToStart(
-    tracingChannel2,
+    tracingChannel,
     states,
     config,
     channelName,
@@ -11738,18 +11806,18 @@ function traceAsyncChannel(channel, config) {
       logErrorAndEnd(states, event, channelName);
     }
   };
-  tracingChannel2.subscribe(handlers);
+  tracingChannel.subscribe(handlers);
   return () => {
     unbindCurrentSpanStore?.();
-    tracingChannel2.unsubscribe(handlers);
+    tracingChannel.unsubscribe(handlers);
   };
 }
 function traceStreamingChannel(channel, config) {
-  const tracingChannel2 = channel.tracingChannel();
+  const tracingChannel = channel.tracingChannel();
   const states = /* @__PURE__ */ new WeakMap();
   const channelName = channel.channelName;
   const unbindCurrentSpanStore = bindCurrentSpanStoreToStart(
-    tracingChannel2,
+    tracingChannel,
     states,
     config,
     channelName,
@@ -11960,18 +12028,18 @@ function traceStreamingChannel(channel, config) {
       }
     }
   };
-  tracingChannel2.subscribe(handlers);
+  tracingChannel.subscribe(handlers);
   return () => {
     unbindCurrentSpanStore?.();
-    tracingChannel2.unsubscribe(handlers);
+    tracingChannel.unsubscribe(handlers);
   };
 }
 function traceSyncStreamChannel(channel, config) {
-  const tracingChannel2 = channel.tracingChannel();
+  const tracingChannel = channel.tracingChannel();
   const states = /* @__PURE__ */ new WeakMap();
   const channelName = channel.channelName;
   const unbindCurrentSpanStore = bindCurrentSpanStoreToStart(
-    tracingChannel2,
+    tracingChannel,
     states,
     config,
     channelName,
@@ -12084,10 +12152,10 @@ function traceSyncStreamChannel(channel, config) {
       logErrorAndEnd(states, event, channelName);
     }
   };
-  tracingChannel2.subscribe(handlers);
+  tracingChannel.subscribe(handlers);
   return () => {
     unbindCurrentSpanStore?.();
-    tracingChannel2.unsubscribe(handlers);
+    tracingChannel.unsubscribe(handlers);
   };
 }
 function unsubscribeAll(unsubscribers) {
@@ -12560,8 +12628,8 @@ function registerHarnessSessionParent(session, parent) {
 function currentHarnessTurnParent() {
   return harnessTurnParentStore?.getStore() ?? wrapperTurnParents.get(currentSpan());
 }
-function bindHarnessTurnParentToStart(tracingChannel2, parentFromEvent) {
-  const startChannel = tracingChannel2.start;
+function bindHarnessTurnParentToStart(tracingChannel, parentFromEvent) {
+  const startChannel = tracingChannel.start;
   if (!startChannel) {
     return () => {
     };
@@ -19024,7 +19092,7 @@ var AnthropicPlugin = class extends BasePlugin {
     );
   }
   subscribeToAnthropicToolRunner() {
-    const tracingChannel2 = anthropicChannels.betaMessagesToolRunner.tracingChannel();
+    const tracingChannel = anthropicChannels.betaMessagesToolRunner.tracingChannel();
     const states = /* @__PURE__ */ new WeakMap();
     const handlers = {
       start: (event) => {
@@ -19081,9 +19149,9 @@ var AnthropicPlugin = class extends BasePlugin {
         states.delete(event);
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
 };
@@ -21168,7 +21236,7 @@ var CursorSDKPlugin = class extends BasePlugin {
     this.subscribeToAgentFactory(cursorSDKChannels.resume);
   }
   subscribeToAgentFactory(channel) {
-    const tracingChannel2 = channel.tracingChannel();
+    const tracingChannel = channel.tracingChannel();
     const handlers = {
       asyncEnd: (event) => {
         patchCursorAgentInPlace(event.result);
@@ -21176,9 +21244,9 @@ var CursorSDKPlugin = class extends BasePlugin {
       error: () => {
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
   subscribeToPrompt() {
@@ -22486,10 +22554,10 @@ var GoogleGenAIPlugin = class extends BasePlugin {
     this.subscribeToInteractionsCreateChannel();
   }
   subscribeToGenerateContentChannel() {
-    const tracingChannel2 = googleGenAIChannels.generateContent.tracingChannel();
+    const tracingChannel = googleGenAIChannels.generateContent.tracingChannel();
     const states = /* @__PURE__ */ new WeakMap();
     const unbindCurrentSpanStore = bindCurrentSpanStoreToStart2(
-      tracingChannel2,
+      tracingChannel,
       states,
       (event) => {
         const params = event.arguments[0];
@@ -22563,14 +22631,14 @@ var GoogleGenAIPlugin = class extends BasePlugin {
         logErrorAndEndSpan(states, event);
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
       unbindCurrentSpanStore?.();
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
   subscribeToGenerateContentStreamChannel() {
-    const tracingChannel2 = googleGenAIChannels.generateContentStream.tracingChannel();
+    const tracingChannel = googleGenAIChannels.generateContentStream.tracingChannel();
     const handlers = {
       start: (event) => {
         const streamEvent = event;
@@ -22591,16 +22659,16 @@ var GoogleGenAIPlugin = class extends BasePlugin {
       error: () => {
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
   subscribeToEmbedContentChannel() {
-    const tracingChannel2 = googleGenAIChannels.embedContent.tracingChannel();
+    const tracingChannel = googleGenAIChannels.embedContent.tracingChannel();
     const states = /* @__PURE__ */ new WeakMap();
     const unbindCurrentSpanStore = bindCurrentSpanStoreToStart2(
-      tracingChannel2,
+      tracingChannel,
       states,
       (event) => {
         const params = event.arguments[0];
@@ -22670,10 +22738,10 @@ var GoogleGenAIPlugin = class extends BasePlugin {
         logErrorAndEndSpan(states, event);
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
       unbindCurrentSpanStore?.();
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
   subscribeToInteractionsCreateChannel() {
@@ -22709,10 +22777,10 @@ function ensureSpanState(states, event, create) {
   states.set(event, created);
   return created;
 }
-function bindCurrentSpanStoreToStart2(tracingChannel2, states, create) {
+function bindCurrentSpanStoreToStart2(tracingChannel, states, create) {
   const state = _internalGetGlobalState();
   const contextManager = state?.contextManager;
-  const startChannel = tracingChannel2.start;
+  const startChannel = tracingChannel.start;
   const currentSpanStore = contextManager ? contextManager[BRAINTRUST_CURRENT_SPAN_STORE] : void 0;
   if (!startChannel?.bindStore || !currentSpanStore) {
     return void 0;
@@ -23215,11 +23283,11 @@ function summarizeEmbedContentOutput(response) {
   };
 }
 function populateUsageMetrics(metrics, usage) {
-  if (usage.promptTokenCount !== void 0) {
-    metrics.prompt_tokens = usage.promptTokenCount;
+  if (usage.promptTokenCount !== void 0 || usage.toolUsePromptTokenCount !== void 0) {
+    metrics.prompt_tokens = (usage.promptTokenCount ?? 0) + (usage.toolUsePromptTokenCount ?? 0);
   }
-  if (usage.candidatesTokenCount !== void 0) {
-    metrics.completion_tokens = usage.candidatesTokenCount;
+  if (usage.candidatesTokenCount !== void 0 || usage.thoughtsTokenCount !== void 0) {
+    metrics.completion_tokens = (usage.candidatesTokenCount ?? 0) + (usage.thoughtsTokenCount ?? 0);
   }
   if (usage.totalTokenCount !== void 0) {
     metrics.tokens = usage.totalTokenCount;
@@ -23230,13 +23298,26 @@ function populateUsageMetrics(metrics, usage) {
   if (usage.thoughtsTokenCount !== void 0) {
     metrics.completion_reasoning_tokens = usage.thoughtsTokenCount;
   }
+  for (const detail of usage.promptTokensDetails ?? []) {
+    if (detail.modality === "AUDIO" && detail.tokenCount !== void 0) {
+      metrics.prompt_audio_tokens = (metrics.prompt_audio_tokens ?? 0) + detail.tokenCount;
+    }
+  }
+  for (const detail of usage.candidatesTokensDetails ?? []) {
+    if (detail.modality === "AUDIO" && detail.tokenCount !== void 0) {
+      metrics.completion_audio_tokens = (metrics.completion_audio_tokens ?? 0) + detail.tokenCount;
+    }
+    if (detail.modality === "IMAGE" && detail.tokenCount !== void 0) {
+      metrics.completion_image_tokens = (metrics.completion_image_tokens ?? 0) + detail.tokenCount;
+    }
+  }
 }
 function populateInteractionUsageMetrics(metrics, usage) {
   if (typeof usage.total_input_tokens === "number") {
     metrics.prompt_tokens = usage.total_input_tokens;
   }
   if (typeof usage.total_output_tokens === "number") {
-    metrics.completion_tokens = usage.total_output_tokens;
+    metrics.completion_tokens = usage.total_output_tokens + (typeof usage.total_thought_tokens === "number" ? usage.total_thought_tokens : 0);
   }
   if (typeof usage.total_tokens === "number") {
     metrics.tokens = usage.total_tokens;
@@ -23247,6 +23328,24 @@ function populateInteractionUsageMetrics(metrics, usage) {
   if (typeof usage.total_thought_tokens === "number") {
     metrics.completion_reasoning_tokens = usage.total_thought_tokens;
   }
+  for (const detail of Array.isArray(usage.input_tokens_by_modality) ? usage.input_tokens_by_modality : []) {
+    if (typeof detail?.modality === "string" && detail.modality.toLowerCase() === "audio" && typeof detail.tokens === "number") {
+      metrics.prompt_audio_tokens = (metrics.prompt_audio_tokens ?? 0) + detail.tokens;
+    }
+  }
+  for (const detail of Array.isArray(usage.output_tokens_by_modality) ? usage.output_tokens_by_modality : []) {
+    if (typeof detail?.modality !== "string" || typeof detail.tokens !== "number") {
+      continue;
+    }
+    switch (detail.modality.toLowerCase()) {
+      case "audio":
+        metrics.completion_audio_tokens = (metrics.completion_audio_tokens ?? 0) + detail.tokens;
+        break;
+      case "image":
+        metrics.completion_image_tokens = (metrics.completion_image_tokens ?? 0) + detail.tokens;
+        break;
+    }
+  }
 }
 function aggregateGenerateContentChunks(chunks, startTime, firstTokenTime) {
   const end = getCurrentUnixTimestamp();
@@ -23957,6 +24056,259 @@ function extractTextGenerationStreamMetadata(chunks) {
   return void 0;
 }
 
+// src/instrumentation/plugins/huggingface-transformers-plugin.ts
+var REQUEST_METADATA_KEYS = [
+  "temperature",
+  "top_p",
+  "max_tokens",
+  "stop"
+];
+var HuggingFaceTransformersPlugin = class extends BasePlugin {
+  onEnable() {
+    this.subscribeToPipelineFactory();
+    this.unsubscribers.push(
+      traceAsyncChannel(huggingFaceTransformersChannels.pipelineCall, {
+        name: (_args, event) => {
+          const task = getTask(event);
+          const operation = task?.replaceAll("-", "_") ?? "unknown";
+          return `huggingface.transformers.${operation}`;
+        },
+        type: "llm" /* LLM */,
+        shouldTrace: (_args, event) => isSupportedHuggingFaceTransformersTask(
+          getTask(event)
+        ),
+        extractInput: (args, event) => ({
+          input: extractInput(
+            getTask(event),
+            args
+          ),
+          metadata: extractMetadata(
+            event,
+            args
+          )
+        }),
+        extractOutput: (result, event) => extractOutput(
+          getTask(event),
+          result
+        ),
+        extractMetrics: () => ({})
+      })
+    );
+  }
+  onDisable() {
+    this.unsubscribers = unsubscribeAll(this.unsubscribers);
+  }
+  subscribeToPipelineFactory() {
+    const channel = huggingFaceTransformersChannels.pipeline.tracingChannel();
+    const handlers = {
+      asyncEnd: (event) => {
+        if (typeof event.result !== "function") {
+          return;
+        }
+        registerHuggingFaceTransformersPipeline(
+          event.result,
+          event.arguments?.[0],
+          event.arguments?.[1]
+        );
+      }
+    };
+    channel.subscribe(handlers);
+    this.unsubscribers.push(() => channel.unsubscribe(handlers));
+  }
+};
+function getTask(event) {
+  const self = event.self;
+  const registeredTask = getHuggingFaceTransformersPipelineInfo(self)?.task;
+  if (registeredTask !== void 0) {
+    return registeredTask;
+  }
+  if (typeof self?.task === "string") {
+    return self.task;
+  }
+  return void 0;
+}
+function extractMetadata(event, args) {
+  const metadata = {
+    provider: "huggingface"
+  };
+  const registeredModel = getHuggingFaceTransformersPipelineInfo(
+    event.self
+  )?.model;
+  const model = registeredModel ?? modelIdentifier(event.self);
+  if (model) {
+    metadata.model = model;
+  }
+  const task = getTask(event);
+  const options = task === "question-answering" ? args[2] : args[1];
+  if (isObject(options)) {
+    for (const key of REQUEST_METADATA_KEYS) {
+      if (options[key] !== void 0) {
+        metadata[key] = options[key];
+      }
+    }
+  }
+  return metadata;
+}
+function modelIdentifier(pipeline) {
+  const model = pipeline?.model;
+  if (!isObject(model)) {
+    return void 0;
+  }
+  const config = isObject(model.config) ? model.config : void 0;
+  for (const value of [
+    config?._name_or_path,
+    config?.name_or_path,
+    config?.model_id,
+    config?.modelId,
+    model.name
+  ]) {
+    if (typeof value === "string" && value.length > 0) {
+      return value;
+    }
+  }
+  return void 0;
+}
+function extractInput(task, args) {
+  switch (task) {
+    case "feature-extraction":
+      return args[0];
+    case "question-answering": {
+      const question = args[0];
+      const context = args[1];
+      if (Array.isArray(question) && Array.isArray(context) && question.every((value) => typeof value === "string") && context.every((value) => typeof value === "string")) {
+        return question.map((value, index) => [
+          {
+            role: "user",
+            content: `Context:
+${context[index] ?? ""}
+
+Question:
+${value}`
+          }
+        ]);
+      }
+      if (typeof question === "string" && typeof context === "string") {
+        return [
+          {
+            role: "user",
+            content: `Context:
+${context}
+
+Question:
+${question}`
+          }
+        ];
+      }
+      return { context, question };
+    }
+  }
+  const input = args[0];
+  if (task === "text-generation" && isChat(input)) {
+    return input;
+  }
+  if (task === "text-generation" && Array.isArray(input) && input.every(isChat)) {
+    return input;
+  }
+  if (typeof input === "string") {
+    return [{ role: "user", content: input }];
+  }
+  if (Array.isArray(input) && input.every((value) => typeof value === "string")) {
+    return input.map((value) => [
+      {
+        role: "user",
+        content: value
+      }
+    ]);
+  }
+  return input;
+}
+function isChat(value) {
+  return Array.isArray(value) && value.length > 0 && value.every(
+    (message) => isObject(message) && typeof message.role === "string" && "content" in message
+  );
+}
+function extractOutput(task, result) {
+  switch (task) {
+    case "feature-extraction":
+      return summarizeEmbedding(result);
+    case "question-answering":
+      return choicesFromAnswers(result);
+    default:
+      return choicesFromGenerations(result);
+  }
+}
+function summarizeEmbedding(result) {
+  if (!isObject(result) || !Array.isArray(result.dims)) {
+    return void 0;
+  }
+  const { dims } = result;
+  if (dims.length === 0 || !dims.every((dimension) => typeof dimension === "number")) {
+    return void 0;
+  }
+  if (dims.length === 1) {
+    return { embedding_length: dims[0] };
+  }
+  if (dims.length === 2) {
+    return {
+      embedding_count: dims[0],
+      embedding_length: dims[1]
+    };
+  }
+  return {
+    embedding_batch_count: dims[0],
+    embedding_count: dims[1],
+    embedding_length: dims.at(-1) ?? 0
+  };
+}
+function choicesFromAnswers(result) {
+  const answers = Array.isArray(result) ? result.flat() : [result];
+  const choices = answers.flatMap((answer, index) => {
+    if (!isObject(answer) || typeof answer.answer !== "string") {
+      return [];
+    }
+    return [
+      {
+        index,
+        finish_reason: "stop",
+        message: {
+          role: "assistant",
+          content: answer.answer
+        }
+      }
+    ];
+  });
+  return choices.length > 0 ? choices : void 0;
+}
+function choicesFromGenerations(result) {
+  const generations = Array.isArray(result) ? result.flat() : [result];
+  const choices = generations.flatMap((generation, index) => {
+    if (!isObject(generation)) {
+      return [];
+    }
+    const generated = generation.generated_text ?? generation.summary_text;
+    let content;
+    if (typeof generated === "string") {
+      content = generated;
+    } else if (isChat(generated)) {
+      content = generated.at(-1)?.content;
+    }
+    if (typeof content !== "string") {
+      return [];
+    }
+    return [
+      {
+        index,
+        finish_reason: "stop",
+        message: {
+          role: "assistant",
+          content
+        }
+      }
+    ];
+  });
+  return choices.length > 0 ? choices : void 0;
+}
+
 // src/instrumentation/plugins/openrouter-agent-plugin.ts
 var OpenRouterAgentPlugin = class extends BasePlugin {
   onEnable() {
@@ -24336,7 +24688,7 @@ function isWrappedTool(tool) {
   return Boolean(tool[OPENROUTER_WRAPPED_TOOL]);
 }
 function traceToolExecution(args) {
-  const tracingChannel2 = openRouterAgentChannels.toolExecute.tracingChannel();
+  const tracingChannel = openRouterAgentChannels.toolExecute.tracingChannel();
   const input = args.args.length > 0 ? args.args[0] : void 0;
   const event = {
     arguments: [input],
@@ -24346,33 +24698,33 @@ function traceToolExecution(args) {
     toolCallId: args.toolCallId,
     toolName: args.toolName
   };
-  tracingChannel2.start.publish(event);
+  tracingChannel.start.publish(event);
   try {
     const result = args.execute();
-    return publishToolResult(tracingChannel2, event, result);
+    return publishToolResult(tracingChannel, event, result);
   } catch (error) {
     event.error = normalizeError(error);
-    tracingChannel2.error.publish(event);
+    tracingChannel.error.publish(event);
     throw error;
   }
 }
-function publishToolResult(tracingChannel2, event, result) {
+function publishToolResult(tracingChannel, event, result) {
   if (isPromiseLike(result)) {
     return result.then(
       (resolved) => {
         event.result = resolved;
-        tracingChannel2.asyncEnd.publish(event);
+        tracingChannel.asyncEnd.publish(event);
         return resolved;
       },
       (error) => {
         event.error = normalizeError(error);
-        tracingChannel2.error.publish(event);
+        tracingChannel.error.publish(event);
         throw error;
       }
     );
   }
   event.result = result;
-  tracingChannel2.asyncEnd.publish(event);
+  tracingChannel.asyncEnd.publish(event);
   return result;
 }
 function getToolCallId(context) {
@@ -25242,7 +25594,7 @@ function isWrappedTool2(tool) {
   return Boolean(tool[OPENROUTER_WRAPPED_TOOL2]);
 }
 function traceToolExecution2(args) {
-  const tracingChannel2 = openRouterChannels.toolExecute.tracingChannel();
+  const tracingChannel = openRouterChannels.toolExecute.tracingChannel();
   const input = args.args.length > 0 ? args.args[0] : void 0;
   const event = {
     arguments: [input],
@@ -25252,33 +25604,33 @@ function traceToolExecution2(args) {
     toolCallId: args.toolCallId,
     toolName: args.toolName
   };
-  tracingChannel2.start.publish(event);
+  tracingChannel.start.publish(event);
   try {
     const result = args.execute();
-    return publishToolResult2(tracingChannel2, event, result);
+    return publishToolResult2(tracingChannel, event, result);
   } catch (error) {
     event.error = normalizeError2(error);
-    tracingChannel2.error.publish(event);
+    tracingChannel.error.publish(event);
     throw error;
   }
 }
-function publishToolResult2(tracingChannel2, event, result) {
+function publishToolResult2(tracingChannel, event, result) {
   if (isPromiseLike(result)) {
     return result.then(
       (resolved) => {
         event.result = resolved;
-        tracingChannel2.asyncEnd.publish(event);
+        tracingChannel.asyncEnd.publish(event);
         return resolved;
       },
       (error) => {
         event.error = normalizeError2(error);
-        tracingChannel2.error.publish(event);
+        tracingChannel.error.publish(event);
         throw error;
       }
     );
   }
   event.result = result;
-  tracingChannel2.asyncEnd.publish(event);
+  tracingChannel.asyncEnd.publish(event);
   return result;
 }
 function getToolCallId2(context) {
@@ -26396,7 +26748,7 @@ var GoogleADKPlugin = class extends BasePlugin {
     this.activeAgentSpans.clear();
   }
   subscribeToRunnerRunAsync() {
-    const tracingChannel2 = googleADKChannels.runnerRunAsync.tracingChannel();
+    const tracingChannel = googleADKChannels.runnerRunAsync.tracingChannel();
     const states = /* @__PURE__ */ new WeakMap();
     const createState = (event) => {
       const params = event.arguments[0] ?? {};
@@ -26427,7 +26779,7 @@ var GoogleADKPlugin = class extends BasePlugin {
       return { span, startTime, events: [], contextKey };
     };
     const unbindCurrentSpanStore = bindCurrentSpanStoreToStart3(
-      tracingChannel2,
+      tracingChannel,
       states,
       createState
     );
@@ -26479,14 +26831,14 @@ var GoogleADKPlugin = class extends BasePlugin {
         states.delete(event);
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
       unbindCurrentSpanStore?.();
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
   subscribeToAgentRunAsync() {
-    const tracingChannel2 = googleADKChannels.agentRunAsync.tracingChannel();
+    const tracingChannel = googleADKChannels.agentRunAsync.tracingChannel();
     const states = /* @__PURE__ */ new WeakMap();
     const createState = (event) => {
       const parentContext = event.arguments[0];
@@ -26535,7 +26887,7 @@ var GoogleADKPlugin = class extends BasePlugin {
       return { span, startTime, events: [], contextKey, name: agentName };
     };
     const unbindCurrentSpanStore = bindCurrentSpanStoreToStart3(
-      tracingChannel2,
+      tracingChannel,
       states,
       createState
     );
@@ -26587,14 +26939,14 @@ var GoogleADKPlugin = class extends BasePlugin {
         states.delete(event);
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
       unbindCurrentSpanStore?.();
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
   subscribeToToolRunAsync() {
-    const tracingChannel2 = googleADKChannels.toolRunAsync.tracingChannel();
+    const tracingChannel = googleADKChannels.toolRunAsync.tracingChannel();
     const states = /* @__PURE__ */ new WeakMap();
     const handlers = {
       start: (event) => {
@@ -26661,9 +27013,9 @@ var GoogleADKPlugin = class extends BasePlugin {
         states.delete(event);
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
 };
@@ -26739,10 +27091,10 @@ function bindAsyncIterableToCurrentSpan(stream, span) {
   }
   return stream;
 }
-function bindCurrentSpanStoreToStart3(tracingChannel2, states, create) {
+function bindCurrentSpanStoreToStart3(tracingChannel, states, create) {
   const state = _internalGetGlobalState();
   const contextManager = state?.contextManager;
-  const startChannel = tracingChannel2.start;
+  const startChannel = tracingChannel.start;
   const currentSpanStore = contextManager ? contextManager[BRAINTRUST_CURRENT_SPAN_STORE] : void 0;
   if (!startChannel?.bindStore || !currentSpanStore) {
     return void 0;
@@ -26994,15 +27346,32 @@ function aggregateUsageFromEvents(events) {
     if (usage.thoughtsTokenCount !== void 0) {
       aggregated.thoughtsTokenCount = (aggregated.thoughtsTokenCount ?? 0) + usage.thoughtsTokenCount;
     }
+    if (usage.toolUsePromptTokenCount !== void 0) {
+      aggregated.toolUsePromptTokenCount = (aggregated.toolUsePromptTokenCount ?? 0) + usage.toolUsePromptTokenCount;
+    }
+    for (const detailsField of [
+      "promptTokensDetails",
+      "cacheTokensDetails",
+      "candidatesTokensDetails",
+      "toolUsePromptTokensDetails"
+    ]) {
+      const details = usage[detailsField];
+      if (details !== void 0) {
+        aggregated[detailsField] = [
+          ...aggregated[detailsField] ?? [],
+          ...details
+        ];
+      }
+    }
   }
   return hasUsage ? aggregated : void 0;
 }
 function populateUsageMetrics2(metrics, usage) {
-  if (usage.promptTokenCount !== void 0) {
-    metrics.prompt_tokens = usage.promptTokenCount;
+  if (usage.promptTokenCount !== void 0 || usage.toolUsePromptTokenCount !== void 0) {
+    metrics.prompt_tokens = (usage.promptTokenCount ?? 0) + (usage.toolUsePromptTokenCount ?? 0);
   }
-  if (usage.candidatesTokenCount !== void 0) {
-    metrics.completion_tokens = usage.candidatesTokenCount;
+  if (usage.candidatesTokenCount !== void 0 || usage.thoughtsTokenCount !== void 0) {
+    metrics.completion_tokens = (usage.candidatesTokenCount ?? 0) + (usage.thoughtsTokenCount ?? 0);
   }
   if (usage.totalTokenCount !== void 0) {
     metrics.tokens = usage.totalTokenCount;
@@ -27013,6 +27382,19 @@ function populateUsageMetrics2(metrics, usage) {
   if (usage.thoughtsTokenCount !== void 0) {
     metrics.completion_reasoning_tokens = usage.thoughtsTokenCount;
   }
+  for (const detail of usage.promptTokensDetails ?? []) {
+    if (detail.modality === "AUDIO" && detail.tokenCount !== void 0) {
+      metrics.prompt_audio_tokens = (metrics.prompt_audio_tokens ?? 0) + detail.tokenCount;
+    }
+  }
+  for (const detail of usage.candidatesTokensDetails ?? []) {
+    if (detail.modality === "AUDIO" && detail.tokenCount !== void 0) {
+      metrics.completion_audio_tokens = (metrics.completion_audio_tokens ?? 0) + detail.tokenCount;
+    }
+    if (detail.modality === "IMAGE" && detail.tokenCount !== void 0) {
+      metrics.completion_image_tokens = (metrics.completion_image_tokens ?? 0) + detail.tokenCount;
+    }
+  }
 }
 function cleanMetrics4(metrics) {
   const cleaned = {};
@@ -28469,10 +28851,10 @@ var GenkitPlugin = class extends BasePlugin {
     this.subscribeToActionStream();
   }
   subscribeToActionRun() {
-    const tracingChannel2 = genkitChannels.actionRun.tracingChannel();
+    const tracingChannel = genkitChannels.actionRun.tracingChannel();
     const states = /* @__PURE__ */ new WeakMap();
     const unbindCurrentSpanStore = bindActionCurrentSpanStoreToStart(
-      tracingChannel2,
+      tracingChannel,
       states,
       (event) => startActionRunSpan(event)
     );
@@ -28509,17 +28891,17 @@ var GenkitPlugin = class extends BasePlugin {
         states.delete(event);
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
       unbindCurrentSpanStore?.();
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
   subscribeToActionSpan() {
-    const tracingChannel2 = genkitCoreChannels.actionSpan.tracingChannel();
+    const tracingChannel = genkitCoreChannels.actionSpan.tracingChannel();
     const states = /* @__PURE__ */ new WeakMap();
     const unbindCurrentSpanStore = bindActionCurrentSpanStoreToStart(
-      tracingChannel2,
+      tracingChannel,
       states,
       (event) => startActionSpan(event)
     );
@@ -28557,10 +28939,10 @@ var GenkitPlugin = class extends BasePlugin {
         states.delete(event);
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
       unbindCurrentSpanStore?.();
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
   subscribeToActionStream() {
@@ -28629,10 +29011,10 @@ function ensureActionSpanState(states, event, create) {
   }
   return created;
 }
-function bindActionCurrentSpanStoreToStart(tracingChannel2, states, create) {
+function bindActionCurrentSpanStoreToStart(tracingChannel, states, create) {
   const state = _internalGetGlobalState();
   const contextManager = state?.contextManager;
-  const startChannel = tracingChannel2.start;
+  const startChannel = tracingChannel.start;
   const currentSpanStore = contextManager ? contextManager[BRAINTRUST_CURRENT_SPAN_STORE] : void 0;
   if (!startChannel?.bindStore || !currentSpanStore) {
     return void 0;
@@ -29524,7 +29906,7 @@ var LangChainPlugin = class extends BasePlugin {
     this.injectedManagers = /* @__PURE__ */ new WeakSet();
   }
   subscribeToConfigure(channel) {
-    const tracingChannel2 = channel.tracingChannel();
+    const tracingChannel = channel.tracingChannel();
     const handlers = {
       start: (event) => {
         injectHandlerIntoArguments(event.arguments);
@@ -29533,9 +29915,9 @@ var LangChainPlugin = class extends BasePlugin {
         this.injectHandler(event.result);
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
   injectHandler(result) {
@@ -29779,7 +30161,7 @@ var LangSmithPlugin = class extends BasePlugin {
     const inputs = preferOwnValue(run, previous, "inputs");
     const outputs = preferOwnValue(run, previous, "outputs");
     const error = errorMessage(preferOwnValue(run, previous, "error"));
-    const metadata = extractMetadata(run, previous);
+    const metadata = extractMetadata2(run, previous);
     const tags = extractTags(preferOwnValue(run, previous, "tags"));
     const metrics = extractMetrics(run, previous);
     span.log({
@@ -29884,7 +30266,7 @@ function mapRunType(runType) {
       return "task" /* TASK */;
   }
 }
-function extractMetadata(run, previous) {
+function extractMetadata2(run, previous) {
   const extra = preferOwnValue(run, previous, "extra");
   const rawMetadata = ownValue(extra, "metadata");
   if (!isRecord2(rawMetadata)) {
@@ -30983,9 +31365,9 @@ var StrandsAgentSDKPlugin = class extends BasePlugin {
     });
   }
   subscribeToMultiAgentStream(channel, operation) {
-    const tracingChannel2 = channel.tracingChannel();
+    const tracingChannel = channel.tracingChannel();
     const states = /* @__PURE__ */ new WeakMap();
-    const unbindAutoInstrumentationSuppression = bindAutoInstrumentationSuppressionToStart(tracingChannel2);
+    const unbindAutoInstrumentationSuppression = bindAutoInstrumentationSuppressionToStart(tracingChannel);
     const handlers = {
       start: (event) => {
         const state = startMultiAgentStream(
@@ -31039,10 +31421,10 @@ var StrandsAgentSDKPlugin = class extends BasePlugin {
         states.delete(event);
       }
     };
-    tracingChannel2.subscribe(handlers);
+    tracingChannel.subscribe(handlers);
     this.unsubscribers.push(() => {
       unbindAutoInstrumentationSuppression?.();
-      tracingChannel2.unsubscribe(handlers);
+      tracingChannel.unsubscribe(handlers);
     });
   }
 };
@@ -31225,6 +31607,15 @@ function handleMultiAgentStreamEvent(state, event, activeChildParents) {
     logInstrumentationError5("Strands Agent SDK multi-agent event", error);
   }
 }
+function buildModelSpanInput(event, attachmentCache) {
+  const systemPrompt = typeof event.agent?.systemPrompt === "string" && event.agent.systemPrompt.length > 0 ? event.agent.systemPrompt : void 0;
+  const processedMessages = Array.isArray(event.agent?.messages) ? processStrandsInputAttachments(event.agent.messages, attachmentCache) : void 0;
+  if (!systemPrompt) {
+    return processedMessages;
+  }
+  const systemMessage = { role: "system", content: systemPrompt };
+  return Array.isArray(processedMessages) ? [systemMessage, ...processedMessages] : [systemMessage];
+}
 function startModelSpan(state, event) {
   if (state.activeModel) {
     finalizeModelSpan(state);
@@ -31242,10 +31633,7 @@ function startModelSpan(state, event) {
       withSpanInstrumentationName(
         {
           event: {
-            input: Array.isArray(event.agent?.messages) ? processStrandsInputAttachments(
-              event.agent.messages,
-              state.attachmentCache
-            ) : void 0,
+            input: buildModelSpanInput(event, state.attachmentCache),
             metadata
           },
           name: formatModelSpanName(model),
@@ -31885,6 +32273,527 @@ function logInstrumentationError5(context, error) {
   debugLogger.debug(`${context}:`, error);
 }
 
+// src/instrumentation/plugins/cloudflare-ai-chat-instrumentation.ts
+var wrappedTurnRunners = /* @__PURE__ */ new WeakSet();
+var wrappedResponseHooks = /* @__PURE__ */ new WeakSet();
+function instrumentCloudflareAIChatAgent(agent) {
+  try {
+    const original = Reflect.get(agent, "_runExclusiveChatTurn");
+    if (typeof original !== "function" || wrappedTurnRunners.has(original)) {
+      return agent;
+    }
+    const wrapped = async function(...args) {
+      return cloudflareAIChatChannels.runExclusiveChatTurn.tracePromise(
+        () => Reflect.apply(original, this, args),
+        {
+          arguments: args,
+          self: this
+        }
+      );
+    };
+    wrappedTurnRunners.add(wrapped);
+    defineWrappedMethod(agent, "_runExclusiveChatTurn", original, wrapped);
+  } catch (error) {
+    debugLogger.debug("Failed to wrap @cloudflare/ai-chat turn runner:", error);
+  }
+  return agent;
+}
+function instrumentCloudflareAIChatResponseHook(agent) {
+  try {
+    const original = Reflect.get(agent, "onChatResponse");
+    if (typeof original !== "function" || wrappedResponseHooks.has(original)) {
+      return;
+    }
+    const wrapped = function(result) {
+      const args = [result];
+      return cloudflareAIChatChannels.onChatResponse.traceSync(
+        () => Reflect.apply(original, this, args),
+        {
+          arguments: args,
+          self: this
+        }
+      );
+    };
+    wrappedResponseHooks.add(wrapped);
+    defineWrappedMethod(agent, "onChatResponse", original, wrapped);
+  } catch (error) {
+    debugLogger.debug(
+      "Failed to wrap @cloudflare/ai-chat response hook:",
+      error
+    );
+  }
+}
+function defineWrappedMethod(target, property, original, wrapped) {
+  const ownDescriptor = Object.getOwnPropertyDescriptor(target, property);
+  Object.defineProperty(target, property, {
+    configurable: ownDescriptor?.configurable ?? true,
+    enumerable: ownDescriptor?.enumerable ?? false,
+    value: wrapped,
+    writable: ownDescriptor && "writable" in ownDescriptor ? ownDescriptor.writable : true
+  });
+  try {
+    Object.defineProperty(wrapped, "name", {
+      configurable: true,
+      value: original.name
+    });
+  } catch {
+  }
+}
+
+// src/instrumentation/plugins/cloudflare-ai-chat-plugin.ts
+var BLOCKED_KEYS = /* @__PURE__ */ new Set(["__proto__", "constructor", "prototype"]);
+var RESPONSE_STATE_RETENTION_MS = 6e4;
+var CloudflareAIChatPlugin = class extends BasePlugin {
+  activeStates = /* @__PURE__ */ new Set();
+  activeTurns = /* @__PURE__ */ new WeakMap();
+  eventStates = /* @__PURE__ */ new WeakMap();
+  onEnable() {
+    this.subscribeToResponseHook();
+    this.subscribeToTurnRunner();
+  }
+  onDisable() {
+    for (const unsubscribe of this.unsubscribers) {
+      unsubscribe();
+    }
+    this.unsubscribers = [];
+    for (const state of this.activeStates) {
+      this.cleanupState(state);
+    }
+    this.activeStates.clear();
+  }
+  subscribeToTurnRunner() {
+    const tracingChannel = cloudflareAIChatChannels.runExclusiveChatTurn.tracingChannel();
+    const unbindCurrentSpanStore = this.bindCurrentSpanStore(tracingChannel);
+    const handlers = {
+      start: (event) => {
+        this.ensureEventState(event);
+      },
+      asyncEnd: (event) => {
+        this.finishEvent(event);
+      },
+      error: (event) => {
+        this.finishEvent(event, event.error);
+      }
+    };
+    tracingChannel.subscribe(handlers);
+    this.unsubscribers.push(() => {
+      unbindCurrentSpanStore?.();
+      tracingChannel.unsubscribe(handlers);
+    });
+  }
+  subscribeToResponseHook() {
+    const tracingChannel = cloudflareAIChatChannels.onChatResponse.tracingChannel();
+    const handlers = {
+      start: (event) => {
+        let state;
+        try {
+          const agent = asObject(event.self);
+          const result = event.arguments[0];
+          state = agent ? this.findResponseState(
+            agent,
+            stringValue3(ownValue2(result, "requestId"))
+          ) : void 0;
+          if (!state) {
+            return;
+          }
+          state.responseObserved = true;
+          const output = serializeMessage(ownValue2(result, "message"));
+          const status = stringValue3(ownValue2(result, "status"));
+          const error = ownValue2(result, "error");
+          const input = ownValue2(result, "continuation") === true ? state.input : serializeMessages(readProperty(agent, "messages"))?.filter(
+            (message) => typeof output?.id !== "string" || message.id !== output.id
+          );
+          state.span.log({
+            ...input !== void 0 ? { input } : {},
+            ...output !== void 0 ? { output } : {},
+            ...status === "error" && error !== void 0 ? { error } : {}
+          });
+        } catch (error) {
+          debugLogger.debug(
+            "Failed to process @cloudflare/ai-chat response hook:",
+            error
+          );
+        } finally {
+          if (state?.settled) {
+            this.cleanupState(state);
+          }
+        }
+      }
+    };
+    tracingChannel.subscribe(handlers);
+    this.unsubscribers.push(() => tracingChannel.unsubscribe(handlers));
+  }
+  bindCurrentSpanStore(tracingChannel) {
+    const globalState = _internalGetGlobalState();
+    const contextManager = globalState?.contextManager;
+    const startChannel = tracingChannel.start;
+    const currentSpanStore = contextManager ? contextManager[BRAINTRUST_CURRENT_SPAN_STORE] : void 0;
+    if (!startChannel || !currentSpanStore || !contextManager) {
+      return void 0;
+    }
+    startChannel.bindStore(currentSpanStore, (event) => {
+      const state = this.ensureEventState(event);
+      return state ? contextManager.wrapSpanForStore(state.span) : currentSpanStore.getStore();
+    });
+    return () => startChannel.unbindStore(currentSpanStore);
+  }
+  ensureEventState(event) {
+    const eventKey = event;
+    const known = this.eventStates.get(eventKey);
+    if (known) {
+      return known;
+    }
+    try {
+      const agent = asObject(event.self);
+      const requestId = stringValue3(event.arguments[0]);
+      const key = requestId ?? /* @__PURE__ */ Symbol("cloudflare-ai-chat-turn");
+      const activeForAgent = agent ? this.getActiveTurns(agent) : void 0;
+      const existing = activeForAgent?.get(key);
+      const state = existing && !existing.settled ? existing : this.createTurnState(agent, key);
+      if (existing && !existing.settled) {
+        state.depth += 1;
+      }
+      this.eventStates.set(eventKey, state);
+      this.bindTurnCallback(event, agent, state);
+      if (agent) {
+        instrumentCloudflareAIChatResponseHook(agent);
+      }
+      return state;
+    } catch (error) {
+      debugLogger.debug(
+        "Failed to start @cloudflare/ai-chat instrumentation:",
+        error
+      );
+      return void 0;
+    }
+  }
+  createTurnState(agent, key) {
+    const span = startSpan(
+      withSpanInstrumentationName(
+        {
+          name: "AIChatAgent.onChatMessage",
+          spanAttributes: { type: "task" /* TASK */ }
+        },
+        INSTRUMENTATION_NAMES.CLOUDFLARE_AI_CHAT
+      )
+    );
+    const state = {
+      agent,
+      depth: 1,
+      ended: false,
+      key,
+      responseObserved: false,
+      settled: false,
+      span
+    };
+    this.activeStates.add(state);
+    if (agent) {
+      this.getActiveTurns(agent).set(key, state);
+    }
+    return state;
+  }
+  bindTurnCallback(event, agent, state) {
+    const callback = event.arguments[1];
+    if (typeof callback !== "function") {
+      return;
+    }
+    try {
+      event.arguments[1] = function(...args) {
+        return withCurrent(state.span, () => {
+          try {
+            const input = serializeMessages(readProperty(agent, "messages"));
+            if (input !== void 0 && state.input === void 0) {
+              state.input = input;
+              state.span.log({ input });
+            }
+          } catch (error) {
+            debugLogger.debug(
+              "Failed to capture @cloudflare/ai-chat messages:",
+              error
+            );
+          }
+          return Reflect.apply(callback, this, args);
+        });
+      };
+    } catch (error) {
+      debugLogger.debug(
+        "Failed to bind @cloudflare/ai-chat turn callback context:",
+        error
+      );
+    }
+  }
+  finishEvent(event, error) {
+    const eventKey = event;
+    const state = this.eventStates.get(eventKey);
+    if (!state || state.ended) {
+      return;
+    }
+    this.eventStates.delete(eventKey);
+    if (error !== void 0 && state.pendingError === void 0) {
+      state.pendingError = error;
+    }
+    state.depth -= 1;
+    if (state.depth === 0) {
+      state.settled = true;
+      this.endState(state);
+      if (state.responseObserved || error !== void 0) {
+        this.cleanupState(state);
+      } else {
+        state.cleanupTimer = setTimeout(
+          () => this.cleanupState(state),
+          RESPONSE_STATE_RETENTION_MS
+        );
+        if (typeof state.cleanupTimer === "object" && state.cleanupTimer !== null && "unref" in state.cleanupTimer && typeof state.cleanupTimer.unref === "function") {
+          state.cleanupTimer.unref();
+        }
+      }
+    }
+  }
+  endState(state) {
+    if (state.ended) {
+      return;
+    }
+    state.ended = true;
+    try {
+      if (state.pendingError !== void 0) {
+        state.span.log({ error: state.pendingError });
+      }
+    } finally {
+      state.span.end();
+    }
+  }
+  cleanupState(state) {
+    if (state.cleanupTimer !== void 0) {
+      clearTimeout(state.cleanupTimer);
+      state.cleanupTimer = void 0;
+    }
+    try {
+      this.endState(state);
+    } finally {
+      this.activeStates.delete(state);
+      if (state.agent) {
+        const activeForAgent = this.activeTurns.get(state.agent);
+        if (activeForAgent?.get(state.key) === state) {
+          activeForAgent.delete(state.key);
+        }
+      }
+    }
+  }
+  getActiveTurns(agent) {
+    let turns = this.activeTurns.get(agent);
+    if (!turns) {
+      turns = /* @__PURE__ */ new Map();
+      this.activeTurns.set(agent, turns);
+    }
+    return turns;
+  }
+  findResponseState(agent, requestId) {
+    const turns = this.activeTurns.get(agent);
+    if (!turns) {
+      return void 0;
+    }
+    if (requestId !== void 0) {
+      return turns.get(requestId);
+    }
+    const active = [...turns.values()];
+    return active.length === 1 ? active[0] : void 0;
+  }
+};
+function asObject(value) {
+  return value !== null && (typeof value === "object" || typeof value === "function") ? value : void 0;
+}
+function readProperty(value, key) {
+  const object = asObject(value);
+  return object ? Reflect.get(object, key) : void 0;
+}
+function ownValue2(value, key) {
+  const object = asObject(value);
+  if (!object) {
+    return void 0;
+  }
+  const descriptor = Object.getOwnPropertyDescriptor(object, key);
+  return descriptor && "value" in descriptor ? descriptor.value : void 0;
+}
+function stringValue3(value) {
+  return typeof value === "string" ? value : void 0;
+}
+function serializeMessages(value) {
+  if (!Array.isArray(value)) {
+    return void 0;
+  }
+  return value.flatMap((message) => {
+    const serialized = serializeMessage(message);
+    return serialized === void 0 ? [] : [serialized];
+  });
+}
+function serializeMessage(value) {
+  const message = asObject(value);
+  if (!message) {
+    return void 0;
+  }
+  const serialized = [];
+  const id = ownValue2(message, "id");
+  const role = ownValue2(message, "role");
+  const parts = ownValue2(message, "parts");
+  if (typeof id === "string") {
+    serialized.push(["id", id]);
+  }
+  if (typeof role === "string") {
+    serialized.push(["role", role]);
+  }
+  const sanitizedParts = sanitizeLoggedValue2(parts);
+  if (sanitizedParts !== void 0) {
+    serialized.push(["parts", sanitizedParts]);
+  }
+  return serialized.length > 0 ? Object.fromEntries(serialized) : void 0;
+}
+function sanitizeLoggedValue2(value, seen = /* @__PURE__ */ new WeakSet(), depth = 0) {
+  if (value === null || typeof value === "string" || typeof value === "boolean") {
+    return value;
+  }
+  if (typeof value === "number") {
+    return Number.isFinite(value) ? value : void 0;
+  }
+  const object = asObject(value);
+  if (!object || depth >= 20 || typeof value === "function") {
+    return void 0;
+  }
+  if (seen.has(object)) {
+    return "[Circular]";
+  }
+  seen.add(object);
+  if (Array.isArray(value)) {
+    return value.map((entry) => sanitizeLoggedValue2(entry, seen, depth + 1));
+  }
+  const entries = [];
+  for (const [key, descriptor] of Object.entries(
+    Object.getOwnPropertyDescriptors(object)
+  )) {
+    if (BLOCKED_KEYS.has(key) || !descriptor.enumerable || !("value" in descriptor)) {
+      continue;
+    }
+    const sanitized = sanitizeLoggedValue2(descriptor.value, seen, depth + 1);
+    if (sanitized !== void 0) {
+      entries.push([key, sanitized]);
+    }
+  }
+  return Object.fromEntries(entries);
+}
+
+// src/instrumentation/plugins/cloudflare-agents-plugin.ts
+var CLOUDFLARE_WORKERS_CONTEXT = {
+  span_origin: {
+    environment: { type: "server", name: "cloudflare_workers" }
+  }
+};
+var CloudflareAgentsPlugin = class extends BasePlugin {
+  onEnable() {
+    const channel = cloudflareAgentsChannels.runAgentTool.tracingChannel();
+    const spans = /* @__PURE__ */ new WeakMap();
+    const handlers = {
+      start: (event) => {
+        try {
+          const agentClass = event.arguments[0];
+          const options = event.arguments[1];
+          if (ownValue3(options, "detached")) {
+            return;
+          }
+          const name = ownValue3(agentClass, "name");
+          if (typeof name !== "string" || name.length === 0) {
+            debugLogger.warn(
+              "Skipping Cloudflare Agents runAgentTool span because the child agent class has no name."
+            );
+            return;
+          }
+          const span = _internalStartSpanWithContext(
+            withSpanInstrumentationName(
+              {
+                name,
+                spanAttributes: { type: "tool" /* TOOL */ },
+                event: {
+                  input: ownValue3(options, "input")
+                }
+              },
+              INSTRUMENTATION_NAMES.CLOUDFLARE_AGENTS
+            ),
+            CLOUDFLARE_WORKERS_CONTEXT
+          );
+          spans.set(event, span);
+        } catch (error) {
+          logInstrumentationError6("start", error);
+        }
+      },
+      asyncEnd: (event) => {
+        const span = spans.get(event);
+        if (!span) {
+          return;
+        }
+        spans.delete(event);
+        try {
+          const status = ownValue3(event.result, "status");
+          if (status === "completed") {
+            span.log({ output: ownValue3(event.result, "output") });
+          } else {
+            const error = ownValue3(event.result, "error");
+            if (typeof error === "string") {
+              span.log({ error });
+            }
+          }
+        } catch (error) {
+          logInstrumentationError6("completion", error);
+        } finally {
+          safelyEndSpan(span);
+        }
+      },
+      error: (event) => {
+        const span = spans.get(event);
+        if (!span) {
+          return;
+        }
+        spans.delete(event);
+        try {
+          span.log({ error: event.error });
+        } catch (error) {
+          logInstrumentationError6("rejection", error);
+        } finally {
+          safelyEndSpan(span);
+        }
+      }
+    };
+    channel.subscribe(handlers);
+    this.unsubscribers.push(() => channel.unsubscribe(handlers));
+  }
+  onDisable() {
+    for (const unsubscribe of this.unsubscribers) {
+      unsubscribe();
+    }
+    this.unsubscribers = [];
+  }
+};
+function ownValue3(value, key) {
+  if (!isObjectLike2(value)) {
+    return void 0;
+  }
+  const descriptor = Object.getOwnPropertyDescriptor(value, key);
+  return descriptor && "value" in descriptor ? descriptor.value : void 0;
+}
+function isObjectLike2(value) {
+  return typeof value === "object" && value !== null || typeof value === "function";
+}
+function safelyEndSpan(span) {
+  try {
+    span.end();
+  } catch (error) {
+    logInstrumentationError6("span end", error);
+  }
+}
+function logInstrumentationError6(operation, error) {
+  debugLogger.error(
+    `Failed to process Cloudflare Agents ${operation} instrumentation:`,
+    error
+  );
+}
+
 // src/instrumentation/braintrust-plugin.ts
 var BraintrustPlugin = class extends BasePlugin {
   config;
@@ -31897,6 +32806,7 @@ var BraintrustPlugin = class extends BasePlugin {
   openAIAgentsPlugin = null;
   googleGenAIPlugin = null;
   huggingFacePlugin = null;
+  huggingFaceTransformersPlugin = null;
   openRouterPlugin = null;
   openRouterAgentPlugin = null;
   mistralPlugin = null;
@@ -31911,6 +32821,8 @@ var BraintrustPlugin = class extends BasePlugin {
   langSmithPlugin = null;
   piCodingAgentPlugin = null;
   strandsAgentSDKPlugin = null;
+  cloudflareAIChatPlugin = null;
+  cloudflareAgentsPlugin = null;
   constructor(config = {}) {
     super();
     this.config = config;
@@ -31952,6 +32864,8 @@ var BraintrustPlugin = class extends BasePlugin {
     if (integrations.huggingface !== false) {
       this.huggingFacePlugin = new HuggingFacePlugin();
       this.huggingFacePlugin.enable();
+      this.huggingFaceTransformersPlugin = new HuggingFaceTransformersPlugin();
+      this.huggingFaceTransformersPlugin.enable();
     }
     if (integrations.openrouter !== false) {
       this.openRouterPlugin = new OpenRouterPlugin();
@@ -31997,6 +32911,14 @@ var BraintrustPlugin = class extends BasePlugin {
       this.strandsAgentSDKPlugin = new StrandsAgentSDKPlugin();
       this.strandsAgentSDKPlugin.enable();
     }
+    if (integrations.cloudflareAIChat !== false) {
+      this.cloudflareAIChatPlugin = new CloudflareAIChatPlugin();
+      this.cloudflareAIChatPlugin.enable();
+    }
+    if (integrations.cloudflareAgents !== false) {
+      this.cloudflareAgentsPlugin = new CloudflareAgentsPlugin();
+      this.cloudflareAgentsPlugin.enable();
+    }
     if (integrations.flue !== false) {
       this.fluePlugin = new FluePlugin();
       this.fluePlugin.enable();
@@ -32049,6 +32971,10 @@ var BraintrustPlugin = class extends BasePlugin {
       this.huggingFacePlugin.disable();
       this.huggingFacePlugin = null;
     }
+    if (this.huggingFaceTransformersPlugin) {
+      this.huggingFaceTransformersPlugin.disable();
+      this.huggingFaceTransformersPlugin = null;
+    }
     if (this.openRouterPlugin) {
       this.openRouterPlugin.disable();
       this.openRouterPlugin = null;
@@ -32093,6 +33019,14 @@ var BraintrustPlugin = class extends BasePlugin {
       this.strandsAgentSDKPlugin.disable();
       this.strandsAgentSDKPlugin = null;
     }
+    if (this.cloudflareAIChatPlugin) {
+      this.cloudflareAIChatPlugin.disable();
+      this.cloudflareAIChatPlugin = null;
+    }
+    if (this.cloudflareAgentsPlugin) {
+      this.cloudflareAgentsPlugin.disable();
+      this.cloudflareAgentsPlugin = null;
+    }
     if (this.fluePlugin) {
       this.fluePlugin.disable();
       this.fluePlugin = null;
@@ -32109,7 +33043,9 @@ var BraintrustPlugin = class extends BasePlugin {
 };
 
 // src/instrumentation/registry.ts
-var REGISTRY_STATE_KEY = /* @__PURE__ */ Symbol.for("braintrust.registry");
+var REGISTRY_STATE_KEY = /* @__PURE__ */ Symbol.for(
+  `braintrust.registry.global-hooks.v${GLOBAL_INSTRUMENTATION_HOOKS_PROTOCOL_VERSION}`
+);
 function getSharedState() {
   const state = globalThis[/* @__PURE__ */ Symbol.for("braintrust-state")];
   return state && typeof state === "object" ? state : void 0;
@@ -32425,7 +33361,6 @@ function wrapMastraAgent(agent, _options) {
 
 // src/node/config.ts
 import { AsyncLocalStorage } from "node:async_hooks";
-import * as diagnostics_channel from "node:diagnostics_channel";
 import * as path from "node:path";
 import * as fs from "node:fs/promises";
 import * as os from "node:os";
@@ -32712,8 +33647,6 @@ function configureNode() {
   };
   isomorph_default.getCallerLocation = getCallerLocation;
   isomorph_default.newAsyncLocalStorage = () => new AsyncLocalStorage();
-  isomorph_default.newTracingChannel = (nameOrChannels) => diagnostics_channel.tracingChannel(nameOrChannels);
-  patchTracingChannel(diagnostics_channel.tracingChannel);
   isomorph_default.processOn = (event, handler) => {
     process.on(event, handler);
   };
@@ -32950,6 +33883,7 @@ export {
   braintrustFlueObserver,
   BRAINTRUST_LANGCHAIN_CALLBACK_HANDLER_NAME,
   BraintrustLangChainCallbackHandler,
+  instrumentCloudflareAIChatAgent,
   configureInstrumentation,
   BraintrustObservabilityExporter,
   wrapMastraAgent,