@tanstack/ai 0.42.0 → 0.43.0

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      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      // Model-message arguments are complete, so surface the parsed input.\n      // A malformed arguments string just leaves `input` undefined.\n      let input: unknown\n      try {\n        input = JSON.parse(toolCall.function.arguments)\n      } catch {\n        input = undefined\n      }\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        ...(input !== undefined && { input }),\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;AAG7C,UAAI;AACJ,UAAI;AACF,gBAAQ,KAAK,MAAM,SAAS,SAAS,SAAS;AAAA,MAChD,QAAQ;AACN,gBAAQ;AAAA,MACV;AACA,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,UAAU,UAAa,EAAE,MAAA;AAAA,QAC7B,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","names":[],"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-requested' ||\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      // Model-message arguments are complete, so surface the parsed input.\n      // A malformed arguments string just leaves `input` undefined.\n      let input: unknown\n      try {\n        input = JSON.parse(toolCall.function.arguments)\n      } catch {\n        input = undefined\n      }\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        ...(input !== undefined && { input }),\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, msg.id)\n        uiMessages.push(toolResultUIMessage)\n      }\n    } else {\n      // Regular message. Preserve a persisted stable id so a hydrated message\n      // keeps the same identity as its live stream (enables in-place resume).\n      const uiMessage = modelMessageToUIMessage(msg, msg.id)\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"],"mappings":";;;;;;AAkBA,SAAS,cAAc,MAAwC;CAC7D,OACE,KAAK,SAAS,UACd,KAAK,SAAS,WACd,KAAK,SAAS,WACd,KAAK,SAAS,WACd,KAAK,SAAS;AAElB;AAEA,SAAS,kBAAkB,OAAwB;CACjD,IAAI;EACF,OAAO,KAAK,UAAU,KAAK;CAC7B,QAAQ;EACN,OAAO;CACT;AACF;AAEA,SAAS,uBAAuB,SAA0B;CACxD,IAAI;EACF,OAAO,KAAK,MAAM,OAAO;CAC3B,QAAQ;EACN,OAAO;CACT;AACF;;;;;;;AAQA,SAAS,qBACP,OACoC;CACpC,IAAI,MAAM,WAAW,GAAG,OAAO;CAG/B,IADgB,MAAM,OAAO,MAAM,EAAE,SAAS,MAC1C,GAEF,OADe,MAAM,KAAK,MAAM,EAAE,OAAO,CAAC,CAAC,KAAK,EACzC,KAAU;CAGnB,OAAO;AACT;;;;;AAMA,SAAS,eAAe,SAAqD;CAC3E,IAAI,YAAY,MAAM,OAAO;CAC7B,IAAI,OAAO,YAAY,UAAU,OAAO;CACxC,OAAO,QACJ,QAAQ,SAA2B,KAAK,SAAS,MAAM,CAAC,CACxD,KAAK,SAAS,KAAK,OAAO,CAAC,CAC3B,KAAK,EAAE;AACZ;;;;AAKA,SAAgB,+BACd,UACqB;CAIrB,MAAM,sCAAsB,IAAI,IAAY;CAC5C,KAAK,MAAM,OAAO,UAChB,IAAI,WAAW;OACR,MAAM,QAAQ,IAAI,OACrB,IAAI,KAAK,SAAS,eAChB,oBAAoB,IAAI,KAAK,UAAU;CAAA;CAM/C,MAAM,gBAAqC,CAAC;CAC5C,KAAK,MAAM,OAAO,UAAU;EAC1B,IAAI,WAAW,KAAK;GAElB,cAAc,KAAK,GAAG,yBAAyB,GAAG,CAAC;GACnD;EACF;EAEA,MAAM,OAAQ,IAAyB;EAGvC,IACE,SAAS,UACT,IAAI,cACJ,oBAAoB,IAAI,IAAI,UAAU,GAEtC;EAIF,IAAI,SAAS,eAAe,SAAS,YACnC;EAIF,IAAI,SAAS,aAAa;GACxB,cAAc,KAAK;IACjB,MAAM;IACN,SAAU,IAA4B;GACxC,CAAC;GACD;EACF;EAGA,cAAc,KAAK,GAAG;CACxB;CACA,OAAO;AACT;;;;;;;;;;;;;;;;;;;;AAqBA,SAAgB,yBACd,WACqB;CAErB,IAAI,UAAU,SAAS,UACrB,OAAO,CAAC;CAKV,IAAI,UAAU,SAAS,aACrB,OAAO,CAAC,uBAAuB,SAAS,CAAC;CAI3C,OAAO,uBAAuB,SAAS;AACzC;;;;;AAMA,SAAS,uBAAuB,WAAoC;CAClE,MAAM,eAAmC,CAAC;CAC1C,KAAK,MAAM,QAAQ,UAAU,OAC3B,IAAI,cAAc,IAAI,GACpB,aAAa,KAAK,IAAI;CAI1B,OAAO;EACL,MAAM,UAAU;EAChB,SAAS,qBAAqB,YAAY;CAC5C;AACF;AAgBA,SAAS,gBAAkC;CACzC,OAAO;EAAE,cAAc,CAAC;EAAG,WAAW,CAAC;CAAE;AAC3C;AAEA,SAAS,mBAAmB,MAA6B;CACvD,OACE,KAAK,UAAU,oBACf,KAAK,UAAU,cACf,KAAK,UAAU,wBACf,KAAK,UAAU,wBACf,KAAK,UAAU,WACf,KAAK,WAAW,KAAA;AAEpB;;;;;;;;;;AAWA,SAAS,uBAAuB,WAA2C;CACzE,MAAM,cAAmC,CAAC;CAC1C,IAAI,UAAU,cAAc;CAC5B,IAAI,kBAAkE,CAAC;CAKvE,MAAM,uCAAuB,IAAI,IAAY;CAE7C,SAAS,eAAqB;EAC5B,MAAM,UAAU,qBAAqB,QAAQ,YAAY;EACzD,MAAM,aAAa,YAAY;EAC/B,MAAM,eAAe,QAAQ,UAAU,SAAS;EAEhD,IAAI,cAAc,cAAc;GAC9B,YAAY,KAAK;IACf,MAAM;IACN;IACA,GAAI,gBAAgB,EAAE,WAAW,QAAQ,UAAU;IACnD,GAAI,gBAAgB,SAAS,KAAK,EAAE,UAAU,gBAAgB;GAChE,CAAC;GACD,kBAAkB,CAAC;EACrB;EACA,UAAU,cAAc;CAC1B;CAEA,KAAK,MAAM,QAAQ,UAAU,OAC3B,QAAQ,KAAK,MAAb;EACE,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;GACH,QAAQ,aAAa,KAAK,IAAI;GAC9B;EAEF,KAAK;GACH,IAAI,mBAAmB,IAAI,GACzB,QAAQ,UAAU,KAAK;IACrB,IAAI,KAAK;IACT,MAAM;IACN,UAAU;KACR,MAAM,KAAK;KACX,WAAW,KAAK;IAClB;IACA,GAAI,KAAK,aAAa,KAAA,KAAa,EAAE,UAAU,KAAK,SAAS;GAC/D,CAAC;GAEH;EAEF,KAAK;GAEH,aAAa;GAGb,KACG,KAAK,UAAU,cAAc,KAAK,UAAU,YAC7C,CAAC,qBAAqB,IAAI,KAAK,UAAU,GACzC;IACA,YAAY,KAAK;KACf,MAAM;KACN,SAAS,KAAK;KACd,YAAY,KAAK;IACnB,CAAC;IACD,qBAAqB,IAAI,KAAK,UAAU;GAC1C;GACA;EAEF,KAAK;GACH,IAAI,KAAK,SACP,gBAAgB,KAAK;IACnB,SAAS,KAAK;IACd,GAAI,KAAK,aAAa,EAAE,WAAW,KAAK,UAAU;GACpD,CAAC;GAEH;EAEF,KAAK;GAKH,IAAI,KAAK,WAAW,YAAY;IAC9B,MAAM,aACJ,KAAK,QAAQ,KACT,KAAK,MACL,KAAK,SAAS,KAAA,IACZ,kBAAkB,KAAK,IAAI,IAC3B;IACR,IAAI,eAAe,IACjB,QAAQ,aAAa,KAAK;KAAE,MAAM;KAAQ,SAAS;IAAW,CAAC;GAEnE;GACA;EAEF,KAAK,eAGH;EAEF,SACE;CACJ;CAIF,aAAa;CAMb,KAAK,MAAM,QAAQ,UAAU,OAAO;EAClC,IAAI,KAAK,SAAS,aAAa;EAI/B,IAAI,KAAK,WAAW,KAAA,KAAa,CAAC,qBAAqB,IAAI,KAAK,EAAE,GAAG;GACnE,YAAY,KAAK;IACf,MAAM;IACN,SAAS,oBAAoB,KAAK,MAAM;IACxC,YAAY,KAAK;GACnB,CAAC;GACD,qBAAqB,IAAI,KAAK,EAAE;EAClC;EAGA,IACE,KAAK,WAAW,KAAA,KAChB,KAAK,UAAU,wBACf,KAAK,UAAU,aAAa,KAAA,KAC5B,CAAC,qBAAqB,IAAI,KAAK,EAAE,GACjC;GACA,MAAM,WAAW,KAAK,SAAS;GAC/B,YAAY,KAAK;IACf,MAAM;IACN,SAAS,KAAK,UAAU;KACtB;KACA,GAAI,YAAY,EAAE,kBAAkB,KAAK;KACzC,SAAS,WACL,8BACA;IACN,CAAC;IACD,YAAY,KAAK;GACnB,CAAC;GACD,qBAAqB,IAAI,KAAK,EAAE;EAClC;CACF;CAGA,IAAI,YAAY,WAAW,GACzB,YAAY,KAAK;EACf,MAAM;EACN,SAAS;CACX,CAAC;CAGH,OAAO;AACT;;;;;;;;;;;;;AAcA,SAAgB,wBACd,cACA,IACW;CACX,MAAM,QAA4B,CAAC;CAEnC,IAAI,aAAa,SAAS,eAAe,aAAa,UAAU,QAC9D,KAAK,MAAM,YAAY,aAAa,UAAU;EAC5C,IAAI,CAAC,SAAS,SAAS;EACvB,MAAM,KAAK;GACT,MAAM;GACN,SAAS,SAAS;GAClB,GAAI,SAAS,aAAa,EAAE,WAAW,SAAS,UAAU;EAC5D,CAAC;CACH;CAKF,IAAI,aAAa,SAAS,UAAU,aAAa,YAC/C,MAAM,KAAK;EACT,MAAM;EACN,YAAY,aAAa;EACzB,SAAS,eAAe,aAAa,OAAO;EAC5C,OAAO;CACT,CAAC;MACI,IAAI,MAAM,QAAQ,aAAa,OAAO,GAE3C,KAAK,MAAM,QAAQ,aAAa,SAC9B,MAAM,KAAK,IAAI;MAEZ;EAEL,MAAM,cAAc,eAAe,aAAa,OAAO;EACvD,IAAI,aACF,MAAM,KAAK;GACT,MAAM;GACN,SAAS;EACX,CAAC;CAEL;CAGA,IAAI,aAAa,aAAa,aAAa,UAAU,SAAS,GAC5D,KAAK,MAAM,YAAY,aAAa,WAAW;EAG7C,IAAI;EACJ,IAAI;GACF,QAAQ,KAAK,MAAM,SAAS,SAAS,SAAS;EAChD,QAAQ;GACN,QAAQ,KAAA;EACV;EACA,MAAM,KAAK;GACT,MAAM;GACN,IAAI,SAAS;GACb,MAAM,SAAS,SAAS;GACxB,WAAW,SAAS,SAAS;GAC7B,OAAO;GACP,GAAI,UAAU,KAAA,KAAa,EAAE,MAAM;GACnC,GAAI,SAAS,aAAa,KAAA,KAAa,EAAE,UAAU,SAAS,SAAS;EACvE,CAAC;CACH;CAGF,OAAO;EACL,IAAI,MAAM,kBAAkB;EAC5B,MAAM,aAAa,SAAS,SAAS,cAAc,aAAa;EAChE;CACF;AACF;;;;;;;;;;;;;;;;;AAkBA,SAAgB,+BACd,SACW;CACX,IAAI,WAAW,SACb,OAAO;EAAE,GAAG;EAAS,IAAI,QAAQ,MAAM,kBAAkB;CAAE;CAG7D,MAAM,KAAK,QAAQ,MAAM,kBAAkB;CAE3C,QAAQ,QAAQ,MAAhB;EACE,KAAK,QACH,OAAO;GACL;GACA,MAAM;GACN,OAAO,uBAAuB,QAAQ,OAAO;EAC/C;EACF,KAAK,aACH,OAAO,wBACL;GACE,MAAM;GACN,SAAS,QAAQ,WAAW;GAC5B,GAAI,QAAQ,aAAa,EAAE,WAAW,QAAQ,UAAU;EAC1D,GACA,EACF;EACF,KAAK,QACH,OAAO,wBACL;GACE,MAAM;GACN,SAAS,QAAQ;GACjB,YAAY,QAAQ;EACtB,GACA,EACF;EACF,KAAK;EACL,KAAK,aAEH,OAAO;GACL;GACA,MAAM;GACN,OAAO,QAAQ,UACX,CAAC;IAAE,MAAM;IAAQ,SAAS,QAAQ;GAAQ,CAAC,IAC3C,CAAC;EACP;EACF,KAAK,aACH,OAAO;GACL;GACA,MAAM;GACN,OAAO,QAAQ,UACX,CAAC;IAAE,MAAM;IAAY,SAAS,QAAQ;GAAQ,CAAC,IAC/C,CAAC;EACP;EAEF,SAEE,OAAO;GAAE;GAAI,MAAM;GAAa,OAAO,CAAC;EAAE;CAC9C;AACF;;;;;;;;;;AAWA,SAAS,uBACP,SACoB;CACpB,IAAI,OAAO,YAAY,UACrB,OAAO,UAAU,CAAC;EAAE,MAAM;EAAQ;CAAQ,CAAC,IAAI,CAAC;CAGlD,MAAM,QAA4B,CAAC;CACnC,KAAK,MAAM,QAAQ,SACjB,IAAI,KAAK,SAAS,QAChB,MAAM,KAAK;EAAE,MAAM;EAAQ,SAAS,KAAK;CAAK,CAAC;MAC1C,IAAI,KAAK,SAAS,UACvB,MAAM,KAAK,IAAI;CAGnB,OAAO;AACT;;;;;;;;;AAUA,SAAgB,0BACd,eACkB;CAClB,MAAM,aAA+B,CAAC;CACtC,IAAI,0BAA4C;CAEhD,KAAK,MAAM,OAAO,eAChB,IAAI,IAAI,SAAS,QAEf,IACE,IAAI,eAAe,KAAA,KACnB,2BACA,wBAAwB,SAAS,aACjC;EACA,MAAM,UAAU,eAAe,IAAI,OAAO;EAC1C,MAAM,eAAe,wBAAwB,MAAM,MAChD,SACC,KAAK,SAAS,eAAe,KAAK,OAAO,IAAI,UACjD;EAEA,IAAI,cAAc;GAChB,aAAa,SAAS,uBAAuB,OAAO;GACpD,aAAa,QAAQ;EACvB;EAEA,wBAAwB,MAAM,KAAK;GACjC,MAAM;GACN,YAAY,IAAI;GAChB;GACA,OAAO;EACT,CAAC;CACH,OAAO;EAEL,MAAM,sBAAsB,wBAAwB,KAAK,IAAI,EAAE;EAC/D,WAAW,KAAK,mBAAmB;CACrC;MACK;EAGL,MAAM,YAAY,wBAAwB,KAAK,IAAI,EAAE;EACrD,WAAW,KAAK,SAAS;EAGzB,IAAI,IAAI,SAAS,aACf,0BAA0B;OAE1B,0BAA0B;CAE9B;CAGF,OAAO;AACT;;;;;;;;;AAUA,SAAgB,qBACd,SACA,YACW;CACX,IAAI,WAAW,SAEb,OAAO;EACL,GAAG;EACH,IAAI,QAAQ,MAAM,WAAW;EAC7B,WAAW,QAAQ,6BAAa,IAAI,KAAK;CAC3C;MAGA,OAAO;EACL,GAAG,wBAAwB,SAAS,WAAW,CAAC;EAChD,2BAAW,IAAI,KAAK;CACtB;AAEJ;;;;AAKA,SAAgB,oBAA4B;CAC1C,OAAO,OAAO,KAAK,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,UAAU,CAAC;AACpE"}

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

@@ -1,17 +1,19 @@
+//#region src/activities/chat/middleware/builder.ts
+/** Create an order-aware middleware builder. */
 function createChatMiddleware() {
-  const list = [];
-  const builder = {
-    use(middleware) {
-      list.push(middleware);
-      return builder;
-    },
-    build() {
-      return list;
-    }
-  };
-  return builder;
+	const list = [];
+	const builder = {
+		use(middleware) {
+			list.push(middleware);
+			return builder;
+		},
+		build() {
+			return list;
+		}
+	};
+	return builder;
 }
-export {
-  createChatMiddleware
-};
-//# sourceMappingURL=builder.js.map
+//#endregion
+export { createChatMiddleware };
+
+//# sourceMappingURL=builder.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"builder.js","sources":["../../../../../src/activities/chat/middleware/builder.ts"],"sourcesContent":["import type { CapabilityHandle } from './capabilities'\nimport type { AnyChatMiddleware, ChatMiddleware } from './types'\nimport type { DefinedChatMiddleware } from './define'\n\n/** Union of capability NAME literals from a tuple of handles. */\nexport type NamesOf<T extends ReadonlyArray<CapabilityHandle>> =\n  T[number]['capabilityName']\n\n/** Names provided across a middleware array (imprecise middleware → `string`). */\nexport type ProvidedNames<TList extends ReadonlyArray<AnyChatMiddleware>> =\n  NonNullable<TList[number]['provides']> extends infer P\n    ? P extends ReadonlyArray<CapabilityHandle>\n      ? NamesOf<P>\n      : never\n    : never\n\n/** Names required across a middleware array. */\nexport type RequiredNames<TList extends ReadonlyArray<AnyChatMiddleware>> =\n  NonNullable<TList[number]['requires']> extends infer P\n    ? P extends ReadonlyArray<CapabilityHandle>\n      ? NamesOf<P>\n      : never\n    : never\n\n/**\n * Branded marker surfaced when required capability names are missing from the\n * provided set, so the compiler error names the gap instead of emitting an\n * opaque \"not assignable\".\n */\nexport type MissingCapabilities<TMissing extends string> = {\n  // The human-readable message lives in the property KEY, so TypeScript's\n  // \"Property '<key>' is missing in type ... but required in type ...\" error\n  // prints the explanation instead of an opaque `__missingCapabilities`. The\n  // key distributes over a union of missing names (one required key each).\n  [K in `✖ Missing capability \"${TMissing}\": no configured middleware provides it. Add a middleware whose \\`provides\\` includes it (and, with createChatMiddleware().use(), order the provider before this consumer).`]: never\n}\n\n/**\n * Missing capability names. When required names are imprecise (`string`, i.e.\n * plain `ChatMiddleware` not authored via `defineChatMiddleware`), we cannot\n * prove a gap, so we allow it (→ `never`). Otherwise the precise literals not\n * present in the provided set.\n */\ntype MissingNames<TList extends ReadonlyArray<AnyChatMiddleware>> =\n  string extends RequiredNames<TList>\n    ? never\n    : Exclude<RequiredNames<TList>, ProvidedNames<TList>>\n\n/**\n * Resolves to `TList` when coverage holds, otherwise to a `MissingCapabilities`\n * marker (not assignable to a middleware array) — producing a compile error at\n * the `middleware` option that names the missing capability.\n */\nexport type CheckCoverage<TList extends ReadonlyArray<AnyChatMiddleware>> = [\n  MissingNames<TList>,\n] extends [never]\n  ? TList\n  : MissingCapabilities<MissingNames<TList>>\n\n/**\n * Order-aware middleware builder. Each `.use()` requires that the middleware's\n * required capability names are already in the accumulated provided set, then\n * adds its provided names. `.build()` returns the ordered array.\n *\n * `TProvided` is the running union of provided capability name literals.\n */\nexport interface ChatMiddlewareBuilder<\n  TList extends ReadonlyArray<AnyChatMiddleware>,\n  TProvided extends string,\n> {\n  use: <\n    TRequires extends ReadonlyArray<CapabilityHandle>,\n    TProvides extends ReadonlyArray<CapabilityHandle>,\n    TContext = unknown,\n  >(\n    middleware: [NamesOf<TRequires>] extends [TProvided]\n      ? DefinedChatMiddleware<TContext, TRequires, TProvides>\n      : DefinedChatMiddleware<TContext, TRequires, TProvides> &\n          MissingCapabilities<Exclude<NamesOf<TRequires>, TProvided>>,\n  ) => ChatMiddlewareBuilder<\n    readonly [...TList, DefinedChatMiddleware<TContext, TRequires, TProvides>],\n    TProvided | NamesOf<TProvides>\n  >\n\n  build: () => [...TList]\n}\n\n/** Create an order-aware middleware builder. */\nexport function createChatMiddleware(): ChatMiddlewareBuilder<\n  readonly [],\n  never\n> {\n  const list: Array<ChatMiddleware<unknown>> = []\n  const builder = {\n    use(middleware: ChatMiddleware<unknown>) {\n      list.push(middleware)\n      return builder\n    },\n    build() {\n      return list\n    },\n  }\n  // The only sanctioned assertion in this PR: the runtime `builder` is a single\n  // object reused across `.use()` calls, but the type accumulates `TProvided`\n  // and `TList` per call — TypeScript cannot derive that from runtime values, so\n  // a structural `as` is impossible and the double assertion is irreducible.\n  // eslint-disable-next-line no-restricted-syntax -- irreducible: type-level accumulation cannot be expressed from a single runtime object\n  return builder as unknown as ChatMiddlewareBuilder<readonly [], never>\n}\n"],"names":[],"mappings":"AAwFO,SAAS,uBAGd;AACA,QAAM,OAAuC,CAAA;AAC7C,QAAM,UAAU;AAAA,IACd,IAAI,YAAqC;AACvC,WAAK,KAAK,UAAU;AACpB,aAAO;AAAA,IACT;AAAA,IACA,QAAQ;AACN,aAAO;AAAA,IACT;AAAA,EAAA;AAOF,SAAO;AACT;"}
+{"version":3,"file":"builder.js","names":[],"sources":["../../../../../src/activities/chat/middleware/builder.ts"],"sourcesContent":["import type { CapabilityHandle } from './capabilities'\nimport type { AnyChatMiddleware, ChatMiddleware } from './types'\nimport type { DefinedChatMiddleware } from './define'\n\n/** Union of capability NAME literals from a tuple of handles. */\nexport type NamesOf<T extends ReadonlyArray<CapabilityHandle>> =\n  T[number]['capabilityName']\n\n/** Names provided across a middleware array (imprecise middleware → `string`). */\nexport type ProvidedNames<TList extends ReadonlyArray<AnyChatMiddleware>> =\n  NonNullable<TList[number]['provides']> extends infer P\n    ? P extends ReadonlyArray<CapabilityHandle>\n      ? NamesOf<P>\n      : never\n    : never\n\n/** Names required across a middleware array. */\nexport type RequiredNames<TList extends ReadonlyArray<AnyChatMiddleware>> =\n  NonNullable<TList[number]['requires']> extends infer P\n    ? P extends ReadonlyArray<CapabilityHandle>\n      ? NamesOf<P>\n      : never\n    : never\n\n/**\n * Branded marker surfaced when required capability names are missing from the\n * provided set, so the compiler error names the gap instead of emitting an\n * opaque \"not assignable\".\n */\nexport type MissingCapabilities<TMissing extends string> = {\n  // The human-readable message lives in the property KEY, so TypeScript's\n  // \"Property '<key>' is missing in type ... but required in type ...\" error\n  // prints the explanation instead of an opaque `__missingCapabilities`. The\n  // key distributes over a union of missing names (one required key each).\n  [K in `✖ Missing capability \"${TMissing}\": no configured middleware provides it. Add a middleware whose \\`provides\\` includes it (and, with createChatMiddleware().use(), order the provider before this consumer).`]: never\n}\n\n/**\n * Missing capability names. When required names are imprecise (`string`, i.e.\n * plain `ChatMiddleware` not authored via `defineChatMiddleware`), we cannot\n * prove a gap, so we allow it (→ `never`). Otherwise the precise literals not\n * present in the provided set.\n */\ntype MissingNames<TList extends ReadonlyArray<AnyChatMiddleware>> =\n  string extends RequiredNames<TList>\n    ? never\n    : Exclude<RequiredNames<TList>, ProvidedNames<TList>>\n\n/**\n * Resolves to `TList` when coverage holds, otherwise to a `MissingCapabilities`\n * marker (not assignable to a middleware array) — producing a compile error at\n * the `middleware` option that names the missing capability.\n */\nexport type CheckCoverage<TList extends ReadonlyArray<AnyChatMiddleware>> = [\n  MissingNames<TList>,\n] extends [never]\n  ? TList\n  : MissingCapabilities<MissingNames<TList>>\n\n/**\n * Order-aware middleware builder. Each `.use()` requires that the middleware's\n * required capability names are already in the accumulated provided set, then\n * adds its provided names. `.build()` returns the ordered array.\n *\n * `TProvided` is the running union of provided capability name literals.\n */\nexport interface ChatMiddlewareBuilder<\n  TList extends ReadonlyArray<AnyChatMiddleware>,\n  TProvided extends string,\n> {\n  use: <\n    TRequires extends ReadonlyArray<CapabilityHandle>,\n    TProvides extends ReadonlyArray<CapabilityHandle>,\n    TContext = unknown,\n  >(\n    middleware: [NamesOf<TRequires>] extends [TProvided]\n      ? DefinedChatMiddleware<TContext, TRequires, TProvides>\n      : DefinedChatMiddleware<TContext, TRequires, TProvides> &\n          MissingCapabilities<Exclude<NamesOf<TRequires>, TProvided>>,\n  ) => ChatMiddlewareBuilder<\n    readonly [...TList, DefinedChatMiddleware<TContext, TRequires, TProvides>],\n    TProvided | NamesOf<TProvides>\n  >\n\n  build: () => [...TList]\n}\n\n/** Create an order-aware middleware builder. */\nexport function createChatMiddleware(): ChatMiddlewareBuilder<\n  readonly [],\n  never\n> {\n  const list: Array<ChatMiddleware<unknown>> = []\n  const builder = {\n    use(middleware: ChatMiddleware<unknown>) {\n      list.push(middleware)\n      return builder\n    },\n    build() {\n      return list\n    },\n  }\n  // The only sanctioned assertion in this PR: the runtime `builder` is a single\n  // object reused across `.use()` calls, but the type accumulates `TProvided`\n  // and `TList` per call — TypeScript cannot derive that from runtime values, so\n  // a structural `as` is impossible and the double assertion is irreducible.\n  // oxlint-disable-next-line eslint-js/no-restricted-syntax -- irreducible: type-level accumulation cannot be expressed from a single runtime object\n  return builder as unknown as ChatMiddlewareBuilder<readonly [], never>\n}\n"],"mappings":";;AAwFA,SAAgB,uBAGd;CACA,MAAM,OAAuC,CAAC;CAC9C,MAAM,UAAU;EACd,IAAI,YAAqC;GACvC,KAAK,KAAK,UAAU;GACpB,OAAO;EACT;EACA,QAAQ;GACN,OAAO;EACT;CACF;CAMA,OAAO;AACT"}

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

@@ -1,45 +1,80 @@
-class CapabilityRegistry {
-  provided = /* @__PURE__ */ new Set();
-  onDuplicate;
-  /** Register a callback fired when a handle is provided more than once. */
-  setOnDuplicate(cb) {
-    this.onDuplicate = cb;
-  }
-  /** Record that `handle` was provided; fire the duplicate callback on repeats. */
-  markProvided(handle) {
-    if (this.provided.has(handle)) this.onDuplicate?.(handle.capabilityName);
-    this.provided.add(handle);
-  }
-  has(handle) {
-    return this.provided.has(handle);
-  }
-}
+//#region src/activities/chat/middleware/capabilities.ts
+/**
+* Per-request bookkeeping: which capabilities were provided, plus the
+* duplicate-provide notification. Capability VALUES live in per-capability
+* WeakMaps (see `createCapability`), not here — this only tracks presence.
+*/
+var CapabilityRegistry = class {
+	provided = /* @__PURE__ */ new Set();
+	onDuplicate;
+	/** Register a callback fired when a handle is provided more than once. */
+	setOnDuplicate(cb) {
+		this.onDuplicate = cb;
+	}
+	/** Record that `handle` was provided; fire the duplicate callback on repeats. */
+	markProvided(handle) {
+		if (this.provided.has(handle)) this.onDuplicate?.(handle.capabilityName);
+		this.provided.add(handle);
+	}
+	has(handle) {
+		return this.provided.has(handle);
+	}
+};
+/**
+* Create a capability. Returns a hybrid handle that destructures to
+* `[get, provide]` and is itself the identity for `requires`/`provides`.
+*
+* Curried so the value type is supplied explicitly while the name literal is
+* INFERRED from the argument: `createCapability<T>()('name')`. (A single call
+* `createCapability<T>('name')` cannot work — supplying `T` explicitly stops
+* TypeScript inferring the name, collapsing it to `string` and defeating the
+* compile-time coverage check that keys on the literal name.)
+*
+* @example Provider + consumer middleware
+* ```ts
+* const counterCapability = createCapability<{ value: number }>()('counter')
+* const [getCounter, provideCounter] = counterCapability
+*
+* const withCounter = defineChatMiddleware({
+*   name: 'counter',
+*   provides: [counterCapability],
+*   setup(ctx) { provideCounter(ctx, { value: 0 }) },
+* })
+*
+* const readsCounter = defineChatMiddleware({
+*   name: 'reads-counter',
+*   requires: [counterCapability],
+*   onChunk(ctx) { getCounter(ctx).value++ },
+* })
+*
+* chat({ adapter, messages, middleware: [withCounter, readsCounter] })
+* ```
+*
+* @remarks Capability `name`s must be unique across your app: compile-time
+* coverage tracking keys on the name literal (runtime keys on reference).
+*/
 function createCapability() {
-  return (name) => {
-    const values = /* @__PURE__ */ new WeakMap();
-    function get(ctx, opts) {
-      if (!values.has(ctx)) {
-        if (opts?.optional) return void 0;
-        throw new Error(
-          `Capability "${name}" was requested but never provided. Ensure a middleware provides it in setup(), ordered before this consumer.`
-        );
-      }
-      return values.get(ctx);
-    }
-    const provide = (ctx, value) => {
-      values.set(ctx, value);
-      ctx.capabilities.markProvided(handle);
-    };
-    const pair = [get, provide];
-    const handle = Object.assign(pair, {
-      capabilityName: name,
-      has: (ctx) => values.has(ctx)
-    });
-    return handle;
-  };
+	return (name) => {
+		const values = /* @__PURE__ */ new WeakMap();
+		function get(ctx, opts) {
+			if (!values.has(ctx)) {
+				if (opts?.optional) return void 0;
+				throw new Error(`Capability "${name}" was requested but never provided. Ensure a middleware provides it in setup(), ordered before this consumer.`);
+			}
+			return values.get(ctx);
+		}
+		const provide = (ctx, value) => {
+			values.set(ctx, value);
+			ctx.capabilities.markProvided(handle);
+		};
+		const handle = Object.assign([get, provide], {
+			capabilityName: name,
+			has: (ctx) => values.has(ctx)
+		});
+		return handle;
+	};
 }
-export {
-  CapabilityRegistry,
-  createCapability
-};
-//# sourceMappingURL=capabilities.js.map
+//#endregion
+export { CapabilityRegistry, createCapability };
+
+//# sourceMappingURL=capabilities.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"capabilities.js","sources":["../../../../../src/activities/chat/middleware/capabilities.ts"],"sourcesContent":["/** Options accepted by a capability getter. */\nexport interface CapabilityGetOptions {\n  /** When true, return undefined instead of throwing if the capability is absent. */\n  optional?: boolean\n}\n\n/**\n * The minimal context shape a capability accessor needs. The full\n * `ChatMiddlewareContext` satisfies this (it has `capabilities`), so accessors\n * accept any middleware context without referencing `any`.\n */\nexport interface CapabilityContext {\n  capabilities: CapabilityRegistry\n}\n\n/** Reads a capability value off a context. Overloaded so the flag narrows the return. */\nexport interface CapabilityGetter<TValue> {\n  (ctx: CapabilityContext): TValue\n  (ctx: CapabilityContext, opts: { optional: true }): TValue | undefined\n}\n\n/** Writes a capability value onto a context. */\nexport type CapabilityProvider<TValue> = (\n  ctx: CapabilityContext,\n  value: TValue,\n) => void\n\n/**\n * A capability handle. It is BOTH a `[get, provide]` tuple (array-destructurable)\n * AND the identity used in middleware `requires`/`provides` declarations.\n *\n * Runtime identity is this object's reference. The `capabilityName` literal is\n * used for diagnostics and COMPILE-TIME tracking only — capability names MUST be\n * unique across an app or the type-level coverage check conflates them.\n */\nexport type Capability<\n  TValue = unknown,\n  TName extends string = string,\n> = readonly [\n  get: CapabilityGetter<TValue>,\n  provide: CapabilityProvider<TValue>,\n] & {\n  readonly capabilityName: TName\n  /** @internal Presence check for the post-setup assertion. */\n  has: (ctx: CapabilityContext) => boolean\n}\n\n/**\n * A capability handle with permissive value/name — for use as a constraint in\n * `requires`/`provides` arrays. Concentrates `any` in one named alias (same\n * convention as `AnyTextAdapter`/`AnyTool`); needed so `Capability<SpecificT>`\n * is assignable to the handle-array element type.\n */\nexport type CapabilityHandle = Capability<any, string>\n\n/**\n * Per-request bookkeeping: which capabilities were provided, plus the\n * duplicate-provide notification. Capability VALUES live in per-capability\n * WeakMaps (see `createCapability`), not here — this only tracks presence.\n */\nexport class CapabilityRegistry {\n  private readonly provided = new Set<CapabilityHandle>()\n  private onDuplicate?: (name: string) => void\n\n  /** Register a callback fired when a handle is provided more than once. */\n  setOnDuplicate(cb: (name: string) => void): void {\n    this.onDuplicate = cb\n  }\n\n  /** Record that `handle` was provided; fire the duplicate callback on repeats. */\n  markProvided(handle: CapabilityHandle): void {\n    if (this.provided.has(handle)) this.onDuplicate?.(handle.capabilityName)\n    this.provided.add(handle)\n  }\n\n  has(handle: CapabilityHandle): boolean {\n    return this.provided.has(handle)\n  }\n}\n\n/**\n * Create a capability. Returns a hybrid handle that destructures to\n * `[get, provide]` and is itself the identity for `requires`/`provides`.\n *\n * Curried so the value type is supplied explicitly while the name literal is\n * INFERRED from the argument: `createCapability<T>()('name')`. (A single call\n * `createCapability<T>('name')` cannot work — supplying `T` explicitly stops\n * TypeScript inferring the name, collapsing it to `string` and defeating the\n * compile-time coverage check that keys on the literal name.)\n *\n * @example Provider + consumer middleware\n * ```ts\n * const counterCapability = createCapability<{ value: number }>()('counter')\n * const [getCounter, provideCounter] = counterCapability\n *\n * const withCounter = defineChatMiddleware({\n *   name: 'counter',\n *   provides: [counterCapability],\n *   setup(ctx) { provideCounter(ctx, { value: 0 }) },\n * })\n *\n * const readsCounter = defineChatMiddleware({\n *   name: 'reads-counter',\n *   requires: [counterCapability],\n *   onChunk(ctx) { getCounter(ctx).value++ },\n * })\n *\n * chat({ adapter, messages, middleware: [withCounter, readsCounter] })\n * ```\n *\n * @remarks Capability `name`s must be unique across your app: compile-time\n * coverage tracking keys on the name literal (runtime keys on reference).\n */\nexport function createCapability<TValue = unknown>(): <\n  const TName extends string,\n>(\n  name: TName,\n) => Capability<TValue, TName> {\n  return <const TName extends string>(\n    name: TName,\n  ): Capability<TValue, TName> => {\n    // Each capability owns a typed WeakMap keyed by the context object. Because\n    // the value type is TValue, reads are typed with no assertion.\n    const values = new WeakMap<CapabilityContext, TValue>()\n\n    function get(ctx: CapabilityContext): TValue\n    function get(\n      ctx: CapabilityContext,\n      opts: { optional: true },\n    ): TValue | undefined\n    function get(\n      ctx: CapabilityContext,\n      opts?: CapabilityGetOptions,\n    ): TValue | undefined {\n      if (!values.has(ctx)) {\n        if (opts?.optional) return undefined\n        throw new Error(\n          `Capability \"${name}\" was requested but never provided. Ensure a ` +\n            `middleware provides it in setup(), ordered before this consumer.`,\n        )\n      }\n      return values.get(ctx)\n    }\n\n    const provide: CapabilityProvider<TValue> = (ctx, value) => {\n      values.set(ctx, value)\n      ctx.capabilities.markProvided(handle)\n    }\n\n    const pair: readonly [\n      CapabilityGetter<TValue>,\n      CapabilityProvider<TValue>,\n    ] = [get, provide]\n    // Object.assign's return type is the intersection of the tuple and the\n    // props, which IS Capability<TValue, TName> — no cast needed.\n    const handle = Object.assign(pair, {\n      capabilityName: name,\n      has: (ctx: CapabilityContext) => values.has(ctx),\n    })\n    return handle\n  }\n}\n"],"names":[],"mappings":"AA4DO,MAAM,mBAAmB;AAAA,EACb,+BAAe,IAAA;AAAA,EACxB;AAAA;AAAA,EAGR,eAAe,IAAkC;AAC/C,SAAK,cAAc;AAAA,EACrB;AAAA;AAAA,EAGA,aAAa,QAAgC;AAC3C,QAAI,KAAK,SAAS,IAAI,MAAM,EAAG,MAAK,cAAc,OAAO,cAAc;AACvE,SAAK,SAAS,IAAI,MAAM;AAAA,EAC1B;AAAA,EAEA,IAAI,QAAmC;AACrC,WAAO,KAAK,SAAS,IAAI,MAAM;AAAA,EACjC;AACF;AAmCO,SAAS,mBAIe;AAC7B,SAAO,CACL,SAC8B;AAG9B,UAAM,6BAAa,QAAA;AAOnB,aAAS,IACP,KACA,MACoB;AACpB,UAAI,CAAC,OAAO,IAAI,GAAG,GAAG;AACpB,YAAI,MAAM,SAAU,QAAO;AAC3B,cAAM,IAAI;AAAA,UACR,eAAe,IAAI;AAAA,QAAA;AAAA,MAGvB;AACA,aAAO,OAAO,IAAI,GAAG;AAAA,IACvB;AAEA,UAAM,UAAsC,CAAC,KAAK,UAAU;AAC1D,aAAO,IAAI,KAAK,KAAK;AACrB,UAAI,aAAa,aAAa,MAAM;AAAA,IACtC;AAEA,UAAM,OAGF,CAAC,KAAK,OAAO;AAGjB,UAAM,SAAS,OAAO,OAAO,MAAM;AAAA,MACjC,gBAAgB;AAAA,MAChB,KAAK,CAAC,QAA2B,OAAO,IAAI,GAAG;AAAA,IAAA,CAChD;AACD,WAAO;AAAA,EACT;AACF;"}
+{"version":3,"file":"capabilities.js","names":[],"sources":["../../../../../src/activities/chat/middleware/capabilities.ts"],"sourcesContent":["/** Options accepted by a capability getter. */\nexport interface CapabilityGetOptions {\n  /** When true, return undefined instead of throwing if the capability is absent. */\n  optional?: boolean\n}\n\n/**\n * The minimal context shape a capability accessor needs. The full\n * `ChatMiddlewareContext` satisfies this (it has `capabilities`), so accessors\n * accept any middleware context without referencing `any`.\n */\nexport interface CapabilityContext {\n  capabilities: CapabilityRegistry\n}\n\n/** Reads a capability value off a context. Overloaded so the flag narrows the return. */\nexport interface CapabilityGetter<TValue> {\n  (ctx: CapabilityContext): TValue\n  (ctx: CapabilityContext, opts: { optional: true }): TValue | undefined\n}\n\n/** Writes a capability value onto a context. */\nexport type CapabilityProvider<TValue> = (\n  ctx: CapabilityContext,\n  value: TValue,\n) => void\n\n/**\n * A capability handle. It is BOTH a `[get, provide]` tuple (array-destructurable)\n * AND the identity used in middleware `requires`/`provides` declarations.\n *\n * Runtime identity is this object's reference. The `capabilityName` literal is\n * used for diagnostics and COMPILE-TIME tracking only — capability names MUST be\n * unique across an app or the type-level coverage check conflates them.\n */\nexport type Capability<\n  TValue = unknown,\n  TName extends string = string,\n> = readonly [\n  get: CapabilityGetter<TValue>,\n  provide: CapabilityProvider<TValue>,\n] & {\n  readonly capabilityName: TName\n  /** @internal Presence check for the post-setup assertion. */\n  has: (ctx: CapabilityContext) => boolean\n}\n\n/**\n * A capability handle with permissive value/name — for use as a constraint in\n * `requires`/`provides` arrays. Concentrates `any` in one named alias (same\n * convention as `AnyTextAdapter`/`AnyTool`); needed so `Capability<SpecificT>`\n * is assignable to the handle-array element type.\n */\nexport type CapabilityHandle = Capability<any, string>\n\n/**\n * Per-request bookkeeping: which capabilities were provided, plus the\n * duplicate-provide notification. Capability VALUES live in per-capability\n * WeakMaps (see `createCapability`), not here — this only tracks presence.\n */\nexport class CapabilityRegistry {\n  private readonly provided = new Set<CapabilityHandle>()\n  private onDuplicate?: (name: string) => void\n\n  /** Register a callback fired when a handle is provided more than once. */\n  setOnDuplicate(cb: (name: string) => void): void {\n    this.onDuplicate = cb\n  }\n\n  /** Record that `handle` was provided; fire the duplicate callback on repeats. */\n  markProvided(handle: CapabilityHandle): void {\n    if (this.provided.has(handle)) this.onDuplicate?.(handle.capabilityName)\n    this.provided.add(handle)\n  }\n\n  has(handle: CapabilityHandle): boolean {\n    return this.provided.has(handle)\n  }\n}\n\n/**\n * Create a capability. Returns a hybrid handle that destructures to\n * `[get, provide]` and is itself the identity for `requires`/`provides`.\n *\n * Curried so the value type is supplied explicitly while the name literal is\n * INFERRED from the argument: `createCapability<T>()('name')`. (A single call\n * `createCapability<T>('name')` cannot work — supplying `T` explicitly stops\n * TypeScript inferring the name, collapsing it to `string` and defeating the\n * compile-time coverage check that keys on the literal name.)\n *\n * @example Provider + consumer middleware\n * ```ts\n * const counterCapability = createCapability<{ value: number }>()('counter')\n * const [getCounter, provideCounter] = counterCapability\n *\n * const withCounter = defineChatMiddleware({\n *   name: 'counter',\n *   provides: [counterCapability],\n *   setup(ctx) { provideCounter(ctx, { value: 0 }) },\n * })\n *\n * const readsCounter = defineChatMiddleware({\n *   name: 'reads-counter',\n *   requires: [counterCapability],\n *   onChunk(ctx) { getCounter(ctx).value++ },\n * })\n *\n * chat({ adapter, messages, middleware: [withCounter, readsCounter] })\n * ```\n *\n * @remarks Capability `name`s must be unique across your app: compile-time\n * coverage tracking keys on the name literal (runtime keys on reference).\n */\nexport function createCapability<TValue = unknown>(): <\n  const TName extends string,\n>(\n  name: TName,\n) => Capability<TValue, TName> {\n  return <const TName extends string>(\n    name: TName,\n  ): Capability<TValue, TName> => {\n    // Each capability owns a typed WeakMap keyed by the context object. Because\n    // the value type is TValue, reads are typed with no assertion.\n    const values = new WeakMap<CapabilityContext, TValue>()\n\n    function get(ctx: CapabilityContext): TValue\n    function get(\n      ctx: CapabilityContext,\n      opts: { optional: true },\n    ): TValue | undefined\n    function get(\n      ctx: CapabilityContext,\n      opts?: CapabilityGetOptions,\n    ): TValue | undefined {\n      if (!values.has(ctx)) {\n        if (opts?.optional) return undefined\n        throw new Error(\n          `Capability \"${name}\" was requested but never provided. Ensure a ` +\n            `middleware provides it in setup(), ordered before this consumer.`,\n        )\n      }\n      return values.get(ctx)\n    }\n\n    const provide: CapabilityProvider<TValue> = (ctx, value) => {\n      values.set(ctx, value)\n      ctx.capabilities.markProvided(handle)\n    }\n\n    const pair: readonly [\n      CapabilityGetter<TValue>,\n      CapabilityProvider<TValue>,\n    ] = [get, provide]\n    // Object.assign's return type is the intersection of the tuple and the\n    // props, which IS Capability<TValue, TName> — no cast needed.\n    const handle = Object.assign(pair, {\n      capabilityName: name,\n      has: (ctx: CapabilityContext) => values.has(ctx),\n    })\n    return handle\n  }\n}\n"],"mappings":";;;;;;AA4DA,IAAa,qBAAb,MAAgC;CAC9B,2BAA4B,IAAI,IAAsB;CACtD;;CAGA,eAAe,IAAkC;EAC/C,KAAK,cAAc;CACrB;;CAGA,aAAa,QAAgC;EAC3C,IAAI,KAAK,SAAS,IAAI,MAAM,GAAG,KAAK,cAAc,OAAO,cAAc;EACvE,KAAK,SAAS,IAAI,MAAM;CAC1B;CAEA,IAAI,QAAmC;EACrC,OAAO,KAAK,SAAS,IAAI,MAAM;CACjC;AACF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmCA,SAAgB,mBAIe;CAC7B,QACE,SAC8B;EAG9B,MAAM,yBAAS,IAAI,QAAmC;EAOtD,SAAS,IACP,KACA,MACoB;GACpB,IAAI,CAAC,OAAO,IAAI,GAAG,GAAG;IACpB,IAAI,MAAM,UAAU,OAAO,KAAA;IAC3B,MAAM,IAAI,MACR,eAAe,KAAK,8GAEtB;GACF;GACA,OAAO,OAAO,IAAI,GAAG;EACvB;EAEA,MAAM,WAAuC,KAAK,UAAU;GAC1D,OAAO,IAAI,KAAK,KAAK;GACrB,IAAI,aAAa,aAAa,MAAM;EACtC;EAQA,MAAM,SAAS,OAAO,OAAO,CAHxB,KAAK,OAGmB,GAAM;GACjC,gBAAgB;GAChB,MAAM,QAA2B,OAAO,IAAI,GAAG;EACjD,CAAC;EACD,OAAO;CACT;AACF"}

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

@@ -1,4 +1,4 @@
-import { StreamChunk } from '../../../types.js';
+import { AgentLoopState, StreamChunk } from '../../../types.js';
 import { InternalLogger } from '../../../logger/internal-logger.js';
 import { AbortInfo, AfterToolCallInfo, BeforeToolCallDecision, ChatMiddleware, ChatMiddlewareConfig, ChatMiddlewareContext, ErrorInfo, FinishInfo, IterationInfo, SandboxFileHookEvent, StructuredOutputMiddlewareConfig, ToolCallHookContext, ToolPhaseCompleteInfo, UsageInfo } from './types.js';
 /**
@@ -68,16 +68,103 @@ export declare class MiddlewareRunner<TContext = unknown> {
      * Run onUsage on all middleware in order.
      */
     runOnUsage(ctx: ChatMiddlewareContext<TContext>, usage: UsageInfo): Promise<void>;
+    /**
+     * Await ONE terminal hook and RETURN its throw instead of letting it escape
+     * the caller's loop, logging it on the `errors` channel first so the failure
+     * is never invisible. `undefined` means the hook completed.
+     *
+     * Capturing (rather than swallowing at this level) is what lets isolation and
+     * reporting coexist: every caller gives every middleware its turn, and then
+     * each decides on its own whether the collected failures are worth telling the
+     * caller about. See {@link runOnFinish} vs {@link runOnAbort} /
+     * {@link runOnError}.
+     */
+    private captureTerminalHook;
     /**
      * Run onFinish on all middleware in order.
+     *
+     * ISOLATED **and** REPORTED. `onFinish` is the only terminal fan-out on the
+     * SUCCESS path, and it is where `withPersistence.onFinish` writes the
+     * assistant turn through the store. So the two properties are needed together
+     * and neither may be traded for the other:
+     *
+     * - ISOLATION: every middleware's hook runs even if an earlier one threw, so a
+     *   transient store error cannot skip a later middleware's own bookkeeping.
+     *   Each failure is captured by {@link captureTerminalHook}, not propagated
+     *   mid-loop.
+     * - REPORTING: after the loop, the failures are rethrown. `chat()`'s catch
+     *   treats what we throw as a genuine error (it is not a
+     *   `MiddlewareAbortError`, and `structuralInterruptFailure` does not match
+     *   it) and rethrows it out of the generator.
+     *
+     * What that rethrow can and cannot achieve depends on the transport, because
+     * this fan-out is awaited AFTER the adapter's `RUN_FINISHED` has already been
+     * yielded (`chat()` yields terminal chunks while streaming, then awaits this
+     * hook on its way out). The success terminal is therefore already gone; the
+     * rethrow can only append to what the consumer saw, never retract it:
+     *
+     * - NON-DURABLE transport: the throw escapes the generator mid-response, and
+     *   the SSE / HTTP-stream encoder turns it into a TRAILING `RUN_ERROR` on the
+     *   wire carrying the store's own message and `code`. `ai-client` surfaces
+     *   that as an error status, so the user is not told the turn was saved when
+     *   it was not.
+     * - DURABLE transport: the throw reaches the durability sink instead. The
+     *   terminal was already persisted AND forwarded, so the sink deliberately
+     *   does NOT append a second, contradictory terminal, and `terminalForwarded`
+     *   (see `stream-to-response.ts`) suppresses the rethrow to the live consumer.
+     *   The `RUN_FINISHED` stands and the failure is RECORDED SERVER-SIDE on the
+     *   sink's `errors` channel. That is the intended outcome, not a gap: the save
+     *   failed, not the run — the consumer did receive the complete stream, so
+     *   telling it the run errored would be the lie. What the rethrow buys here is
+     *   that the sink sees the failure at all; while this loop swallowed, the only
+     *   trace anywhere was {@link captureTerminalHook}'s log line.
+     *
+     * Either way, swallowing is the one option ruled out: a failed
+     * `messages.append` would otherwise leave a `completed` run record with the
+     * assistant turn missing from storage and nothing beyond a middleware log
+     * line, and the client would go on to send a history the server has no record
+     * of.
+     *
+     * A single failure is rethrown AS-IS so the store's own error — its message,
+     * `cause`, `code` and `instanceof` identity — is what reaches the caller and
+     * the wire; wrapping the common case would bury it. Two or more become an
+     * `AggregateError` (never a `MiddlewareAbortError`, so it cannot be mistaken
+     * for an abort) rather than picking a winner and dropping the rest.
      */
     runOnFinish(ctx: ChatMiddlewareContext<TContext>, info: FinishInfo): Promise<void>;
     /**
      * Run onAbort on all middleware in order.
+     *
+     * ISOLATED and DELIBERATELY SWALLOWED. `onAbort` is a pure teardown fan-out
+     * released from `chat()`'s `finally`, on a path where the outcome is already
+     * decided: the run stopped, and the caller is being told why. A throw here has
+     * nothing better to report than the abort reason it would DISPLACE — the
+     * `finally` would surface a flaky store's error in place of "client
+     * disconnected" — so failures are logged on the `errors` channel and go no
+     * further. That is not a silent failure; it is refusing to let teardown
+     * rewrite an outcome it did not produce.
+     *
+     * Isolation matters independently: these hooks release PER-MIDDLEWARE
+     * resources (`withSandbox.onAbort` detaches or destroys the sandbox and stamps
+     * `detachedSince`; `withPersistence.onAbort` records the run status through the
+     * store), so an unguarded loop turns one transient store error into a
+     * permanently leaked sandbox for every middleware ordered after it.
      */
     runOnAbort(ctx: ChatMiddlewareContext<TContext>, info: AbortInfo): Promise<void>;
     /**
      * Run onError on all middleware in order.
+     *
+     * ISOLATED and DELIBERATELY SWALLOWED, for the same reason as
+     * {@link runOnAbort} and NOT merely because it is teardown: the run has
+     * already failed, `info.error` IS that failure, and `chat()` rethrows it to the
+     * caller the moment this fan-out returns. A propagated hook throw could only
+     * REPLACE the run's real error with a teardown artifact — strictly less
+     * information for the caller, who is already learning the run failed. Reporting
+     * would buy nothing and cost the diagnosis, so failures are logged on the
+     * `errors` channel and stop there.
+     *
+     * Contrast {@link runOnFinish}, where nothing else is telling the caller
+     * anything is wrong — which is why that one reports.
      */
     runOnError(ctx: ChatMiddlewareContext<TContext>, info: ErrorInfo): Promise<void>;
     /**
@@ -85,6 +172,12 @@ export declare class MiddlewareRunner<TContext = unknown> {
      * Called at the start of each agent loop iteration.
      */
     runOnIteration(ctx: ChatMiddlewareContext<TContext>, info: IterationInfo): Promise<void>;
+    /**
+     * Run onShouldContinue through middleware in order (AND semantics).
+     * Any explicit `false` stops further iterations; `true` / void / undefined pass.
+     * Called after `agentLoopStrategy` has already approved continuation.
+     */
+    runOnShouldContinue(ctx: ChatMiddlewareContext<TContext>, state: AgentLoopState): Promise<boolean>;
     /**
      * Run onToolPhaseComplete on all middleware in order.
      * Called after all tool calls in an iteration have been processed.