@assistant-ui/ai-sdk 0.0.9 → 0.0.11

package/dist/runtime/useExternalHistory.js.map

@@ -1 +1 @@
-{"version":3,"file":"useExternalHistory.js","names":["AssistantRuntime","ThreadHistoryAdapter","ThreadMessage","MessageFormatAdapter","MessageFormatItem","MessageFormatRepository","ExportedMessageRepository","RespondToToolApprovalOptions","Unstable_ToolInteractionLog","getExternalStoreMessages","MessageRepository","readToolInteractionLog","useAui","useRef","useEffect","useState","RefObject","useCallback","useMemo","toExportedMessageRepository","toThreadMessages","messages","TMessage","survivingIds","Set","survivors","flatMap","m","message","console","warn","parentId","has","add","id","headId","hasUnansweredApproval","content","some","part","type","approval","approved","undefined","resolution","isAwaitingToolApproval","status","reason","isTerminalMessage","TOOL_ARTIFACTS_METADATA_KEY","TOOL_INTERACTIONS_METADATA_KEY","TOOL_APPROVAL_RESPONSES_METADATA_KEY","StoredToolApprovalResponse","Omit","isRecord","value","Record","Array","isArray","collectToolArtifacts","toolArtifacts","ReadonlyMap","entries","artifact","get","toolCallId","const","length","Object","fromEntries","collectToolInteractions","toolInteractions","interactions","collectToolApprovalResponses","toolApprovalResponses","response","optionId","text","addToolData","parts","toolCallIds","artifacts","hasOwn","approvalIds","approvalId","approvalResponses","hasArtifacts","keys","hasInteractions","hasApprovalResponses","metadata","restoreToolData","Map","set","log","_","__","___","restMetadata","_metadata","restMessage","encodeContent","storageFormatAdapter","item","JSON","stringify","encode","useExternalHistory","runtimeRef","historyAdapter","onSetMessages","onToolArtifactsRestored","onToolInteractionsRestored","onToolApprovalResponsesRestored","loadedRef","itemEpoch","setItemEpoch","aui","optionalThreadListItem","threadListItem","source","hasLoaded","setHasLoaded","historyIds","deferredTelemetryIds","persistedInnerMessages","onSetMessagesRef","current","formatAdapter","withFormat","Error","isLoading","loadHistory","repo","load","clear","restoredMessages","map","restoredRepo","index","getId","converted","thread","import","tempRepo","getMessages","error","pin","remoteId","getState","subscribe","n","threadState","isRunning","runStartRef","persistTimerRef","ReturnType","setTimeout","persistInFlightRef","Promise","resolve","stepBoundariesRef","wasRunningRef","toolCallCountRef","persistToolInteractions","messageId","persistence","then","update","find","innerMessage","previousMessages","slice","indexOf","at","storedToolArtifacts","storedToolInteractions","storedToolApprovalResponses","innerId","persisted","catch","adapter","idleScheduledInnerIds","unpersistedInnerIds","filter","unsubscribe","wasRunning","lastMsg","role","currentToolCallCount","p","push","Date","now","clearTimeout","pending","every","persistSettled","initialThreadState","ignoreRunning","latest","boundaries","durationMs","lastAssistant","findLast","tcCount","totalSteps","stepDur","i","Math","round","stepTimestamps","endMs","start_ms","end_ms","telemetryOptions","lastInnerMessageId","getLastInnerId","msgs","toBatchItems","msg","idx","innerMessages","isTerminal","isAwaitingToolCalls","isReady","batchItems","append","delete","reportTelemetry","deleteMessage","deleteMessages","bind","messageIndex","findIndex","previousInnerMessages","itemsToDelete","deletion","persistToolApprovalResponses"],"sources":["../../src/runtime/useExternalHistory.ts"],"sourcesContent":["\"use client\";\n\nimport type {\n  AssistantRuntime,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatItem,\n  MessageFormatRepository,\n  ExportedMessageRepository,\n  RespondToToolApprovalOptions,\n  Unstable_ToolInteractionLog,\n} from \"@assistant-ui/core\";\nimport { getExternalStoreMessages } from \"@assistant-ui/core\";\nimport {\n  MessageRepository,\n  readToolInteractionLog,\n} from \"@assistant-ui/core/internal\";\nimport { useAui } from \"@assistant-ui/store\";\nimport {\n  useRef,\n  useEffect,\n  useState,\n  type RefObject,\n  useCallback,\n  useMemo,\n} from \"react\";\n\nexport const toExportedMessageRepository = <TMessage>(\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  messages: MessageFormatRepository<TMessage>,\n): ExportedMessageRepository => {\n  const survivingIds = new Set<string>();\n  const survivors = messages.messages.flatMap((m) => {\n    const message = toThreadMessages([m.message])[0];\n    if (!message) {\n      console.warn(\"Skipping a stored message that could not be loaded.\");\n      return [];\n    }\n    if (m.parentId && !survivingIds.has(m.parentId)) return [];\n    survivingIds.add(message.id);\n    return [{ ...m, message }];\n  });\n\n  return {\n    headId:\n      messages.headId && survivingIds.has(messages.headId)\n        ? messages.headId\n        : null,\n    messages: survivors,\n  };\n};\n\nconst hasUnansweredApproval = (message: ThreadMessage) =>\n  message.content.some(\n    (part) =>\n      part.type === \"tool-call\" &&\n      part.approval != null &&\n      part.approval.approved === undefined &&\n      part.approval.resolution === undefined,\n  );\n\n// Core reports a run paused on an unanswered approval as an interrupt, not as tool calls.\nconst isAwaitingToolApproval = (message: ThreadMessage) =>\n  message.status?.type === \"requires-action\" &&\n  (message.status.reason === \"tool-calls\" ||\n    (message.status.reason === \"interrupt\" && hasUnansweredApproval(message)));\n\nconst isTerminalMessage = (message: ThreadMessage) =>\n  message.status === undefined ||\n  message.status.type === \"complete\" ||\n  message.status.type === \"incomplete\";\n\nconst TOOL_ARTIFACTS_METADATA_KEY = \"__aui_toolArtifacts\";\nconst TOOL_INTERACTIONS_METADATA_KEY = \"__aui_toolInteractions\";\nconst TOOL_APPROVAL_RESPONSES_METADATA_KEY = \"__aui_toolApprovalResponses\";\n\ntype StoredToolApprovalResponse = Omit<\n  RespondToToolApprovalOptions,\n  \"approvalId\"\n>;\n\nconst isRecord = (value: unknown): value is Record<string, unknown> =>\n  value !== null && typeof value === \"object\" && !Array.isArray(value);\n\nconst collectToolArtifacts = (\n  message: ThreadMessage,\n  toolArtifacts: ReadonlyMap<string, unknown> | undefined,\n) => {\n  if (!toolArtifacts) return undefined;\n  const entries = message.content.flatMap((part) => {\n    if (part.type !== \"tool-call\") return [];\n    const artifact = toolArtifacts.get(part.toolCallId);\n    return artifact === undefined ? [] : [[part.toolCallId, artifact] as const];\n  });\n  return entries.length > 0 ? Object.fromEntries(entries) : undefined;\n};\n\nconst collectToolInteractions = (\n  message: ThreadMessage,\n  toolInteractions:\n    | ReadonlyMap<string, Unstable_ToolInteractionLog>\n    | undefined,\n) => {\n  if (!toolInteractions) return undefined;\n  const entries = message.content.flatMap((part) => {\n    if (part.type !== \"tool-call\") return [];\n    const interactions = toolInteractions.get(part.toolCallId);\n    return interactions === undefined\n      ? []\n      : [[part.toolCallId, interactions] as const];\n  });\n  return entries.length > 0 ? Object.fromEntries(entries) : undefined;\n};\n\nconst collectToolApprovalResponses = (\n  message: ThreadMessage,\n  toolApprovalResponses:\n    | ReadonlyMap<string, RespondToToolApprovalOptions>\n    | undefined,\n) => {\n  if (!toolApprovalResponses) return undefined;\n  const entries = message.content.flatMap((part) => {\n    if (part.type !== \"tool-call\" || !part.approval) return [];\n    const response = toolApprovalResponses.get(part.approval.id);\n    if (!response) return [];\n    return [\n      [\n        part.approval.id,\n        {\n          approved: response.approved,\n          ...(response.optionId != null && { optionId: response.optionId }),\n          ...(response.text != null && { text: response.text }),\n          ...(response.reason != null && { reason: response.reason }),\n        },\n      ] as const,\n    ];\n  });\n  return entries.length > 0 ? Object.fromEntries(entries) : undefined;\n};\n\nconst addToolData = <TMessage>(\n  message: TMessage,\n  toolArtifacts: Record<string, unknown> | undefined,\n  toolInteractions: Record<string, Unstable_ToolInteractionLog> | undefined,\n  toolApprovalResponses: Record<string, StoredToolApprovalResponse> | undefined,\n): TMessage => {\n  if (\n    (!toolArtifacts && !toolInteractions && !toolApprovalResponses) ||\n    !isRecord(message) ||\n    !Array.isArray(message.parts)\n  )\n    return message;\n  const toolCallIds = message.parts.flatMap((part) => {\n    if (!isRecord(part) || typeof part.toolCallId !== \"string\") return [];\n    return [part.toolCallId];\n  });\n  const artifacts = toolArtifacts\n    ? Object.fromEntries(\n        toolCallIds.flatMap((toolCallId) =>\n          Object.hasOwn(toolArtifacts, toolCallId)\n            ? [[toolCallId, toolArtifacts[toolCallId]] as const]\n            : [],\n        ),\n      )\n    : undefined;\n  const interactions = toolInteractions\n    ? Object.fromEntries(\n        toolCallIds.flatMap((toolCallId) =>\n          Object.hasOwn(toolInteractions, toolCallId)\n            ? [[toolCallId, toolInteractions[toolCallId]] as const]\n            : [],\n        ),\n      )\n    : undefined;\n  const approvalIds = message.parts.flatMap((part) => {\n    if (!isRecord(part) || !isRecord(part.approval)) return [];\n    const approvalId = part.approval.id;\n    return typeof approvalId === \"string\" ? [approvalId] : [];\n  });\n  const approvalResponses = toolApprovalResponses\n    ? Object.fromEntries(\n        approvalIds.flatMap((approvalId) =>\n          Object.hasOwn(toolApprovalResponses, approvalId)\n            ? [[approvalId, toolApprovalResponses[approvalId]] as const]\n            : [],\n        ),\n      )\n    : undefined;\n  const hasArtifacts = !!artifacts && Object.keys(artifacts).length > 0;\n  const hasInteractions =\n    !!interactions && Object.keys(interactions).length > 0;\n  const hasApprovalResponses =\n    !!approvalResponses && Object.keys(approvalResponses).length > 0;\n  if (!hasArtifacts && !hasInteractions && !hasApprovalResponses)\n    return message;\n  const metadata = isRecord(message.metadata) ? message.metadata : {};\n  return {\n    ...message,\n    metadata: {\n      ...metadata,\n      ...(hasArtifacts && { [TOOL_ARTIFACTS_METADATA_KEY]: artifacts }),\n      ...(hasInteractions && {\n        [TOOL_INTERACTIONS_METADATA_KEY]: interactions,\n      }),\n      ...(hasApprovalResponses && {\n        [TOOL_APPROVAL_RESPONSES_METADATA_KEY]: approvalResponses,\n      }),\n    },\n  } as TMessage;\n};\n\nconst restoreToolData = <TMessage>(\n  message: TMessage,\n  toolArtifacts: Map<string, unknown> | undefined,\n  toolInteractions: Map<string, Unstable_ToolInteractionLog> | undefined,\n  toolApprovalResponses: Map<string, RespondToToolApprovalOptions> | undefined,\n): TMessage => {\n  if (!isRecord(message) || !isRecord(message.metadata)) return message;\n  const metadata = message.metadata;\n  const hasArtifacts = Object.hasOwn(metadata, TOOL_ARTIFACTS_METADATA_KEY);\n  const hasInteractions = Object.hasOwn(\n    metadata,\n    TOOL_INTERACTIONS_METADATA_KEY,\n  );\n  const hasApprovalResponses = Object.hasOwn(\n    metadata,\n    TOOL_APPROVAL_RESPONSES_METADATA_KEY,\n  );\n  if (!hasArtifacts && !hasInteractions && !hasApprovalResponses)\n    return message;\n  const artifacts = metadata[TOOL_ARTIFACTS_METADATA_KEY];\n  if (toolArtifacts && isRecord(artifacts)) {\n    for (const [toolCallId, artifact] of Object.entries(artifacts)) {\n      toolArtifacts.set(toolCallId, artifact);\n    }\n  }\n  const interactions = metadata[TOOL_INTERACTIONS_METADATA_KEY];\n  if (toolInteractions && isRecord(interactions)) {\n    for (const [toolCallId, value] of Object.entries(interactions)) {\n      const log = readToolInteractionLog(value);\n      if (log) toolInteractions.set(toolCallId, log);\n    }\n  }\n  const approvalResponses = metadata[TOOL_APPROVAL_RESPONSES_METADATA_KEY];\n  if (toolApprovalResponses && isRecord(approvalResponses)) {\n    for (const [approvalId, value] of Object.entries(approvalResponses)) {\n      if (!isRecord(value) || typeof value.approved !== \"boolean\") continue;\n      toolApprovalResponses.set(approvalId, {\n        approvalId,\n        approved: value.approved,\n        ...(typeof value.optionId === \"string\" && {\n          optionId: value.optionId,\n        }),\n        ...(typeof value.text === \"string\" && { text: value.text }),\n        ...(typeof value.reason === \"string\" && { reason: value.reason }),\n      });\n    }\n  }\n  const {\n    [TOOL_ARTIFACTS_METADATA_KEY]: _,\n    [TOOL_INTERACTIONS_METADATA_KEY]: __,\n    [TOOL_APPROVAL_RESPONSES_METADATA_KEY]: ___,\n    ...restMetadata\n  } = metadata;\n  const { metadata: _metadata, ...restMessage } = message;\n  return (\n    Object.keys(restMetadata).length === 0\n      ? restMessage\n      : { ...restMessage, metadata: restMetadata }\n  ) as TMessage;\n};\n\nconst encodeContent = <TMessage>(\n  storageFormatAdapter: MessageFormatAdapter<TMessage, any>,\n  item: MessageFormatItem<TMessage>,\n) => JSON.stringify(storageFormatAdapter.encode(item));\n\nexport const useExternalHistory = <TMessage>(\n  runtimeRef: RefObject<AssistantRuntime>,\n  historyAdapter: ThreadHistoryAdapter | undefined,\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  storageFormatAdapter: MessageFormatAdapter<TMessage, any>,\n  onSetMessages: (messages: TMessage[]) => void,\n  toolArtifacts?: Map<string, unknown>,\n  onToolArtifactsRestored?: () => void,\n  toolInteractions?: Map<string, Unstable_ToolInteractionLog>,\n  onToolInteractionsRestored?: () => void,\n  toolApprovalResponses?: Map<string, RespondToToolApprovalOptions>,\n  onToolApprovalResponsesRestored?: () => void,\n) => {\n  const loadedRef = useRef(false);\n  const [itemEpoch, setItemEpoch] = useState(0);\n\n  const aui = useAui();\n  const optionalThreadListItem = useCallback(\n    () => (aui.threadListItem.source ? aui.threadListItem : null),\n    [aui],\n  );\n\n  const [hasLoaded, setHasLoaded] = useState(false);\n\n  const historyIds = useRef(new Set<string>());\n  const deferredTelemetryIds = useRef(new Set<string>());\n  // `content` is a snapshot taken at write time rather than a re-encode of `source`, because a retained message object can be mutated in place by the runtime that produced it.\n  const persistedInnerMessages = useRef(\n    new Map<string, { source: TMessage; content: string }>(),\n  );\n\n  const onSetMessagesRef = useRef(onSetMessages);\n  useEffect(() => {\n    onSetMessagesRef.current = onSetMessages;\n  });\n\n  const formatAdapter = useMemo(() => {\n    if (!historyAdapter) return undefined;\n    if (!historyAdapter.withFormat) {\n      throw new Error(\n        \"useAISDKRuntime: ThreadHistoryAdapter is missing the required `withFormat` method.\",\n      );\n    }\n    return historyAdapter.withFormat<TMessage, any>(storageFormatAdapter);\n  }, [historyAdapter, storageFormatAdapter]);\n\n  const isLoading = formatAdapter != null && !hasLoaded;\n\n  useEffect(() => {\n    if (!formatAdapter || loadedRef.current) return undefined;\n\n    const loadHistory = async () => {\n      try {\n        const repo = await formatAdapter.load();\n        toolArtifacts?.clear();\n        toolInteractions?.clear();\n        toolApprovalResponses?.clear();\n        const restoredMessages =\n          repo?.messages.map((item) => ({\n            ...item,\n            message: restoreToolData(\n              item.message,\n              toolArtifacts,\n              toolInteractions,\n              toolApprovalResponses,\n            ),\n          })) ?? [];\n        onToolArtifactsRestored?.();\n        onToolInteractionsRestored?.();\n        onToolApprovalResponsesRestored?.();\n        if (repo && restoredMessages.length > 0) {\n          const restoredRepo = { ...repo, messages: restoredMessages };\n          for (const [index, m] of repo.messages.entries()) {\n            persistedInnerMessages.current.set(\n              storageFormatAdapter.getId(m.message),\n              {\n                source: restoredMessages[index]!.message,\n                content: encodeContent(storageFormatAdapter, m),\n              },\n            );\n          }\n          const converted = toExportedMessageRepository(\n            toThreadMessages,\n            restoredRepo,\n          );\n          runtimeRef.current.thread.import(converted);\n\n          const tempRepo = new MessageRepository();\n          tempRepo.import(converted);\n          const messages = tempRepo.getMessages();\n\n          onSetMessagesRef.current(\n            messages.flatMap(getExternalStoreMessages<TMessage>),\n          );\n\n          historyIds.current = new Set();\n          for (const m of converted.messages) {\n            historyIds.current.add(m.message.id);\n            if (isAwaitingToolApproval(m.message)) {\n              deferredTelemetryIds.current.add(m.message.id);\n            }\n          }\n        }\n      } catch (error) {\n        console.error(\"Failed to load message history:\", error);\n      } finally {\n        setHasLoaded(true);\n      }\n    };\n\n    formatAdapter.pin?.();\n\n    const remoteId = optionalThreadListItem()?.getState().remoteId;\n    if (!remoteId) {\n      // History loads asynchronously against the thread list item; without a\n      // remote id there is nothing to await, so the flag settles here.\n      // eslint-disable-next-line react-hooks/set-state-in-effect\n      setHasLoaded(true);\n      return aui.subscribe(() => {\n        if (optionalThreadListItem()?.getState().remoteId) {\n          setItemEpoch((n) => n + 1);\n        }\n      });\n    }\n\n    const threadState = runtimeRef.current.thread.getState();\n    if (threadState.isRunning || threadState.messages.length > 0) {\n      loadedRef.current = true;\n      setHasLoaded(true);\n      return undefined;\n    }\n\n    loadedRef.current = true;\n    void loadHistory();\n    return undefined;\n  }, [\n    formatAdapter,\n    toThreadMessages,\n    runtimeRef,\n    optionalThreadListItem,\n    aui,\n    itemEpoch,\n    storageFormatAdapter,\n    toolArtifacts,\n    onToolArtifactsRestored,\n    toolInteractions,\n    onToolInteractionsRestored,\n    toolApprovalResponses,\n    onToolApprovalResponsesRestored,\n  ]);\n\n  const runStartRef = useRef<number | null>(null);\n  const persistTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);\n  const persistInFlightRef = useRef<Promise<void>>(Promise.resolve());\n  const stepBoundariesRef = useRef<number[]>([]);\n  const wasRunningRef = useRef(false);\n  const toolCallCountRef = useRef(0);\n\n  const persistToolInteractions = useCallback(\n    (messageId: string) => {\n      const persistence = persistInFlightRef.current.then(async () => {\n        if (!formatAdapter?.update) return;\n        const messages = runtimeRef.current.thread.getState().messages;\n        const message = messages.find(\n          (item) =>\n            item.id === messageId ||\n            getExternalStoreMessages<TMessage>(item).some(\n              (innerMessage) =>\n                storageFormatAdapter.getId(innerMessage) === messageId,\n            ),\n        );\n        if (!message) return;\n\n        const previousMessages = messages\n          .slice(0, messages.indexOf(message))\n          .flatMap(getExternalStoreMessages<TMessage>);\n        let parentId = previousMessages.at(-1)\n          ? storageFormatAdapter.getId(previousMessages.at(-1)!)\n          : null;\n        const storedToolArtifacts = collectToolArtifacts(\n          message,\n          toolArtifacts,\n        );\n        const storedToolInteractions = collectToolInteractions(\n          message,\n          toolInteractions,\n        );\n        const storedToolApprovalResponses = collectToolApprovalResponses(\n          message,\n          toolApprovalResponses,\n        );\n\n        for (const innerMessage of getExternalStoreMessages<TMessage>(\n          message,\n        )) {\n          const item = {\n            parentId,\n            message: addToolData(\n              innerMessage,\n              storedToolArtifacts,\n              storedToolInteractions,\n              storedToolApprovalResponses,\n            ),\n          };\n          const innerId = storageFormatAdapter.getId(item.message);\n          const persisted = persistedInnerMessages.current.get(innerId);\n          if (persisted) {\n            const content = encodeContent(storageFormatAdapter, item);\n            if (content === persisted.content) {\n              persisted.source = item.message;\n            } else {\n              await formatAdapter.update(item, innerId);\n              persistedInnerMessages.current.set(innerId, {\n                source: item.message,\n                content,\n              });\n            }\n          }\n          parentId = innerId;\n        }\n      });\n      persistInFlightRef.current = persistence.catch(() => {});\n      return persistence.catch((error) => {\n        console.error(\"Failed to persist tool data:\", error);\n      });\n    },\n    [\n      formatAdapter,\n      runtimeRef,\n      storageFormatAdapter,\n      toolArtifacts,\n      toolInteractions,\n      toolApprovalResponses,\n    ],\n  );\n\n  useEffect(() => {\n    if (!formatAdapter) return;\n    const adapter = formatAdapter;\n\n    const idleScheduledInnerIds = new Set<string>();\n    const unpersistedInnerIds = (message: ThreadMessage) =>\n      isTerminalMessage(message)\n        ? getExternalStoreMessages<TMessage>(message)\n            .map((innerMessage) => storageFormatAdapter.getId(innerMessage))\n            .filter((innerId) => !persistedInnerMessages.current.has(innerId))\n        : [];\n\n    const unsubscribe = runtimeRef.current.thread.subscribe(() => {\n      const threadState = runtimeRef.current.thread.getState();\n      const { isRunning } = threadState;\n      const wasRunning = wasRunningRef.current;\n      wasRunningRef.current = isRunning;\n\n      // Track step boundaries by content changes (more reliable than isRunning)\n      if (runStartRef.current != null) {\n        const lastMsg = threadState.messages.at(-1);\n        if (lastMsg?.role === \"assistant\") {\n          const currentToolCallCount = lastMsg.content.filter(\n            (p) => p.type === \"tool-call\",\n          ).length;\n          while (toolCallCountRef.current < currentToolCallCount) {\n            stepBoundariesRef.current.push(Date.now() - runStartRef.current);\n            toolCallCountRef.current++;\n          }\n        }\n      }\n\n      if (isRunning) {\n        if (runStartRef.current == null) {\n          runStartRef.current = Date.now();\n          stepBoundariesRef.current = [];\n          toolCallCountRef.current = 0;\n          adapter.pin?.();\n        }\n        // Cancel any pending persist — isRunning went back to true\n        if (persistTimerRef.current) {\n          clearTimeout(persistTimerRef.current);\n          persistTimerRef.current = null;\n        }\n        return;\n      }\n\n      if (wasRunning) {\n        // Record step boundary offset (synchronous for accuracy)\n        if (runStartRef.current != null) {\n          stepBoundariesRef.current.push(Date.now() - runStartRef.current);\n        }\n      } else {\n        const pending = threadState.messages.flatMap(unpersistedInnerIds);\n        if (pending.every((innerId) => idleScheduledInnerIds.has(innerId)))\n          return;\n        for (const innerId of pending) idleScheduledInnerIds.add(innerId);\n      }\n\n      // Debounce: wait one macrotask so agentic step flickers are absorbed\n      if (persistTimerRef.current) clearTimeout(persistTimerRef.current);\n      persistTimerRef.current = setTimeout(() => {\n        persistSettled(false);\n      }, 0);\n    });\n\n    const initialThreadState = runtimeRef.current.thread.getState();\n    wasRunningRef.current = initialThreadState.isRunning;\n    if (initialThreadState.isRunning) {\n      if (runStartRef.current == null) {\n        runStartRef.current = Date.now();\n        stepBoundariesRef.current = [];\n        toolCallCountRef.current = 0;\n        adapter.pin?.();\n      }\n    } else if (initialThreadState.messages.length > 0) {\n      persistSettled(true);\n    }\n\n    function persistSettled(ignoreRunning: boolean) {\n      persistTimerRef.current = null;\n      const latest = runtimeRef.current.thread.getState();\n      if (!ignoreRunning && latest.isRunning) return;\n\n      const boundaries = stepBoundariesRef.current;\n      const durationMs = boundaries.length > 0 ? boundaries.at(-1) : undefined;\n\n      // Fallback: if only 1 boundary but message has multiple steps, distribute evenly\n      if (boundaries.length === 1 && durationMs != null) {\n        const lastAssistant = latest.messages.findLast(\n          (m) => m.role === \"assistant\",\n        );\n        if (lastAssistant) {\n          const tcCount = lastAssistant.content.filter(\n            (p) => p.type === \"tool-call\",\n          ).length;\n          if (tcCount > 0) {\n            const totalSteps = tcCount + 1;\n            const stepDur = durationMs / totalSteps;\n            boundaries.length = 0;\n            for (let i = 0; i < totalSteps; i++) {\n              boundaries.push(Math.round((i + 1) * stepDur));\n            }\n          }\n        }\n      }\n\n      // Build per-step timestamps when there are multiple steps\n      const stepTimestamps =\n        boundaries.length > 1\n          ? boundaries.map((endMs, i) => ({\n              start_ms: i === 0 ? 0 : boundaries[i - 1]!,\n              end_ms: endMs,\n            }))\n          : undefined;\n\n      runStartRef.current = null;\n      stepBoundariesRef.current = [];\n\n      const telemetryOptions = {\n        ...(durationMs != null ? { durationMs } : undefined),\n        ...(stepTimestamps != null ? { stepTimestamps } : undefined),\n      };\n\n      persistInFlightRef.current = persistInFlightRef.current\n        .then(async () => {\n          const { messages } = latest;\n          let lastInnerMessageId: string | null = null;\n\n          const getLastInnerId = (msgs: TMessage[]): string | null =>\n            msgs.length > 0 ? storageFormatAdapter.getId(msgs.at(-1)!) : null;\n\n          const toBatchItems = (\n            msgs: TMessage[],\n            toolArtifacts: Record<string, unknown> | undefined,\n            toolInteractions:\n              | Record<string, Unstable_ToolInteractionLog>\n              | undefined,\n            toolApprovalResponses:\n              | Record<string, StoredToolApprovalResponse>\n              | undefined,\n          ) =>\n            msgs.map((msg, idx) => ({\n              parentId:\n                idx === 0\n                  ? lastInnerMessageId\n                  : storageFormatAdapter.getId(msgs[idx - 1]!),\n              message: addToolData(\n                msg,\n                toolArtifacts,\n                toolInteractions,\n                toolApprovalResponses,\n              ),\n            }));\n\n          for (const message of messages) {\n            const innerMessages = getExternalStoreMessages<TMessage>(message);\n\n            const isTerminal = isTerminalMessage(message);\n            const isAwaitingToolCalls = isAwaitingToolApproval(message);\n            // A paused message's later content can only reach storage via update, so it is persisted early only when the adapter supports update.\n            const isReady =\n              isTerminal ||\n              (isAwaitingToolCalls && adapter.update !== undefined);\n\n            if (!isReady) {\n              lastInnerMessageId =\n                getLastInnerId(innerMessages) ?? lastInnerMessageId;\n              continue;\n            }\n\n            if (!historyIds.current.has(message.id)) {\n              historyIds.current.add(message.id);\n              deferredTelemetryIds.current.add(message.id);\n            }\n\n            const batchItems = toBatchItems(\n              innerMessages,\n              collectToolArtifacts(message, toolArtifacts),\n              collectToolInteractions(message, toolInteractions),\n              collectToolApprovalResponses(message, toolApprovalResponses),\n            );\n            for (const item of batchItems) {\n              const innerId = storageFormatAdapter.getId(item.message);\n              const persisted = persistedInnerMessages.current.get(innerId);\n              if (!persisted) {\n                await adapter.append(item);\n                persistedInnerMessages.current.set(innerId, {\n                  source: item.message,\n                  content: encodeContent(storageFormatAdapter, item),\n                });\n              } else if (\n                persisted.source !== item.message &&\n                durationMs !== undefined &&\n                adapter.update\n              ) {\n                const content = encodeContent(storageFormatAdapter, item);\n                if (content === persisted.content) {\n                  persisted.source = item.message;\n                } else {\n                  try {\n                    await adapter.update(item, innerId);\n                    persistedInnerMessages.current.set(innerId, {\n                      source: item.message,\n                      content,\n                    });\n                  } catch {\n                    // A failed update leaves the stale baseline behind so the next run stop retries it.\n                  }\n                }\n              }\n            }\n\n            lastInnerMessageId =\n              getLastInnerId(innerMessages) ?? lastInnerMessageId;\n\n            if (deferredTelemetryIds.current.has(message.id) && isTerminal) {\n              deferredTelemetryIds.current.delete(message.id);\n              adapter.reportTelemetry?.(\n                batchItems.map((item, index) => ({\n                  ...item,\n                  message: innerMessages[index]!,\n                })),\n                {\n                  ...telemetryOptions,\n                  message,\n                },\n              );\n            }\n          }\n        })\n        .catch((error) => {\n          console.error(\"Failed to persist message history:\", error);\n        });\n    }\n\n    return () => {\n      unsubscribe();\n      if (persistTimerRef.current) {\n        clearTimeout(persistTimerRef.current);\n        persistTimerRef.current = null;\n        persistSettled(false);\n      }\n    };\n  }, [\n    formatAdapter,\n    storageFormatAdapter,\n    runtimeRef,\n    toolArtifacts,\n    toolInteractions,\n    toolApprovalResponses,\n  ]);\n\n  const deleteMessage = useCallback(\n    async (messageId: string) => {\n      const deleteMessages = formatAdapter?.delete?.bind(formatAdapter);\n      if (!deleteMessages) return;\n\n      const messages = runtimeRef.current.thread.getState().messages;\n      const messageIndex = messages.findIndex((m) => m.id === messageId);\n      if (messageIndex === -1) return;\n\n      const previousInnerMessages = messages\n        .slice(0, messageIndex)\n        .flatMap(getExternalStoreMessages<TMessage>);\n      let parentId = previousInnerMessages.at(-1)\n        ? storageFormatAdapter.getId(previousInnerMessages.at(-1)!)\n        : null;\n      const itemsToDelete = getExternalStoreMessages<TMessage>(\n        messages[messageIndex]!,\n      ).map((message) => {\n        const item = { parentId, message };\n        parentId = storageFormatAdapter.getId(message);\n        return item;\n      });\n\n      const deletion = persistInFlightRef.current.then(async () => {\n        await deleteMessages(itemsToDelete);\n\n        historyIds.current.delete(messageId);\n        deferredTelemetryIds.current.delete(messageId);\n        for (const item of itemsToDelete) {\n          persistedInnerMessages.current.delete(\n            storageFormatAdapter.getId(item.message),\n          );\n        }\n      });\n\n      persistInFlightRef.current = deletion.catch(() => {});\n      await deletion;\n    },\n    [formatAdapter, runtimeRef, storageFormatAdapter],\n  );\n\n  return {\n    isLoading,\n    deleteMessage,\n    persistToolInteractions,\n    persistToolApprovalResponses: persistToolInteractions,\n  };\n};\n"],"mappings":";;;;;;AA4BA,MAAamB,+BACXC,kBACAC,aAC8B;CAC9B,MAAME,+BAAe,IAAIC,IAAY;CACrC,MAAMC,YAAYJ,SAASA,SAASK,SAASC,MAAM;EACjD,MAAMC,UAAUR,iBAAiB,CAACO,EAAEC,OAAO,CAAC,CAAC,CAAC;EAC9C,IAAI,CAACA,SAAS;GACZC,QAAQC,KAAK,qDAAqD;GAClE,OAAO,CAAA;EACT;EACA,IAAIH,EAAEI,YAAY,CAACR,aAAaS,IAAIL,EAAEI,QAAQ,GAAG,OAAO,CAAA;EACxDR,aAAaU,IAAIL,QAAQM,EAAE;EAC3B,OAAO,CAAC;GAAE,GAAGP;GAAGC;EAAQ,CAAC;CAC3B,CAAC;CAED,OAAO;EACLO,QACEd,SAASc,UAAUZ,aAAaS,IAAIX,SAASc,MAAM,IAC/Cd,SAASc,SACT;EACNd,UAAUI;CACZ;AACF;AAEA,MAAMW,yBAAyBR,YAC7BA,QAAQS,QAAQC,MACbC,SACCA,KAAKC,SAAS,eACdD,KAAKE,YAAY,QACjBF,KAAKE,SAASC,aAAaC,KAAAA,KAC3BJ,KAAKE,SAASG,eAAeD,KAAAA,CACjC;AAGF,MAAME,0BAA0BjB,YAC9BA,QAAQkB,QAAQN,SAAS,sBACxBZ,QAAQkB,OAAOC,WAAW,gBACxBnB,QAAQkB,OAAOC,WAAW,eAAeX,sBAAsBR,OAAO;AAE3E,MAAMoB,qBAAqBpB,YACzBA,QAAQkB,WAAWH,KAAAA,KACnBf,QAAQkB,OAAON,SAAS,cACxBZ,QAAQkB,OAAON,SAAS;AAE1B,MAAMS,8BAA8B;AACpC,MAAMC,iCAAiC;AACvC,MAAMC,uCAAuC;AAO7C,MAAMG,YAAYC,UAChBA,UAAU,QAAQ,OAAOA,UAAU,YAAY,CAACE,MAAMC,QAAQH,KAAK;AAErE,MAAMI,wBACJ/B,SACAgC,kBACG;CACH,IAAI,CAACA,eAAe,OAAOjB,KAAAA;CAC3B,MAAMmB,UAAUlC,QAAQS,QAAQX,SAASa,SAAS;EAChD,IAAIA,KAAKC,SAAS,aAAa,OAAO,CAAA;EACtC,MAAMuB,WAAWH,cAAcI,IAAIzB,KAAK0B,UAAU;EAClD,OAAOF,aAAapB,KAAAA,IAAY,CAAA,IAAK,CAAC,CAACJ,KAAK0B,YAAYF,QAAQ,CAAU;CAC5E,CAAC;CACD,OAAOD,QAAQK,SAAS,IAAIC,OAAOC,YAAYP,OAAO,IAAInB,KAAAA;AAC5D;AAEA,MAAM2B,2BACJ1C,SACA2C,qBAGG;CACH,IAAI,CAACA,kBAAkB,OAAO5B,KAAAA;CAC9B,MAAMmB,UAAUlC,QAAQS,QAAQX,SAASa,SAAS;EAChD,IAAIA,KAAKC,SAAS,aAAa,OAAO,CAAA;EACtC,MAAMgC,eAAeD,iBAAiBP,IAAIzB,KAAK0B,UAAU;EACzD,OAAOO,iBAAiB7B,KAAAA,IACpB,CAAA,IACA,CAAC,CAACJ,KAAK0B,YAAYO,YAAY,CAAU;CAC/C,CAAC;CACD,OAAOV,QAAQK,SAAS,IAAIC,OAAOC,YAAYP,OAAO,IAAInB,KAAAA;AAC5D;AAEA,MAAM8B,gCACJ7C,SACA8C,0BAGG;CACH,IAAI,CAACA,uBAAuB,OAAO/B,KAAAA;CACnC,MAAMmB,UAAUlC,QAAQS,QAAQX,SAASa,SAAS;EAChD,IAAIA,KAAKC,SAAS,eAAe,CAACD,KAAKE,UAAU,OAAO,CAAA;EACxD,MAAMkC,WAAWD,sBAAsBV,IAAIzB,KAAKE,SAASP,EAAE;EAC3D,IAAI,CAACyC,UAAU,OAAO,CAAA;EACtB,OAAO,CACL,CACEpC,KAAKE,SAASP,IACd;GACEQ,UAAUiC,SAASjC;GACnB,GAAIiC,SAASC,YAAY,QAAQ,EAAEA,UAAUD,SAASC,SAAS;GAC/D,GAAID,SAASE,QAAQ,QAAQ,EAAEA,MAAMF,SAASE,KAAK;GACnD,GAAIF,SAAS5B,UAAU,QAAQ,EAAEA,QAAQ4B,SAAS5B,OAAO;EAC3D,CAAC,CACO;CAEd,CAAC;CACD,OAAOe,QAAQK,SAAS,IAAIC,OAAOC,YAAYP,OAAO,IAAInB,KAAAA;AAC5D;AAEA,MAAMmC,eACJlD,SACAgC,eACAW,kBACAG,0BACa;CACb,IACG,CAACd,iBAAiB,CAACW,oBAAoB,CAACG,yBACzC,CAACpB,SAAS1B,OAAO,KACjB,CAAC6B,MAAMC,QAAQ9B,QAAQmD,KAAK,GAE5B,OAAOnD;CACT,MAAMoD,cAAcpD,QAAQmD,MAAMrD,SAASa,SAAS;EAClD,IAAI,CAACe,SAASf,IAAI,KAAK,OAAOA,KAAK0B,eAAe,UAAU,OAAO,CAAA;EACnE,OAAO,CAAC1B,KAAK0B,UAAU;CACzB,CAAC;CACD,MAAMgB,YAAYrB,gBACdQ,OAAOC,YACLW,YAAYtD,SAASuC,eACnBG,OAAOc,OAAOtB,eAAeK,UAAU,IACnC,CAAC,CAACA,YAAYL,cAAcK,WAAW,CAAU,IACjD,CAAA,CACN,CACF,IACAtB,KAAAA;CACJ,MAAM6B,eAAeD,mBACjBH,OAAOC,YACLW,YAAYtD,SAASuC,eACnBG,OAAOc,OAAOX,kBAAkBN,UAAU,IACtC,CAAC,CAACA,YAAYM,iBAAiBN,WAAW,CAAU,IACpD,CAAA,CACN,CACF,IACAtB,KAAAA;CACJ,MAAMwC,cAAcvD,QAAQmD,MAAMrD,SAASa,SAAS;EAClD,IAAI,CAACe,SAASf,IAAI,KAAK,CAACe,SAASf,KAAKE,QAAQ,GAAG,OAAO,CAAA;EACxD,MAAM2C,aAAa7C,KAAKE,SAASP;EACjC,OAAO,OAAOkD,eAAe,WAAW,CAACA,UAAU,IAAI,CAAA;CACzD,CAAC;CACD,MAAMC,oBAAoBX,wBACtBN,OAAOC,YACLc,YAAYzD,SAAS0D,eACnBhB,OAAOc,OAAOR,uBAAuBU,UAAU,IAC3C,CAAC,CAACA,YAAYV,sBAAsBU,WAAW,CAAU,IACzD,CAAA,CACN,CACF,IACAzC,KAAAA;CACJ,MAAM2C,eAAe,CAAC,CAACL,aAAab,OAAOmB,KAAKN,SAAS,CAAC,CAACd,SAAS;CACpE,MAAMqB,kBACJ,CAAC,CAAChB,gBAAgBJ,OAAOmB,KAAKf,YAAY,CAAC,CAACL,SAAS;CACvD,MAAMsB,uBACJ,CAAC,CAACJ,qBAAqBjB,OAAOmB,KAAKF,iBAAiB,CAAC,CAAClB,SAAS;CACjE,IAAI,CAACmB,gBAAgB,CAACE,mBAAmB,CAACC,sBACxC,OAAO7D;CACT,MAAM8D,WAAWpC,SAAS1B,QAAQ8D,QAAQ,IAAI9D,QAAQ8D,WAAW,CAAC;CAClE,OAAO;EACL,GAAG9D;EACH8D,UAAU;GACR,GAAGA;GACH,GAAIJ,gBAAgB,GAAGrC,8BAA8BgC,UAAU;GAC/D,GAAIO,mBAAmB,GACpBtC,iCAAiCsB,aACpC;GACA,GAAIiB,wBAAwB,GACzBtC,uCAAuCkC,kBAC1C;EACF;CACF;AACF;AAEA,MAAMM,mBACJ/D,SACAgC,eACAW,kBACAG,0BACa;CACb,IAAI,CAACpB,SAAS1B,OAAO,KAAK,CAAC0B,SAAS1B,QAAQ8D,QAAQ,GAAG,OAAO9D;CAC9D,MAAM8D,WAAW9D,QAAQ8D;CACzB,MAAMJ,eAAelB,OAAOc,OAAOQ,UAAUzC,2BAA2B;CACxE,MAAMuC,kBAAkBpB,OAAOc,OAC7BQ,UACAxC,8BACF;CACA,MAAMuC,uBAAuBrB,OAAOc,OAClCQ,UACAvC,oCACF;CACA,IAAI,CAACmC,gBAAgB,CAACE,mBAAmB,CAACC,sBACxC,OAAO7D;CACT,MAAMqD,YAAYS,SAASzC;CAC3B,IAAIW,iBAAiBN,SAAS2B,SAAS,GACrC,KAAK,MAAM,CAAChB,YAAYF,aAAaK,OAAON,QAAQmB,SAAS,GAC3DrB,cAAciC,IAAI5B,YAAYF,QAAQ;CAG1C,MAAMS,eAAekB,SAASxC;CAC9B,IAAIqB,oBAAoBjB,SAASkB,YAAY,GAC3C,KAAK,MAAM,CAACP,YAAYV,UAAUa,OAAON,QAAQU,YAAY,GAAG;EAC9D,MAAMsB,MAAMnF,uBAAuB4C,KAAK;EACxC,IAAIuC,KAAKvB,iBAAiBsB,IAAI5B,YAAY6B,GAAG;CAC/C;CAEF,MAAMT,oBAAoBK,SAASvC;CACnC,IAAIuB,yBAAyBpB,SAAS+B,iBAAiB,GACrD,KAAK,MAAM,CAACD,YAAY7B,UAAUa,OAAON,QAAQuB,iBAAiB,GAAG;EACnE,IAAI,CAAC/B,SAASC,KAAK,KAAK,OAAOA,MAAMb,aAAa,WAAW;EAC7DgC,sBAAsBmB,IAAIT,YAAY;GACpCA;GACA1C,UAAUa,MAAMb;GAChB,GAAI,OAAOa,MAAMqB,aAAa,YAAY,EACxCA,UAAUrB,MAAMqB,SAClB;GACA,GAAI,OAAOrB,MAAMsB,SAAS,YAAY,EAAEA,MAAMtB,MAAMsB,KAAK;GACzD,GAAI,OAAOtB,MAAMR,WAAW,YAAY,EAAEA,QAAQQ,MAAMR,OAAO;EACjE,CAAC;CACH;CAEF,MAAM,GACHE,8BAA8B8C,IAC9B7C,iCAAiC8C,KACjC7C,uCAAuC8C,KACxC,GAAGC,iBACDR;CACJ,MAAM,EAAEA,UAAUS,WAAW,GAAGC,gBAAgBxE;CAChD,OACEwC,OAAOmB,KAAKW,YAAY,CAAC,CAAC/B,WAAW,IACjCiC,cACA;EAAE,GAAGA;EAAaV,UAAUQ;CAAa;AAEjD;AAEA,MAAMG,iBACJC,sBACAC,SACGC,KAAKC,UAAUH,qBAAqBI,OAAOH,IAAI,CAAC;AAErD,MAAaI,sBACXC,YACAC,gBACAzF,kBACAkF,sBACAQ,eACAlD,eACAmD,yBACAxC,kBACAyC,4BACAtC,uBACAuC,oCACG;CACH,MAAMC,YAAYrG,OAAO,KAAK;CAC9B,MAAM,CAACsG,WAAWC,gBAAgBrG,SAAS,CAAC;CAE5C,MAAMsG,MAAMzG,OAAO;CACnB,MAAM0G,yBAAyBrG,kBACtBoG,IAAIE,eAAeC,SAASH,IAAIE,iBAAiB,MACxD,CAACF,GAAG,CACN;CAEA,MAAM,CAACI,WAAWC,gBAAgB3G,SAAS,KAAK;CAEhD,MAAM4G,aAAa9G,uBAAO,IAAIW,IAAY,CAAC;CAC3C,MAAMoG,uBAAuB/G,uBAAO,IAAIW,IAAY,CAAC;CAErD,MAAMqG,yBAAyBhH,uBAC7B,IAAI+E,IAAmD,CACzD;CAEA,MAAMkC,mBAAmBjH,OAAOiG,aAAa;CAC7ChG,gBAAgB;EACdgH,iBAAiBC,UAAUjB;CAC7B,CAAC;CAED,MAAMkB,gBAAgB9G,cAAc;EAClC,IAAI,CAAC2F,gBAAgB,OAAOlE,KAAAA;EAC5B,IAAI,CAACkE,eAAeoB,YAClB,MAAM,IAAIC,MACR,oFACF;EAEF,OAAOrB,eAAeoB,WAA0B3B,oBAAoB;CACtE,GAAG,CAACO,gBAAgBP,oBAAoB,CAAC;CAEzC,MAAM6B,YAAYH,iBAAiB,QAAQ,CAACP;CAE5C3G,gBAAgB;EACd,IAAI,CAACkH,iBAAiBd,UAAUa,SAAS,OAAOpF,KAAAA;EAEhD,MAAMyF,cAAc,YAAY;GAC9B,IAAI;IACF,MAAMC,OAAO,MAAML,cAAcM,KAAK;IACtC1E,eAAe2E,MAAM;IACrBhE,kBAAkBgE,MAAM;IACxB7D,uBAAuB6D,MAAM;IAC7B,MAAMC,mBACJH,MAAMhH,SAASoH,KAAKlC,UAAU;KAC5B,GAAGA;KACH3E,SAAS+D,gBACPY,KAAK3E,SACLgC,eACAW,kBACAG,qBACF;IACF,EAAE,KAAK,CAAA;IACTqC,0BAA0B;IAC1BC,6BAA6B;IAC7BC,kCAAkC;IAClC,IAAIoB,QAAQG,iBAAiBrE,SAAS,GAAG;KACvC,MAAMuE,eAAe;MAAE,GAAGL;MAAMhH,UAAUmH;KAAiB;KAC3D,KAAK,MAAM,CAACG,OAAOhH,MAAM0G,KAAKhH,SAASyC,QAAQ,GAC7C+D,uBAAuBE,QAAQlC,IAC7BS,qBAAqBsC,MAAMjH,EAAEC,OAAO,GACpC;MACE4F,QAAQgB,iBAAiBG,MAAM,CAAE/G;MACjCS,SAASgE,cAAcC,sBAAsB3E,CAAC;KAChD,CACF;KAEF,MAAMkH,YAAY1H,4BAChBC,kBACAsH,YACF;KACA9B,WAAWmB,QAAQe,OAAOC,OAAOF,SAAS;KAE1C,MAAMG,WAAW,IAAItI,kBAAkB;KACvCsI,SAASD,OAAOF,SAAS;KACzB,MAAMxH,WAAW2H,SAASC,YAAY;KAEtCnB,iBAAiBC,QACf1G,SAASK,QAAQjB,wBAAkC,CACrD;KAEAkH,WAAWI,0BAAU,IAAIvG,IAAI;KAC7B,KAAK,MAAMG,OAAKkH,UAAUxH,UAAU;MAClCsG,WAAWI,QAAQ9F,IAAIN,IAAEC,QAAQM,EAAE;MACnC,IAAIW,uBAAuBlB,IAAEC,OAAO,GAClCgG,qBAAqBG,QAAQ9F,IAAIN,IAAEC,QAAQM,EAAE;KAEjD;IACF;GACF,SAASgH,OAAO;IACdrH,QAAQqH,MAAM,mCAAmCA,KAAK;GACxD,UAAU;IACRxB,aAAa,IAAI;GACnB;EACF;EAEAM,cAAcmB,MAAM;EAGpB,IAAI,CADa7B,uBAAuB,CAAC,EAAE+B,SAAS,CAAC,CAACD,UACvC;GAIb1B,aAAa,IAAI;GACjB,OAAOL,IAAIiC,gBAAgB;IACzB,IAAIhC,uBAAuB,CAAC,EAAE+B,SAAS,CAAC,CAACD,UACvChC,cAAcmC,MAAMA,IAAI,CAAC;GAE7B,CAAC;EACH;EAEA,MAAMC,cAAc5C,WAAWmB,QAAQe,OAAOO,SAAS;EACvD,IAAIG,YAAYC,aAAaD,YAAYnI,SAAS8C,SAAS,GAAG;GAC5D+C,UAAUa,UAAU;GACpBL,aAAa,IAAI;GACjB;EACF;EAEAR,UAAUa,UAAU;EACpB,YAAiB;CAEnB,GAAG;EACDC;EACA5G;EACAwF;EACAU;EACAD;EACAF;EACAb;EACA1C;EACAmD;EACAxC;EACAyC;EACAtC;EACAuC;CAA+B,CAChC;CAED,MAAMyC,cAAc7I,OAAsB,IAAI;CAC9C,MAAM8I,kBAAkB9I,OAA6C,IAAI;CACzE,MAAMiJ,qBAAqBjJ,OAAsBkJ,QAAQC,QAAQ,CAAC;CAClE,MAAMC,oBAAoBpJ,OAAiB,CAAA,CAAE;CAC7C,MAAMqJ,gBAAgBrJ,OAAO,KAAK;CAClC,MAAMsJ,mBAAmBtJ,OAAO,CAAC;CAEjC,MAAMuJ,0BAA0BnJ,aAC7BoJ,cAAsB;EACrB,MAAMC,cAAcR,mBAAmB/B,QAAQwC,KAAK,YAAY;GAC9D,IAAI,CAACvC,eAAewC,QAAQ;GAC5B,MAAMnJ,aAAWuF,WAAWmB,QAAQe,OAAOO,SAAS,CAAC,CAAChI;GACtD,MAAMO,UAAUP,WAASoJ,MACtBlE,WACCA,OAAKrE,OAAOmI,aACZ5J,yBAAmC8F,MAAI,CAAC,CAACjE,MACtCoI,iBACCpE,qBAAqBsC,MAAM8B,YAAY,MAAML,SACjD,CACJ;GACA,IAAI,CAACzI,SAAS;GAEd,MAAM+I,mBAAmBtJ,WACtBuJ,MAAM,GAAGvJ,WAASwJ,QAAQjJ,OAAO,CAAC,CAAC,CACnCF,QAAQjB,wBAAkC;GAC7C,IAAIsB,WAAW4I,iBAAiBG,GAAG,EAAE,IACjCxE,qBAAqBsC,MAAM+B,iBAAiBG,GAAG,EAAE,CAAE,IACnD;GACJ,MAAMC,sBAAsBpH,qBAC1B/B,SACAgC,aACF;GACA,MAAMoH,yBAAyB1G,wBAC7B1C,SACA2C,gBACF;GACA,MAAM0G,8BAA8BxG,6BAClC7C,SACA8C,qBACF;GAEA,KAAK,MAAMgG,kBAAgBjK,yBACzBmB,OACF,GAAG;IACD,MAAM2E,SAAO;KACXxE;KACAH,SAASkD,YACP4F,gBACAK,qBACAC,wBACAC,2BACF;IACF;IACA,MAAMC,UAAU5E,qBAAqBsC,MAAMrC,OAAK3E,OAAO;IACvD,MAAMuJ,YAAYtD,uBAAuBE,QAAQ/D,IAAIkH,OAAO;IAC5D,IAAIC,WAAW;KACb,MAAM9I,UAAUgE,cAAcC,sBAAsBC,MAAI;KACxD,IAAIlE,YAAY8I,UAAU9I,SACxB8I,UAAU3D,SAASjB,OAAK3E;UACnB;MACL,MAAMoG,cAAcwC,OAAOjE,QAAM2E,OAAO;MACxCrD,uBAAuBE,QAAQlC,IAAIqF,SAAS;OAC1C1D,QAAQjB,OAAK3E;OACbS;MACF,CAAC;KACH;IACF;IACAN,WAAWmJ;GACb;EACF,CAAC;EACDpB,mBAAmB/B,UAAUuC,YAAYc,YAAY,CAAC,CAAC;EACvD,OAAOd,YAAYc,OAAOlC,YAAU;GAClCrH,QAAQqH,MAAM,gCAAgCA,OAAK;EACrD,CAAC;CACH,GACA;EACElB;EACApB;EACAN;EACA1C;EACAW;EACAG;CAAqB,CAEzB;CAEA5D,gBAAgB;EACd,IAAI,CAACkH,eAAe;EACpB,MAAMqD,UAAUrD;EAEhB,MAAMsD,wCAAwB,IAAI9J,IAAY;EAC9C,MAAM+J,uBAAuB3J,cAC3BoB,kBAAkBpB,SAAO,IACrBnB,yBAAmCmB,SAAO,CAAC,CACxC6G,KAAKiC,mBAAiBpE,qBAAqBsC,MAAM8B,cAAY,CAAC,CAAC,CAC/Dc,QAAQN,cAAY,CAACrD,uBAAuBE,QAAQ/F,IAAIkJ,SAAO,CAAC,IACnE,CAAA;EAEN,MAAMO,cAAc7E,WAAWmB,QAAQe,OAAOQ,gBAAgB;GAC5D,MAAME,gBAAc5C,WAAWmB,QAAQe,OAAOO,SAAS;GACvD,MAAM,EAAEI,cAAcD;GACtB,MAAMkC,aAAaxB,cAAcnC;GACjCmC,cAAcnC,UAAU0B;GAGxB,IAAIC,YAAY3B,WAAW,MAAM;IAC/B,MAAM4D,UAAUnC,cAAYnI,SAASyJ,GAAG,EAAE;IAC1C,IAAIa,SAASC,SAAS,aAAa;KACjC,MAAMC,uBAAuBF,QAAQtJ,QAAQmJ,QAC1CM,MAAMA,EAAEtJ,SAAS,WACpB,CAAC,CAAC2B;KACF,OAAOgG,iBAAiBpC,UAAU8D,sBAAsB;MACtD5B,kBAAkBlC,QAAQgE,KAAKC,KAAKC,IAAI,IAAIvC,YAAY3B,OAAO;MAC/DoC,iBAAiBpC;KACnB;IACF;GACF;GAEA,IAAI0B,WAAW;IACb,IAAIC,YAAY3B,WAAW,MAAM;KAC/B2B,YAAY3B,UAAUiE,KAAKC,IAAI;KAC/BhC,kBAAkBlC,UAAU,CAAA;KAC5BoC,iBAAiBpC,UAAU;KAC3BsD,QAAQlC,MAAM;IAChB;IAEA,IAAIQ,gBAAgB5B,SAAS;KAC3BmE,aAAavC,gBAAgB5B,OAAO;KACpC4B,gBAAgB5B,UAAU;IAC5B;IACA;GACF;GAEA,IAAI2D,YAEEhC;QAAAA,YAAY3B,WAAW,MACzBkC,kBAAkBlC,QAAQgE,KAAKC,KAAKC,IAAI,IAAIvC,YAAY3B,OAAO;GAAA,OAE5D;IACL,MAAMoE,UAAU3C,cAAYnI,SAASK,QAAQ6J,mBAAmB;IAChE,IAAIY,QAAQC,OAAOlB,cAAYI,sBAAsBtJ,IAAIkJ,SAAO,CAAC,GAC/D;IACF,KAAK,MAAMA,aAAWiB,SAASb,sBAAsBrJ,IAAIiJ,SAAO;GAClE;GAGA,IAAIvB,gBAAgB5B,SAASmE,aAAavC,gBAAgB5B,OAAO;GACjE4B,gBAAgB5B,UAAU8B,iBAAiB;IACzCwC,eAAe,KAAK;GACtB,GAAG,CAAC;EACN,CAAC;EAED,MAAMC,qBAAqB1F,WAAWmB,QAAQe,OAAOO,SAAS;EAC9Da,cAAcnC,UAAUuE,mBAAmB7C;EAC3C,IAAI6C,mBAAmB7C,WACjBC;OAAAA,YAAY3B,WAAW,MAAM;IAC/B2B,YAAY3B,UAAUiE,KAAKC,IAAI;IAC/BhC,kBAAkBlC,UAAU,CAAA;IAC5BoC,iBAAiBpC,UAAU;IAC3BsD,QAAQlC,MAAM;GAChB;SACK,IAAImD,mBAAmBjL,SAAS8C,SAAS,GAC9CkI,eAAe,IAAI;EAGrB,SAASA,eAAeE,eAAwB;GAC9C5C,gBAAgB5B,UAAU;GAC1B,MAAMyE,SAAS5F,WAAWmB,QAAQe,OAAOO,SAAS;GAClD,IAAI,CAACkD,iBAAiBC,OAAO/C,WAAW;GAExC,MAAMgD,aAAaxC,kBAAkBlC;GACrC,MAAM2E,aAAaD,WAAWtI,SAAS,IAAIsI,WAAW3B,GAAG,EAAE,IAAInI,KAAAA;GAG/D,IAAI8J,WAAWtI,WAAW,KAAKuI,cAAc,MAAM;IACjD,MAAMC,gBAAgBH,OAAOnL,SAASuL,UACnCjL,QAAMA,IAAEiK,SAAS,WACpB;IACA,IAAIe,eAAe;KACjB,MAAME,UAAUF,cAActK,QAAQmJ,QACnCM,QAAMA,IAAEtJ,SAAS,WACpB,CAAC,CAAC2B;KACF,IAAI0I,UAAU,GAAG;MACf,MAAMC,aAAaD,UAAU;MAC7B,MAAME,UAAUL,aAAaI;MAC7BL,WAAWtI,SAAS;MACpB,KAAK,IAAI6I,IAAI,GAAGA,IAAIF,YAAYE,KAC9BP,WAAWV,KAAKkB,KAAKC,OAAOF,IAAI,KAAKD,OAAO,CAAC;KAEjD;IACF;GACF;GAGA,MAAMI,iBACJV,WAAWtI,SAAS,IAChBsI,WAAWhE,KAAK2E,OAAOJ,SAAO;IAC5BK,UAAUL,QAAM,IAAI,IAAIP,WAAWO,MAAI;IACvCM,QAAQF;GACV,EAAE,IACFzK,KAAAA;GAEN+G,YAAY3B,UAAU;GACtBkC,kBAAkBlC,UAAU,CAAA;GAE5B,MAAMwF,mBAAmB;IACvB,GAAIb,cAAc,OAAO,EAAEA,WAAW,IAAI/J,KAAAA;IAC1C,GAAIwK,kBAAkB,OAAO,EAAEA,eAAe,IAAIxK,KAAAA;GACpD;GAEAmH,mBAAmB/B,UAAU+B,mBAAmB/B,QAC7CwC,KAAK,YAAY;IAChB,MAAM,EAAElJ,UAAAA,eAAamL;IACrB,IAAIgB,qBAAoC;IAExC,MAAMC,kBAAkBC,SACtBA,KAAKvJ,SAAS,IAAImC,qBAAqBsC,MAAM8E,KAAK5C,GAAG,EAAE,CAAE,IAAI;IAE/D,MAAM6C,gBACJD,QACA9J,iBACAW,oBAGAG,4BAIAgJ,OAAKjF,KAAKmF,KAAKC,SAAS;KACtB9L,UACE8L,QAAQ,IACJL,qBACAlH,qBAAqBsC,MAAM8E,OAAKG,MAAM,EAAG;KAC/CjM,SAASkD,YACP8I,KACAhK,iBACAW,oBACAG,uBACF;IACF,EAAE;IAEJ,KAAK,MAAM9C,aAAWP,YAAU;KAC9B,MAAMyM,gBAAgBrN,yBAAmCmB,SAAO;KAEhE,MAAMmM,aAAa/K,kBAAkBpB,SAAO;KAC5C,MAAMoM,sBAAsBnL,uBAAuBjB,SAAO;KAM1D,IAAI,EAHFmM,cACCC,uBAAuB3C,QAAQb,WAAW7H,KAAAA,IAE/B;MACZ6K,qBACEC,eAAeK,aAAa,KAAKN;MACnC;KACF;KAEA,IAAI,CAAC7F,WAAWI,QAAQ/F,IAAIJ,UAAQM,EAAE,GAAG;MACvCyF,WAAWI,QAAQ9F,IAAIL,UAAQM,EAAE;MACjC0F,qBAAqBG,QAAQ9F,IAAIL,UAAQM,EAAE;KAC7C;KAEA,MAAMgM,aAAaP,aACjBG,eACAnK,qBAAqB/B,WAASgC,aAAa,GAC3CU,wBAAwB1C,WAAS2C,gBAAgB,GACjDE,6BAA6B7C,WAAS8C,qBAAqB,CAC7D;KACA,KAAK,MAAM6B,UAAQ2H,YAAY;MAC7B,MAAMhD,YAAU5E,qBAAqBsC,MAAMrC,OAAK3E,OAAO;MACvD,MAAMuJ,cAAYtD,uBAAuBE,QAAQ/D,IAAIkH,SAAO;MAC5D,IAAI,CAACC,aAAW;OACd,MAAME,QAAQ8C,OAAO5H,MAAI;OACzBsB,uBAAuBE,QAAQlC,IAAIqF,WAAS;QAC1C1D,QAAQjB,OAAK3E;QACbS,SAASgE,cAAcC,sBAAsBC,MAAI;OACnD,CAAC;MACH,OAAO,IACL4E,YAAU3D,WAAWjB,OAAK3E,WAC1B8K,eAAe/J,KAAAA,KACf0I,QAAQb,QACR;OACA,MAAMnI,YAAUgE,cAAcC,sBAAsBC,MAAI;OACxD,IAAIlE,cAAY8I,YAAU9I,SACxB8I,YAAU3D,SAASjB,OAAK3E;YAExB,IAAI;QACF,MAAMyJ,QAAQb,OAAOjE,QAAM2E,SAAO;QAClCrD,uBAAuBE,QAAQlC,IAAIqF,WAAS;SAC1C1D,QAAQjB,OAAK3E;SACbS,SAAAA;QACF,CAAC;OACH,QAAQ,CACN;MAGN;KACF;KAEAmL,qBACEC,eAAeK,aAAa,KAAKN;KAEnC,IAAI5F,qBAAqBG,QAAQ/F,IAAIJ,UAAQM,EAAE,KAAK6L,YAAY;MAC9DnG,qBAAqBG,QAAQqG,OAAOxM,UAAQM,EAAE;MAC9CmJ,QAAQgD,kBACNH,WAAWzF,KAAKlC,QAAMoC,aAAW;OAC/B,GAAGpC;OACH3E,SAASkM,cAAcnF;MACzB,EAAE,GACF;OACE,GAAG4E;OACH3L,SAAAA;MACF,CACF;KACF;IACF;GACF,CAAC,CAAC,CACDwJ,OAAOlC,YAAU;IAChBrH,QAAQqH,MAAM,sCAAsCA,OAAK;GAC3D,CAAC;EACL;EAEA,aAAa;GACXuC,YAAY;GACZ,IAAI9B,gBAAgB5B,SAAS;IAC3BmE,aAAavC,gBAAgB5B,OAAO;IACpC4B,gBAAgB5B,UAAU;IAC1BsE,eAAe,KAAK;GACtB;EACF;CACF,GAAG;EACDrE;EACA1B;EACAM;EACAhD;EACAW;EACAG;CAAqB,CACtB;CA2CD,OAAO;EACLyD;EACAmG,eA3CoBrN,YACpB,OAAOoJ,gBAAsB;GAC3B,MAAMkE,iBAAiBvG,eAAeoG,QAAQI,KAAKxG,aAAa;GAChE,IAAI,CAACuG,gBAAgB;GAErB,MAAMlN,aAAWuF,WAAWmB,QAAQe,OAAOO,SAAS,CAAC,CAAChI;GACtD,MAAMoN,eAAepN,WAASqN,WAAW/M,QAAMA,IAAEO,OAAOmI,WAAS;GACjE,IAAIoE,iBAAiB,IAAI;GAEzB,MAAME,wBAAwBtN,WAC3BuJ,MAAM,GAAG6D,YAAY,CAAC,CACtB/M,QAAQjB,wBAAkC;GAC7C,IAAIsB,aAAW4M,sBAAsB7D,GAAG,EAAE,IACtCxE,qBAAqBsC,MAAM+F,sBAAsB7D,GAAG,EAAE,CAAE,IACxD;GACJ,MAAM8D,gBAAgBnO,yBACpBY,WAASoN,aACX,CAAC,CAAChG,KAAK7G,cAAY;IACjB,MAAM2E,SAAO;KAAExE,UAAAA;KAAUH,SAAAA;IAAQ;IACjCG,aAAWuE,qBAAqBsC,MAAMhH,SAAO;IAC7C,OAAO2E;GACT,CAAC;GAED,MAAMsI,WAAW/E,mBAAmB/B,QAAQwC,KAAK,YAAY;IAC3D,MAAMgE,eAAeK,aAAa;IAElCjH,WAAWI,QAAQqG,OAAO/D,WAAS;IACnCzC,qBAAqBG,QAAQqG,OAAO/D,WAAS;IAC7C,KAAK,MAAM9D,UAAQqI,eACjB/G,uBAAuBE,QAAQqG,OAC7B9H,qBAAqBsC,MAAMrC,OAAK3E,OAAO,CACzC;GAEJ,CAAC;GAEDkI,mBAAmB/B,UAAU8G,SAASzD,YAAY,CAAC,CAAC;GACpD,MAAMyD;EACR,GACA;GAAC7G;GAAepB;GAAYN;EAAoB,CAKhDgI;EACAlE;EACA0E,8BAA8B1E;CAChC;AACF"}
+{"version":3,"file":"useExternalHistory.js","names":["AssistantRuntime","ThreadHistoryAdapter","ThreadMessage","MessageFormatAdapter","MessageFormatItem","MessageFormatRepository","ExportedMessageRepository","RespondToToolApprovalOptions","Unstable_ToolInteractionLog","getExternalStoreMessages","MessageRepository","readToolInteractionLog","useAui","useRef","useEffect","useState","RefObject","useCallback","useMemo","normalizeToolApprovalAnswers","toExportedMessageRepository","toThreadMessages","messages","TMessage","survivingIds","Set","survivors","flatMap","m","message","console","warn","parentId","has","add","id","headId","hasUnansweredApproval","content","some","part","type","approval","approved","undefined","resolution","isAwaitingToolApproval","status","reason","isTerminalMessage","TOOL_ARTIFACTS_METADATA_KEY","TOOL_INTERACTIONS_METADATA_KEY","TOOL_APPROVAL_RESPONSES_METADATA_KEY","StoredToolApprovalResponse","Omit","isRecord","value","Record","Array","isArray","collectToolArtifacts","toolArtifacts","ReadonlyMap","entries","artifact","get","toolCallId","const","length","Object","fromEntries","collectToolInteractions","toolInteractions","interactions","collectToolApprovalResponses","toolApprovalResponses","response","optionId","text","answers","addToolData","parts","toolCallIds","artifacts","hasOwn","approvalIds","approvalId","approvalResponses","hasArtifacts","keys","hasInteractions","hasApprovalResponses","metadata","restoreToolData","Map","set","log","_","__","___","restMetadata","_metadata","restMessage","encodeContent","storageFormatAdapter","item","JSON","stringify","encode","useExternalHistory","runtimeRef","historyAdapter","onSetMessages","onToolArtifactsRestored","onToolInteractionsRestored","onToolApprovalResponsesRestored","loadedRef","itemEpoch","setItemEpoch","aui","optionalThreadListItem","threadListItem","source","hasLoaded","setHasLoaded","historyIds","deferredTelemetryIds","persistedInnerMessages","onSetMessagesRef","current","formatAdapter","withFormat","Error","isLoading","loadHistory","repo","load","clear","restoredMessages","map","restoredRepo","index","getId","converted","thread","import","tempRepo","getMessages","error","pin","remoteId","getState","subscribe","n","threadState","isRunning","runStartRef","persistTimerRef","ReturnType","setTimeout","persistInFlightRef","Promise","resolve","stepBoundariesRef","wasRunningRef","toolCallCountRef","persistToolInteractions","messageId","persistence","then","update","find","innerMessage","previousMessages","slice","indexOf","at","storedToolArtifacts","storedToolInteractions","storedToolApprovalResponses","innerId","persisted","catch","adapter","idleScheduledInnerIds","unpersistedInnerIds","filter","unsubscribe","wasRunning","lastMsg","role","currentToolCallCount","p","push","Date","now","clearTimeout","pending","every","persistSettled","initialThreadState","ignoreRunning","latest","boundaries","durationMs","lastAssistant","findLast","tcCount","totalSteps","stepDur","i","Math","round","stepTimestamps","endMs","start_ms","end_ms","telemetryOptions","lastInnerMessageId","getLastInnerId","msgs","toBatchItems","msg","idx","innerMessages","isTerminal","isAwaitingToolCalls","isReady","batchItems","append","delete","reportTelemetry","deleteMessage","deleteMessages","bind","messageIndex","findIndex","previousInnerMessages","itemsToDelete","deletion","persistToolApprovalResponses"],"sources":["../../src/runtime/useExternalHistory.ts"],"sourcesContent":["\"use client\";\n\nimport type {\n  AssistantRuntime,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatItem,\n  MessageFormatRepository,\n  ExportedMessageRepository,\n  RespondToToolApprovalOptions,\n  Unstable_ToolInteractionLog,\n} from \"@assistant-ui/core\";\nimport { getExternalStoreMessages } from \"@assistant-ui/core\";\nimport {\n  MessageRepository,\n  readToolInteractionLog,\n} from \"@assistant-ui/core/internal\";\nimport { useAui } from \"@assistant-ui/store\";\nimport {\n  useRef,\n  useEffect,\n  useState,\n  type RefObject,\n  useCallback,\n  useMemo,\n} from \"react\";\nimport { normalizeToolApprovalAnswers } from \"../converters/toolApprovalAnswers\";\n\nexport const toExportedMessageRepository = <TMessage>(\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  messages: MessageFormatRepository<TMessage>,\n): ExportedMessageRepository => {\n  const survivingIds = new Set<string>();\n  const survivors = messages.messages.flatMap((m) => {\n    const message = toThreadMessages([m.message])[0];\n    if (!message) {\n      console.warn(\"Skipping a stored message that could not be loaded.\");\n      return [];\n    }\n    if (m.parentId && !survivingIds.has(m.parentId)) return [];\n    survivingIds.add(message.id);\n    return [{ ...m, message }];\n  });\n\n  return {\n    headId:\n      messages.headId && survivingIds.has(messages.headId)\n        ? messages.headId\n        : null,\n    messages: survivors,\n  };\n};\n\nconst hasUnansweredApproval = (message: ThreadMessage) =>\n  message.content.some(\n    (part) =>\n      part.type === \"tool-call\" &&\n      part.approval != null &&\n      part.approval.approved === undefined &&\n      part.approval.resolution === undefined,\n  );\n\n// Core reports a run paused on an unanswered approval as an interrupt, not as tool calls.\nconst isAwaitingToolApproval = (message: ThreadMessage) =>\n  message.status?.type === \"requires-action\" &&\n  (message.status.reason === \"tool-calls\" ||\n    (message.status.reason === \"interrupt\" && hasUnansweredApproval(message)));\n\nconst isTerminalMessage = (message: ThreadMessage) =>\n  message.status === undefined ||\n  message.status.type === \"complete\" ||\n  message.status.type === \"incomplete\";\n\nconst TOOL_ARTIFACTS_METADATA_KEY = \"__aui_toolArtifacts\";\nconst TOOL_INTERACTIONS_METADATA_KEY = \"__aui_toolInteractions\";\nconst TOOL_APPROVAL_RESPONSES_METADATA_KEY = \"__aui_toolApprovalResponses\";\n\ntype StoredToolApprovalResponse = Omit<\n  RespondToToolApprovalOptions,\n  \"approvalId\"\n>;\n\nconst isRecord = (value: unknown): value is Record<string, unknown> =>\n  value !== null && typeof value === \"object\" && !Array.isArray(value);\n\nconst collectToolArtifacts = (\n  message: ThreadMessage,\n  toolArtifacts: ReadonlyMap<string, unknown> | undefined,\n) => {\n  if (!toolArtifacts) return undefined;\n  const entries = message.content.flatMap((part) => {\n    if (part.type !== \"tool-call\") return [];\n    const artifact = toolArtifacts.get(part.toolCallId);\n    return artifact === undefined ? [] : [[part.toolCallId, artifact] as const];\n  });\n  return entries.length > 0 ? Object.fromEntries(entries) : undefined;\n};\n\nconst collectToolInteractions = (\n  message: ThreadMessage,\n  toolInteractions:\n    | ReadonlyMap<string, Unstable_ToolInteractionLog>\n    | undefined,\n) => {\n  if (!toolInteractions) return undefined;\n  const entries = message.content.flatMap((part) => {\n    if (part.type !== \"tool-call\") return [];\n    const interactions = toolInteractions.get(part.toolCallId);\n    return interactions === undefined\n      ? []\n      : [[part.toolCallId, interactions] as const];\n  });\n  return entries.length > 0 ? Object.fromEntries(entries) : undefined;\n};\n\nconst collectToolApprovalResponses = (\n  message: ThreadMessage,\n  toolApprovalResponses:\n    | ReadonlyMap<string, RespondToToolApprovalOptions>\n    | undefined,\n) => {\n  if (!toolApprovalResponses) return undefined;\n  const entries = message.content.flatMap((part) => {\n    if (part.type !== \"tool-call\" || !part.approval) return [];\n    const response = toolApprovalResponses.get(part.approval.id);\n    if (!response) return [];\n    return [\n      [\n        part.approval.id,\n        {\n          approved: response.approved,\n          ...(response.optionId != null && { optionId: response.optionId }),\n          ...(response.text != null && { text: response.text }),\n          ...(response.answers != null && { answers: response.answers }),\n          ...(response.reason != null && { reason: response.reason }),\n        },\n      ] as const,\n    ];\n  });\n  return entries.length > 0 ? Object.fromEntries(entries) : undefined;\n};\n\nconst addToolData = <TMessage>(\n  message: TMessage,\n  toolArtifacts: Record<string, unknown> | undefined,\n  toolInteractions: Record<string, Unstable_ToolInteractionLog> | undefined,\n  toolApprovalResponses: Record<string, StoredToolApprovalResponse> | undefined,\n): TMessage => {\n  if (\n    (!toolArtifacts && !toolInteractions && !toolApprovalResponses) ||\n    !isRecord(message) ||\n    !Array.isArray(message.parts)\n  )\n    return message;\n  const toolCallIds = message.parts.flatMap((part) => {\n    if (!isRecord(part) || typeof part.toolCallId !== \"string\") return [];\n    return [part.toolCallId];\n  });\n  const artifacts = toolArtifacts\n    ? Object.fromEntries(\n        toolCallIds.flatMap((toolCallId) =>\n          Object.hasOwn(toolArtifacts, toolCallId)\n            ? [[toolCallId, toolArtifacts[toolCallId]] as const]\n            : [],\n        ),\n      )\n    : undefined;\n  const interactions = toolInteractions\n    ? Object.fromEntries(\n        toolCallIds.flatMap((toolCallId) =>\n          Object.hasOwn(toolInteractions, toolCallId)\n            ? [[toolCallId, toolInteractions[toolCallId]] as const]\n            : [],\n        ),\n      )\n    : undefined;\n  const approvalIds = message.parts.flatMap((part) => {\n    if (!isRecord(part) || !isRecord(part.approval)) return [];\n    const approvalId = part.approval.id;\n    return typeof approvalId === \"string\" ? [approvalId] : [];\n  });\n  const approvalResponses = toolApprovalResponses\n    ? Object.fromEntries(\n        approvalIds.flatMap((approvalId) =>\n          Object.hasOwn(toolApprovalResponses, approvalId)\n            ? [[approvalId, toolApprovalResponses[approvalId]] as const]\n            : [],\n        ),\n      )\n    : undefined;\n  const hasArtifacts = !!artifacts && Object.keys(artifacts).length > 0;\n  const hasInteractions =\n    !!interactions && Object.keys(interactions).length > 0;\n  const hasApprovalResponses =\n    !!approvalResponses && Object.keys(approvalResponses).length > 0;\n  if (!hasArtifacts && !hasInteractions && !hasApprovalResponses)\n    return message;\n  const metadata = isRecord(message.metadata) ? message.metadata : {};\n  return {\n    ...message,\n    metadata: {\n      ...metadata,\n      ...(hasArtifacts && { [TOOL_ARTIFACTS_METADATA_KEY]: artifacts }),\n      ...(hasInteractions && {\n        [TOOL_INTERACTIONS_METADATA_KEY]: interactions,\n      }),\n      ...(hasApprovalResponses && {\n        [TOOL_APPROVAL_RESPONSES_METADATA_KEY]: approvalResponses,\n      }),\n    },\n  } as TMessage;\n};\n\nconst restoreToolData = <TMessage>(\n  message: TMessage,\n  toolArtifacts: Map<string, unknown> | undefined,\n  toolInteractions: Map<string, Unstable_ToolInteractionLog> | undefined,\n  toolApprovalResponses: Map<string, RespondToToolApprovalOptions> | undefined,\n): TMessage => {\n  if (!isRecord(message) || !isRecord(message.metadata)) return message;\n  const metadata = message.metadata;\n  const hasArtifacts = Object.hasOwn(metadata, TOOL_ARTIFACTS_METADATA_KEY);\n  const hasInteractions = Object.hasOwn(\n    metadata,\n    TOOL_INTERACTIONS_METADATA_KEY,\n  );\n  const hasApprovalResponses = Object.hasOwn(\n    metadata,\n    TOOL_APPROVAL_RESPONSES_METADATA_KEY,\n  );\n  if (!hasArtifacts && !hasInteractions && !hasApprovalResponses)\n    return message;\n  const artifacts = metadata[TOOL_ARTIFACTS_METADATA_KEY];\n  if (toolArtifacts && isRecord(artifacts)) {\n    for (const [toolCallId, artifact] of Object.entries(artifacts)) {\n      toolArtifacts.set(toolCallId, artifact);\n    }\n  }\n  const interactions = metadata[TOOL_INTERACTIONS_METADATA_KEY];\n  if (toolInteractions && isRecord(interactions)) {\n    for (const [toolCallId, value] of Object.entries(interactions)) {\n      const log = readToolInteractionLog(value);\n      if (log) toolInteractions.set(toolCallId, log);\n    }\n  }\n  const approvalResponses = metadata[TOOL_APPROVAL_RESPONSES_METADATA_KEY];\n  if (toolApprovalResponses && isRecord(approvalResponses)) {\n    for (const [approvalId, value] of Object.entries(approvalResponses)) {\n      if (!isRecord(value) || typeof value.approved !== \"boolean\") continue;\n      const answers = normalizeToolApprovalAnswers(value.answers);\n      toolApprovalResponses.set(approvalId, {\n        approvalId,\n        approved: value.approved,\n        ...(typeof value.optionId === \"string\" && {\n          optionId: value.optionId,\n        }),\n        ...(typeof value.text === \"string\" && { text: value.text }),\n        ...(answers && Object.keys(answers).length > 0 && { answers }),\n        ...(typeof value.reason === \"string\" && { reason: value.reason }),\n      });\n    }\n  }\n  const {\n    [TOOL_ARTIFACTS_METADATA_KEY]: _,\n    [TOOL_INTERACTIONS_METADATA_KEY]: __,\n    [TOOL_APPROVAL_RESPONSES_METADATA_KEY]: ___,\n    ...restMetadata\n  } = metadata;\n  const { metadata: _metadata, ...restMessage } = message;\n  return (\n    Object.keys(restMetadata).length === 0\n      ? restMessage\n      : { ...restMessage, metadata: restMetadata }\n  ) as TMessage;\n};\n\nconst encodeContent = <TMessage>(\n  storageFormatAdapter: MessageFormatAdapter<TMessage, any>,\n  item: MessageFormatItem<TMessage>,\n) => JSON.stringify(storageFormatAdapter.encode(item));\n\nexport const useExternalHistory = <TMessage>(\n  runtimeRef: RefObject<AssistantRuntime>,\n  historyAdapter: ThreadHistoryAdapter | undefined,\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  storageFormatAdapter: MessageFormatAdapter<TMessage, any>,\n  onSetMessages: (messages: TMessage[]) => void,\n  toolArtifacts?: Map<string, unknown>,\n  onToolArtifactsRestored?: () => void,\n  toolInteractions?: Map<string, Unstable_ToolInteractionLog>,\n  onToolInteractionsRestored?: () => void,\n  toolApprovalResponses?: Map<string, RespondToToolApprovalOptions>,\n  onToolApprovalResponsesRestored?: () => void,\n) => {\n  const loadedRef = useRef(false);\n  const [itemEpoch, setItemEpoch] = useState(0);\n\n  const aui = useAui();\n  const optionalThreadListItem = useCallback(\n    () => (aui.threadListItem.source ? aui.threadListItem : null),\n    [aui],\n  );\n\n  const [hasLoaded, setHasLoaded] = useState(false);\n\n  const historyIds = useRef(new Set<string>());\n  const deferredTelemetryIds = useRef(new Set<string>());\n  // `content` is a snapshot taken at write time rather than a re-encode of `source`, because a retained message object can be mutated in place by the runtime that produced it.\n  const persistedInnerMessages = useRef(\n    new Map<string, { source: TMessage; content: string }>(),\n  );\n\n  const onSetMessagesRef = useRef(onSetMessages);\n  useEffect(() => {\n    onSetMessagesRef.current = onSetMessages;\n  });\n\n  const formatAdapter = useMemo(() => {\n    if (!historyAdapter) return undefined;\n    if (!historyAdapter.withFormat) {\n      throw new Error(\n        \"useAISDKRuntime: ThreadHistoryAdapter is missing the required `withFormat` method.\",\n      );\n    }\n    return historyAdapter.withFormat<TMessage, any>(storageFormatAdapter);\n  }, [historyAdapter, storageFormatAdapter]);\n\n  const isLoading = formatAdapter != null && !hasLoaded;\n\n  useEffect(() => {\n    if (!formatAdapter || loadedRef.current) return undefined;\n\n    const loadHistory = async () => {\n      try {\n        const repo = await formatAdapter.load();\n        toolArtifacts?.clear();\n        toolInteractions?.clear();\n        toolApprovalResponses?.clear();\n        const restoredMessages =\n          repo?.messages.map((item) => ({\n            ...item,\n            message: restoreToolData(\n              item.message,\n              toolArtifacts,\n              toolInteractions,\n              toolApprovalResponses,\n            ),\n          })) ?? [];\n        onToolArtifactsRestored?.();\n        onToolInteractionsRestored?.();\n        onToolApprovalResponsesRestored?.();\n        if (repo && restoredMessages.length > 0) {\n          const restoredRepo = { ...repo, messages: restoredMessages };\n          for (const [index, m] of repo.messages.entries()) {\n            persistedInnerMessages.current.set(\n              storageFormatAdapter.getId(m.message),\n              {\n                source: restoredMessages[index]!.message,\n                content: encodeContent(storageFormatAdapter, m),\n              },\n            );\n          }\n          const converted = toExportedMessageRepository(\n            toThreadMessages,\n            restoredRepo,\n          );\n          runtimeRef.current.thread.import(converted);\n\n          const tempRepo = new MessageRepository();\n          tempRepo.import(converted);\n          const messages = tempRepo.getMessages();\n\n          onSetMessagesRef.current(\n            messages.flatMap(getExternalStoreMessages<TMessage>),\n          );\n\n          historyIds.current = new Set();\n          for (const m of converted.messages) {\n            historyIds.current.add(m.message.id);\n            if (isAwaitingToolApproval(m.message)) {\n              deferredTelemetryIds.current.add(m.message.id);\n            }\n          }\n        }\n      } catch (error) {\n        console.error(\"Failed to load message history:\", error);\n      } finally {\n        setHasLoaded(true);\n      }\n    };\n\n    formatAdapter.pin?.();\n\n    const remoteId = optionalThreadListItem()?.getState().remoteId;\n    if (!remoteId) {\n      // History loads asynchronously against the thread list item; without a\n      // remote id there is nothing to await, so the flag settles here.\n      // eslint-disable-next-line react-hooks/set-state-in-effect\n      setHasLoaded(true);\n      return aui.subscribe(() => {\n        if (optionalThreadListItem()?.getState().remoteId) {\n          setItemEpoch((n) => n + 1);\n        }\n      });\n    }\n\n    const threadState = runtimeRef.current.thread.getState();\n    if (threadState.isRunning || threadState.messages.length > 0) {\n      loadedRef.current = true;\n      setHasLoaded(true);\n      return undefined;\n    }\n\n    loadedRef.current = true;\n    void loadHistory();\n    return undefined;\n  }, [\n    formatAdapter,\n    toThreadMessages,\n    runtimeRef,\n    optionalThreadListItem,\n    aui,\n    itemEpoch,\n    storageFormatAdapter,\n    toolArtifacts,\n    onToolArtifactsRestored,\n    toolInteractions,\n    onToolInteractionsRestored,\n    toolApprovalResponses,\n    onToolApprovalResponsesRestored,\n  ]);\n\n  const runStartRef = useRef<number | null>(null);\n  const persistTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);\n  const persistInFlightRef = useRef<Promise<void>>(Promise.resolve());\n  const stepBoundariesRef = useRef<number[]>([]);\n  const wasRunningRef = useRef(false);\n  const toolCallCountRef = useRef(0);\n\n  const persistToolInteractions = useCallback(\n    (messageId: string) => {\n      const persistence = persistInFlightRef.current.then(async () => {\n        if (!formatAdapter?.update) return;\n        const messages = runtimeRef.current.thread.getState().messages;\n        const message = messages.find(\n          (item) =>\n            item.id === messageId ||\n            getExternalStoreMessages<TMessage>(item).some(\n              (innerMessage) =>\n                storageFormatAdapter.getId(innerMessage) === messageId,\n            ),\n        );\n        if (!message) return;\n\n        const previousMessages = messages\n          .slice(0, messages.indexOf(message))\n          .flatMap(getExternalStoreMessages<TMessage>);\n        let parentId = previousMessages.at(-1)\n          ? storageFormatAdapter.getId(previousMessages.at(-1)!)\n          : null;\n        const storedToolArtifacts = collectToolArtifacts(\n          message,\n          toolArtifacts,\n        );\n        const storedToolInteractions = collectToolInteractions(\n          message,\n          toolInteractions,\n        );\n        const storedToolApprovalResponses = collectToolApprovalResponses(\n          message,\n          toolApprovalResponses,\n        );\n\n        for (const innerMessage of getExternalStoreMessages<TMessage>(\n          message,\n        )) {\n          const item = {\n            parentId,\n            message: addToolData(\n              innerMessage,\n              storedToolArtifacts,\n              storedToolInteractions,\n              storedToolApprovalResponses,\n            ),\n          };\n          const innerId = storageFormatAdapter.getId(item.message);\n          const persisted = persistedInnerMessages.current.get(innerId);\n          if (persisted) {\n            const content = encodeContent(storageFormatAdapter, item);\n            if (content === persisted.content) {\n              persisted.source = item.message;\n            } else {\n              await formatAdapter.update(item, innerId);\n              persistedInnerMessages.current.set(innerId, {\n                source: item.message,\n                content,\n              });\n            }\n          }\n          parentId = innerId;\n        }\n      });\n      persistInFlightRef.current = persistence.catch(() => {});\n      return persistence.catch((error) => {\n        console.error(\"Failed to persist tool data:\", error);\n      });\n    },\n    [\n      formatAdapter,\n      runtimeRef,\n      storageFormatAdapter,\n      toolArtifacts,\n      toolInteractions,\n      toolApprovalResponses,\n    ],\n  );\n\n  useEffect(() => {\n    if (!formatAdapter) return;\n    const adapter = formatAdapter;\n\n    const idleScheduledInnerIds = new Set<string>();\n    const unpersistedInnerIds = (message: ThreadMessage) =>\n      isTerminalMessage(message)\n        ? getExternalStoreMessages<TMessage>(message)\n            .map((innerMessage) => storageFormatAdapter.getId(innerMessage))\n            .filter((innerId) => !persistedInnerMessages.current.has(innerId))\n        : [];\n\n    const unsubscribe = runtimeRef.current.thread.subscribe(() => {\n      const threadState = runtimeRef.current.thread.getState();\n      const { isRunning } = threadState;\n      const wasRunning = wasRunningRef.current;\n      wasRunningRef.current = isRunning;\n\n      // Track step boundaries by content changes (more reliable than isRunning)\n      if (runStartRef.current != null) {\n        const lastMsg = threadState.messages.at(-1);\n        if (lastMsg?.role === \"assistant\") {\n          const currentToolCallCount = lastMsg.content.filter(\n            (p) => p.type === \"tool-call\",\n          ).length;\n          while (toolCallCountRef.current < currentToolCallCount) {\n            stepBoundariesRef.current.push(Date.now() - runStartRef.current);\n            toolCallCountRef.current++;\n          }\n        }\n      }\n\n      if (isRunning) {\n        if (runStartRef.current == null) {\n          runStartRef.current = Date.now();\n          stepBoundariesRef.current = [];\n          toolCallCountRef.current = 0;\n          adapter.pin?.();\n        }\n        // Cancel any pending persist — isRunning went back to true\n        if (persistTimerRef.current) {\n          clearTimeout(persistTimerRef.current);\n          persistTimerRef.current = null;\n        }\n        return;\n      }\n\n      if (wasRunning) {\n        // Record step boundary offset (synchronous for accuracy)\n        if (runStartRef.current != null) {\n          stepBoundariesRef.current.push(Date.now() - runStartRef.current);\n        }\n      } else {\n        const pending = threadState.messages.flatMap(unpersistedInnerIds);\n        if (pending.every((innerId) => idleScheduledInnerIds.has(innerId)))\n          return;\n        for (const innerId of pending) idleScheduledInnerIds.add(innerId);\n      }\n\n      // Debounce: wait one macrotask so agentic step flickers are absorbed\n      if (persistTimerRef.current) clearTimeout(persistTimerRef.current);\n      persistTimerRef.current = setTimeout(() => {\n        persistSettled(false);\n      }, 0);\n    });\n\n    const initialThreadState = runtimeRef.current.thread.getState();\n    wasRunningRef.current = initialThreadState.isRunning;\n    if (initialThreadState.isRunning) {\n      if (runStartRef.current == null) {\n        runStartRef.current = Date.now();\n        stepBoundariesRef.current = [];\n        toolCallCountRef.current = 0;\n        adapter.pin?.();\n      }\n    } else if (initialThreadState.messages.length > 0) {\n      persistSettled(true);\n    }\n\n    function persistSettled(ignoreRunning: boolean) {\n      persistTimerRef.current = null;\n      const latest = runtimeRef.current.thread.getState();\n      if (!ignoreRunning && latest.isRunning) return;\n\n      const boundaries = stepBoundariesRef.current;\n      const durationMs = boundaries.length > 0 ? boundaries.at(-1) : undefined;\n\n      // Fallback: if only 1 boundary but message has multiple steps, distribute evenly\n      if (boundaries.length === 1 && durationMs != null) {\n        const lastAssistant = latest.messages.findLast(\n          (m) => m.role === \"assistant\",\n        );\n        if (lastAssistant) {\n          const tcCount = lastAssistant.content.filter(\n            (p) => p.type === \"tool-call\",\n          ).length;\n          if (tcCount > 0) {\n            const totalSteps = tcCount + 1;\n            const stepDur = durationMs / totalSteps;\n            boundaries.length = 0;\n            for (let i = 0; i < totalSteps; i++) {\n              boundaries.push(Math.round((i + 1) * stepDur));\n            }\n          }\n        }\n      }\n\n      // Build per-step timestamps when there are multiple steps\n      const stepTimestamps =\n        boundaries.length > 1\n          ? boundaries.map((endMs, i) => ({\n              start_ms: i === 0 ? 0 : boundaries[i - 1]!,\n              end_ms: endMs,\n            }))\n          : undefined;\n\n      runStartRef.current = null;\n      stepBoundariesRef.current = [];\n\n      const telemetryOptions = {\n        ...(durationMs != null ? { durationMs } : undefined),\n        ...(stepTimestamps != null ? { stepTimestamps } : undefined),\n      };\n\n      persistInFlightRef.current = persistInFlightRef.current\n        .then(async () => {\n          const { messages } = latest;\n          let lastInnerMessageId: string | null = null;\n\n          const getLastInnerId = (msgs: TMessage[]): string | null =>\n            msgs.length > 0 ? storageFormatAdapter.getId(msgs.at(-1)!) : null;\n\n          const toBatchItems = (\n            msgs: TMessage[],\n            toolArtifacts: Record<string, unknown> | undefined,\n            toolInteractions:\n              | Record<string, Unstable_ToolInteractionLog>\n              | undefined,\n            toolApprovalResponses:\n              | Record<string, StoredToolApprovalResponse>\n              | undefined,\n          ) =>\n            msgs.map((msg, idx) => ({\n              parentId:\n                idx === 0\n                  ? lastInnerMessageId\n                  : storageFormatAdapter.getId(msgs[idx - 1]!),\n              message: addToolData(\n                msg,\n                toolArtifacts,\n                toolInteractions,\n                toolApprovalResponses,\n              ),\n            }));\n\n          for (const message of messages) {\n            const innerMessages = getExternalStoreMessages<TMessage>(message);\n\n            const isTerminal = isTerminalMessage(message);\n            const isAwaitingToolCalls = isAwaitingToolApproval(message);\n            // A paused message's later content can only reach storage via update, so it is persisted early only when the adapter supports update.\n            const isReady =\n              isTerminal ||\n              (isAwaitingToolCalls && adapter.update !== undefined);\n\n            if (!isReady) {\n              lastInnerMessageId =\n                getLastInnerId(innerMessages) ?? lastInnerMessageId;\n              continue;\n            }\n\n            if (!historyIds.current.has(message.id)) {\n              historyIds.current.add(message.id);\n              deferredTelemetryIds.current.add(message.id);\n            }\n\n            const batchItems = toBatchItems(\n              innerMessages,\n              collectToolArtifacts(message, toolArtifacts),\n              collectToolInteractions(message, toolInteractions),\n              collectToolApprovalResponses(message, toolApprovalResponses),\n            );\n            for (const item of batchItems) {\n              const innerId = storageFormatAdapter.getId(item.message);\n              const persisted = persistedInnerMessages.current.get(innerId);\n              if (!persisted) {\n                await adapter.append(item);\n                persistedInnerMessages.current.set(innerId, {\n                  source: item.message,\n                  content: encodeContent(storageFormatAdapter, item),\n                });\n              } else if (\n                persisted.source !== item.message &&\n                durationMs !== undefined &&\n                adapter.update\n              ) {\n                const content = encodeContent(storageFormatAdapter, item);\n                if (content === persisted.content) {\n                  persisted.source = item.message;\n                } else {\n                  try {\n                    await adapter.update(item, innerId);\n                    persistedInnerMessages.current.set(innerId, {\n                      source: item.message,\n                      content,\n                    });\n                  } catch {\n                    // A failed update leaves the stale baseline behind so the next run stop retries it.\n                  }\n                }\n              }\n            }\n\n            lastInnerMessageId =\n              getLastInnerId(innerMessages) ?? lastInnerMessageId;\n\n            if (deferredTelemetryIds.current.has(message.id) && isTerminal) {\n              deferredTelemetryIds.current.delete(message.id);\n              adapter.reportTelemetry?.(\n                batchItems.map((item, index) => ({\n                  ...item,\n                  message: innerMessages[index]!,\n                })),\n                {\n                  ...telemetryOptions,\n                  message,\n                },\n              );\n            }\n          }\n        })\n        .catch((error) => {\n          console.error(\"Failed to persist message history:\", error);\n        });\n    }\n\n    return () => {\n      unsubscribe();\n      if (persistTimerRef.current) {\n        clearTimeout(persistTimerRef.current);\n        persistTimerRef.current = null;\n        persistSettled(false);\n      }\n    };\n  }, [\n    formatAdapter,\n    storageFormatAdapter,\n    runtimeRef,\n    toolArtifacts,\n    toolInteractions,\n    toolApprovalResponses,\n  ]);\n\n  const deleteMessage = useCallback(\n    async (messageId: string) => {\n      const deleteMessages = formatAdapter?.delete?.bind(formatAdapter);\n      if (!deleteMessages) return;\n\n      const messages = runtimeRef.current.thread.getState().messages;\n      const messageIndex = messages.findIndex((m) => m.id === messageId);\n      if (messageIndex === -1) return;\n\n      const previousInnerMessages = messages\n        .slice(0, messageIndex)\n        .flatMap(getExternalStoreMessages<TMessage>);\n      let parentId = previousInnerMessages.at(-1)\n        ? storageFormatAdapter.getId(previousInnerMessages.at(-1)!)\n        : null;\n      const itemsToDelete = getExternalStoreMessages<TMessage>(\n        messages[messageIndex]!,\n      ).map((message) => {\n        const item = { parentId, message };\n        parentId = storageFormatAdapter.getId(message);\n        return item;\n      });\n\n      const deletion = persistInFlightRef.current.then(async () => {\n        await deleteMessages(itemsToDelete);\n\n        historyIds.current.delete(messageId);\n        deferredTelemetryIds.current.delete(messageId);\n        for (const item of itemsToDelete) {\n          persistedInnerMessages.current.delete(\n            storageFormatAdapter.getId(item.message),\n          );\n        }\n      });\n\n      persistInFlightRef.current = deletion.catch(() => {});\n      await deletion;\n    },\n    [formatAdapter, runtimeRef, storageFormatAdapter],\n  );\n\n  return {\n    isLoading,\n    deleteMessage,\n    persistToolInteractions,\n    persistToolApprovalResponses: persistToolInteractions,\n  };\n};\n"],"mappings":";;;;;;;AA6BA,MAAaoB,+BACXC,kBACAC,aAC8B;CAC9B,MAAME,+BAAe,IAAIC,IAAY;CACrC,MAAMC,YAAYJ,SAASA,SAASK,SAASC,MAAM;EACjD,MAAMC,UAAUR,iBAAiB,CAACO,EAAEC,OAAO,CAAC,CAAC,CAAC;EAC9C,IAAI,CAACA,SAAS;GACZC,QAAQC,KAAK,qDAAqD;GAClE,OAAO,CAAA;EACT;EACA,IAAIH,EAAEI,YAAY,CAACR,aAAaS,IAAIL,EAAEI,QAAQ,GAAG,OAAO,CAAA;EACxDR,aAAaU,IAAIL,QAAQM,EAAE;EAC3B,OAAO,CAAC;GAAE,GAAGP;GAAGC;EAAQ,CAAC;CAC3B,CAAC;CAED,OAAO;EACLO,QACEd,SAASc,UAAUZ,aAAaS,IAAIX,SAASc,MAAM,IAC/Cd,SAASc,SACT;EACNd,UAAUI;CACZ;AACF;AAEA,MAAMW,yBAAyBR,YAC7BA,QAAQS,QAAQC,MACbC,SACCA,KAAKC,SAAS,eACdD,KAAKE,YAAY,QACjBF,KAAKE,SAASC,aAAaC,KAAAA,KAC3BJ,KAAKE,SAASG,eAAeD,KAAAA,CACjC;AAGF,MAAME,0BAA0BjB,YAC9BA,QAAQkB,QAAQN,SAAS,sBACxBZ,QAAQkB,OAAOC,WAAW,gBACxBnB,QAAQkB,OAAOC,WAAW,eAAeX,sBAAsBR,OAAO;AAE3E,MAAMoB,qBAAqBpB,YACzBA,QAAQkB,WAAWH,KAAAA,KACnBf,QAAQkB,OAAON,SAAS,cACxBZ,QAAQkB,OAAON,SAAS;AAE1B,MAAMS,8BAA8B;AACpC,MAAMC,iCAAiC;AACvC,MAAMC,uCAAuC;AAO7C,MAAMG,YAAYC,UAChBA,UAAU,QAAQ,OAAOA,UAAU,YAAY,CAACE,MAAMC,QAAQH,KAAK;AAErE,MAAMI,wBACJ/B,SACAgC,kBACG;CACH,IAAI,CAACA,eAAe,OAAOjB,KAAAA;CAC3B,MAAMmB,UAAUlC,QAAQS,QAAQX,SAASa,SAAS;EAChD,IAAIA,KAAKC,SAAS,aAAa,OAAO,CAAA;EACtC,MAAMuB,WAAWH,cAAcI,IAAIzB,KAAK0B,UAAU;EAClD,OAAOF,aAAapB,KAAAA,IAAY,CAAA,IAAK,CAAC,CAACJ,KAAK0B,YAAYF,QAAQ,CAAU;CAC5E,CAAC;CACD,OAAOD,QAAQK,SAAS,IAAIC,OAAOC,YAAYP,OAAO,IAAInB,KAAAA;AAC5D;AAEA,MAAM2B,2BACJ1C,SACA2C,qBAGG;CACH,IAAI,CAACA,kBAAkB,OAAO5B,KAAAA;CAC9B,MAAMmB,UAAUlC,QAAQS,QAAQX,SAASa,SAAS;EAChD,IAAIA,KAAKC,SAAS,aAAa,OAAO,CAAA;EACtC,MAAMgC,eAAeD,iBAAiBP,IAAIzB,KAAK0B,UAAU;EACzD,OAAOO,iBAAiB7B,KAAAA,IACpB,CAAA,IACA,CAAC,CAACJ,KAAK0B,YAAYO,YAAY,CAAU;CAC/C,CAAC;CACD,OAAOV,QAAQK,SAAS,IAAIC,OAAOC,YAAYP,OAAO,IAAInB,KAAAA;AAC5D;AAEA,MAAM8B,gCACJ7C,SACA8C,0BAGG;CACH,IAAI,CAACA,uBAAuB,OAAO/B,KAAAA;CACnC,MAAMmB,UAAUlC,QAAQS,QAAQX,SAASa,SAAS;EAChD,IAAIA,KAAKC,SAAS,eAAe,CAACD,KAAKE,UAAU,OAAO,CAAA;EACxD,MAAMkC,WAAWD,sBAAsBV,IAAIzB,KAAKE,SAASP,EAAE;EAC3D,IAAI,CAACyC,UAAU,OAAO,CAAA;EACtB,OAAO,CACL,CACEpC,KAAKE,SAASP,IACd;GACEQ,UAAUiC,SAASjC;GACnB,GAAIiC,SAASC,YAAY,QAAQ,EAAEA,UAAUD,SAASC,SAAS;GAC/D,GAAID,SAASE,QAAQ,QAAQ,EAAEA,MAAMF,SAASE,KAAK;GACnD,GAAIF,SAASG,WAAW,QAAQ,EAAEA,SAASH,SAASG,QAAQ;GAC5D,GAAIH,SAAS5B,UAAU,QAAQ,EAAEA,QAAQ4B,SAAS5B,OAAO;EAC3D,CAAC,CACO;CAEd,CAAC;CACD,OAAOe,QAAQK,SAAS,IAAIC,OAAOC,YAAYP,OAAO,IAAInB,KAAAA;AAC5D;AAEA,MAAMoC,eACJnD,SACAgC,eACAW,kBACAG,0BACa;CACb,IACG,CAACd,iBAAiB,CAACW,oBAAoB,CAACG,yBACzC,CAACpB,SAAS1B,OAAO,KACjB,CAAC6B,MAAMC,QAAQ9B,QAAQoD,KAAK,GAE5B,OAAOpD;CACT,MAAMqD,cAAcrD,QAAQoD,MAAMtD,SAASa,SAAS;EAClD,IAAI,CAACe,SAASf,IAAI,KAAK,OAAOA,KAAK0B,eAAe,UAAU,OAAO,CAAA;EACnE,OAAO,CAAC1B,KAAK0B,UAAU;CACzB,CAAC;CACD,MAAMiB,YAAYtB,gBACdQ,OAAOC,YACLY,YAAYvD,SAASuC,eACnBG,OAAOe,OAAOvB,eAAeK,UAAU,IACnC,CAAC,CAACA,YAAYL,cAAcK,WAAW,CAAU,IACjD,CAAA,CACN,CACF,IACAtB,KAAAA;CACJ,MAAM6B,eAAeD,mBACjBH,OAAOC,YACLY,YAAYvD,SAASuC,eACnBG,OAAOe,OAAOZ,kBAAkBN,UAAU,IACtC,CAAC,CAACA,YAAYM,iBAAiBN,WAAW,CAAU,IACpD,CAAA,CACN,CACF,IACAtB,KAAAA;CACJ,MAAMyC,cAAcxD,QAAQoD,MAAMtD,SAASa,SAAS;EAClD,IAAI,CAACe,SAASf,IAAI,KAAK,CAACe,SAASf,KAAKE,QAAQ,GAAG,OAAO,CAAA;EACxD,MAAM4C,aAAa9C,KAAKE,SAASP;EACjC,OAAO,OAAOmD,eAAe,WAAW,CAACA,UAAU,IAAI,CAAA;CACzD,CAAC;CACD,MAAMC,oBAAoBZ,wBACtBN,OAAOC,YACLe,YAAY1D,SAAS2D,eACnBjB,OAAOe,OAAOT,uBAAuBW,UAAU,IAC3C,CAAC,CAACA,YAAYX,sBAAsBW,WAAW,CAAU,IACzD,CAAA,CACN,CACF,IACA1C,KAAAA;CACJ,MAAM4C,eAAe,CAAC,CAACL,aAAad,OAAOoB,KAAKN,SAAS,CAAC,CAACf,SAAS;CACpE,MAAMsB,kBACJ,CAAC,CAACjB,gBAAgBJ,OAAOoB,KAAKhB,YAAY,CAAC,CAACL,SAAS;CACvD,MAAMuB,uBACJ,CAAC,CAACJ,qBAAqBlB,OAAOoB,KAAKF,iBAAiB,CAAC,CAACnB,SAAS;CACjE,IAAI,CAACoB,gBAAgB,CAACE,mBAAmB,CAACC,sBACxC,OAAO9D;CACT,MAAM+D,WAAWrC,SAAS1B,QAAQ+D,QAAQ,IAAI/D,QAAQ+D,WAAW,CAAC;CAClE,OAAO;EACL,GAAG/D;EACH+D,UAAU;GACR,GAAGA;GACH,GAAIJ,gBAAgB,GAAGtC,8BAA8BiC,UAAU;GAC/D,GAAIO,mBAAmB,GACpBvC,iCAAiCsB,aACpC;GACA,GAAIkB,wBAAwB,GACzBvC,uCAAuCmC,kBAC1C;EACF;CACF;AACF;AAEA,MAAMM,mBACJhE,SACAgC,eACAW,kBACAG,0BACa;CACb,IAAI,CAACpB,SAAS1B,OAAO,KAAK,CAAC0B,SAAS1B,QAAQ+D,QAAQ,GAAG,OAAO/D;CAC9D,MAAM+D,WAAW/D,QAAQ+D;CACzB,MAAMJ,eAAenB,OAAOe,OAAOQ,UAAU1C,2BAA2B;CACxE,MAAMwC,kBAAkBrB,OAAOe,OAC7BQ,UACAzC,8BACF;CACA,MAAMwC,uBAAuBtB,OAAOe,OAClCQ,UACAxC,oCACF;CACA,IAAI,CAACoC,gBAAgB,CAACE,mBAAmB,CAACC,sBACxC,OAAO9D;CACT,MAAMsD,YAAYS,SAAS1C;CAC3B,IAAIW,iBAAiBN,SAAS4B,SAAS,GACrC,KAAK,MAAM,CAACjB,YAAYF,aAAaK,OAAON,QAAQoB,SAAS,GAC3DtB,cAAckC,IAAI7B,YAAYF,QAAQ;CAG1C,MAAMS,eAAemB,SAASzC;CAC9B,IAAIqB,oBAAoBjB,SAASkB,YAAY,GAC3C,KAAK,MAAM,CAACP,YAAYV,UAAUa,OAAON,QAAQU,YAAY,GAAG;EAC9D,MAAMuB,MAAMrF,uBAAuB6C,KAAK;EACxC,IAAIwC,KAAKxB,iBAAiBuB,IAAI7B,YAAY8B,GAAG;CAC/C;CAEF,MAAMT,oBAAoBK,SAASxC;CACnC,IAAIuB,yBAAyBpB,SAASgC,iBAAiB,GACrD,KAAK,MAAM,CAACD,YAAY9B,UAAUa,OAAON,QAAQwB,iBAAiB,GAAG;EACnE,IAAI,CAAChC,SAASC,KAAK,KAAK,OAAOA,MAAMb,aAAa,WAAW;EAC7D,MAAMoC,UAAU5D,6BAA6BqC,MAAMuB,OAAO;EAC1DJ,sBAAsBoB,IAAIT,YAAY;GACpCA;GACA3C,UAAUa,MAAMb;GAChB,GAAI,OAAOa,MAAMqB,aAAa,YAAY,EACxCA,UAAUrB,MAAMqB,SAClB;GACA,GAAI,OAAOrB,MAAMsB,SAAS,YAAY,EAAEA,MAAMtB,MAAMsB,KAAK;GACzD,GAAIC,WAAWV,OAAOoB,KAAKV,OAAO,CAAC,CAACX,SAAS,KAAK,EAAEW,QAAQ;GAC5D,GAAI,OAAOvB,MAAMR,WAAW,YAAY,EAAEA,QAAQQ,MAAMR,OAAO;EACjE,CAAC;CACH;CAEF,MAAM,GACHE,8BAA8B+C,IAC9B9C,iCAAiC+C,KACjC9C,uCAAuC+C,KACxC,GAAGC,iBACDR;CACJ,MAAM,EAAEA,UAAUS,WAAW,GAAGC,gBAAgBzE;CAChD,OACEwC,OAAOoB,KAAKW,YAAY,CAAC,CAAChC,WAAW,IACjCkC,cACA;EAAE,GAAGA;EAAaV,UAAUQ;CAAa;AAEjD;AAEA,MAAMG,iBACJC,sBACAC,SACGC,KAAKC,UAAUH,qBAAqBI,OAAOH,IAAI,CAAC;AAErD,MAAaI,sBACXC,YACAC,gBACA1F,kBACAmF,sBACAQ,eACAnD,eACAoD,yBACAzC,kBACA0C,4BACAvC,uBACAwC,oCACG;CACH,MAAMC,YAAYvG,OAAO,KAAK;CAC9B,MAAM,CAACwG,WAAWC,gBAAgBvG,SAAS,CAAC;CAE5C,MAAMwG,MAAM3G,OAAO;CACnB,MAAM4G,yBAAyBvG,kBACtBsG,IAAIE,eAAeC,SAASH,IAAIE,iBAAiB,MACxD,CAACF,GAAG,CACN;CAEA,MAAM,CAACI,WAAWC,gBAAgB7G,SAAS,KAAK;CAEhD,MAAM8G,aAAahH,uBAAO,IAAIY,IAAY,CAAC;CAC3C,MAAMqG,uBAAuBjH,uBAAO,IAAIY,IAAY,CAAC;CAErD,MAAMsG,yBAAyBlH,uBAC7B,IAAIiF,IAAmD,CACzD;CAEA,MAAMkC,mBAAmBnH,OAAOmG,aAAa;CAC7ClG,gBAAgB;EACdkH,iBAAiBC,UAAUjB;CAC7B,CAAC;CAED,MAAMkB,gBAAgBhH,cAAc;EAClC,IAAI,CAAC6F,gBAAgB,OAAOnE,KAAAA;EAC5B,IAAI,CAACmE,eAAeoB,YAClB,MAAM,IAAIC,MACR,oFACF;EAEF,OAAOrB,eAAeoB,WAA0B3B,oBAAoB;CACtE,GAAG,CAACO,gBAAgBP,oBAAoB,CAAC;CAEzC,MAAM6B,YAAYH,iBAAiB,QAAQ,CAACP;CAE5C7G,gBAAgB;EACd,IAAI,CAACoH,iBAAiBd,UAAUa,SAAS,OAAOrF,KAAAA;EAEhD,MAAM0F,cAAc,YAAY;GAC9B,IAAI;IACF,MAAMC,OAAO,MAAML,cAAcM,KAAK;IACtC3E,eAAe4E,MAAM;IACrBjE,kBAAkBiE,MAAM;IACxB9D,uBAAuB8D,MAAM;IAC7B,MAAMC,mBACJH,MAAMjH,SAASqH,KAAKlC,UAAU;KAC5B,GAAGA;KACH5E,SAASgE,gBACPY,KAAK5E,SACLgC,eACAW,kBACAG,qBACF;IACF,EAAE,KAAK,CAAA;IACTsC,0BAA0B;IAC1BC,6BAA6B;IAC7BC,kCAAkC;IAClC,IAAIoB,QAAQG,iBAAiBtE,SAAS,GAAG;KACvC,MAAMwE,eAAe;MAAE,GAAGL;MAAMjH,UAAUoH;KAAiB;KAC3D,KAAK,MAAM,CAACG,OAAOjH,MAAM2G,KAAKjH,SAASyC,QAAQ,GAC7CgE,uBAAuBE,QAAQlC,IAC7BS,qBAAqBsC,MAAMlH,EAAEC,OAAO,GACpC;MACE6F,QAAQgB,iBAAiBG,MAAM,CAAEhH;MACjCS,SAASiE,cAAcC,sBAAsB5E,CAAC;KAChD,CACF;KAEF,MAAMmH,YAAY3H,4BAChBC,kBACAuH,YACF;KACA9B,WAAWmB,QAAQe,OAAOC,OAAOF,SAAS;KAE1C,MAAMG,WAAW,IAAIxI,kBAAkB;KACvCwI,SAASD,OAAOF,SAAS;KACzB,MAAMzH,WAAW4H,SAASC,YAAY;KAEtCnB,iBAAiBC,QACf3G,SAASK,QAAQlB,wBAAkC,CACrD;KAEAoH,WAAWI,0BAAU,IAAIxG,IAAI;KAC7B,KAAK,MAAMG,OAAKmH,UAAUzH,UAAU;MAClCuG,WAAWI,QAAQ/F,IAAIN,IAAEC,QAAQM,EAAE;MACnC,IAAIW,uBAAuBlB,IAAEC,OAAO,GAClCiG,qBAAqBG,QAAQ/F,IAAIN,IAAEC,QAAQM,EAAE;KAEjD;IACF;GACF,SAASiH,OAAO;IACdtH,QAAQsH,MAAM,mCAAmCA,KAAK;GACxD,UAAU;IACRxB,aAAa,IAAI;GACnB;EACF;EAEAM,cAAcmB,MAAM;EAGpB,IAAI,CADa7B,uBAAuB,CAAC,EAAE+B,SAAS,CAAC,CAACD,UACvC;GAIb1B,aAAa,IAAI;GACjB,OAAOL,IAAIiC,gBAAgB;IACzB,IAAIhC,uBAAuB,CAAC,EAAE+B,SAAS,CAAC,CAACD,UACvChC,cAAcmC,MAAMA,IAAI,CAAC;GAE7B,CAAC;EACH;EAEA,MAAMC,cAAc5C,WAAWmB,QAAQe,OAAOO,SAAS;EACvD,IAAIG,YAAYC,aAAaD,YAAYpI,SAAS8C,SAAS,GAAG;GAC5DgD,UAAUa,UAAU;GACpBL,aAAa,IAAI;GACjB;EACF;EAEAR,UAAUa,UAAU;EACpB,YAAiB;CAEnB,GAAG;EACDC;EACA7G;EACAyF;EACAU;EACAD;EACAF;EACAb;EACA3C;EACAoD;EACAzC;EACA0C;EACAvC;EACAwC;CAA+B,CAChC;CAED,MAAMyC,cAAc/I,OAAsB,IAAI;CAC9C,MAAMgJ,kBAAkBhJ,OAA6C,IAAI;CACzE,MAAMmJ,qBAAqBnJ,OAAsBoJ,QAAQC,QAAQ,CAAC;CAClE,MAAMC,oBAAoBtJ,OAAiB,CAAA,CAAE;CAC7C,MAAMuJ,gBAAgBvJ,OAAO,KAAK;CAClC,MAAMwJ,mBAAmBxJ,OAAO,CAAC;CAEjC,MAAMyJ,0BAA0BrJ,aAC7BsJ,cAAsB;EACrB,MAAMC,cAAcR,mBAAmB/B,QAAQwC,KAAK,YAAY;GAC9D,IAAI,CAACvC,eAAewC,QAAQ;GAC5B,MAAMpJ,aAAWwF,WAAWmB,QAAQe,OAAOO,SAAS,CAAC,CAACjI;GACtD,MAAMO,UAAUP,WAASqJ,MACtBlE,WACCA,OAAKtE,OAAOoI,aACZ9J,yBAAmCgG,MAAI,CAAC,CAAClE,MACtCqI,iBACCpE,qBAAqBsC,MAAM8B,YAAY,MAAML,SACjD,CACJ;GACA,IAAI,CAAC1I,SAAS;GAEd,MAAMgJ,mBAAmBvJ,WACtBwJ,MAAM,GAAGxJ,WAASyJ,QAAQlJ,OAAO,CAAC,CAAC,CACnCF,QAAQlB,wBAAkC;GAC7C,IAAIuB,WAAW6I,iBAAiBG,GAAG,EAAE,IACjCxE,qBAAqBsC,MAAM+B,iBAAiBG,GAAG,EAAE,CAAE,IACnD;GACJ,MAAMC,sBAAsBrH,qBAC1B/B,SACAgC,aACF;GACA,MAAMqH,yBAAyB3G,wBAC7B1C,SACA2C,gBACF;GACA,MAAM2G,8BAA8BzG,6BAClC7C,SACA8C,qBACF;GAEA,KAAK,MAAMiG,kBAAgBnK,yBACzBoB,OACF,GAAG;IACD,MAAM4E,SAAO;KACXzE;KACAH,SAASmD,YACP4F,gBACAK,qBACAC,wBACAC,2BACF;IACF;IACA,MAAMC,UAAU5E,qBAAqBsC,MAAMrC,OAAK5E,OAAO;IACvD,MAAMwJ,YAAYtD,uBAAuBE,QAAQhE,IAAImH,OAAO;IAC5D,IAAIC,WAAW;KACb,MAAM/I,UAAUiE,cAAcC,sBAAsBC,MAAI;KACxD,IAAInE,YAAY+I,UAAU/I,SACxB+I,UAAU3D,SAASjB,OAAK5E;UACnB;MACL,MAAMqG,cAAcwC,OAAOjE,QAAM2E,OAAO;MACxCrD,uBAAuBE,QAAQlC,IAAIqF,SAAS;OAC1C1D,QAAQjB,OAAK5E;OACbS;MACF,CAAC;KACH;IACF;IACAN,WAAWoJ;GACb;EACF,CAAC;EACDpB,mBAAmB/B,UAAUuC,YAAYc,YAAY,CAAC,CAAC;EACvD,OAAOd,YAAYc,OAAOlC,YAAU;GAClCtH,QAAQsH,MAAM,gCAAgCA,OAAK;EACrD,CAAC;CACH,GACA;EACElB;EACApB;EACAN;EACA3C;EACAW;EACAG;CAAqB,CAEzB;CAEA7D,gBAAgB;EACd,IAAI,CAACoH,eAAe;EACpB,MAAMqD,UAAUrD;EAEhB,MAAMsD,wCAAwB,IAAI/J,IAAY;EAC9C,MAAMgK,uBAAuB5J,cAC3BoB,kBAAkBpB,SAAO,IACrBpB,yBAAmCoB,SAAO,CAAC,CACxC8G,KAAKiC,mBAAiBpE,qBAAqBsC,MAAM8B,cAAY,CAAC,CAAC,CAC/Dc,QAAQN,cAAY,CAACrD,uBAAuBE,QAAQhG,IAAImJ,SAAO,CAAC,IACnE,CAAA;EAEN,MAAMO,cAAc7E,WAAWmB,QAAQe,OAAOQ,gBAAgB;GAC5D,MAAME,gBAAc5C,WAAWmB,QAAQe,OAAOO,SAAS;GACvD,MAAM,EAAEI,cAAcD;GACtB,MAAMkC,aAAaxB,cAAcnC;GACjCmC,cAAcnC,UAAU0B;GAGxB,IAAIC,YAAY3B,WAAW,MAAM;IAC/B,MAAM4D,UAAUnC,cAAYpI,SAAS0J,GAAG,EAAE;IAC1C,IAAIa,SAASC,SAAS,aAAa;KACjC,MAAMC,uBAAuBF,QAAQvJ,QAAQoJ,QAC1CM,MAAMA,EAAEvJ,SAAS,WACpB,CAAC,CAAC2B;KACF,OAAOiG,iBAAiBpC,UAAU8D,sBAAsB;MACtD5B,kBAAkBlC,QAAQgE,KAAKC,KAAKC,IAAI,IAAIvC,YAAY3B,OAAO;MAC/DoC,iBAAiBpC;KACnB;IACF;GACF;GAEA,IAAI0B,WAAW;IACb,IAAIC,YAAY3B,WAAW,MAAM;KAC/B2B,YAAY3B,UAAUiE,KAAKC,IAAI;KAC/BhC,kBAAkBlC,UAAU,CAAA;KAC5BoC,iBAAiBpC,UAAU;KAC3BsD,QAAQlC,MAAM;IAChB;IAEA,IAAIQ,gBAAgB5B,SAAS;KAC3BmE,aAAavC,gBAAgB5B,OAAO;KACpC4B,gBAAgB5B,UAAU;IAC5B;IACA;GACF;GAEA,IAAI2D,YAEEhC;QAAAA,YAAY3B,WAAW,MACzBkC,kBAAkBlC,QAAQgE,KAAKC,KAAKC,IAAI,IAAIvC,YAAY3B,OAAO;GAAA,OAE5D;IACL,MAAMoE,UAAU3C,cAAYpI,SAASK,QAAQ8J,mBAAmB;IAChE,IAAIY,QAAQC,OAAOlB,cAAYI,sBAAsBvJ,IAAImJ,SAAO,CAAC,GAC/D;IACF,KAAK,MAAMA,aAAWiB,SAASb,sBAAsBtJ,IAAIkJ,SAAO;GAClE;GAGA,IAAIvB,gBAAgB5B,SAASmE,aAAavC,gBAAgB5B,OAAO;GACjE4B,gBAAgB5B,UAAU8B,iBAAiB;IACzCwC,eAAe,KAAK;GACtB,GAAG,CAAC;EACN,CAAC;EAED,MAAMC,qBAAqB1F,WAAWmB,QAAQe,OAAOO,SAAS;EAC9Da,cAAcnC,UAAUuE,mBAAmB7C;EAC3C,IAAI6C,mBAAmB7C,WACjBC;OAAAA,YAAY3B,WAAW,MAAM;IAC/B2B,YAAY3B,UAAUiE,KAAKC,IAAI;IAC/BhC,kBAAkBlC,UAAU,CAAA;IAC5BoC,iBAAiBpC,UAAU;IAC3BsD,QAAQlC,MAAM;GAChB;SACK,IAAImD,mBAAmBlL,SAAS8C,SAAS,GAC9CmI,eAAe,IAAI;EAGrB,SAASA,eAAeE,eAAwB;GAC9C5C,gBAAgB5B,UAAU;GAC1B,MAAMyE,SAAS5F,WAAWmB,QAAQe,OAAOO,SAAS;GAClD,IAAI,CAACkD,iBAAiBC,OAAO/C,WAAW;GAExC,MAAMgD,aAAaxC,kBAAkBlC;GACrC,MAAM2E,aAAaD,WAAWvI,SAAS,IAAIuI,WAAW3B,GAAG,EAAE,IAAIpI,KAAAA;GAG/D,IAAI+J,WAAWvI,WAAW,KAAKwI,cAAc,MAAM;IACjD,MAAMC,gBAAgBH,OAAOpL,SAASwL,UACnClL,QAAMA,IAAEkK,SAAS,WACpB;IACA,IAAIe,eAAe;KACjB,MAAME,UAAUF,cAAcvK,QAAQoJ,QACnCM,QAAMA,IAAEvJ,SAAS,WACpB,CAAC,CAAC2B;KACF,IAAI2I,UAAU,GAAG;MACf,MAAMC,aAAaD,UAAU;MAC7B,MAAME,UAAUL,aAAaI;MAC7BL,WAAWvI,SAAS;MACpB,KAAK,IAAI8I,IAAI,GAAGA,IAAIF,YAAYE,KAC9BP,WAAWV,KAAKkB,KAAKC,OAAOF,IAAI,KAAKD,OAAO,CAAC;KAEjD;IACF;GACF;GAGA,MAAMI,iBACJV,WAAWvI,SAAS,IAChBuI,WAAWhE,KAAK2E,OAAOJ,SAAO;IAC5BK,UAAUL,QAAM,IAAI,IAAIP,WAAWO,MAAI;IACvCM,QAAQF;GACV,EAAE,IACF1K,KAAAA;GAENgH,YAAY3B,UAAU;GACtBkC,kBAAkBlC,UAAU,CAAA;GAE5B,MAAMwF,mBAAmB;IACvB,GAAIb,cAAc,OAAO,EAAEA,WAAW,IAAIhK,KAAAA;IAC1C,GAAIyK,kBAAkB,OAAO,EAAEA,eAAe,IAAIzK,KAAAA;GACpD;GAEAoH,mBAAmB/B,UAAU+B,mBAAmB/B,QAC7CwC,KAAK,YAAY;IAChB,MAAM,EAAEnJ,UAAAA,eAAaoL;IACrB,IAAIgB,qBAAoC;IAExC,MAAMC,kBAAkBC,SACtBA,KAAKxJ,SAAS,IAAIoC,qBAAqBsC,MAAM8E,KAAK5C,GAAG,EAAE,CAAE,IAAI;IAE/D,MAAM6C,gBACJD,QACA/J,iBACAW,oBAGAG,4BAIAiJ,OAAKjF,KAAKmF,KAAKC,SAAS;KACtB/L,UACE+L,QAAQ,IACJL,qBACAlH,qBAAqBsC,MAAM8E,OAAKG,MAAM,EAAG;KAC/ClM,SAASmD,YACP8I,KACAjK,iBACAW,oBACAG,uBACF;IACF,EAAE;IAEJ,KAAK,MAAM9C,aAAWP,YAAU;KAC9B,MAAM0M,gBAAgBvN,yBAAmCoB,SAAO;KAEhE,MAAMoM,aAAahL,kBAAkBpB,SAAO;KAC5C,MAAMqM,sBAAsBpL,uBAAuBjB,SAAO;KAM1D,IAAI,EAHFoM,cACCC,uBAAuB3C,QAAQb,WAAW9H,KAAAA,IAE/B;MACZ8K,qBACEC,eAAeK,aAAa,KAAKN;MACnC;KACF;KAEA,IAAI,CAAC7F,WAAWI,QAAQhG,IAAIJ,UAAQM,EAAE,GAAG;MACvC0F,WAAWI,QAAQ/F,IAAIL,UAAQM,EAAE;MACjC2F,qBAAqBG,QAAQ/F,IAAIL,UAAQM,EAAE;KAC7C;KAEA,MAAMiM,aAAaP,aACjBG,eACApK,qBAAqB/B,WAASgC,aAAa,GAC3CU,wBAAwB1C,WAAS2C,gBAAgB,GACjDE,6BAA6B7C,WAAS8C,qBAAqB,CAC7D;KACA,KAAK,MAAM8B,UAAQ2H,YAAY;MAC7B,MAAMhD,YAAU5E,qBAAqBsC,MAAMrC,OAAK5E,OAAO;MACvD,MAAMwJ,cAAYtD,uBAAuBE,QAAQhE,IAAImH,SAAO;MAC5D,IAAI,CAACC,aAAW;OACd,MAAME,QAAQ8C,OAAO5H,MAAI;OACzBsB,uBAAuBE,QAAQlC,IAAIqF,WAAS;QAC1C1D,QAAQjB,OAAK5E;QACbS,SAASiE,cAAcC,sBAAsBC,MAAI;OACnD,CAAC;MACH,OAAO,IACL4E,YAAU3D,WAAWjB,OAAK5E,WAC1B+K,eAAehK,KAAAA,KACf2I,QAAQb,QACR;OACA,MAAMpI,YAAUiE,cAAcC,sBAAsBC,MAAI;OACxD,IAAInE,cAAY+I,YAAU/I,SACxB+I,YAAU3D,SAASjB,OAAK5E;YAExB,IAAI;QACF,MAAM0J,QAAQb,OAAOjE,QAAM2E,SAAO;QAClCrD,uBAAuBE,QAAQlC,IAAIqF,WAAS;SAC1C1D,QAAQjB,OAAK5E;SACbS,SAAAA;QACF,CAAC;OACH,QAAQ,CACN;MAGN;KACF;KAEAoL,qBACEC,eAAeK,aAAa,KAAKN;KAEnC,IAAI5F,qBAAqBG,QAAQhG,IAAIJ,UAAQM,EAAE,KAAK8L,YAAY;MAC9DnG,qBAAqBG,QAAQqG,OAAOzM,UAAQM,EAAE;MAC9CoJ,QAAQgD,kBACNH,WAAWzF,KAAKlC,QAAMoC,aAAW;OAC/B,GAAGpC;OACH5E,SAASmM,cAAcnF;MACzB,EAAE,GACF;OACE,GAAG4E;OACH5L,SAAAA;MACF,CACF;KACF;IACF;GACF,CAAC,CAAC,CACDyJ,OAAOlC,YAAU;IAChBtH,QAAQsH,MAAM,sCAAsCA,OAAK;GAC3D,CAAC;EACL;EAEA,aAAa;GACXuC,YAAY;GACZ,IAAI9B,gBAAgB5B,SAAS;IAC3BmE,aAAavC,gBAAgB5B,OAAO;IACpC4B,gBAAgB5B,UAAU;IAC1BsE,eAAe,KAAK;GACtB;EACF;CACF,GAAG;EACDrE;EACA1B;EACAM;EACAjD;EACAW;EACAG;CAAqB,CACtB;CA2CD,OAAO;EACL0D;EACAmG,eA3CoBvN,YACpB,OAAOsJ,gBAAsB;GAC3B,MAAMkE,iBAAiBvG,eAAeoG,QAAQI,KAAKxG,aAAa;GAChE,IAAI,CAACuG,gBAAgB;GAErB,MAAMnN,aAAWwF,WAAWmB,QAAQe,OAAOO,SAAS,CAAC,CAACjI;GACtD,MAAMqN,eAAerN,WAASsN,WAAWhN,QAAMA,IAAEO,OAAOoI,WAAS;GACjE,IAAIoE,iBAAiB,IAAI;GAEzB,MAAME,wBAAwBvN,WAC3BwJ,MAAM,GAAG6D,YAAY,CAAC,CACtBhN,QAAQlB,wBAAkC;GAC7C,IAAIuB,aAAW6M,sBAAsB7D,GAAG,EAAE,IACtCxE,qBAAqBsC,MAAM+F,sBAAsB7D,GAAG,EAAE,CAAE,IACxD;GACJ,MAAM8D,gBAAgBrO,yBACpBa,WAASqN,aACX,CAAC,CAAChG,KAAK9G,cAAY;IACjB,MAAM4E,SAAO;KAAEzE,UAAAA;KAAUH,SAAAA;IAAQ;IACjCG,aAAWwE,qBAAqBsC,MAAMjH,SAAO;IAC7C,OAAO4E;GACT,CAAC;GAED,MAAMsI,WAAW/E,mBAAmB/B,QAAQwC,KAAK,YAAY;IAC3D,MAAMgE,eAAeK,aAAa;IAElCjH,WAAWI,QAAQqG,OAAO/D,WAAS;IACnCzC,qBAAqBG,QAAQqG,OAAO/D,WAAS;IAC7C,KAAK,MAAM9D,UAAQqI,eACjB/G,uBAAuBE,QAAQqG,OAC7B9H,qBAAqBsC,MAAMrC,OAAK5E,OAAO,CACzC;GAEJ,CAAC;GAEDmI,mBAAmB/B,UAAU8G,SAASzD,YAAY,CAAC,CAAC;GACpD,MAAMyD;EACR,GACA;GAAC7G;GAAepB;GAAYN;EAAoB,CAKhDgI;EACAlE;EACA0E,8BAA8B1E;CAChC;AACF"}

package/dist/runtime/useResourceCleanup.js.map

@@ -1 +1 @@
-{"version":3,"file":"useResourceCleanup.js","names":["useEffect","useRef","useAssistantClientDestroySignal","useResourceCleanup","enabled","cleanup","hostDestroySignal","$","_c","destroySignal","cleanupRef","enabledRef","registrationRef","t0","current","t1","t2","controller","abort","registration","AbortController","run","signal","aborted","options","once","addEventListener"],"sources":["../../src/runtime/useResourceCleanup.ts"],"sourcesContent":["import { useEffect, useRef } from \"react\";\nimport { useAssistantClientDestroySignal } from \"@assistant-ui/store/internal\";\n\nexport const useResourceCleanup = (\n  enabled: boolean,\n  cleanup: () => void,\n  hostDestroySignal?: AbortSignal,\n) => {\n  const destroySignal = useAssistantClientDestroySignal();\n  const cleanupRef = useRef(cleanup);\n  const enabledRef = useRef(enabled);\n  const registrationRef = useRef<{\n    destroySignal: AbortSignal | undefined;\n    hostDestroySignal: AbortSignal | undefined;\n    controller: AbortController;\n  } | null>(null);\n\n  useEffect(() => {\n    cleanupRef.current = cleanup;\n    enabledRef.current = enabled;\n  });\n\n  useEffect(() => {\n    if (!enabled || (!destroySignal && !hostDestroySignal)) return undefined;\n    const current = registrationRef.current;\n    if (\n      current !== null &&\n      current.destroySignal === destroySignal &&\n      current.hostDestroySignal === hostDestroySignal\n    ) {\n      return undefined;\n    }\n    current?.controller.abort();\n\n    const registration = new AbortController();\n    registrationRef.current = {\n      destroySignal,\n      hostDestroySignal,\n      controller: registration,\n    };\n    const run = () => {\n      if (registration.signal.aborted) return;\n      registration.abort();\n      if (enabledRef.current) cleanupRef.current();\n    };\n    if (destroySignal?.aborted || hostDestroySignal?.aborted) {\n      run();\n      return undefined;\n    }\n    const options = { once: true, signal: registration.signal };\n    destroySignal?.addEventListener(\"abort\", run, options);\n    hostDestroySignal?.addEventListener(\"abort\", run, options);\n\n    // The listeners must survive standalone soft unmounts so a later\n    // permanent destroy still cleans up the retained resource state.\n    return undefined;\n  }, [destroySignal, enabled, hostDestroySignal]);\n};\n"],"mappings":";;;;AAGA,MAAaG,sBAAqBC,SAAAC,SAAAC,sBAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAKhC,MAAAC,gBAAsBP,gCAAgC;CACtD,MAAAQ,aAAmBT,OAAOI,OAAO;CACjC,MAAAM,aAAmBV,OAAOG,OAAO;CACjC,MAAAQ,kBAAwBX,OAId,IAAI;CAAE,IAAAY;CAAA,IAAAN,EAAA,OAAAF,WAAAE,EAAA,OAAAH,SAAA;EAENS,WAAA;GACRH,WAAUI,UAAWT;GACrBM,WAAUG,UAAWV;EAAH;EACnBG,EAAA,KAAAF;EAAAE,EAAA,KAAAH;EAAAG,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAHDP,UAAUa,EAGT;CAAC,IAAAE;CAAA,IAAAC;CAAA,IAAAT,EAAA,OAAAE,iBAAAF,EAAA,OAAAH,WAAAG,EAAA,OAAAD,mBAAA;EAEQS,WAAA;GACR,IAAI,CAACX,WAAY,CAACK,iBAAD,CAAmBH,mBAAkB;GACtD,MAAAQ,UAAgBF,gBAAeE;GAC/B,IACEA,YAAY,QACZA,QAAOL,kBAAmBA,iBAC1BK,QAAOR,sBAAuBA,mBAAiB;GAIjDQ,SAAOG,WAAkBC,MAAC;GAE1B,MAAAC,eAAqB,IAAIC,gBAAgB;GACzCR,gBAAeE,UAAW;IAAAL;IAAAH;IAAAW,YAGZE;GACd;GACA,MAAAE,YAAY;IACV,IAAIF,aAAYG,OAAOC,SAAQ;IAC/BJ,aAAYD,MAAO;IACnB,IAAIP,WAAUG,SAAUJ,WAAUI,QAAS;GAAE;GAE/C,IAAIL,eAAac,WAAajB,mBAAiBiB,SAAS;IACtDF,IAAI;IAAC;GAAA;GAGP,MAAAG,UAAgB;IAAAC,MAAQ;IAAIH,QAAUH,aAAYG;GAAQ;GAC1Db,eAAaiB,iBAAmB,SAASL,KAAKG,OAAO;GACrDlB,mBAAiBoB,iBAAmB,SAASL,KAAKG,OAAO;EAAC;EAKzDR,KAAA;GAACP;GAAeL;GAASE;EAAiB;EAACC,EAAA,KAAAE;EAAAF,EAAA,KAAAH;EAAAG,EAAA,KAAAD;EAAAC,EAAA,KAAAQ;EAAAR,EAAA,KAAAS;CAAA,OAAA;EAAAD,KAAAR,EAAA;EAAAS,KAAAT,EAAA;CAAA;CAlC9CP,UAAUe,IAkCPC,EAA2C;AAAC"}
+{"version":3,"file":"useResourceCleanup.js","names":["useEffect","useRef","useAssistantClientDestroySignal","useResourceCleanup","enabled","cleanup","hostDestroySignal","$","_c","destroySignal","cleanupRef","enabledRef","registrationRef","t0","current","t1","t2","controller","abort","registration","AbortController","run","signal","aborted","options","once","addEventListener","_c"],"sources":["../../src/runtime/useResourceCleanup.ts"],"sourcesContent":["import { useEffect, useRef } from \"react\";\nimport { useAssistantClientDestroySignal } from \"@assistant-ui/store/internal\";\n\nexport const useResourceCleanup = (\n  enabled: boolean,\n  cleanup: () => void,\n  hostDestroySignal?: AbortSignal,\n) => {\n  const destroySignal = useAssistantClientDestroySignal();\n  const cleanupRef = useRef(cleanup);\n  const enabledRef = useRef(enabled);\n  const registrationRef = useRef<{\n    destroySignal: AbortSignal | undefined;\n    hostDestroySignal: AbortSignal | undefined;\n    controller: AbortController;\n  } | null>(null);\n\n  useEffect(() => {\n    cleanupRef.current = cleanup;\n    enabledRef.current = enabled;\n  });\n\n  useEffect(() => {\n    if (!enabled || (!destroySignal && !hostDestroySignal)) return undefined;\n    const current = registrationRef.current;\n    if (\n      current !== null &&\n      current.destroySignal === destroySignal &&\n      current.hostDestroySignal === hostDestroySignal\n    ) {\n      return undefined;\n    }\n    current?.controller.abort();\n\n    const registration = new AbortController();\n    registrationRef.current = {\n      destroySignal,\n      hostDestroySignal,\n      controller: registration,\n    };\n    const run = () => {\n      if (registration.signal.aborted) return;\n      registration.abort();\n      if (enabledRef.current) cleanupRef.current();\n    };\n    if (destroySignal?.aborted || hostDestroySignal?.aborted) {\n      run();\n      return undefined;\n    }\n    const options = { once: true, signal: registration.signal };\n    destroySignal?.addEventListener(\"abort\", run, options);\n    hostDestroySignal?.addEventListener(\"abort\", run, options);\n\n    // The listeners must survive standalone soft unmounts so a later\n    // permanent destroy still cleans up the retained resource state.\n    return undefined;\n  }, [destroySignal, enabled, hostDestroySignal]);\n};\n"],"mappings":";;;;AAGA,MAAaG,sBAAqBC,SAAAC,SAAAC,sBAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAKhC,MAAAC,gBAAsBP,gCAAgC;CACtD,MAAAQ,aAAmBT,OAAOI,OAAO;CACjC,MAAAM,aAAmBV,OAAOG,OAAO;CACjC,MAAAQ,kBAAwBX,OAId,IAAI;CAAE,IAAAY;CAAA,IAAAN,EAAA,OAAAF,WAAAE,EAAA,OAAAH,SAAA;EAENS,WAAA;GACRH,WAAUI,UAAWT;GACrBM,WAAUG,UAAWV;EAAH;EACnBG,EAAA,KAAAF;EAAAE,EAAA,KAAAH;EAAAG,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAHDP,UAAUa,EAGT;CAAC,IAAAE;CAAA,IAAAC;CAAA,IAAAT,EAAA,OAAAE,iBAAAF,EAAA,OAAAH,WAAAG,EAAA,OAAAD,mBAAA;EAEQS,WAAA;GACR,IAAI,CAACX,WAAY,CAACK,iBAAD,CAAmBH,mBAAkB;GACtD,MAAAQ,UAAgBF,gBAAeE;GAC/B,IACEA,YAAY,QACZA,QAAOL,kBAAmBA,iBAC1BK,QAAOR,sBAAuBA,mBAAiB;GAIjDQ,SAAOG,WAAkBC,MAAC;GAE1B,MAAAC,eAAqB,IAAIC,gBAAgB;GACzCR,gBAAeE,UAAW;IAAAL;IAAAH;IAAAW,YAGZE;GACd;GACA,MAAAE,YAAY;IACV,IAAIF,aAAYG,OAAOC,SAAQ;IAC/BJ,aAAYD,MAAO;IACnB,IAAIP,WAAUG,SAAUJ,WAAUI,QAAS;GAAE;GAE/C,IAAIL,eAAac,WAAajB,mBAAiBiB,SAAS;IACtDF,IAAI;IAAC;GAAA;GAGP,MAAAG,UAAgB;IAAAC,MAAQ;IAAIH,QAAUH,aAAYG;GAAQ;GAC1Db,eAAaiB,iBAAmB,SAASL,KAAKG,OAAO;GACrDlB,mBAAiBoB,iBAAmB,SAASL,KAAKG,OAAO;EAAC;EAKzDR,KAAA;GAACP;GAAeL;GAASE;EAAiB;EAACC,EAAA,KAAAE;EAAAF,EAAA,KAAAH;EAAAG,EAAA,KAAAD;EAAAC,EAAA,KAAAQ;EAAAR,EAAA,KAAAS;CAAA,OAAA;EAAAD,KAAAR,EAAA;EAAAS,KAAAT,EAAA;CAAA;CAlC9CP,UAAUe,IAkCPC,EAA2C;AAAC"}

package/dist/runtime/useStreamingTiming.js.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamingTiming.js","names":["UIMessage","isToolUIPart","MessageTiming","useStreamingTiming","useStreamingTimingPrimitive","StreamingTimingAccessors","findAssistant","messages","messageId","find","m","role","id","getTextLength","message","parts","len","part","type","text","length","getToolCallCount","count","getAssistantMessageId","findLast","aiSdkStreamingTimingAccessors","isRunning"],"sources":["../../src/runtime/useStreamingTiming.ts"],"sourcesContent":["\"use client\";\n\nimport type { UIMessage } from \"@ai-sdk/react\";\nimport { isToolUIPart } from \"ai\";\nimport type { MessageTiming } from \"@assistant-ui/core\";\nimport {\n  useStreamingTiming as useStreamingTimingPrimitive,\n  type StreamingTimingAccessors,\n} from \"@assistant-ui/core/react\";\n\nconst findAssistant = (\n  messages: readonly UIMessage[],\n  messageId: string,\n): UIMessage | undefined =>\n  messages.find((m) => m.role === \"assistant\" && m.id === messageId);\n\nconst getTextLength = (\n  messages: readonly UIMessage[],\n  messageId: string,\n): number => {\n  const message = findAssistant(messages, messageId);\n  if (!message?.parts) return 0;\n  let len = 0;\n  for (const part of message.parts) {\n    if (part.type === \"text\") len += part.text.length;\n  }\n  return len;\n};\n\nconst getToolCallCount = (\n  messages: readonly UIMessage[],\n  messageId: string,\n): number => {\n  const message = findAssistant(messages, messageId);\n  if (!message?.parts) return 0;\n  let count = 0;\n  for (const part of message.parts) {\n    if (isToolUIPart(part)) count++;\n  }\n  return count;\n};\n\nconst getAssistantMessageId = (\n  messages: readonly UIMessage[],\n): string | undefined => messages.findLast((m) => m.role === \"assistant\")?.id;\n\nexport const aiSdkStreamingTimingAccessors: StreamingTimingAccessors<UIMessage> =\n  {\n    getAssistantMessageId,\n    getTextLength,\n    getToolCallCount,\n  };\n\n/**\n * Tracks streaming timing for AI SDK messages client-side. Delegates to the\n * shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,\n * adapted to the `UIMessage` shape. Timing is finalized when streaming ends\n * and stored per message id.\n */\nexport const useStreamingTiming = (\n  messages: UIMessage[],\n  isRunning: boolean,\n): Record<string, MessageTiming> =>\n  useStreamingTimingPrimitive(\n    messages,\n    isRunning,\n    aiSdkStreamingTimingAccessors,\n  );\n"],"mappings":";;;;AAUA,MAAMM,iBACJC,UACAC,cAEAD,SAASE,MAAMC,MAAMA,EAAEC,SAAS,eAAeD,EAAEE,OAAOJ,SAAS;AAEnE,MAAMK,iBACJN,UACAC,cACW;CACX,MAAMM,UAAUR,cAAcC,UAAUC,SAAS;CACjD,IAAI,CAACM,SAASC,OAAO,OAAO;CAC5B,IAAIC,MAAM;CACV,KAAK,MAAMC,QAAQH,QAAQC,OACzB,IAAIE,KAAKC,SAAS,QAAQF,OAAOC,KAAKE,KAAKC;CAE7C,OAAOJ;AACT;AAEA,MAAMK,oBACJd,UACAC,cACW;CACX,MAAMM,UAAUR,cAAcC,UAAUC,SAAS;CACjD,IAAI,CAACM,SAASC,OAAO,OAAO;CAC5B,IAAIO,QAAQ;CACZ,KAAK,MAAML,QAAQH,QAAQC,OACzB,IAAId,aAAagB,IAAI,GAAGK;CAE1B,OAAOA;AACT;AAEA,MAAMC,yBACJhB,aACuBA,SAASiB,UAAUd,MAAMA,EAAEC,SAAS,WAAW,CAAC,EAAEC;AAE3E,MAAaa,gCACX;CACEF;CACAV;CACAQ;AACF;;;;;;;AAQF,MAAalB,sBAAqBI,UAAAmB,cAAA;CAAA,OAIhCtB,qBACEG,UACAmB,WACAD,6BACF;AAAC"}
+{"version":3,"file":"useStreamingTiming.js","names":["UIMessage","isToolUIPart","MessageTiming","useStreamingTiming","useStreamingTimingPrimitive","StreamingTimingAccessors","findAssistant","messages","messageId","find","m","role","id","getTextLength","message","parts","len","part","type","text","length","getToolCallCount","count","getAssistantMessageId","findLast","aiSdkStreamingTimingAccessors","isRunning","useStreamingTimingPrimitive"],"sources":["../../src/runtime/useStreamingTiming.ts"],"sourcesContent":["\"use client\";\n\nimport type { UIMessage } from \"@ai-sdk/react\";\nimport { isToolUIPart } from \"ai\";\nimport type { MessageTiming } from \"@assistant-ui/core\";\nimport {\n  useStreamingTiming as useStreamingTimingPrimitive,\n  type StreamingTimingAccessors,\n} from \"@assistant-ui/core/react\";\n\nconst findAssistant = (\n  messages: readonly UIMessage[],\n  messageId: string,\n): UIMessage | undefined =>\n  messages.find((m) => m.role === \"assistant\" && m.id === messageId);\n\nconst getTextLength = (\n  messages: readonly UIMessage[],\n  messageId: string,\n): number => {\n  const message = findAssistant(messages, messageId);\n  if (!message?.parts) return 0;\n  let len = 0;\n  for (const part of message.parts) {\n    if (part.type === \"text\") len += part.text.length;\n  }\n  return len;\n};\n\nconst getToolCallCount = (\n  messages: readonly UIMessage[],\n  messageId: string,\n): number => {\n  const message = findAssistant(messages, messageId);\n  if (!message?.parts) return 0;\n  let count = 0;\n  for (const part of message.parts) {\n    if (isToolUIPart(part)) count++;\n  }\n  return count;\n};\n\nconst getAssistantMessageId = (\n  messages: readonly UIMessage[],\n): string | undefined => messages.findLast((m) => m.role === \"assistant\")?.id;\n\nexport const aiSdkStreamingTimingAccessors: StreamingTimingAccessors<UIMessage> =\n  {\n    getAssistantMessageId,\n    getTextLength,\n    getToolCallCount,\n  };\n\n/**\n * Tracks streaming timing for AI SDK messages client-side. Delegates to the\n * shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,\n * adapted to the `UIMessage` shape. Timing is finalized when streaming ends\n * and stored per message id.\n */\nexport const useStreamingTiming = (\n  messages: UIMessage[],\n  isRunning: boolean,\n): Record<string, MessageTiming> =>\n  useStreamingTimingPrimitive(\n    messages,\n    isRunning,\n    aiSdkStreamingTimingAccessors,\n  );\n"],"mappings":";;;;AAUA,MAAMM,iBACJC,UACAC,cAEAD,SAASE,MAAMC,MAAMA,EAAEC,SAAS,eAAeD,EAAEE,OAAOJ,SAAS;AAEnE,MAAMK,iBACJN,UACAC,cACW;CACX,MAAMM,UAAUR,cAAcC,UAAUC,SAAS;CACjD,IAAI,CAACM,SAASC,OAAO,OAAO;CAC5B,IAAIC,MAAM;CACV,KAAK,MAAMC,QAAQH,QAAQC,OACzB,IAAIE,KAAKC,SAAS,QAAQF,OAAOC,KAAKE,KAAKC;CAE7C,OAAOJ;AACT;AAEA,MAAMK,oBACJd,UACAC,cACW;CACX,MAAMM,UAAUR,cAAcC,UAAUC,SAAS;CACjD,IAAI,CAACM,SAASC,OAAO,OAAO;CAC5B,IAAIO,QAAQ;CACZ,KAAK,MAAML,QAAQH,QAAQC,OACzB,IAAId,aAAagB,IAAI,GAAGK;CAE1B,OAAOA;AACT;AAEA,MAAMC,yBACJhB,aACuBA,SAASiB,UAAUd,MAAMA,EAAEC,SAAS,WAAW,CAAC,EAAEC;AAE3E,MAAaa,gCACX;CACEF;CACAV;CACAQ;AACF;;;;;;;AAQF,MAAalB,sBAAqBI,UAAAmB,cAAA;CAAA,OAIhCtB,qBACEG,UACAmB,WACAD,6BACF;AAAC"}

package/dist/tools/generativeTools.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generativeTools.d.ts","sourceRoot":"","sources":["../../src/tools/generativeTools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,KAAK,OAAO,EAAE,MAAM,IAAI,CAAC;AAU9C,OAAO,KAAK,EAEV,OAAO,EAER,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAiB,KAAK,aAAa,EAAE,2BAAwB;AAkEpE;;;;GAIG;AACH,MAAM,WAAW,sBAAsB;IACrC;;;;OAIG;IACH,OAAO,EAAE,OAAO,CAAC;IACjB;;;;OAIG;IACH,aAAa,CAAC,EAAE,aAAa,CAAC;CAC/B;AAED,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,EAAE,OAAO,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC;;OAEG;IACH,QAAQ,CAAC,EAAE,aAAa,CAAC;CAC1B,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,eAAO,MAAM,eAAe,GAAI,SAAS,sBAAsB,KAAG,OAUjE,CAAC;AAEF,qBAAa,YAAY;;gBAIX,OAAO,EAAE,mBAAmB;IAIlC,KAAK,CAAC,OAAO,GAAE,wBAA6B,GAAG,OAAO,CAAC,OAAO,CAAC;IAsB/D,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAmH7B"}
+{"version":3,"file":"generativeTools.d.ts","sourceRoot":"","sources":["../../src/tools/generativeTools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,KAAK,OAAO,EAAE,MAAM,IAAI,CAAC;AAU9C,OAAO,KAAK,EAEV,OAAO,EAER,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAiB,KAAK,aAAa,EAAE,2BAAwB;AAuFpE;;;;GAIG;AACH,MAAM,WAAW,sBAAsB;IACrC;;;;OAIG;IACH,OAAO,EAAE,OAAO,CAAC;IACjB;;;;OAIG;IACH,aAAa,CAAC,EAAE,aAAa,CAAC;CAC/B;AAED,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,EAAE,OAAO,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC;;OAEG;IACH,QAAQ,CAAC,EAAE,aAAa,CAAC;CAC1B,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,eAAO,MAAM,eAAe,GAAI,SAAS,sBAAsB,KAAG,OAUjE,CAAC;AAEF,qBAAa,YAAY;;gBAIX,OAAO,EAAE,mBAAmB;IAIlC,KAAK,CAAC,OAAO,GAAE,wBAA6B,GAAG,OAAO,CAAC,OAAO,CAAC;IAsB/D,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAmH7B"}

package/dist/tools/generativeTools.js

@@ -31,7 +31,18 @@ const withMcpConnectionTimeout = async (promise, options) => {
 		if (timeout !== void 0) clearTimeout(timeout);
 	}
 };
-const parametersToInputSchema = (parameters) => jsonSchema(parameters ? toJSONSchema(parameters) : EMPTY_SCHEMA);
+const convertedParameterSchemas = /* @__PURE__ */ new WeakMap();
+const getOrConvertParameterSchema = (parameters) => {
+	const cached = convertedParameterSchemas.get(parameters);
+	if (cached) return cached;
+	const converted = toJSONSchema(parameters);
+	convertedParameterSchemas.set(parameters, converted);
+	return converted;
+};
+const parametersToInputSchema = (parameters) => {
+	if (!parameters) return jsonSchema(EMPTY_SCHEMA);
+	return jsonSchema(getOrConvertParameterSchema(parameters));
+};
 /**
 * Builds an AI SDK `ToolSet` for server-side use with `streamText` /
 * `generateText` from a generative `toolkit` and the frontend-uploaded tools.

package/dist/tools/generativeTools.js.map

@@ -1 +1 @@
-{"version":3,"file":"generativeTools.js","names":["jsonSchema","ToolSet","MCPClient","MCPClientConfig","createMCPClient","Experimental_StdioMCPTransport","toJSONSchema","Tool","McpServerConfig","ToolModelOutputFunction","McpToolkitToolConfig","Toolkit","ToolkitDefinition","frontendTools","FrontendTools","toAISDKContent","toAISDKDefaultOutput","unwrapModelContentEnvelope","ModelContentEnvelope","EMPTY_SCHEMA","type","const","properties","humanNotSupported","Error","neverAbort","AbortController","signal","MCPConnectionTimeoutPhase","MCPConnectionTimeoutError","createMcpConnectionTimeoutError","name","phase","timeoutMs","withMcpConnectionTimeout","promise","Promise","T","options","config","startedAt","connectionTimeout","undefined","remainingMs","Date","now","timeoutError","timeout","ReturnType","setTimeout","race","_","reject","clearTimeout","parametersToInputSchema","parameters","GenerativeToolsOptions","toolkit","AISDKToolkitOptions","AISDKToolkitToolsOptions","frontend","generativeTools","assertNoMcpToolkitTools","toProviderToolSet","toServerToolSet","AISDKToolkit","mcpClients","Map","constructor","tools","frontendToolSet","mcpToolSet","mcpTools","providerToolSet","serverToolSet","assertNoMcpToolNameCollisions","source","close","clientEntries","entries","clientNames","map","clear","clientResults","allSettled","clientPromise","clients","flatMap","result","index","status","value","closeResults","client","errors","toMcpToolkitError","reason","length","AggregateError","#mcpTools","McpToolSet","toolSets","all","Object","filter","entry","McpToolkitTool","isMcpToolkitTool","tool","mcpClient","server","catch","error","get","delete","create","toolSources","serverName","mcpTool","toolSet","toolName","isDisabledMcpTool","exposedName","prefix","existingServerName","set","sources","#mcpClient","existing","createPromise","toMCPClientConfig","next","then","transport","command","args","env","cwd","url","headers","redirect","ToolkitTool","Record","mcp","prototype","hasOwnProperty","call","disabled","getErrorMessage","message","String","entryName","action","cause","mcpToolName","find","AISDKToModelOutputOptions","Omit","Parameters","TArgs","TResult","output","toAISDKToModelOutput","toModelOutput","modelContent","parts","fromEntries","t","execute","description","inputSchema","providerOptions","callOptions","toolCallId","abortSignal","AbortSignal","human","ProviderToolkitTool","isProviderToolkitTool","id","providerId","supportsDeferredResults","Extract"],"sources":["../../src/tools/generativeTools.ts"],"sourcesContent":["import { jsonSchema, type ToolSet } from \"ai\";\nimport type { MCPClient, MCPClientConfig } from \"@ai-sdk/mcp\";\nimport { createMCPClient } from \"@ai-sdk/mcp\";\nimport { Experimental_StdioMCPTransport } from \"#mcp-stdio\";\nimport {\n  toJSONSchema,\n  type Tool,\n  type McpServerConfig,\n  type ToolModelOutputFunction,\n} from \"assistant-stream\";\nimport type {\n  McpToolkitToolConfig,\n  Toolkit,\n  ToolkitDefinition,\n} from \"@assistant-ui/core/react\";\nimport { frontendTools, type FrontendTools } from \"./frontendTools\";\nimport {\n  toAISDKContent,\n  toAISDKDefaultOutput,\n} from \"../converters/toolOutputConversion\";\nimport {\n  unwrapModelContentEnvelope,\n  type ModelContentEnvelope,\n} from \"../converters/modelContentEnvelope\";\n\nconst EMPTY_SCHEMA = { type: \"object\" as const, properties: {} };\n\nconst humanNotSupported = (): never => {\n  throw new Error(\n    \"`human()` is not available during server-side tool execution.\",\n  );\n};\n\n// AI SDK leaves `abortSignal` optional; assistant-ui's execute requires one.\nconst neverAbort = new AbortController().signal;\n\ntype MCPConnectionTimeoutPhase = \"connecting\" | \"listing tools\";\n\nclass MCPConnectionTimeoutError extends Error {}\n\nconst createMcpConnectionTimeoutError = (\n  name: string,\n  phase: MCPConnectionTimeoutPhase,\n  timeoutMs: number,\n) =>\n  new MCPConnectionTimeoutError(\n    `MCP toolkit entry \"${name}\" timed out while ${phase} after ${timeoutMs}ms.`,\n  );\n\nconst withMcpConnectionTimeout = async <T>(\n  promise: Promise<T>,\n  options: {\n    name: string;\n    config: McpServerConfig;\n    phase: MCPConnectionTimeoutPhase;\n    startedAt: number;\n  },\n): Promise<T> => {\n  const timeoutMs = options.config.connectionTimeout;\n  if (timeoutMs === undefined) return await promise;\n  const remainingMs = timeoutMs - (Date.now() - options.startedAt);\n  const timeoutError = () =>\n    createMcpConnectionTimeoutError(options.name, options.phase, timeoutMs);\n  if (remainingMs <= 0) throw timeoutError();\n\n  let timeout: ReturnType<typeof setTimeout> | undefined;\n  try {\n    return await Promise.race([\n      promise,\n      new Promise<never>((_, reject) => {\n        timeout = setTimeout(() => reject(timeoutError()), remainingMs);\n      }),\n    ]);\n  } finally {\n    if (timeout !== undefined) clearTimeout(timeout);\n  }\n};\n\nconst parametersToInputSchema = (parameters: Tool[\"parameters\"] | undefined) =>\n  jsonSchema(parameters ? toJSONSchema(parameters) : EMPTY_SCHEMA);\n\n/**\n * @deprecated Options for the deprecated {@link generativeTools}. Use\n * {@link AISDKToolkit} with {@link AISDKToolkitOptions} /\n * {@link AISDKToolkitToolsOptions} instead.\n */\nexport interface GenerativeToolsOptions {\n  /**\n   * The server build of a generative toolkit (schema + server `execute`). Typed\n   * as the canonical {@link Toolkit} so callers don't need to cast; the server\n   * build carries `execute`, recovered internally as {@link ToolkitDefinition}.\n   */\n  toolkit: Toolkit;\n  /**\n   * Tools uploaded by the frontend (the request body's `tools`). Merged in\n   * alongside the `toolkit`; a server `execute` from `toolkit` takes precedence\n   * over an uploaded entry of the same name.\n   */\n  frontendTools?: FrontendTools;\n}\n\nexport type AISDKToolkitOptions = {\n  toolkit: Toolkit;\n};\n\nexport type AISDKToolkitToolsOptions = {\n  /**\n   * Tools uploaded by the frontend request body.\n   */\n  frontend?: FrontendTools;\n};\n\n/**\n * Builds an AI SDK `ToolSet` for server-side use with `streamText` /\n * `generateText` from a generative `toolkit` and the frontend-uploaded tools.\n *\n * Each toolkit tool's `execute` runs on the server. Pair this with the\n * `\"use generative\"` compiler: import the toolkit in a server route (where it\n * resolves to the server build — schema + `execute`, with `render` stripped) and\n * pass it here. Tools without an `execute` are still exposed to the model but\n * left for the client to fulfill. `frontendTools` lets the client contribute\n * tools that aren't in the static toolkit.\n *\n * @deprecated Use {@link AISDKToolkit} instead:\n * `new AISDKToolkit({ toolkit }).tools({ frontend })`. It is a strict superset\n * (it also opens MCP server connections), so it replaces `generativeTools`\n * everywhere. The `frontendTools` option is named `frontend` on `.tools()`, and\n * `.tools()` is async. `generativeTools` will be removed in a future version.\n *\n * @example\n * ```ts\n * // Define once at module scope so any MCP connections pool across requests.\n * const aiToolkit = new AISDKToolkit({ toolkit: docsToolkit });\n *\n * // In your route handler:\n * const { tools } = await req.json();\n * streamText({\n *   model,\n *   messages,\n *   tools: await aiToolkit.tools({ frontend: tools }),\n * });\n * ```\n */\nexport const generativeTools = (options: GenerativeToolsOptions): ToolSet => {\n  assertNoMcpToolkitTools(options.toolkit);\n  return {\n    ...(options.frontendTools ? frontendTools(options.frontendTools) : {}),\n    // `toolkit` last so its server-side `execute` wins over an uploaded entry of\n    // the same name. The cast recovers the declaration shape — the server build\n    // carries `execute`, which the canonical `Toolkit` type erases.\n    ...toProviderToolSet(options.toolkit),\n    ...toServerToolSet(options.toolkit as ToolkitDefinition),\n  };\n};\n\nexport class AISDKToolkit {\n  readonly #toolkit: Toolkit;\n  readonly #mcpClients = new Map<string, Promise<MCPClient>>();\n\n  constructor(options: AISDKToolkitOptions) {\n    this.#toolkit = options.toolkit;\n  }\n\n  async tools(options: AISDKToolkitToolsOptions = {}): Promise<ToolSet> {\n    const frontendToolSet = options.frontend\n      ? frontendTools(options.frontend)\n      : {};\n    const mcpToolSet = await this.#mcpTools();\n    const providerToolSet = toProviderToolSet(this.#toolkit);\n    const serverToolSet = toServerToolSet(this.#toolkit as ToolkitDefinition);\n\n    assertNoMcpToolNameCollisions(mcpToolSet, [\n      { source: \"frontend\", tools: frontendToolSet },\n      { source: \"provider\", tools: providerToolSet },\n      { source: \"toolkit\", tools: serverToolSet },\n    ]);\n\n    return {\n      ...frontendToolSet,\n      ...mcpToolSet.tools,\n      ...providerToolSet,\n      ...serverToolSet,\n    };\n  }\n\n  async close(): Promise<void> {\n    const clientEntries = [...this.#mcpClients.entries()];\n    const clientNames = clientEntries.map(([name]) => name);\n    this.#mcpClients.clear();\n    const clientResults = await Promise.allSettled(\n      clientEntries.map(([, clientPromise]) => clientPromise),\n    );\n    const clients = clientResults.flatMap((result, index) =>\n      result.status === \"fulfilled\"\n        ? [[clientNames[index]!, result.value] as const]\n        : [],\n    );\n    const closeResults = await Promise.allSettled(\n      clients.map(([, client]) => client.close()),\n    );\n    const errors = [\n      ...clientResults.flatMap((result, index) =>\n        result.status === \"rejected\"\n          ? [toMcpToolkitError(clientNames[index]!, \"connect\", result.reason)]\n          : [],\n      ),\n      ...closeResults.flatMap((result, index) =>\n        result.status === \"rejected\"\n          ? [toMcpToolkitError(clients[index]![0], \"close\", result.reason)]\n          : [],\n      ),\n    ];\n    if (errors.length === 1) throw errors[0];\n    if (errors.length > 1) {\n      throw new AggregateError(\n        errors,\n        \"Failed to close one or more MCP clients\",\n      );\n    }\n  }\n\n  async #mcpTools(): Promise<McpToolSet> {\n    const toolSets = await Promise.all(\n      Object.entries(this.#toolkit)\n        .filter((entry): entry is [string, McpToolkitTool] =>\n          isMcpToolkitTool(entry[1]),\n        )\n        .map(async ([name, tool]) => {\n          const startedAt = Date.now();\n          const clientPromise = this.#mcpClient(name, tool.server, startedAt);\n          const client = await clientPromise.catch((error: unknown) => {\n            if (error instanceof MCPConnectionTimeoutError) throw error;\n            throw toMcpToolkitError(name, \"connect\", error);\n          });\n          try {\n            const tools = await withMcpConnectionTimeout(client.tools(), {\n              name,\n              config: tool.server,\n              phase: \"listing tools\",\n              startedAt,\n            });\n            return [name, tool, tools] as const;\n          } catch (error) {\n            if (error instanceof MCPConnectionTimeoutError) {\n              if (this.#mcpClients.get(name) === clientPromise) {\n                this.#mcpClients.delete(name);\n                void client.close().catch(() => {});\n              }\n              throw error;\n            }\n            throw toMcpToolkitError(name, \"list tools\", error);\n          }\n        }),\n    );\n\n    const tools = Object.create(null) as ToolSet;\n    const toolSources = new Map<string, string>();\n    for (const [serverName, mcpTool, toolSet] of toolSets) {\n      for (const [toolName, tool] of Object.entries(toolSet)) {\n        if (isDisabledMcpTool(mcpTool.tools?.[toolName])) continue;\n        const exposedName = `${mcpTool.prefix ?? \"\"}${toolName}`;\n        const existingServerName = toolSources.get(exposedName);\n        if (existingServerName) {\n          throw new Error(\n            `MCP tool name collision: \"${exposedName}\" is exposed by both \"${existingServerName}\" and \"${serverName}\". Rename one of the toolkit entries or expose distinct MCP tool names.`,\n          );\n        }\n        toolSources.set(exposedName, serverName);\n        tools[exposedName] = tool as ToolSet[string];\n      }\n    }\n    return { tools, sources: toolSources };\n  }\n\n  #mcpClient(\n    name: string,\n    config: McpServerConfig,\n    startedAt: number,\n  ): Promise<MCPClient> {\n    const existing = this.#mcpClients.get(name);\n    if (existing) return existing;\n    const createPromise = createMCPClient(toMCPClientConfig(config));\n    let next: Promise<MCPClient>;\n    next = withMcpConnectionTimeout(createPromise, {\n      name,\n      config,\n      phase: \"connecting\",\n      startedAt,\n    }).catch((error) => {\n      if (this.#mcpClients.get(name) === next) {\n        this.#mcpClients.delete(name);\n      }\n      if (error instanceof MCPConnectionTimeoutError) {\n        void createPromise.then((client) => client.close()).catch(() => {});\n      }\n      throw error;\n    });\n    this.#mcpClients.set(name, next);\n    return next;\n  }\n}\n\nconst toMCPClientConfig = (config: McpServerConfig): MCPClientConfig => {\n  if (config.type === \"stdio\") {\n    return {\n      transport: new Experimental_StdioMCPTransport({\n        command: config.command,\n        ...(config.args && { args: [...config.args] }),\n        ...(config.env && { env: config.env }),\n        ...(config.cwd && { cwd: config.cwd }),\n      }),\n    };\n  }\n\n  return {\n    transport: {\n      type: config.type,\n      url: config.url,\n      ...(config.headers && { headers: config.headers }),\n      ...(config.redirect && { redirect: config.redirect }),\n    },\n  };\n};\n\ntype ToolkitTool = Toolkit[string];\n\ntype McpToolkitTool = ToolkitTool & {\n  type: \"mcp\";\n  server: McpServerConfig;\n  prefix?: string | undefined;\n  tools?: Record<string, McpToolkitToolConfig> | undefined;\n};\n\ntype McpToolSet = {\n  tools: ToolSet;\n  sources: Map<string, string>;\n};\n\nconst assertNoMcpToolNameCollisions = (\n  mcp: McpToolSet,\n  toolSets: readonly { source: string; tools: ToolSet }[],\n): void => {\n  for (const [toolName, serverName] of mcp.sources) {\n    for (const { source, tools } of toolSets) {\n      if (!Object.prototype.hasOwnProperty.call(tools, toolName)) continue;\n      throw new Error(\n        `MCP tool \"${toolName}\" from \"${serverName}\" conflicts with ${source} tool \"${toolName}\". Rename one of the tools so each model-visible tool name is unique.`,\n      );\n    }\n  }\n};\n\nconst isMcpToolkitTool = (tool: ToolkitTool): tool is McpToolkitTool =>\n  tool.type === \"mcp\" && !tool.disabled;\n\nconst getErrorMessage = (error: unknown): string =>\n  error instanceof Error ? error.message || error.name : String(error);\n\nconst toMcpToolkitError = (\n  entryName: string,\n  action: \"connect\" | \"list tools\" | \"close\",\n  error: unknown,\n): Error => {\n  return new Error(\n    `MCP toolkit entry \"${entryName}\" failed to ${action}: ${getErrorMessage(error)}`,\n    { cause: error },\n  );\n};\n\nconst isDisabledMcpTool = (config: McpToolkitToolConfig | undefined): boolean =>\n  config?.disabled === true;\n\nconst assertNoMcpToolkitTools = (toolkit: Toolkit): void => {\n  const mcpToolName = Object.entries(toolkit).find(([, tool]) =>\n    isMcpToolkitTool(tool),\n  )?.[0];\n  if (!mcpToolName) return;\n\n  throw new Error(\n    `MCP toolkit entry \"${mcpToolName}\" requires AISDKToolkit. Use new AISDKToolkit({ toolkit }).tools(...) instead of generativeTools(...).`,\n  );\n};\n\ntype AISDKToModelOutputOptions<TArgs, TResult> = Omit<\n  Parameters<ToolModelOutputFunction<TArgs, TResult>>[0],\n  \"output\"\n> & {\n  output: TResult | ModelContentEnvelope<TResult>;\n};\n\nconst toAISDKToModelOutput =\n  <TArgs, TResult>(toModelOutput?: ToolModelOutputFunction<TArgs, TResult>) =>\n  async (options: AISDKToModelOutputOptions<TArgs, TResult>) => {\n    const { result, modelContent } = unwrapModelContentEnvelope(options.output);\n\n    if (modelContent !== undefined) {\n      return toAISDKContent(modelContent);\n    }\n\n    if (!toModelOutput) {\n      return toAISDKDefaultOutput(result);\n    }\n\n    const parts = await toModelOutput({\n      ...options,\n      output: result,\n    });\n    return toAISDKContent(parts);\n  };\n\nconst toServerToolSet = (toolkit: ToolkitDefinition): ToolSet =>\n  Object.fromEntries(\n    Object.entries(toolkit)\n      .filter(\n        ([, t]) => t.type !== \"mcp\" && t.type !== \"provider\" && !t.disabled,\n      )\n      .map(([name, t]) => {\n        const execute = t.execute;\n        return [\n          name,\n          {\n            ...(t.description !== undefined && { description: t.description }),\n            inputSchema: parametersToInputSchema(t.parameters),\n            toModelOutput: toAISDKToModelOutput(t.toModelOutput),\n            ...(t.providerOptions && { providerOptions: t.providerOptions }),\n            ...(execute && {\n              execute: (\n                args: unknown,\n                callOptions: { toolCallId: string; abortSignal?: AbortSignal },\n              ) =>\n                execute(args as never, {\n                  toolCallId: callOptions.toolCallId,\n                  abortSignal: callOptions.abortSignal ?? neverAbort,\n                  human: humanNotSupported,\n                }),\n            }),\n          },\n        ];\n      }),\n  ) as ToolSet;\n\nconst toProviderToolSet = (toolkit: Toolkit): ToolSet =>\n  Object.fromEntries(\n    Object.entries(toolkit)\n      .filter((entry): entry is [string, ProviderToolkitTool] =>\n        isProviderToolkitTool(entry[1]),\n      )\n      .map(([name, t]) => [\n        name,\n        {\n          type: \"provider\",\n          id: t.providerId,\n          args: t.args,\n          ...(t.parameters && {\n            inputSchema: parametersToInputSchema(t.parameters),\n          }),\n          ...(t.providerOptions && { providerOptions: t.providerOptions }),\n          ...(t.supportsDeferredResults !== undefined && {\n            supportsDeferredResults: t.supportsDeferredResults,\n          }),\n        },\n      ]),\n  ) as ToolSet;\n\ntype ProviderToolkitTool = Extract<Toolkit[string], { type: \"provider\" }>;\n\nconst isProviderToolkitTool = (\n  tool: Toolkit[string],\n): tool is ProviderToolkitTool => tool.type === \"provider\" && !tool.disabled;\n"],"mappings":";;;;;;;;AAyBA,MAAMmB,eAAe;CAAEC,MAAM;CAAmBE,YAAY,CAAC;AAAE;AAE/D,MAAMC,0BAAiC;CACrC,MAAM,IAAIC,MACR,+DACF;AACF;AAGA,MAAMC,aAAa,IAAIC,gBAAgB,CAAC,CAACC;AAIzC,IAAME,4BAAN,cAAwCL,MAAM,CAAA;AAE9C,MAAMM,mCACJC,MACAC,OACAC,cAEA,IAAIJ,0BACF,sBAAsBE,KAAI,oBAAqBC,MAAK,SAAUC,UAAS,IACzE;AAEF,MAAMC,2BAA2B,OAC/BC,SACAG,YAMe;CACf,MAAML,YAAYK,QAAQC,OAAOE;CACjC,IAAIR,cAAcS,KAAAA,GAAW,OAAO,MAAMP;CAC1C,MAAMQ,cAAcV,aAAaW,KAAKC,IAAI,IAAIP,QAAQE;CACtD,MAAMM,qBACJhB,gCAAgCQ,QAAQP,MAAMO,QAAQN,OAAOC,SAAS;CACxE,IAAIU,eAAe,GAAG,MAAMG,aAAa;CAEzC,IAAIC;CACJ,IAAI;EACF,OAAO,MAAMX,QAAQc,KAAK,CACxBf,SACA,IAAIC,SAAgBe,GAAGC,WAAW;GAChCL,UAAUE,iBAAiBG,OAAON,aAAa,CAAC,GAAGH,WAAW;EAChE,CAAC,CAAC,CACH;CACH,UAAU;EACR,IAAII,YAAYL,KAAAA,GAAWW,aAAaN,OAAO;CACjD;AACF;AAEA,MAAMO,2BAA2BC,eAC/BvD,WAAWuD,aAAajD,aAAaiD,UAAU,IAAIpC,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgEjE,MAAa0C,mBAAmBvB,YAA6C;CAC3EwB,wBAAwBxB,QAAQmB,OAAO;CACvC,OAAO;EACL,GAAInB,QAAQzB,gBAAgBA,cAAcyB,QAAQzB,aAAa,IAAI,CAAC;EAIpE,GAAGkD,kBAAkBzB,QAAQmB,OAAO;EACpC,GAAGO,gBAAgB1B,QAAQmB,OAA4B;CACzD;AACF;AAEA,IAAaQ,eAAb,MAA0B;CACxB;CACA,8BAAuB,IAAIE,IAAgC;CAE3DC,YAAY9B,SAA8B;EACxC,KAAK,WAAWA,QAAQmB;CAC1B;CAEA,MAAMY,MAAM/B,UAAoC,CAAC,GAAqB;EACpE,MAAMgC,kBAAkBhC,QAAQsB,WAC5B/C,cAAcyB,QAAQsB,QAAQ,IAC9B,CAAC;EACL,MAAMW,aAAa,MAAM,KAAK,UAAU;EACxC,MAAME,kBAAkBV,kBAAkB,KAAK,QAAQ;EACvD,MAAMW,gBAAgBV,gBAAgB,KAAK,QAA6B;EAExEW,8BAA8BJ,YAAY;GACxC;IAAEK,QAAQ;IAAYP,OAAOC;GAAgB;GAC7C;IAAEM,QAAQ;IAAYP,OAAOI;GAAgB;GAC7C;IAAEG,QAAQ;IAAWP,OAAOK;GAAc;EAAC,CAC5C;EAED,OAAO;GACL,GAAGJ;GACH,GAAGC,WAAWF;GACd,GAAGI;GACH,GAAGC;EACL;CACF;CAEA,MAAMG,QAAuB;EAC3B,MAAMC,gBAAgB,CAAC,GAAG,KAAK,YAAYC,QAAQ,CAAC;EACpD,MAAMC,cAAcF,cAAcG,KAAK,CAAClD,UAAUA,IAAI;EACtD,KAAK,YAAYmD,MAAM;EACvB,MAAMC,gBAAgB,MAAM/C,QAAQgD,WAClCN,cAAcG,KAAK,GAAGI,mBAAmBA,aAAa,CACxD;EACA,MAAMC,UAAUH,cAAcI,SAASC,QAAQC,UAC7CD,OAAOE,WAAW,cACd,CAAC,CAACV,YAAYS,QAASD,OAAOG,KAAK,CAAU,IAC7C,CAAA,CACN;EACA,MAAMC,eAAe,MAAMxD,QAAQgD,WACjCE,QAAQL,KAAK,GAAGY,YAAYA,OAAOhB,MAAM,CAAC,CAC5C;EACA,MAAMiB,SAAS,CACb,GAAGX,cAAcI,SAASC,QAAQC,UAChCD,OAAOE,WAAW,aACd,CAACK,kBAAkBf,YAAYS,QAAS,WAAWD,OAAOQ,MAAM,CAAC,IACjE,CAAA,CACN,GACA,GAAGJ,aAAaL,SAASC,QAAQC,UAC/BD,OAAOE,WAAW,aACd,CAACK,kBAAkBT,QAAQG,MAAM,CAAE,IAAI,SAASD,OAAOQ,MAAM,CAAC,IAC9D,CAAA,CACN,CAAC;EAEH,IAAIF,OAAOG,WAAW,GAAG,MAAMH,OAAO;EACtC,IAAIA,OAAOG,SAAS,GAClB,MAAM,IAAIC,eACRJ,QACA,yCACF;CAEJ;CAEA,MAAM,YAAiC;EACrC,MAAMO,WAAW,MAAMjE,QAAQkE,IAC7BC,OAAOxB,QAAQ,KAAK,QAAQ,CAAC,CAC1ByB,QAAQC,UACPE,iBAAiBF,MAAM,EAAE,CAC3B,CAAC,CACAxB,IAAI,OAAO,CAAClD,MAAM6E,UAAU;GAC3B,MAAMpE,YAAYI,KAAKC,IAAI;GAC3B,MAAMwC,gBAAgB,KAAK,WAAWtD,MAAM6E,KAAKE,QAAQtE,SAAS;GAClE,MAAMqD,SAAS,MAAMR,cAAc0B,OAAOC,UAAmB;IAC3D,IAAIA,iBAAiBnF,2BAA2B,MAAMmF;IACtD,MAAMjB,kBAAkBhE,MAAM,WAAWiF,KAAK;GAChD,CAAC;GACD,IAAI;IAOF,OAAO;KAACjF;KAAM6E;KAAMvC,MANAnC,yBAAyB2D,OAAOxB,MAAM,GAAG;MAC3DtC;MACAQ,QAAQqE,KAAKE;MACb9E,OAAO;MACPQ;KACF,CAAC;IACwB;GAC3B,SAASwE,OAAO;IACd,IAAIA,iBAAiBnF,2BAA2B;KAC9C,IAAI,KAAK,YAAYoF,IAAIlF,IAAI,MAAMsD,eAAe;MAChD,KAAK,YAAY6B,OAAOnF,IAAI;MAC5B,OAAY8C,MAAM,CAAC,CAACkC,YAAY,CAAC,CAAC;KACpC;KACA,MAAMC;IACR;IACA,MAAMjB,kBAAkBhE,MAAM,cAAciF,KAAK;GACnD;EACF,CAAC,CACL;EAEA,MAAM3C,QAAQkC,OAAOY,OAAO,IAAI;EAChC,MAAMC,8BAAc,IAAIjD,IAAoB;EAC5C,KAAK,MAAM,CAACkD,YAAYC,SAASC,YAAYlB,UAC3C,KAAK,MAAM,CAACmB,UAAUZ,SAASL,OAAOxB,QAAQwC,OAAO,GAAG;GACtD,IAAIE,kBAAkBH,QAAQjD,QAAQmD,SAAS,GAAG;GAClD,MAAME,cAAc,GAAGJ,QAAQK,UAAU,KAAKH;GAC9C,MAAMI,qBAAqBR,YAAYH,IAAIS,WAAW;GACtD,IAAIE,oBACF,MAAM,IAAIpG,MACR,6BAA6BkG,YAAW,wBAAyBE,mBAAkB,SAAUP,WAAU,wEACzG;GAEFD,YAAYS,IAAIH,aAAaL,UAAU;GACvChD,MAAMqD,eAAed;EACvB;EAEF,OAAO;GAAEvC;GAAOyD,SAASV;EAAY;CACvC;CAEA,WACErF,MACAQ,QACAC,WACoB;EACpB,MAAMwF,WAAW,KAAK,YAAYf,IAAIlF,IAAI;EAC1C,IAAIiG,UAAU,OAAOA;EACrB,MAAMC,gBAAgB7H,gBAAgB8H,kBAAkB3F,MAAM,CAAC;EAC/D,IAAI4F;EACJA,OAAOjG,yBAAyB+F,eAAe;GAC7ClG;GACAQ;GACAP,OAAO;GACPQ;EACF,CAAC,CAAC,CAACuE,OAAOC,UAAU;GAClB,IAAI,KAAK,YAAYC,IAAIlF,IAAI,MAAMoG,MACjC,KAAK,YAAYjB,OAAOnF,IAAI;GAE9B,IAAIiF,iBAAiBnF,2BACnB,cAAmBuG,MAAMvC,WAAWA,OAAOhB,MAAM,CAAC,CAAC,CAACkC,YAAY,CAAC,CAAC;GAEpE,MAAMC;EACR,CAAC;EACD,KAAK,YAAYa,IAAI9F,MAAMoG,IAAI;EAC/B,OAAOA;CACT;AACF;AAEA,MAAMD,qBAAqB3F,WAA6C;CACtE,IAAIA,OAAOnB,SAAS,SAClB,OAAO,EACLiH,WAAW,IAAIhI,+BAA+B;EAC5CiI,SAAS/F,OAAO+F;EAChB,GAAI/F,OAAOgG,QAAQ,EAAEA,MAAM,CAAC,GAAGhG,OAAOgG,IAAI,EAAE;EAC5C,GAAIhG,OAAOiG,OAAO,EAAEA,KAAKjG,OAAOiG,IAAI;EACpC,GAAIjG,OAAOkG,OAAO,EAAEA,KAAKlG,OAAOkG,IAAI;CACtC,CAAC,EACH;CAGF,OAAO,EACLJ,WAAW;EACTjH,MAAMmB,OAAOnB;EACbsH,KAAKnG,OAAOmG;EACZ,GAAInG,OAAOoG,WAAW,EAAEA,SAASpG,OAAOoG,QAAQ;EAChD,GAAIpG,OAAOqG,YAAY,EAAEA,UAAUrG,OAAOqG,SAAS;CACrD,EACF;AACF;AAgBA,MAAMjE,iCACJoE,KACA1C,aACS;CACT,KAAK,MAAM,CAACmB,UAAUH,eAAe0B,IAAIjB,SACvC,KAAK,MAAM,EAAElD,QAAQP,WAAWgC,UAAU;EACxC,IAAI,CAACE,OAAOyC,UAAUC,eAAeC,KAAK7E,OAAOmD,QAAQ,GAAG;EAC5D,MAAM,IAAIhG,MACR,aAAagG,SAAQ,UAAWH,WAAU,mBAAoBzC,OAAM,SAAU4C,SAAQ,sEACxF;CACF;AAEJ;AAEA,MAAMb,oBAAoBC,SACxBA,KAAKxF,SAAS,SAAS,CAACwF,KAAKuC;AAE/B,MAAMC,mBAAmBpC,UACvBA,iBAAiBxF,QAAQwF,MAAMqC,WAAWrC,MAAMjF,OAAOuH,OAAOtC,KAAK;AAErE,MAAMjB,qBACJwD,WACAC,QACAxC,UACU;CACV,OAAO,IAAIxF,MACT,sBAAsB+H,UAAS,cAAeC,OAAM,IAAKJ,gBAAgBpC,KAAK,KAC9E,EAAEyC,OAAOzC,MAAM,CACjB;AACF;AAEA,MAAMS,qBAAqBlF,WACzBA,QAAQ4G,aAAa;AAEvB,MAAMrF,2BAA2BL,YAA2B;CAC1D,MAAMiG,cAAcnD,OAAOxB,QAAQtB,OAAO,CAAC,CAACkG,MAAM,GAAG/C,UACnDD,iBAAiBC,IAAI,CACvB,CAAC,GAAG;CACJ,IAAI,CAAC8C,aAAa;CAElB,MAAM,IAAIlI,MACR,sBAAsBkI,YAAW,uGACnC;AACF;AASA,MAAMQ,wBACaC,kBACjB,OAAO7H,YAAuD;CAC5D,MAAM,EAAEkD,QAAQ4E,iBAAiBnJ,2BAA2BqB,QAAQ2H,MAAM;CAE1E,IAAIG,iBAAiB1H,KAAAA,GACnB,OAAO3B,eAAeqJ,YAAY;CAGpC,IAAI,CAACD,eACH,OAAOnJ,qBAAqBwE,MAAM;CAGpC,MAAM6E,QAAQ,MAAMF,cAAc;EAChC,GAAG7H;EACH2H,QAAQzE;CACV,CAAC;CACD,OAAOzE,eAAesJ,KAAK;AAC7B;AAEF,MAAMrG,mBAAmBP,YACvB8C,OAAO+D,YACL/D,OAAOxB,QAAQtB,OAAO,CAAC,CACpB+C,QACE,GAAG+D,OAAOA,EAAEnJ,SAAS,SAASmJ,EAAEnJ,SAAS,cAAc,CAACmJ,EAAEpB,QAC7D,CAAC,CACAlE,KAAK,CAAClD,MAAMwI,OAAO;CAClB,MAAMC,UAAUD,EAAEC;CAClB,OAAO,CACLzI,MACA;EACE,GAAIwI,EAAEE,gBAAgB/H,KAAAA,KAAa,EAAE+H,aAAaF,EAAEE,YAAY;EAChEC,aAAapH,wBAAwBiH,EAAEhH,UAAU;EACjD4G,eAAeD,qBAAqBK,EAAEJ,aAAa;EACnD,GAAII,EAAEI,mBAAmB,EAAEA,iBAAiBJ,EAAEI,gBAAgB;EAC9D,GAAIH,WAAW,EACbA,UACEjC,MACAqC,gBAEAJ,QAAQjC,MAAe;GACrBsC,YAAYD,YAAYC;GACxBC,aAAaF,YAAYE,eAAerJ;GACxCuJ,OAAOzJ;EACT,CAAC,EACL;CACF,CAAC;AAEL,CAAC,CACL;AAEF,MAAMwC,qBAAqBN,YACzB8C,OAAO+D,YACL/D,OAAOxB,QAAQtB,OAAO,CAAC,CACpB+C,QAAQC,UACPyE,sBAAsBzE,MAAM,EAAE,CAChC,CAAC,CACAxB,KAAK,CAAClD,MAAMwI,OAAO,CAClBxI,MACA;CACEX,MAAM;CACN+J,IAAIZ,EAAEa;CACN7C,MAAMgC,EAAEhC;CACR,GAAIgC,EAAEhH,cAAc,EAClBmH,aAAapH,wBAAwBiH,EAAEhH,UAAU,EACnD;CACA,GAAIgH,EAAEI,mBAAmB,EAAEA,iBAAiBJ,EAAEI,gBAAgB;CAC9D,GAAIJ,EAAEc,4BAA4B3I,KAAAA,KAAa,EAC7C2I,yBAAyBd,EAAEc,wBAC7B;AACF,CAAC,CACF,CACL;AAIF,MAAMH,yBACJtE,SACgCA,KAAKxF,SAAS,cAAc,CAACwF,KAAKuC"}
+{"version":3,"file":"generativeTools.js","names":["jsonSchema","ToolSet","MCPClient","MCPClientConfig","createMCPClient","Experimental_StdioMCPTransport","toJSONSchema","Tool","McpServerConfig","ToolModelOutputFunction","McpToolkitToolConfig","Toolkit","ToolkitDefinition","frontendTools","FrontendTools","toAISDKContent","toAISDKDefaultOutput","unwrapModelContentEnvelope","ModelContentEnvelope","EMPTY_SCHEMA","type","const","properties","humanNotSupported","Error","neverAbort","AbortController","signal","MCPConnectionTimeoutPhase","MCPConnectionTimeoutError","createMcpConnectionTimeoutError","name","phase","timeoutMs","withMcpConnectionTimeout","promise","Promise","T","options","config","startedAt","connectionTimeout","undefined","remainingMs","Date","now","timeoutError","timeout","ReturnType","setTimeout","race","_","reject","clearTimeout","convertedParameterSchemas","WeakMap","getOrConvertParameterSchema","parameters","NonNullable","cached","get","converted","set","parametersToInputSchema","GenerativeToolsOptions","toolkit","AISDKToolkitOptions","AISDKToolkitToolsOptions","frontend","generativeTools","assertNoMcpToolkitTools","toProviderToolSet","toServerToolSet","AISDKToolkit","mcpClients","Map","constructor","tools","frontendToolSet","mcpToolSet","mcpTools","providerToolSet","serverToolSet","assertNoMcpToolNameCollisions","source","close","clientEntries","entries","clientNames","map","clear","clientResults","allSettled","clientPromise","clients","flatMap","result","index","status","value","closeResults","client","errors","toMcpToolkitError","reason","length","AggregateError","#mcpTools","McpToolSet","toolSets","all","Object","filter","entry","McpToolkitTool","isMcpToolkitTool","tool","mcpClient","server","catch","error","delete","create","toolSources","serverName","mcpTool","toolSet","toolName","isDisabledMcpTool","exposedName","prefix","existingServerName","sources","#mcpClient","existing","createPromise","toMCPClientConfig","next","then","transport","command","args","env","cwd","url","headers","redirect","ToolkitTool","Record","mcp","prototype","hasOwnProperty","call","disabled","getErrorMessage","message","String","entryName","action","cause","mcpToolName","find","AISDKToModelOutputOptions","Omit","Parameters","TArgs","TResult","output","toAISDKToModelOutput","toModelOutput","modelContent","parts","fromEntries","t","execute","description","inputSchema","providerOptions","callOptions","toolCallId","abortSignal","AbortSignal","human","ProviderToolkitTool","isProviderToolkitTool","id","providerId","supportsDeferredResults","Extract"],"sources":["../../src/tools/generativeTools.ts"],"sourcesContent":["import { jsonSchema, type ToolSet } from \"ai\";\nimport type { MCPClient, MCPClientConfig } from \"@ai-sdk/mcp\";\nimport { createMCPClient } from \"@ai-sdk/mcp\";\nimport { Experimental_StdioMCPTransport } from \"#mcp-stdio\";\nimport {\n  toJSONSchema,\n  type Tool,\n  type McpServerConfig,\n  type ToolModelOutputFunction,\n} from \"assistant-stream\";\nimport type {\n  McpToolkitToolConfig,\n  Toolkit,\n  ToolkitDefinition,\n} from \"@assistant-ui/core/react\";\nimport { frontendTools, type FrontendTools } from \"./frontendTools\";\nimport {\n  toAISDKContent,\n  toAISDKDefaultOutput,\n} from \"../converters/toolOutputConversion\";\nimport {\n  unwrapModelContentEnvelope,\n  type ModelContentEnvelope,\n} from \"../converters/modelContentEnvelope\";\n\nconst EMPTY_SCHEMA = { type: \"object\" as const, properties: {} };\n\nconst humanNotSupported = (): never => {\n  throw new Error(\n    \"`human()` is not available during server-side tool execution.\",\n  );\n};\n\n// AI SDK leaves `abortSignal` optional; assistant-ui's execute requires one.\nconst neverAbort = new AbortController().signal;\n\ntype MCPConnectionTimeoutPhase = \"connecting\" | \"listing tools\";\n\nclass MCPConnectionTimeoutError extends Error {}\n\nconst createMcpConnectionTimeoutError = (\n  name: string,\n  phase: MCPConnectionTimeoutPhase,\n  timeoutMs: number,\n) =>\n  new MCPConnectionTimeoutError(\n    `MCP toolkit entry \"${name}\" timed out while ${phase} after ${timeoutMs}ms.`,\n  );\n\nconst withMcpConnectionTimeout = async <T>(\n  promise: Promise<T>,\n  options: {\n    name: string;\n    config: McpServerConfig;\n    phase: MCPConnectionTimeoutPhase;\n    startedAt: number;\n  },\n): Promise<T> => {\n  const timeoutMs = options.config.connectionTimeout;\n  if (timeoutMs === undefined) return await promise;\n  const remainingMs = timeoutMs - (Date.now() - options.startedAt);\n  const timeoutError = () =>\n    createMcpConnectionTimeoutError(options.name, options.phase, timeoutMs);\n  if (remainingMs <= 0) throw timeoutError();\n\n  let timeout: ReturnType<typeof setTimeout> | undefined;\n  try {\n    return await Promise.race([\n      promise,\n      new Promise<never>((_, reject) => {\n        timeout = setTimeout(() => reject(timeoutError()), remainingMs);\n      }),\n    ]);\n  } finally {\n    if (timeout !== undefined) clearTimeout(timeout);\n  }\n};\n\n// Converted schemas are shared by parameter identity and treated as immutable.\nconst convertedParameterSchemas = new WeakMap<\n  object,\n  ReturnType<typeof toJSONSchema>\n>();\n\nconst getOrConvertParameterSchema = (\n  parameters: NonNullable<Tool[\"parameters\"]>,\n) => {\n  const cached = convertedParameterSchemas.get(parameters);\n  if (cached) return cached;\n\n  const converted = toJSONSchema(parameters);\n  convertedParameterSchemas.set(parameters, converted);\n  return converted;\n};\n\nconst parametersToInputSchema = (\n  parameters: Tool[\"parameters\"] | undefined,\n) => {\n  if (!parameters) return jsonSchema(EMPTY_SCHEMA);\n  return jsonSchema(getOrConvertParameterSchema(parameters));\n};\n\n/**\n * @deprecated Options for the deprecated {@link generativeTools}. Use\n * {@link AISDKToolkit} with {@link AISDKToolkitOptions} /\n * {@link AISDKToolkitToolsOptions} instead.\n */\nexport interface GenerativeToolsOptions {\n  /**\n   * The server build of a generative toolkit (schema + server `execute`). Typed\n   * as the canonical {@link Toolkit} so callers don't need to cast; the server\n   * build carries `execute`, recovered internally as {@link ToolkitDefinition}.\n   */\n  toolkit: Toolkit;\n  /**\n   * Tools uploaded by the frontend (the request body's `tools`). Merged in\n   * alongside the `toolkit`; a server `execute` from `toolkit` takes precedence\n   * over an uploaded entry of the same name.\n   */\n  frontendTools?: FrontendTools;\n}\n\nexport type AISDKToolkitOptions = {\n  toolkit: Toolkit;\n};\n\nexport type AISDKToolkitToolsOptions = {\n  /**\n   * Tools uploaded by the frontend request body.\n   */\n  frontend?: FrontendTools;\n};\n\n/**\n * Builds an AI SDK `ToolSet` for server-side use with `streamText` /\n * `generateText` from a generative `toolkit` and the frontend-uploaded tools.\n *\n * Each toolkit tool's `execute` runs on the server. Pair this with the\n * `\"use generative\"` compiler: import the toolkit in a server route (where it\n * resolves to the server build — schema + `execute`, with `render` stripped) and\n * pass it here. Tools without an `execute` are still exposed to the model but\n * left for the client to fulfill. `frontendTools` lets the client contribute\n * tools that aren't in the static toolkit.\n *\n * @deprecated Use {@link AISDKToolkit} instead:\n * `new AISDKToolkit({ toolkit }).tools({ frontend })`. It is a strict superset\n * (it also opens MCP server connections), so it replaces `generativeTools`\n * everywhere. The `frontendTools` option is named `frontend` on `.tools()`, and\n * `.tools()` is async. `generativeTools` will be removed in a future version.\n *\n * @example\n * ```ts\n * // Define once at module scope so any MCP connections pool across requests.\n * const aiToolkit = new AISDKToolkit({ toolkit: docsToolkit });\n *\n * // In your route handler:\n * const { tools } = await req.json();\n * streamText({\n *   model,\n *   messages,\n *   tools: await aiToolkit.tools({ frontend: tools }),\n * });\n * ```\n */\nexport const generativeTools = (options: GenerativeToolsOptions): ToolSet => {\n  assertNoMcpToolkitTools(options.toolkit);\n  return {\n    ...(options.frontendTools ? frontendTools(options.frontendTools) : {}),\n    // `toolkit` last so its server-side `execute` wins over an uploaded entry of\n    // the same name. The cast recovers the declaration shape — the server build\n    // carries `execute`, which the canonical `Toolkit` type erases.\n    ...toProviderToolSet(options.toolkit),\n    ...toServerToolSet(options.toolkit as ToolkitDefinition),\n  };\n};\n\nexport class AISDKToolkit {\n  readonly #toolkit: Toolkit;\n  readonly #mcpClients = new Map<string, Promise<MCPClient>>();\n\n  constructor(options: AISDKToolkitOptions) {\n    this.#toolkit = options.toolkit;\n  }\n\n  async tools(options: AISDKToolkitToolsOptions = {}): Promise<ToolSet> {\n    const frontendToolSet = options.frontend\n      ? frontendTools(options.frontend)\n      : {};\n    const mcpToolSet = await this.#mcpTools();\n    const providerToolSet = toProviderToolSet(this.#toolkit);\n    const serverToolSet = toServerToolSet(this.#toolkit as ToolkitDefinition);\n\n    assertNoMcpToolNameCollisions(mcpToolSet, [\n      { source: \"frontend\", tools: frontendToolSet },\n      { source: \"provider\", tools: providerToolSet },\n      { source: \"toolkit\", tools: serverToolSet },\n    ]);\n\n    return {\n      ...frontendToolSet,\n      ...mcpToolSet.tools,\n      ...providerToolSet,\n      ...serverToolSet,\n    };\n  }\n\n  async close(): Promise<void> {\n    const clientEntries = [...this.#mcpClients.entries()];\n    const clientNames = clientEntries.map(([name]) => name);\n    this.#mcpClients.clear();\n    const clientResults = await Promise.allSettled(\n      clientEntries.map(([, clientPromise]) => clientPromise),\n    );\n    const clients = clientResults.flatMap((result, index) =>\n      result.status === \"fulfilled\"\n        ? [[clientNames[index]!, result.value] as const]\n        : [],\n    );\n    const closeResults = await Promise.allSettled(\n      clients.map(([, client]) => client.close()),\n    );\n    const errors = [\n      ...clientResults.flatMap((result, index) =>\n        result.status === \"rejected\"\n          ? [toMcpToolkitError(clientNames[index]!, \"connect\", result.reason)]\n          : [],\n      ),\n      ...closeResults.flatMap((result, index) =>\n        result.status === \"rejected\"\n          ? [toMcpToolkitError(clients[index]![0], \"close\", result.reason)]\n          : [],\n      ),\n    ];\n    if (errors.length === 1) throw errors[0];\n    if (errors.length > 1) {\n      throw new AggregateError(\n        errors,\n        \"Failed to close one or more MCP clients\",\n      );\n    }\n  }\n\n  async #mcpTools(): Promise<McpToolSet> {\n    const toolSets = await Promise.all(\n      Object.entries(this.#toolkit)\n        .filter((entry): entry is [string, McpToolkitTool] =>\n          isMcpToolkitTool(entry[1]),\n        )\n        .map(async ([name, tool]) => {\n          const startedAt = Date.now();\n          const clientPromise = this.#mcpClient(name, tool.server, startedAt);\n          const client = await clientPromise.catch((error: unknown) => {\n            if (error instanceof MCPConnectionTimeoutError) throw error;\n            throw toMcpToolkitError(name, \"connect\", error);\n          });\n          try {\n            const tools = await withMcpConnectionTimeout(client.tools(), {\n              name,\n              config: tool.server,\n              phase: \"listing tools\",\n              startedAt,\n            });\n            return [name, tool, tools] as const;\n          } catch (error) {\n            if (error instanceof MCPConnectionTimeoutError) {\n              if (this.#mcpClients.get(name) === clientPromise) {\n                this.#mcpClients.delete(name);\n                void client.close().catch(() => {});\n              }\n              throw error;\n            }\n            throw toMcpToolkitError(name, \"list tools\", error);\n          }\n        }),\n    );\n\n    const tools = Object.create(null) as ToolSet;\n    const toolSources = new Map<string, string>();\n    for (const [serverName, mcpTool, toolSet] of toolSets) {\n      for (const [toolName, tool] of Object.entries(toolSet)) {\n        if (isDisabledMcpTool(mcpTool.tools?.[toolName])) continue;\n        const exposedName = `${mcpTool.prefix ?? \"\"}${toolName}`;\n        const existingServerName = toolSources.get(exposedName);\n        if (existingServerName) {\n          throw new Error(\n            `MCP tool name collision: \"${exposedName}\" is exposed by both \"${existingServerName}\" and \"${serverName}\". Rename one of the toolkit entries or expose distinct MCP tool names.`,\n          );\n        }\n        toolSources.set(exposedName, serverName);\n        tools[exposedName] = tool as ToolSet[string];\n      }\n    }\n    return { tools, sources: toolSources };\n  }\n\n  #mcpClient(\n    name: string,\n    config: McpServerConfig,\n    startedAt: number,\n  ): Promise<MCPClient> {\n    const existing = this.#mcpClients.get(name);\n    if (existing) return existing;\n    const createPromise = createMCPClient(toMCPClientConfig(config));\n    let next: Promise<MCPClient>;\n    next = withMcpConnectionTimeout(createPromise, {\n      name,\n      config,\n      phase: \"connecting\",\n      startedAt,\n    }).catch((error) => {\n      if (this.#mcpClients.get(name) === next) {\n        this.#mcpClients.delete(name);\n      }\n      if (error instanceof MCPConnectionTimeoutError) {\n        void createPromise.then((client) => client.close()).catch(() => {});\n      }\n      throw error;\n    });\n    this.#mcpClients.set(name, next);\n    return next;\n  }\n}\n\nconst toMCPClientConfig = (config: McpServerConfig): MCPClientConfig => {\n  if (config.type === \"stdio\") {\n    return {\n      transport: new Experimental_StdioMCPTransport({\n        command: config.command,\n        ...(config.args && { args: [...config.args] }),\n        ...(config.env && { env: config.env }),\n        ...(config.cwd && { cwd: config.cwd }),\n      }),\n    };\n  }\n\n  return {\n    transport: {\n      type: config.type,\n      url: config.url,\n      ...(config.headers && { headers: config.headers }),\n      ...(config.redirect && { redirect: config.redirect }),\n    },\n  };\n};\n\ntype ToolkitTool = Toolkit[string];\n\ntype McpToolkitTool = ToolkitTool & {\n  type: \"mcp\";\n  server: McpServerConfig;\n  prefix?: string | undefined;\n  tools?: Record<string, McpToolkitToolConfig> | undefined;\n};\n\ntype McpToolSet = {\n  tools: ToolSet;\n  sources: Map<string, string>;\n};\n\nconst assertNoMcpToolNameCollisions = (\n  mcp: McpToolSet,\n  toolSets: readonly { source: string; tools: ToolSet }[],\n): void => {\n  for (const [toolName, serverName] of mcp.sources) {\n    for (const { source, tools } of toolSets) {\n      if (!Object.prototype.hasOwnProperty.call(tools, toolName)) continue;\n      throw new Error(\n        `MCP tool \"${toolName}\" from \"${serverName}\" conflicts with ${source} tool \"${toolName}\". Rename one of the tools so each model-visible tool name is unique.`,\n      );\n    }\n  }\n};\n\nconst isMcpToolkitTool = (tool: ToolkitTool): tool is McpToolkitTool =>\n  tool.type === \"mcp\" && !tool.disabled;\n\nconst getErrorMessage = (error: unknown): string =>\n  error instanceof Error ? error.message || error.name : String(error);\n\nconst toMcpToolkitError = (\n  entryName: string,\n  action: \"connect\" | \"list tools\" | \"close\",\n  error: unknown,\n): Error => {\n  return new Error(\n    `MCP toolkit entry \"${entryName}\" failed to ${action}: ${getErrorMessage(error)}`,\n    { cause: error },\n  );\n};\n\nconst isDisabledMcpTool = (config: McpToolkitToolConfig | undefined): boolean =>\n  config?.disabled === true;\n\nconst assertNoMcpToolkitTools = (toolkit: Toolkit): void => {\n  const mcpToolName = Object.entries(toolkit).find(([, tool]) =>\n    isMcpToolkitTool(tool),\n  )?.[0];\n  if (!mcpToolName) return;\n\n  throw new Error(\n    `MCP toolkit entry \"${mcpToolName}\" requires AISDKToolkit. Use new AISDKToolkit({ toolkit }).tools(...) instead of generativeTools(...).`,\n  );\n};\n\ntype AISDKToModelOutputOptions<TArgs, TResult> = Omit<\n  Parameters<ToolModelOutputFunction<TArgs, TResult>>[0],\n  \"output\"\n> & {\n  output: TResult | ModelContentEnvelope<TResult>;\n};\n\nconst toAISDKToModelOutput =\n  <TArgs, TResult>(toModelOutput?: ToolModelOutputFunction<TArgs, TResult>) =>\n  async (options: AISDKToModelOutputOptions<TArgs, TResult>) => {\n    const { result, modelContent } = unwrapModelContentEnvelope(options.output);\n\n    if (modelContent !== undefined) {\n      return toAISDKContent(modelContent);\n    }\n\n    if (!toModelOutput) {\n      return toAISDKDefaultOutput(result);\n    }\n\n    const parts = await toModelOutput({\n      ...options,\n      output: result,\n    });\n    return toAISDKContent(parts);\n  };\n\nconst toServerToolSet = (toolkit: ToolkitDefinition): ToolSet =>\n  Object.fromEntries(\n    Object.entries(toolkit)\n      .filter(\n        ([, t]) => t.type !== \"mcp\" && t.type !== \"provider\" && !t.disabled,\n      )\n      .map(([name, t]) => {\n        const execute = t.execute;\n        return [\n          name,\n          {\n            ...(t.description !== undefined && { description: t.description }),\n            inputSchema: parametersToInputSchema(t.parameters),\n            toModelOutput: toAISDKToModelOutput(t.toModelOutput),\n            ...(t.providerOptions && { providerOptions: t.providerOptions }),\n            ...(execute && {\n              execute: (\n                args: unknown,\n                callOptions: { toolCallId: string; abortSignal?: AbortSignal },\n              ) =>\n                execute(args as never, {\n                  toolCallId: callOptions.toolCallId,\n                  abortSignal: callOptions.abortSignal ?? neverAbort,\n                  human: humanNotSupported,\n                }),\n            }),\n          },\n        ];\n      }),\n  ) as ToolSet;\n\nconst toProviderToolSet = (toolkit: Toolkit): ToolSet =>\n  Object.fromEntries(\n    Object.entries(toolkit)\n      .filter((entry): entry is [string, ProviderToolkitTool] =>\n        isProviderToolkitTool(entry[1]),\n      )\n      .map(([name, t]) => [\n        name,\n        {\n          type: \"provider\",\n          id: t.providerId,\n          args: t.args,\n          ...(t.parameters && {\n            inputSchema: parametersToInputSchema(t.parameters),\n          }),\n          ...(t.providerOptions && { providerOptions: t.providerOptions }),\n          ...(t.supportsDeferredResults !== undefined && {\n            supportsDeferredResults: t.supportsDeferredResults,\n          }),\n        },\n      ]),\n  ) as ToolSet;\n\ntype ProviderToolkitTool = Extract<Toolkit[string], { type: \"provider\" }>;\n\nconst isProviderToolkitTool = (\n  tool: Toolkit[string],\n): tool is ProviderToolkitTool => tool.type === \"provider\" && !tool.disabled;\n"],"mappings":";;;;;;;;AAyBA,MAAMmB,eAAe;CAAEC,MAAM;CAAmBE,YAAY,CAAC;AAAE;AAE/D,MAAMC,0BAAiC;CACrC,MAAM,IAAIC,MACR,+DACF;AACF;AAGA,MAAMC,aAAa,IAAIC,gBAAgB,CAAC,CAACC;AAIzC,IAAME,4BAAN,cAAwCL,MAAM,CAAA;AAE9C,MAAMM,mCACJC,MACAC,OACAC,cAEA,IAAIJ,0BACF,sBAAsBE,KAAI,oBAAqBC,MAAK,SAAUC,UAAS,IACzE;AAEF,MAAMC,2BAA2B,OAC/BC,SACAG,YAMe;CACf,MAAML,YAAYK,QAAQC,OAAOE;CACjC,IAAIR,cAAcS,KAAAA,GAAW,OAAO,MAAMP;CAC1C,MAAMQ,cAAcV,aAAaW,KAAKC,IAAI,IAAIP,QAAQE;CACtD,MAAMM,qBACJhB,gCAAgCQ,QAAQP,MAAMO,QAAQN,OAAOC,SAAS;CACxE,IAAIU,eAAe,GAAG,MAAMG,aAAa;CAEzC,IAAIC;CACJ,IAAI;EACF,OAAO,MAAMX,QAAQc,KAAK,CACxBf,SACA,IAAIC,SAAgBe,GAAGC,WAAW;GAChCL,UAAUE,iBAAiBG,OAAON,aAAa,CAAC,GAAGH,WAAW;EAChE,CAAC,CAAC,CACH;CACH,UAAU;EACR,IAAII,YAAYL,KAAAA,GAAWW,aAAaN,OAAO;CACjD;AACF;AAGA,MAAMO,4CAA4B,IAAIC,QAGpC;AAEF,MAAMC,+BACJC,eACG;CACH,MAAME,SAASL,0BAA0BM,IAAIH,UAAU;CACvD,IAAIE,QAAQ,OAAOA;CAEnB,MAAME,YAAYvD,aAAamD,UAAU;CACzCH,0BAA0BQ,IAAIL,YAAYI,SAAS;CACnD,OAAOA;AACT;AAEA,MAAME,2BACJN,eACG;CACH,IAAI,CAACA,YAAY,OAAOzD,WAAWmB,YAAY;CAC/C,OAAOnB,WAAWwD,4BAA4BC,UAAU,CAAC;AAC3D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgEA,MAAaY,mBAAmB/B,YAA6C;CAC3EgC,wBAAwBhC,QAAQ2B,OAAO;CACvC,OAAO;EACL,GAAI3B,QAAQzB,gBAAgBA,cAAcyB,QAAQzB,aAAa,IAAI,CAAC;EAIpE,GAAG0D,kBAAkBjC,QAAQ2B,OAAO;EACpC,GAAGO,gBAAgBlC,QAAQ2B,OAA4B;CACzD;AACF;AAEA,IAAaQ,eAAb,MAA0B;CACxB;CACA,8BAAuB,IAAIE,IAAgC;CAE3DC,YAAYtC,SAA8B;EACxC,KAAK,WAAWA,QAAQ2B;CAC1B;CAEA,MAAMY,MAAMvC,UAAoC,CAAC,GAAqB;EACpE,MAAMwC,kBAAkBxC,QAAQ8B,WAC5BvD,cAAcyB,QAAQ8B,QAAQ,IAC9B,CAAC;EACL,MAAMW,aAAa,MAAM,KAAK,UAAU;EACxC,MAAME,kBAAkBV,kBAAkB,KAAK,QAAQ;EACvD,MAAMW,gBAAgBV,gBAAgB,KAAK,QAA6B;EAExEW,8BAA8BJ,YAAY;GACxC;IAAEK,QAAQ;IAAYP,OAAOC;GAAgB;GAC7C;IAAEM,QAAQ;IAAYP,OAAOI;GAAgB;GAC7C;IAAEG,QAAQ;IAAWP,OAAOK;GAAc;EAAC,CAC5C;EAED,OAAO;GACL,GAAGJ;GACH,GAAGC,WAAWF;GACd,GAAGI;GACH,GAAGC;EACL;CACF;CAEA,MAAMG,QAAuB;EAC3B,MAAMC,gBAAgB,CAAC,GAAG,KAAK,YAAYC,QAAQ,CAAC;EACpD,MAAMC,cAAcF,cAAcG,KAAK,CAAC1D,UAAUA,IAAI;EACtD,KAAK,YAAY2D,MAAM;EACvB,MAAMC,gBAAgB,MAAMvD,QAAQwD,WAClCN,cAAcG,KAAK,GAAGI,mBAAmBA,aAAa,CACxD;EACA,MAAMC,UAAUH,cAAcI,SAASC,QAAQC,UAC7CD,OAAOE,WAAW,cACd,CAAC,CAACV,YAAYS,QAASD,OAAOG,KAAK,CAAU,IAC7C,CAAA,CACN;EACA,MAAMC,eAAe,MAAMhE,QAAQwD,WACjCE,QAAQL,KAAK,GAAGY,YAAYA,OAAOhB,MAAM,CAAC,CAC5C;EACA,MAAMiB,SAAS,CACb,GAAGX,cAAcI,SAASC,QAAQC,UAChCD,OAAOE,WAAW,aACd,CAACK,kBAAkBf,YAAYS,QAAS,WAAWD,OAAOQ,MAAM,CAAC,IACjE,CAAA,CACN,GACA,GAAGJ,aAAaL,SAASC,QAAQC,UAC/BD,OAAOE,WAAW,aACd,CAACK,kBAAkBT,QAAQG,MAAM,CAAE,IAAI,SAASD,OAAOQ,MAAM,CAAC,IAC9D,CAAA,CACN,CAAC;EAEH,IAAIF,OAAOG,WAAW,GAAG,MAAMH,OAAO;EACtC,IAAIA,OAAOG,SAAS,GAClB,MAAM,IAAIC,eACRJ,QACA,yCACF;CAEJ;CAEA,MAAM,YAAiC;EACrC,MAAMO,WAAW,MAAMzE,QAAQ0E,IAC7BC,OAAOxB,QAAQ,KAAK,QAAQ,CAAC,CAC1ByB,QAAQC,UACPE,iBAAiBF,MAAM,EAAE,CAC3B,CAAC,CACAxB,IAAI,OAAO,CAAC1D,MAAMqF,UAAU;GAC3B,MAAM5E,YAAYI,KAAKC,IAAI;GAC3B,MAAMgD,gBAAgB,KAAK,WAAW9D,MAAMqF,KAAKE,QAAQ9E,SAAS;GAClE,MAAM6D,SAAS,MAAMR,cAAc0B,OAAOC,UAAmB;IAC3D,IAAIA,iBAAiB3F,2BAA2B,MAAM2F;IACtD,MAAMjB,kBAAkBxE,MAAM,WAAWyF,KAAK;GAChD,CAAC;GACD,IAAI;IAOF,OAAO;KAACzF;KAAMqF;KAAMvC,MANA3C,yBAAyBmE,OAAOxB,MAAM,GAAG;MAC3D9C;MACAQ,QAAQ6E,KAAKE;MACbtF,OAAO;MACPQ;KACF,CAAC;IACwB;GAC3B,SAASgF,OAAO;IACd,IAAIA,iBAAiB3F,2BAA2B;KAC9C,IAAI,KAAK,YAAY+B,IAAI7B,IAAI,MAAM8D,eAAe;MAChD,KAAK,YAAY4B,OAAO1F,IAAI;MAC5B,OAAYsD,MAAM,CAAC,CAACkC,YAAY,CAAC,CAAC;KACpC;KACA,MAAMC;IACR;IACA,MAAMjB,kBAAkBxE,MAAM,cAAcyF,KAAK;GACnD;EACF,CAAC,CACL;EAEA,MAAM3C,QAAQkC,OAAOW,OAAO,IAAI;EAChC,MAAMC,8BAAc,IAAIhD,IAAoB;EAC5C,KAAK,MAAM,CAACiD,YAAYC,SAASC,YAAYjB,UAC3C,KAAK,MAAM,CAACkB,UAAUX,SAASL,OAAOxB,QAAQuC,OAAO,GAAG;GACtD,IAAIE,kBAAkBH,QAAQhD,QAAQkD,SAAS,GAAG;GAClD,MAAME,cAAc,GAAGJ,QAAQK,UAAU,KAAKH;GAC9C,MAAMI,qBAAqBR,YAAY/D,IAAIqE,WAAW;GACtD,IAAIE,oBACF,MAAM,IAAI3G,MACR,6BAA6ByG,YAAW,wBAAyBE,mBAAkB,SAAUP,WAAU,wEACzG;GAEFD,YAAY7D,IAAImE,aAAaL,UAAU;GACvC/C,MAAMoD,eAAeb;EACvB;EAEF,OAAO;GAAEvC;GAAOuD,SAAST;EAAY;CACvC;CAEA,WACE5F,MACAQ,QACAC,WACoB;EACpB,MAAM8F,WAAW,KAAK,YAAY1E,IAAI7B,IAAI;EAC1C,IAAIuG,UAAU,OAAOA;EACrB,MAAMC,gBAAgBnI,gBAAgBoI,kBAAkBjG,MAAM,CAAC;EAC/D,IAAIkG;EACJA,OAAOvG,yBAAyBqG,eAAe;GAC7CxG;GACAQ;GACAP,OAAO;GACPQ;EACF,CAAC,CAAC,CAAC+E,OAAOC,UAAU;GAClB,IAAI,KAAK,YAAY5D,IAAI7B,IAAI,MAAM0G,MACjC,KAAK,YAAYhB,OAAO1F,IAAI;GAE9B,IAAIyF,iBAAiB3F,2BACnB,cAAmB6G,MAAMrC,WAAWA,OAAOhB,MAAM,CAAC,CAAC,CAACkC,YAAY,CAAC,CAAC;GAEpE,MAAMC;EACR,CAAC;EACD,KAAK,YAAY1D,IAAI/B,MAAM0G,IAAI;EAC/B,OAAOA;CACT;AACF;AAEA,MAAMD,qBAAqBjG,WAA6C;CACtE,IAAIA,OAAOnB,SAAS,SAClB,OAAO,EACLuH,WAAW,IAAItI,+BAA+B;EAC5CuI,SAASrG,OAAOqG;EAChB,GAAIrG,OAAOsG,QAAQ,EAAEA,MAAM,CAAC,GAAGtG,OAAOsG,IAAI,EAAE;EAC5C,GAAItG,OAAOuG,OAAO,EAAEA,KAAKvG,OAAOuG,IAAI;EACpC,GAAIvG,OAAOwG,OAAO,EAAEA,KAAKxG,OAAOwG,IAAI;CACtC,CAAC,EACH;CAGF,OAAO,EACLJ,WAAW;EACTvH,MAAMmB,OAAOnB;EACb4H,KAAKzG,OAAOyG;EACZ,GAAIzG,OAAO0G,WAAW,EAAEA,SAAS1G,OAAO0G,QAAQ;EAChD,GAAI1G,OAAO2G,YAAY,EAAEA,UAAU3G,OAAO2G,SAAS;CACrD,EACF;AACF;AAgBA,MAAM/D,iCACJkE,KACAxC,aACS;CACT,KAAK,MAAM,CAACkB,UAAUH,eAAeyB,IAAIjB,SACvC,KAAK,MAAM,EAAEhD,QAAQP,WAAWgC,UAAU;EACxC,IAAI,CAACE,OAAOuC,UAAUC,eAAeC,KAAK3E,OAAOkD,QAAQ,GAAG;EAC5D,MAAM,IAAIvG,MACR,aAAauG,SAAQ,UAAWH,WAAU,mBAAoBxC,OAAM,SAAU2C,SAAQ,sEACxF;CACF;AAEJ;AAEA,MAAMZ,oBAAoBC,SACxBA,KAAKhG,SAAS,SAAS,CAACgG,KAAKqC;AAE/B,MAAMC,mBAAmBlC,UACvBA,iBAAiBhG,QAAQgG,MAAMmC,WAAWnC,MAAMzF,OAAO6H,OAAOpC,KAAK;AAErE,MAAMjB,qBACJsD,WACAC,QACAtC,UACU;CACV,OAAO,IAAIhG,MACT,sBAAsBqI,UAAS,cAAeC,OAAM,IAAKJ,gBAAgBlC,KAAK,KAC9E,EAAEuC,OAAOvC,MAAM,CACjB;AACF;AAEA,MAAMQ,qBAAqBzF,WACzBA,QAAQkH,aAAa;AAEvB,MAAMnF,2BAA2BL,YAA2B;CAC1D,MAAM+F,cAAcjD,OAAOxB,QAAQtB,OAAO,CAAC,CAACgG,MAAM,GAAG7C,UACnDD,iBAAiBC,IAAI,CACvB,CAAC,GAAG;CACJ,IAAI,CAAC4C,aAAa;CAElB,MAAM,IAAIxI,MACR,sBAAsBwI,YAAW,uGACnC;AACF;AASA,MAAMQ,wBACaC,kBACjB,OAAOnI,YAAuD;CAC5D,MAAM,EAAE0D,QAAQ0E,iBAAiBzJ,2BAA2BqB,QAAQiI,MAAM;CAE1E,IAAIG,iBAAiBhI,KAAAA,GACnB,OAAO3B,eAAe2J,YAAY;CAGpC,IAAI,CAACD,eACH,OAAOzJ,qBAAqBgF,MAAM;CAGpC,MAAM2E,QAAQ,MAAMF,cAAc;EAChC,GAAGnI;EACHiI,QAAQvE;CACV,CAAC;CACD,OAAOjF,eAAe4J,KAAK;AAC7B;AAEF,MAAMnG,mBAAmBP,YACvB8C,OAAO6D,YACL7D,OAAOxB,QAAQtB,OAAO,CAAC,CACpB+C,QACE,GAAG6D,OAAOA,EAAEzJ,SAAS,SAASyJ,EAAEzJ,SAAS,cAAc,CAACyJ,EAAEpB,QAC7D,CAAC,CACAhE,KAAK,CAAC1D,MAAM8I,OAAO;CAClB,MAAMC,UAAUD,EAAEC;CAClB,OAAO,CACL/I,MACA;EACE,GAAI8I,EAAEE,gBAAgBrI,KAAAA,KAAa,EAAEqI,aAAaF,EAAEE,YAAY;EAChEC,aAAajH,wBAAwB8G,EAAEpH,UAAU;EACjDgH,eAAeD,qBAAqBK,EAAEJ,aAAa;EACnD,GAAII,EAAEI,mBAAmB,EAAEA,iBAAiBJ,EAAEI,gBAAgB;EAC9D,GAAIH,WAAW,EACbA,UACEjC,MACAqC,gBAEAJ,QAAQjC,MAAe;GACrBsC,YAAYD,YAAYC;GACxBC,aAAaF,YAAYE,eAAe3J;GACxC6J,OAAO/J;EACT,CAAC,EACL;CACF,CAAC;AAEL,CAAC,CACL;AAEF,MAAMgD,qBAAqBN,YACzB8C,OAAO6D,YACL7D,OAAOxB,QAAQtB,OAAO,CAAC,CACpB+C,QAAQC,UACPuE,sBAAsBvE,MAAM,EAAE,CAChC,CAAC,CACAxB,KAAK,CAAC1D,MAAM8I,OAAO,CAClB9I,MACA;CACEX,MAAM;CACNqK,IAAIZ,EAAEa;CACN7C,MAAMgC,EAAEhC;CACR,GAAIgC,EAAEpH,cAAc,EAClBuH,aAAajH,wBAAwB8G,EAAEpH,UAAU,EACnD;CACA,GAAIoH,EAAEI,mBAAmB,EAAEA,iBAAiBJ,EAAEI,gBAAgB;CAC9D,GAAIJ,EAAEc,4BAA4BjJ,KAAAA,KAAa,EAC7CiJ,yBAAyBd,EAAEc,wBAC7B;AACF,CAAC,CACF,CACL;AAIF,MAAMH,yBACJpE,SACgCA,KAAKhG,SAAS,cAAc,CAACgG,KAAKqC"}

package/dist/transport/resumable.js.map

@@ -1 +1 @@
-{"version":3,"file":"resumable.js","names":["RESUMABLE_STREAM_ID_HEADER","RESUMABLE_STREAM_ID_HEADER_VALUE","DEFAULT_STORAGE_KEY","ResumableClientStorage","getStreamId","threadId","setStreamId","id","clear","subscribe","listener","getSessionStorage","Storage","window","sessionStorage","StorageSlot","value","owner","createResumableSessionStorage","options","key","keyOption","resolveKey","undefined","slots","Map","listeners","Set","readSlot","slot","get","storage","getItem","set","notify","subscription","error","console","setItem","Array","from","find","removeItem","add","delete","AssistantChatResumableOptions","resumeApi","streamId","isFinishEvent","chunk","Uint8Array","accumulator"],"sources":["../../src/transport/resumable.ts"],"sourcesContent":["\"use client\";\n\nimport { RESUMABLE_STREAM_ID_HEADER as RESUMABLE_STREAM_ID_HEADER_VALUE } from \"assistant-stream/resumable\";\n\n/** Response header used by the [Resumable Streams](/docs/guides/resumable-streams) server and client wiring. */\nexport const RESUMABLE_STREAM_ID_HEADER = RESUMABLE_STREAM_ID_HEADER_VALUE;\n\nconst DEFAULT_STORAGE_KEY = \"aui-resumable-stream-id\";\n\nexport type ResumableClientStorage = {\n  getStreamId(threadId?: string): string | null;\n  setStreamId(id: string, threadId?: string): void;\n  clear(threadId?: string): void;\n  /** Subscribes to stream id changes so automatic resume can react after mount. */\n  subscribe?(listener: () => void, threadId?: string): () => void;\n};\n\nconst getSessionStorage = (): Storage | null => {\n  if (typeof window === \"undefined\") return null;\n  try {\n    return window.sessionStorage;\n  } catch {\n    return null;\n  }\n};\n\ntype StorageSlot = {\n  value: string | null;\n  owner: string | undefined;\n};\n\n/** `sessionStorage`-backed storage for the pending resumable stream id. See the [Resumable Streams](/docs/guides/resumable-streams) guide for end-to-end wiring. */\nexport function createResumableSessionStorage(options?: {\n  /**\n   * Storage key for the pending stream id. A static string namespaces per route\n   * or chat surface. A getter is read lazily on every access, so the key can be\n   * derived from the active thread's identity; while the getter returns\n   * `undefined`, reads report no pending stream and writes are dropped, so a\n   * thread whose identity is not known yet never touches another thread's key.\n   *\n   * Under a remote thread list with more than one thread, scope the key per\n   * thread and create one storage instance per thread runtime rather than a\n   * single shared one. A shared key is written and cleared by whichever thread\n   * acts last, so one conversation's stream can resume inside another.\n   */\n  key?: string | (() => string | undefined);\n}): ResumableClientStorage {\n  const keyOption = options?.key;\n  const resolveKey = (): string | undefined => {\n    if (typeof keyOption !== \"function\")\n      return keyOption ?? DEFAULT_STORAGE_KEY;\n    try {\n      return keyOption();\n    } catch {\n      return undefined;\n    }\n  };\n  const slots = new Map<string, StorageSlot>();\n  const listeners = new Set<{\n    listener: () => void;\n    threadId: string | undefined;\n  }>();\n  const readSlot = (key: string): StorageSlot => {\n    let slot = slots.get(key);\n    if (slot) return slot;\n    const storage = getSessionStorage();\n    let value: string | null = null;\n    if (storage) {\n      try {\n        value = storage.getItem(key);\n      } catch {\n        value = null;\n      }\n    }\n    slot = { value, owner: undefined };\n    slots.set(key, slot);\n    return slot;\n  };\n  const notify = (threadId?: string) => {\n    for (const subscription of listeners) {\n      if (threadId && subscription.threadId !== threadId) continue;\n      try {\n        subscription.listener();\n      } catch (error) {\n        console.error(\n          \"[assistant-ui] resumable storage listener failed\",\n          error,\n        );\n      }\n    }\n  };\n\n  return {\n    getStreamId(threadId) {\n      const key = resolveKey();\n      if (!key) return null;\n      const slot = readSlot(key);\n      if (slot.value === null) return null;\n      if (slot.owner && threadId && slot.owner !== threadId) return null;\n      return slot.value;\n    },\n    setStreamId(id, threadId) {\n      const key = resolveKey();\n      const storage = getSessionStorage();\n      if (!key || !storage) return;\n      try {\n        storage.setItem(key, id);\n      } catch {\n        // Ignore blocked or unavailable sessionStorage.\n        return;\n      }\n      const slot = readSlot(key);\n      slot.value = id;\n      if (threadId) {\n        slot.owner = threadId;\n      } else if (!slot.owner) {\n        slot.owner = Array.from(listeners).find(\n          (subscription) => subscription.threadId !== undefined,\n        )?.threadId;\n      }\n      notify(slot.owner);\n    },\n    clear(threadId) {\n      const key = resolveKey();\n      if (!key) return;\n      const slot = readSlot(key);\n      if (slot.owner && threadId && slot.owner !== threadId) return;\n      const storage = getSessionStorage();\n      if (!storage) return;\n      try {\n        storage.removeItem(key);\n      } catch {\n        // Ignore blocked or unavailable sessionStorage.\n        return;\n      }\n      slot.value = null;\n      slot.owner = undefined;\n      notify(threadId);\n    },\n    subscribe(listener, threadId) {\n      const key = resolveKey();\n      if (key && threadId) {\n        const slot = readSlot(key);\n        if (slot.value !== null) slot.owner ??= threadId;\n      }\n      const subscription = { listener, threadId };\n      listeners.add(subscription);\n      return () => listeners.delete(subscription);\n    },\n  };\n}\n\nexport type AssistantChatResumableOptions = {\n  storage: ResumableClientStorage;\n  resumeApi: string | ((streamId: string) => string);\n  /**\n   * Defaults to scanning for the AI SDK UIMessageStream `finish` marker.\n   * Cancellation never invokes this callback, only natural completion does.\n   */\n  isFinishEvent?: (chunk: Uint8Array, accumulator: string) => boolean;\n};\n"],"mappings":";;;;AAKA,MAAaA,6BAA6BC;AAE1C,MAAMC,sBAAsB;AAU5B,MAAMS,0BAA0C;CAC9C,IAAI,OAAOE,WAAW,aAAa,OAAO;CAC1C,IAAI;EACF,OAAOA,OAAOC;CAChB,QAAQ;EACN,OAAO;CACT;AACF;;AAQA,SAAgBI,8BAA8BC,SAcnB;CACzB,MAAME,YAAYF,SAASC;CAC3B,MAAME,mBAAuC;EAC3C,IAAI,OAAOD,cAAc,YACvB,OAAOA,aAAanB;EACtB,IAAI;GACF,OAAOmB,UAAU;EACnB,QAAQ;GACN;EACF;CACF;CACA,MAAMG,wBAAQ,IAAIC,IAAyB;CAC3C,MAAMC,4BAAY,IAAIC,IAGnB;CACH,MAAMC,YAAYR,QAA6B;EAC7C,IAAIS,OAAOL,MAAMM,IAAIV,GAAG;EACxB,IAAIS,MAAM,OAAOA;EACjB,MAAME,UAAUpB,kBAAkB;EAClC,IAAIK,QAAuB;EAC3B,IAAIe,SACF,IAAI;GACFf,QAAQe,QAAQC,QAAQZ,GAAG;EAC7B,QAAQ;GACNJ,QAAQ;EACV;EAEFa,OAAO;GAAEb;GAAOC,OAAOM,KAAAA;EAAU;EACjCC,MAAMS,IAAIb,KAAKS,IAAI;EACnB,OAAOA;CACT;CACA,MAAMK,UAAU7B,aAAsB;EACpC,KAAK,MAAM8B,gBAAgBT,WAAW;GACpC,IAAIrB,YAAY8B,aAAa9B,aAAaA,UAAU;GACpD,IAAI;IACF8B,aAAazB,SAAS;GACxB,SAAS0B,OAAO;IACdC,QAAQD,MACN,oDACAA,KACF;GACF;EACF;CACF;CAEA,OAAO;EACLhC,YAAYC,UAAU;GACpB,MAAMe,MAAME,WAAW;GACvB,IAAI,CAACF,KAAK,OAAO;GACjB,MAAMS,OAAOD,SAASR,GAAG;GACzB,IAAIS,KAAKb,UAAU,MAAM,OAAO;GAChC,IAAIa,KAAKZ,SAASZ,YAAYwB,KAAKZ,UAAUZ,UAAU,OAAO;GAC9D,OAAOwB,KAAKb;EACd;EACAV,YAAYC,IAAIF,UAAU;GACxB,MAAMe,MAAME,WAAW;GACvB,MAAMS,UAAUpB,kBAAkB;GAClC,IAAI,CAACS,OAAO,CAACW,SAAS;GACtB,IAAI;IACFA,QAAQO,QAAQlB,KAAKb,EAAE;GACzB,QAAQ;IAEN;GACF;GACA,MAAMsB,OAAOD,SAASR,GAAG;GACzBS,KAAKb,QAAQT;GACb,IAAIF,UACFwB,KAAKZ,QAAQZ;QACR,IAAI,CAACwB,KAAKZ,OACfY,KAAKZ,QAAQsB,MAAMC,KAAKd,SAAS,CAAC,CAACe,MAChCN,iBAAiBA,aAAa9B,aAAakB,KAAAA,CAC9C,CAAC,EAAElB;GAEL6B,OAAOL,KAAKZ,KAAK;EACnB;EACAT,MAAMH,UAAU;GACd,MAAMe,MAAME,WAAW;GACvB,IAAI,CAACF,KAAK;GACV,MAAMS,OAAOD,SAASR,GAAG;GACzB,IAAIS,KAAKZ,SAASZ,YAAYwB,KAAKZ,UAAUZ,UAAU;GACvD,MAAM0B,UAAUpB,kBAAkB;GAClC,IAAI,CAACoB,SAAS;GACd,IAAI;IACFA,QAAQW,WAAWtB,GAAG;GACxB,QAAQ;IAEN;GACF;GACAS,KAAKb,QAAQ;GACba,KAAKZ,QAAQM,KAAAA;GACbW,OAAO7B,QAAQ;EACjB;EACAI,UAAUC,UAAUL,UAAU;GAC5B,MAAMe,MAAME,WAAW;GACvB,IAAIF,OAAOf,UAAU;IACnB,MAAMwB,OAAOD,SAASR,GAAG;IACzB,IAAIS,KAAKb,UAAU,MAAMa,KAAKZ,UAAUZ;GAC1C;GACA,MAAM8B,eAAe;IAAEzB;IAAUL;GAAS;GAC1CqB,UAAUiB,IAAIR,YAAY;GAC1B,aAAaT,UAAUkB,OAAOT,YAAY;EAC5C;CACF;AACF"}
+{"version":3,"file":"resumable.js","names":["RESUMABLE_STREAM_ID_HEADER","RESUMABLE_STREAM_ID_HEADER_VALUE","DEFAULT_STORAGE_KEY","ResumableClientStorage","getStreamId","threadId","setStreamId","id","clear","subscribe","listener","getSessionStorage","Storage","window","sessionStorage","StorageSlot","value","owner","createResumableSessionStorage","options","key","keyOption","resolveKey","undefined","slots","Map","listeners","Set","readSlot","slot","get","storage","getItem","set","notify","subscription","error","console","setItem","Array","from","find","removeItem","add","delete","AssistantChatResumableOptions","resumeApi","streamId","isFinishEvent","chunk","Uint8Array","accumulator","RESUMABLE_STREAM_ID_HEADER_VALUE"],"sources":["../../src/transport/resumable.ts"],"sourcesContent":["\"use client\";\n\nimport { RESUMABLE_STREAM_ID_HEADER as RESUMABLE_STREAM_ID_HEADER_VALUE } from \"assistant-stream/resumable\";\n\n/** Response header used by the [Resumable Streams](/docs/guides/resumable-streams) server and client wiring. */\nexport const RESUMABLE_STREAM_ID_HEADER = RESUMABLE_STREAM_ID_HEADER_VALUE;\n\nconst DEFAULT_STORAGE_KEY = \"aui-resumable-stream-id\";\n\nexport type ResumableClientStorage = {\n  getStreamId(threadId?: string): string | null;\n  setStreamId(id: string, threadId?: string): void;\n  clear(threadId?: string): void;\n  /** Subscribes to stream id changes so automatic resume can react after mount. */\n  subscribe?(listener: () => void, threadId?: string): () => void;\n};\n\nconst getSessionStorage = (): Storage | null => {\n  if (typeof window === \"undefined\") return null;\n  try {\n    return window.sessionStorage;\n  } catch {\n    return null;\n  }\n};\n\ntype StorageSlot = {\n  value: string | null;\n  owner: string | undefined;\n};\n\n/** `sessionStorage`-backed storage for the pending resumable stream id. See the [Resumable Streams](/docs/guides/resumable-streams) guide for end-to-end wiring. */\nexport function createResumableSessionStorage(options?: {\n  /**\n   * Storage key for the pending stream id. A static string namespaces per route\n   * or chat surface. A getter is read lazily on every access, so the key can be\n   * derived from the active thread's identity; while the getter returns\n   * `undefined`, reads report no pending stream and writes are dropped, so a\n   * thread whose identity is not known yet never touches another thread's key.\n   *\n   * Under a remote thread list with more than one thread, scope the key per\n   * thread and create one storage instance per thread runtime rather than a\n   * single shared one. A shared key is written and cleared by whichever thread\n   * acts last, so one conversation's stream can resume inside another.\n   */\n  key?: string | (() => string | undefined);\n}): ResumableClientStorage {\n  const keyOption = options?.key;\n  const resolveKey = (): string | undefined => {\n    if (typeof keyOption !== \"function\")\n      return keyOption ?? DEFAULT_STORAGE_KEY;\n    try {\n      return keyOption();\n    } catch {\n      return undefined;\n    }\n  };\n  const slots = new Map<string, StorageSlot>();\n  const listeners = new Set<{\n    listener: () => void;\n    threadId: string | undefined;\n  }>();\n  const readSlot = (key: string): StorageSlot => {\n    let slot = slots.get(key);\n    if (slot) return slot;\n    const storage = getSessionStorage();\n    let value: string | null = null;\n    if (storage) {\n      try {\n        value = storage.getItem(key);\n      } catch {\n        value = null;\n      }\n    }\n    slot = { value, owner: undefined };\n    slots.set(key, slot);\n    return slot;\n  };\n  const notify = (threadId?: string) => {\n    for (const subscription of listeners) {\n      if (threadId && subscription.threadId !== threadId) continue;\n      try {\n        subscription.listener();\n      } catch (error) {\n        console.error(\n          \"[assistant-ui] resumable storage listener failed\",\n          error,\n        );\n      }\n    }\n  };\n\n  return {\n    getStreamId(threadId) {\n      const key = resolveKey();\n      if (!key) return null;\n      const slot = readSlot(key);\n      if (slot.value === null) return null;\n      if (slot.owner && threadId && slot.owner !== threadId) return null;\n      return slot.value;\n    },\n    setStreamId(id, threadId) {\n      const key = resolveKey();\n      const storage = getSessionStorage();\n      if (!key || !storage) return;\n      try {\n        storage.setItem(key, id);\n      } catch {\n        // Ignore blocked or unavailable sessionStorage.\n        return;\n      }\n      const slot = readSlot(key);\n      slot.value = id;\n      if (threadId) {\n        slot.owner = threadId;\n      } else if (!slot.owner) {\n        slot.owner = Array.from(listeners).find(\n          (subscription) => subscription.threadId !== undefined,\n        )?.threadId;\n      }\n      notify(slot.owner);\n    },\n    clear(threadId) {\n      const key = resolveKey();\n      if (!key) return;\n      const slot = readSlot(key);\n      if (slot.owner && threadId && slot.owner !== threadId) return;\n      const storage = getSessionStorage();\n      if (!storage) return;\n      try {\n        storage.removeItem(key);\n      } catch {\n        // Ignore blocked or unavailable sessionStorage.\n        return;\n      }\n      slot.value = null;\n      slot.owner = undefined;\n      notify(threadId);\n    },\n    subscribe(listener, threadId) {\n      const key = resolveKey();\n      if (key && threadId) {\n        const slot = readSlot(key);\n        if (slot.value !== null) slot.owner ??= threadId;\n      }\n      const subscription = { listener, threadId };\n      listeners.add(subscription);\n      return () => listeners.delete(subscription);\n    },\n  };\n}\n\nexport type AssistantChatResumableOptions = {\n  storage: ResumableClientStorage;\n  resumeApi: string | ((streamId: string) => string);\n  /**\n   * Defaults to scanning for the AI SDK UIMessageStream `finish` marker.\n   * Cancellation never invokes this callback, only natural completion does.\n   */\n  isFinishEvent?: (chunk: Uint8Array, accumulator: string) => boolean;\n};\n"],"mappings":";;;;AAKA,MAAaA,6BAA6BC;AAE1C,MAAMC,sBAAsB;AAU5B,MAAMS,0BAA0C;CAC9C,IAAI,OAAOE,WAAW,aAAa,OAAO;CAC1C,IAAI;EACF,OAAOA,OAAOC;CAChB,QAAQ;EACN,OAAO;CACT;AACF;;AAQA,SAAgBI,8BAA8BC,SAcnB;CACzB,MAAME,YAAYF,SAASC;CAC3B,MAAME,mBAAuC;EAC3C,IAAI,OAAOD,cAAc,YACvB,OAAOA,aAAanB;EACtB,IAAI;GACF,OAAOmB,UAAU;EACnB,QAAQ;GACN;EACF;CACF;CACA,MAAMG,wBAAQ,IAAIC,IAAyB;CAC3C,MAAMC,4BAAY,IAAIC,IAGnB;CACH,MAAMC,YAAYR,QAA6B;EAC7C,IAAIS,OAAOL,MAAMM,IAAIV,GAAG;EACxB,IAAIS,MAAM,OAAOA;EACjB,MAAME,UAAUpB,kBAAkB;EAClC,IAAIK,QAAuB;EAC3B,IAAIe,SACF,IAAI;GACFf,QAAQe,QAAQC,QAAQZ,GAAG;EAC7B,QAAQ;GACNJ,QAAQ;EACV;EAEFa,OAAO;GAAEb;GAAOC,OAAOM,KAAAA;EAAU;EACjCC,MAAMS,IAAIb,KAAKS,IAAI;EACnB,OAAOA;CACT;CACA,MAAMK,UAAU7B,aAAsB;EACpC,KAAK,MAAM8B,gBAAgBT,WAAW;GACpC,IAAIrB,YAAY8B,aAAa9B,aAAaA,UAAU;GACpD,IAAI;IACF8B,aAAazB,SAAS;GACxB,SAAS0B,OAAO;IACdC,QAAQD,MACN,oDACAA,KACF;GACF;EACF;CACF;CAEA,OAAO;EACLhC,YAAYC,UAAU;GACpB,MAAMe,MAAME,WAAW;GACvB,IAAI,CAACF,KAAK,OAAO;GACjB,MAAMS,OAAOD,SAASR,GAAG;GACzB,IAAIS,KAAKb,UAAU,MAAM,OAAO;GAChC,IAAIa,KAAKZ,SAASZ,YAAYwB,KAAKZ,UAAUZ,UAAU,OAAO;GAC9D,OAAOwB,KAAKb;EACd;EACAV,YAAYC,IAAIF,UAAU;GACxB,MAAMe,MAAME,WAAW;GACvB,MAAMS,UAAUpB,kBAAkB;GAClC,IAAI,CAACS,OAAO,CAACW,SAAS;GACtB,IAAI;IACFA,QAAQO,QAAQlB,KAAKb,EAAE;GACzB,QAAQ;IAEN;GACF;GACA,MAAMsB,OAAOD,SAASR,GAAG;GACzBS,KAAKb,QAAQT;GACb,IAAIF,UACFwB,KAAKZ,QAAQZ;QACR,IAAI,CAACwB,KAAKZ,OACfY,KAAKZ,QAAQsB,MAAMC,KAAKd,SAAS,CAAC,CAACe,MAChCN,iBAAiBA,aAAa9B,aAAakB,KAAAA,CAC9C,CAAC,EAAElB;GAEL6B,OAAOL,KAAKZ,KAAK;EACnB;EACAT,MAAMH,UAAU;GACd,MAAMe,MAAME,WAAW;GACvB,IAAI,CAACF,KAAK;GACV,MAAMS,OAAOD,SAASR,GAAG;GACzB,IAAIS,KAAKZ,SAASZ,YAAYwB,KAAKZ,UAAUZ,UAAU;GACvD,MAAM0B,UAAUpB,kBAAkB;GAClC,IAAI,CAACoB,SAAS;GACd,IAAI;IACFA,QAAQW,WAAWtB,GAAG;GACxB,QAAQ;IAEN;GACF;GACAS,KAAKb,QAAQ;GACba,KAAKZ,QAAQM,KAAAA;GACbW,OAAO7B,QAAQ;EACjB;EACAI,UAAUC,UAAUL,UAAU;GAC5B,MAAMe,MAAME,WAAW;GACvB,IAAIF,OAAOf,UAAU;IACnB,MAAMwB,OAAOD,SAASR,GAAG;IACzB,IAAIS,KAAKb,UAAU,MAAMa,KAAKZ,UAAUZ;GAC1C;GACA,MAAM8B,eAAe;IAAEzB;IAAUL;GAAS;GAC1CqB,UAAUiB,IAAIR,YAAY;GAC1B,aAAaT,UAAUkB,OAAOT,YAAY;EAC5C;CACF;AACF"}

package/dist/usage.js.map

@@ -1 +1 @@
-{"version":3,"file":"usage.js","names":["useMemo","useAuiState","ThreadTokenUsage","totalTokens","inputTokens","outputTokens","reasoningTokens","cachedInputTokens","TokenUsageExtractableMessage","role","metadata","UsageRecord","Record","USAGE_KEYS","const","asRecord","value","Array","isArray","undefined","asPositiveTokenCount","Number","isFinite","computeTotalTokens","usage","normalizeUsage","record","result","hasFields","key","count","outputTokenDetails","inputTokenDetails","cacheReadTokens","withComputedTotal","usageFromSteps","steps","sums","present","stepsWithUsage","stepsWithComputableTotal","step","stepTotal","getThreadMessageTokenUsage","message","topLevelUsage","customUsage","custom","findLatestMessageWithUsage","messages","idx","length","useThreadTokenUsage","$","_c","msg","_temp","t0","s","thread"],"sources":["../src/usage.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/react\" preserve=\"true\" />\nimport { useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\n\nexport type ThreadTokenUsage = {\n  totalTokens?: number;\n  inputTokens?: number;\n  outputTokens?: number;\n  reasoningTokens?: number;\n  cachedInputTokens?: number;\n};\n\nexport interface TokenUsageExtractableMessage {\n  role?: string;\n  metadata?: unknown;\n}\n\ntype UsageRecord = Record<string, unknown>;\n\nconst USAGE_KEYS = [\n  \"inputTokens\",\n  \"outputTokens\",\n  \"reasoningTokens\",\n  \"cachedInputTokens\",\n  \"totalTokens\",\n] as const satisfies (keyof ThreadTokenUsage)[];\n\nfunction asRecord(value: unknown): UsageRecord | undefined {\n  if (!value || typeof value !== \"object\" || Array.isArray(value))\n    return undefined;\n  return value as UsageRecord;\n}\n\nfunction asPositiveTokenCount(value: unknown): number | undefined {\n  if (typeof value !== \"number\" || !Number.isFinite(value) || value < 0) {\n    return undefined;\n  }\n  return value;\n}\n\nfunction computeTotalTokens(usage: ThreadTokenUsage): number | undefined {\n  if (usage.totalTokens !== undefined) return usage.totalTokens;\n  if (usage.inputTokens !== undefined && usage.outputTokens !== undefined) {\n    return usage.inputTokens + usage.outputTokens;\n  }\n  return undefined;\n}\n\nfunction normalizeUsage(value: unknown): ThreadTokenUsage | undefined {\n  const record = asRecord(value);\n  if (!record) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  let hasFields = false;\n  for (const key of USAGE_KEYS) {\n    const count = asPositiveTokenCount(record[key]);\n    if (count !== undefined) {\n      result[key] = count;\n      hasFields = true;\n    }\n  }\n  // AI SDK v7 moved these under token detail objects; v6 kept them top-level.\n  if (result.reasoningTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.outputTokenDetails)?.reasoningTokens,\n    );\n    if (count !== undefined) {\n      result.reasoningTokens = count;\n      hasFields = true;\n    }\n  }\n  if (result.cachedInputTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.inputTokenDetails)?.cacheReadTokens,\n    );\n    if (count !== undefined) {\n      result.cachedInputTokens = count;\n      hasFields = true;\n    }\n  }\n  return hasFields ? result : undefined;\n}\n\nfunction withComputedTotal(\n  usage: ThreadTokenUsage,\n): ThreadTokenUsage | undefined {\n  const totalTokens = computeTotalTokens(usage);\n  return { ...usage, ...(totalTokens !== undefined && { totalTokens }) };\n}\n\nfunction usageFromSteps(value: unknown): ThreadTokenUsage | undefined {\n  const steps = Array.isArray(value) ? value : [];\n\n  const sums: Record<string, number> = {};\n  const present: Record<string, boolean> = {};\n  let stepsWithUsage = 0;\n  let stepsWithComputableTotal = 0;\n\n  for (const step of steps) {\n    const usage = normalizeUsage(asRecord(step)?.usage);\n    if (!usage) continue;\n    stepsWithUsage++;\n\n    const stepTotal = computeTotalTokens(usage);\n    if (stepTotal !== undefined) {\n      sums.totalTokens = (sums.totalTokens ?? 0) + stepTotal;\n      stepsWithComputableTotal++;\n    }\n\n    for (const key of USAGE_KEYS) {\n      if (key === \"totalTokens\") continue;\n      if (usage[key] !== undefined) {\n        sums[key] = (sums[key] ?? 0) + usage[key];\n        present[key] = true;\n      }\n    }\n  }\n\n  if (stepsWithUsage === 0) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  if (stepsWithComputableTotal === stepsWithUsage) {\n    result.totalTokens = sums.totalTokens!;\n  }\n  for (const key of USAGE_KEYS) {\n    if (key === \"totalTokens\") continue;\n    if (present[key]) {\n      result[key] = sums[key]!;\n    }\n  }\n  return result;\n}\n\nexport function getThreadMessageTokenUsage(\n  message: TokenUsageExtractableMessage | undefined,\n): ThreadTokenUsage | undefined {\n  if (!message || message.role !== \"assistant\") return undefined;\n\n  const metadata = asRecord(message.metadata);\n  if (!metadata) return undefined;\n\n  const topLevelUsage = normalizeUsage(metadata.usage);\n  if (topLevelUsage) return withComputedTotal(topLevelUsage);\n\n  const customUsage = normalizeUsage(asRecord(metadata.custom)?.usage);\n  if (customUsage) return withComputedTotal(customUsage);\n\n  return usageFromSteps(metadata.steps);\n}\n\nfunction findLatestMessageWithUsage(\n  messages: readonly TokenUsageExtractableMessage[] | undefined,\n): TokenUsageExtractableMessage | undefined {\n  if (!messages) return undefined;\n\n  for (let idx = messages.length - 1; idx >= 0; idx -= 1) {\n    const message = messages[idx];\n    if (getThreadMessageTokenUsage(message)) {\n      return message;\n    }\n  }\n\n  return undefined;\n}\n\n/**\n * Reads token usage from the newest assistant message that reports any.\n *\n * A route attaches usage through the AI SDK's `messageMetadata` option. Because\n * a thread message carries a fixed metadata shape, the converter moves every\n * other key the route returns into `metadata.custom`, which is where this hook\n * looks.\n */\nexport function useThreadTokenUsage(): ThreadTokenUsage | undefined {\n  const msg = useAuiState((s) => findLatestMessageWithUsage(s.thread.messages));\n  return useMemo(() => getThreadMessageTokenUsage(msg), [msg]);\n}\n"],"mappings":";;;;AAmBA,MAAMa,aAAa;CACjB;CACA;CACA;CACA;CACA;AAAa;AAGf,SAASE,SAASC,OAAyC;CACzD,IAAI,CAACA,SAAS,OAAOA,UAAU,YAAYC,MAAMC,QAAQF,KAAK,GAC5D,OAAOG,KAAAA;CACT,OAAOH;AACT;AAEA,SAASI,qBAAqBJ,OAAoC;CAChE,IAAI,OAAOA,UAAU,YAAY,CAACK,OAAOC,SAASN,KAAK,KAAKA,QAAQ,GAClE;CAEF,OAAOA;AACT;AAEA,SAASO,mBAAmBC,OAA6C;CACvE,IAAIA,MAAMrB,gBAAgBgB,KAAAA,GAAW,OAAOK,MAAMrB;CAClD,IAAIqB,MAAMpB,gBAAgBe,KAAAA,KAAaK,MAAMnB,iBAAiBc,KAAAA,GAC5D,OAAOK,MAAMpB,cAAcoB,MAAMnB;AAGrC;AAEA,SAASoB,eAAeT,OAA8C;CACpE,MAAMU,SAASX,SAASC,KAAK;CAC7B,IAAI,CAACU,QAAQ,OAAOP,KAAAA;CAEpB,MAAMQ,SAA2B,CAAC;CAClC,IAAIC,YAAY;CAChB,KAAK,MAAMC,OAAOhB,YAAY;EAC5B,MAAMiB,QAAQV,qBAAqBM,OAAOG,IAAI;EAC9C,IAAIC,UAAUX,KAAAA,GAAW;GACvBQ,OAAOE,OAAOC;GACdF,YAAY;EACd;CACF;CAEA,IAAID,OAAOrB,oBAAoBa,KAAAA,GAAW;EACxC,MAAMW,QAAQV,qBACZL,SAASW,OAAOK,kBAAkB,CAAC,EAAEzB,eACvC;EACA,IAAIwB,UAAUX,KAAAA,GAAW;GACvBQ,OAAOrB,kBAAkBwB;GACzBF,YAAY;EACd;CACF;CACA,IAAID,OAAOpB,sBAAsBY,KAAAA,GAAW;EAC1C,MAAMW,QAAQV,qBACZL,SAASW,OAAOM,iBAAiB,CAAC,EAAEC,eACtC;EACA,IAAIH,UAAUX,KAAAA,GAAW;GACvBQ,OAAOpB,oBAAoBuB;GAC3BF,YAAY;EACd;CACF;CACA,OAAOA,YAAYD,SAASR,KAAAA;AAC9B;AAEA,SAASe,kBACPV,OAC8B;CAC9B,MAAMrB,cAAcoB,mBAAmBC,KAAK;CAC5C,OAAO;EAAE,GAAGA;EAAO,GAAIrB,gBAAgBgB,KAAAA,KAAa,EAAEhB,YAAY;CAAG;AACvE;AAEA,SAASgC,eAAenB,OAA8C;CACpE,MAAMoB,QAAQnB,MAAMC,QAAQF,KAAK,IAAIA,QAAQ,CAAA;CAE7C,MAAMqB,OAA+B,CAAC;CACtC,MAAMC,UAAmC,CAAC;CAC1C,IAAIC,iBAAiB;CACrB,IAAIC,2BAA2B;CAE/B,KAAK,MAAMC,QAAQL,OAAO;EACxB,MAAMZ,QAAQC,eAAeV,SAAS0B,IAAI,CAAC,EAAEjB,KAAK;EAClD,IAAI,CAACA,OAAO;EACZe;EAEA,MAAMG,YAAYnB,mBAAmBC,KAAK;EAC1C,IAAIkB,cAAcvB,KAAAA,GAAW;GAC3BkB,KAAKlC,eAAekC,KAAKlC,eAAe,KAAKuC;GAC7CF;EACF;EAEA,KAAK,MAAMX,OAAOhB,YAAY;GAC5B,IAAIgB,QAAQ,eAAe;GAC3B,IAAIL,MAAMK,SAASV,KAAAA,GAAW;IAC5BkB,KAAKR,QAAQQ,KAAKR,QAAQ,KAAKL,MAAMK;IACrCS,QAAQT,OAAO;GACjB;EACF;CACF;CAEA,IAAIU,mBAAmB,GAAG,OAAOpB,KAAAA;CAEjC,MAAMQ,SAA2B,CAAC;CAClC,IAAIa,6BAA6BD,gBAC/BZ,OAAOxB,cAAckC,KAAKlC;CAE5B,KAAK,MAAM0B,OAAOhB,YAAY;EAC5B,IAAIgB,QAAQ,eAAe;EAC3B,IAAIS,QAAQT,MACVF,OAAOE,OAAOQ,KAAKR;CAEvB;CACA,OAAOF;AACT;AAEA,SAAgBgB,2BACdC,SAC8B;CAC9B,IAAI,CAACA,WAAWA,QAAQnC,SAAS,aAAa,OAAOU,KAAAA;CAErD,MAAMT,WAAWK,SAAS6B,QAAQlC,QAAQ;CAC1C,IAAI,CAACA,UAAU,OAAOS,KAAAA;CAEtB,MAAM0B,gBAAgBpB,eAAef,SAASc,KAAK;CACnD,IAAIqB,eAAe,OAAOX,kBAAkBW,aAAa;CAEzD,MAAMC,cAAcrB,eAAeV,SAASL,SAASqC,MAAM,CAAC,EAAEvB,KAAK;CACnE,IAAIsB,aAAa,OAAOZ,kBAAkBY,WAAW;CAErD,OAAOX,eAAezB,SAAS0B,KAAK;AACtC;AAEA,SAASY,2BACPC,UAC0C;CAC1C,IAAI,CAACA,UAAU,OAAO9B,KAAAA;CAEtB,KAAK,IAAI+B,MAAMD,SAASE,SAAS,GAAGD,OAAO,GAAGA,OAAO,GAAG;EACtD,MAAMN,UAAUK,SAASC;EACzB,IAAIP,2BAA2BC,OAAO,GACpC,OAAOA;CAEX;AAGF;;;;;;;;;AAUA,SAAOQ,sBAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CACL,MAAAC,MAAYtD,YAAYuD,KAAoD;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAAE,KAAA;EACzDE,KAAAd,2BAA2BY,GAAG;EAACF,EAAA,KAAAE;EAAAF,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,OAA/BI;AAA+B;AAF/C,SAAAD,MAAAE,GAAA;CAAA,OAC0BV,2BAA2BU,EAACC,OAAOV,QAAS;AAAC"}
+{"version":3,"file":"usage.js","names":["useMemo","useAuiState","ThreadTokenUsage","totalTokens","inputTokens","outputTokens","reasoningTokens","cachedInputTokens","TokenUsageExtractableMessage","role","metadata","UsageRecord","Record","USAGE_KEYS","const","asRecord","value","Array","isArray","undefined","asPositiveTokenCount","Number","isFinite","computeTotalTokens","usage","normalizeUsage","record","result","hasFields","key","count","outputTokenDetails","inputTokenDetails","cacheReadTokens","withComputedTotal","usageFromSteps","steps","sums","present","stepsWithUsage","stepsWithComputableTotal","step","stepTotal","getThreadMessageTokenUsage","message","topLevelUsage","customUsage","custom","findLatestMessageWithUsage","messages","idx","length","useThreadTokenUsage","$","_c","msg","_temp","t0","s","thread","_c"],"sources":["../src/usage.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/react\" preserve=\"true\" />\nimport { useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\n\nexport type ThreadTokenUsage = {\n  totalTokens?: number;\n  inputTokens?: number;\n  outputTokens?: number;\n  reasoningTokens?: number;\n  cachedInputTokens?: number;\n};\n\nexport interface TokenUsageExtractableMessage {\n  role?: string;\n  metadata?: unknown;\n}\n\ntype UsageRecord = Record<string, unknown>;\n\nconst USAGE_KEYS = [\n  \"inputTokens\",\n  \"outputTokens\",\n  \"reasoningTokens\",\n  \"cachedInputTokens\",\n  \"totalTokens\",\n] as const satisfies (keyof ThreadTokenUsage)[];\n\nfunction asRecord(value: unknown): UsageRecord | undefined {\n  if (!value || typeof value !== \"object\" || Array.isArray(value))\n    return undefined;\n  return value as UsageRecord;\n}\n\nfunction asPositiveTokenCount(value: unknown): number | undefined {\n  if (typeof value !== \"number\" || !Number.isFinite(value) || value < 0) {\n    return undefined;\n  }\n  return value;\n}\n\nfunction computeTotalTokens(usage: ThreadTokenUsage): number | undefined {\n  if (usage.totalTokens !== undefined) return usage.totalTokens;\n  if (usage.inputTokens !== undefined && usage.outputTokens !== undefined) {\n    return usage.inputTokens + usage.outputTokens;\n  }\n  return undefined;\n}\n\nfunction normalizeUsage(value: unknown): ThreadTokenUsage | undefined {\n  const record = asRecord(value);\n  if (!record) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  let hasFields = false;\n  for (const key of USAGE_KEYS) {\n    const count = asPositiveTokenCount(record[key]);\n    if (count !== undefined) {\n      result[key] = count;\n      hasFields = true;\n    }\n  }\n  // AI SDK v7 moved these under token detail objects; v6 kept them top-level.\n  if (result.reasoningTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.outputTokenDetails)?.reasoningTokens,\n    );\n    if (count !== undefined) {\n      result.reasoningTokens = count;\n      hasFields = true;\n    }\n  }\n  if (result.cachedInputTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.inputTokenDetails)?.cacheReadTokens,\n    );\n    if (count !== undefined) {\n      result.cachedInputTokens = count;\n      hasFields = true;\n    }\n  }\n  return hasFields ? result : undefined;\n}\n\nfunction withComputedTotal(\n  usage: ThreadTokenUsage,\n): ThreadTokenUsage | undefined {\n  const totalTokens = computeTotalTokens(usage);\n  return { ...usage, ...(totalTokens !== undefined && { totalTokens }) };\n}\n\nfunction usageFromSteps(value: unknown): ThreadTokenUsage | undefined {\n  const steps = Array.isArray(value) ? value : [];\n\n  const sums: Record<string, number> = {};\n  const present: Record<string, boolean> = {};\n  let stepsWithUsage = 0;\n  let stepsWithComputableTotal = 0;\n\n  for (const step of steps) {\n    const usage = normalizeUsage(asRecord(step)?.usage);\n    if (!usage) continue;\n    stepsWithUsage++;\n\n    const stepTotal = computeTotalTokens(usage);\n    if (stepTotal !== undefined) {\n      sums.totalTokens = (sums.totalTokens ?? 0) + stepTotal;\n      stepsWithComputableTotal++;\n    }\n\n    for (const key of USAGE_KEYS) {\n      if (key === \"totalTokens\") continue;\n      if (usage[key] !== undefined) {\n        sums[key] = (sums[key] ?? 0) + usage[key];\n        present[key] = true;\n      }\n    }\n  }\n\n  if (stepsWithUsage === 0) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  if (stepsWithComputableTotal === stepsWithUsage) {\n    result.totalTokens = sums.totalTokens!;\n  }\n  for (const key of USAGE_KEYS) {\n    if (key === \"totalTokens\") continue;\n    if (present[key]) {\n      result[key] = sums[key]!;\n    }\n  }\n  return result;\n}\n\nexport function getThreadMessageTokenUsage(\n  message: TokenUsageExtractableMessage | undefined,\n): ThreadTokenUsage | undefined {\n  if (!message || message.role !== \"assistant\") return undefined;\n\n  const metadata = asRecord(message.metadata);\n  if (!metadata) return undefined;\n\n  const topLevelUsage = normalizeUsage(metadata.usage);\n  if (topLevelUsage) return withComputedTotal(topLevelUsage);\n\n  const customUsage = normalizeUsage(asRecord(metadata.custom)?.usage);\n  if (customUsage) return withComputedTotal(customUsage);\n\n  return usageFromSteps(metadata.steps);\n}\n\nfunction findLatestMessageWithUsage(\n  messages: readonly TokenUsageExtractableMessage[] | undefined,\n): TokenUsageExtractableMessage | undefined {\n  if (!messages) return undefined;\n\n  for (let idx = messages.length - 1; idx >= 0; idx -= 1) {\n    const message = messages[idx];\n    if (getThreadMessageTokenUsage(message)) {\n      return message;\n    }\n  }\n\n  return undefined;\n}\n\n/**\n * Reads token usage from the newest assistant message that reports any.\n *\n * A route attaches usage through the AI SDK's `messageMetadata` option. Because\n * a thread message carries a fixed metadata shape, the converter moves every\n * other key the route returns into `metadata.custom`, which is where this hook\n * looks.\n */\nexport function useThreadTokenUsage(): ThreadTokenUsage | undefined {\n  const msg = useAuiState((s) => findLatestMessageWithUsage(s.thread.messages));\n  return useMemo(() => getThreadMessageTokenUsage(msg), [msg]);\n}\n"],"mappings":";;;;AAmBA,MAAMa,aAAa;CACjB;CACA;CACA;CACA;CACA;AAAa;AAGf,SAASE,SAASC,OAAyC;CACzD,IAAI,CAACA,SAAS,OAAOA,UAAU,YAAYC,MAAMC,QAAQF,KAAK,GAC5D,OAAOG,KAAAA;CACT,OAAOH;AACT;AAEA,SAASI,qBAAqBJ,OAAoC;CAChE,IAAI,OAAOA,UAAU,YAAY,CAACK,OAAOC,SAASN,KAAK,KAAKA,QAAQ,GAClE;CAEF,OAAOA;AACT;AAEA,SAASO,mBAAmBC,OAA6C;CACvE,IAAIA,MAAMrB,gBAAgBgB,KAAAA,GAAW,OAAOK,MAAMrB;CAClD,IAAIqB,MAAMpB,gBAAgBe,KAAAA,KAAaK,MAAMnB,iBAAiBc,KAAAA,GAC5D,OAAOK,MAAMpB,cAAcoB,MAAMnB;AAGrC;AAEA,SAASoB,eAAeT,OAA8C;CACpE,MAAMU,SAASX,SAASC,KAAK;CAC7B,IAAI,CAACU,QAAQ,OAAOP,KAAAA;CAEpB,MAAMQ,SAA2B,CAAC;CAClC,IAAIC,YAAY;CAChB,KAAK,MAAMC,OAAOhB,YAAY;EAC5B,MAAMiB,QAAQV,qBAAqBM,OAAOG,IAAI;EAC9C,IAAIC,UAAUX,KAAAA,GAAW;GACvBQ,OAAOE,OAAOC;GACdF,YAAY;EACd;CACF;CAEA,IAAID,OAAOrB,oBAAoBa,KAAAA,GAAW;EACxC,MAAMW,QAAQV,qBACZL,SAASW,OAAOK,kBAAkB,CAAC,EAAEzB,eACvC;EACA,IAAIwB,UAAUX,KAAAA,GAAW;GACvBQ,OAAOrB,kBAAkBwB;GACzBF,YAAY;EACd;CACF;CACA,IAAID,OAAOpB,sBAAsBY,KAAAA,GAAW;EAC1C,MAAMW,QAAQV,qBACZL,SAASW,OAAOM,iBAAiB,CAAC,EAAEC,eACtC;EACA,IAAIH,UAAUX,KAAAA,GAAW;GACvBQ,OAAOpB,oBAAoBuB;GAC3BF,YAAY;EACd;CACF;CACA,OAAOA,YAAYD,SAASR,KAAAA;AAC9B;AAEA,SAASe,kBACPV,OAC8B;CAC9B,MAAMrB,cAAcoB,mBAAmBC,KAAK;CAC5C,OAAO;EAAE,GAAGA;EAAO,GAAIrB,gBAAgBgB,KAAAA,KAAa,EAAEhB,YAAY;CAAG;AACvE;AAEA,SAASgC,eAAenB,OAA8C;CACpE,MAAMoB,QAAQnB,MAAMC,QAAQF,KAAK,IAAIA,QAAQ,CAAA;CAE7C,MAAMqB,OAA+B,CAAC;CACtC,MAAMC,UAAmC,CAAC;CAC1C,IAAIC,iBAAiB;CACrB,IAAIC,2BAA2B;CAE/B,KAAK,MAAMC,QAAQL,OAAO;EACxB,MAAMZ,QAAQC,eAAeV,SAAS0B,IAAI,CAAC,EAAEjB,KAAK;EAClD,IAAI,CAACA,OAAO;EACZe;EAEA,MAAMG,YAAYnB,mBAAmBC,KAAK;EAC1C,IAAIkB,cAAcvB,KAAAA,GAAW;GAC3BkB,KAAKlC,eAAekC,KAAKlC,eAAe,KAAKuC;GAC7CF;EACF;EAEA,KAAK,MAAMX,OAAOhB,YAAY;GAC5B,IAAIgB,QAAQ,eAAe;GAC3B,IAAIL,MAAMK,SAASV,KAAAA,GAAW;IAC5BkB,KAAKR,QAAQQ,KAAKR,QAAQ,KAAKL,MAAMK;IACrCS,QAAQT,OAAO;GACjB;EACF;CACF;CAEA,IAAIU,mBAAmB,GAAG,OAAOpB,KAAAA;CAEjC,MAAMQ,SAA2B,CAAC;CAClC,IAAIa,6BAA6BD,gBAC/BZ,OAAOxB,cAAckC,KAAKlC;CAE5B,KAAK,MAAM0B,OAAOhB,YAAY;EAC5B,IAAIgB,QAAQ,eAAe;EAC3B,IAAIS,QAAQT,MACVF,OAAOE,OAAOQ,KAAKR;CAEvB;CACA,OAAOF;AACT;AAEA,SAAgBgB,2BACdC,SAC8B;CAC9B,IAAI,CAACA,WAAWA,QAAQnC,SAAS,aAAa,OAAOU,KAAAA;CAErD,MAAMT,WAAWK,SAAS6B,QAAQlC,QAAQ;CAC1C,IAAI,CAACA,UAAU,OAAOS,KAAAA;CAEtB,MAAM0B,gBAAgBpB,eAAef,SAASc,KAAK;CACnD,IAAIqB,eAAe,OAAOX,kBAAkBW,aAAa;CAEzD,MAAMC,cAAcrB,eAAeV,SAASL,SAASqC,MAAM,CAAC,EAAEvB,KAAK;CACnE,IAAIsB,aAAa,OAAOZ,kBAAkBY,WAAW;CAErD,OAAOX,eAAezB,SAAS0B,KAAK;AACtC;AAEA,SAASY,2BACPC,UAC0C;CAC1C,IAAI,CAACA,UAAU,OAAO9B,KAAAA;CAEtB,KAAK,IAAI+B,MAAMD,SAASE,SAAS,GAAGD,OAAO,GAAGA,OAAO,GAAG;EACtD,MAAMN,UAAUK,SAASC;EACzB,IAAIP,2BAA2BC,OAAO,GACpC,OAAOA;CAEX;AAGF;;;;;;;;;AAUA,SAAOQ,sBAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CACL,MAAAC,MAAYtD,YAAYuD,KAAoD;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAAE,KAAA;EACzDE,KAAAd,2BAA2BY,GAAG;EAACF,EAAA,KAAAE;EAAAF,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,OAA/BI;AAA+B;AAF/C,SAAAD,MAAAE,GAAA;CAAA,OAC0BV,2BAA2BU,EAACC,OAAOV,QAAS;AAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/ai-sdk",
-  "version": "0.0.9",
+  "version": "0.0.11",
   "description": "Vercel AI SDK adapter for assistant-ui",
   "keywords": [
     "ai-sdk",
@@ -46,13 +46,13 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@ai-sdk/mcp": "^2.0.49",
-    "@ai-sdk/react": "^4.0.104",
-    "@assistant-ui/core": "^0.3.22",
-    "@assistant-ui/store": "^0.3.16",
-    "@assistant-ui/tap": "^0.9.20",
-    "assistant-cloud": "^0.2.4",
-    "assistant-stream": "^0.3.46"
+    "@ai-sdk/mcp": "^2.0.63",
+    "@ai-sdk/react": "^4.0.126",
+    "@assistant-ui/core": "^0.3.24",
+    "@assistant-ui/store": "^0.3.17",
+    "@assistant-ui/tap": "^0.9.21",
+    "assistant-cloud": "^0.2.5",
+    "assistant-stream": "^0.3.48"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -68,16 +68,16 @@
     }
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.29",
+    "@assistant-ui/x-buildutils": "0.0.30",
     "@assistant-ui/x-react18": "0.0.0",
     "@testing-library/react": "^16.3.3",
     "@types/json-schema": "^7.0.15",
     "@types/react": "^19.3.0",
     "@types/react-dom": "^19.3.0",
-    "ai": "^7.0.101",
+    "ai": "^7.0.123",
     "jsdom": "^30.0.1",
     "react": "^19.3.0",
-    "vitest": "^5.0.1"
+    "vitest": "^5.0.3"
   },
   "publishConfig": {
     "access": "public",
@@ -95,6 +95,7 @@
   "scripts": {
     "build": "aui-build",
     "test": "vitest run",
+    "test:coverage": "vitest run --coverage --coverage.reporter=text --coverage.reporter=html --coverage.reporter=json-summary",
     "typecheck": "tsc --noEmit && tsc --noEmit -p tsconfig.test.json",
     "test:peer-react18": "vitest run --config vitest.peer-react18.config.ts",
     "test:react-compiler": "vitest run --config vitest.react-compiler.config.ts",