@tanstack/ai 0.37.0 → 0.39.0

package/dist/esm/activities/chat/mcp/manager.js

@@ -1,3 +1,9 @@
+function bindReadResource(tool, source) {
+  if (!source.readResource) return;
+  const meta = tool.metadata?.mcp;
+  if (!meta?.uiResourceUri) return;
+  meta.readResource = source.readResource.bind(source);
+}
 class MCPDuplicateToolNameError extends Error {
   constructor(toolName) {
     super(
@@ -40,7 +46,10 @@ class MCPManager {
       for (const [source, result] of zipped) {
         if (result === void 0) continue;
         if (result.status === "fulfilled") {
-          tools.push(...result.value);
+          for (const t of result.value) {
+            bindReadResource(t, source);
+            tools.push(t);
+          }
         } else if (this.#onDiscoveryError) {
           await this.#onDiscoveryError(result.reason, source);
         } else {

package/dist/esm/activities/chat/mcp/manager.js.map

@@ -1 +1 @@
-{"version":3,"file":"manager.js","sources":["../../../../../src/activities/chat/mcp/manager.ts"],"sourcesContent":["import type { ServerTool } from '../tools/tool-definition'\nimport type { ChatMCPOptions, MCPToolSource } from './types'\n\nexport class MCPDuplicateToolNameError extends Error {\n  constructor(public readonly toolName: string) {\n    super(\n      `Duplicate MCP tool name \"${toolName}\" in chat({ mcp.clients }). ` +\n        `Set a unique \\`prefix\\` on one of the MCP clients (or use a pool, ` +\n        `which auto-prefixes) to disambiguate.`,\n    )\n    this.name = 'MCPDuplicateToolNameError'\n  }\n}\n\n/**\n * Encapsulates MCP tool discovery + connection lifecycle for chat().\n * Built from chat()'s `mcp` option; runners only call `discover()` then\n * `dispose()`. A manager built from `undefined` is an inert no-op\n * (`discover()` → `[]`, `dispose()` → no-op), so runners need no branching.\n */\nexport class MCPManager {\n  static from(options: ChatMCPOptions | undefined): MCPManager {\n    return new MCPManager(options)\n  }\n\n  readonly #sources: ReadonlyArray<MCPToolSource>\n  readonly #shouldClose: boolean\n  readonly #lazyTools: boolean\n  readonly #onDiscoveryError?: (\n    error: unknown,\n    source: MCPToolSource,\n  ) => void | Promise<void>\n\n  private constructor(options: ChatMCPOptions | undefined) {\n    this.#sources = options?.clients ?? []\n    // default 'close'; only 'keep-alive' disables closing\n    this.#shouldClose = options ? options.connection !== 'keep-alive' : false\n    this.#lazyTools = options?.lazyTools ?? false\n    this.#onDiscoveryError = options?.onDiscoveryError\n  }\n\n  /**\n   * Discover + merge tools from all sources. Throws on a fatal discovery error\n   * (no `onDiscoveryError`, or it re-threw) or a duplicate tool name; in that\n   * case it first closes any connected sources when the policy is 'close'.\n   */\n  async discover(): Promise<Array<ServerTool>> {\n    if (this.#sources.length === 0) return []\n    try {\n      const settled = await Promise.allSettled(\n        this.#sources.map((s) => s.tools({ lazy: this.#lazyTools })),\n      )\n      const tools: Array<ServerTool> = []\n      const zipped = this.#sources.map(\n        (source, i) => [source, settled[i]] as const,\n      )\n      for (const [source, result] of zipped) {\n        if (result === undefined) continue\n        if (result.status === 'fulfilled') {\n          tools.push(...result.value)\n        } else if (this.#onDiscoveryError) {\n          // throw/reject inside handler ⇒ propagate (fail-fast); return ⇒ skip\n          await this.#onDiscoveryError(result.reason, source)\n        } else {\n          throw result.reason\n        }\n      }\n      const seen = new Set<string>()\n      for (const t of tools) {\n        if (seen.has(t.name)) throw new MCPDuplicateToolNameError(t.name)\n        seen.add(t.name)\n      }\n      return tools\n    } catch (err) {\n      await this.dispose() // cleanup-on-failure (no-op if keep-alive)\n      throw err\n    }\n  }\n\n  /** Close sources iff policy is 'close'. Idempotent; never throws. */\n  async dispose(): Promise<void> {\n    if (!this.#shouldClose || this.#sources.length === 0) return\n    await Promise.allSettled(this.#sources.map((s) => s.close()))\n  }\n}\n"],"names":[],"mappings":"AAGO,MAAM,kCAAkC,MAAM;AAAA,EACnD,YAA4B,UAAkB;AAC5C;AAAA,MACE,4BAA4B,QAAQ;AAAA,IAAA;AAFZ,SAAA,WAAA;AAM1B,SAAK,OAAO;AAAA,EACd;AAAA,EAP4B;AAQ9B;AAQO,MAAM,WAAW;AAAA,EACtB,OAAO,KAAK,SAAiD;AAC3D,WAAO,IAAI,WAAW,OAAO;AAAA,EAC/B;AAAA,EAES;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAKD,YAAY,SAAqC;AACvD,SAAK,WAAW,SAAS,WAAW,CAAA;AAEpC,SAAK,eAAe,UAAU,QAAQ,eAAe,eAAe;AACpE,SAAK,aAAa,SAAS,aAAa;AACxC,SAAK,oBAAoB,SAAS;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,WAAuC;AAC3C,QAAI,KAAK,SAAS,WAAW,UAAU,CAAA;AACvC,QAAI;AACF,YAAM,UAAU,MAAM,QAAQ;AAAA,QAC5B,KAAK,SAAS,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,KAAK,YAAY,CAAC;AAAA,MAAA;AAE7D,YAAM,QAA2B,CAAA;AACjC,YAAM,SAAS,KAAK,SAAS;AAAA,QAC3B,CAAC,QAAQ,MAAM,CAAC,QAAQ,QAAQ,CAAC,CAAC;AAAA,MAAA;AAEpC,iBAAW,CAAC,QAAQ,MAAM,KAAK,QAAQ;AACrC,YAAI,WAAW,OAAW;AAC1B,YAAI,OAAO,WAAW,aAAa;AACjC,gBAAM,KAAK,GAAG,OAAO,KAAK;AAAA,QAC5B,WAAW,KAAK,mBAAmB;AAEjC,gBAAM,KAAK,kBAAkB,OAAO,QAAQ,MAAM;AAAA,QACpD,OAAO;AACL,gBAAM,OAAO;AAAA,QACf;AAAA,MACF;AACA,YAAM,2BAAW,IAAA;AACjB,iBAAW,KAAK,OAAO;AACrB,YAAI,KAAK,IAAI,EAAE,IAAI,EAAG,OAAM,IAAI,0BAA0B,EAAE,IAAI;AAChE,aAAK,IAAI,EAAE,IAAI;AAAA,MACjB;AACA,aAAO;AAAA,IACT,SAAS,KAAK;AACZ,YAAM,KAAK,QAAA;AACX,YAAM;AAAA,IACR;AAAA,EACF;AAAA;AAAA,EAGA,MAAM,UAAyB;AAC7B,QAAI,CAAC,KAAK,gBAAgB,KAAK,SAAS,WAAW,EAAG;AACtD,UAAM,QAAQ,WAAW,KAAK,SAAS,IAAI,CAAC,MAAM,EAAE,MAAA,CAAO,CAAC;AAAA,EAC9D;AACF;"}
+{"version":3,"file":"manager.js","sources":["../../../../../src/activities/chat/mcp/manager.ts"],"sourcesContent":["import type { ServerTool } from '../tools/tool-definition'\nimport type { ChatMCPOptions, MCPToolSource } from './types'\n\n/**\n * Bind the source's `readResource` onto a ui-linked tool's `metadata.mcp` so it\n * travels with the tool to the server-tool execution/emit site (`tool-calls.ts`).\n *\n * `discover()` is the single place in `@tanstack/ai` that has both a tool and\n * its originating source, and `@tanstack/ai` must not import `@tanstack/ai-mcp`,\n * so this is where the source handle is threaded onto the tool. Only tools that\n * actually link a `ui://` resource (their discovery stamped\n * `metadata.mcp.uiResourceUri`) and whose source can read resources get bound;\n * everything else is left untouched.\n *\n * This mutates the discovered tool's `metadata.mcp` in place. That is safe\n * because discovery returns fresh tool objects per `discover()` call: the bound\n * `readResource` closes over `source`, whose connection stays live until the run\n * drains. If discovery results were ever cached and reused across runs, this\n * would bind a closure over an already-closed source — bind onto a copy then.\n */\nfunction bindReadResource(tool: ServerTool, source: MCPToolSource): void {\n  if (!source.readResource) return\n  const meta = (\n    tool.metadata as { mcp?: { uiResourceUri?: string } } | undefined\n  )?.mcp\n  if (!meta?.uiResourceUri) return\n  ;(meta as { readResource?: MCPToolSource['readResource'] }).readResource =\n    source.readResource.bind(source)\n}\n\nexport class MCPDuplicateToolNameError extends Error {\n  constructor(public readonly toolName: string) {\n    super(\n      `Duplicate MCP tool name \"${toolName}\" in chat({ mcp.clients }). ` +\n        `Set a unique \\`prefix\\` on one of the MCP clients (or use a pool, ` +\n        `which auto-prefixes) to disambiguate.`,\n    )\n    this.name = 'MCPDuplicateToolNameError'\n  }\n}\n\n/**\n * Encapsulates MCP tool discovery + connection lifecycle for chat().\n * Built from chat()'s `mcp` option; runners only call `discover()` then\n * `dispose()`. A manager built from `undefined` is an inert no-op\n * (`discover()` → `[]`, `dispose()` → no-op), so runners need no branching.\n */\nexport class MCPManager {\n  static from(options: ChatMCPOptions | undefined): MCPManager {\n    return new MCPManager(options)\n  }\n\n  readonly #sources: ReadonlyArray<MCPToolSource>\n  readonly #shouldClose: boolean\n  readonly #lazyTools: boolean\n  readonly #onDiscoveryError?: (\n    error: unknown,\n    source: MCPToolSource,\n  ) => void | Promise<void>\n\n  private constructor(options: ChatMCPOptions | undefined) {\n    this.#sources = options?.clients ?? []\n    // default 'close'; only 'keep-alive' disables closing\n    this.#shouldClose = options ? options.connection !== 'keep-alive' : false\n    this.#lazyTools = options?.lazyTools ?? false\n    this.#onDiscoveryError = options?.onDiscoveryError\n  }\n\n  /**\n   * Discover + merge tools from all sources. Throws on a fatal discovery error\n   * (no `onDiscoveryError`, or it re-threw) or a duplicate tool name; in that\n   * case it first closes any connected sources when the policy is 'close'.\n   */\n  async discover(): Promise<Array<ServerTool>> {\n    if (this.#sources.length === 0) return []\n    try {\n      const settled = await Promise.allSettled(\n        this.#sources.map((s) => s.tools({ lazy: this.#lazyTools })),\n      )\n      const tools: Array<ServerTool> = []\n      const zipped = this.#sources.map(\n        (source, i) => [source, settled[i]] as const,\n      )\n      for (const [source, result] of zipped) {\n        if (result === undefined) continue\n        if (result.status === 'fulfilled') {\n          for (const t of result.value) {\n            bindReadResource(t, source)\n            tools.push(t)\n          }\n        } else if (this.#onDiscoveryError) {\n          // throw/reject inside handler ⇒ propagate (fail-fast); return ⇒ skip\n          await this.#onDiscoveryError(result.reason, source)\n        } else {\n          throw result.reason\n        }\n      }\n      const seen = new Set<string>()\n      for (const t of tools) {\n        if (seen.has(t.name)) throw new MCPDuplicateToolNameError(t.name)\n        seen.add(t.name)\n      }\n      return tools\n    } catch (err) {\n      await this.dispose() // cleanup-on-failure (no-op if keep-alive)\n      throw err\n    }\n  }\n\n  /** Close sources iff policy is 'close'. Idempotent; never throws. */\n  async dispose(): Promise<void> {\n    if (!this.#shouldClose || this.#sources.length === 0) return\n    await Promise.allSettled(this.#sources.map((s) => s.close()))\n  }\n}\n"],"names":[],"mappings":"AAoBA,SAAS,iBAAiB,MAAkB,QAA6B;AACvE,MAAI,CAAC,OAAO,aAAc;AAC1B,QAAM,OACJ,KAAK,UACJ;AACH,MAAI,CAAC,MAAM,cAAe;AACxB,OAA0D,eAC1D,OAAO,aAAa,KAAK,MAAM;AACnC;AAEO,MAAM,kCAAkC,MAAM;AAAA,EACnD,YAA4B,UAAkB;AAC5C;AAAA,MACE,4BAA4B,QAAQ;AAAA,IAAA;AAFZ,SAAA,WAAA;AAM1B,SAAK,OAAO;AAAA,EACd;AAAA,EAP4B;AAQ9B;AAQO,MAAM,WAAW;AAAA,EACtB,OAAO,KAAK,SAAiD;AAC3D,WAAO,IAAI,WAAW,OAAO;AAAA,EAC/B;AAAA,EAES;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAKD,YAAY,SAAqC;AACvD,SAAK,WAAW,SAAS,WAAW,CAAA;AAEpC,SAAK,eAAe,UAAU,QAAQ,eAAe,eAAe;AACpE,SAAK,aAAa,SAAS,aAAa;AACxC,SAAK,oBAAoB,SAAS;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,WAAuC;AAC3C,QAAI,KAAK,SAAS,WAAW,UAAU,CAAA;AACvC,QAAI;AACF,YAAM,UAAU,MAAM,QAAQ;AAAA,QAC5B,KAAK,SAAS,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,KAAK,YAAY,CAAC;AAAA,MAAA;AAE7D,YAAM,QAA2B,CAAA;AACjC,YAAM,SAAS,KAAK,SAAS;AAAA,QAC3B,CAAC,QAAQ,MAAM,CAAC,QAAQ,QAAQ,CAAC,CAAC;AAAA,MAAA;AAEpC,iBAAW,CAAC,QAAQ,MAAM,KAAK,QAAQ;AACrC,YAAI,WAAW,OAAW;AAC1B,YAAI,OAAO,WAAW,aAAa;AACjC,qBAAW,KAAK,OAAO,OAAO;AAC5B,6BAAiB,GAAG,MAAM;AAC1B,kBAAM,KAAK,CAAC;AAAA,UACd;AAAA,QACF,WAAW,KAAK,mBAAmB;AAEjC,gBAAM,KAAK,kBAAkB,OAAO,QAAQ,MAAM;AAAA,QACpD,OAAO;AACL,gBAAM,OAAO;AAAA,QACf;AAAA,MACF;AACA,YAAM,2BAAW,IAAA;AACjB,iBAAW,KAAK,OAAO;AACrB,YAAI,KAAK,IAAI,EAAE,IAAI,EAAG,OAAM,IAAI,0BAA0B,EAAE,IAAI;AAChE,aAAK,IAAI,EAAE,IAAI;AAAA,MACjB;AACA,aAAO;AAAA,IACT,SAAS,KAAK;AACZ,YAAM,KAAK,QAAA;AACX,YAAM;AAAA,IACR;AAAA,EACF;AAAA;AAAA,EAGA,MAAM,UAAyB;AAC7B,QAAI,CAAC,KAAK,gBAAgB,KAAK,SAAS,WAAW,EAAG;AACtD,UAAM,QAAQ,WAAW,KAAK,SAAS,IAAI,CAAC,MAAM,EAAE,MAAA,CAAO,CAAC;AAAA,EAC9D;AACF;"}

package/dist/esm/activities/chat/mcp/types.d.ts

@@ -1,4 +1,18 @@
 import { ServerTool } from '../tools/tool-definition.js';
+/**
+ * The shape `readResource` resolves to — a structural subset of MCP's
+ * `ReadResourceResult`. Single source of truth shared by
+ * `MCPToolSource.readResource` (this file) and the tool-bound
+ * `McpToolAppMeta.readResource` (tool-calls.ts) so the two copies cannot drift.
+ */
+export interface McpResourceReadResult {
+    contents: Array<{
+        uri: string;
+        mimeType?: string;
+        text?: string;
+        blob?: string;
+    }>;
+}
 /**
  * Minimal structural shape that `chat({ mcp })` needs from an MCP client.
  *
@@ -11,6 +25,14 @@ export interface MCPToolSource {
         lazy?: boolean;
     }) => Promise<Array<ServerTool>>;
     close: () => Promise<void>;
+    /**
+     * Reads an MCP resource by URI. Used by the chat manager to eagerly fetch
+     * `ui://` resource widgets (MCP Apps) after a tool result resolves.
+     *
+     * Optional — sources that do not serve `ui://` resources need not implement
+     * this method. `ai-mcp`'s `MCPClient` satisfies this structurally.
+     */
+    readResource?: (uri: string) => Promise<McpResourceReadResult>;
 }
 /**
  * Controls what happens to MCP connections when the chat run ends.

package/dist/esm/activities/chat/messages.js.map

@@ -1 +1 @@
-{"version":3,"file":"messages.js","sources":["../../../../src/activities/chat/messages.ts"],"sourcesContent":["import { normalizeToolResult } from '../../utilities/tool-result'\nimport type { Message as AGUIMessage } from '@ag-ui/core'\nimport type {\n  ContentPart,\n  MessagePart,\n  ModelMessage,\n  TextPart,\n  ToolCallPart,\n  UIMessage,\n} from '../../types'\n// ===========================\n// Message Converters\n// ===========================\n\n/**\n * Check if a MessagePart is a content part (text, image, audio, video, document)\n * that maps directly to a ModelMessage ContentPart.\n */\nfunction isContentPart(part: MessagePart): part is ContentPart {\n  return (\n    part.type === 'text' ||\n    part.type === 'image' ||\n    part.type === 'audio' ||\n    part.type === 'video' ||\n    part.type === 'document'\n  )\n}\n\nfunction safeJsonStringify(value: unknown): string {\n  try {\n    return JSON.stringify(value)\n  } catch {\n    return ''\n  }\n}\n\nfunction parseToolResultContent(content: string): unknown {\n  try {\n    return JSON.parse(content)\n  } catch {\n    return content\n  }\n}\n\n/**\n * Collapse an array of ContentParts into the most compact ModelMessage content:\n * - Empty array → null\n * - All text parts → joined string (or null if empty)\n * - Mixed content → ContentPart array as-is\n */\nfunction collapseContentParts(\n  parts: Array<ContentPart>,\n): string | null | Array<ContentPart> {\n  if (parts.length === 0) return null\n\n  const allText = parts.every((p) => p.type === 'text')\n  if (allText) {\n    const joined = parts.map((p) => p.content).join('')\n    return joined || null\n  }\n\n  return parts\n}\n\n/**\n * Extract text content from ModelMessage content (string, null, or ContentPart array).\n * Used when only the text portion is needed (e.g., tool result content).\n */\nfunction getTextContent(content: string | null | Array<ContentPart>): string {\n  if (content === null) return ''\n  if (typeof content === 'string') return content\n  return content\n    .filter((part): part is TextPart => part.type === 'text')\n    .map((part) => part.content)\n    .join('')\n}\n\n/**\n * Convert UIMessages or ModelMessages to ModelMessages\n */\nexport function convertMessagesToModelMessages(\n  messages: Array<UIMessage | ModelMessage>,\n): Array<ModelMessage> {\n  // Pre-pass: collect toolCallIds already represented in anchor UIMessage parts.\n  // Fan-out tool messages whose toolCallId matches an anchored ToolResultPart\n  // are AG-UI duplicates and must be dropped to avoid double-feeding the LLM.\n  const anchoredToolCallIds = new Set<string>()\n  for (const msg of messages) {\n    if ('parts' in msg) {\n      for (const part of msg.parts) {\n        if (part.type === 'tool-result') {\n          anchoredToolCallIds.add(part.toolCallId)\n        }\n      }\n    }\n  }\n\n  const modelMessages: Array<ModelMessage> = []\n  for (const msg of messages) {\n    if ('parts' in msg) {\n      // UIMessage anchor — existing fan-out path\n      modelMessages.push(...uiMessageToModelMessages(msg))\n      continue\n    }\n\n    const role = (msg as { role: string }).role\n\n    // AG-UI tool fan-out duplicate — drop if anchor already covers it\n    if (\n      role === 'tool' &&\n      msg.toolCallId &&\n      anchoredToolCallIds.has(msg.toolCallId)\n    ) {\n      continue\n    }\n\n    // AG-UI reasoning and activity — no ModelMessage equivalent today\n    if (role === 'reasoning' || role === 'activity') {\n      continue\n    }\n\n    // AG-UI developer — collapse to system\n    if (role === 'developer') {\n      modelMessages.push({\n        role: 'system' as ModelMessage['role'],\n        content: (msg as { content: string }).content,\n      })\n      continue\n    }\n\n    // Already a ModelMessage (user, assistant, system, tool with no anchor) — pass through\n    modelMessages.push(msg)\n  }\n  return modelMessages\n}\n\n/**\n * Convert a UIMessage to ModelMessage(s)\n *\n * Walks the parts array IN ORDER to preserve the interleaving of text,\n * tool calls, and tool results. This is critical for multi-round tool\n * flows where the model generates text, calls a tool, gets the result,\n * then generates more text and calls another tool.\n *\n * The output preserves the sequential structure:\n *   text1 → toolCall1 → toolResult1 → text2 → toolCall2 → toolResult2\n * becomes:\n *   assistant: {content: \"text1\", toolCalls: [toolCall1]}\n *   tool: toolResult1\n *   assistant: {content: \"text2\", toolCalls: [toolCall2]}\n *   tool: toolResult2\n *\n * @param uiMessage - The UIMessage to convert\n * @returns An array of ModelMessages preserving part ordering\n */\nexport function uiMessageToModelMessages(\n  uiMessage: UIMessage,\n): Array<ModelMessage> {\n  // Skip system messages - they're handled via systemPrompts, not ModelMessages\n  if (uiMessage.role === 'system') {\n    return []\n  }\n\n  // For non-assistant messages (user), use the simpler path since they\n  // don't have tool calls or tool results to interleave\n  if (uiMessage.role !== 'assistant') {\n    return [buildUserOrToolMessage(uiMessage)]\n  }\n\n  // For assistant messages, walk parts in order to preserve interleaving\n  return buildAssistantMessages(uiMessage)\n}\n\n/**\n * Build a single ModelMessage for user messages (simple path).\n * Preserves ordering of text and multimodal content parts.\n */\nfunction buildUserOrToolMessage(uiMessage: UIMessage): ModelMessage {\n  const contentParts: Array<ContentPart> = []\n  for (const part of uiMessage.parts) {\n    if (isContentPart(part)) {\n      contentParts.push(part)\n    }\n  }\n\n  return {\n    role: uiMessage.role as 'user' | 'assistant' | 'tool',\n    content: collapseContentParts(contentParts),\n  }\n}\n\n// Accumulator for building an assistant segment (content + tool calls)\ninterface AssistantSegment {\n  contentParts: Array<ContentPart>\n  toolCalls: Array<{\n    id: string\n    type: 'function'\n    function: { name: string; arguments: string }\n    /** Provider-specific metadata that round-trips with the tool call.\n     * Untyped at this framework layer; adapters narrow it via their\n     * `TToolCallMetadata` generic. */\n    metadata?: unknown\n  }>\n}\n\nfunction createSegment(): AssistantSegment {\n  return { contentParts: [], toolCalls: [] }\n}\n\nfunction isToolCallIncluded(part: ToolCallPart): boolean {\n  return (\n    part.state === 'input-complete' ||\n    part.state === 'complete' ||\n    part.state === 'approval-responded' ||\n    part.state === 'error' ||\n    part.output !== undefined\n  )\n}\n\n/**\n * Build ModelMessages for an assistant UIMessage, preserving the\n * sequential interleaving of text, tool calls, and tool results.\n *\n * Walks parts in order. Text and tool-call parts accumulate into the\n * current \"segment\". When a tool-result part is encountered, the\n * current segment is flushed as an assistant message, then the tool\n * result is emitted as a tool message.\n */\nfunction buildAssistantMessages(uiMessage: UIMessage): Array<ModelMessage> {\n  const messageList: Array<ModelMessage> = []\n  let current = createSegment()\n  let pendingThinking: Array<{ content: string; signature?: string }> = []\n\n  // Track emitted tool result IDs to avoid duplicates.\n  // A tool call can have BOTH an explicit tool-result part AND an output\n  // field on the tool-call part. We only want one per tool call ID.\n  const emittedToolResultIds = new Set<string>()\n\n  function flushSegment(): void {\n    const content = collapseContentParts(current.contentParts)\n    const hasContent = content !== null\n    const hasToolCalls = current.toolCalls.length > 0\n\n    if (hasContent || hasToolCalls) {\n      messageList.push({\n        role: 'assistant',\n        content,\n        ...(hasToolCalls && { toolCalls: current.toolCalls }),\n        ...(pendingThinking.length > 0 && { thinking: pendingThinking }),\n      })\n      pendingThinking = []\n    }\n    current = createSegment()\n  }\n\n  for (const part of uiMessage.parts) {\n    switch (part.type) {\n      case 'text':\n      case 'image':\n      case 'audio':\n      case 'video':\n      case 'document':\n        current.contentParts.push(part)\n        break\n\n      case 'tool-call':\n        if (isToolCallIncluded(part)) {\n          current.toolCalls.push({\n            id: part.id,\n            type: 'function' as const,\n            function: {\n              name: part.name,\n              arguments: part.arguments,\n            },\n            ...(part.metadata !== undefined && { metadata: part.metadata }),\n          })\n        }\n        break\n\n      case 'tool-result':\n        // Flush the current assistant segment before emitting the tool result\n        flushSegment()\n\n        // Emit the tool result\n        if (\n          (part.state === 'complete' || part.state === 'error') &&\n          !emittedToolResultIds.has(part.toolCallId)\n        ) {\n          messageList.push({\n            role: 'tool',\n            content: part.content,\n            toolCallId: part.toolCallId,\n          })\n          emittedToolResultIds.add(part.toolCallId)\n        }\n        break\n\n      case 'thinking':\n        if (part.content) {\n          pendingThinking.push({\n            content: part.content,\n            ...(part.signature && { signature: part.signature }),\n          })\n        }\n        break\n\n      case 'structured-output':\n        // Only emit completed structured responses into history. Streaming or\n        // errored buffers would push malformed JSON into the next LLM turn's\n        // assistant content. `raw` is the source of truth; `data` is the\n        // defensive fallback for terminal-only completes that didn't ship raw.\n        if (part.status === 'complete') {\n          const serialized =\n            part.raw !== ''\n              ? part.raw\n              : part.data !== undefined\n                ? safeJsonStringify(part.data)\n                : ''\n          if (serialized !== '') {\n            current.contentParts.push({ type: 'text', content: serialized })\n          }\n        }\n        break\n\n      default:\n        break\n    }\n  }\n\n  // Flush any remaining accumulated content\n  flushSegment()\n\n  // Emit tool results from client tool-call parts with output or approval,\n  // but only if not already covered by an explicit tool-result part above.\n  // These are appended at the end since they don't have explicit tool-result\n  // parts in the parts array to trigger inline emission.\n  for (const part of uiMessage.parts) {\n    if (part.type !== 'tool-call') continue\n\n    // Output takes priority — if the tool has already produced a result,\n    // emit the concrete output regardless of approval metadata.\n    if (part.output !== undefined && !emittedToolResultIds.has(part.id)) {\n      messageList.push({\n        role: 'tool',\n        content: normalizeToolResult(part.output),\n        toolCallId: part.id,\n      })\n      emittedToolResultIds.add(part.id)\n    }\n\n    // Approval response without output — emit approval status for iteration tracking\n    if (\n      part.output === undefined &&\n      part.state === 'approval-responded' &&\n      part.approval?.approved !== undefined &&\n      !emittedToolResultIds.has(part.id)\n    ) {\n      const approved = part.approval.approved\n      messageList.push({\n        role: 'tool',\n        content: JSON.stringify({\n          approved,\n          ...(approved && { pendingExecution: true }),\n          message: approved\n            ? 'User approved this action'\n            : 'User denied this action',\n        }),\n        toolCallId: part.id,\n      })\n      emittedToolResultIds.add(part.id)\n    }\n  }\n\n  // If no messages were produced (e.g., empty parts), emit a minimal assistant message\n  if (messageList.length === 0) {\n    messageList.push({\n      role: 'assistant',\n      content: null,\n    })\n  }\n\n  return messageList\n}\n\n/**\n * Convert a ModelMessage to UIMessage\n *\n * This conversion creates a parts-based structure:\n * - content field → TextPart\n * - toolCalls array → ToolCallPart[]\n * - role=\"tool\" messages should be converted separately and merged\n *\n * @param modelMessage - The ModelMessage to convert\n * @param id - Optional ID for the UIMessage (generated if not provided)\n * @returns A UIMessage with parts\n */\nexport function modelMessageToUIMessage(\n  modelMessage: ModelMessage,\n  id?: string,\n): UIMessage {\n  const parts: Array<MessagePart> = []\n\n  if (modelMessage.role === 'assistant' && modelMessage.thinking?.length) {\n    for (const thinking of modelMessage.thinking) {\n      if (!thinking.content) continue\n      parts.push({\n        type: 'thinking',\n        content: thinking.content,\n        ...(thinking.signature && { signature: thinking.signature }),\n      })\n    }\n  }\n\n  // Handle tool results (when role is \"tool\") - only produce tool-result part,\n  // not a text part (the content IS the tool result, not display text)\n  if (modelMessage.role === 'tool' && modelMessage.toolCallId) {\n    parts.push({\n      type: 'tool-result',\n      toolCallId: modelMessage.toolCallId,\n      content: getTextContent(modelMessage.content),\n      state: 'complete',\n    })\n  } else if (Array.isArray(modelMessage.content)) {\n    // Multimodal content - preserve all content parts as MessageParts\n    for (const part of modelMessage.content) {\n      parts.push(part)\n    }\n  } else {\n    // String or null content\n    const textContent = getTextContent(modelMessage.content)\n    if (textContent) {\n      parts.push({\n        type: 'text',\n        content: textContent,\n      })\n    }\n  }\n\n  // Handle tool calls\n  if (modelMessage.toolCalls && modelMessage.toolCalls.length > 0) {\n    for (const toolCall of modelMessage.toolCalls) {\n      parts.push({\n        type: 'tool-call',\n        id: toolCall.id,\n        name: toolCall.function.name,\n        arguments: toolCall.function.arguments,\n        state: 'input-complete', // Model messages have complete arguments\n        ...(toolCall.metadata !== undefined && { metadata: toolCall.metadata }),\n      })\n    }\n  }\n\n  return {\n    id: id || generateMessageId(),\n    role: modelMessage.role === 'tool' ? 'assistant' : modelMessage.role,\n    parts,\n  }\n}\n\n/**\n * Normalize a single AG-UI `MESSAGES_SNAPSHOT` message into a `UIMessage`.\n *\n * AG-UI snapshot messages use the wire shape `{ id, role, content }` and have\n * no `parts` array. Casting them directly to `UIMessage` is unsafe: any code\n * that later reads `message.parts` (e.g. the devtools `onToolCallStateChange`\n * handler) crashes with \"Cannot read properties of undefined (reading 'find')\".\n *\n * Each role is mapped to the canonical `UIMessage` shape, reusing\n * `modelMessageToUIMessage` for the roles that share `ModelMessage`'s structure.\n * The original AG-UI `id` is preserved so later `TEXT_MESSAGE_CONTENT` /\n * `TOOL_CALL_*` events still route by `messageId` (falling back to a generated\n * id only when the snapshot omits one). Messages that already carry `parts`\n * (e.g. a TanStack server echoing `UIMessage`s back over the wire) pass through\n * unchanged apart from ensuring an id.\n */\nexport function aguiSnapshotMessageToUIMessage(\n  message: AGUIMessage | UIMessage,\n): UIMessage {\n  if ('parts' in message) {\n    return { ...message, id: message.id || generateMessageId() }\n  }\n\n  const id = message.id || generateMessageId()\n\n  switch (message.role) {\n    case 'user':\n      return {\n        id,\n        role: 'user',\n        parts: aguiUserContentToParts(message.content),\n      }\n    case 'assistant':\n      return modelMessageToUIMessage(\n        {\n          role: 'assistant',\n          content: message.content ?? null,\n          ...(message.toolCalls && { toolCalls: message.toolCalls }),\n        },\n        id,\n      )\n    case 'tool':\n      return modelMessageToUIMessage(\n        {\n          role: 'tool',\n          content: message.content,\n          toolCallId: message.toolCallId,\n        },\n        id,\n      )\n    case 'system':\n    case 'developer':\n      // `ModelMessage` has no system/developer role; build the part directly.\n      return {\n        id,\n        role: 'system',\n        parts: message.content\n          ? [{ type: 'text', content: message.content }]\n          : [],\n      }\n    case 'reasoning':\n      return {\n        id,\n        role: 'assistant',\n        parts: message.content\n          ? [{ type: 'thinking', content: message.content }]\n          : [],\n      }\n    case 'activity':\n    default:\n      // `activity` (and any future role) has no text/parts equivalent today.\n      return { id, role: 'assistant', parts: [] }\n  }\n}\n\n/**\n * Convert AG-UI user message content into `UIMessage` parts.\n *\n * AG-UI user content is either a plain string or a multimodal array whose text\n * entries use `{ type: 'text', text }` (vs. TanStack's `{ type: 'text', content }`).\n * Text entries are rewritten to the TanStack shape; image/audio/video/document\n * entries already match `ContentPart` and pass through. `binary` entries have no\n * TanStack equivalent and are dropped.\n */\nfunction aguiUserContentToParts(\n  content: Extract<AGUIMessage, { role: 'user' }>['content'],\n): Array<MessagePart> {\n  if (typeof content === 'string') {\n    return content ? [{ type: 'text', content }] : []\n  }\n\n  const parts: Array<MessagePart> = []\n  for (const part of content) {\n    if (part.type === 'text') {\n      parts.push({ type: 'text', content: part.text })\n    } else if (part.type !== 'binary') {\n      parts.push(part)\n    }\n  }\n  return parts\n}\n\n/**\n * Convert an array of ModelMessages to UIMessages\n *\n * This handles merging tool result messages with their corresponding assistant messages\n *\n * @param modelMessages - Array of ModelMessages to convert\n * @returns Array of UIMessages\n */\nexport function modelMessagesToUIMessages(\n  modelMessages: Array<ModelMessage>,\n): Array<UIMessage> {\n  const uiMessages: Array<UIMessage> = []\n  let currentAssistantMessage: UIMessage | null = null\n\n  for (const msg of modelMessages) {\n    if (msg.role === 'tool') {\n      // Tool result - merge into the last assistant message if possible\n      if (\n        msg.toolCallId !== undefined &&\n        currentAssistantMessage &&\n        currentAssistantMessage.role === 'assistant'\n      ) {\n        const content = getTextContent(msg.content)\n        const toolCallPart = currentAssistantMessage.parts.find(\n          (part): part is ToolCallPart =>\n            part.type === 'tool-call' && part.id === msg.toolCallId,\n        )\n\n        if (toolCallPart) {\n          toolCallPart.output = parseToolResultContent(content)\n          toolCallPart.state = 'complete'\n        }\n\n        currentAssistantMessage.parts.push({\n          type: 'tool-result',\n          toolCallId: msg.toolCallId,\n          content,\n          state: 'complete',\n        })\n      } else {\n        // No assistant message to merge into, create a standalone one\n        const toolResultUIMessage = modelMessageToUIMessage(msg)\n        uiMessages.push(toolResultUIMessage)\n      }\n    } else {\n      // Regular message\n      const uiMessage = modelMessageToUIMessage(msg)\n      uiMessages.push(uiMessage)\n\n      // Track assistant messages for potential tool result merging\n      if (msg.role === 'assistant') {\n        currentAssistantMessage = uiMessage\n      } else {\n        currentAssistantMessage = null\n      }\n    }\n  }\n\n  return uiMessages\n}\n\n/**\n * Normalize a message (UIMessage or ModelMessage) to a UIMessage\n * Ensures the message has an ID and createdAt timestamp\n *\n * @param message - Either a UIMessage or ModelMessage\n * @param generateId - Function to generate a message ID if needed\n * @returns A UIMessage with guaranteed id and createdAt\n */\nexport function normalizeToUIMessage(\n  message: UIMessage | ModelMessage,\n  generateId: () => string,\n): UIMessage {\n  if ('parts' in message) {\n    // Already a UIMessage\n    return {\n      ...message,\n      id: message.id || generateId(),\n      createdAt: message.createdAt || new Date(),\n    }\n  } else {\n    // ModelMessage - convert to UIMessage\n    return {\n      ...modelMessageToUIMessage(message, generateId()),\n      createdAt: new Date(),\n    }\n  }\n}\n\n/**\n * Generate a unique message ID\n */\nexport function generateMessageId(): string {\n  return `msg-${Date.now()}-${Math.random().toString(36).substring(7)}`\n}\n"],"names":[],"mappings":";AAkBA,SAAS,cAAc,MAAwC;AAC7D,SACE,KAAK,SAAS,UACd,KAAK,SAAS,WACd,KAAK,SAAS,WACd,KAAK,SAAS,WACd,KAAK,SAAS;AAElB;AAEA,SAAS,kBAAkB,OAAwB;AACjD,MAAI;AACF,WAAO,KAAK,UAAU,KAAK;AAAA,EAC7B,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEA,SAAS,uBAAuB,SAA0B;AACxD,MAAI;AACF,WAAO,KAAK,MAAM,OAAO;AAAA,EAC3B,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAQA,SAAS,qBACP,OACoC;AACpC,MAAI,MAAM,WAAW,EAAG,QAAO;AAE/B,QAAM,UAAU,MAAM,MAAM,CAAC,MAAM,EAAE,SAAS,MAAM;AACpD,MAAI,SAAS;AACX,UAAM,SAAS,MAAM,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE;AAClD,WAAO,UAAU;AAAA,EACnB;AAEA,SAAO;AACT;AAMA,SAAS,eAAe,SAAqD;AAC3E,MAAI,YAAY,KAAM,QAAO;AAC7B,MAAI,OAAO,YAAY,SAAU,QAAO;AACxC,SAAO,QACJ,OAAO,CAAC,SAA2B,KAAK,SAAS,MAAM,EACvD,IAAI,CAAC,SAAS,KAAK,OAAO,EAC1B,KAAK,EAAE;AACZ;AAKO,SAAS,+BACd,UACqB;AAIrB,QAAM,0CAA0B,IAAA;AAChC,aAAW,OAAO,UAAU;AAC1B,QAAI,WAAW,KAAK;AAClB,iBAAW,QAAQ,IAAI,OAAO;AAC5B,YAAI,KAAK,SAAS,eAAe;AAC/B,8BAAoB,IAAI,KAAK,UAAU;AAAA,QACzC;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,QAAM,gBAAqC,CAAA;AAC3C,aAAW,OAAO,UAAU;AAC1B,QAAI,WAAW,KAAK;AAElB,oBAAc,KAAK,GAAG,yBAAyB,GAAG,CAAC;AACnD;AAAA,IACF;AAEA,UAAM,OAAQ,IAAyB;AAGvC,QACE,SAAS,UACT,IAAI,cACJ,oBAAoB,IAAI,IAAI,UAAU,GACtC;AACA;AAAA,IACF;AAGA,QAAI,SAAS,eAAe,SAAS,YAAY;AAC/C;AAAA,IACF;AAGA,QAAI,SAAS,aAAa;AACxB,oBAAc,KAAK;AAAA,QACjB,MAAM;AAAA,QACN,SAAU,IAA4B;AAAA,MAAA,CACvC;AACD;AAAA,IACF;AAGA,kBAAc,KAAK,GAAG;AAAA,EACxB;AACA,SAAO;AACT;AAqBO,SAAS,yBACd,WACqB;AAErB,MAAI,UAAU,SAAS,UAAU;AAC/B,WAAO,CAAA;AAAA,EACT;AAIA,MAAI,UAAU,SAAS,aAAa;AAClC,WAAO,CAAC,uBAAuB,SAAS,CAAC;AAAA,EAC3C;AAGA,SAAO,uBAAuB,SAAS;AACzC;AAMA,SAAS,uBAAuB,WAAoC;AAClE,QAAM,eAAmC,CAAA;AACzC,aAAW,QAAQ,UAAU,OAAO;AAClC,QAAI,cAAc,IAAI,GAAG;AACvB,mBAAa,KAAK,IAAI;AAAA,IACxB;AAAA,EACF;AAEA,SAAO;AAAA,IACL,MAAM,UAAU;AAAA,IAChB,SAAS,qBAAqB,YAAY;AAAA,EAAA;AAE9C;AAgBA,SAAS,gBAAkC;AACzC,SAAO,EAAE,cAAc,IAAI,WAAW,CAAA,EAAC;AACzC;AAEA,SAAS,mBAAmB,MAA6B;AACvD,SACE,KAAK,UAAU,oBACf,KAAK,UAAU,cACf,KAAK,UAAU,wBACf,KAAK,UAAU,WACf,KAAK,WAAW;AAEpB;AAWA,SAAS,uBAAuB,WAA2C;AACzE,QAAM,cAAmC,CAAA;AACzC,MAAI,UAAU,cAAA;AACd,MAAI,kBAAkE,CAAA;AAKtE,QAAM,2CAA2B,IAAA;AAEjC,WAAS,eAAqB;AAC5B,UAAM,UAAU,qBAAqB,QAAQ,YAAY;AACzD,UAAM,aAAa,YAAY;AAC/B,UAAM,eAAe,QAAQ,UAAU,SAAS;AAEhD,QAAI,cAAc,cAAc;AAC9B,kBAAY,KAAK;AAAA,QACf,MAAM;AAAA,QACN;AAAA,QACA,GAAI,gBAAgB,EAAE,WAAW,QAAQ,UAAA;AAAA,QACzC,GAAI,gBAAgB,SAAS,KAAK,EAAE,UAAU,gBAAA;AAAA,MAAgB,CAC/D;AACD,wBAAkB,CAAA;AAAA,IACpB;AACA,cAAU,cAAA;AAAA,EACZ;AAEA,aAAW,QAAQ,UAAU,OAAO;AAClC,YAAQ,KAAK,MAAA;AAAA,MACX,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AACH,gBAAQ,aAAa,KAAK,IAAI;AAC9B;AAAA,MAEF,KAAK;AACH,YAAI,mBAAmB,IAAI,GAAG;AAC5B,kBAAQ,UAAU,KAAK;AAAA,YACrB,IAAI,KAAK;AAAA,YACT,MAAM;AAAA,YACN,UAAU;AAAA,cACR,MAAM,KAAK;AAAA,cACX,WAAW,KAAK;AAAA,YAAA;AAAA,YAElB,GAAI,KAAK,aAAa,UAAa,EAAE,UAAU,KAAK,SAAA;AAAA,UAAS,CAC9D;AAAA,QACH;AACA;AAAA,MAEF,KAAK;AAEH,qBAAA;AAGA,aACG,KAAK,UAAU,cAAc,KAAK,UAAU,YAC7C,CAAC,qBAAqB,IAAI,KAAK,UAAU,GACzC;AACA,sBAAY,KAAK;AAAA,YACf,MAAM;AAAA,YACN,SAAS,KAAK;AAAA,YACd,YAAY,KAAK;AAAA,UAAA,CAClB;AACD,+BAAqB,IAAI,KAAK,UAAU;AAAA,QAC1C;AACA;AAAA,MAEF,KAAK;AACH,YAAI,KAAK,SAAS;AAChB,0BAAgB,KAAK;AAAA,YACnB,SAAS,KAAK;AAAA,YACd,GAAI,KAAK,aAAa,EAAE,WAAW,KAAK,UAAA;AAAA,UAAU,CACnD;AAAA,QACH;AACA;AAAA,MAEF,KAAK;AAKH,YAAI,KAAK,WAAW,YAAY;AAC9B,gBAAM,aACJ,KAAK,QAAQ,KACT,KAAK,MACL,KAAK,SAAS,SACZ,kBAAkB,KAAK,IAAI,IAC3B;AACR,cAAI,eAAe,IAAI;AACrB,oBAAQ,aAAa,KAAK,EAAE,MAAM,QAAQ,SAAS,YAAY;AAAA,UACjE;AAAA,QACF;AACA;AAAA,IAGA;AAAA,EAEN;AAGA,eAAA;AAMA,aAAW,QAAQ,UAAU,OAAO;AAClC,QAAI,KAAK,SAAS,YAAa;AAI/B,QAAI,KAAK,WAAW,UAAa,CAAC,qBAAqB,IAAI,KAAK,EAAE,GAAG;AACnE,kBAAY,KAAK;AAAA,QACf,MAAM;AAAA,QACN,SAAS,oBAAoB,KAAK,MAAM;AAAA,QACxC,YAAY,KAAK;AAAA,MAAA,CAClB;AACD,2BAAqB,IAAI,KAAK,EAAE;AAAA,IAClC;AAGA,QACE,KAAK,WAAW,UAChB,KAAK,UAAU,wBACf,KAAK,UAAU,aAAa,UAC5B,CAAC,qBAAqB,IAAI,KAAK,EAAE,GACjC;AACA,YAAM,WAAW,KAAK,SAAS;AAC/B,kBAAY,KAAK;AAAA,QACf,MAAM;AAAA,QACN,SAAS,KAAK,UAAU;AAAA,UACtB;AAAA,UACA,GAAI,YAAY,EAAE,kBAAkB,KAAA;AAAA,UACpC,SAAS,WACL,8BACA;AAAA,QAAA,CACL;AAAA,QACD,YAAY,KAAK;AAAA,MAAA,CAClB;AACD,2BAAqB,IAAI,KAAK,EAAE;AAAA,IAClC;AAAA,EACF;AAGA,MAAI,YAAY,WAAW,GAAG;AAC5B,gBAAY,KAAK;AAAA,MACf,MAAM;AAAA,MACN,SAAS;AAAA,IAAA,CACV;AAAA,EACH;AAEA,SAAO;AACT;AAcO,SAAS,wBACd,cACA,IACW;AACX,QAAM,QAA4B,CAAA;AAElC,MAAI,aAAa,SAAS,eAAe,aAAa,UAAU,QAAQ;AACtE,eAAW,YAAY,aAAa,UAAU;AAC5C,UAAI,CAAC,SAAS,QAAS;AACvB,YAAM,KAAK;AAAA,QACT,MAAM;AAAA,QACN,SAAS,SAAS;AAAA,QAClB,GAAI,SAAS,aAAa,EAAE,WAAW,SAAS,UAAA;AAAA,MAAU,CAC3D;AAAA,IACH;AAAA,EACF;AAIA,MAAI,aAAa,SAAS,UAAU,aAAa,YAAY;AAC3D,UAAM,KAAK;AAAA,MACT,MAAM;AAAA,MACN,YAAY,aAAa;AAAA,MACzB,SAAS,eAAe,aAAa,OAAO;AAAA,MAC5C,OAAO;AAAA,IAAA,CACR;AAAA,EACH,WAAW,MAAM,QAAQ,aAAa,OAAO,GAAG;AAE9C,eAAW,QAAQ,aAAa,SAAS;AACvC,YAAM,KAAK,IAAI;AAAA,IACjB;AAAA,EACF,OAAO;AAEL,UAAM,cAAc,eAAe,aAAa,OAAO;AACvD,QAAI,aAAa;AACf,YAAM,KAAK;AAAA,QACT,MAAM;AAAA,QACN,SAAS;AAAA,MAAA,CACV;AAAA,IACH;AAAA,EACF;AAGA,MAAI,aAAa,aAAa,aAAa,UAAU,SAAS,GAAG;AAC/D,eAAW,YAAY,aAAa,WAAW;AAC7C,YAAM,KAAK;AAAA,QACT,MAAM;AAAA,QACN,IAAI,SAAS;AAAA,QACb,MAAM,SAAS,SAAS;AAAA,QACxB,WAAW,SAAS,SAAS;AAAA,QAC7B,OAAO;AAAA;AAAA,QACP,GAAI,SAAS,aAAa,UAAa,EAAE,UAAU,SAAS,SAAA;AAAA,MAAS,CACtE;AAAA,IACH;AAAA,EACF;AAEA,SAAO;AAAA,IACL,IAAI,MAAM,kBAAA;AAAA,IACV,MAAM,aAAa,SAAS,SAAS,cAAc,aAAa;AAAA,IAChE;AAAA,EAAA;AAEJ;AAkBO,SAAS,+BACd,SACW;AACX,MAAI,WAAW,SAAS;AACtB,WAAO,EAAE,GAAG,SAAS,IAAI,QAAQ,MAAM,oBAAkB;AAAA,EAC3D;AAEA,QAAM,KAAK,QAAQ,MAAM,kBAAA;AAEzB,UAAQ,QAAQ,MAAA;AAAA,IACd,KAAK;AACH,aAAO;AAAA,QACL;AAAA,QACA,MAAM;AAAA,QACN,OAAO,uBAAuB,QAAQ,OAAO;AAAA,MAAA;AAAA,IAEjD,KAAK;AACH,aAAO;AAAA,QACL;AAAA,UACE,MAAM;AAAA,UACN,SAAS,QAAQ,WAAW;AAAA,UAC5B,GAAI,QAAQ,aAAa,EAAE,WAAW,QAAQ,UAAA;AAAA,QAAU;AAAA,QAE1D;AAAA,MAAA;AAAA,IAEJ,KAAK;AACH,aAAO;AAAA,QACL;AAAA,UACE,MAAM;AAAA,UACN,SAAS,QAAQ;AAAA,UACjB,YAAY,QAAQ;AAAA,QAAA;AAAA,QAEtB;AAAA,MAAA;AAAA,IAEJ,KAAK;AAAA,IACL,KAAK;AAEH,aAAO;AAAA,QACL;AAAA,QACA,MAAM;AAAA,QACN,OAAO,QAAQ,UACX,CAAC,EAAE,MAAM,QAAQ,SAAS,QAAQ,QAAA,CAAS,IAC3C,CAAA;AAAA,MAAC;AAAA,IAET,KAAK;AACH,aAAO;AAAA,QACL;AAAA,QACA,MAAM;AAAA,QACN,OAAO,QAAQ,UACX,CAAC,EAAE,MAAM,YAAY,SAAS,QAAQ,QAAA,CAAS,IAC/C,CAAA;AAAA,MAAC;AAAA,IAET,KAAK;AAAA,IACL;AAEE,aAAO,EAAE,IAAI,MAAM,aAAa,OAAO,CAAA,EAAC;AAAA,EAAE;AAEhD;AAWA,SAAS,uBACP,SACoB;AACpB,MAAI,OAAO,YAAY,UAAU;AAC/B,WAAO,UAAU,CAAC,EAAE,MAAM,QAAQ,QAAA,CAAS,IAAI,CAAA;AAAA,EACjD;AAEA,QAAM,QAA4B,CAAA;AAClC,aAAW,QAAQ,SAAS;AAC1B,QAAI,KAAK,SAAS,QAAQ;AACxB,YAAM,KAAK,EAAE,MAAM,QAAQ,SAAS,KAAK,MAAM;AAAA,IACjD,WAAW,KAAK,SAAS,UAAU;AACjC,YAAM,KAAK,IAAI;AAAA,IACjB;AAAA,EACF;AACA,SAAO;AACT;AAUO,SAAS,0BACd,eACkB;AAClB,QAAM,aAA+B,CAAA;AACrC,MAAI,0BAA4C;AAEhD,aAAW,OAAO,eAAe;AAC/B,QAAI,IAAI,SAAS,QAAQ;AAEvB,UACE,IAAI,eAAe,UACnB,2BACA,wBAAwB,SAAS,aACjC;AACA,cAAM,UAAU,eAAe,IAAI,OAAO;AAC1C,cAAM,eAAe,wBAAwB,MAAM;AAAA,UACjD,CAAC,SACC,KAAK,SAAS,eAAe,KAAK,OAAO,IAAI;AAAA,QAAA;AAGjD,YAAI,cAAc;AAChB,uBAAa,SAAS,uBAAuB,OAAO;AACpD,uBAAa,QAAQ;AAAA,QACvB;AAEA,gCAAwB,MAAM,KAAK;AAAA,UACjC,MAAM;AAAA,UACN,YAAY,IAAI;AAAA,UAChB;AAAA,UACA,OAAO;AAAA,QAAA,CACR;AAAA,MACH,OAAO;AAEL,cAAM,sBAAsB,wBAAwB,GAAG;AACvD,mBAAW,KAAK,mBAAmB;AAAA,MACrC;AAAA,IACF,OAAO;AAEL,YAAM,YAAY,wBAAwB,GAAG;AAC7C,iBAAW,KAAK,SAAS;AAGzB,UAAI,IAAI,SAAS,aAAa;AAC5B,kCAA0B;AAAA,MAC5B,OAAO;AACL,kCAA0B;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;AAUO,SAAS,qBACd,SACA,YACW;AACX,MAAI,WAAW,SAAS;AAEtB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,IAAI,QAAQ,MAAM,WAAA;AAAA,MAClB,WAAW,QAAQ,aAAa,oBAAI,KAAA;AAAA,IAAK;AAAA,EAE7C,OAAO;AAEL,WAAO;AAAA,MACL,GAAG,wBAAwB,SAAS,YAAY;AAAA,MAChD,+BAAe,KAAA;AAAA,IAAK;AAAA,EAExB;AACF;AAKO,SAAS,oBAA4B;AAC1C,SAAO,OAAO,KAAK,IAAA,CAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,UAAU,CAAC,CAAC;AACrE;"}
+{"version":3,"file":"messages.js","sources":["../../../../src/activities/chat/messages.ts"],"sourcesContent":["import { normalizeToolResult } from '../../utilities/tool-result'\nimport type { Message as AGUIMessage } from '@ag-ui/core'\nimport type {\n  ContentPart,\n  MessagePart,\n  ModelMessage,\n  TextPart,\n  ToolCallPart,\n  UIMessage,\n} from '../../types'\n// ===========================\n// Message Converters\n// ===========================\n\n/**\n * Check if a MessagePart is a content part (text, image, audio, video, document)\n * that maps directly to a ModelMessage ContentPart.\n */\nfunction isContentPart(part: MessagePart): part is ContentPart {\n  return (\n    part.type === 'text' ||\n    part.type === 'image' ||\n    part.type === 'audio' ||\n    part.type === 'video' ||\n    part.type === 'document'\n  )\n}\n\nfunction safeJsonStringify(value: unknown): string {\n  try {\n    return JSON.stringify(value)\n  } catch {\n    return ''\n  }\n}\n\nfunction parseToolResultContent(content: string): unknown {\n  try {\n    return JSON.parse(content)\n  } catch {\n    return content\n  }\n}\n\n/**\n * Collapse an array of ContentParts into the most compact ModelMessage content:\n * - Empty array → null\n * - All text parts → joined string (or null if empty)\n * - Mixed content → ContentPart array as-is\n */\nfunction collapseContentParts(\n  parts: Array<ContentPart>,\n): string | null | Array<ContentPart> {\n  if (parts.length === 0) return null\n\n  const allText = parts.every((p) => p.type === 'text')\n  if (allText) {\n    const joined = parts.map((p) => p.content).join('')\n    return joined || null\n  }\n\n  return parts\n}\n\n/**\n * Extract text content from ModelMessage content (string, null, or ContentPart array).\n * Used when only the text portion is needed (e.g., tool result content).\n */\nfunction getTextContent(content: string | null | Array<ContentPart>): string {\n  if (content === null) return ''\n  if (typeof content === 'string') return content\n  return content\n    .filter((part): part is TextPart => part.type === 'text')\n    .map((part) => part.content)\n    .join('')\n}\n\n/**\n * Convert UIMessages or ModelMessages to ModelMessages\n */\nexport function convertMessagesToModelMessages(\n  messages: Array<UIMessage | ModelMessage>,\n): Array<ModelMessage> {\n  // Pre-pass: collect toolCallIds already represented in anchor UIMessage parts.\n  // Fan-out tool messages whose toolCallId matches an anchored ToolResultPart\n  // are AG-UI duplicates and must be dropped to avoid double-feeding the LLM.\n  const anchoredToolCallIds = new Set<string>()\n  for (const msg of messages) {\n    if ('parts' in msg) {\n      for (const part of msg.parts) {\n        if (part.type === 'tool-result') {\n          anchoredToolCallIds.add(part.toolCallId)\n        }\n      }\n    }\n  }\n\n  const modelMessages: Array<ModelMessage> = []\n  for (const msg of messages) {\n    if ('parts' in msg) {\n      // UIMessage anchor — existing fan-out path\n      modelMessages.push(...uiMessageToModelMessages(msg))\n      continue\n    }\n\n    const role = (msg as { role: string }).role\n\n    // AG-UI tool fan-out duplicate — drop if anchor already covers it\n    if (\n      role === 'tool' &&\n      msg.toolCallId &&\n      anchoredToolCallIds.has(msg.toolCallId)\n    ) {\n      continue\n    }\n\n    // AG-UI reasoning and activity — no ModelMessage equivalent today\n    if (role === 'reasoning' || role === 'activity') {\n      continue\n    }\n\n    // AG-UI developer — collapse to system\n    if (role === 'developer') {\n      modelMessages.push({\n        role: 'system' as ModelMessage['role'],\n        content: (msg as { content: string }).content,\n      })\n      continue\n    }\n\n    // Already a ModelMessage (user, assistant, system, tool with no anchor) — pass through\n    modelMessages.push(msg)\n  }\n  return modelMessages\n}\n\n/**\n * Convert a UIMessage to ModelMessage(s)\n *\n * Walks the parts array IN ORDER to preserve the interleaving of text,\n * tool calls, and tool results. This is critical for multi-round tool\n * flows where the model generates text, calls a tool, gets the result,\n * then generates more text and calls another tool.\n *\n * The output preserves the sequential structure:\n *   text1 → toolCall1 → toolResult1 → text2 → toolCall2 → toolResult2\n * becomes:\n *   assistant: {content: \"text1\", toolCalls: [toolCall1]}\n *   tool: toolResult1\n *   assistant: {content: \"text2\", toolCalls: [toolCall2]}\n *   tool: toolResult2\n *\n * @param uiMessage - The UIMessage to convert\n * @returns An array of ModelMessages preserving part ordering\n */\nexport function uiMessageToModelMessages(\n  uiMessage: UIMessage,\n): Array<ModelMessage> {\n  // Skip system messages - they're handled via systemPrompts, not ModelMessages\n  if (uiMessage.role === 'system') {\n    return []\n  }\n\n  // For non-assistant messages (user), use the simpler path since they\n  // don't have tool calls or tool results to interleave\n  if (uiMessage.role !== 'assistant') {\n    return [buildUserOrToolMessage(uiMessage)]\n  }\n\n  // For assistant messages, walk parts in order to preserve interleaving\n  return buildAssistantMessages(uiMessage)\n}\n\n/**\n * Build a single ModelMessage for user messages (simple path).\n * Preserves ordering of text and multimodal content parts.\n */\nfunction buildUserOrToolMessage(uiMessage: UIMessage): ModelMessage {\n  const contentParts: Array<ContentPart> = []\n  for (const part of uiMessage.parts) {\n    if (isContentPart(part)) {\n      contentParts.push(part)\n    }\n  }\n\n  return {\n    role: uiMessage.role as 'user' | 'assistant' | 'tool',\n    content: collapseContentParts(contentParts),\n  }\n}\n\n// Accumulator for building an assistant segment (content + tool calls)\ninterface AssistantSegment {\n  contentParts: Array<ContentPart>\n  toolCalls: Array<{\n    id: string\n    type: 'function'\n    function: { name: string; arguments: string }\n    /** Provider-specific metadata that round-trips with the tool call.\n     * Untyped at this framework layer; adapters narrow it via their\n     * `TToolCallMetadata` generic. */\n    metadata?: unknown\n  }>\n}\n\nfunction createSegment(): AssistantSegment {\n  return { contentParts: [], toolCalls: [] }\n}\n\nfunction isToolCallIncluded(part: ToolCallPart): boolean {\n  return (\n    part.state === 'input-complete' ||\n    part.state === 'complete' ||\n    part.state === 'approval-responded' ||\n    part.state === 'error' ||\n    part.output !== undefined\n  )\n}\n\n/**\n * Build ModelMessages for an assistant UIMessage, preserving the\n * sequential interleaving of text, tool calls, and tool results.\n *\n * Walks parts in order. Text and tool-call parts accumulate into the\n * current \"segment\". When a tool-result part is encountered, the\n * current segment is flushed as an assistant message, then the tool\n * result is emitted as a tool message.\n */\nfunction buildAssistantMessages(uiMessage: UIMessage): Array<ModelMessage> {\n  const messageList: Array<ModelMessage> = []\n  let current = createSegment()\n  let pendingThinking: Array<{ content: string; signature?: string }> = []\n\n  // Track emitted tool result IDs to avoid duplicates.\n  // A tool call can have BOTH an explicit tool-result part AND an output\n  // field on the tool-call part. We only want one per tool call ID.\n  const emittedToolResultIds = new Set<string>()\n\n  function flushSegment(): void {\n    const content = collapseContentParts(current.contentParts)\n    const hasContent = content !== null\n    const hasToolCalls = current.toolCalls.length > 0\n\n    if (hasContent || hasToolCalls) {\n      messageList.push({\n        role: 'assistant',\n        content,\n        ...(hasToolCalls && { toolCalls: current.toolCalls }),\n        ...(pendingThinking.length > 0 && { thinking: pendingThinking }),\n      })\n      pendingThinking = []\n    }\n    current = createSegment()\n  }\n\n  for (const part of uiMessage.parts) {\n    switch (part.type) {\n      case 'text':\n      case 'image':\n      case 'audio':\n      case 'video':\n      case 'document':\n        current.contentParts.push(part)\n        break\n\n      case 'tool-call':\n        if (isToolCallIncluded(part)) {\n          current.toolCalls.push({\n            id: part.id,\n            type: 'function' as const,\n            function: {\n              name: part.name,\n              arguments: part.arguments,\n            },\n            ...(part.metadata !== undefined && { metadata: part.metadata }),\n          })\n        }\n        break\n\n      case 'tool-result':\n        // Flush the current assistant segment before emitting the tool result\n        flushSegment()\n\n        // Emit the tool result\n        if (\n          (part.state === 'complete' || part.state === 'error') &&\n          !emittedToolResultIds.has(part.toolCallId)\n        ) {\n          messageList.push({\n            role: 'tool',\n            content: part.content,\n            toolCallId: part.toolCallId,\n          })\n          emittedToolResultIds.add(part.toolCallId)\n        }\n        break\n\n      case 'thinking':\n        if (part.content) {\n          pendingThinking.push({\n            content: part.content,\n            ...(part.signature && { signature: part.signature }),\n          })\n        }\n        break\n\n      case 'structured-output':\n        // Only emit completed structured responses into history. Streaming or\n        // errored buffers would push malformed JSON into the next LLM turn's\n        // assistant content. `raw` is the source of truth; `data` is the\n        // defensive fallback for terminal-only completes that didn't ship raw.\n        if (part.status === 'complete') {\n          const serialized =\n            part.raw !== ''\n              ? part.raw\n              : part.data !== undefined\n                ? safeJsonStringify(part.data)\n                : ''\n          if (serialized !== '') {\n            current.contentParts.push({ type: 'text', content: serialized })\n          }\n        }\n        break\n\n      case 'ui-resource':\n        // MCP Apps widget — rendered client-side only. It must never enter\n        // model input, so it is intentionally dropped from the model message.\n        break\n\n      default:\n        break\n    }\n  }\n\n  // Flush any remaining accumulated content\n  flushSegment()\n\n  // Emit tool results from client tool-call parts with output or approval,\n  // but only if not already covered by an explicit tool-result part above.\n  // These are appended at the end since they don't have explicit tool-result\n  // parts in the parts array to trigger inline emission.\n  for (const part of uiMessage.parts) {\n    if (part.type !== 'tool-call') continue\n\n    // Output takes priority — if the tool has already produced a result,\n    // emit the concrete output regardless of approval metadata.\n    if (part.output !== undefined && !emittedToolResultIds.has(part.id)) {\n      messageList.push({\n        role: 'tool',\n        content: normalizeToolResult(part.output),\n        toolCallId: part.id,\n      })\n      emittedToolResultIds.add(part.id)\n    }\n\n    // Approval response without output — emit approval status for iteration tracking\n    if (\n      part.output === undefined &&\n      part.state === 'approval-responded' &&\n      part.approval?.approved !== undefined &&\n      !emittedToolResultIds.has(part.id)\n    ) {\n      const approved = part.approval.approved\n      messageList.push({\n        role: 'tool',\n        content: JSON.stringify({\n          approved,\n          ...(approved && { pendingExecution: true }),\n          message: approved\n            ? 'User approved this action'\n            : 'User denied this action',\n        }),\n        toolCallId: part.id,\n      })\n      emittedToolResultIds.add(part.id)\n    }\n  }\n\n  // If no messages were produced (e.g., empty parts), emit a minimal assistant message\n  if (messageList.length === 0) {\n    messageList.push({\n      role: 'assistant',\n      content: null,\n    })\n  }\n\n  return messageList\n}\n\n/**\n * Convert a ModelMessage to UIMessage\n *\n * This conversion creates a parts-based structure:\n * - content field → TextPart\n * - toolCalls array → ToolCallPart[]\n * - role=\"tool\" messages should be converted separately and merged\n *\n * @param modelMessage - The ModelMessage to convert\n * @param id - Optional ID for the UIMessage (generated if not provided)\n * @returns A UIMessage with parts\n */\nexport function modelMessageToUIMessage(\n  modelMessage: ModelMessage,\n  id?: string,\n): UIMessage {\n  const parts: Array<MessagePart> = []\n\n  if (modelMessage.role === 'assistant' && modelMessage.thinking?.length) {\n    for (const thinking of modelMessage.thinking) {\n      if (!thinking.content) continue\n      parts.push({\n        type: 'thinking',\n        content: thinking.content,\n        ...(thinking.signature && { signature: thinking.signature }),\n      })\n    }\n  }\n\n  // Handle tool results (when role is \"tool\") - only produce tool-result part,\n  // not a text part (the content IS the tool result, not display text)\n  if (modelMessage.role === 'tool' && modelMessage.toolCallId) {\n    parts.push({\n      type: 'tool-result',\n      toolCallId: modelMessage.toolCallId,\n      content: getTextContent(modelMessage.content),\n      state: 'complete',\n    })\n  } else if (Array.isArray(modelMessage.content)) {\n    // Multimodal content - preserve all content parts as MessageParts\n    for (const part of modelMessage.content) {\n      parts.push(part)\n    }\n  } else {\n    // String or null content\n    const textContent = getTextContent(modelMessage.content)\n    if (textContent) {\n      parts.push({\n        type: 'text',\n        content: textContent,\n      })\n    }\n  }\n\n  // Handle tool calls\n  if (modelMessage.toolCalls && modelMessage.toolCalls.length > 0) {\n    for (const toolCall of modelMessage.toolCalls) {\n      parts.push({\n        type: 'tool-call',\n        id: toolCall.id,\n        name: toolCall.function.name,\n        arguments: toolCall.function.arguments,\n        state: 'input-complete', // Model messages have complete arguments\n        ...(toolCall.metadata !== undefined && { metadata: toolCall.metadata }),\n      })\n    }\n  }\n\n  return {\n    id: id || generateMessageId(),\n    role: modelMessage.role === 'tool' ? 'assistant' : modelMessage.role,\n    parts,\n  }\n}\n\n/**\n * Normalize a single AG-UI `MESSAGES_SNAPSHOT` message into a `UIMessage`.\n *\n * AG-UI snapshot messages use the wire shape `{ id, role, content }` and have\n * no `parts` array. Casting them directly to `UIMessage` is unsafe: any code\n * that later reads `message.parts` (e.g. the devtools `onToolCallStateChange`\n * handler) crashes with \"Cannot read properties of undefined (reading 'find')\".\n *\n * Each role is mapped to the canonical `UIMessage` shape, reusing\n * `modelMessageToUIMessage` for the roles that share `ModelMessage`'s structure.\n * The original AG-UI `id` is preserved so later `TEXT_MESSAGE_CONTENT` /\n * `TOOL_CALL_*` events still route by `messageId` (falling back to a generated\n * id only when the snapshot omits one). Messages that already carry `parts`\n * (e.g. a TanStack server echoing `UIMessage`s back over the wire) pass through\n * unchanged apart from ensuring an id.\n */\nexport function aguiSnapshotMessageToUIMessage(\n  message: AGUIMessage | UIMessage,\n): UIMessage {\n  if ('parts' in message) {\n    return { ...message, id: message.id || generateMessageId() }\n  }\n\n  const id = message.id || generateMessageId()\n\n  switch (message.role) {\n    case 'user':\n      return {\n        id,\n        role: 'user',\n        parts: aguiUserContentToParts(message.content),\n      }\n    case 'assistant':\n      return modelMessageToUIMessage(\n        {\n          role: 'assistant',\n          content: message.content ?? null,\n          ...(message.toolCalls && { toolCalls: message.toolCalls }),\n        },\n        id,\n      )\n    case 'tool':\n      return modelMessageToUIMessage(\n        {\n          role: 'tool',\n          content: message.content,\n          toolCallId: message.toolCallId,\n        },\n        id,\n      )\n    case 'system':\n    case 'developer':\n      // `ModelMessage` has no system/developer role; build the part directly.\n      return {\n        id,\n        role: 'system',\n        parts: message.content\n          ? [{ type: 'text', content: message.content }]\n          : [],\n      }\n    case 'reasoning':\n      return {\n        id,\n        role: 'assistant',\n        parts: message.content\n          ? [{ type: 'thinking', content: message.content }]\n          : [],\n      }\n    case 'activity':\n    default:\n      // `activity` (and any future role) has no text/parts equivalent today.\n      return { id, role: 'assistant', parts: [] }\n  }\n}\n\n/**\n * Convert AG-UI user message content into `UIMessage` parts.\n *\n * AG-UI user content is either a plain string or a multimodal array whose text\n * entries use `{ type: 'text', text }` (vs. TanStack's `{ type: 'text', content }`).\n * Text entries are rewritten to the TanStack shape; image/audio/video/document\n * entries already match `ContentPart` and pass through. `binary` entries have no\n * TanStack equivalent and are dropped.\n */\nfunction aguiUserContentToParts(\n  content: Extract<AGUIMessage, { role: 'user' }>['content'],\n): Array<MessagePart> {\n  if (typeof content === 'string') {\n    return content ? [{ type: 'text', content }] : []\n  }\n\n  const parts: Array<MessagePart> = []\n  for (const part of content) {\n    if (part.type === 'text') {\n      parts.push({ type: 'text', content: part.text })\n    } else if (part.type !== 'binary') {\n      parts.push(part)\n    }\n  }\n  return parts\n}\n\n/**\n * Convert an array of ModelMessages to UIMessages\n *\n * This handles merging tool result messages with their corresponding assistant messages\n *\n * @param modelMessages - Array of ModelMessages to convert\n * @returns Array of UIMessages\n */\nexport function modelMessagesToUIMessages(\n  modelMessages: Array<ModelMessage>,\n): Array<UIMessage> {\n  const uiMessages: Array<UIMessage> = []\n  let currentAssistantMessage: UIMessage | null = null\n\n  for (const msg of modelMessages) {\n    if (msg.role === 'tool') {\n      // Tool result - merge into the last assistant message if possible\n      if (\n        msg.toolCallId !== undefined &&\n        currentAssistantMessage &&\n        currentAssistantMessage.role === 'assistant'\n      ) {\n        const content = getTextContent(msg.content)\n        const toolCallPart = currentAssistantMessage.parts.find(\n          (part): part is ToolCallPart =>\n            part.type === 'tool-call' && part.id === msg.toolCallId,\n        )\n\n        if (toolCallPart) {\n          toolCallPart.output = parseToolResultContent(content)\n          toolCallPart.state = 'complete'\n        }\n\n        currentAssistantMessage.parts.push({\n          type: 'tool-result',\n          toolCallId: msg.toolCallId,\n          content,\n          state: 'complete',\n        })\n      } else {\n        // No assistant message to merge into, create a standalone one\n        const toolResultUIMessage = modelMessageToUIMessage(msg)\n        uiMessages.push(toolResultUIMessage)\n      }\n    } else {\n      // Regular message\n      const uiMessage = modelMessageToUIMessage(msg)\n      uiMessages.push(uiMessage)\n\n      // Track assistant messages for potential tool result merging\n      if (msg.role === 'assistant') {\n        currentAssistantMessage = uiMessage\n      } else {\n        currentAssistantMessage = null\n      }\n    }\n  }\n\n  return uiMessages\n}\n\n/**\n * Normalize a message (UIMessage or ModelMessage) to a UIMessage\n * Ensures the message has an ID and createdAt timestamp\n *\n * @param message - Either a UIMessage or ModelMessage\n * @param generateId - Function to generate a message ID if needed\n * @returns A UIMessage with guaranteed id and createdAt\n */\nexport function normalizeToUIMessage(\n  message: UIMessage | ModelMessage,\n  generateId: () => string,\n): UIMessage {\n  if ('parts' in message) {\n    // Already a UIMessage\n    return {\n      ...message,\n      id: message.id || generateId(),\n      createdAt: message.createdAt || new Date(),\n    }\n  } else {\n    // ModelMessage - convert to UIMessage\n    return {\n      ...modelMessageToUIMessage(message, generateId()),\n      createdAt: new Date(),\n    }\n  }\n}\n\n/**\n * Generate a unique message ID\n */\nexport function generateMessageId(): string {\n  return `msg-${Date.now()}-${Math.random().toString(36).substring(7)}`\n}\n"],"names":[],"mappings":";AAkBA,SAAS,cAAc,MAAwC;AAC7D,SACE,KAAK,SAAS,UACd,KAAK,SAAS,WACd,KAAK,SAAS,WACd,KAAK,SAAS,WACd,KAAK,SAAS;AAElB;AAEA,SAAS,kBAAkB,OAAwB;AACjD,MAAI;AACF,WAAO,KAAK,UAAU,KAAK;AAAA,EAC7B,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEA,SAAS,uBAAuB,SAA0B;AACxD,MAAI;AACF,WAAO,KAAK,MAAM,OAAO;AAAA,EAC3B,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAQA,SAAS,qBACP,OACoC;AACpC,MAAI,MAAM,WAAW,EAAG,QAAO;AAE/B,QAAM,UAAU,MAAM,MAAM,CAAC,MAAM,EAAE,SAAS,MAAM;AACpD,MAAI,SAAS;AACX,UAAM,SAAS,MAAM,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE;AAClD,WAAO,UAAU;AAAA,EACnB;AAEA,SAAO;AACT;AAMA,SAAS,eAAe,SAAqD;AAC3E,MAAI,YAAY,KAAM,QAAO;AAC7B,MAAI,OAAO,YAAY,SAAU,QAAO;AACxC,SAAO,QACJ,OAAO,CAAC,SAA2B,KAAK,SAAS,MAAM,EACvD,IAAI,CAAC,SAAS,KAAK,OAAO,EAC1B,KAAK,EAAE;AACZ;AAKO,SAAS,+BACd,UACqB;AAIrB,QAAM,0CAA0B,IAAA;AAChC,aAAW,OAAO,UAAU;AAC1B,QAAI,WAAW,KAAK;AAClB,iBAAW,QAAQ,IAAI,OAAO;AAC5B,YAAI,KAAK,SAAS,eAAe;AAC/B,8BAAoB,IAAI,KAAK,UAAU;AAAA,QACzC;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,QAAM,gBAAqC,CAAA;AAC3C,aAAW,OAAO,UAAU;AAC1B,QAAI,WAAW,KAAK;AAElB,oBAAc,KAAK,GAAG,yBAAyB,GAAG,CAAC;AACnD;AAAA,IACF;AAEA,UAAM,OAAQ,IAAyB;AAGvC,QACE,SAAS,UACT,IAAI,cACJ,oBAAoB,IAAI,IAAI,UAAU,GACtC;AACA;AAAA,IACF;AAGA,QAAI,SAAS,eAAe,SAAS,YAAY;AAC/C;AAAA,IACF;AAGA,QAAI,SAAS,aAAa;AACxB,oBAAc,KAAK;AAAA,QACjB,MAAM;AAAA,QACN,SAAU,IAA4B;AAAA,MAAA,CACvC;AACD;AAAA,IACF;AAGA,kBAAc,KAAK,GAAG;AAAA,EACxB;AACA,SAAO;AACT;AAqBO,SAAS,yBACd,WACqB;AAErB,MAAI,UAAU,SAAS,UAAU;AAC/B,WAAO,CAAA;AAAA,EACT;AAIA,MAAI,UAAU,SAAS,aAAa;AAClC,WAAO,CAAC,uBAAuB,SAAS,CAAC;AAAA,EAC3C;AAGA,SAAO,uBAAuB,SAAS;AACzC;AAMA,SAAS,uBAAuB,WAAoC;AAClE,QAAM,eAAmC,CAAA;AACzC,aAAW,QAAQ,UAAU,OAAO;AAClC,QAAI,cAAc,IAAI,GAAG;AACvB,mBAAa,KAAK,IAAI;AAAA,IACxB;AAAA,EACF;AAEA,SAAO;AAAA,IACL,MAAM,UAAU;AAAA,IAChB,SAAS,qBAAqB,YAAY;AAAA,EAAA;AAE9C;AAgBA,SAAS,gBAAkC;AACzC,SAAO,EAAE,cAAc,IAAI,WAAW,CAAA,EAAC;AACzC;AAEA,SAAS,mBAAmB,MAA6B;AACvD,SACE,KAAK,UAAU,oBACf,KAAK,UAAU,cACf,KAAK,UAAU,wBACf,KAAK,UAAU,WACf,KAAK,WAAW;AAEpB;AAWA,SAAS,uBAAuB,WAA2C;AACzE,QAAM,cAAmC,CAAA;AACzC,MAAI,UAAU,cAAA;AACd,MAAI,kBAAkE,CAAA;AAKtE,QAAM,2CAA2B,IAAA;AAEjC,WAAS,eAAqB;AAC5B,UAAM,UAAU,qBAAqB,QAAQ,YAAY;AACzD,UAAM,aAAa,YAAY;AAC/B,UAAM,eAAe,QAAQ,UAAU,SAAS;AAEhD,QAAI,cAAc,cAAc;AAC9B,kBAAY,KAAK;AAAA,QACf,MAAM;AAAA,QACN;AAAA,QACA,GAAI,gBAAgB,EAAE,WAAW,QAAQ,UAAA;AAAA,QACzC,GAAI,gBAAgB,SAAS,KAAK,EAAE,UAAU,gBAAA;AAAA,MAAgB,CAC/D;AACD,wBAAkB,CAAA;AAAA,IACpB;AACA,cAAU,cAAA;AAAA,EACZ;AAEA,aAAW,QAAQ,UAAU,OAAO;AAClC,YAAQ,KAAK,MAAA;AAAA,MACX,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AACH,gBAAQ,aAAa,KAAK,IAAI;AAC9B;AAAA,MAEF,KAAK;AACH,YAAI,mBAAmB,IAAI,GAAG;AAC5B,kBAAQ,UAAU,KAAK;AAAA,YACrB,IAAI,KAAK;AAAA,YACT,MAAM;AAAA,YACN,UAAU;AAAA,cACR,MAAM,KAAK;AAAA,cACX,WAAW,KAAK;AAAA,YAAA;AAAA,YAElB,GAAI,KAAK,aAAa,UAAa,EAAE,UAAU,KAAK,SAAA;AAAA,UAAS,CAC9D;AAAA,QACH;AACA;AAAA,MAEF,KAAK;AAEH,qBAAA;AAGA,aACG,KAAK,UAAU,cAAc,KAAK,UAAU,YAC7C,CAAC,qBAAqB,IAAI,KAAK,UAAU,GACzC;AACA,sBAAY,KAAK;AAAA,YACf,MAAM;AAAA,YACN,SAAS,KAAK;AAAA,YACd,YAAY,KAAK;AAAA,UAAA,CAClB;AACD,+BAAqB,IAAI,KAAK,UAAU;AAAA,QAC1C;AACA;AAAA,MAEF,KAAK;AACH,YAAI,KAAK,SAAS;AAChB,0BAAgB,KAAK;AAAA,YACnB,SAAS,KAAK;AAAA,YACd,GAAI,KAAK,aAAa,EAAE,WAAW,KAAK,UAAA;AAAA,UAAU,CACnD;AAAA,QACH;AACA;AAAA,MAEF,KAAK;AAKH,YAAI,KAAK,WAAW,YAAY;AAC9B,gBAAM,aACJ,KAAK,QAAQ,KACT,KAAK,MACL,KAAK,SAAS,SACZ,kBAAkB,KAAK,IAAI,IAC3B;AACR,cAAI,eAAe,IAAI;AACrB,oBAAQ,aAAa,KAAK,EAAE,MAAM,QAAQ,SAAS,YAAY;AAAA,UACjE;AAAA,QACF;AACA;AAAA,IAQA;AAAA,EAEN;AAGA,eAAA;AAMA,aAAW,QAAQ,UAAU,OAAO;AAClC,QAAI,KAAK,SAAS,YAAa;AAI/B,QAAI,KAAK,WAAW,UAAa,CAAC,qBAAqB,IAAI,KAAK,EAAE,GAAG;AACnE,kBAAY,KAAK;AAAA,QACf,MAAM;AAAA,QACN,SAAS,oBAAoB,KAAK,MAAM;AAAA,QACxC,YAAY,KAAK;AAAA,MAAA,CAClB;AACD,2BAAqB,IAAI,KAAK,EAAE;AAAA,IAClC;AAGA,QACE,KAAK,WAAW,UAChB,KAAK,UAAU,wBACf,KAAK,UAAU,aAAa,UAC5B,CAAC,qBAAqB,IAAI,KAAK,EAAE,GACjC;AACA,YAAM,WAAW,KAAK,SAAS;AAC/B,kBAAY,KAAK;AAAA,QACf,MAAM;AAAA,QACN,SAAS,KAAK,UAAU;AAAA,UACtB;AAAA,UACA,GAAI,YAAY,EAAE,kBAAkB,KAAA;AAAA,UACpC,SAAS,WACL,8BACA;AAAA,QAAA,CACL;AAAA,QACD,YAAY,KAAK;AAAA,MAAA,CAClB;AACD,2BAAqB,IAAI,KAAK,EAAE;AAAA,IAClC;AAAA,EACF;AAGA,MAAI,YAAY,WAAW,GAAG;AAC5B,gBAAY,KAAK;AAAA,MACf,MAAM;AAAA,MACN,SAAS;AAAA,IAAA,CACV;AAAA,EACH;AAEA,SAAO;AACT;AAcO,SAAS,wBACd,cACA,IACW;AACX,QAAM,QAA4B,CAAA;AAElC,MAAI,aAAa,SAAS,eAAe,aAAa,UAAU,QAAQ;AACtE,eAAW,YAAY,aAAa,UAAU;AAC5C,UAAI,CAAC,SAAS,QAAS;AACvB,YAAM,KAAK;AAAA,QACT,MAAM;AAAA,QACN,SAAS,SAAS;AAAA,QAClB,GAAI,SAAS,aAAa,EAAE,WAAW,SAAS,UAAA;AAAA,MAAU,CAC3D;AAAA,IACH;AAAA,EACF;AAIA,MAAI,aAAa,SAAS,UAAU,aAAa,YAAY;AAC3D,UAAM,KAAK;AAAA,MACT,MAAM;AAAA,MACN,YAAY,aAAa;AAAA,MACzB,SAAS,eAAe,aAAa,OAAO;AAAA,MAC5C,OAAO;AAAA,IAAA,CACR;AAAA,EACH,WAAW,MAAM,QAAQ,aAAa,OAAO,GAAG;AAE9C,eAAW,QAAQ,aAAa,SAAS;AACvC,YAAM,KAAK,IAAI;AAAA,IACjB;AAAA,EACF,OAAO;AAEL,UAAM,cAAc,eAAe,aAAa,OAAO;AACvD,QAAI,aAAa;AACf,YAAM,KAAK;AAAA,QACT,MAAM;AAAA,QACN,SAAS;AAAA,MAAA,CACV;AAAA,IACH;AAAA,EACF;AAGA,MAAI,aAAa,aAAa,aAAa,UAAU,SAAS,GAAG;AAC/D,eAAW,YAAY,aAAa,WAAW;AAC7C,YAAM,KAAK;AAAA,QACT,MAAM;AAAA,QACN,IAAI,SAAS;AAAA,QACb,MAAM,SAAS,SAAS;AAAA,QACxB,WAAW,SAAS,SAAS;AAAA,QAC7B,OAAO;AAAA;AAAA,QACP,GAAI,SAAS,aAAa,UAAa,EAAE,UAAU,SAAS,SAAA;AAAA,MAAS,CACtE;AAAA,IACH;AAAA,EACF;AAEA,SAAO;AAAA,IACL,IAAI,MAAM,kBAAA;AAAA,IACV,MAAM,aAAa,SAAS,SAAS,cAAc,aAAa;AAAA,IAChE;AAAA,EAAA;AAEJ;AAkBO,SAAS,+BACd,SACW;AACX,MAAI,WAAW,SAAS;AACtB,WAAO,EAAE,GAAG,SAAS,IAAI,QAAQ,MAAM,oBAAkB;AAAA,EAC3D;AAEA,QAAM,KAAK,QAAQ,MAAM,kBAAA;AAEzB,UAAQ,QAAQ,MAAA;AAAA,IACd,KAAK;AACH,aAAO;AAAA,QACL;AAAA,QACA,MAAM;AAAA,QACN,OAAO,uBAAuB,QAAQ,OAAO;AAAA,MAAA;AAAA,IAEjD,KAAK;AACH,aAAO;AAAA,QACL;AAAA,UACE,MAAM;AAAA,UACN,SAAS,QAAQ,WAAW;AAAA,UAC5B,GAAI,QAAQ,aAAa,EAAE,WAAW,QAAQ,UAAA;AAAA,QAAU;AAAA,QAE1D;AAAA,MAAA;AAAA,IAEJ,KAAK;AACH,aAAO;AAAA,QACL;AAAA,UACE,MAAM;AAAA,UACN,SAAS,QAAQ;AAAA,UACjB,YAAY,QAAQ;AAAA,QAAA;AAAA,QAEtB;AAAA,MAAA;AAAA,IAEJ,KAAK;AAAA,IACL,KAAK;AAEH,aAAO;AAAA,QACL;AAAA,QACA,MAAM;AAAA,QACN,OAAO,QAAQ,UACX,CAAC,EAAE,MAAM,QAAQ,SAAS,QAAQ,QAAA,CAAS,IAC3C,CAAA;AAAA,MAAC;AAAA,IAET,KAAK;AACH,aAAO;AAAA,QACL;AAAA,QACA,MAAM;AAAA,QACN,OAAO,QAAQ,UACX,CAAC,EAAE,MAAM,YAAY,SAAS,QAAQ,QAAA,CAAS,IAC/C,CAAA;AAAA,MAAC;AAAA,IAET,KAAK;AAAA,IACL;AAEE,aAAO,EAAE,IAAI,MAAM,aAAa,OAAO,CAAA,EAAC;AAAA,EAAE;AAEhD;AAWA,SAAS,uBACP,SACoB;AACpB,MAAI,OAAO,YAAY,UAAU;AAC/B,WAAO,UAAU,CAAC,EAAE,MAAM,QAAQ,QAAA,CAAS,IAAI,CAAA;AAAA,EACjD;AAEA,QAAM,QAA4B,CAAA;AAClC,aAAW,QAAQ,SAAS;AAC1B,QAAI,KAAK,SAAS,QAAQ;AACxB,YAAM,KAAK,EAAE,MAAM,QAAQ,SAAS,KAAK,MAAM;AAAA,IACjD,WAAW,KAAK,SAAS,UAAU;AACjC,YAAM,KAAK,IAAI;AAAA,IACjB;AAAA,EACF;AACA,SAAO;AACT;AAUO,SAAS,0BACd,eACkB;AAClB,QAAM,aAA+B,CAAA;AACrC,MAAI,0BAA4C;AAEhD,aAAW,OAAO,eAAe;AAC/B,QAAI,IAAI,SAAS,QAAQ;AAEvB,UACE,IAAI,eAAe,UACnB,2BACA,wBAAwB,SAAS,aACjC;AACA,cAAM,UAAU,eAAe,IAAI,OAAO;AAC1C,cAAM,eAAe,wBAAwB,MAAM;AAAA,UACjD,CAAC,SACC,KAAK,SAAS,eAAe,KAAK,OAAO,IAAI;AAAA,QAAA;AAGjD,YAAI,cAAc;AAChB,uBAAa,SAAS,uBAAuB,OAAO;AACpD,uBAAa,QAAQ;AAAA,QACvB;AAEA,gCAAwB,MAAM,KAAK;AAAA,UACjC,MAAM;AAAA,UACN,YAAY,IAAI;AAAA,UAChB;AAAA,UACA,OAAO;AAAA,QAAA,CACR;AAAA,MACH,OAAO;AAEL,cAAM,sBAAsB,wBAAwB,GAAG;AACvD,mBAAW,KAAK,mBAAmB;AAAA,MACrC;AAAA,IACF,OAAO;AAEL,YAAM,YAAY,wBAAwB,GAAG;AAC7C,iBAAW,KAAK,SAAS;AAGzB,UAAI,IAAI,SAAS,aAAa;AAC5B,kCAA0B;AAAA,MAC5B,OAAO;AACL,kCAA0B;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;AAUO,SAAS,qBACd,SACA,YACW;AACX,MAAI,WAAW,SAAS;AAEtB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,IAAI,QAAQ,MAAM,WAAA;AAAA,MAClB,WAAW,QAAQ,aAAa,oBAAI,KAAA;AAAA,IAAK;AAAA,EAE7C,OAAO;AAEL,WAAO;AAAA,MACL,GAAG,wBAAwB,SAAS,YAAY;AAAA,MAChD,+BAAe,KAAA;AAAA,IAAK;AAAA,EAExB;AACF;AAKO,SAAS,oBAA4B;AAC1C,SAAO,OAAO,KAAK,IAAA,CAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,UAAU,CAAC,CAAC;AACrE;"}

package/dist/esm/activities/chat/middleware/compose.d.ts

@@ -1,6 +1,6 @@
 import { StreamChunk } from '../../../types.js';
 import { InternalLogger } from '../../../logger/internal-logger.js';
-import { AbortInfo, AfterToolCallInfo, BeforeToolCallDecision, ChatMiddleware, ChatMiddlewareConfig, ChatMiddlewareContext, ErrorInfo, FinishInfo, IterationInfo, StructuredOutputMiddlewareConfig, ToolCallHookContext, ToolPhaseCompleteInfo, UsageInfo } from './types.js';
+import { AbortInfo, AfterToolCallInfo, BeforeToolCallDecision, ChatMiddleware, ChatMiddlewareConfig, ChatMiddlewareContext, ErrorInfo, FinishInfo, IterationInfo, SandboxFileEvent, StructuredOutputMiddlewareConfig, ToolCallHookContext, ToolPhaseCompleteInfo, UsageInfo } from './types.js';
 /**
  * Internal middleware runner that manages composed execution of middleware hooks.
  * Created once per chat() invocation.
@@ -49,6 +49,12 @@ export declare class MiddlewareRunner<TContext = unknown> {
      * - null: drop the chunk entirely
      */
     runOnChunk(ctx: ChatMiddlewareContext<TContext>, chunk: StreamChunk): Promise<Array<StreamChunk>>;
+    /**
+     * Dispatch a sandbox file event to every middleware's `sandbox` hooks, in
+     * array order: the catch-all `onFile` then the type-specific hook. Errors are
+     * logged and swallowed so one bad hook can't break the run.
+     */
+    runSandboxFile(ctx: ChatMiddlewareContext<TContext>, event: SandboxFileEvent): Promise<void>;
     /**
      * Run onBeforeToolCall through middleware in order.
      * Returns the first non-void decision, or undefined to continue normally.

package/dist/esm/activities/chat/middleware/compose.js

@@ -287,6 +287,33 @@ class MiddlewareRunner {
     }
     return chunks;
   }
+  /**
+   * Dispatch a sandbox file event to every middleware's `sandbox` hooks, in
+   * array order: the catch-all `onFile` then the type-specific hook. Errors are
+   * logged and swallowed so one bad hook can't break the run.
+   */
+  async runSandboxFile(ctx, event) {
+    const typed = {
+      create: "onFileCreate",
+      change: "onFileChange",
+      delete: "onFileDelete"
+    }[event.type];
+    for (const mw of this.middlewares) {
+      const hooks = mw.sandbox;
+      if (!hooks) continue;
+      for (const fn of [hooks.onFile, hooks[typed]]) {
+        if (!fn) continue;
+        try {
+          await fn(ctx, event);
+        } catch (error) {
+          this.logger.sandbox(
+            `hook=${typed} middleware=${mw.name ?? "unnamed"} threw`,
+            { middleware: mw.name ?? "unnamed", error }
+          );
+        }
+      }
+    }
+  }
   /**
    * Run onBeforeToolCall through middleware in order.
    * Returns the first non-void decision, or undefined to continue normally.

package/dist/esm/activities/chat/middleware/compose.js.map

@@ -1 +1 @@
-{"version":3,"file":"compose.js","sources":["../../../../../src/activities/chat/middleware/compose.ts"],"sourcesContent":["import { aiEventClient } from '@tanstack/ai-event-client'\nimport type { StreamChunk } from '../../../types'\nimport type { InternalLogger } from '../../../logger/internal-logger'\nimport type {\n  AbortInfo,\n  AfterToolCallInfo,\n  BeforeToolCallDecision,\n  ChatMiddleware,\n  ChatMiddlewareConfig,\n  ChatMiddlewareContext,\n  ErrorInfo,\n  FinishInfo,\n  IterationInfo,\n  StructuredOutputMiddlewareConfig,\n  ToolCallHookContext,\n  ToolPhaseCompleteInfo,\n  UsageInfo,\n} from './types'\n\n/** Check if a middleware should be skipped for instrumentation events. */\nfunction shouldSkipInstrumentation(mw: ChatMiddleware<any>): boolean {\n  return mw.name === 'devtools' || mw.name === 'strip-to-spec'\n}\n\n/** Build the base context for middleware instrumentation events. */\nfunction instrumentCtx(ctx: ChatMiddlewareContext<any>) {\n  return {\n    requestId: ctx.requestId,\n    streamId: ctx.streamId,\n    clientId: ctx.threadId,\n    timestamp: Date.now(),\n  }\n}\n\n/**\n * Internal middleware runner that manages composed execution of middleware hooks.\n * Created once per chat() invocation.\n */\nexport class MiddlewareRunner<TContext = unknown> {\n  private readonly middlewares: ReadonlyArray<ChatMiddleware<TContext>>\n  private readonly logger: InternalLogger\n\n  constructor(\n    middlewares: ReadonlyArray<ChatMiddleware<TContext>>,\n    logger: InternalLogger,\n  ) {\n    this.middlewares = middlewares\n    this.logger = logger\n  }\n\n  get hasMiddleware(): boolean {\n    return this.middlewares.length > 0\n  }\n\n  /**\n   * Pipe config through all middleware onConfig hooks in order.\n   * Each middleware receives the merged config from previous middleware.\n   * Partial returns are shallow-merged with the current config.\n   */\n  async runOnConfig(\n    ctx: ChatMiddlewareContext<TContext>,\n    config: ChatMiddlewareConfig,\n  ): Promise<ChatMiddlewareConfig> {\n    let current = config\n    for (const mw of this.middlewares) {\n      if (mw.onConfig) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const result = await mw.onConfig(ctx, current)\n        const hasTransform = result !== undefined && result !== null\n        if (hasTransform) {\n          current = { ...current, ...result }\n          if (!skip) {\n            this.logger.config(\n              `middleware=${mw.name ?? 'unnamed'} keys=${Object.keys(result).join(',')}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                changes: result,\n              },\n            )\n          }\n        }\n        if (!skip) {\n          const base = instrumentCtx(ctx)\n          aiEventClient.emit('middleware:hook:executed', {\n            ...base,\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onConfig',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n          if (hasTransform) {\n            aiEventClient.emit('middleware:config:transformed', {\n              ...base,\n              middlewareName: mw.name || 'unnamed',\n              iteration: ctx.iteration,\n              changes: result,\n            })\n          }\n        }\n      }\n    }\n    return current\n  }\n\n  /**\n   * Pipe config through all middleware onStructuredOutputConfig hooks in order.\n   * Each middleware receives the merged config from previous middleware.\n   * Partial returns are shallow-merged with the current config.\n   *\n   * Called once at the structured-output boundary, before runOnConfig at the\n   * same boundary (which receives a ChatMiddlewareConfig view, no outputSchema).\n   */\n  async runOnStructuredOutputConfig(\n    ctx: ChatMiddlewareContext<TContext>,\n    config: StructuredOutputMiddlewareConfig,\n  ): Promise<StructuredOutputMiddlewareConfig> {\n    let current = config\n    for (const mw of this.middlewares) {\n      if (mw.onStructuredOutputConfig) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const result = await mw.onStructuredOutputConfig(ctx, current)\n        const hasTransform = result !== undefined && result !== null\n        if (hasTransform) {\n          current = { ...current, ...result }\n          if (!skip) {\n            this.logger.config(\n              `middleware=${mw.name ?? 'unnamed'} keys=${Object.keys(result).join(',')}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                changes: result,\n              },\n            )\n          }\n        }\n        if (!skip) {\n          const base = instrumentCtx(ctx)\n          aiEventClient.emit('middleware:hook:executed', {\n            ...base,\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onStructuredOutputConfig',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n          if (hasTransform) {\n            aiEventClient.emit('middleware:config:transformed', {\n              ...base,\n              middlewareName: mw.name || 'unnamed',\n              iteration: ctx.iteration,\n              // `result` is `Partial<StructuredOutputMiddlewareConfig>` —\n              // Object.fromEntries(Object.entries(result)) yields the\n              // structural `Record<string, unknown>` the event emitter wants\n              // without an `as` cast.\n              changes: Object.fromEntries(Object.entries(result)),\n            })\n          }\n        }\n      }\n    }\n    return current\n  }\n\n  /**\n   * Run all `setup` hooks in array order, then assert every declared `provides`\n   * capability was actually provided. Wires the last-wins duplicate-provide\n   * warning into the registry. Runs before init `onConfig`.\n   *\n   * Takes the full `ChatMiddlewareContext` — the same stable context the engine\n   * threads through every other hook — because it both forwards `ctx` to each\n   * `setup` hook and emits instrumentation events from it.\n   */\n  async runSetup(ctx: ChatMiddlewareContext<TContext>): Promise<void> {\n    ctx.capabilities.setOnDuplicate((name) => {\n      this.logger.warn(\n        `capability \"${name}\" was provided more than once; last provider wins`,\n        { capability: name },\n      )\n    })\n\n    for (const mw of this.middlewares) {\n      if (mw.setup) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.setup(ctx)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=setup middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'setup' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'setup',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n\n    for (const mw of this.middlewares) {\n      for (const handle of mw.provides ?? []) {\n        if (!ctx.capabilities.has(handle)) {\n          throw new Error(\n            `Middleware \"${mw.name ?? 'unnamed'}\" declares it provides ` +\n              `\"${handle.capabilityName}\" but never called provide() in setup().`,\n          )\n        }\n      }\n    }\n  }\n\n  /**\n   * Call onStart on all middleware in order.\n   */\n  async runOnStart(ctx: ChatMiddlewareContext<TContext>): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onStart) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onStart(ctx)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onStart middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onStart' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onStart',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Pipe a single chunk through all middleware onChunk hooks in order.\n   * Returns the resulting chunks (0..N) to yield to the consumer.\n   *\n   * - void: pass through unchanged\n   * - chunk: replace with this chunk\n   * - chunk[]: expand to multiple chunks\n   * - null: drop the chunk entirely\n   */\n  async runOnChunk(\n    ctx: ChatMiddlewareContext<TContext>,\n    chunk: StreamChunk,\n  ): Promise<Array<StreamChunk>> {\n    let chunks: Array<StreamChunk> = [chunk]\n\n    for (const mw of this.middlewares) {\n      if (!mw.onChunk) continue\n      const skip = shouldSkipInstrumentation(mw)\n\n      const nextChunks: Array<StreamChunk> = []\n      for (const c of chunks) {\n        // Cast: @ag-ui/core Zod passthrough types prevent direct `.type` access\n        const chunkType = c.type\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onChunk', in: c },\n          )\n        }\n        const result = await mw.onChunk(ctx, c)\n        if (result === null) {\n          // Drop this chunk\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=<dropped>`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                dropped: true,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: 0,\n              wasDropped: true,\n            })\n          }\n          continue\n        } else if (result === undefined) {\n          // Pass through — no instrumentation for pass-throughs\n          nextChunks.push(c)\n        } else if (Array.isArray(result)) {\n          // Expand\n          nextChunks.push(...result)\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=[${result.map((r: StreamChunk) => r.type).join(',')}]`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                in: c,\n                out: result,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: result.length,\n              wasDropped: false,\n            })\n          }\n        } else {\n          // Replace\n          nextChunks.push(result)\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=${result.type}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                in: c,\n                out: result,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: 1,\n              wasDropped: false,\n            })\n          }\n        }\n      }\n      chunks = nextChunks\n    }\n\n    return chunks\n  }\n\n  /**\n   * Run onBeforeToolCall through middleware in order.\n   * Returns the first non-void decision, or undefined to continue normally.\n   */\n  async runOnBeforeToolCall(\n    ctx: ChatMiddlewareContext<TContext>,\n    hookCtx: ToolCallHookContext,\n  ): Promise<BeforeToolCallDecision> {\n    for (const mw of this.middlewares) {\n      if (mw.onBeforeToolCall) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const decision = await mw.onBeforeToolCall(ctx, hookCtx)\n        const hasTransform = decision !== undefined && decision !== null\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onBeforeToolCall middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onBeforeToolCall' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onBeforeToolCall',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n        }\n        if (hasTransform) {\n          return decision\n        }\n      }\n    }\n    return undefined\n  }\n\n  /**\n   * Run onAfterToolCall on all middleware in order.\n   */\n  async runOnAfterToolCall(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: AfterToolCallInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onAfterToolCall) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onAfterToolCall(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onAfterToolCall middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onAfterToolCall' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onAfterToolCall',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onUsage on all middleware in order.\n   */\n  async runOnUsage(\n    ctx: ChatMiddlewareContext<TContext>,\n    usage: UsageInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onUsage) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onUsage(ctx, usage)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onUsage middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onUsage' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onUsage',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onFinish on all middleware in order.\n   */\n  async runOnFinish(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: FinishInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onFinish) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onFinish(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onFinish middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onFinish' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onFinish',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onAbort on all middleware in order.\n   */\n  async runOnAbort(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: AbortInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onAbort) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onAbort(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onAbort middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onAbort' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onAbort',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onError on all middleware in order.\n   */\n  async runOnError(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: ErrorInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onError) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onError(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onError middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onError' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onError',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onIteration on all middleware in order.\n   * Called at the start of each agent loop iteration.\n   */\n  async runOnIteration(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: IterationInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onIteration) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onIteration(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onIteration middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onIteration' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onIteration',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onToolPhaseComplete on all middleware in order.\n   * Called after all tool calls in an iteration have been processed.\n   */\n  async runOnToolPhaseComplete(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: ToolPhaseCompleteInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onToolPhaseComplete) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onToolPhaseComplete(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onToolPhaseComplete middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onToolPhaseComplete' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onToolPhaseComplete',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n}\n"],"names":[],"mappings":";AAoBA,SAAS,0BAA0B,IAAkC;AACnE,SAAO,GAAG,SAAS,cAAc,GAAG,SAAS;AAC/C;AAGA,SAAS,cAAc,KAAiC;AACtD,SAAO;AAAA,IACL,WAAW,IAAI;AAAA,IACf,UAAU,IAAI;AAAA,IACd,UAAU,IAAI;AAAA,IACd,WAAW,KAAK,IAAA;AAAA,EAAI;AAExB;AAMO,MAAM,iBAAqC;AAAA,EAC/B;AAAA,EACA;AAAA,EAEjB,YACE,aACA,QACA;AACA,SAAK,cAAc;AACnB,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,IAAI,gBAAyB;AAC3B,WAAO,KAAK,YAAY,SAAS;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,YACJ,KACA,QAC+B;AAC/B,QAAI,UAAU;AACd,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,UAAU;AACf,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,SAAS,MAAM,GAAG,SAAS,KAAK,OAAO;AAC7C,cAAM,eAAe,WAAW,UAAa,WAAW;AACxD,YAAI,cAAc;AAChB,oBAAU,EAAE,GAAG,SAAS,GAAG,OAAA;AAC3B,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,cAAc,GAAG,QAAQ,SAAS,SAAS,OAAO,KAAK,MAAM,EAAE,KAAK,GAAG,CAAC;AAAA,cACxE;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,SAAS;AAAA,cAAA;AAAA,YACX;AAAA,UAEJ;AAAA,QACF;AACA,YAAI,CAAC,MAAM;AACT,gBAAM,OAAO,cAAc,GAAG;AAC9B,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG;AAAA,YACH,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AACD,cAAI,cAAc;AAChB,0BAAc,KAAK,iCAAiC;AAAA,cAClD,GAAG;AAAA,cACH,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,WAAW,IAAI;AAAA,cACf,SAAS;AAAA,YAAA,CACV;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,4BACJ,KACA,QAC2C;AAC3C,QAAI,UAAU;AACd,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,0BAA0B;AAC/B,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,SAAS,MAAM,GAAG,yBAAyB,KAAK,OAAO;AAC7D,cAAM,eAAe,WAAW,UAAa,WAAW;AACxD,YAAI,cAAc;AAChB,oBAAU,EAAE,GAAG,SAAS,GAAG,OAAA;AAC3B,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,cAAc,GAAG,QAAQ,SAAS,SAAS,OAAO,KAAK,MAAM,EAAE,KAAK,GAAG,CAAC;AAAA,cACxE;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,SAAS;AAAA,cAAA;AAAA,YACX;AAAA,UAEJ;AAAA,QACF;AACA,YAAI,CAAC,MAAM;AACT,gBAAM,OAAO,cAAc,GAAG;AAC9B,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG;AAAA,YACH,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AACD,cAAI,cAAc;AAChB,0BAAc,KAAK,iCAAiC;AAAA,cAClD,GAAG;AAAA,cACH,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,WAAW,IAAI;AAAA;AAAA;AAAA;AAAA;AAAA,cAKf,SAAS,OAAO,YAAY,OAAO,QAAQ,MAAM,CAAC;AAAA,YAAA,CACnD;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,SAAS,KAAqD;AAClE,QAAI,aAAa,eAAe,CAAC,SAAS;AACxC,WAAK,OAAO;AAAA,QACV,eAAe,IAAI;AAAA,QACnB,EAAE,YAAY,KAAA;AAAA,MAAK;AAAA,IAEvB,CAAC;AAED,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,OAAO;AACZ,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,MAAM,GAAG;AAClB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,yBAAyB,GAAG,QAAQ,SAAS;AAAA,YAC7C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,QAAA;AAAA,UAAQ;AAEpD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAEA,eAAW,MAAM,KAAK,aAAa;AACjC,iBAAW,UAAU,GAAG,YAAY,CAAA,GAAI;AACtC,YAAI,CAAC,IAAI,aAAa,IAAI,MAAM,GAAG;AACjC,gBAAM,IAAI;AAAA,YACR,eAAe,GAAG,QAAQ,SAAS,2BAC7B,OAAO,cAAc;AAAA,UAAA;AAAA,QAE/B;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WAAW,KAAqD;AACpE,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,GAAG;AACpB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,WACJ,KACA,OAC6B;AAC7B,QAAI,SAA6B,CAAC,KAAK;AAEvC,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,CAAC,GAAG,QAAS;AACjB,YAAM,OAAO,0BAA0B,EAAE;AAEzC,YAAM,aAAiC,CAAA;AACvC,iBAAW,KAAK,QAAQ;AAEtB,cAAM,YAAY,EAAE;AACpB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS;AAAA,YAC/D,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,WAAW,IAAI,EAAA;AAAA,UAAE;AAAA,QAE/D;AACA,cAAM,SAAS,MAAM,GAAG,QAAQ,KAAK,CAAC;AACtC,YAAI,WAAW,MAAM;AAEnB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS;AAAA,cAC/D;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,SAAS;AAAA,cAAA;AAAA,YACX;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa;AAAA,cACb,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AACA;AAAA,QACF,WAAW,WAAW,QAAW;AAE/B,qBAAW,KAAK,CAAC;AAAA,QACnB,WAAW,MAAM,QAAQ,MAAM,GAAG;AAEhC,qBAAW,KAAK,GAAG,MAAM;AACzB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS,SAAS,OAAO,IAAI,CAAC,MAAmB,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC;AAAA,cACxH;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,IAAI;AAAA,gBACJ,KAAK;AAAA,cAAA;AAAA,YACP;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa,OAAO;AAAA,cACpB,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AAAA,QACF,OAAO;AAEL,qBAAW,KAAK,MAAM;AACtB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS,QAAQ,OAAO,IAAI;AAAA,cAClF;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,IAAI;AAAA,gBACJ,KAAK;AAAA,cAAA;AAAA,YACP;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa;AAAA,cACb,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AAAA,QACF;AAAA,MACF;AACA,eAAS;AAAA,IACX;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,oBACJ,KACA,SACiC;AACjC,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,kBAAkB;AACvB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,WAAW,MAAM,GAAG,iBAAiB,KAAK,OAAO;AACvD,cAAM,eAAe,aAAa,UAAa,aAAa;AAC5D,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,oCAAoC,GAAG,QAAQ,SAAS;AAAA,YACxD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,mBAAA;AAAA,UAAmB;AAE/D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AAAA,QACH;AACA,YAAI,cAAc;AAChB,iBAAO;AAAA,QACT;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,mBACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,iBAAiB;AACtB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,gBAAgB,KAAK,IAAI;AAClC,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,mCAAmC,GAAG,QAAQ,SAAS;AAAA,YACvD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,kBAAA;AAAA,UAAkB;AAE9D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,OACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,KAAK;AAC3B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,UAAU;AACf,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,SAAS,KAAK,IAAI;AAC3B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,4BAA4B,GAAG,QAAQ,SAAS;AAAA,YAChD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,WAAA;AAAA,UAAW;AAEvD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,IAAI;AAC1B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,IAAI;AAC1B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,eACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,aAAa;AAClB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,YAAY,KAAK,IAAI;AAC9B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,+BAA+B,GAAG,QAAQ,SAAS;AAAA,YACnD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,cAAA;AAAA,UAAc;AAE1D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,uBACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,qBAAqB;AAC1B,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,oBAAoB,KAAK,IAAI;AACtC,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,uCAAuC,GAAG,QAAQ,SAAS;AAAA,YAC3D,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,sBAAA;AAAA,UAAsB;AAElE,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;"}
+{"version":3,"file":"compose.js","sources":["../../../../../src/activities/chat/middleware/compose.ts"],"sourcesContent":["import { aiEventClient } from '@tanstack/ai-event-client'\nimport type { StreamChunk } from '../../../types'\nimport type { InternalLogger } from '../../../logger/internal-logger'\nimport type {\n  AbortInfo,\n  AfterToolCallInfo,\n  BeforeToolCallDecision,\n  ChatMiddleware,\n  ChatMiddlewareConfig,\n  ChatMiddlewareContext,\n  ErrorInfo,\n  FinishInfo,\n  IterationInfo,\n  SandboxFileEvent,\n  StructuredOutputMiddlewareConfig,\n  ToolCallHookContext,\n  ToolPhaseCompleteInfo,\n  UsageInfo,\n} from './types'\n\n/** Check if a middleware should be skipped for instrumentation events. */\nfunction shouldSkipInstrumentation(mw: ChatMiddleware<any>): boolean {\n  return mw.name === 'devtools' || mw.name === 'strip-to-spec'\n}\n\n/** Build the base context for middleware instrumentation events. */\nfunction instrumentCtx(ctx: ChatMiddlewareContext<any>) {\n  return {\n    requestId: ctx.requestId,\n    streamId: ctx.streamId,\n    clientId: ctx.threadId,\n    timestamp: Date.now(),\n  }\n}\n\n/**\n * Internal middleware runner that manages composed execution of middleware hooks.\n * Created once per chat() invocation.\n */\nexport class MiddlewareRunner<TContext = unknown> {\n  private readonly middlewares: ReadonlyArray<ChatMiddleware<TContext>>\n  private readonly logger: InternalLogger\n\n  constructor(\n    middlewares: ReadonlyArray<ChatMiddleware<TContext>>,\n    logger: InternalLogger,\n  ) {\n    this.middlewares = middlewares\n    this.logger = logger\n  }\n\n  get hasMiddleware(): boolean {\n    return this.middlewares.length > 0\n  }\n\n  /**\n   * Pipe config through all middleware onConfig hooks in order.\n   * Each middleware receives the merged config from previous middleware.\n   * Partial returns are shallow-merged with the current config.\n   */\n  async runOnConfig(\n    ctx: ChatMiddlewareContext<TContext>,\n    config: ChatMiddlewareConfig,\n  ): Promise<ChatMiddlewareConfig> {\n    let current = config\n    for (const mw of this.middlewares) {\n      if (mw.onConfig) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const result = await mw.onConfig(ctx, current)\n        const hasTransform = result !== undefined && result !== null\n        if (hasTransform) {\n          current = { ...current, ...result }\n          if (!skip) {\n            this.logger.config(\n              `middleware=${mw.name ?? 'unnamed'} keys=${Object.keys(result).join(',')}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                changes: result,\n              },\n            )\n          }\n        }\n        if (!skip) {\n          const base = instrumentCtx(ctx)\n          aiEventClient.emit('middleware:hook:executed', {\n            ...base,\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onConfig',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n          if (hasTransform) {\n            aiEventClient.emit('middleware:config:transformed', {\n              ...base,\n              middlewareName: mw.name || 'unnamed',\n              iteration: ctx.iteration,\n              changes: result,\n            })\n          }\n        }\n      }\n    }\n    return current\n  }\n\n  /**\n   * Pipe config through all middleware onStructuredOutputConfig hooks in order.\n   * Each middleware receives the merged config from previous middleware.\n   * Partial returns are shallow-merged with the current config.\n   *\n   * Called once at the structured-output boundary, before runOnConfig at the\n   * same boundary (which receives a ChatMiddlewareConfig view, no outputSchema).\n   */\n  async runOnStructuredOutputConfig(\n    ctx: ChatMiddlewareContext<TContext>,\n    config: StructuredOutputMiddlewareConfig,\n  ): Promise<StructuredOutputMiddlewareConfig> {\n    let current = config\n    for (const mw of this.middlewares) {\n      if (mw.onStructuredOutputConfig) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const result = await mw.onStructuredOutputConfig(ctx, current)\n        const hasTransform = result !== undefined && result !== null\n        if (hasTransform) {\n          current = { ...current, ...result }\n          if (!skip) {\n            this.logger.config(\n              `middleware=${mw.name ?? 'unnamed'} keys=${Object.keys(result).join(',')}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                changes: result,\n              },\n            )\n          }\n        }\n        if (!skip) {\n          const base = instrumentCtx(ctx)\n          aiEventClient.emit('middleware:hook:executed', {\n            ...base,\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onStructuredOutputConfig',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n          if (hasTransform) {\n            aiEventClient.emit('middleware:config:transformed', {\n              ...base,\n              middlewareName: mw.name || 'unnamed',\n              iteration: ctx.iteration,\n              // `result` is `Partial<StructuredOutputMiddlewareConfig>` —\n              // Object.fromEntries(Object.entries(result)) yields the\n              // structural `Record<string, unknown>` the event emitter wants\n              // without an `as` cast.\n              changes: Object.fromEntries(Object.entries(result)),\n            })\n          }\n        }\n      }\n    }\n    return current\n  }\n\n  /**\n   * Run all `setup` hooks in array order, then assert every declared `provides`\n   * capability was actually provided. Wires the last-wins duplicate-provide\n   * warning into the registry. Runs before init `onConfig`.\n   *\n   * Takes the full `ChatMiddlewareContext` — the same stable context the engine\n   * threads through every other hook — because it both forwards `ctx` to each\n   * `setup` hook and emits instrumentation events from it.\n   */\n  async runSetup(ctx: ChatMiddlewareContext<TContext>): Promise<void> {\n    ctx.capabilities.setOnDuplicate((name) => {\n      this.logger.warn(\n        `capability \"${name}\" was provided more than once; last provider wins`,\n        { capability: name },\n      )\n    })\n\n    for (const mw of this.middlewares) {\n      if (mw.setup) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.setup(ctx)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=setup middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'setup' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'setup',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n\n    for (const mw of this.middlewares) {\n      for (const handle of mw.provides ?? []) {\n        if (!ctx.capabilities.has(handle)) {\n          throw new Error(\n            `Middleware \"${mw.name ?? 'unnamed'}\" declares it provides ` +\n              `\"${handle.capabilityName}\" but never called provide() in setup().`,\n          )\n        }\n      }\n    }\n  }\n\n  /**\n   * Call onStart on all middleware in order.\n   */\n  async runOnStart(ctx: ChatMiddlewareContext<TContext>): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onStart) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onStart(ctx)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onStart middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onStart' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onStart',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Pipe a single chunk through all middleware onChunk hooks in order.\n   * Returns the resulting chunks (0..N) to yield to the consumer.\n   *\n   * - void: pass through unchanged\n   * - chunk: replace with this chunk\n   * - chunk[]: expand to multiple chunks\n   * - null: drop the chunk entirely\n   */\n  async runOnChunk(\n    ctx: ChatMiddlewareContext<TContext>,\n    chunk: StreamChunk,\n  ): Promise<Array<StreamChunk>> {\n    let chunks: Array<StreamChunk> = [chunk]\n\n    for (const mw of this.middlewares) {\n      if (!mw.onChunk) continue\n      const skip = shouldSkipInstrumentation(mw)\n\n      const nextChunks: Array<StreamChunk> = []\n      for (const c of chunks) {\n        // Cast: @ag-ui/core Zod passthrough types prevent direct `.type` access\n        const chunkType = c.type\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onChunk', in: c },\n          )\n        }\n        const result = await mw.onChunk(ctx, c)\n        if (result === null) {\n          // Drop this chunk\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=<dropped>`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                dropped: true,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: 0,\n              wasDropped: true,\n            })\n          }\n          continue\n        } else if (result === undefined) {\n          // Pass through — no instrumentation for pass-throughs\n          nextChunks.push(c)\n        } else if (Array.isArray(result)) {\n          // Expand\n          nextChunks.push(...result)\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=[${result.map((r: StreamChunk) => r.type).join(',')}]`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                in: c,\n                out: result,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: result.length,\n              wasDropped: false,\n            })\n          }\n        } else {\n          // Replace\n          nextChunks.push(result)\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=${result.type}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                in: c,\n                out: result,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: 1,\n              wasDropped: false,\n            })\n          }\n        }\n      }\n      chunks = nextChunks\n    }\n\n    return chunks\n  }\n\n  /**\n   * Dispatch a sandbox file event to every middleware's `sandbox` hooks, in\n   * array order: the catch-all `onFile` then the type-specific hook. Errors are\n   * logged and swallowed so one bad hook can't break the run.\n   */\n  async runSandboxFile(\n    ctx: ChatMiddlewareContext<TContext>,\n    event: SandboxFileEvent,\n  ): Promise<void> {\n    const typed = (\n      {\n        create: 'onFileCreate',\n        change: 'onFileChange',\n        delete: 'onFileDelete',\n      } as const\n    )[event.type]\n    for (const mw of this.middlewares) {\n      const hooks = mw.sandbox\n      if (!hooks) continue\n      for (const fn of [hooks.onFile, hooks[typed]]) {\n        if (!fn) continue\n        try {\n          await fn(ctx, event)\n        } catch (error) {\n          this.logger.sandbox(\n            `hook=${typed} middleware=${mw.name ?? 'unnamed'} threw`,\n            { middleware: mw.name ?? 'unnamed', error },\n          )\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onBeforeToolCall through middleware in order.\n   * Returns the first non-void decision, or undefined to continue normally.\n   */\n  async runOnBeforeToolCall(\n    ctx: ChatMiddlewareContext<TContext>,\n    hookCtx: ToolCallHookContext,\n  ): Promise<BeforeToolCallDecision> {\n    for (const mw of this.middlewares) {\n      if (mw.onBeforeToolCall) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const decision = await mw.onBeforeToolCall(ctx, hookCtx)\n        const hasTransform = decision !== undefined && decision !== null\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onBeforeToolCall middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onBeforeToolCall' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onBeforeToolCall',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n        }\n        if (hasTransform) {\n          return decision\n        }\n      }\n    }\n    return undefined\n  }\n\n  /**\n   * Run onAfterToolCall on all middleware in order.\n   */\n  async runOnAfterToolCall(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: AfterToolCallInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onAfterToolCall) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onAfterToolCall(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onAfterToolCall middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onAfterToolCall' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onAfterToolCall',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onUsage on all middleware in order.\n   */\n  async runOnUsage(\n    ctx: ChatMiddlewareContext<TContext>,\n    usage: UsageInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onUsage) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onUsage(ctx, usage)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onUsage middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onUsage' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onUsage',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onFinish on all middleware in order.\n   */\n  async runOnFinish(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: FinishInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onFinish) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onFinish(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onFinish middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onFinish' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onFinish',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onAbort on all middleware in order.\n   */\n  async runOnAbort(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: AbortInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onAbort) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onAbort(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onAbort middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onAbort' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onAbort',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onError on all middleware in order.\n   */\n  async runOnError(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: ErrorInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onError) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onError(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onError middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onError' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onError',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onIteration on all middleware in order.\n   * Called at the start of each agent loop iteration.\n   */\n  async runOnIteration(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: IterationInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onIteration) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onIteration(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onIteration middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onIteration' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onIteration',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onToolPhaseComplete on all middleware in order.\n   * Called after all tool calls in an iteration have been processed.\n   */\n  async runOnToolPhaseComplete(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: ToolPhaseCompleteInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onToolPhaseComplete) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onToolPhaseComplete(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onToolPhaseComplete middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onToolPhaseComplete' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onToolPhaseComplete',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n}\n"],"names":[],"mappings":";AAqBA,SAAS,0BAA0B,IAAkC;AACnE,SAAO,GAAG,SAAS,cAAc,GAAG,SAAS;AAC/C;AAGA,SAAS,cAAc,KAAiC;AACtD,SAAO;AAAA,IACL,WAAW,IAAI;AAAA,IACf,UAAU,IAAI;AAAA,IACd,UAAU,IAAI;AAAA,IACd,WAAW,KAAK,IAAA;AAAA,EAAI;AAExB;AAMO,MAAM,iBAAqC;AAAA,EAC/B;AAAA,EACA;AAAA,EAEjB,YACE,aACA,QACA;AACA,SAAK,cAAc;AACnB,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,IAAI,gBAAyB;AAC3B,WAAO,KAAK,YAAY,SAAS;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,YACJ,KACA,QAC+B;AAC/B,QAAI,UAAU;AACd,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,UAAU;AACf,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,SAAS,MAAM,GAAG,SAAS,KAAK,OAAO;AAC7C,cAAM,eAAe,WAAW,UAAa,WAAW;AACxD,YAAI,cAAc;AAChB,oBAAU,EAAE,GAAG,SAAS,GAAG,OAAA;AAC3B,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,cAAc,GAAG,QAAQ,SAAS,SAAS,OAAO,KAAK,MAAM,EAAE,KAAK,GAAG,CAAC;AAAA,cACxE;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,SAAS;AAAA,cAAA;AAAA,YACX;AAAA,UAEJ;AAAA,QACF;AACA,YAAI,CAAC,MAAM;AACT,gBAAM,OAAO,cAAc,GAAG;AAC9B,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG;AAAA,YACH,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AACD,cAAI,cAAc;AAChB,0BAAc,KAAK,iCAAiC;AAAA,cAClD,GAAG;AAAA,cACH,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,WAAW,IAAI;AAAA,cACf,SAAS;AAAA,YAAA,CACV;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,4BACJ,KACA,QAC2C;AAC3C,QAAI,UAAU;AACd,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,0BAA0B;AAC/B,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,SAAS,MAAM,GAAG,yBAAyB,KAAK,OAAO;AAC7D,cAAM,eAAe,WAAW,UAAa,WAAW;AACxD,YAAI,cAAc;AAChB,oBAAU,EAAE,GAAG,SAAS,GAAG,OAAA;AAC3B,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,cAAc,GAAG,QAAQ,SAAS,SAAS,OAAO,KAAK,MAAM,EAAE,KAAK,GAAG,CAAC;AAAA,cACxE;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,SAAS;AAAA,cAAA;AAAA,YACX;AAAA,UAEJ;AAAA,QACF;AACA,YAAI,CAAC,MAAM;AACT,gBAAM,OAAO,cAAc,GAAG;AAC9B,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG;AAAA,YACH,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AACD,cAAI,cAAc;AAChB,0BAAc,KAAK,iCAAiC;AAAA,cAClD,GAAG;AAAA,cACH,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,WAAW,IAAI;AAAA;AAAA;AAAA;AAAA;AAAA,cAKf,SAAS,OAAO,YAAY,OAAO,QAAQ,MAAM,CAAC;AAAA,YAAA,CACnD;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,SAAS,KAAqD;AAClE,QAAI,aAAa,eAAe,CAAC,SAAS;AACxC,WAAK,OAAO;AAAA,QACV,eAAe,IAAI;AAAA,QACnB,EAAE,YAAY,KAAA;AAAA,MAAK;AAAA,IAEvB,CAAC;AAED,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,OAAO;AACZ,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,MAAM,GAAG;AAClB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,yBAAyB,GAAG,QAAQ,SAAS;AAAA,YAC7C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,QAAA;AAAA,UAAQ;AAEpD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAEA,eAAW,MAAM,KAAK,aAAa;AACjC,iBAAW,UAAU,GAAG,YAAY,CAAA,GAAI;AACtC,YAAI,CAAC,IAAI,aAAa,IAAI,MAAM,GAAG;AACjC,gBAAM,IAAI;AAAA,YACR,eAAe,GAAG,QAAQ,SAAS,2BAC7B,OAAO,cAAc;AAAA,UAAA;AAAA,QAE/B;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WAAW,KAAqD;AACpE,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,GAAG;AACpB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,WACJ,KACA,OAC6B;AAC7B,QAAI,SAA6B,CAAC,KAAK;AAEvC,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,CAAC,GAAG,QAAS;AACjB,YAAM,OAAO,0BAA0B,EAAE;AAEzC,YAAM,aAAiC,CAAA;AACvC,iBAAW,KAAK,QAAQ;AAEtB,cAAM,YAAY,EAAE;AACpB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS;AAAA,YAC/D,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,WAAW,IAAI,EAAA;AAAA,UAAE;AAAA,QAE/D;AACA,cAAM,SAAS,MAAM,GAAG,QAAQ,KAAK,CAAC;AACtC,YAAI,WAAW,MAAM;AAEnB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS;AAAA,cAC/D;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,SAAS;AAAA,cAAA;AAAA,YACX;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa;AAAA,cACb,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AACA;AAAA,QACF,WAAW,WAAW,QAAW;AAE/B,qBAAW,KAAK,CAAC;AAAA,QACnB,WAAW,MAAM,QAAQ,MAAM,GAAG;AAEhC,qBAAW,KAAK,GAAG,MAAM;AACzB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS,SAAS,OAAO,IAAI,CAAC,MAAmB,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC;AAAA,cACxH;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,IAAI;AAAA,gBACJ,KAAK;AAAA,cAAA;AAAA,YACP;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa,OAAO;AAAA,cACpB,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AAAA,QACF,OAAO;AAEL,qBAAW,KAAK,MAAM;AACtB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS,QAAQ,OAAO,IAAI;AAAA,cAClF;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,IAAI;AAAA,gBACJ,KAAK;AAAA,cAAA;AAAA,YACP;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa;AAAA,cACb,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AAAA,QACF;AAAA,MACF;AACA,eAAS;AAAA,IACX;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,eACJ,KACA,OACe;AACf,UAAM,QACJ;AAAA,MACE,QAAQ;AAAA,MACR,QAAQ;AAAA,MACR,QAAQ;AAAA,IAAA,EAEV,MAAM,IAAI;AACZ,eAAW,MAAM,KAAK,aAAa;AACjC,YAAM,QAAQ,GAAG;AACjB,UAAI,CAAC,MAAO;AACZ,iBAAW,MAAM,CAAC,MAAM,QAAQ,MAAM,KAAK,CAAC,GAAG;AAC7C,YAAI,CAAC,GAAI;AACT,YAAI;AACF,gBAAM,GAAG,KAAK,KAAK;AAAA,QACrB,SAAS,OAAO;AACd,eAAK,OAAO;AAAA,YACV,QAAQ,KAAK,eAAe,GAAG,QAAQ,SAAS;AAAA,YAChD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAA;AAAA,UAAM;AAAA,QAE9C;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,oBACJ,KACA,SACiC;AACjC,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,kBAAkB;AACvB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,WAAW,MAAM,GAAG,iBAAiB,KAAK,OAAO;AACvD,cAAM,eAAe,aAAa,UAAa,aAAa;AAC5D,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,oCAAoC,GAAG,QAAQ,SAAS;AAAA,YACxD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,mBAAA;AAAA,UAAmB;AAE/D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AAAA,QACH;AACA,YAAI,cAAc;AAChB,iBAAO;AAAA,QACT;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,mBACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,iBAAiB;AACtB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,gBAAgB,KAAK,IAAI;AAClC,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,mCAAmC,GAAG,QAAQ,SAAS;AAAA,YACvD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,kBAAA;AAAA,UAAkB;AAE9D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,OACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,KAAK;AAC3B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,UAAU;AACf,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,SAAS,KAAK,IAAI;AAC3B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,4BAA4B,GAAG,QAAQ,SAAS;AAAA,YAChD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,WAAA;AAAA,UAAW;AAEvD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,IAAI;AAC1B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,IAAI;AAC1B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,eACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,aAAa;AAClB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,YAAY,KAAK,IAAI;AAC9B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,+BAA+B,GAAG,QAAQ,SAAS;AAAA,YACnD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,cAAA;AAAA,UAAc;AAE1D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,uBACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,qBAAqB;AAC1B,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,oBAAoB,KAAK,IAAI;AACtC,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,uCAAuC,GAAG,QAAQ,SAAS;AAAA,YAC3D,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,sBAAA;AAAA,UAAsB;AAElE,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;"}

package/dist/esm/activities/chat/middleware/index.d.ts

@@ -1,4 +1,4 @@
-export type { ChatMiddleware, ChatMiddlewareContext, ChatMiddlewarePhase, ChatMiddlewareConfig, StructuredOutputMiddlewareConfig, ToolCallHookContext, BeforeToolCallDecision, AfterToolCallInfo, IterationInfo, ToolPhaseCompleteInfo, UsageInfo, FinishInfo, AbortInfo, ErrorInfo, } from './types.js';
+export type { ChatMiddleware, ChatMiddlewareContext, ChatMiddlewarePhase, ChatMiddlewareConfig, StructuredOutputMiddlewareConfig, ToolCallHookContext, BeforeToolCallDecision, AfterToolCallInfo, IterationInfo, ToolPhaseCompleteInfo, UsageInfo, FinishInfo, AbortInfo, ErrorInfo, SandboxFileEvent, ChatSandboxHooks, } from './types.js';
 export { MiddlewareRunner } from './compose.js';
 export { createCapability, CapabilityRegistry } from './capabilities.js';
 export type { Capability, CapabilityHandle, CapabilityContext, CapabilityGetter, CapabilityProvider, CapabilityGetOptions, } from './capabilities.js';

package/dist/esm/activities/chat/middleware/sandbox-runtime.d.ts

@@ -0,0 +1,8 @@
+import { InternalLogger } from '../../../logger/internal-logger.js';
+import { SandboxFileEvent } from './types.js';
+export interface SandboxRuntime {
+    emit: (event: SandboxFileEvent) => void;
+    logger: InternalLogger;
+}
+export declare const SandboxRuntimeCapability: import('./capabilities.js').Capability<SandboxRuntime, "sandbox-runtime">;
+export declare const getSandboxRuntime: import('./capabilities.js').CapabilityGetter<SandboxRuntime>, provideSandboxRuntime: import('./capabilities.js').CapabilityProvider<SandboxRuntime>;

package/dist/esm/activities/chat/middleware/sandbox-runtime.js

@@ -0,0 +1,9 @@
+import { createCapability } from "./capabilities.js";
+const SandboxRuntimeCapability = createCapability()("sandbox-runtime");
+const [getSandboxRuntime, provideSandboxRuntime] = SandboxRuntimeCapability;
+export {
+  SandboxRuntimeCapability,
+  getSandboxRuntime,
+  provideSandboxRuntime
+};
+//# sourceMappingURL=sandbox-runtime.js.map

package/dist/esm/activities/chat/middleware/sandbox-runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sandbox-runtime.js","sources":["../../../../../src/activities/chat/middleware/sandbox-runtime.ts"],"sourcesContent":["/**\n * Internal runtime seam the chat engine PROVIDES so the sandbox middleware can\n * surface file events without a public ctx method. `emit` runs every\n * middleware's `sandbox` hooks AND emits a CUSTOM `sandbox.file` chunk into the\n * stream; `logger` lets the sandbox layer log under the `sandbox` debug\n * category. Consumed (optionally) by `withSandbox` in `@tanstack/ai-sandbox`.\n */\nimport { createCapability } from './capabilities'\nimport type { InternalLogger } from '../../../logger/internal-logger'\nimport type { SandboxFileEvent } from './types'\n\nexport interface SandboxRuntime {\n  emit: (event: SandboxFileEvent) => void\n  logger: InternalLogger\n}\n\nexport const SandboxRuntimeCapability =\n  createCapability<SandboxRuntime>()('sandbox-runtime')\n\nexport const [getSandboxRuntime, provideSandboxRuntime] =\n  SandboxRuntimeCapability\n"],"names":[],"mappings":";AAgBO,MAAM,2BACX,iBAAA,EAAmC,iBAAiB;AAE/C,MAAM,CAAC,mBAAmB,qBAAqB,IACpD;"}

package/dist/esm/activities/chat/middleware/types.d.ts

@@ -1,6 +1,23 @@
 import { JSONSchema, ModelMessage, StreamChunk, TokenUsage, Tool, ToolCall } from '../../../types.js';
 import { SystemPrompt } from '../../../system-prompts.js';
 import { Capability, CapabilityHandle, CapabilityRegistry } from './capabilities.js';
+/** A file change observed inside a sandbox during a chat run. */
+export interface SandboxFileEvent {
+    type: 'create' | 'change' | 'delete';
+    /** Absolute path inside the sandbox (under the workspace root). */
+    path: string;
+    timestamp: number;
+}
+/**
+ * Sandbox file-event hooks a chat middleware can declare. Fire server-side for
+ * every file create/change/delete observed in the sandbox during the run.
+ */
+export interface ChatSandboxHooks<TContext = unknown> {
+    onFile?: (ctx: ChatMiddlewareContext<TContext>, e: SandboxFileEvent) => void | Promise<void>;
+    onFileCreate?: (ctx: ChatMiddlewareContext<TContext>, e: SandboxFileEvent) => void | Promise<void>;
+    onFileChange?: (ctx: ChatMiddlewareContext<TContext>, e: SandboxFileEvent) => void | Promise<void>;
+    onFileDelete?: (ctx: ChatMiddlewareContext<TContext>, e: SandboxFileEvent) => void | Promise<void>;
+}
 /**
  * Phase of the chat middleware lifecycle.
  * - 'init': Initial config transform before the chat engine starts
@@ -400,6 +417,11 @@ export interface ChatMiddleware<TContext = unknown> {
      * Exactly one of onFinish/onAbort/onError will be called per run.
      */
     onError?: (ctx: ChatMiddlewareContext<TContext>, info: ErrorInfo) => void | Promise<void>;
+    /**
+     * Sandbox file-event hooks. Fire when a sandbox provided by `withSandbox` is
+     * active during the run and a file is created/changed/deleted. Server-side.
+     */
+    sandbox?: ChatSandboxHooks<TContext>;
 }
 /** A `ChatMiddleware` with a permissive context — for use as a constraint. */
 export type AnyChatMiddleware = ChatMiddleware<any>;

package/dist/esm/activities/chat/stream/processor.js

@@ -1082,6 +1082,29 @@ class StreamProcessor {
       });
       return;
     }
+    if (chunk.name === "ui-resource" && chunk.value) {
+      const v = chunk.value;
+      const resolvedMessageId = this.toolCallToMessage.get(v.toolCallId) ?? messageId;
+      if (resolvedMessageId) {
+        const part = {
+          type: "ui-resource",
+          resource: v.resource,
+          toolCallId: v.toolCallId,
+          toolName: v.toolName,
+          ...v.serverId !== void 0 && { serverId: v.serverId },
+          ...v.meta !== void 0 && { meta: v.meta }
+        };
+        this.messages = this.messages.map(
+          (msg) => msg.id === resolvedMessageId ? { ...msg, parts: [...msg.parts, part] } : msg
+        );
+        this.emitMessagesChange();
+      } else {
+        console.warn(
+          `[mcp-apps] dropped ui-resource: no target message for toolCallId "${v.toolCallId}" (toolName "${v.toolName}")`
+        );
+      }
+      return;
+    }
     if (this.events.onCustomEvent) {
       const toolCallId = chunk.value && typeof chunk.value === "object" ? chunk.value.toolCallId : void 0;
       this.events.onCustomEvent(chunk.name, chunk.value, { toolCallId });