braintrust 3.8.0 → 3.9.0

package/dist/instrumentation/index.mjs

@@ -99,6 +99,16 @@ var isomorph_default = iso;
 function isAsyncIterable(value) {
   return value !== null && typeof value === "object" && Symbol.asyncIterator in value && typeof value[Symbol.asyncIterator] === "function";
 }
+function hasAsyncIteratorMethods(value) {
+  return value !== null && typeof value === "object" && "next" in value && typeof value.next === "function";
+}
+function isSelfAsyncIterator(value) {
+  try {
+    return value[Symbol.asyncIterator]() === value;
+  } catch {
+    return false;
+  }
+}
 function patchStreamIfNeeded(stream, options) {
   if (!isAsyncIterable(stream)) {
     return stream;
@@ -109,6 +119,98 @@ function patchStreamIfNeeded(stream, options) {
     );
     return stream;
   }
+  if (hasAsyncIteratorMethods(stream) && isSelfAsyncIterator(stream)) {
+    if ("__braintrust_patched_iterator_methods" in stream) {
+      return stream;
+    }
+    try {
+      const originalNext = stream.next.bind(stream);
+      const originalReturn = typeof stream.return === "function" ? stream.return.bind(stream) : null;
+      const originalThrow = typeof stream.throw === "function" ? stream.throw.bind(stream) : null;
+      const chunks = [];
+      let completed = false;
+      stream.next = async (...args) => {
+        try {
+          const result = await originalNext(...args);
+          if (result.done) {
+            if (!completed) {
+              completed = true;
+              try {
+                await options.onComplete(chunks);
+              } catch (error) {
+                console.error("Error in stream onComplete handler:", error);
+              }
+            }
+          } else {
+            const chunk = result.value;
+            const shouldCollect = options.shouldCollect ? options.shouldCollect(chunk) : true;
+            if (shouldCollect) {
+              chunks.push(chunk);
+              if (options.onChunk) {
+                try {
+                  await options.onChunk(chunk);
+                } catch (error) {
+                  console.error("Error in stream onChunk handler:", error);
+                }
+              }
+            }
+          }
+          return result;
+        } catch (error) {
+          if (!completed) {
+            completed = true;
+            if (options.onError) {
+              try {
+                await options.onError(
+                  error instanceof Error ? error : new Error(String(error)),
+                  chunks
+                );
+              } catch (handlerError) {
+                console.error("Error in stream onError handler:", handlerError);
+              }
+            }
+          }
+          throw error;
+        }
+      };
+      if (originalReturn) {
+        stream.return = async (...args) => {
+          if (!completed) {
+            completed = true;
+            try {
+              await options.onComplete(chunks);
+            } catch (error) {
+              console.error("Error in stream onComplete handler:", error);
+            }
+          }
+          return originalReturn(...args);
+        };
+      }
+      if (originalThrow) {
+        stream.throw = async (...args) => {
+          if (!completed) {
+            completed = true;
+            const rawError = args[0];
+            const error = rawError instanceof Error ? rawError : new Error(String(rawError));
+            if (options.onError) {
+              try {
+                await options.onError(error, chunks);
+              } catch (handlerError) {
+                console.error("Error in stream onError handler:", handlerError);
+              }
+            }
+          }
+          return originalThrow(...args);
+        };
+      }
+      Object.defineProperty(stream, "__braintrust_patched_iterator_methods", {
+        value: true
+      });
+      return stream;
+    } catch (error) {
+      console.warn("Failed to patch stream iterator methods:", error);
+    }
+  }
   const originalIteratorFn = stream[Symbol.asyncIterator];
   if ("__braintrust_patched" in originalIteratorFn && originalIteratorFn["__braintrust_patched"]) {
     return stream;
@@ -1018,6 +1120,9 @@ function isArray(value) {
 function isObjectOrArray(value) {
   return value instanceof Object;
 }
+function isPromiseLike(value) {
+  return value != null && (typeof value === "object" || typeof value === "function") && "then" in value && typeof value.then === "function";
+}
 
 // util/object_util.ts
 var SET_UNION_FIELDS = /* @__PURE__ */ new Set(["tags"]);
@@ -1973,7 +2078,11 @@ var CodeBundle = z6.object({
       eval_name: z6.string(),
       position: z6.union([
         z6.object({ type: z6.literal("task") }),
-        z6.object({ type: z6.literal("scorer"), index: z6.number().int().gte(0) })
+        z6.object({ type: z6.literal("scorer"), index: z6.number().int().gte(0) }),
+        z6.object({
+          type: z6.literal("classifier"),
+          index: z6.number().int().gte(0)
+        })
       ])
     }),
     z6.object({ type: z6.literal("function"), index: z6.number().int().gte(0) }),
@@ -3095,11 +3204,15 @@ var RunEval = z6.object({
   data: z6.union([
     z6.object({
       dataset_id: z6.string(),
+      dataset_version: z6.union([z6.string(), z6.null()]).optional(),
+      dataset_environment: z6.union([z6.string(), z6.null()]).optional(),
       _internal_btql: z6.union([z6.object({}).partial().passthrough(), z6.null()]).optional()
     }),
     z6.object({
       project_name: z6.string(),
       dataset_name: z6.string(),
+      dataset_version: z6.union([z6.string(), z6.null()]).optional(),
+      dataset_environment: z6.union([z6.string(), z6.null()]).optional(),
       _internal_btql: z6.union([z6.object({}).partial().passthrough(), z6.null()]).optional()
     }),
     z6.object({ data: z6.array(z6.unknown()) })
@@ -3227,6 +3340,7 @@ var ViewOptions = z6.union([
     rowHeight: z6.union([z6.string(), z6.null()]),
     tallGroupRows: z6.union([z6.boolean(), z6.null()]),
     layout: z6.union([z6.string(), z6.null()]),
+    topicMapReportKey: z6.union([z6.string(), z6.null()]),
     chartHeight: z6.union([z6.number(), z6.null()]),
     excludedMeasures: z6.union([
       z6.array(
@@ -3269,6 +3383,7 @@ var ViewOptions = z6.union([
       z6.null()
     ]),
     queryShape: z6.union([z6.enum(["traces", "spans"]), z6.null()]),
+    cluster: z6.union([z6.string(), z6.null()]),
     freezeColumns: z6.union([z6.boolean(), z6.null()])
   }).partial(),
   z6.null()
@@ -6538,8 +6653,9 @@ function validateAndSanitizeExperimentLogPartialArgs(event) {
 }
 function deepCopyEvent(event) {
   const attachments = [];
-  const IDENTIFIER = "_bt_internal_saved_attachment";
-  const savedAttachmentSchema = z8.strictObject({ [IDENTIFIER]: z8.number() });
+  const ATTACHMENT_INDEX_KEY = "_bt_internal_saved_attachment_idx";
+  const ATTACHMENT_MARKER_KEY = "_bt_internal_saved_attachment_marker";
+  const attachmentMarker = ++deepCopyEventMarkerCounter;
   const serialized = JSON.stringify(event, (_k, v) => {
     if (v instanceof SpanImpl || v instanceof NoopSpan) {
       return `<span>`;
@@ -6551,21 +6667,39 @@ function deepCopyEvent(event) {
       return `<logger>`;
     } else if (v instanceof BaseAttachment) {
       const idx = attachments.push(v);
-      return { [IDENTIFIER]: idx - 1 };
+      return {
+        [ATTACHMENT_INDEX_KEY]: idx - 1,
+        [ATTACHMENT_MARKER_KEY]: attachmentMarker
+      };
     } else if (v instanceof ReadonlyAttachment) {
       return v.reference;
     }
     return v;
   });
   const x = JSON.parse(serialized, (_k, v) => {
-    const parsedAttachment = savedAttachmentSchema.safeParse(v);
-    if (parsedAttachment.success) {
-      return attachments[parsedAttachment.data[IDENTIFIER]];
+    if (isDeepCopyAttachmentPlaceholder(v, attachmentMarker)) {
+      return attachments[v[ATTACHMENT_INDEX_KEY]];
     }
     return v;
   });
   return x;
 }
+var deepCopyEventMarkerCounter = 0;
+function isDeepCopyAttachmentPlaceholder(value, attachmentMarker) {
+  if (!value || typeof value !== "object" || Array.isArray(value)) {
+    return false;
+  }
+  const record = value;
+  if (!Object.hasOwn(record, "_bt_internal_saved_attachment_idx") || !Object.hasOwn(record, "_bt_internal_saved_attachment_marker")) {
+    return false;
+  }
+  if (Object.keys(record).length !== 2) {
+    return false;
+  }
+  const attachmentIndex = record._bt_internal_saved_attachment_idx;
+  const marker = record._bt_internal_saved_attachment_marker;
+  return Number.isInteger(attachmentIndex) && attachmentIndex >= 0 && marker === attachmentMarker;
+}
 function extractAttachments(event, attachments) {
   for (const [key, value] of Object.entries(event)) {
     if (!value) {
@@ -9144,10 +9278,42 @@ function processImagesInOutput(output) {
   }
   return output;
 }
+function mergeLogprobTokens(existing, incoming) {
+  if (incoming === void 0) {
+    return existing;
+  }
+  if (incoming === null) {
+    return existing ?? null;
+  }
+  if (Array.isArray(existing)) {
+    return [...existing, ...incoming];
+  }
+  return [...incoming];
+}
+function aggregateChatLogprobs(existing, incoming) {
+  if (incoming === void 0) {
+    return existing;
+  }
+  if (incoming === null) {
+    return existing ?? null;
+  }
+  const aggregated = existing && existing !== null ? { ...existing, ...incoming } : { ...incoming };
+  const content = mergeLogprobTokens(existing?.content, incoming.content);
+  if (content !== void 0) {
+    aggregated.content = content;
+  }
+  const refusal = mergeLogprobTokens(existing?.refusal, incoming.refusal);
+  if (refusal !== void 0) {
+    aggregated.refusal = refusal;
+  }
+  return aggregated;
+}
 function aggregateChatCompletionChunks(chunks, streamResult, endEvent) {
   let role = void 0;
   let content = void 0;
+  let refusal = void 0;
   let tool_calls = void 0;
+  let logprobs = void 0;
   let finish_reason = void 0;
   let metrics = {};
   for (const chunk of chunks) {
@@ -9157,19 +9323,30 @@ function aggregateChatCompletionChunks(chunks, streamResult, endEvent) {
         ...parseMetricsFromUsage(chunk.usage)
       };
     }
-    const delta = chunk.choices?.[0]?.delta;
-    if (!delta) {
+    const choice = chunk.choices?.[0];
+    if (!choice) {
       continue;
     }
-    if (!role && delta.role) {
-      role = delta.role;
+    if (choice.finish_reason) {
+      finish_reason = choice.finish_reason;
+    }
+    logprobs = aggregateChatLogprobs(logprobs, choice.logprobs);
+    const delta = choice.delta;
+    if (!delta) {
+      continue;
     }
     if (delta.finish_reason) {
       finish_reason = delta.finish_reason;
     }
+    if (!role && delta.role) {
+      role = delta.role;
+    }
     if (delta.content) {
       content = (content || "") + delta.content;
     }
+    if (delta.refusal) {
+      refusal = (refusal || "") + delta.refusal;
+    }
     if (delta.tool_calls) {
       const toolDelta = delta.tool_calls[0];
       if (!tool_calls || toolDelta.id && tool_calls[tool_calls.length - 1].id !== toolDelta.id) {
@@ -9195,9 +9372,10 @@ function aggregateChatCompletionChunks(chunks, streamResult, endEvent) {
         message: {
           role,
           content,
+          ...refusal !== void 0 ? { refusal } : {},
           tool_calls
         },
-        logprobs: null,
+        logprobs: logprobs ?? null,
         finish_reason
       }
     ]
@@ -9260,13 +9438,21 @@ var anthropicChannels = defineChannels("@anthropic-ai/sdk", {
   betaMessagesCreate: channel({
     channelName: "beta.messages.create",
     kind: "async"
+  }),
+  betaMessagesToolRunner: channel({
+    channelName: "beta.messages.toolRunner",
+    kind: "sync-stream"
   })
 });
 
 // src/instrumentation/plugins/anthropic-plugin.ts
+var ANTHROPIC_TOOL_RUNNER_TOOL_WRAPPED = Symbol.for(
+  "braintrust.anthropic_tool_runner_tool_wrapped"
+);
 var AnthropicPlugin = class extends BasePlugin {
   onEnable() {
     this.subscribeToAnthropicChannels();
+    this.subscribeToAnthropicToolRunner();
   }
   onDisable() {
     this.unsubscribers = unsubscribeAll(this.unsubscribers);
@@ -9321,6 +9507,61 @@ var AnthropicPlugin = class extends BasePlugin {
       })
     );
   }
+  subscribeToAnthropicToolRunner() {
+    const tracingChannel = anthropicChannels.betaMessagesToolRunner.tracingChannel();
+    const states = /* @__PURE__ */ new WeakMap();
+    const handlers = {
+      start: (event) => {
+        const params = event.arguments[0] ?? {};
+        const span = startSpan({
+          name: "anthropic.beta.messages.toolRunner",
+          spanAttributes: {
+            type: "task" /* TASK */
+          }
+        });
+        span.log({
+          input: processAttachmentsInInput(
+            coalesceInput(params.messages ?? [], params.system)
+          ),
+          metadata: {
+            ...extractAnthropicToolRunnerMetadata(params),
+            provider: "anthropic"
+          }
+        });
+        const state = {
+          aggregatedMetrics: {},
+          finalized: false,
+          iterationCount: 0,
+          seenMessages: /* @__PURE__ */ new WeakSet(),
+          span,
+          startTime: getCurrentUnixTimestamp()
+        };
+        states.set(event, state);
+      },
+      end: (event) => {
+        const state = states.get(event);
+        if (!state) {
+          return;
+        }
+        patchAnthropicToolRunner({
+          runner: event.result,
+          state
+        });
+      },
+      error: (event) => {
+        const state = states.get(event);
+        if (!state || !event.error) {
+          return;
+        }
+        finalizeAnthropicToolRunnerError(state, event.error);
+        states.delete(event);
+      }
+    };
+    tracingChannel.subscribe(handlers);
+    this.unsubscribers.push(() => {
+      tracingChannel.unsubscribe(handlers);
+    });
+  }
 };
 function parseMetricsFromUsage2(usage) {
   if (!usage) {
@@ -9337,72 +9578,411 @@ function parseMetricsFromUsage2(usage) {
   saveIfExistsTo("output_tokens", "completion_tokens");
   saveIfExistsTo("cache_read_input_tokens", "prompt_cached_tokens");
   saveIfExistsTo("cache_creation_input_tokens", "prompt_cache_creation_tokens");
+  if (isObject(usage.server_tool_use)) {
+    for (const [name, value] of Object.entries(usage.server_tool_use)) {
+      if (typeof value === "number") {
+        metrics[`server_tool_use_${name}`] = value;
+      }
+    }
+  }
   return metrics;
 }
-function aggregateAnthropicStreamChunks(chunks) {
-  const fallbackTextDeltas = [];
-  const contentBlocks = {};
-  const contentBlockDeltas = {};
-  let metrics = {};
-  let metadata = {};
-  let role;
-  for (const event of chunks) {
-    switch (event?.type) {
-      case "message_start":
-        if (event.message?.usage) {
-          const initialMetrics = parseMetricsFromUsage2(event.message.usage);
-          metrics = { ...metrics, ...initialMetrics };
-        }
-        if (typeof event.message?.role === "string") {
-          role = event.message.role;
-        }
-        break;
-      case "content_block_start":
-        if (event.content_block) {
-          contentBlocks[event.index] = event.content_block;
-          contentBlockDeltas[event.index] = { textDeltas: [], citations: [] };
-        }
-        break;
-      case "content_block_delta": {
-        const acc = contentBlockDeltas[event.index];
-        const delta = event.delta;
-        if (!delta) break;
-        if (delta.type === "text_delta" && "text" in delta) {
-          const text = delta.text;
-          if (text) {
-            if (acc !== void 0) {
-              acc.textDeltas.push(text);
-            } else {
-              fallbackTextDeltas.push(text);
+function extractAnthropicToolRunnerMetadata(params) {
+  const metadata = filterFrom(params, ["messages", "system", "tools"]);
+  const toolNames = extractAnthropicToolNames(params.tools);
+  return {
+    ...metadata,
+    operation: "toolRunner",
+    ...toolNames.length > 0 ? { tool_names: toolNames } : {}
+  };
+}
+function extractAnthropicToolNames(tools) {
+  const toolNames = [];
+  for (const tool of tools) {
+    if (!tool || typeof tool !== "object") {
+      continue;
+    }
+    const toolRecord = tool;
+    if (typeof toolRecord.name === "string") {
+      toolNames.push(toolRecord.name);
+      continue;
+    }
+    if (typeof toolRecord.mcp_server_name === "string") {
+      toolNames.push(toolRecord.mcp_server_name);
+    }
+  }
+  return toolNames;
+}
+function toErrorMessage(error) {
+  return error instanceof Error ? error.message : String(error);
+}
+function getAnthropicToolRunnerInput(args) {
+  return args.length > 0 ? args[0] : void 0;
+}
+function wrapAnthropicToolRunnerTools(params, state) {
+  if (!Array.isArray(params.tools)) {
+    return;
+  }
+  params.tools = params.tools.map(
+    (tool, index) => wrapAnthropicToolRunnerTool(tool, index, state)
+  );
+}
+function wrapAnthropicToolRunnerTool(tool, index, state) {
+  if (!tool || typeof tool !== "object" || typeof tool.run !== "function" || tool[ANTHROPIC_TOOL_RUNNER_TOOL_WRAPPED]) {
+    return tool;
+  }
+  const toolName = typeof tool.name === "string" ? tool.name : `tool[${index}]`;
+  const originalRun = tool.run;
+  const runDescriptor = Object.getOwnPropertyDescriptor(tool, "run");
+  const wrappedTool = Object.create(
+    Object.getPrototypeOf(tool),
+    Object.getOwnPropertyDescriptors(tool)
+  );
+  Object.defineProperty(wrappedTool, "run", {
+    configurable: runDescriptor?.configurable ?? true,
+    enumerable: runDescriptor?.enumerable ?? true,
+    value: function braintrustAnthropicToolRunnerRun(...args) {
+      return state.span.traced(
+        (span) => {
+          const finalizeSuccess = (result) => {
+            span.log({ output: result });
+            return result;
+          };
+          const finalizeError = (error) => {
+            span.log({ error: toErrorMessage(error) });
+            throw error;
+          };
+          try {
+            const result = Reflect.apply(originalRun, this, args);
+            if (isPromiseLike(result)) {
+              return result.then(finalizeSuccess, finalizeError);
             }
+            return finalizeSuccess(result);
+          } catch (error) {
+            return finalizeError(error);
           }
-        } else if (delta.type === "input_json_delta" && "partial_json" in delta) {
-          const partialJson = delta.partial_json;
-          if (partialJson && acc !== void 0) {
-            acc.textDeltas.push(partialJson);
-          }
-        } else if (delta.type === "thinking_delta" && "thinking" in delta) {
-          const thinking = delta.thinking;
-          if (thinking && acc !== void 0) {
-            acc.textDeltas.push(thinking);
-          }
-        } else if (delta.type === "citations_delta" && "citation" in delta) {
-          const citation = delta.citation;
-          if (citation && acc !== void 0) {
-            acc.citations.push(citation);
+        },
+        {
+          event: {
+            input: getAnthropicToolRunnerInput(args),
+            metadata: {
+              "gen_ai.tool.name": toolName,
+              provider: "anthropic"
+            }
+          },
+          name: `tool: ${toolName}`,
+          spanAttributes: {
+            type: "tool" /* TOOL */
           }
         }
-        break;
-      }
-      case "content_block_stop":
-        finalizeContentBlock(
-          event.index,
-          contentBlocks,
-          contentBlockDeltas,
-          fallbackTextDeltas
-        );
-        break;
-      case "message_delta":
+      );
+    },
+    writable: runDescriptor?.writable ?? true
+  });
+  Object.defineProperty(wrappedTool, ANTHROPIC_TOOL_RUNNER_TOOL_WRAPPED, {
+    configurable: false,
+    enumerable: false,
+    value: true,
+    writable: false
+  });
+  return wrappedTool;
+}
+function getAnthropicToolRunnerParams(runnerRecord) {
+  const params = Reflect.get(runnerRecord, "params");
+  return params && typeof params === "object" ? params : void 0;
+}
+function ensureAnthropicToolRunnerToolsWrapped(runnerRecord, state) {
+  const params = getAnthropicToolRunnerParams(runnerRecord);
+  if (params) {
+    wrapAnthropicToolRunnerTools(params, state);
+  }
+}
+function wrapAnthropicToolRunnerSetMessagesParams(runnerRecord, state) {
+  const setMessagesParams = Reflect.get(runnerRecord, "setMessagesParams");
+  if (typeof setMessagesParams !== "function") {
+    return;
+  }
+  Reflect.set(
+    runnerRecord,
+    "setMessagesParams",
+    function patchedSetMessagesParams(...args) {
+      const result = Reflect.apply(setMessagesParams, this, args);
+      const nextParams = getAnthropicToolRunnerParams(runnerRecord);
+      if (nextParams) {
+        wrapAnthropicToolRunnerTools(nextParams, state);
+      }
+      return result;
+    }
+  );
+}
+function isAnthropicMessage(value) {
+  return !!value && typeof value === "object" && typeof value.role === "string" && Array.isArray(value.content);
+}
+function isAnthropicMessageStream(value) {
+  return !!value && typeof value === "object" && isAsyncIterable(value) && "finalMessage" in value && typeof value.finalMessage === "function";
+}
+function recordAnthropicToolRunnerMessage(state, message) {
+  if (typeof message !== "object" || message === null) {
+    return;
+  }
+  if (state.seenMessages.has(message)) {
+    state.lastMessage = message;
+    return;
+  }
+  state.seenMessages.add(message);
+  state.lastMessage = message;
+  const parsedMetrics = parseMetricsFromUsage2(message.usage);
+  for (const [key, value] of Object.entries(parsedMetrics)) {
+    if (typeof value === "number") {
+      state.aggregatedMetrics[key] = (state.aggregatedMetrics[key] ?? 0) + value;
+    }
+  }
+}
+function instrumentAnthropicMessageStream(stream, state) {
+  if ("__braintrust_tool_runner_stream_patched" in stream) {
+    return;
+  }
+  if (!Object.isFrozen(stream) && !Object.isSealed(stream)) {
+    patchStreamIfNeeded(stream, {
+      onChunk: () => {
+        if (state.firstTokenTime === void 0) {
+          state.firstTokenTime = getCurrentUnixTimestamp();
+        }
+      },
+      onComplete: () => void 0
+    });
+  }
+  if (typeof stream.finalMessage === "function") {
+    const originalFinalMessage = stream.finalMessage.bind(stream);
+    stream.finalMessage = async () => {
+      const message = await originalFinalMessage();
+      recordAnthropicToolRunnerMessage(state, message);
+      return message;
+    };
+  }
+  Object.defineProperty(stream, "__braintrust_tool_runner_stream_patched", {
+    value: true
+  });
+}
+async function finalizeAnthropicToolRunner(state, finalMessage) {
+  if (state.finalized) {
+    return;
+  }
+  state.finalized = true;
+  const message = finalMessage ?? state.lastMessage;
+  if (message) {
+    recordAnthropicToolRunnerMessage(state, message);
+  }
+  const metrics = finalizeAnthropicTokens({
+    ...state.aggregatedMetrics
+  });
+  if (state.firstTokenTime !== void 0) {
+    metrics.time_to_first_token = state.firstTokenTime - state.startTime;
+  }
+  const metadata = {
+    anthropic_tool_runner_iterations: state.iterationCount
+  };
+  if (message?.stop_reason !== void 0) {
+    metadata.stop_reason = message.stop_reason;
+  }
+  if (message?.stop_sequence !== void 0) {
+    metadata.stop_sequence = message.stop_sequence;
+  }
+  state.span.log({
+    ...message ? { output: { role: message.role, content: message.content } } : {},
+    metadata,
+    metrics: Object.fromEntries(
+      Object.entries(metrics).filter(
+        (entry) => entry[1] !== void 0
+      )
+    )
+  });
+  state.span.end();
+}
+function finalizeAnthropicToolRunnerError(state, error) {
+  if (state.finalized) {
+    return;
+  }
+  state.finalized = true;
+  state.span.log({
+    error: error instanceof Error ? error.message : String(error)
+  });
+  state.span.end();
+}
+async function resolveAnthropicToolRunnerFinalMessage(runner) {
+  if (typeof runner.done === "function") {
+    return await runner.done();
+  }
+  if (typeof runner.runUntilDone === "function") {
+    return await runner.runUntilDone();
+  }
+  return void 0;
+}
+function wrapAnthropicToolRunnerPromiseMethod(runnerRecord, methodName, state) {
+  const method = runnerRecord[methodName];
+  if (typeof method !== "function") {
+    return;
+  }
+  runnerRecord[methodName] = async (...args) => {
+    ensureAnthropicToolRunnerToolsWrapped(runnerRecord, state);
+    return await withCurrent(state.span, async () => {
+      try {
+        const message = await method.apply(runnerRecord, args);
+        recordAnthropicToolRunnerMessage(state, message);
+        await finalizeAnthropicToolRunner(state, message);
+        return message;
+      } catch (error) {
+        finalizeAnthropicToolRunnerError(state, error);
+        throw error;
+      }
+    });
+  };
+}
+function patchAnthropicToolRunner(args) {
+  const { runner, state } = args;
+  if (!runner || typeof runner !== "object") {
+    void finalizeAnthropicToolRunner(state);
+    return;
+  }
+  const runnerRecord = runner;
+  if ("__braintrust_tool_runner_patched" in runnerRecord) {
+    return;
+  }
+  ensureAnthropicToolRunnerToolsWrapped(runnerRecord, state);
+  wrapAnthropicToolRunnerSetMessagesParams(runnerRecord, state);
+  wrapAnthropicToolRunnerPromiseMethod(runnerRecord, "done", state);
+  wrapAnthropicToolRunnerPromiseMethod(runnerRecord, "runUntilDone", state);
+  if (!isAsyncIterable(runnerRecord)) {
+    Object.defineProperty(runnerRecord, "__braintrust_tool_runner_patched", {
+      value: true
+    });
+    return;
+  }
+  const originalIterator = runnerRecord[Symbol.asyncIterator].bind(runnerRecord);
+  runnerRecord[Symbol.asyncIterator] = function() {
+    const iterator = originalIterator();
+    return {
+      async next(value) {
+        try {
+          ensureAnthropicToolRunnerToolsWrapped(runnerRecord, state);
+          const result = await withCurrent(
+            state.span,
+            () => value === void 0 ? iterator.next() : iterator.next(value)
+          );
+          if (result.done) {
+            const finalMessage = await resolveAnthropicToolRunnerFinalMessage(runner);
+            await finalizeAnthropicToolRunner(state, finalMessage);
+            return result;
+          }
+          state.iterationCount += 1;
+          if (isAnthropicMessage(result.value)) {
+            if (state.firstTokenTime === void 0) {
+              state.firstTokenTime = getCurrentUnixTimestamp();
+            }
+            recordAnthropicToolRunnerMessage(state, result.value);
+          } else if (isAnthropicMessageStream(result.value)) {
+            instrumentAnthropicMessageStream(result.value, state);
+          }
+          return result;
+        } catch (error) {
+          finalizeAnthropicToolRunnerError(state, error);
+          throw error;
+        }
+      },
+      async return(value) {
+        try {
+          ensureAnthropicToolRunnerToolsWrapped(runnerRecord, state);
+          const result = typeof iterator.return === "function" ? await withCurrent(state.span, () => iterator.return(value)) : { done: true, value };
+          const finalMessage = await resolveAnthropicToolRunnerFinalMessage(
+            runner
+          ).catch(() => void 0);
+          await finalizeAnthropicToolRunner(state, finalMessage);
+          return result;
+        } catch (error) {
+          finalizeAnthropicToolRunnerError(state, error);
+          throw error;
+        }
+      },
+      async throw(error) {
+        finalizeAnthropicToolRunnerError(state, error);
+        if (typeof iterator.throw === "function") {
+          return await withCurrent(state.span, () => iterator.throw(error));
+        }
+        throw error;
+      },
+      [Symbol.asyncIterator]() {
+        return this;
+      }
+    };
+  };
+  Object.defineProperty(runnerRecord, "__braintrust_tool_runner_patched", {
+    value: true
+  });
+}
+function aggregateAnthropicStreamChunks(chunks) {
+  const fallbackTextDeltas = [];
+  const contentBlocks = {};
+  const contentBlockDeltas = {};
+  let metrics = {};
+  let metadata = {};
+  let role;
+  for (const event of chunks) {
+    switch (event?.type) {
+      case "message_start":
+        if (event.message?.usage) {
+          const initialMetrics = parseMetricsFromUsage2(event.message.usage);
+          metrics = { ...metrics, ...initialMetrics };
+        }
+        if (typeof event.message?.role === "string") {
+          role = event.message.role;
+        }
+        break;
+      case "content_block_start":
+        if (event.content_block) {
+          contentBlocks[event.index] = event.content_block;
+          contentBlockDeltas[event.index] = { textDeltas: [], citations: [] };
+        }
+        break;
+      case "content_block_delta": {
+        const acc = contentBlockDeltas[event.index];
+        const delta = event.delta;
+        if (!delta) break;
+        if (delta.type === "text_delta" && "text" in delta) {
+          const text = delta.text;
+          if (text) {
+            if (acc !== void 0) {
+              acc.textDeltas.push(text);
+            } else {
+              fallbackTextDeltas.push(text);
+            }
+          }
+        } else if (delta.type === "input_json_delta" && "partial_json" in delta) {
+          const partialJson = delta.partial_json;
+          if (partialJson && acc !== void 0) {
+            acc.textDeltas.push(partialJson);
+          }
+        } else if (delta.type === "thinking_delta" && "thinking" in delta) {
+          const thinking = delta.thinking;
+          if (thinking && acc !== void 0) {
+            acc.textDeltas.push(thinking);
+          }
+        } else if (delta.type === "citations_delta" && "citation" in delta) {
+          const citation = delta.citation;
+          if (citation && acc !== void 0) {
+            acc.citations.push(citation);
+          }
+        }
+        break;
+      }
+      case "content_block_stop":
+        finalizeContentBlock(
+          event.index,
+          contentBlocks,
+          contentBlockDeltas,
+          fallbackTextDeltas
+        );
+        break;
+      case "message_delta":
         if (event.usage) {
           const finalMetrics = parseMetricsFromUsage2(event.usage);
           metrics = { ...metrics, ...finalMetrics };
@@ -9447,15 +10027,24 @@ function finalizeContentBlock(index, contentBlocks, contentBlockDeltas, fallback
   }
   const acc = contentBlockDeltas[index];
   const text = acc?.textDeltas.join("") ?? "";
-  if (isToolUseContentBlock(contentBlock)) {
+  if (isToolUseLikeContentBlock(contentBlock)) {
     if (!text) {
       return;
     }
     try {
-      contentBlocks[index] = {
-        ...contentBlock,
-        input: JSON.parse(text)
+      const parsedInput = JSON.parse(text);
+      if (!isObject(parsedInput)) {
+        fallbackTextDeltas.push(text);
+        delete contentBlocks[index];
+        return;
+      }
+      const parsedToolUseBlock = {
+        type: contentBlock.type,
+        id: contentBlock.id,
+        name: contentBlock.name,
+        input: parsedInput
       };
+      contentBlocks[index] = parsedToolUseBlock;
     } catch {
       fallbackTextDeltas.push(text);
       delete contentBlocks[index];
@@ -9489,8 +10078,8 @@ function finalizeContentBlock(index, contentBlocks, contentBlockDeltas, fallback
 function isTextContentBlock(contentBlock) {
   return contentBlock.type === "text";
 }
-function isToolUseContentBlock(contentBlock) {
-  return contentBlock.type === "tool_use";
+function isToolUseLikeContentBlock(contentBlock) {
+  return (contentBlock.type === "tool_use" || contentBlock.type === "server_tool_use") && typeof contentBlock.id === "string" && typeof contentBlock.name === "string" && isObject(contentBlock.input);
 }
 function isThinkingContentBlock(contentBlock) {
   return contentBlock.type === "thinking";
@@ -9688,6 +10277,10 @@ var aiSDKChannels = defineChannels("ai", {
     channelName: "embedMany",
     kind: "async"
   }),
+  rerank: channel({
+    channelName: "rerank",
+    kind: "async"
+  }),
   agentGenerate: channel({
     channelName: "Agent.generate",
     kind: "async"
@@ -9747,7 +10340,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.generateText, {
         name: "generateText",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         extractOutput: (result, endEvent) => {
           finalizeAISDKChildTracing(endEvent);
@@ -9763,7 +10356,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.streamText, {
         name: "streamText",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         extractOutput: (result, endEvent) => processAISDKOutput(
           result,
@@ -9783,7 +10376,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceSyncStreamChannel(aiSDKChannels.streamTextSync, {
         name: "streamText",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         patchResult: ({ endEvent, result, span, startTime }) => patchAISDKStreamingResult({
           defaultDenyOutputPaths: denyOutputPaths,
@@ -9797,7 +10390,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.generateObject, {
         name: "generateObject",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         extractOutput: (result, endEvent) => {
           finalizeAISDKChildTracing(endEvent);
@@ -9813,7 +10406,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.streamObject, {
         name: "streamObject",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         extractOutput: (result, endEvent) => processAISDKOutput(
           result,
@@ -9833,7 +10426,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceSyncStreamChannel(aiSDKChannels.streamObjectSync, {
         name: "streamObject",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         patchResult: ({ endEvent, result, span, startTime }) => patchAISDKStreamingResult({
           defaultDenyOutputPaths: denyOutputPaths,
@@ -9847,7 +10440,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceAsyncChannel(aiSDKChannels.embed, {
         name: "embed",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event) => prepareAISDKEmbedInput(params, event.self),
         extractOutput: (result, endEvent) => processAISDKEmbeddingOutput(
           result,
@@ -9859,7 +10452,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceAsyncChannel(aiSDKChannels.embedMany, {
         name: "embedMany",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event) => prepareAISDKEmbedInput(params, event.self),
         extractOutput: (result, endEvent) => processAISDKEmbeddingOutput(
           result,
@@ -9868,10 +10461,22 @@ var AISDKPlugin = class extends BasePlugin {
         extractMetrics: (result, _startTime, endEvent) => extractTopLevelAISDKMetrics(result, endEvent)
       })
     );
+    this.unsubscribers.push(
+      traceAsyncChannel(aiSDKChannels.rerank, {
+        name: "rerank",
+        type: "function" /* FUNCTION */,
+        extractInput: ([params], event) => prepareAISDKRerankInput(params, event.self),
+        extractOutput: (result, endEvent) => processAISDKRerankOutput(
+          result,
+          resolveDenyOutputPaths(endEvent, denyOutputPaths)
+        ),
+        extractMetrics: (result, _startTime, endEvent) => extractTopLevelAISDKMetrics(result, endEvent)
+      })
+    );
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.agentGenerate, {
         name: "Agent.generate",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         extractOutput: (result, endEvent) => {
           finalizeAISDKChildTracing(endEvent);
@@ -9887,7 +10492,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.agentStream, {
         name: "Agent.stream",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         extractOutput: (result, endEvent) => processAISDKOutput(
           result,
@@ -9907,7 +10512,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceSyncStreamChannel(aiSDKChannels.agentStreamSync, {
         name: "Agent.stream",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         patchResult: ({ endEvent, result, span, startTime }) => patchAISDKStreamingResult({
           defaultDenyOutputPaths: denyOutputPaths,
@@ -9921,7 +10526,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.toolLoopAgentGenerate, {
         name: "ToolLoopAgent.generate",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         extractOutput: (result, endEvent) => {
           finalizeAISDKChildTracing(endEvent);
@@ -9937,7 +10542,7 @@ var AISDKPlugin = class extends BasePlugin {
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.toolLoopAgentStream, {
         name: "ToolLoopAgent.stream",
-        type: "llm" /* LLM */,
+        type: "function" /* FUNCTION */,
         extractInput: ([params], event, span) => prepareAISDKCallInput(params, event, span, denyOutputPaths),
         extractOutput: (result, endEvent) => processAISDKOutput(
           result,
@@ -10287,6 +10892,16 @@ function prepareAISDKEmbedInput(params, self) {
     metadata: extractMetadataFromEmbedParams(params, self)
   };
 }
+function prepareAISDKRerankInput(params, self) {
+  const { documents, query } = params;
+  return {
+    input: {
+      documents,
+      query
+    },
+    metadata: extractMetadataFromRerankParams(params, self)
+  };
+}
 function extractTopLevelAISDKMetrics(result, event, startTime) {
   const metrics = hasModelChildTracing(event) ? {} : extractTokenMetrics(result);
   if (startTime) {
@@ -10332,6 +10947,16 @@ function extractMetadataFromCallParams(params, self) {
 function extractMetadataFromEmbedParams(params, self) {
   return extractBaseMetadata(params.model, self);
 }
+function extractMetadataFromRerankParams(params, self) {
+  const metadata = extractBaseMetadata(params.model, self);
+  if (typeof params.topN === "number") {
+    metadata.topN = params.topN;
+  }
+  if (Array.isArray(params.documents)) {
+    metadata.document_count = params.documents.length;
+  }
+  return metadata;
+}
 function prepareAISDKChildTracing(params, self, parentSpan, denyOutputPaths, aiSDK) {
   const cleanup = [];
   const patchedModels = /* @__PURE__ */ new WeakSet();
@@ -10879,6 +11504,22 @@ function processAISDKEmbeddingOutput(output, denyOutputPaths) {
   }
   return normalizeAISDKLoggedOutput(omit(summarized, denyOutputPaths));
 }
+function processAISDKRerankOutput(output, _denyOutputPaths) {
+  if (!output || typeof output !== "object") {
+    return output;
+  }
+  const ranking = safeSerializableFieldRead(output, "ranking");
+  if (Array.isArray(ranking)) {
+    return ranking.slice(0, 100).map((item) => {
+      const entry = item && typeof item === "object" ? item : void 0;
+      return {
+        index: typeof entry?.originalIndex === "number" ? entry.originalIndex : void 0,
+        relevance_score: typeof entry?.score === "number" ? entry.score : void 0
+      };
+    });
+  }
+  return void 0;
+}
 function extractTokenMetrics(result) {
   const metrics = {};
   let usage;
@@ -10921,12 +11562,80 @@ function extractTokenMetrics(result) {
   if (totalTokens !== void 0) {
     metrics.tokens = totalTokens;
   }
+  const promptCachedTokens = firstNumber(
+    usage.inputTokens?.cacheRead,
+    usage.inputTokenDetails?.cacheReadTokens,
+    usage.cachedInputTokens,
+    usage.promptCachedTokens,
+    usage.prompt_cached_tokens
+  );
+  if (promptCachedTokens !== void 0) {
+    metrics.prompt_cached_tokens = promptCachedTokens;
+  }
+  const promptCacheCreationTokens = firstNumber(
+    usage.inputTokens?.cacheWrite,
+    usage.inputTokenDetails?.cacheWriteTokens,
+    usage.promptCacheCreationTokens,
+    usage.prompt_cache_creation_tokens,
+    extractAnthropicCacheCreationTokens(result)
+  );
+  if (promptCacheCreationTokens !== void 0) {
+    metrics.prompt_cache_creation_tokens = promptCacheCreationTokens;
+  }
+  const promptReasoningTokens = firstNumber(
+    usage.promptReasoningTokens,
+    usage.prompt_reasoning_tokens
+  );
+  if (promptReasoningTokens !== void 0) {
+    metrics.prompt_reasoning_tokens = promptReasoningTokens;
+  }
+  const completionCachedTokens = firstNumber(
+    usage.completionCachedTokens,
+    usage.completion_cached_tokens
+  );
+  if (completionCachedTokens !== void 0) {
+    metrics.completion_cached_tokens = completionCachedTokens;
+  }
+  const reasoningTokenCount = firstNumber(
+    usage.outputTokens?.reasoning,
+    usage.reasoningTokens,
+    usage.completionReasoningTokens,
+    usage.completion_reasoning_tokens,
+    usage.reasoning_tokens,
+    usage.thinkingTokens,
+    usage.thinking_tokens
+  );
+  if (reasoningTokenCount !== void 0) {
+    metrics.completion_reasoning_tokens = reasoningTokenCount;
+    metrics.reasoning_tokens = reasoningTokenCount;
+  }
+  const completionAudioTokens = firstNumber(
+    usage.completionAudioTokens,
+    usage.completion_audio_tokens
+  );
+  if (completionAudioTokens !== void 0) {
+    metrics.completion_audio_tokens = completionAudioTokens;
+  }
   const cost = extractCostFromResult(result);
   if (cost !== void 0) {
     metrics.estimated_cost = cost;
   }
   return metrics;
 }
+function extractAnthropicCacheCreationTokens(result) {
+  const providerMetadata = safeSerializableFieldRead(
+    result,
+    "providerMetadata"
+  );
+  const anthropicMetadata = providerMetadata?.anthropic;
+  if (!anthropicMetadata) {
+    return void 0;
+  }
+  return firstNumber(
+    anthropicMetadata.cacheCreationInputTokens,
+    anthropicMetadata.usage?.cache_creation_input_tokens
+  );
+}
 function safeResultFieldRead(result, field) {
   return safeSerializableFieldRead(result, field);
 }
@@ -11043,14 +11752,11 @@ function extractSerializableOutputFields(output) {
     ...extractGetterValues(output)
   };
 }
-function isPromiseLike(value) {
-  return value != null && typeof value === "object" && typeof value.then === "function";
-}
 function isSerializableOutputValue(value) {
   if (typeof value === "function") {
     return false;
   }
-  if (value && typeof value === "object" && typeof value.then === "function") {
+  if (isPromiseLike(value)) {
     return false;
   }
   if (value && typeof value === "object" && typeof value.getReader === "function") {
@@ -11370,12 +12076,9 @@ function bindClaudeLocalToolContextToAsyncIterable(result, localToolContext) {
 var LOCAL_TOOL_HANDLER_WRAPPED = Symbol.for(
   "braintrust.claude_agent_sdk.local_tool_handler_wrapped"
 );
-function toErrorMessage(error) {
+function toErrorMessage2(error) {
   return error instanceof Error ? error.message : String(error);
 }
-function isPromiseLike2(value) {
-  return value !== null && typeof value === "object" && "then" in value && typeof value.then === "function";
-}
 function getToolUseIdFromExtra(extra) {
   if (!extra || typeof extra !== "object" || !("_meta" in extra)) {
     return void 0;
@@ -11422,14 +12125,14 @@ function wrapLocalClaudeToolHandler(handler, getMetadata) {
         return result;
       };
       const finalizeError = (error) => {
-        span.log({ error: toErrorMessage(error) });
+        span.log({ error: toErrorMessage2(error) });
         span.end();
         throw error;
       };
       return withCurrent(span, () => {
         try {
           const result = runHandler();
-          if (isPromiseLike2(result)) {
+          if (isPromiseLike(result)) {
             return result.then(finalizeSuccess, finalizeError);
           }
           return finalizeSuccess(result);
@@ -11549,7 +12252,7 @@ var ROOT_LLM_PARENT_KEY = "__root__";
 function llmParentKey(parentToolUseId) {
   return parentToolUseId ?? ROOT_LLM_PARENT_KEY;
 }
-function isSubAgentToolName(toolName) {
+function isSubAgentDelegationToolName(toolName) {
   return toolName === "Agent" || toolName === "Task";
 }
 function filterSerializableOptions(options) {
@@ -11583,24 +12286,92 @@ function getNumberProperty(obj, key) {
   const value = Reflect.get(obj, key);
   return typeof value === "number" ? value : void 0;
 }
-function extractUsageFromMessage(message) {
-  const metrics = {};
-  let usage;
-  if (message.type === "assistant") {
-    usage = message.message?.usage;
-  } else if (message.type === "result") {
-    usage = message.usage;
+function getStringProperty(obj, key) {
+  if (!obj || typeof obj !== "object" || !(key in obj)) {
+    return void 0;
   }
-  if (!usage || typeof usage !== "object") {
-    return metrics;
+  const value = Reflect.get(obj, key);
+  return typeof value === "string" ? value : void 0;
+}
+function upsertSubAgentDetails(detailsByToolUseId, toolUseId, update) {
+  const existing = detailsByToolUseId.get(toolUseId) ?? {};
+  const merged = {
+    ...existing,
+    ...Object.fromEntries(
+      Object.entries(update).filter(([, value]) => value !== void 0)
+    )
+  };
+  detailsByToolUseId.set(toolUseId, merged);
+  return merged;
+}
+function formatSubAgentSpanName(details) {
+  if (details?.description) {
+    return `Agent: ${details.description}`;
   }
-  const inputTokens = getNumberProperty(usage, "input_tokens");
-  if (inputTokens !== void 0) {
-    metrics.prompt_tokens = inputTokens;
+  if (details?.agentType) {
+    return `Agent: ${details.agentType}`;
   }
-  const outputTokens = getNumberProperty(usage, "output_tokens");
-  if (outputTokens !== void 0) {
-    metrics.completion_tokens = outputTokens;
+  return "Agent: sub-agent";
+}
+function subAgentDetailsToMetadata(details) {
+  if (!details) {
+    return {};
+  }
+  const metadata = {};
+  if (details.agentId) {
+    metadata["claude_agent_sdk.agent_id"] = details.agentId;
+  }
+  if (details.agentType) {
+    metadata["claude_agent_sdk.agent_type"] = details.agentType;
+  }
+  if (details.description) {
+    metadata["claude_agent_sdk.description"] = details.description;
+  }
+  if (details.taskId) {
+    metadata["claude_agent_sdk.task_id"] = details.taskId;
+  }
+  if (details.taskType) {
+    metadata["claude_agent_sdk.task_type"] = details.taskType;
+  }
+  if (details.toolUseId) {
+    metadata["claude_agent_sdk.tool_use_id"] = details.toolUseId;
+  }
+  if (details.workflowName) {
+    metadata["claude_agent_sdk.workflow_name"] = details.workflowName;
+  }
+  return metadata;
+}
+function resolveTaskToolUseId(taskIdToToolUseId, message) {
+  const messageToolUseId = typeof message.tool_use_id === "string" ? message.tool_use_id : void 0;
+  if (messageToolUseId && typeof message.task_id === "string") {
+    taskIdToToolUseId.set(message.task_id, messageToolUseId);
+  }
+  if (messageToolUseId) {
+    return messageToolUseId;
+  }
+  if (typeof message.task_id === "string") {
+    return taskIdToToolUseId.get(message.task_id);
+  }
+  return void 0;
+}
+function extractUsageFromMessage(message) {
+  const metrics = {};
+  let usage;
+  if (message.type === "assistant") {
+    usage = message.message?.usage;
+  } else if (message.type === "result") {
+    usage = message.usage;
+  }
+  if (!usage || typeof usage !== "object") {
+    return metrics;
+  }
+  const inputTokens = getNumberProperty(usage, "input_tokens");
+  if (inputTokens !== void 0) {
+    metrics.prompt_tokens = inputTokens;
+  }
+  const outputTokens = getNumberProperty(usage, "output_tokens");
+  if (outputTokens !== void 0) {
+    metrics.completion_tokens = outputTokens;
   }
   const cacheReadTokens = getNumberProperty(usage, "cache_read_input_tokens") || 0;
   const cacheCreationTokens = getNumberProperty(usage, "cache_creation_input_tokens") || 0;
@@ -11755,7 +12526,7 @@ function prepareLocalToolHandlersInMcpServers(mcpServers) {
   }
   return { hasLocalToolHandlers, localToolHookNames };
 }
-function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers, localToolHookNames, skipLocalToolHooks, subAgentSpans, endedSubAgentSpans) {
+function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers, localToolHookNames, skipLocalToolHooks, subAgentDetailsByToolUseId, subAgentSpans, endedSubAgentSpans) {
   const preToolUse = async (input, toolUseID) => {
     if (input.hook_event_name !== "PreToolUse" || !toolUseID) {
       return {};
@@ -11763,9 +12534,6 @@ function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers,
     if (skipLocalToolHooks && (isLocalToolUse(input.tool_name, mcpServers) || localToolHookNames.has(input.tool_name))) {
       return {};
     }
-    if (isSubAgentToolName(input.tool_name)) {
-      return {};
-    }
     const parsed = parseToolName(input.tool_name);
     const toolSpan = startSpan({
       event: {
@@ -11795,10 +12563,18 @@ function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers,
       return {};
     }
     const subAgentSpan = subAgentSpans.get(toolUseID);
+    const toolSpan = activeToolSpans.get(toolUseID);
     if (subAgentSpan) {
+      if (endedSubAgentSpans.has(toolUseID)) {
+        return {};
+      }
       try {
         const response = input.tool_response;
-        const metadata = {};
+        const metadata = {
+          ...subAgentDetailsToMetadata(
+            subAgentDetailsByToolUseId.get(toolUseID)
+          )
+        };
         if (response?.status) {
           metadata["claude_agent_sdk.status"] = response.status;
         }
@@ -11816,9 +12592,16 @@ function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers,
         subAgentSpan.end();
         endedSubAgentSpans.add(toolUseID);
       }
+      if (toolSpan) {
+        try {
+          toolSpan.log({ output: input.tool_response });
+        } finally {
+          toolSpan.end();
+          activeToolSpans.delete(toolUseID);
+        }
+      }
       return {};
     }
-    const toolSpan = activeToolSpans.get(toolUseID);
     if (!toolSpan) {
       return {};
     }
@@ -11838,16 +12621,43 @@ function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers,
       return {};
     }
     const subAgentSpan = subAgentSpans.get(toolUseID);
+    const toolSpan = activeToolSpans.get(toolUseID);
     if (subAgentSpan) {
+      if (endedSubAgentSpans.has(toolUseID)) {
+        return {};
+      }
       try {
-        subAgentSpan.log({ error: input.error });
+        subAgentSpan.log({
+          error: input.error,
+          metadata: subAgentDetailsToMetadata(
+            subAgentDetailsByToolUseId.get(toolUseID)
+          )
+        });
       } finally {
         subAgentSpan.end();
         endedSubAgentSpans.add(toolUseID);
       }
+      if (toolSpan) {
+        const parsed2 = parseToolName(input.tool_name);
+        try {
+          toolSpan.log({
+            error: input.error,
+            metadata: {
+              "claude_agent_sdk.is_interrupt": input.is_interrupt,
+              "claude_agent_sdk.session_id": input.session_id,
+              "claude_agent_sdk.raw_tool_name": parsed2.rawToolName,
+              "gen_ai.tool.call.id": toolUseID,
+              "gen_ai.tool.name": parsed2.toolName,
+              ...parsed2.mcpServer && { "mcp.server": parsed2.mcpServer }
+            }
+          });
+        } finally {
+          toolSpan.end();
+          activeToolSpans.delete(toolUseID);
+        }
+      }
       return {};
     }
-    const toolSpan = activeToolSpans.get(toolUseID);
     if (!toolSpan) {
       return {};
     }
@@ -11869,15 +12679,84 @@ function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers,
     }
     return {};
   };
-  return { postToolUse, postToolUseFailure, preToolUse };
+  const subagentStart = async (input, toolUseID) => {
+    if (input.hook_event_name !== "SubagentStart" || !toolUseID) {
+      return {};
+    }
+    const details = upsertSubAgentDetails(
+      subAgentDetailsByToolUseId,
+      toolUseID,
+      {
+        agentId: input.agent_id,
+        agentType: input.agent_type,
+        toolUseId: toolUseID
+      }
+    );
+    const subAgentSpan = subAgentSpans.get(toolUseID);
+    if (subAgentSpan && !endedSubAgentSpans.has(toolUseID)) {
+      subAgentSpan.log({
+        metadata: subAgentDetailsToMetadata(details)
+      });
+    }
+    return {};
+  };
+  const subagentStop = async (input, toolUseID) => {
+    if (input.hook_event_name !== "SubagentStop" || !toolUseID) {
+      return {};
+    }
+    const details = upsertSubAgentDetails(
+      subAgentDetailsByToolUseId,
+      toolUseID,
+      {
+        agentId: input.agent_id,
+        agentType: input.agent_type,
+        toolUseId: toolUseID
+      }
+    );
+    const subAgentSpan = subAgentSpans.get(toolUseID);
+    if (!subAgentSpan || endedSubAgentSpans.has(toolUseID)) {
+      return {};
+    }
+    const metadata = {
+      ...subAgentDetailsToMetadata(details),
+      ...input.agent_transcript_path && {
+        "claude_agent_sdk.agent_transcript_path": input.agent_transcript_path
+      },
+      "claude_agent_sdk.stop_hook_active": input.stop_hook_active
+    };
+    try {
+      subAgentSpan.log({
+        metadata,
+        output: input.last_assistant_message
+      });
+    } finally {
+      subAgentSpan.end();
+      endedSubAgentSpans.add(toolUseID);
+    }
+    return {};
+  };
+  return {
+    postToolUse,
+    postToolUseFailure,
+    preToolUse,
+    subagentStart,
+    subagentStop
+  };
 }
-function injectTracingHooks(options, resolveParentSpan, activeToolSpans, localToolHookNames, skipLocalToolHooks, subAgentSpans, endedSubAgentSpans) {
-  const { preToolUse, postToolUse, postToolUseFailure } = createToolTracingHooks(
+function injectTracingHooks(options, resolveParentSpan, activeToolSpans, localToolHookNames, skipLocalToolHooks, subAgentDetailsByToolUseId, subAgentSpans, endedSubAgentSpans) {
+  const {
+    preToolUse,
+    postToolUse,
+    postToolUseFailure,
+    subagentStart,
+    subagentStop
+  } = createToolTracingHooks(
     resolveParentSpan,
     activeToolSpans,
     options.mcpServers,
     localToolHookNames,
     skipLocalToolHooks,
+    subAgentDetailsByToolUseId,
     subAgentSpans,
     endedSubAgentSpans
   );
@@ -11899,6 +12778,18 @@ function injectTracingHooks(options, resolveParentSpan, activeToolSpans, localTo
       PreToolUse: [
         ...existingHooks.PreToolUse ?? [],
         { hooks: [preToolUse] }
+      ],
+      SubagentStart: [
+        ...existingHooks.SubagentStart ?? [],
+        {
+          hooks: [subagentStart]
+        }
+      ],
+      SubagentStop: [
+        ...existingHooks.SubagentStop ?? [],
+        {
+          hooks: [subagentStop]
+        }
       ]
     }
   };
@@ -11927,7 +12818,6 @@ async function finalizeCurrentMessageGroup(state) {
     parentSpan,
     existingLlmSpan
   );
-  state.activeLlmSpansByParentToolUse.delete(parentKey);
   if (llmSpanResult) {
     if (parentToolUseId) {
       state.latestLlmParentBySubAgentToolUse.set(
@@ -11941,6 +12831,7 @@ async function finalizeCurrentMessageGroup(state) {
       state.finalResults.push(llmSpanResult.finalMessage);
     }
   }
+  state.activeLlmSpansByParentToolUse.delete(parentKey);
   const lastMessage = state.currentMessages[state.currentMessages.length - 1];
   if (lastMessage?.message?.usage) {
     state.accumulatedOutputTokens += getNumberProperty(lastMessage.message.usage, "output_tokens") || 0;
@@ -11957,8 +12848,12 @@ function maybeTrackToolUseContext(state, message) {
       continue;
     }
     state.toolUseToParent.set(block.id, parentToolUseId);
-    if (block.name === "Task" && typeof block.input === "object" && block.input !== null && "subagent_type" in block.input && typeof block.input.subagent_type === "string") {
-      state.pendingSubAgentNames.set(block.id, block.input.subagent_type);
+    if (typeof block.name === "string" && isSubAgentDelegationToolName(block.name) && typeof block.input === "object" && block.input !== null) {
+      upsertSubAgentDetails(state.subAgentDetailsByToolUseId, block.id, {
+        agentType: getStringProperty(block.input, "subagent_type"),
+        description: getStringProperty(block.input, "description"),
+        toolUseId: block.id
+      });
     }
   }
 }
@@ -11971,34 +12866,133 @@ async function maybeStartSubAgentSpan(state, message) {
     return;
   }
   await ensureSubAgentSpan(
-    state.pendingSubAgentNames,
+    state.subAgentDetailsByToolUseId,
     state.span,
+    state.activeToolSpans,
     state.subAgentSpans,
     parentToolUseId
   );
 }
-async function ensureSubAgentSpan(pendingSubAgentNames, rootSpan, subAgentSpans, parentToolUseId) {
+async function ensureSubAgentSpan(subAgentDetailsByToolUseId, rootSpan, activeToolSpans, subAgentSpans, parentToolUseId) {
   const existingSpan = subAgentSpans.get(parentToolUseId);
   if (existingSpan) {
     return existingSpan;
   }
-  const agentName = pendingSubAgentNames.get(parentToolUseId);
-  const spanName = agentName ? `Agent: ${agentName}` : "Agent: sub-agent";
+  const details = subAgentDetailsByToolUseId.get(parentToolUseId);
+  const spanName = formatSubAgentSpanName(details);
+  const parentToolSpan = activeToolSpans.get(parentToolUseId);
+  const parentSpan = parentToolSpan ? await parentToolSpan.export() : await rootSpan.export();
   const subAgentSpan = startSpan({
     event: {
-      metadata: {
-        ...agentName && { "claude_agent_sdk.agent_type": agentName }
-      }
+      metadata: subAgentDetailsToMetadata(details)
     },
     name: spanName,
-    parent: await rootSpan.export(),
+    parent: parentSpan,
     spanAttributes: { type: "task" /* TASK */ }
   });
   subAgentSpans.set(parentToolUseId, subAgentSpan);
   return subAgentSpan;
 }
+async function maybeHandleTaskLifecycleMessage(state, message) {
+  if (message.type !== "system") {
+    return false;
+  }
+  if (message.subtype !== "task_started" && message.subtype !== "task_progress" && message.subtype !== "task_notification") {
+    return false;
+  }
+  const toolUseId = resolveTaskToolUseId(state.taskIdToToolUseId, message);
+  if (!toolUseId) {
+    return true;
+  }
+  const details = upsertSubAgentDetails(
+    state.subAgentDetailsByToolUseId,
+    toolUseId,
+    {
+      description: getStringProperty(message, "description"),
+      taskId: getStringProperty(message, "task_id"),
+      taskType: getStringProperty(message, "task_type"),
+      toolUseId,
+      workflowName: getStringProperty(message, "workflow_name")
+    }
+  );
+  const subAgentSpan = await ensureSubAgentSpan(
+    state.subAgentDetailsByToolUseId,
+    state.span,
+    state.activeToolSpans,
+    state.subAgentSpans,
+    toolUseId
+  );
+  if (state.endedSubAgentSpans.has(toolUseId)) {
+    return true;
+  }
+  const usage = message.usage;
+  const usageTotalTokens = getNumberProperty(usage, "total_tokens");
+  const usageToolUses = getNumberProperty(usage, "tool_uses");
+  const usageDurationMs = getNumberProperty(usage, "duration_ms");
+  const metadata = {
+    ...subAgentDetailsToMetadata(details),
+    "claude_agent_sdk.tool_use_id": toolUseId,
+    ...usageTotalTokens !== void 0 && {
+      "claude_agent_sdk.total_tokens": usageTotalTokens
+    },
+    ...usageToolUses !== void 0 && {
+      "claude_agent_sdk.tool_use_count": usageToolUses
+    },
+    ...usageDurationMs !== void 0 && {
+      "claude_agent_sdk.duration_ms": usageDurationMs
+    }
+  };
+  if (message.subtype === "task_started") {
+    const prompt = getStringProperty(message, "prompt");
+    subAgentSpan.log({
+      input: prompt,
+      metadata
+    });
+    return true;
+  }
+  const summary = getStringProperty(message, "summary");
+  if (summary) {
+    metadata["claude_agent_sdk.summary"] = summary;
+  }
+  if (message.subtype === "task_progress") {
+    const lastToolName = getStringProperty(message, "last_tool_name");
+    if (lastToolName) {
+      metadata["claude_agent_sdk.last_tool_name"] = lastToolName;
+    }
+    subAgentSpan.log({
+      metadata,
+      output: summary
+    });
+    return true;
+  }
+  const status = getStringProperty(message, "status");
+  const outputFile = getStringProperty(message, "output_file");
+  if (status) {
+    metadata["claude_agent_sdk.task_status"] = status;
+  }
+  if (outputFile) {
+    metadata["claude_agent_sdk.output_file"] = outputFile;
+  }
+  const output = summary || outputFile ? {
+    ...summary && { summary },
+    ...outputFile && { output_file: outputFile }
+  } : void 0;
+  try {
+    subAgentSpan.log({
+      metadata,
+      output
+    });
+  } finally {
+    subAgentSpan.end();
+    state.endedSubAgentSpans.add(toolUseId);
+  }
+  return true;
+}
 async function handleStreamMessage(state, message) {
   maybeTrackToolUseContext(state, message);
+  if (await maybeHandleTaskLifecycleMessage(state, message)) {
+    return;
+  }
   await maybeStartSubAgentSpan(state, message);
   const messageId = message.message?.id;
   if (messageId && messageId !== state.currentMessageId) {
@@ -12013,8 +13007,9 @@ async function handleStreamMessage(state, message) {
       let llmParentSpan = await state.span.export();
       if (parentToolUseId) {
         const subAgentSpan = await ensureSubAgentSpan(
-          state.pendingSubAgentNames,
+          state.subAgentDetailsByToolUseId,
           state.span,
+          state.activeToolSpans,
           state.subAgentSpans,
           parentToolUseId
         );
@@ -12094,6 +13089,10 @@ async function finalizeQuerySpan(state) {
       llmSpan.end();
     }
     state.activeLlmSpansByParentToolUse.clear();
+    for (const toolSpan of state.activeToolSpans.values()) {
+      toolSpan.end();
+    }
+    state.activeToolSpans.clear();
     for (const [id, subAgentSpan] of state.subAgentSpans) {
       if (!state.endedSubAgentSpans.has(id)) {
         subAgentSpan.end();
@@ -12167,7 +13166,8 @@ var ClaudeAgentSDKPlugin = class extends BasePlugin {
         const latestRootLlmParentRef = {
           value: void 0
         };
-        const pendingSubAgentNames = /* @__PURE__ */ new Map();
+        const subAgentDetailsByToolUseId = /* @__PURE__ */ new Map();
+        const taskIdToToolUseId = /* @__PURE__ */ new Map();
         const localToolContext = createClaudeLocalToolContext();
         const { hasLocalToolHandlers, localToolHookNames } = prepareLocalToolHandlersInMcpServers(options.mcpServers);
         const skipLocalToolHooks = options[CLAUDE_AGENT_SDK_SKIP_LOCAL_TOOL_HOOKS_OPTION] === true || hasLocalToolHandlers;
@@ -12184,8 +13184,9 @@ var ClaudeAgentSDKPlugin = class extends BasePlugin {
               return parentLlm;
             }
             const subAgentSpan = await ensureSubAgentSpan(
-              pendingSubAgentNames,
+              subAgentDetailsByToolUseId,
               span,
+              activeToolSpans,
               subAgentSpans,
               parentToolUseId
             );
@@ -12204,6 +13205,7 @@ var ClaudeAgentSDKPlugin = class extends BasePlugin {
           activeToolSpans,
           localToolHookNames,
           skipLocalToolHooks,
+          subAgentDetailsByToolUseId,
           subAgentSpans,
           endedSubAgentSpans
         );
@@ -12221,12 +13223,13 @@ var ClaudeAgentSDKPlugin = class extends BasePlugin {
           finalResults: [],
           options: optionsWithHooks,
           originalPrompt,
-          pendingSubAgentNames,
           processing: Promise.resolve(),
           promptDone,
           promptStarted: () => promptStarted,
           span,
+          subAgentDetailsByToolUseId,
           subAgentSpans,
+          taskIdToToolUseId,
           latestLlmParentBySubAgentToolUse,
           latestRootLlmParentRef,
           toolUseToParent,
@@ -12308,6 +13311,10 @@ var googleGenAIChannels = defineChannels("@google/genai", {
   generateContentStream: channel({
     channelName: "models.generateContentStream",
     kind: "async"
+  }),
+  embedContent: channel({
+    channelName: "models.embedContent",
+    kind: "async"
   })
 });
 
@@ -12334,6 +13341,7 @@ var GoogleGenAIPlugin = class extends BasePlugin {
   subscribeToGoogleGenAIChannels() {
     this.subscribeToGenerateContentChannel();
     this.subscribeToGenerateContentStreamChannel();
+    this.subscribeToEmbedContentChannel();
   }
   subscribeToGenerateContentChannel() {
     const tracingChannel = googleGenAIChannels.generateContent.tracingChannel();
@@ -12343,8 +13351,8 @@ var GoogleGenAIPlugin = class extends BasePlugin {
       states,
       (event) => {
         const params = event.arguments[0];
-        const input = serializeInput(params);
-        const metadata = extractMetadata(params);
+        const input = serializeGenerateContentInput(params);
+        const metadata = extractGenerateContentMetadata(params);
         const span = startSpan({
           name: "generate_content",
           spanAttributes: {
@@ -12362,8 +13370,8 @@ var GoogleGenAIPlugin = class extends BasePlugin {
       start: (event) => {
         ensureSpanState(states, event, () => {
           const params = event.arguments[0];
-          const input = serializeInput(params);
-          const metadata = extractMetadata(params);
+          const input = serializeGenerateContentInput(params);
+          const metadata = extractGenerateContentMetadata(params);
           const span = startSpan({
             name: "generate_content",
             spanAttributes: {
@@ -12383,7 +13391,9 @@ var GoogleGenAIPlugin = class extends BasePlugin {
           return;
         }
         try {
+          const responseMetadata = extractResponseMetadata(event.result);
           spanState.span.log({
+            ...responseMetadata ? { metadata: responseMetadata } : {},
             metrics: cleanMetrics(
               extractGenerateContentMetrics(
                 event.result,
@@ -12413,8 +13423,8 @@ var GoogleGenAIPlugin = class extends BasePlugin {
       start: (event) => {
         const streamEvent = event;
         const params = event.arguments[0];
-        streamEvent.googleGenAIInput = serializeInput(params);
-        streamEvent.googleGenAIMetadata = extractMetadata(params);
+        streamEvent.googleGenAIInput = serializeGenerateContentInput(params);
+        streamEvent.googleGenAIMetadata = extractGenerateContentMetadata(params);
         streamEvent.googleGenAIStartTime = getCurrentUnixTimestamp();
       },
       asyncEnd: (event) => {
@@ -12434,6 +13444,76 @@ var GoogleGenAIPlugin = class extends BasePlugin {
       tracingChannel.unsubscribe(handlers);
     });
   }
+  subscribeToEmbedContentChannel() {
+    const tracingChannel = googleGenAIChannels.embedContent.tracingChannel();
+    const states = /* @__PURE__ */ new WeakMap();
+    const unbindCurrentSpanStore = bindCurrentSpanStoreToStart2(
+      tracingChannel,
+      states,
+      (event) => {
+        const params = event.arguments[0];
+        const input = serializeEmbedContentInput(params);
+        const metadata = extractEmbedContentMetadata(params);
+        const span = startSpan({
+          name: "embed_content",
+          spanAttributes: {
+            type: "llm" /* LLM */
+          },
+          event: createWrapperParityEvent({ input, metadata })
+        });
+        return {
+          span,
+          startTime: getCurrentUnixTimestamp()
+        };
+      }
+    );
+    const handlers = {
+      start: (event) => {
+        ensureSpanState(states, event, () => {
+          const params = event.arguments[0];
+          const input = serializeEmbedContentInput(params);
+          const metadata = extractEmbedContentMetadata(params);
+          const span = startSpan({
+            name: "embed_content",
+            spanAttributes: {
+              type: "llm" /* LLM */
+            },
+            event: createWrapperParityEvent({ input, metadata })
+          });
+          return {
+            span,
+            startTime: getCurrentUnixTimestamp()
+          };
+        });
+      },
+      asyncEnd: (event) => {
+        const spanState = states.get(event);
+        if (!spanState) {
+          return;
+        }
+        try {
+          const output = summarizeEmbedContentOutput(event.result);
+          spanState.span.log({
+            ...output ? { output } : {},
+            metrics: cleanMetrics(
+              extractEmbedContentMetrics(event.result, spanState.startTime)
+            )
+          });
+        } finally {
+          spanState.span.end();
+          states.delete(event);
+        }
+      },
+      error: (event) => {
+        logErrorAndEndSpan(states, event);
+      }
+    };
+    tracingChannel.subscribe(handlers);
+    this.unsubscribers.push(() => {
+      unbindCurrentSpanStore?.();
+      tracingChannel.unsubscribe(handlers);
+    });
+  }
 };
 function ensureSpanState(states, event, create) {
   const existing = states.get(event);
@@ -12507,7 +13587,11 @@ function patchGoogleGenAIStreamingResult(args) {
     finalized = true;
     if (options.result) {
       const { end, ...metricsWithoutEnd } = options.result.metrics;
+      const responseMetadata = extractResponseMetadata(
+        options.result.aggregated
+      );
       span.log({
+        ...responseMetadata ? { metadata: responseMetadata } : {},
         metrics: cleanMetrics(metricsWithoutEnd),
         output: options.result.aggregated
       });
@@ -12610,26 +13694,35 @@ function patchGoogleGenAIStreamingResult(args) {
   patchIterator(result);
   return true;
 }
-function serializeInput(params) {
+function serializeGenerateContentInput(params) {
   const input = {
     model: params.model,
-    contents: serializeContents(params.contents)
+    contents: serializeContentCollection(params.contents)
   };
-  if (params.config) {
-    const config = tryToDict(params.config);
-    if (config) {
-      const filteredConfig = {};
-      Object.keys(config).forEach((key) => {
-        if (key !== "tools") {
-          filteredConfig[key] = config[key];
-        }
-      });
-      input.config = filteredConfig;
-    }
+  const config = params.config ? tryToDict(params.config) : null;
+  if (config) {
+    const filteredConfig = {};
+    Object.keys(config).forEach((key) => {
+      if (key !== "tools") {
+        filteredConfig[key] = config[key];
+      }
+    });
+    input.config = filteredConfig;
+  }
+  return input;
+}
+function serializeEmbedContentInput(params) {
+  const input = {
+    model: params.model,
+    contents: serializeContentCollection(params.contents)
+  };
+  const config = params.config ? tryToDict(params.config) : null;
+  if (config) {
+    input.config = config;
   }
   return input;
 }
-function serializeContents(contents) {
+function serializeContentCollection(contents) {
   if (contents === null || contents === void 0) {
     return null;
   }
@@ -12681,22 +13774,26 @@ function serializePart(part) {
   }
   return part;
 }
-function serializeTools(params) {
-  if (!params.config?.tools) {
+function serializeGenerateContentTools(params) {
+  const config = params.config ? tryToDict(params.config) : null;
+  const tools = config?.tools;
+  if (!Array.isArray(tools)) {
     return null;
   }
   try {
-    return params.config.tools.map((tool) => {
-      if (typeof tool === "object" && tool.functionDeclarations) {
-        return tool;
+    const serializedTools = [];
+    for (const tool of tools) {
+      const toolDict = tryToDict(tool);
+      if (toolDict) {
+        serializedTools.push(toolDict);
       }
-      return tool;
-    });
+    }
+    return serializedTools.length > 0 ? serializedTools : null;
   } catch {
     return null;
   }
 }
-function extractMetadata(params) {
+function extractGenerateContentMetadata(params) {
   const metadata = {};
   if (params.model) {
     metadata.model = params.model;
@@ -12711,12 +13808,25 @@ function extractMetadata(params) {
       });
     }
   }
-  const tools = serializeTools(params);
+  const tools = serializeGenerateContentTools(params);
   if (tools) {
     metadata.tools = tools;
   }
   return metadata;
 }
+function extractEmbedContentMetadata(params) {
+  const metadata = {};
+  if (params.model) {
+    metadata.model = params.model;
+  }
+  const config = params.config ? tryToDict(params.config) : null;
+  if (config) {
+    Object.keys(config).forEach((key) => {
+      metadata[key] = config[key];
+    });
+  }
+  return metadata;
+}
 function extractGenerateContentMetrics(response, startTime) {
   const metrics = {};
   if (startTime !== void 0) {
@@ -12730,24 +13840,87 @@ function extractGenerateContentMetrics(response, startTime) {
   }
   return metrics;
 }
-function populateUsageMetrics(metrics, usage) {
-  if (usage.promptTokenCount !== void 0) {
-    metrics.prompt_tokens = usage.promptTokenCount;
+function extractEmbedContentMetrics(response, startTime) {
+  const metrics = {};
+  if (startTime !== void 0) {
+    const end = getCurrentUnixTimestamp();
+    metrics.start = startTime;
+    metrics.end = end;
+    metrics.duration = end - startTime;
   }
-  if (usage.candidatesTokenCount !== void 0) {
-    metrics.completion_tokens = usage.candidatesTokenCount;
+  if (response?.usageMetadata) {
+    populateUsageMetrics(metrics, response.usageMetadata);
   }
-  if (usage.totalTokenCount !== void 0) {
-    metrics.tokens = usage.totalTokenCount;
+  const embeddingTokenCount = extractEmbedPromptTokenCount(response);
+  if (embeddingTokenCount !== void 0) {
+    metrics.prompt_tokens = embeddingTokenCount;
+    metrics.tokens = embeddingTokenCount;
   }
-  if (usage.cachedContentTokenCount !== void 0) {
-    metrics.prompt_cached_tokens = usage.cachedContentTokenCount;
+  return metrics;
+}
+function extractEmbedPromptTokenCount(response) {
+  if (!response) {
+    return void 0;
   }
-  if (usage.thoughtsTokenCount !== void 0) {
-    metrics.completion_reasoning_tokens = usage.thoughtsTokenCount;
+  const usagePromptTokens = response.usageMetadata?.promptTokenCount;
+  if (typeof usagePromptTokens === "number" && Number.isFinite(usagePromptTokens)) {
+    return usagePromptTokens;
+  }
+  const usageTotalTokens = response.usageMetadata?.totalTokenCount;
+  if (typeof usageTotalTokens === "number" && Number.isFinite(usageTotalTokens)) {
+    return usageTotalTokens;
+  }
+  const embeddings = Array.isArray(response.embeddings) ? response.embeddings : response.embedding ? [response.embedding] : [];
+  if (embeddings.length === 0) {
+    return void 0;
+  }
+  let total = 0;
+  let sawAny = false;
+  for (const embedding of embeddings) {
+    const embeddingStats = tryToDict(tryToDict(embedding)?.statistics);
+    const tokenCount = embeddingStats?.tokenCount;
+    if (typeof tokenCount === "number" && Number.isFinite(tokenCount)) {
+      total += tokenCount;
+      sawAny = true;
+    }
   }
+  return sawAny ? total : void 0;
 }
-function aggregateGenerateContentChunks(chunks, startTime, firstTokenTime) {
+function summarizeEmbedContentOutput(response) {
+  if (!response) {
+    return void 0;
+  }
+  const embeddings = Array.isArray(response.embeddings) ? response.embeddings : response.embedding ? [response.embedding] : [];
+  if (embeddings.length === 0) {
+    return void 0;
+  }
+  const firstValues = embeddings[0]?.values;
+  if (!Array.isArray(firstValues)) {
+    return void 0;
+  }
+  return {
+    embedding_count: embeddings.length,
+    embedding_length: firstValues.length
+  };
+}
+function populateUsageMetrics(metrics, usage) {
+  if (usage.promptTokenCount !== void 0) {
+    metrics.prompt_tokens = usage.promptTokenCount;
+  }
+  if (usage.candidatesTokenCount !== void 0) {
+    metrics.completion_tokens = usage.candidatesTokenCount;
+  }
+  if (usage.totalTokenCount !== void 0) {
+    metrics.tokens = usage.totalTokenCount;
+  }
+  if (usage.cachedContentTokenCount !== void 0) {
+    metrics.prompt_cached_tokens = usage.cachedContentTokenCount;
+  }
+  if (usage.thoughtsTokenCount !== void 0) {
+    metrics.completion_reasoning_tokens = usage.thoughtsTokenCount;
+  }
+}
+function aggregateGenerateContentChunks(chunks, startTime, firstTokenTime) {
   const end = getCurrentUnixTimestamp();
   const metrics = {
     start: startTime,
@@ -12763,6 +13936,7 @@ function aggregateGenerateContentChunks(chunks, startTime, firstTokenTime) {
   let text = "";
   let thoughtText = "";
   const otherParts = [];
+  let groundingMetadata = void 0;
   let usageMetadata = null;
   let lastResponse = null;
   for (const chunk of chunks) {
@@ -12770,6 +13944,9 @@ function aggregateGenerateContentChunks(chunks, startTime, firstTokenTime) {
     if (chunk.usageMetadata) {
       usageMetadata = chunk.usageMetadata;
     }
+    if (chunk.groundingMetadata !== void 0) {
+      groundingMetadata = chunk.groundingMetadata;
+    }
     if (chunk.candidates && Array.isArray(chunk.candidates)) {
       for (const candidate of chunk.candidates) {
         if (candidate.content?.parts) {
@@ -12815,6 +13992,12 @@ function aggregateGenerateContentChunks(chunks, startTime, firstTokenTime) {
       if (candidate.finishReason !== void 0) {
         candidateDict.finishReason = candidate.finishReason;
       }
+      if (candidate.groundingMetadata !== void 0) {
+        candidateDict.groundingMetadata = candidate.groundingMetadata;
+        if (groundingMetadata === void 0) {
+          groundingMetadata = candidate.groundingMetadata;
+        }
+      }
       if (candidate.safetyRatings) {
         candidateDict.safetyRatings = candidate.safetyRatings;
       }
@@ -12826,6 +14009,9 @@ function aggregateGenerateContentChunks(chunks, startTime, firstTokenTime) {
     aggregated.usageMetadata = usageMetadata;
     populateUsageMetrics(metrics, usageMetadata);
   }
+  if (groundingMetadata !== void 0) {
+    aggregated.groundingMetadata = groundingMetadata;
+  }
   if (text) {
     aggregated.text = text;
   }
@@ -12840,6 +14026,31 @@ function cleanMetrics(metrics) {
   }
   return cleaned;
 }
+function extractResponseMetadata(response) {
+  const responseDict = tryToDict(response);
+  if (!responseDict) {
+    return void 0;
+  }
+  const metadata = {};
+  const responseGroundingMetadata = responseDict.groundingMetadata;
+  const candidateGroundingMetadata = [];
+  if (Array.isArray(responseDict.candidates)) {
+    for (const candidate of responseDict.candidates) {
+      const candidateDict = tryToDict(candidate);
+      if (candidateDict?.groundingMetadata !== void 0) {
+        candidateGroundingMetadata.push(candidateDict.groundingMetadata);
+      }
+    }
+  }
+  if (responseGroundingMetadata !== void 0) {
+    metadata.groundingMetadata = responseGroundingMetadata;
+  } else if (candidateGroundingMetadata.length === 1) {
+    [metadata.groundingMetadata] = candidateGroundingMetadata;
+  } else if (candidateGroundingMetadata.length > 1) {
+    metadata.groundingMetadata = candidateGroundingMetadata;
+  }
+  return Object.keys(metadata).length > 0 ? metadata : void 0;
+}
 function tryToDict(obj) {
   if (obj === null || obj === void 0) {
     return null;
@@ -12854,6 +14065,385 @@ function tryToDict(obj) {
   return null;
 }
 
+// src/instrumentation/plugins/huggingface-channels.ts
+var huggingFaceChannels = defineChannels("@huggingface/inference", {
+  chatCompletion: channel({
+    channelName: "chatCompletion",
+    kind: "async"
+  }),
+  chatCompletionStream: channel({
+    channelName: "chatCompletionStream",
+    kind: "sync-stream"
+  }),
+  textGeneration: channel({
+    channelName: "textGeneration",
+    kind: "async"
+  }),
+  textGenerationStream: channel({
+    channelName: "textGenerationStream",
+    kind: "sync-stream"
+  }),
+  featureExtraction: channel({
+    channelName: "featureExtraction",
+    kind: "async"
+  })
+});
+
+// src/instrumentation/plugins/huggingface-plugin.ts
+var REQUEST_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
+  "dimensions",
+  "encoding_format",
+  "endpointUrl",
+  "max_tokens",
+  "model",
+  "provider",
+  "seed",
+  "stop",
+  "stream",
+  "temperature",
+  "top_p"
+]);
+var RESPONSE_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
+  "created",
+  "id",
+  "model",
+  "object"
+]);
+var HuggingFacePlugin = class extends BasePlugin {
+  onEnable() {
+    this.unsubscribers.push(
+      traceAsyncChannel(huggingFaceChannels.chatCompletion, {
+        name: "huggingface.chat_completion",
+        type: "llm" /* LLM */,
+        extractInput: extractChatInputWithMetadata,
+        extractOutput: (result) => result?.choices,
+        extractMetadata: (result) => extractResponseMetadata2(result),
+        extractMetrics: (result) => parseMetricsFromUsage(result?.usage)
+      }),
+      traceSyncStreamChannel(huggingFaceChannels.chatCompletionStream, {
+        name: "huggingface.chat_completion_stream",
+        type: "llm" /* LLM */,
+        extractInput: extractChatInputWithMetadata,
+        patchResult: ({ result, span, startTime }) => patchChatCompletionStream({
+          result,
+          span,
+          startTime
+        })
+      }),
+      traceAsyncChannel(huggingFaceChannels.textGeneration, {
+        name: "huggingface.text_generation",
+        type: "llm" /* LLM */,
+        extractInput: extractTextGenerationInputWithMetadata,
+        extractOutput: (result) => isObject(result) ? { generated_text: result.generated_text } : result,
+        extractMetadata: extractTextGenerationMetadata,
+        extractMetrics: (result) => extractTextGenerationMetrics(result?.details ?? null)
+      }),
+      traceSyncStreamChannel(huggingFaceChannels.textGenerationStream, {
+        name: "huggingface.text_generation_stream",
+        type: "llm" /* LLM */,
+        extractInput: extractTextGenerationInputWithMetadata,
+        patchResult: ({ result, span, startTime }) => patchTextGenerationStream({
+          result,
+          span,
+          startTime
+        })
+      }),
+      traceAsyncChannel(huggingFaceChannels.featureExtraction, {
+        name: "huggingface.feature_extraction",
+        type: "llm" /* LLM */,
+        extractInput: extractFeatureExtractionInputWithMetadata,
+        extractOutput: summarizeFeatureExtractionOutput,
+        extractMetrics: () => ({})
+      })
+    );
+  }
+  onDisable() {
+    this.unsubscribers = unsubscribeAll(this.unsubscribers);
+  }
+};
+function addProviderMetadata(metadata) {
+  return {
+    ...metadata,
+    provider: metadata.provider ?? "huggingface"
+  };
+}
+function normalizeArgs(args) {
+  if (Array.isArray(args)) {
+    return args;
+  }
+  if (isArrayLike(args)) {
+    return Array.from(args);
+  }
+  return [args];
+}
+function isArrayLike(value) {
+  return isObject(value) && "length" in value && typeof value.length === "number" && Number.isInteger(value.length) && value.length >= 0;
+}
+function getFirstObjectArg(args) {
+  const firstObjectArg = normalizeArgs(args).find((arg) => isObject(arg));
+  return isObject(firstObjectArg) ? firstObjectArg : void 0;
+}
+function pickRequestMetadata(params) {
+  if (!params) {
+    return addProviderMetadata({});
+  }
+  const metadata = {};
+  for (const key of REQUEST_METADATA_ALLOWLIST) {
+    const value = params[key];
+    if (value !== void 0) {
+      metadata[key] = value;
+    }
+  }
+  if (isObject(params.parameters)) {
+    metadata.parameters = params.parameters;
+  }
+  return addProviderMetadata(metadata);
+}
+function extractChatInputWithMetadata(args) {
+  const params = getFirstObjectArg(args);
+  const { messages, ...rawMetadata } = params ?? {};
+  return {
+    input: messages,
+    metadata: pickRequestMetadata(rawMetadata)
+  };
+}
+function extractTextGenerationInputWithMetadata(args) {
+  const params = getFirstObjectArg(args);
+  const { inputs, ...rawMetadata } = params ?? {};
+  return {
+    input: inputs,
+    metadata: pickRequestMetadata(rawMetadata)
+  };
+}
+function extractFeatureExtractionInputWithMetadata(args) {
+  const params = getFirstObjectArg(args);
+  const { inputs, ...rawMetadata } = params ?? {};
+  return {
+    input: inputs,
+    metadata: pickRequestMetadata(rawMetadata)
+  };
+}
+function extractResponseMetadata2(result) {
+  if (!isObject(result)) {
+    return void 0;
+  }
+  const metadata = {};
+  for (const key of RESPONSE_METADATA_ALLOWLIST) {
+    const value = result[key];
+    if (value !== void 0) {
+      metadata[key] = value;
+    }
+  }
+  return Object.keys(metadata).length > 0 ? metadata : void 0;
+}
+function extractTextGenerationMetrics(details) {
+  if (!isObject(details)) {
+    return {};
+  }
+  const promptTokens = Array.isArray(details.prefill) ? details.prefill.length : void 0;
+  const completionTokens = typeof details.generated_tokens === "number" ? details.generated_tokens : Array.isArray(details.tokens) ? details.tokens.length : void 0;
+  const metrics = {};
+  if (promptTokens !== void 0) {
+    metrics.prompt_tokens = promptTokens;
+  }
+  if (completionTokens !== void 0) {
+    metrics.completion_tokens = completionTokens;
+  }
+  if (promptTokens !== void 0 || completionTokens !== void 0) {
+    metrics.tokens = (promptTokens ?? 0) + (completionTokens ?? 0);
+  }
+  return metrics;
+}
+function extractTextGenerationMetadata(result) {
+  if (!isObject(result?.details)) {
+    return void 0;
+  }
+  return typeof result.details.finish_reason === "string" ? {
+    finish_reason: result.details.finish_reason
+  } : void 0;
+}
+function summarizeFeatureExtractionOutput(result) {
+  if (!Array.isArray(result)) {
+    return void 0;
+  }
+  const first = result[0];
+  if (typeof first === "number") {
+    return { embedding_length: result.length };
+  }
+  if (Array.isArray(first) && first.every((value) => typeof value === "number")) {
+    return {
+      embedding_count: result.length,
+      embedding_length: first.length
+    };
+  }
+  if (Array.isArray(first) && first.length > 0 && Array.isArray(first[0]) && first[0].every((value) => typeof value === "number")) {
+    return {
+      embedding_batch_count: result.length,
+      embedding_count: first.length,
+      embedding_length: first[0].length
+    };
+  }
+  return void 0;
+}
+function patchChatCompletionStream(args) {
+  const { result, span, startTime } = args;
+  if (!result || !isAsyncIterable(result)) {
+    return false;
+  }
+  let firstChunkTime;
+  patchStreamIfNeeded(result, {
+    onChunk: () => {
+      if (firstChunkTime === void 0) {
+        firstChunkTime = getCurrentUnixTimestamp();
+      }
+    },
+    onComplete: (chunks) => {
+      const lastChunk = chunks.at(-1);
+      const responseMetadata = extractResponseMetadata2(lastChunk);
+      const metrics = {
+        ...parseMetricsFromUsage(lastChunk?.usage),
+        ...firstChunkTime !== void 0 ? { time_to_first_token: firstChunkTime - startTime } : {}
+      };
+      span.log({
+        output: aggregateChatCompletionChunks2(chunks),
+        ...responseMetadata ? { metadata: responseMetadata } : {},
+        metrics
+      });
+      span.end();
+    },
+    onError: (error) => {
+      span.log({
+        error: error.message
+      });
+      span.end();
+    }
+  });
+  return true;
+}
+function patchTextGenerationStream(args) {
+  const { result, span, startTime } = args;
+  if (!result || !isAsyncIterable(result)) {
+    return false;
+  }
+  let firstChunkTime;
+  patchStreamIfNeeded(result, {
+    onChunk: () => {
+      if (firstChunkTime === void 0) {
+        firstChunkTime = getCurrentUnixTimestamp();
+      }
+    },
+    onComplete: (chunks) => {
+      const lastChunk = chunks.at(-1);
+      const streamMetadata = extractTextGenerationStreamMetadata(chunks);
+      span.log({
+        output: aggregateTextGenerationStreamChunks(chunks),
+        ...streamMetadata ? { metadata: streamMetadata } : {},
+        metrics: {
+          ...extractTextGenerationMetrics(lastChunk?.details ?? null),
+          ...parseMetricsFromUsage(lastChunk?.usage),
+          ...firstChunkTime !== void 0 ? { time_to_first_token: firstChunkTime - startTime } : {}
+        }
+      });
+      span.end();
+    },
+    onError: (error) => {
+      span.log({
+        error: error.message
+      });
+      span.end();
+    }
+  });
+  return true;
+}
+function aggregateChatCompletionChunks2(chunks) {
+  if (chunks.length === 0) {
+    return void 0;
+  }
+  const aggregatedChoices = /* @__PURE__ */ new Map();
+  for (const chunk of chunks) {
+    for (const choice of chunk.choices ?? []) {
+      const index = typeof choice.index === "number" ? choice.index : 0;
+      const existing = aggregatedChoices.get(index) ?? { content: "" };
+      const delta = isObject(choice.delta) ? choice.delta : void 0;
+      const message = isObject(choice.message) ? choice.message : void 0;
+      if (typeof delta?.content === "string") {
+        existing.content += delta.content;
+      } else if (typeof message?.content === "string") {
+        existing.content = message.content;
+      }
+      if (typeof delta?.role === "string") {
+        existing.role = delta.role;
+      } else if (typeof message?.role === "string") {
+        existing.role = message.role;
+      }
+      if (choice.finish_reason !== void 0) {
+        existing.finish_reason = choice.finish_reason;
+      }
+      aggregatedChoices.set(index, existing);
+    }
+  }
+  return {
+    choices: [...aggregatedChoices.entries()].map(([index, choice]) => ({
+      index,
+      message: {
+        content: choice.content,
+        role: choice.role ?? "assistant"
+      },
+      ...choice.finish_reason !== void 0 ? { finish_reason: choice.finish_reason } : {}
+    }))
+  };
+}
+function aggregateTextGenerationStreamChunks(chunks) {
+  if (chunks.length === 0) {
+    return void 0;
+  }
+  let generatedText = "";
+  let finishReason;
+  for (const chunk of chunks) {
+    if (typeof chunk.generated_text === "string") {
+      generatedText = chunk.generated_text;
+    } else if (typeof chunk.token?.text === "string" && !chunk.token.special) {
+      generatedText += chunk.token.text;
+    } else if (Array.isArray(chunk.choices)) {
+      for (const choice of chunk.choices) {
+        if (typeof choice.text === "string") {
+          generatedText += choice.text;
+        }
+        if (choice.finish_reason !== void 0) {
+          finishReason = choice.finish_reason;
+        }
+      }
+    }
+    if (isObject(chunk.details) && typeof chunk.details.finish_reason === "string") {
+      finishReason = chunk.details.finish_reason;
+    }
+  }
+  return {
+    generated_text: generatedText,
+    ...finishReason !== void 0 ? { finish_reason: finishReason } : {}
+  };
+}
+function extractTextGenerationStreamMetadata(chunks) {
+  for (let index = chunks.length - 1; index >= 0; index--) {
+    const chunk = chunks[index];
+    if (isObject(chunk?.details) && typeof chunk.details.finish_reason === "string") {
+      return {
+        finish_reason: chunk.details.finish_reason
+      };
+    }
+    if (!Array.isArray(chunk?.choices)) {
+      continue;
+    }
+    for (let choiceIndex = chunk.choices.length - 1; choiceIndex >= 0; choiceIndex--) {
+      const choice = chunk.choices[choiceIndex];
+      if (choice?.finish_reason !== void 0) {
+        return { finish_reason: choice.finish_reason };
+      }
+    }
+  }
+  return void 0;
+}
+
 // src/instrumentation/plugins/openrouter-agent-channels.ts
 var openRouterAgentChannels = defineChannels("@openrouter/agent", {
   callModel: channel({
@@ -12971,20 +14561,20 @@ var OpenRouterAgentPlugin = class extends BasePlugin {
     });
   }
 };
-function normalizeArgs(args) {
+function normalizeArgs2(args) {
   if (Array.isArray(args)) {
     return args;
   }
-  if (isArrayLike(args)) {
+  if (isArrayLike2(args)) {
     return Array.from(args);
   }
   return [args];
 }
-function isArrayLike(value) {
+function isArrayLike2(value) {
   return isObject(value) && "length" in value && typeof value.length === "number" && Number.isInteger(value.length) && value.length >= 0;
 }
 function getOpenRouterCallModelRequestArg(args) {
-  const normalizedArgs = normalizeArgs(args);
+  const normalizedArgs = normalizeArgs2(args);
   const keyedRequestArg = normalizedArgs.find(
     (arg) => isObject(arg) && ("input" in arg || "model" in arg || "tools" in arg)
   );
@@ -13270,7 +14860,7 @@ function traceToolExecution(args) {
   }
 }
 function publishToolResult(tracingChannel, event, result) {
-  if (isPromiseLike3(result)) {
+  if (isPromiseLike(result)) {
     return result.then(
       (resolved) => {
         event.result = resolved;
@@ -13292,9 +14882,6 @@ function getToolCallId(context) {
   const toolContext = context;
   return typeof toolContext?.toolCall?.id === "string" ? toolContext.toolCall.id : void 0;
 }
-function isPromiseLike3(value) {
-  return !!value && (typeof value === "object" || typeof value === "function") && "then" in value && typeof value.then === "function";
-}
 var OPENROUTER_WRAPPED_CALL_MODEL_RESULT = Symbol(
   "braintrust.openrouter.wrappedCallModelResult"
 );
@@ -13649,7 +15236,13 @@ var openRouterChannels = defineChannels("@openrouter/sdk", {
     channelName: "embeddings.generate",
     kind: "async"
   }),
-  betaResponsesSend: channel({
+  rerankRerank: channel(
+    {
+      channelName: "rerank.rerank",
+      kind: "async"
+    }
+  ),
+  betaResponsesSend: channel({
     channelName: "beta.responses.send",
     kind: "async"
   }),
@@ -13741,6 +15334,34 @@ var OpenRouterPlugin = class extends BasePlugin {
         }
       })
     );
+    this.unsubscribers.push(
+      traceAsyncChannel(openRouterChannels.rerankRerank, {
+        name: "openrouter.rerank.rerank",
+        type: "llm" /* LLM */,
+        extractInput: (args) => {
+          const request = getOpenRouterRequestArg(args);
+          const requestBody = isObject(request?.requestBody) ? request.requestBody : {};
+          const httpReferer = request?.httpReferer;
+          const xTitle = request?.xTitle ?? request?.appTitle;
+          const { documents, query, ...metadata } = requestBody;
+          return {
+            input: {
+              documents,
+              query
+            },
+            metadata: buildOpenRouterRerankMetadata(
+              metadata,
+              documents,
+              httpReferer,
+              xTitle
+            )
+          };
+        },
+        extractOutput: (result) => extractOpenRouterRerankOutput(result),
+        extractMetadata: (result) => extractOpenRouterResponseMetadata2(result),
+        extractMetrics: (result) => isObject(result) ? parseOpenRouterMetricsFromUsage2(result.usage) : {}
+      })
+    );
     this.unsubscribers.push(
       traceStreamingChannel(openRouterChannels.betaResponsesSend, {
         name: "openrouter.beta.responses.send",
@@ -13860,20 +15481,20 @@ var OpenRouterPlugin = class extends BasePlugin {
     });
   }
 };
-function normalizeArgs2(args) {
+function normalizeArgs3(args) {
   if (Array.isArray(args)) {
     return args;
   }
-  if (isArrayLike2(args)) {
+  if (isArrayLike3(args)) {
     return Array.from(args);
   }
   return [args];
 }
-function isArrayLike2(value) {
+function isArrayLike3(value) {
   return isObject(value) && "length" in value && typeof value.length === "number" && Number.isInteger(value.length) && value.length >= 0;
 }
 function getOpenRouterRequestArg(args) {
-  const normalizedArgs = normalizeArgs2(args);
+  const normalizedArgs = normalizeArgs3(args);
   const keyedCandidate = normalizedArgs.find(
     (arg) => isObject(arg) && ("chatGenerationParams" in arg || "requestBody" in arg || "openResponsesRequest" in arg)
   );
@@ -13884,7 +15505,7 @@ function getOpenRouterRequestArg(args) {
   return isObject(firstObjectArg) ? firstObjectArg : void 0;
 }
 function getOpenRouterCallModelRequestArg2(args) {
-  const firstObjectArg = normalizeArgs2(args).find((arg) => isObject(arg));
+  const firstObjectArg = normalizeArgs3(args).find((arg) => isObject(arg));
   return isObject(firstObjectArg) ? firstObjectArg : void 0;
 }
 var TOKEN_NAME_MAP3 = {
@@ -14054,6 +15675,13 @@ function buildOpenRouterEmbeddingMetadata(metadata, httpReferer, xTitle) {
     embedding_model: normalized.model
   } : normalized;
 }
+function buildOpenRouterRerankMetadata(metadata, documents, httpReferer, xTitle) {
+  const normalized = buildOpenRouterMetadata2(metadata, httpReferer, xTitle);
+  return {
+    ...normalized,
+    ...Array.isArray(documents) ? { document_count: documents.length } : {}
+  };
+}
 function extractOpenRouterCallModelInput2(request) {
   return isObject(request) && "input" in request ? sanitizeOpenRouterLoggedValue2(request.input) : void 0;
 }
@@ -14073,7 +15701,7 @@ function extractOpenRouterResponseMetadata2(result) {
   const metadataRecord = isObject(sanitized) ? sanitized : {};
   const { model, provider, ...rest } = metadataRecord;
   const normalizedModel = parseOpenRouterModelString2(model);
-  const normalizedProvider = (typeof provider === "string" ? provider : void 0) || normalizedModel.provider;
+  const normalizedProvider = (typeof provider === "string" ? provider.toLowerCase() : void 0) || normalizedModel.provider;
   const usageMetadata = extractOpenRouterUsageMetadata2(usage);
   const combined = {
     ...rest,
@@ -14092,6 +15720,15 @@ function extractOpenRouterResponseOutput2(response, fallbackOutput) {
   }
   return void 0;
 }
+function extractOpenRouterRerankOutput(result) {
+  if (!isObject(result) || !Array.isArray(result.results)) {
+    return void 0;
+  }
+  return result.results.slice(0, 100).map((item) => ({
+    index: isObject(item) && typeof item.index === "number" ? item.index : void 0,
+    relevance_score: isObject(item) && typeof item.relevanceScore === "number" ? item.relevanceScore : isObject(item) && typeof item.relevance_score === "number" ? item.relevance_score : void 0
+  }));
+}
 var OPENROUTER_WRAPPED_TOOL2 = Symbol("braintrust.openrouter.wrappedTool");
 function patchOpenRouterCallModelRequestTools2(request) {
   if (!Array.isArray(request.tools) || request.tools.length === 0) {
@@ -14162,7 +15799,7 @@ function traceToolExecution2(args) {
   }
 }
 function publishToolResult2(tracingChannel, event, result) {
-  if (isPromiseLike4(result)) {
+  if (isPromiseLike(result)) {
     return result.then(
       (resolved) => {
         event.result = resolved;
@@ -14184,9 +15821,6 @@ function getToolCallId2(context) {
   const toolContext = context;
   return typeof toolContext?.toolCall?.id === "string" ? toolContext.toolCall.id : void 0;
 }
-function isPromiseLike4(value) {
-  return !!value && (typeof value === "object" || typeof value === "function") && "then" in value && typeof value.then === "function";
-}
 function aggregateOpenRouterChatChunks(chunks) {
   let role;
   let content = "";
@@ -14826,20 +16460,20 @@ var MISTRAL_RESPONSE_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
 function camelToSnake3(value) {
   return value.replace(/[A-Z]/g, (match) => `_${match.toLowerCase()}`);
 }
-function normalizeArgs3(args) {
+function normalizeArgs4(args) {
   if (Array.isArray(args)) {
     return args;
   }
-  if (isArrayLike3(args)) {
+  if (isArrayLike4(args)) {
     return Array.from(args);
   }
   return [args];
 }
-function isArrayLike3(value) {
+function isArrayLike4(value) {
   return isObject(value) && "length" in value && typeof value.length === "number" && Number.isInteger(value.length) && value.length >= 0;
 }
 function getMistralRequestArg(args) {
-  const firstObjectArg = normalizeArgs3(args).find((arg) => isObject(arg));
+  const firstObjectArg = normalizeArgs4(args).find((arg) => isObject(arg));
   return isObject(firstObjectArg) ? firstObjectArg : void 0;
 }
 function addMistralProviderMetadata(metadata) {
@@ -15144,88 +16778,1326 @@ function aggregateMistralStreamChunks(chunks) {
   };
 }
 
-// src/instrumentation/braintrust-plugin.ts
-var BraintrustPlugin = class extends BasePlugin {
-  config;
-  openaiPlugin = null;
-  anthropicPlugin = null;
-  aiSDKPlugin = null;
-  claudeAgentSDKPlugin = null;
-  googleGenAIPlugin = null;
-  openRouterPlugin = null;
-  openRouterAgentPlugin = null;
-  mistralPlugin = null;
-  constructor(config = {}) {
-    super();
-    this.config = config;
-  }
+// 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/instrumentation/plugins/google-adk-plugin.ts
+var GoogleADKPlugin = class extends BasePlugin {
+  activeRunnerSpans = /* @__PURE__ */ new Map();
+  activeAgentSpans = /* @__PURE__ */ new Map();
   onEnable() {
-    const integrations = this.config.integrations || {};
-    if (integrations.openai !== false) {
-      this.openaiPlugin = new OpenAIPlugin();
-      this.openaiPlugin.enable();
-    }
-    if (integrations.anthropic !== false) {
-      this.anthropicPlugin = new AnthropicPlugin();
-      this.anthropicPlugin.enable();
-    }
-    if (integrations.aisdk !== false && integrations.vercel !== false) {
-      this.aiSDKPlugin = new AISDKPlugin();
-      this.aiSDKPlugin.enable();
-    }
-    if (integrations.claudeAgentSDK !== false) {
-      this.claudeAgentSDKPlugin = new ClaudeAgentSDKPlugin();
-      this.claudeAgentSDKPlugin.enable();
-    }
-    if (integrations.googleGenAI !== false && integrations.google !== false) {
-      this.googleGenAIPlugin = new GoogleGenAIPlugin();
-      this.googleGenAIPlugin.enable();
-    }
-    if (integrations.openrouter !== false) {
-      this.openRouterPlugin = new OpenRouterPlugin();
-      this.openRouterPlugin.enable();
-    }
-    if (integrations.openrouterAgent !== false) {
-      this.openRouterAgentPlugin = new OpenRouterAgentPlugin();
-      this.openRouterAgentPlugin.enable();
-    }
-    if (integrations.mistral !== false) {
-      this.mistralPlugin = new MistralPlugin();
-      this.mistralPlugin.enable();
-    }
+    this.subscribeToRunnerRunAsync();
+    this.subscribeToAgentRunAsync();
+    this.subscribeToToolRunAsync();
   }
   onDisable() {
-    if (this.openaiPlugin) {
-      this.openaiPlugin.disable();
-      this.openaiPlugin = null;
-    }
-    if (this.anthropicPlugin) {
-      this.anthropicPlugin.disable();
-      this.anthropicPlugin = null;
-    }
-    if (this.aiSDKPlugin) {
-      this.aiSDKPlugin.disable();
-      this.aiSDKPlugin = null;
-    }
-    if (this.claudeAgentSDKPlugin) {
-      this.claudeAgentSDKPlugin.disable();
-      this.claudeAgentSDKPlugin = null;
-    }
-    if (this.googleGenAIPlugin) {
-      this.googleGenAIPlugin.disable();
-      this.googleGenAIPlugin = null;
-    }
-    if (this.openRouterPlugin) {
-      this.openRouterPlugin.disable();
-      this.openRouterPlugin = null;
-    }
-    if (this.openRouterAgentPlugin) {
-      this.openRouterAgentPlugin.disable();
-      this.openRouterAgentPlugin = null;
+    for (const unsubscribe of this.unsubscribers) {
+      unsubscribe();
     }
-    if (this.mistralPlugin) {
-      this.mistralPlugin.disable();
-      this.mistralPlugin = null;
+    this.unsubscribers = [];
+    this.activeRunnerSpans.clear();
+    this.activeAgentSpans.clear();
+  }
+  subscribeToRunnerRunAsync() {
+    const tracingChannel = googleADKChannels.runnerRunAsync.tracingChannel();
+    const states = /* @__PURE__ */ new WeakMap();
+    const createState = (event) => {
+      const params = event.arguments[0] ?? {};
+      const contextKey = extractRunnerContextKey(params);
+      const span = startSpan({
+        name: "Google ADK Runner",
+        spanAttributes: {
+          type: "task" /* TASK */
+        }
+      });
+      const startTime = getCurrentUnixTimestamp();
+      try {
+        const metadata = extractRunnerMetadata(params);
+        span.log({
+          input: extractRunnerInput(params),
+          metadata
+        });
+      } catch {
+      }
+      if (contextKey) {
+        this.activeRunnerSpans.set(contextKey, span);
+      }
+      return { span, startTime, events: [], contextKey };
+    };
+    const unbindCurrentSpanStore = bindCurrentSpanStoreToStart3(
+      tracingChannel,
+      states,
+      createState
+    );
+    const handlers = {
+      start: (event) => {
+        ensureState(states, event, () => createState(event));
+      },
+      end: (event) => {
+        const state = states.get(event);
+        if (!state) {
+          return;
+        }
+        const result = event.result;
+        if (isAsyncIterable(result)) {
+          bindAsyncIterableToCurrentSpan(result, state.span);
+          patchStreamIfNeeded(result, {
+            onChunk: (adkEvent) => {
+              state.events.push(adkEvent);
+            },
+            onComplete: () => {
+              finalizeRunnerSpan(state, this.activeRunnerSpans);
+              states.delete(event);
+            },
+            onError: (error) => {
+              cleanupActiveRunnerSpan(state, this.activeRunnerSpans);
+              state.span.log({ error: error.message });
+              state.span.end();
+              states.delete(event);
+            }
+          });
+          return;
+        }
+        try {
+          state.span.log({ output: result });
+        } finally {
+          cleanupActiveRunnerSpan(state, this.activeRunnerSpans);
+          state.span.end();
+          states.delete(event);
+        }
+      },
+      error: (event) => {
+        const state = states.get(event);
+        if (!state || !event.error) {
+          return;
+        }
+        cleanupActiveRunnerSpan(state, this.activeRunnerSpans);
+        state.span.log({ error: event.error.message });
+        state.span.end();
+        states.delete(event);
+      }
+    };
+    tracingChannel.subscribe(handlers);
+    this.unsubscribers.push(() => {
+      unbindCurrentSpanStore?.();
+      tracingChannel.unsubscribe(handlers);
+    });
+  }
+  subscribeToAgentRunAsync() {
+    const tracingChannel = googleADKChannels.agentRunAsync.tracingChannel();
+    const states = /* @__PURE__ */ new WeakMap();
+    const createState = (event) => {
+      const parentContext = event.arguments[0];
+      const agentName = extractAgentName(parentContext);
+      const runnerParentSpan = findRunnerParentSpan(
+        parentContext,
+        this.activeRunnerSpans
+      );
+      const contextKey = extractInvocationContextKey(parentContext);
+      const span = startSpan({
+        name: agentName ? `Agent: ${agentName}` : "Google ADK Agent",
+        spanAttributes: {
+          type: "task" /* TASK */
+        },
+        ...runnerParentSpan ? {
+          parentSpanIds: {
+            spanId: runnerParentSpan.spanId,
+            rootSpanId: runnerParentSpan.rootSpanId
+          }
+        } : {}
+      });
+      const startTime = getCurrentUnixTimestamp();
+      try {
+        const metadata = {
+          provider: "google-adk"
+        };
+        if (agentName) {
+          metadata["google_adk.agent_name"] = agentName;
+        }
+        const modelName = extractModelName(parentContext);
+        if (modelName) {
+          metadata.model = modelName;
+        }
+        span.log({ metadata });
+      } catch {
+      }
+      if (contextKey && agentName) {
+        this.activeAgentSpans.set(agentContextKey(contextKey, agentName), span);
+      }
+      return { span, startTime, events: [], contextKey, name: agentName };
+    };
+    const unbindCurrentSpanStore = bindCurrentSpanStoreToStart3(
+      tracingChannel,
+      states,
+      createState
+    );
+    const handlers = {
+      start: (event) => {
+        ensureState(states, event, () => createState(event));
+      },
+      end: (event) => {
+        const state = states.get(event);
+        if (!state) {
+          return;
+        }
+        const result = event.result;
+        if (isAsyncIterable(result)) {
+          bindAsyncIterableToCurrentSpan(result, state.span);
+          patchStreamIfNeeded(result, {
+            onChunk: (adkEvent) => {
+              state.events.push(adkEvent);
+            },
+            onComplete: () => {
+              finalizeAgentSpan(state, this.activeAgentSpans);
+              states.delete(event);
+            },
+            onError: (error) => {
+              cleanupActiveAgentSpan(state, this.activeAgentSpans);
+              state.span.log({ error: error.message });
+              state.span.end();
+              states.delete(event);
+            }
+          });
+          return;
+        }
+        try {
+          state.span.log({ output: result });
+        } finally {
+          cleanupActiveAgentSpan(state, this.activeAgentSpans);
+          state.span.end();
+          states.delete(event);
+        }
+      },
+      error: (event) => {
+        const state = states.get(event);
+        if (!state || !event.error) {
+          return;
+        }
+        cleanupActiveAgentSpan(state, this.activeAgentSpans);
+        state.span.log({ error: event.error.message });
+        state.span.end();
+        states.delete(event);
+      }
+    };
+    tracingChannel.subscribe(handlers);
+    this.unsubscribers.push(() => {
+      unbindCurrentSpanStore?.();
+      tracingChannel.unsubscribe(handlers);
+    });
+  }
+  subscribeToToolRunAsync() {
+    const tracingChannel = googleADKChannels.toolRunAsync.tracingChannel();
+    const states = /* @__PURE__ */ new WeakMap();
+    const handlers = {
+      start: (event) => {
+        const req = event.arguments[0] ?? {};
+        const tool = event.self;
+        const toolName = extractToolName(req, tool);
+        const parentSpan = findToolParentSpan(
+          req,
+          this.activeAgentSpans,
+          this.activeRunnerSpans
+        );
+        const createSpan = () => startSpan({
+          name: toolName ? `tool: ${toolName}` : "Google ADK Tool",
+          spanAttributes: {
+            type: "tool" /* TOOL */
+          },
+          event: {
+            input: req.args,
+            metadata: {
+              provider: "google-adk",
+              ...toolName && { "google_adk.tool_name": toolName },
+              ...extractToolCallId(req) && {
+                "google_adk.tool_call_id": extractToolCallId(req)
+              }
+            }
+          }
+        });
+        const span = parentSpan ? withCurrent(parentSpan, () => createSpan()) : createSpan();
+        const startTime = getCurrentUnixTimestamp();
+        states.set(event, { span, startTime });
+      },
+      asyncEnd: (event) => {
+        const state = states.get(event);
+        if (!state) {
+          return;
+        }
+        try {
+          const metrics = {};
+          const end = getCurrentUnixTimestamp();
+          metrics.start = state.startTime;
+          metrics.end = end;
+          metrics.duration = end - state.startTime;
+          state.span.log({
+            output: event.result,
+            metrics: cleanMetrics2(metrics)
+          });
+        } finally {
+          state.span.end();
+          states.delete(event);
+        }
+      },
+      error: (event) => {
+        const state = states.get(event);
+        if (!state || !event.error) {
+          return;
+        }
+        state.span.log({ error: event.error.message });
+        state.span.end();
+        states.delete(event);
+      }
+    };
+    tracingChannel.subscribe(handlers);
+    this.unsubscribers.push(() => {
+      tracingChannel.unsubscribe(handlers);
+    });
+  }
+};
+function ensureState(states, event, create) {
+  const existing = states.get(event);
+  if (existing) {
+    return existing;
+  }
+  const created = create();
+  states.set(event, created);
+  return created;
+}
+function bindAsyncIterableToCurrentSpan(stream, span) {
+  if (!isAsyncIterable(stream)) {
+    return stream;
+  }
+  if (Object.isFrozen(stream) || Object.isSealed(stream)) {
+    return stream;
+  }
+  if ("next" in stream && typeof stream.next === "function") {
+    if ("__braintrust_current_span_bound" in stream) {
+      return stream;
+    }
+    try {
+      const iterator = stream;
+      const originalNext = iterator.next.bind(iterator);
+      iterator.next = (...args) => withCurrent(span, () => originalNext(...args));
+      if (typeof iterator.return === "function") {
+        const originalReturn = iterator.return.bind(iterator);
+        iterator.return = (...args) => withCurrent(span, () => originalReturn(...args));
+      }
+      if (typeof iterator.throw === "function") {
+        const originalThrow = iterator.throw.bind(iterator);
+        iterator.throw = (...args) => withCurrent(span, () => originalThrow(...args));
+      }
+      Object.defineProperty(stream, "__braintrust_current_span_bound", {
+        value: true
+      });
+      return stream;
+    } catch {
+      return stream;
+    }
+  }
+  const originalIteratorFn = stream[Symbol.asyncIterator];
+  if ("__braintrust_current_span_bound" in originalIteratorFn && originalIteratorFn.__braintrust_current_span_bound) {
+    return stream;
+  }
+  try {
+    const patchedIteratorFn = function() {
+      const iterator = originalIteratorFn.call(this);
+      const originalNext = iterator.next.bind(iterator);
+      iterator.next = (...args) => withCurrent(span, () => originalNext(...args));
+      if (typeof iterator.return === "function") {
+        const originalReturn = iterator.return.bind(iterator);
+        iterator.return = (...args) => withCurrent(span, () => originalReturn(...args));
+      }
+      if (typeof iterator.throw === "function") {
+        const originalThrow = iterator.throw.bind(iterator);
+        iterator.throw = (...args) => withCurrent(span, () => originalThrow(...args));
+      }
+      return iterator;
+    };
+    Object.defineProperty(
+      patchedIteratorFn,
+      "__braintrust_current_span_bound",
+      {
+        value: true
+      }
+    );
+    stream[Symbol.asyncIterator] = patchedIteratorFn;
+  } catch {
+    return stream;
+  }
+  return stream;
+}
+function bindCurrentSpanStoreToStart3(tracingChannel, states, create) {
+  const state = _internalGetGlobalState();
+  const contextManager = state?.contextManager;
+  const startChannel = tracingChannel.start;
+  const currentSpanStore = contextManager ? contextManager[BRAINTRUST_CURRENT_SPAN_STORE] : void 0;
+  if (!startChannel?.bindStore || !currentSpanStore) {
+    return void 0;
+  }
+  startChannel.bindStore(currentSpanStore, (event) => {
+    const span = ensureState(
+      states,
+      event,
+      () => create(event)
+    ).span;
+    return contextManager.wrapSpanForStore(span);
+  });
+  return () => {
+    startChannel.unbindStore?.(currentSpanStore);
+  };
+}
+function extractRunnerContextKey(paramsOrContext) {
+  const directUserId = "userId" in paramsOrContext ? paramsOrContext.userId : void 0;
+  const directSessionId = "sessionId" in paramsOrContext ? paramsOrContext.sessionId : void 0;
+  if (typeof directUserId === "string" && typeof directSessionId === "string") {
+    return `${directUserId}:${directSessionId}`;
+  }
+  const invocationContext = paramsOrContext;
+  return extractInvocationContextKey(invocationContext);
+}
+function extractInvocationContextKey(parentContext) {
+  const session = parentContext?.session;
+  const userId = session?.userId;
+  const sessionId = session?.id;
+  if (typeof userId !== "string" || typeof sessionId !== "string") {
+    return void 0;
+  }
+  return `${userId}:${sessionId}`;
+}
+function findRunnerParentSpan(parentContext, activeRunnerSpans) {
+  const contextKey = extractInvocationContextKey(parentContext);
+  return contextKey ? activeRunnerSpans.get(contextKey) : void 0;
+}
+function cleanupActiveRunnerSpan(state, activeRunnerSpans) {
+  if (state.contextKey) {
+    activeRunnerSpans.delete(state.contextKey);
+  }
+}
+function agentContextKey(contextKey, agentName) {
+  return `${contextKey}:${agentName}`;
+}
+function cleanupActiveAgentSpan(state, activeAgentSpans) {
+  if (state.contextKey && state.name) {
+    activeAgentSpans.delete(agentContextKey(state.contextKey, state.name));
+  }
+}
+function extractRunnerInput(paramsOrContext) {
+  const content = "newMessage" in paramsOrContext ? paramsOrContext.newMessage : paramsOrContext.userContent;
+  if (!content || typeof content !== "object") {
+    return void 0;
+  }
+  const normalizedContent = content;
+  if (normalizedContent.parts && Array.isArray(normalizedContent.parts)) {
+    const textParts = normalizedContent.parts.filter((p) => p.text !== void 0).map((p) => p.text);
+    if (textParts.length > 0) {
+      return {
+        messages: [
+          {
+            role: normalizedContent.role ?? "user",
+            content: textParts.join("")
+          }
+        ]
+      };
+    }
+  }
+  return { messages: [normalizedContent] };
+}
+function extractRunnerMetadata(paramsOrContext) {
+  const metadata = {
+    provider: "google-adk"
+  };
+  const directUserId = "userId" in paramsOrContext ? paramsOrContext.userId : void 0;
+  const directSessionId = "sessionId" in paramsOrContext ? paramsOrContext.sessionId : void 0;
+  if (typeof directUserId === "string") {
+    metadata["google_adk.user_id"] = directUserId;
+  }
+  if (typeof directSessionId === "string") {
+    metadata["google_adk.session_id"] = directSessionId;
+  }
+  const session = "session" in paramsOrContext ? paramsOrContext.session : void 0;
+  if (metadata["google_adk.user_id"] === void 0 && typeof session?.userId === "string") {
+    metadata["google_adk.user_id"] = session.userId;
+  }
+  if (metadata["google_adk.session_id"] === void 0 && typeof session?.id === "string") {
+    metadata["google_adk.session_id"] = session.id;
+  }
+  return metadata;
+}
+function extractAgentName(parentContext) {
+  if (!parentContext) {
+    return void 0;
+  }
+  const agent = parentContext.agent;
+  return agent?.name;
+}
+function extractModelName(parentContext) {
+  if (!parentContext) {
+    return void 0;
+  }
+  const agent = parentContext.agent;
+  if (!agent?.model) {
+    return void 0;
+  }
+  if (typeof agent.model === "string") {
+    return agent.model;
+  }
+  if (typeof agent.model === "object" && "model" in agent.model) {
+    return agent.model.model;
+  }
+  return void 0;
+}
+function extractToolCallId(req) {
+  const toolContext = req.toolContext;
+  return toolContext?.functionCallId;
+}
+function extractToolName(req, tool) {
+  if (typeof tool?.name === "string" && tool.name.length > 0) {
+    return tool.name;
+  }
+  const toolContext = req.toolContext;
+  const invocationContext = toolContext?.invocationContext;
+  const invocationTool = invocationContext?.tool;
+  const toolName = invocationTool?.name;
+  return typeof toolName === "string" && toolName.length > 0 ? toolName : void 0;
+}
+function extractToolAgentName(req) {
+  const toolContext = req.toolContext;
+  const directName = toolContext?.agentName;
+  if (typeof directName === "string" && directName.length > 0) {
+    return directName;
+  }
+  const invocationContext = toolContext?.invocationContext;
+  return extractAgentName(invocationContext);
+}
+function findToolParentSpan(req, activeAgentSpans, activeRunnerSpans) {
+  const toolContext = req.toolContext;
+  const invocationContext = toolContext?.invocationContext;
+  const contextKey = extractInvocationContextKey(invocationContext);
+  const agentName = extractToolAgentName(req);
+  if (contextKey && agentName) {
+    const agentSpan = activeAgentSpans.get(
+      agentContextKey(contextKey, agentName)
+    );
+    if (agentSpan) {
+      return agentSpan;
+    }
+  }
+  return contextKey ? activeRunnerSpans.get(contextKey) : void 0;
+}
+function finalizeRunnerSpan(state, activeRunnerSpans) {
+  try {
+    const lastEvent = getLastNonPartialEvent(state.events);
+    const metrics = {};
+    const end = getCurrentUnixTimestamp();
+    metrics.start = state.startTime;
+    metrics.end = end;
+    metrics.duration = end - state.startTime;
+    const usage = aggregateUsageFromEvents(state.events);
+    if (usage) {
+      populateUsageMetrics2(metrics, usage);
+    }
+    state.span.log({
+      output: lastEvent ? extractEventOutput(lastEvent) : void 0,
+      metrics: cleanMetrics2(metrics)
+    });
+  } finally {
+    cleanupActiveRunnerSpan(state, activeRunnerSpans);
+    state.span.end();
+  }
+}
+function finalizeAgentSpan(state, activeAgentSpans) {
+  try {
+    const lastEvent = getLastNonPartialEvent(state.events);
+    const metrics = {};
+    const end = getCurrentUnixTimestamp();
+    metrics.start = state.startTime;
+    metrics.end = end;
+    metrics.duration = end - state.startTime;
+    state.span.log({
+      output: lastEvent ? extractEventOutput(lastEvent) : void 0,
+      metrics: cleanMetrics2(metrics)
+    });
+  } finally {
+    cleanupActiveAgentSpan(state, activeAgentSpans);
+    state.span.end();
+  }
+}
+function getLastNonPartialEvent(events) {
+  for (let i = events.length - 1; i >= 0; i--) {
+    if (!events[i].partial) {
+      return events[i];
+    }
+  }
+  return events.length > 0 ? events[events.length - 1] : void 0;
+}
+function extractEventOutput(event) {
+  if (!event.content) {
+    return void 0;
+  }
+  const output = {};
+  if (event.content.role) {
+    output.role = event.content.role;
+  }
+  if (event.content.parts && Array.isArray(event.content.parts)) {
+    const textParts = event.content.parts.filter((p) => p.text !== void 0 && !p.thought).map((p) => p.text);
+    const thoughtParts = event.content.parts.filter((p) => p.text !== void 0 && p.thought).map((p) => p.text);
+    const functionCalls = event.content.parts.filter((p) => p.functionCall).map((p) => p.functionCall);
+    if (textParts.length > 0) {
+      output.content = textParts.join("");
+    }
+    if (thoughtParts.length > 0) {
+      output.thought = thoughtParts.join("");
+    }
+    if (functionCalls.length > 0) {
+      output.functionCalls = functionCalls;
+    }
+  }
+  if (event.author) {
+    output.author = event.author;
+  }
+  return Object.keys(output).length > 0 ? output : void 0;
+}
+function aggregateUsageFromEvents(events) {
+  let hasUsage = false;
+  const aggregated = {};
+  for (const event of events) {
+    if (!event.usageMetadata) {
+      continue;
+    }
+    hasUsage = true;
+    const usage = event.usageMetadata;
+    if (usage.promptTokenCount !== void 0) {
+      aggregated.promptTokenCount = (aggregated.promptTokenCount ?? 0) + usage.promptTokenCount;
+    }
+    if (usage.candidatesTokenCount !== void 0) {
+      aggregated.candidatesTokenCount = (aggregated.candidatesTokenCount ?? 0) + usage.candidatesTokenCount;
+    }
+    if (usage.totalTokenCount !== void 0) {
+      aggregated.totalTokenCount = (aggregated.totalTokenCount ?? 0) + usage.totalTokenCount;
+    }
+    if (usage.cachedContentTokenCount !== void 0) {
+      aggregated.cachedContentTokenCount = (aggregated.cachedContentTokenCount ?? 0) + usage.cachedContentTokenCount;
+    }
+    if (usage.thoughtsTokenCount !== void 0) {
+      aggregated.thoughtsTokenCount = (aggregated.thoughtsTokenCount ?? 0) + usage.thoughtsTokenCount;
+    }
+  }
+  return hasUsage ? aggregated : void 0;
+}
+function populateUsageMetrics2(metrics, usage) {
+  if (usage.promptTokenCount !== void 0) {
+    metrics.prompt_tokens = usage.promptTokenCount;
+  }
+  if (usage.candidatesTokenCount !== void 0) {
+    metrics.completion_tokens = usage.candidatesTokenCount;
+  }
+  if (usage.totalTokenCount !== void 0) {
+    metrics.tokens = usage.totalTokenCount;
+  }
+  if (usage.cachedContentTokenCount !== void 0) {
+    metrics.prompt_cached_tokens = usage.cachedContentTokenCount;
+  }
+  if (usage.thoughtsTokenCount !== void 0) {
+    metrics.completion_reasoning_tokens = usage.thoughtsTokenCount;
+  }
+}
+function cleanMetrics2(metrics) {
+  const cleaned = {};
+  for (const [key, value] of Object.entries(metrics)) {
+    if (value !== null && value !== void 0) {
+      cleaned[key] = value;
+    }
+  }
+  return cleaned;
+}
+
+// src/instrumentation/plugins/cohere-channels.ts
+var cohereChannels = defineChannels("cohere-ai", {
+  chat: channel({
+    channelName: "chat",
+    kind: "async"
+  }),
+  chatStream: channel({
+    channelName: "chatStream",
+    kind: "async"
+  }),
+  embed: channel({
+    channelName: "embed",
+    kind: "async"
+  }),
+  rerank: channel({
+    channelName: "rerank",
+    kind: "async"
+  })
+});
+
+// src/instrumentation/plugins/cohere-plugin.ts
+var CoherePlugin = class extends BasePlugin {
+  onEnable() {
+    this.subscribeToCohereChannels();
+  }
+  onDisable() {
+    this.unsubscribers = unsubscribeAll(this.unsubscribers);
+  }
+  subscribeToCohereChannels() {
+    this.unsubscribers.push(
+      traceStreamingChannel(cohereChannels.chat, {
+        name: "cohere.chat",
+        type: "llm" /* LLM */,
+        extractInput: extractChatInputWithMetadata2,
+        extractOutput: (result) => extractCohereChatOutput(result),
+        extractMetadata: (result) => extractCohereResponseMetadata(result),
+        extractMetrics: (result, startTime) => {
+          const metrics = parseCohereMetricsFromUsage(result);
+          if (startTime) {
+            metrics.time_to_first_token = getCurrentUnixTimestamp() - startTime;
+          }
+          return metrics;
+        }
+      })
+    );
+    this.unsubscribers.push(
+      traceStreamingChannel(cohereChannels.chatStream, {
+        name: "cohere.chatStream",
+        type: "llm" /* LLM */,
+        extractInput: extractChatInputWithMetadata2,
+        extractOutput: () => void 0,
+        extractMetadata: () => void 0,
+        extractMetrics: () => ({}),
+        aggregateChunks: aggregateCohereChatStreamChunks
+      })
+    );
+    this.unsubscribers.push(
+      traceAsyncChannel(cohereChannels.embed, {
+        name: "cohere.embed",
+        type: "llm" /* LLM */,
+        extractInput: extractEmbedInputWithMetadata,
+        extractOutput: extractCohereEmbeddingOutput,
+        extractMetadata: (result) => extractCohereResponseMetadata(result),
+        extractMetrics: (result) => parseCohereMetricsFromUsage(result)
+      })
+    );
+    this.unsubscribers.push(
+      traceAsyncChannel(cohereChannels.rerank, {
+        name: "cohere.rerank",
+        type: "llm" /* LLM */,
+        extractInput: extractRerankInputWithMetadata,
+        extractOutput: (result) => {
+          if (!isObject(result) || !Array.isArray(result.results)) {
+            return void 0;
+          }
+          return result.results.slice(0, 100).map((item) => ({
+            index: isObject(item) ? item.index : void 0,
+            relevance_score: isObject(item) ? (typeof item.relevanceScore === "number" ? item.relevanceScore : item.relevance_score) ?? null : null
+          }));
+        },
+        extractMetadata: (result) => extractCohereResponseMetadata(result),
+        extractMetrics: (result) => parseCohereMetricsFromUsage(result)
+      })
+    );
+  }
+};
+var CHAT_REQUEST_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
+  "conversationId",
+  "conversation_id",
+  "frequencyPenalty",
+  "frequency_penalty",
+  "k",
+  "maxInputTokens",
+  "max_input_tokens",
+  "maxTokens",
+  "max_tokens",
+  "model",
+  "p",
+  "preamble",
+  "presencePenalty",
+  "presence_penalty",
+  "priority",
+  "promptTruncation",
+  "prompt_truncation",
+  "rawPrompting",
+  "raw_prompting",
+  "responseFormat",
+  "response_format",
+  "safetyMode",
+  "safety_mode",
+  "searchQueriesOnly",
+  "search_queries_only",
+  "seed",
+  "stopSequences",
+  "stop_sequences",
+  "strictTools",
+  "strict_tools",
+  "temperature",
+  "toolChoice",
+  "tool_choice"
+]);
+var EMBED_REQUEST_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
+  "embeddingTypes",
+  "embedding_types",
+  "inputType",
+  "input_type",
+  "maxTokens",
+  "max_tokens",
+  "model",
+  "outputDimension",
+  "output_dimension",
+  "priority",
+  "truncate"
+]);
+var RERANK_REQUEST_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
+  "maxChunksPerDoc",
+  "max_chunks_per_doc",
+  "maxTokensPerDoc",
+  "max_tokens_per_doc",
+  "model",
+  "priority",
+  "rankFields",
+  "rank_fields",
+  "returnDocuments",
+  "return_documents",
+  "topN",
+  "top_n"
+]);
+var RESPONSE_METADATA_ALLOWLIST2 = /* @__PURE__ */ new Set([
+  "finishReason",
+  "finish_reason",
+  "generationId",
+  "generation_id",
+  "id",
+  "responseId",
+  "responseType",
+  "response_id",
+  "response_type"
+]);
+function normalizeArgs5(args) {
+  if (Array.isArray(args)) {
+    return args;
+  }
+  if (isArrayLike5(args)) {
+    return Array.from(args);
+  }
+  return [args];
+}
+function isArrayLike5(value) {
+  return isObject(value) && "length" in value && typeof value.length === "number" && Number.isInteger(value.length) && value.length >= 0;
+}
+function getRequestArg(args) {
+  const firstObjectArg = normalizeArgs5(args).find((arg) => isObject(arg));
+  return isObject(firstObjectArg) ? firstObjectArg : void 0;
+}
+function addCohereProviderMetadata(metadata) {
+  return {
+    ...metadata,
+    provider: "cohere"
+  };
+}
+function pickAllowedMetadata2(metadata, allowlist) {
+  if (!metadata) {
+    return {};
+  }
+  const picked = {};
+  for (const key of allowlist) {
+    const value = metadata[key];
+    if (value !== void 0) {
+      picked[key] = value;
+    }
+  }
+  return picked;
+}
+function extractChatInputWithMetadata2(args) {
+  const request = getRequestArg(args);
+  const { message, messages, ...rawMetadata } = request || {};
+  return {
+    input: processInputAttachments(messages ?? message),
+    metadata: addCohereProviderMetadata(
+      pickAllowedMetadata2(rawMetadata, CHAT_REQUEST_METADATA_ALLOWLIST)
+    )
+  };
+}
+function extractEmbedInputWithMetadata(args) {
+  const request = getRequestArg(args);
+  const { inputs, texts, images, ...rawMetadata } = request || {};
+  return {
+    input: inputs ?? texts ?? images,
+    metadata: addCohereProviderMetadata(
+      pickAllowedMetadata2(rawMetadata, EMBED_REQUEST_METADATA_ALLOWLIST)
+    )
+  };
+}
+function extractRerankInputWithMetadata(args) {
+  const request = getRequestArg(args);
+  const { query, documents, ...rawMetadata } = request || {};
+  return {
+    input: {
+      documents,
+      query
+    },
+    metadata: addCohereProviderMetadata({
+      ...pickAllowedMetadata2(rawMetadata, RERANK_REQUEST_METADATA_ALLOWLIST),
+      ...Array.isArray(documents) ? { document_count: documents.length } : {}
+    })
+  };
+}
+function extractCohereResponseMetadata(result) {
+  if (!isObject(result)) {
+    return void 0;
+  }
+  const responseMetadata = pickAllowedMetadata2(
+    result,
+    RESPONSE_METADATA_ALLOWLIST2
+  );
+  const meta = isObject(result.meta) ? result.meta : void 0;
+  const apiVersion = isObject(meta?.apiVersion) && typeof meta.apiVersion.version === "string" && meta.apiVersion.version || isObject(meta?.api_version) && typeof meta.api_version.version === "string" && meta.api_version.version;
+  const metadata = {
+    ...responseMetadata,
+    ...apiVersion ? { api_version: apiVersion } : {}
+  };
+  return Object.keys(metadata).length > 0 ? metadata : void 0;
+}
+function extractCohereChatOutput(result) {
+  if (!isObject(result)) {
+    return void 0;
+  }
+  if (isObject(result.message)) {
+    return result.message;
+  }
+  if (typeof result.text !== "string") {
+    return void 0;
+  }
+  const toolCalls = Array.isArray(result.toolCalls) ? result.toolCalls : Array.isArray(result.tool_calls) ? result.tool_calls : void 0;
+  if (toolCalls && toolCalls.length > 0) {
+    return {
+      content: result.text,
+      role: "assistant",
+      toolCalls
+    };
+  }
+  return result.text;
+}
+function extractCohereEmbeddingOutput(result) {
+  if (!isObject(result)) {
+    return void 0;
+  }
+  const embeddingLength = getFirstEmbeddingLength(result.embeddings);
+  if (embeddingLength === void 0) {
+    return void 0;
+  }
+  return {
+    embedding_length: embeddingLength
+  };
+}
+function getFirstEmbeddingLength(value) {
+  if (Array.isArray(value) && Array.isArray(value[0])) {
+    return value[0].length;
+  }
+  if (!isObject(value)) {
+    return void 0;
+  }
+  for (const key of Object.keys(value)) {
+    const entry = value[key];
+    if (Array.isArray(entry) && Array.isArray(entry[0])) {
+      return entry[0].length;
+    }
+  }
+  return void 0;
+}
+function setMetricIfNumber(metrics, key, value) {
+  if (typeof value === "number" && Number.isFinite(value)) {
+    metrics[key] = value;
+  }
+}
+function mergeUsageMetrics(metrics, usage) {
+  if (!isObject(usage)) {
+    return metrics;
+  }
+  const source = usage;
+  setMetricIfNumber(
+    metrics,
+    "prompt_tokens",
+    source.inputTokens ?? source.input_tokens
+  );
+  setMetricIfNumber(
+    metrics,
+    "completion_tokens",
+    source.outputTokens ?? source.output_tokens
+  );
+  setMetricIfNumber(
+    metrics,
+    "tokens",
+    source.totalTokens ?? source.total_tokens
+  );
+  setMetricIfNumber(
+    metrics,
+    "prompt_cached_tokens",
+    source.cachedTokens ?? source.cached_tokens
+  );
+  const tokenContainer = isObject(source.tokens) ? source.tokens : void 0;
+  if (tokenContainer) {
+    setMetricIfNumber(
+      metrics,
+      "prompt_tokens",
+      tokenContainer.inputTokens ?? tokenContainer.input_tokens
+    );
+    setMetricIfNumber(
+      metrics,
+      "completion_tokens",
+      tokenContainer.outputTokens ?? tokenContainer.output_tokens
+    );
+    setMetricIfNumber(
+      metrics,
+      "tokens",
+      tokenContainer.totalTokens ?? tokenContainer.total_tokens
+    );
+  }
+  const billedUnits = (isObject(source.billedUnits) ? source.billedUnits : void 0) || (isObject(source.billed_units) ? source.billed_units : void 0);
+  if (billedUnits) {
+    setMetricIfNumber(
+      metrics,
+      "prompt_tokens",
+      billedUnits.inputTokens ?? billedUnits.input_tokens
+    );
+    setMetricIfNumber(
+      metrics,
+      "completion_tokens",
+      billedUnits.outputTokens ?? billedUnits.output_tokens
+    );
+    setMetricIfNumber(
+      metrics,
+      "search_units",
+      billedUnits.searchUnits ?? billedUnits.search_units
+    );
+    setMetricIfNumber(metrics, "classifications", billedUnits.classifications);
+    setMetricIfNumber(metrics, "images", billedUnits.images);
+    setMetricIfNumber(
+      metrics,
+      "image_tokens",
+      billedUnits.imageTokens ?? billedUnits.image_tokens
+    );
+  }
+  return metrics;
+}
+function parseCohereMetricsFromUsage(source) {
+  if (!isObject(source)) {
+    return {};
+  }
+  const metrics = {};
+  mergeUsageMetrics(metrics, source);
+  if ("usage" in source) {
+    mergeUsageMetrics(metrics, source.usage);
+  }
+  if ("meta" in source) {
+    mergeUsageMetrics(metrics, source.meta);
+  }
+  if (metrics.tokens === void 0 && typeof metrics.prompt_tokens === "number" && typeof metrics.completion_tokens === "number") {
+    metrics.tokens = metrics.prompt_tokens + metrics.completion_tokens;
+  }
+  return metrics;
+}
+function toToolCallArray(value) {
+  if (Array.isArray(value)) {
+    return value.filter((item) => isObject(item));
+  }
+  return isObject(value) ? [value] : [];
+}
+function getToolCallIndex2(toolCall, fallbackIndex) {
+  return typeof toolCall.index === "number" && Number.isInteger(toolCall.index) ? toolCall.index : fallbackIndex;
+}
+function appendToolCallDelta(existing, incoming) {
+  const currentArguments = isObject(existing?.function) && typeof existing.function.arguments === "string" ? existing.function.arguments : "";
+  const incomingArguments = isObject(incoming.function) && typeof incoming.function.arguments === "string" ? incoming.function.arguments : "";
+  return {
+    ...existing,
+    ...incoming,
+    function: {
+      ...isObject(existing?.function) ? existing.function : {},
+      ...isObject(incoming.function) ? incoming.function : {},
+      ...incomingArguments ? { arguments: `${currentArguments}${incomingArguments}` } : {}
+    }
+  };
+}
+function extractV8DeltaText(chunk) {
+  if (!isObject(chunk.delta) || !isObject(chunk.delta.message)) {
+    return void 0;
+  }
+  const content = chunk.delta.message.content;
+  if (typeof content === "string") {
+    return content;
+  }
+  if (isObject(content) && typeof content.text === "string") {
+    return content.text;
+  }
+  return void 0;
+}
+function aggregateCohereChatStreamChunks(chunks) {
+  const textDeltas = [];
+  const toolCallsByIndex = {};
+  const toolCallOrder = [];
+  let terminalResponse;
+  let role;
+  let finishReason;
+  let metadata = {};
+  let metrics = {};
+  for (const chunk of chunks) {
+    if (!isObject(chunk)) {
+      continue;
+    }
+    const eventType = typeof chunk.eventType === "string" ? chunk.eventType : typeof chunk.event_type === "string" ? chunk.event_type : typeof chunk.type === "string" ? chunk.type : void 0;
+    if (eventType === "text-generation" && typeof chunk.text === "string") {
+      textDeltas.push(chunk.text);
+      continue;
+    }
+    if (eventType === "tool-calls-generation") {
+      const generatedToolCalls = toToolCallArray(
+        Array.isArray(chunk.toolCalls) ? chunk.toolCalls : chunk.tool_calls
+      );
+      for (const [index, toolCall] of generatedToolCalls.entries()) {
+        const normalizedIndex = getToolCallIndex2(toolCall, index);
+        if (!toolCallOrder.includes(normalizedIndex)) {
+          toolCallOrder.push(normalizedIndex);
+        }
+        toolCallsByIndex[normalizedIndex] = {
+          ...toolCallsByIndex[normalizedIndex],
+          ...toolCall
+        };
+      }
+      if (typeof chunk.text === "string") {
+        textDeltas.push(chunk.text);
+      }
+      continue;
+    }
+    if (eventType === "stream-end" && isObject(chunk.response)) {
+      terminalResponse = chunk.response;
+      metrics = {
+        ...metrics,
+        ...parseCohereMetricsFromUsage(chunk.response)
+      };
+      metadata = {
+        ...metadata,
+        ...extractCohereResponseMetadata(chunk.response) || {}
+      };
+      const responseFinishReason = typeof chunk.response.finishReason === "string" ? chunk.response.finishReason : typeof chunk.response.finish_reason === "string" ? chunk.response.finish_reason : void 0;
+      finishReason = responseFinishReason ?? finishReason;
+      continue;
+    }
+    if (eventType === "message-start") {
+      if (typeof chunk.id === "string") {
+        metadata.id = chunk.id;
+      }
+      if (isObject(chunk.delta) && isObject(chunk.delta.message)) {
+        const messageRole = chunk.delta.message.role;
+        if (typeof messageRole === "string") {
+          role = messageRole;
+        }
+      }
+      continue;
+    }
+    if (eventType === "content-delta") {
+      const text = extractV8DeltaText(chunk);
+      if (text) {
+        textDeltas.push(text);
+      }
+      continue;
+    }
+    if (eventType === "tool-call-start") {
+      const toolCalls = isObject(chunk.delta) && isObject(chunk.delta.message) ? toToolCallArray(
+        Array.isArray(chunk.delta.message.toolCalls) ? chunk.delta.message.toolCalls : chunk.delta.message.toolCalls ?? chunk.delta.message.tool_calls
+      ) : [];
+      for (const [index, toolCall] of toolCalls.entries()) {
+        const normalizedIndex = getToolCallIndex2(
+          toolCall,
+          typeof chunk.index === "number" ? chunk.index : index
+        );
+        if (!toolCallOrder.includes(normalizedIndex)) {
+          toolCallOrder.push(normalizedIndex);
+        }
+        toolCallsByIndex[normalizedIndex] = {
+          ...toolCallsByIndex[normalizedIndex],
+          ...toolCall
+        };
+      }
+      continue;
+    }
+    if (eventType === "tool-call-delta") {
+      const toolCallDelta = isObject(chunk.delta) && isObject(chunk.delta.message) ? toToolCallArray(
+        chunk.delta.message.toolCalls ?? chunk.delta.message.tool_calls
+      ) : [];
+      if (toolCallDelta.length > 0) {
+        const delta = toolCallDelta[0];
+        const normalizedIndex = getToolCallIndex2(delta, chunk.index ?? 0);
+        if (!toolCallOrder.includes(normalizedIndex)) {
+          toolCallOrder.push(normalizedIndex);
+        }
+        toolCallsByIndex[normalizedIndex] = appendToolCallDelta(
+          toolCallsByIndex[normalizedIndex],
+          delta
+        );
+      }
+      continue;
+    }
+    if (eventType === "message-end" && isObject(chunk.delta)) {
+      const delta = chunk.delta;
+      if (typeof delta.finishReason === "string") {
+        finishReason = delta.finishReason;
+      } else if (typeof delta.finish_reason === "string") {
+        finishReason = delta.finish_reason;
+      }
+      if (delta.error !== void 0) {
+        metadata.error = delta.error;
+      }
+      metrics = {
+        ...metrics,
+        ...parseCohereMetricsFromUsage(delta.usage)
+      };
+    }
+  }
+  const mergedToolCalls = toolCallOrder.sort((left, right) => left - right).map((index) => toolCallsByIndex[index]).filter((toolCall) => isObject(toolCall));
+  let output = extractCohereChatOutput(terminalResponse);
+  if (output === void 0) {
+    const mergedText = textDeltas.join("");
+    if (mergedToolCalls.length > 0 || role || mergedText.length > 0) {
+      output = {
+        ...role ? { role } : {},
+        ...mergedText.length > 0 ? { content: mergedText } : {},
+        ...mergedToolCalls.length > 0 ? { toolCalls: mergedToolCalls } : {}
+      };
+    }
+  }
+  if (finishReason) {
+    metadata = {
+      ...metadata,
+      finish_reason: finishReason
+    };
+  }
+  return {
+    metadata,
+    metrics,
+    output
+  };
+}
+
+// src/instrumentation/braintrust-plugin.ts
+var BraintrustPlugin = class extends BasePlugin {
+  config;
+  openaiPlugin = null;
+  anthropicPlugin = null;
+  aiSDKPlugin = null;
+  claudeAgentSDKPlugin = null;
+  googleGenAIPlugin = null;
+  huggingFacePlugin = null;
+  openRouterPlugin = null;
+  openRouterAgentPlugin = null;
+  mistralPlugin = null;
+  googleADKPlugin = null;
+  coherePlugin = null;
+  constructor(config = {}) {
+    super();
+    this.config = config;
+  }
+  onEnable() {
+    const integrations = this.config.integrations || {};
+    if (integrations.openai !== false) {
+      this.openaiPlugin = new OpenAIPlugin();
+      this.openaiPlugin.enable();
+    }
+    if (integrations.anthropic !== false) {
+      this.anthropicPlugin = new AnthropicPlugin();
+      this.anthropicPlugin.enable();
+    }
+    if (integrations.aisdk !== false && integrations.vercel !== false) {
+      this.aiSDKPlugin = new AISDKPlugin();
+      this.aiSDKPlugin.enable();
+    }
+    if (integrations.claudeAgentSDK !== false) {
+      this.claudeAgentSDKPlugin = new ClaudeAgentSDKPlugin();
+      this.claudeAgentSDKPlugin.enable();
+    }
+    if (integrations.googleGenAI !== false && integrations.google !== false) {
+      this.googleGenAIPlugin = new GoogleGenAIPlugin();
+      this.googleGenAIPlugin.enable();
+    }
+    if (integrations.huggingface !== false) {
+      this.huggingFacePlugin = new HuggingFacePlugin();
+      this.huggingFacePlugin.enable();
+    }
+    if (integrations.openrouter !== false) {
+      this.openRouterPlugin = new OpenRouterPlugin();
+      this.openRouterPlugin.enable();
+    }
+    if (integrations.openrouterAgent !== false) {
+      this.openRouterAgentPlugin = new OpenRouterAgentPlugin();
+      this.openRouterAgentPlugin.enable();
+    }
+    if (integrations.mistral !== false) {
+      this.mistralPlugin = new MistralPlugin();
+      this.mistralPlugin.enable();
+    }
+    if (integrations.googleADK !== false) {
+      this.googleADKPlugin = new GoogleADKPlugin();
+      this.googleADKPlugin.enable();
+    }
+    if (integrations.cohere !== false) {
+      this.coherePlugin = new CoherePlugin();
+      this.coherePlugin.enable();
+    }
+  }
+  onDisable() {
+    if (this.openaiPlugin) {
+      this.openaiPlugin.disable();
+      this.openaiPlugin = null;
+    }
+    if (this.anthropicPlugin) {
+      this.anthropicPlugin.disable();
+      this.anthropicPlugin = null;
+    }
+    if (this.aiSDKPlugin) {
+      this.aiSDKPlugin.disable();
+      this.aiSDKPlugin = null;
+    }
+    if (this.claudeAgentSDKPlugin) {
+      this.claudeAgentSDKPlugin.disable();
+      this.claudeAgentSDKPlugin = null;
+    }
+    if (this.googleGenAIPlugin) {
+      this.googleGenAIPlugin.disable();
+      this.googleGenAIPlugin = null;
+    }
+    if (this.huggingFacePlugin) {
+      this.huggingFacePlugin.disable();
+      this.huggingFacePlugin = null;
+    }
+    if (this.openRouterPlugin) {
+      this.openRouterPlugin.disable();
+      this.openRouterPlugin = null;
+    }
+    if (this.openRouterAgentPlugin) {
+      this.openRouterAgentPlugin.disable();
+      this.openRouterAgentPlugin = null;
+    }
+    if (this.mistralPlugin) {
+      this.mistralPlugin.disable();
+      this.mistralPlugin = null;
+    }
+    if (this.googleADKPlugin) {
+      this.googleADKPlugin.disable();
+      this.googleADKPlugin = null;
+    }
+    if (this.coherePlugin) {
+      this.coherePlugin.disable();
+      this.coherePlugin = null;
     }
   }
 };
@@ -15298,10 +18170,12 @@ var PluginRegistry = class {
       vercel: true,
       aisdk: true,
       google: true,
+      huggingface: true,
       claudeAgentSDK: true,
       openrouter: true,
       openrouterAgent: true,
-      mistral: true
+      mistral: true,
+      cohere: true
     };
   }
   /**