@assistant-ui/ai-sdk 0.0.1 → 0.0.3

package/dist/runtime/useExternalHistory.js

@@ -140,6 +140,16 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 				persistSettled(false);
 			}, 0);
 		});
+		const initialThreadState = runtimeRef.current.thread.getState();
+		wasRunningRef.current = initialThreadState.isRunning;
+		if (initialThreadState.isRunning) {
+			if (runStartRef.current == null) {
+				runStartRef.current = Date.now();
+				stepBoundariesRef.current = [];
+				toolCallCountRef.current = 0;
+				adapter.pin?.();
+			}
+		} else if (initialThreadState.messages.length > 0) persistSettled(true);
 		function persistSettled(ignoreRunning) {
 			persistTimerRef.current = null;
 			const latest = runtimeRef.current.thread.getState();

package/dist/runtime/useExternalHistory.js.map

@@ -1 +1 @@
-{"version":3,"file":"useExternalHistory.js","names":["AssistantRuntime","ThreadHistoryAdapter","ThreadMessage","MessageFormatAdapter","MessageFormatRepository","ExportedMessageRepository","getExternalStoreMessages","MessageRepository","useAui","useRef","useEffect","useState","RefObject","useCallback","useMemo","toExportedMessageRepository","toThreadMessages","messages","TMessage","survivingIds","Set","survivors","flatMap","m","message","console","warn","parentId","has","add","id","headId","isAwaitingToolApproval","status","type","reason","snapshotExternalMessages","Map","map","useExternalHistory","runtimeRef","historyAdapter","storageFormatAdapter","onSetMessages","loadedRef","itemEpoch","setItemEpoch","aui","optionalThreadListItem","threadListItem","source","hasLoaded","setHasLoaded","historyIds","persistedInnerIds","deferredTelemetryIds","persistedExternalMessages","onSetMessagesRef","current","formatAdapter","undefined","withFormat","Error","isLoading","loadHistory","repo","load","length","getId","converted","thread","import","tempRepo","getMessages","error","pin","remoteId","getState","subscribe","n","threadState","isRunning","runStartRef","persistTimerRef","ReturnType","setTimeout","persistInFlightRef","Promise","resolve","stepBoundariesRef","wasRunningRef","toolCallCountRef","adapter","unsubscribe","wasRunning","lastMsg","at","role","currentToolCallCount","content","filter","p","push","Date","now","clearTimeout","persistSettled","ignoreRunning","latest","boundaries","durationMs","lastAssistant","findLast","tcCount","totalSteps","stepDur","i","Math","round","stepTimestamps","endMs","start_ms","end_ms","telemetryOptions","then","changedRunMessageIds","externalMessages","previous","get","some","item","index","lastInnerMessageId","failedUpdateIds","getLastInnerId","msgs","toBatchItems","msg","idx","innerMessages","isTerminal","isAwaitingToolCalls","isReady","update","isPersistedMessage","batchItems","innerId","append","delete","reportTelemetry","nextSnapshot","catch","deleteMessage","messageId","deleteMessages","bind","messageIndex","findIndex","previousInnerMessages","slice","itemsToDelete","deletion"],"sources":["../../src/runtime/useExternalHistory.ts"],"sourcesContent":["\"use client\";\n\nimport type {\n  AssistantRuntime,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatRepository,\n  ExportedMessageRepository,\n} from \"@assistant-ui/core\";\nimport { getExternalStoreMessages } from \"@assistant-ui/core\";\nimport { MessageRepository } 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 isAwaitingToolApproval = (message: ThreadMessage) =>\n  message.status?.type === \"requires-action\" &&\n  message.status.reason === \"tool-calls\";\n\nconst snapshotExternalMessages = <TMessage>(\n  messages: readonly ThreadMessage[],\n) =>\n  new Map<string, TMessage[]>(\n    messages.map((message) => [\n      message.id,\n      [...getExternalStoreMessages<TMessage>(message)],\n    ]),\n  );\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) => {\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 persistedInnerIds = useRef(new Set<string>());\n  const deferredTelemetryIds = useRef(new Set<string>());\n  const persistedExternalMessages = useRef(new Map<string, TMessage[]>());\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        if (repo && repo.messages.length > 0) {\n          for (const m of repo.messages) {\n            persistedInnerIds.current.add(\n              storageFormatAdapter.getId(m.message),\n            );\n          }\n          const converted = toExportedMessageRepository(toThreadMessages, repo);\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          persistedExternalMessages.current =\n            snapshotExternalMessages<TMessage>(\n              converted.messages.map((m) => m.message),\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      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  ]);\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  useEffect(() => {\n    if (!formatAdapter) return;\n    const adapter = formatAdapter;\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      // Only act on the true→false transition\n      if (!wasRunning) return;\n\n      // Record step boundary offset (synchronous for accuracy)\n      if (runStartRef.current != null) {\n        stepBoundariesRef.current.push(Date.now() - runStartRef.current);\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    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 changedRunMessageIds = new Set<string>();\n          for (const message of latest.messages) {\n            const externalMessages =\n              getExternalStoreMessages<TMessage>(message);\n            const previous = persistedExternalMessages.current.get(message.id);\n            if (\n              previous === undefined ||\n              previous.length !== externalMessages.length ||\n              externalMessages.some((item, index) => item !== previous[index])\n            ) {\n              changedRunMessageIds.add(message.id);\n            }\n          }\n\n          const { messages } = latest;\n          let lastInnerMessageId: string | null = null;\n          const failedUpdateIds = new Set<string>();\n\n          const getLastInnerId = (msgs: TMessage[]): string | null =>\n            msgs.length > 0 ? storageFormatAdapter.getId(msgs.at(-1)!) : null;\n\n          const toBatchItems = (msgs: TMessage[]) =>\n            msgs.map((msg, idx) => ({\n              parentId:\n                idx === 0\n                  ? lastInnerMessageId\n                  : storageFormatAdapter.getId(msgs[idx - 1]!),\n              message: msg,\n            }));\n\n          for (const message of messages) {\n            const innerMessages = getExternalStoreMessages<TMessage>(message);\n\n            const isTerminal =\n              message.status === undefined ||\n              message.status.type === \"complete\" ||\n              message.status.type === \"incomplete\";\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            const isPersistedMessage = historyIds.current.has(message.id);\n            if (isPersistedMessage && !changedRunMessageIds.has(message.id)) {\n              lastInnerMessageId =\n                getLastInnerId(innerMessages) ?? lastInnerMessageId;\n              continue;\n            }\n            if (!isPersistedMessage) {\n              historyIds.current.add(message.id);\n              deferredTelemetryIds.current.add(message.id);\n            }\n\n            const batchItems = toBatchItems(innerMessages);\n            for (const item of batchItems) {\n              const innerId = storageFormatAdapter.getId(item.message);\n              if (!persistedInnerIds.current.has(innerId)) {\n                await adapter.append(item);\n                persistedInnerIds.current.add(innerId);\n              } else if (durationMs !== undefined) {\n                try {\n                  await adapter.update?.(item, innerId);\n                } catch {\n                  // A failed update drops the message from the refreshed baseline so it retries on the next run stop.\n                  failedUpdateIds.add(message.id);\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?.(batchItems, telemetryOptions);\n            }\n          }\n\n          const nextSnapshot = snapshotExternalMessages<TMessage>(\n            latest.messages,\n          );\n          for (const id of failedUpdateIds) {\n            nextSnapshot.delete(id);\n          }\n          persistedExternalMessages.current = nextSnapshot;\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  }, [formatAdapter, storageFormatAdapter, runtimeRef]);\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        persistedExternalMessages.current.delete(messageId);\n        for (const item of itemsToDelete) {\n          persistedInnerIds.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 { isLoading, deleteMessage };\n};\n"],"mappings":";;;;;;AAsBA,MAAae,+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,0BAA0BR,YAC9BA,QAAQS,QAAQC,SAAS,qBACzBV,QAAQS,OAAOE,WAAW;AAE5B,MAAMC,4BACJnB,aAEA,IAAIoB,IACFpB,SAASqB,KAAKd,YAAY,CACxBA,QAAQM,IACR,CAAC,GAAGxB,yBAAmCkB,OAAO,CAAC,CAAC,CACjD,CACH;AAEF,MAAae,sBACXC,YACAC,gBACAzB,kBACA0B,sBACAC,kBACG;CACH,MAAMC,YAAYnC,OAAO,KAAK;CAC9B,MAAM,CAACoC,WAAWC,gBAAgBnC,SAAS,CAAC;CAE5C,MAAMoC,MAAMvC,OAAO;CACnB,MAAMwC,yBAAyBnC,kBACtBkC,IAAIE,eAAeC,SAASH,IAAIE,iBAAiB,MACxD,CAACF,GAAG,CACN;CAEA,MAAM,CAACI,WAAWC,gBAAgBzC,SAAS,KAAK;CAEhD,MAAM0C,aAAa5C,uBAAO,IAAIW,IAAY,CAAC;CAC3C,MAAMkC,oBAAoB7C,uBAAO,IAAIW,IAAY,CAAC;CAClD,MAAMmC,uBAAuB9C,uBAAO,IAAIW,IAAY,CAAC;CACrD,MAAMoC,4BAA4B/C,uBAAO,IAAI4B,IAAwB,CAAC;CAEtE,MAAMoB,mBAAmBhD,OAAOkC,aAAa;CAC7CjC,gBAAgB;EACd+C,iBAAiBC,UAAUf;CAC7B,CAAC;CAED,MAAMgB,gBAAgB7C,cAAc;EAClC,IAAI,CAAC2B,gBAAgB,OAAOmB,KAAAA;EAC5B,IAAI,CAACnB,eAAeoB,YAClB,MAAM,IAAIC,MACR,oFACF;EAEF,OAAOrB,eAAeoB,WAA0BnB,oBAAoB;CACtE,GAAG,CAACD,gBAAgBC,oBAAoB,CAAC;CAEzC,MAAMqB,YAAYJ,iBAAiB,QAAQ,CAACR;CAE5CzC,gBAAgB;EACd,IAAI,CAACiD,iBAAiBf,UAAUc,SAAS,OAAOE,KAAAA;EAEhD,MAAMI,cAAc,YAAY;GAC9B,IAAI;IACF,MAAMC,OAAO,MAAMN,cAAcO,KAAK;IACtC,IAAID,QAAQA,KAAKhD,SAASkD,SAAS,GAAG;KACpC,KAAK,MAAM5C,KAAK0C,KAAKhD,UACnBqC,kBAAkBI,QAAQ7B,IACxBa,qBAAqB0B,MAAM7C,EAAEC,OAAO,CACtC;KAEF,MAAM6C,YAAYtD,4BAA4BC,kBAAkBiD,IAAI;KACpEzB,WAAWkB,QAAQY,OAAOC,OAAOF,SAAS;KAE1C,MAAMG,WAAW,IAAIjE,kBAAkB;KACvCiE,SAASD,OAAOF,SAAS;KACzB,MAAMpD,WAAWuD,SAASC,YAAY;KAEtChB,iBAAiBC,QACfzC,SAASK,QAAQhB,wBAAkC,CACrD;KAEA+C,WAAWK,0BAAU,IAAItC,IAAI;KAC7B,KAAK,MAAMG,OAAK8C,UAAUpD,UAAU;MAClCoC,WAAWK,QAAQ7B,IAAIN,IAAEC,QAAQM,EAAE;MACnC,IAAIE,uBAAuBT,IAAEC,OAAO,GAClC+B,qBAAqBG,QAAQ7B,IAAIN,IAAEC,QAAQM,EAAE;KAEjD;KACA0B,0BAA0BE,UACxBtB,yBACEiC,UAAUpD,SAASqB,KAAKf,QAAMA,IAAEC,OAAO,CACzC;IACJ;GACF,SAASkD,OAAO;IACdjD,QAAQiD,MAAM,mCAAmCA,KAAK;GACxD,UAAU;IACRtB,aAAa,IAAI;GACnB;EACF;EAEAO,cAAcgB,MAAM;EAGpB,IAAI,CADa3B,uBAAuB,CAAC,EAAE6B,SAAS,CAAC,CAACD,UACvC;GACbxB,aAAa,IAAI;GACjB,OAAOL,IAAI+B,gBAAgB;IACzB,IAAI9B,uBAAuB,CAAC,EAAE6B,SAAS,CAAC,CAACD,UACvC9B,cAAciC,MAAMA,IAAI,CAAC;GAE7B,CAAC;EACH;EAEA,MAAMC,cAAcxC,WAAWkB,QAAQY,OAAOO,SAAS;EACvD,IAAIG,YAAYC,aAAaD,YAAY/D,SAASkD,SAAS,GAAG;GAC5DvB,UAAUc,UAAU;GACpBN,aAAa,IAAI;GACjB;EACF;EAEAR,UAAUc,UAAU;EACpB,YAAiB;CAEnB,GAAG;EACDC;EACA3C;EACAwB;EACAQ;EACAD;EACAF;EACAH;CAAoB,CACrB;CAED,MAAMwC,cAAczE,OAAsB,IAAI;CAC9C,MAAM0E,kBAAkB1E,OAA6C,IAAI;CACzE,MAAM6E,qBAAqB7E,OAAsB8E,QAAQC,QAAQ,CAAC;CAClE,MAAMC,oBAAoBhF,OAAiB,CAAA,CAAE;CAC7C,MAAMiF,gBAAgBjF,OAAO,KAAK;CAClC,MAAMkF,mBAAmBlF,OAAO,CAAC;CAEjCC,gBAAgB;EACd,IAAI,CAACiD,eAAe;EACpB,MAAMiC,UAAUjC;EAEhB,MAAMkC,cAAcrD,WAAWkB,QAAQY,OAAOQ,gBAAgB;GAC5D,MAAME,gBAAcxC,WAAWkB,QAAQY,OAAOO,SAAS;GACvD,MAAM,EAAEI,cAAcD;GACtB,MAAMc,aAAaJ,cAAchC;GACjCgC,cAAchC,UAAUuB;GAGxB,IAAIC,YAAYxB,WAAW,MAAM;IAC/B,MAAMqC,UAAUf,cAAY/D,SAAS+E,GAAG,EAAE;IAC1C,IAAID,SAASE,SAAS,aAAa;KACjC,MAAMC,uBAAuBH,QAAQI,QAAQC,QAC1CC,MAAMA,EAAEnE,SAAS,WACpB,CAAC,CAACiC;KACF,OAAOwB,iBAAiBjC,UAAUwC,sBAAsB;MACtDT,kBAAkB/B,QAAQ4C,KAAKC,KAAKC,IAAI,IAAItB,YAAYxB,OAAO;MAC/DiC,iBAAiBjC;KACnB;IACF;GACF;GAEA,IAAIuB,WAAW;IACb,IAAIC,YAAYxB,WAAW,MAAM;KAC/BwB,YAAYxB,UAAU6C,KAAKC,IAAI;KAC/Bf,kBAAkB/B,UAAU,CAAA;KAC5BiC,iBAAiBjC,UAAU;KAC3BkC,QAAQjB,MAAM;IAChB;IAEA,IAAIQ,gBAAgBzB,SAAS;KAC3B+C,aAAatB,gBAAgBzB,OAAO;KACpCyB,gBAAgBzB,UAAU;IAC5B;IACA;GACF;GAGA,IAAI,CAACoC,YAAY;GAGjB,IAAIZ,YAAYxB,WAAW,MACzB+B,kBAAkB/B,QAAQ4C,KAAKC,KAAKC,IAAI,IAAItB,YAAYxB,OAAO;GAIjE,IAAIyB,gBAAgBzB,SAAS+C,aAAatB,gBAAgBzB,OAAO;GACjEyB,gBAAgBzB,UAAU2B,iBAAiB;IACzCqB,eAAe,KAAK;GACtB,GAAG,CAAC;EACN,CAAC;EAED,SAASA,eAAeC,eAAwB;GAC9CxB,gBAAgBzB,UAAU;GAC1B,MAAMkD,SAASpE,WAAWkB,QAAQY,OAAOO,SAAS;GAClD,IAAI,CAAC8B,iBAAiBC,OAAO3B,WAAW;GAExC,MAAM4B,aAAapB,kBAAkB/B;GACrC,MAAMoD,aAAaD,WAAW1C,SAAS,IAAI0C,WAAWb,GAAG,EAAE,IAAIpC,KAAAA;GAG/D,IAAIiD,WAAW1C,WAAW,KAAK2C,cAAc,MAAM;IACjD,MAAMC,gBAAgBH,OAAO3F,SAAS+F,UACnCzF,QAAMA,IAAE0E,SAAS,WACpB;IACA,IAAIc,eAAe;KACjB,MAAME,UAAUF,cAAcZ,QAAQC,QACnCC,QAAMA,IAAEnE,SAAS,WACpB,CAAC,CAACiC;KACF,IAAI8C,UAAU,GAAG;MACf,MAAMC,aAAaD,UAAU;MAC7B,MAAME,UAAUL,aAAaI;MAC7BL,WAAW1C,SAAS;MACpB,KAAK,IAAIiD,IAAI,GAAGA,IAAIF,YAAYE,KAC9BP,WAAWP,KAAKe,KAAKC,OAAOF,IAAI,KAAKD,OAAO,CAAC;KAEjD;IACF;GACF;GAGA,MAAMI,iBACJV,WAAW1C,SAAS,IAChB0C,WAAWvE,KAAKkF,OAAOJ,SAAO;IAC5BK,UAAUL,QAAM,IAAI,IAAIP,WAAWO,MAAI;IACvCM,QAAQF;GACV,EAAE,IACF5D,KAAAA;GAENsB,YAAYxB,UAAU;GACtB+B,kBAAkB/B,UAAU,CAAA;GAE5B,MAAMiE,mBAAmB;IACvB,GAAIb,cAAc,OAAO,EAAEA,WAAW,IAAIlD,KAAAA;IAC1C,GAAI2D,kBAAkB,OAAO,EAAEA,eAAe,IAAI3D,KAAAA;GACpD;GAEA0B,mBAAmB5B,UAAU4B,mBAAmB5B,QAC7CkE,KAAK,YAAY;IAChB,MAAMC,uCAAuB,IAAIzG,IAAY;IAC7C,KAAK,MAAMI,WAAWoF,OAAO3F,UAAU;KACrC,MAAM6G,mBACJxH,yBAAmCkB,OAAO;KAC5C,MAAMuG,WAAWvE,0BAA0BE,QAAQsE,IAAIxG,QAAQM,EAAE;KACjE,IACEiG,aAAanE,KAAAA,KACbmE,SAAS5D,WAAW2D,iBAAiB3D,UACrC2D,iBAAiBG,MAAMC,MAAMC,UAAUD,SAASH,SAASI,MAAM,GAE/DN,qBAAqBhG,IAAIL,QAAQM,EAAE;IAEvC;IAEA,MAAM,EAAEb,UAAAA,eAAa2F;IACrB,IAAIwB,qBAAoC;IACxC,MAAMC,kCAAkB,IAAIjH,IAAY;IAExC,MAAMkH,kBAAkBC,SACtBA,KAAKpE,SAAS,IAAIzB,qBAAqB0B,MAAMmE,KAAKvC,GAAG,EAAE,CAAE,IAAI;IAE/D,MAAMwC,gBAAgBD,WACpBA,OAAKjG,KAAKmG,KAAKC,SAAS;KACtB/G,UACE+G,QAAQ,IACJN,qBACA1F,qBAAqB0B,MAAMmE,OAAKG,MAAM,EAAG;KAC/ClH,SAASiH;IACX,EAAE;IAEJ,KAAK,MAAMjH,aAAWP,YAAU;KAC9B,MAAM0H,gBAAgBrI,yBAAmCkB,SAAO;KAEhE,MAAMoH,aACJpH,UAAQS,WAAW2B,KAAAA,KACnBpC,UAAQS,OAAOC,SAAS,cACxBV,UAAQS,OAAOC,SAAS;KAC1B,MAAM2G,sBAAsB7G,uBAAuBR,SAAO;KAM1D,IAAI,EAHFoH,cACCC,uBAAuBjD,QAAQmD,WAAWnF,KAAAA,IAE/B;MACZwE,qBACEE,eAAeK,aAAa,KAAKP;MACnC;KACF;KAEA,MAAMY,qBAAqB3F,WAAWK,QAAQ9B,IAAIJ,UAAQM,EAAE;KAC5D,IAAIkH,sBAAsB,CAACnB,qBAAqBjG,IAAIJ,UAAQM,EAAE,GAAG;MAC/DsG,qBACEE,eAAeK,aAAa,KAAKP;MACnC;KACF;KACA,IAAI,CAACY,oBAAoB;MACvB3F,WAAWK,QAAQ7B,IAAIL,UAAQM,EAAE;MACjCyB,qBAAqBG,QAAQ7B,IAAIL,UAAQM,EAAE;KAC7C;KAEA,MAAMmH,aAAaT,aAAaG,aAAa;KAC7C,KAAK,MAAMT,UAAQe,YAAY;MAC7B,MAAMC,UAAUxG,qBAAqB0B,MAAM8D,OAAK1G,OAAO;MACvD,IAAI,CAAC8B,kBAAkBI,QAAQ9B,IAAIsH,OAAO,GAAG;OAC3C,MAAMtD,QAAQuD,OAAOjB,MAAI;OACzB5E,kBAAkBI,QAAQ7B,IAAIqH,OAAO;MACvC,OAAO,IAAIpC,eAAelD,KAAAA,GACxB,IAAI;OACF,MAAMgC,QAAQmD,SAASb,QAAMgB,OAAO;MACtC,QAAQ;OAENb,gBAAgBxG,IAAIL,UAAQM,EAAE;MAChC;KAEJ;KAEAsG,qBACEE,eAAeK,aAAa,KAAKP;KAEnC,IAAI7E,qBAAqBG,QAAQ9B,IAAIJ,UAAQM,EAAE,KAAK8G,YAAY;MAC9DrF,qBAAqBG,QAAQ0F,OAAO5H,UAAQM,EAAE;MAC9C8D,QAAQyD,kBAAkBJ,YAAYtB,gBAAgB;KACxD;IACF;IAEA,MAAM2B,eAAelH,yBACnBwE,OAAO3F,QACT;IACA,KAAK,MAAMa,MAAMuG,iBACfiB,aAAaF,OAAOtH,EAAE;IAExB0B,0BAA0BE,UAAU4F;GACtC,CAAC,CAAC,CACDC,OAAO7E,YAAU;IAChBjD,QAAQiD,MAAM,sCAAsCA,OAAK;GAC3D,CAAC;EACL;EAEA,aAAa;GACXmB,YAAY;GACZ,IAAIV,gBAAgBzB,SAAS;IAC3B+C,aAAatB,gBAAgBzB,OAAO;IACpCyB,gBAAgBzB,UAAU;IAC1BgD,eAAe,KAAK;GACtB;EACF;CACF,GAAG;EAAC/C;EAAejB;EAAsBF;CAAU,CAAC;CA4CpD,OAAO;EAAEuB;EAAWyF,eA1CE3I,YACpB,OAAO4I,cAAsB;GAC3B,MAAMC,iBAAiB/F,eAAeyF,QAAQO,KAAKhG,aAAa;GAChE,IAAI,CAAC+F,gBAAgB;GAErB,MAAMzI,aAAWuB,WAAWkB,QAAQY,OAAOO,SAAS,CAAC,CAAC5D;GACtD,MAAM2I,eAAe3I,WAAS4I,WAAWtI,QAAMA,IAAEO,OAAO2H,SAAS;GACjE,IAAIG,iBAAiB,IAAI;GAEzB,MAAME,wBAAwB7I,WAC3B8I,MAAM,GAAGH,YAAY,CAAC,CACtBtI,QAAQhB,wBAAkC;GAC7C,IAAIqB,WAAWmI,sBAAsB9D,GAAG,EAAE,IACtCtD,qBAAqB0B,MAAM0F,sBAAsB9D,GAAG,EAAE,CAAE,IACxD;GACJ,MAAMgE,gBAAgB1J,yBACpBW,WAAS2I,aACX,CAAC,CAACtH,KAAKd,cAAY;IACjB,MAAM0G,SAAO;KAAEvG;KAAUH,SAAAA;IAAQ;IACjCG,WAAWe,qBAAqB0B,MAAM5C,SAAO;IAC7C,OAAO0G;GACT,CAAC;GAED,MAAM+B,WAAW3E,mBAAmB5B,QAAQkE,KAAK,YAAY;IAC3D,MAAM8B,eAAeM,aAAa;IAElC3G,WAAWK,QAAQ0F,OAAOK,SAAS;IACnClG,qBAAqBG,QAAQ0F,OAAOK,SAAS;IAC7CjG,0BAA0BE,QAAQ0F,OAAOK,SAAS;IAClD,KAAK,MAAMvB,UAAQ8B,eACjB1G,kBAAkBI,QAAQ0F,OACxB1G,qBAAqB0B,MAAM8D,OAAK1G,OAAO,CACzC;GAEJ,CAAC;GAED8D,mBAAmB5B,UAAUuG,SAASV,YAAY,CAAC,CAAC;GACpD,MAAMU;EACR,GACA;GAACtG;GAAenB;GAAYE;EAAoB,CAG9B8G;CAAc;AACpC"}
+{"version":3,"file":"useExternalHistory.js","names":["AssistantRuntime","ThreadHistoryAdapter","ThreadMessage","MessageFormatAdapter","MessageFormatRepository","ExportedMessageRepository","getExternalStoreMessages","MessageRepository","useAui","useRef","useEffect","useState","RefObject","useCallback","useMemo","toExportedMessageRepository","toThreadMessages","messages","TMessage","survivingIds","Set","survivors","flatMap","m","message","console","warn","parentId","has","add","id","headId","isAwaitingToolApproval","status","type","reason","snapshotExternalMessages","Map","map","useExternalHistory","runtimeRef","historyAdapter","storageFormatAdapter","onSetMessages","loadedRef","itemEpoch","setItemEpoch","aui","optionalThreadListItem","threadListItem","source","hasLoaded","setHasLoaded","historyIds","persistedInnerIds","deferredTelemetryIds","persistedExternalMessages","onSetMessagesRef","current","formatAdapter","undefined","withFormat","Error","isLoading","loadHistory","repo","load","length","getId","converted","thread","import","tempRepo","getMessages","error","pin","remoteId","getState","subscribe","n","threadState","isRunning","runStartRef","persistTimerRef","ReturnType","setTimeout","persistInFlightRef","Promise","resolve","stepBoundariesRef","wasRunningRef","toolCallCountRef","adapter","unsubscribe","wasRunning","lastMsg","at","role","currentToolCallCount","content","filter","p","push","Date","now","clearTimeout","persistSettled","initialThreadState","ignoreRunning","latest","boundaries","durationMs","lastAssistant","findLast","tcCount","totalSteps","stepDur","i","Math","round","stepTimestamps","endMs","start_ms","end_ms","telemetryOptions","then","changedRunMessageIds","externalMessages","previous","get","some","item","index","lastInnerMessageId","failedUpdateIds","getLastInnerId","msgs","toBatchItems","msg","idx","innerMessages","isTerminal","isAwaitingToolCalls","isReady","update","isPersistedMessage","batchItems","innerId","append","delete","reportTelemetry","nextSnapshot","catch","deleteMessage","messageId","deleteMessages","bind","messageIndex","findIndex","previousInnerMessages","slice","itemsToDelete","deletion"],"sources":["../../src/runtime/useExternalHistory.ts"],"sourcesContent":["\"use client\";\n\nimport type {\n  AssistantRuntime,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatRepository,\n  ExportedMessageRepository,\n} from \"@assistant-ui/core\";\nimport { getExternalStoreMessages } from \"@assistant-ui/core\";\nimport { MessageRepository } 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 isAwaitingToolApproval = (message: ThreadMessage) =>\n  message.status?.type === \"requires-action\" &&\n  message.status.reason === \"tool-calls\";\n\nconst snapshotExternalMessages = <TMessage>(\n  messages: readonly ThreadMessage[],\n) =>\n  new Map<string, TMessage[]>(\n    messages.map((message) => [\n      message.id,\n      [...getExternalStoreMessages<TMessage>(message)],\n    ]),\n  );\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) => {\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 persistedInnerIds = useRef(new Set<string>());\n  const deferredTelemetryIds = useRef(new Set<string>());\n  const persistedExternalMessages = useRef(new Map<string, TMessage[]>());\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        if (repo && repo.messages.length > 0) {\n          for (const m of repo.messages) {\n            persistedInnerIds.current.add(\n              storageFormatAdapter.getId(m.message),\n            );\n          }\n          const converted = toExportedMessageRepository(toThreadMessages, repo);\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          persistedExternalMessages.current =\n            snapshotExternalMessages<TMessage>(\n              converted.messages.map((m) => m.message),\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      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  ]);\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  useEffect(() => {\n    if (!formatAdapter) return;\n    const adapter = formatAdapter;\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      // Only act on the true→false transition\n      if (!wasRunning) return;\n\n      // Record step boundary offset (synchronous for accuracy)\n      if (runStartRef.current != null) {\n        stepBoundariesRef.current.push(Date.now() - runStartRef.current);\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 changedRunMessageIds = new Set<string>();\n          for (const message of latest.messages) {\n            const externalMessages =\n              getExternalStoreMessages<TMessage>(message);\n            const previous = persistedExternalMessages.current.get(message.id);\n            if (\n              previous === undefined ||\n              previous.length !== externalMessages.length ||\n              externalMessages.some((item, index) => item !== previous[index])\n            ) {\n              changedRunMessageIds.add(message.id);\n            }\n          }\n\n          const { messages } = latest;\n          let lastInnerMessageId: string | null = null;\n          const failedUpdateIds = new Set<string>();\n\n          const getLastInnerId = (msgs: TMessage[]): string | null =>\n            msgs.length > 0 ? storageFormatAdapter.getId(msgs.at(-1)!) : null;\n\n          const toBatchItems = (msgs: TMessage[]) =>\n            msgs.map((msg, idx) => ({\n              parentId:\n                idx === 0\n                  ? lastInnerMessageId\n                  : storageFormatAdapter.getId(msgs[idx - 1]!),\n              message: msg,\n            }));\n\n          for (const message of messages) {\n            const innerMessages = getExternalStoreMessages<TMessage>(message);\n\n            const isTerminal =\n              message.status === undefined ||\n              message.status.type === \"complete\" ||\n              message.status.type === \"incomplete\";\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            const isPersistedMessage = historyIds.current.has(message.id);\n            if (isPersistedMessage && !changedRunMessageIds.has(message.id)) {\n              lastInnerMessageId =\n                getLastInnerId(innerMessages) ?? lastInnerMessageId;\n              continue;\n            }\n            if (!isPersistedMessage) {\n              historyIds.current.add(message.id);\n              deferredTelemetryIds.current.add(message.id);\n            }\n\n            const batchItems = toBatchItems(innerMessages);\n            for (const item of batchItems) {\n              const innerId = storageFormatAdapter.getId(item.message);\n              if (!persistedInnerIds.current.has(innerId)) {\n                await adapter.append(item);\n                persistedInnerIds.current.add(innerId);\n              } else if (durationMs !== undefined) {\n                try {\n                  await adapter.update?.(item, innerId);\n                } catch {\n                  // A failed update drops the message from the refreshed baseline so it retries on the next run stop.\n                  failedUpdateIds.add(message.id);\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?.(batchItems, telemetryOptions);\n            }\n          }\n\n          const nextSnapshot = snapshotExternalMessages<TMessage>(\n            latest.messages,\n          );\n          for (const id of failedUpdateIds) {\n            nextSnapshot.delete(id);\n          }\n          persistedExternalMessages.current = nextSnapshot;\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  }, [formatAdapter, storageFormatAdapter, runtimeRef]);\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        persistedExternalMessages.current.delete(messageId);\n        for (const item of itemsToDelete) {\n          persistedInnerIds.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 { isLoading, deleteMessage };\n};\n"],"mappings":";;;;;;AAsBA,MAAae,+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,0BAA0BR,YAC9BA,QAAQS,QAAQC,SAAS,qBACzBV,QAAQS,OAAOE,WAAW;AAE5B,MAAMC,4BACJnB,aAEA,IAAIoB,IACFpB,SAASqB,KAAKd,YAAY,CACxBA,QAAQM,IACR,CAAC,GAAGxB,yBAAmCkB,OAAO,CAAC,CAAC,CACjD,CACH;AAEF,MAAae,sBACXC,YACAC,gBACAzB,kBACA0B,sBACAC,kBACG;CACH,MAAMC,YAAYnC,OAAO,KAAK;CAC9B,MAAM,CAACoC,WAAWC,gBAAgBnC,SAAS,CAAC;CAE5C,MAAMoC,MAAMvC,OAAO;CACnB,MAAMwC,yBAAyBnC,kBACtBkC,IAAIE,eAAeC,SAASH,IAAIE,iBAAiB,MACxD,CAACF,GAAG,CACN;CAEA,MAAM,CAACI,WAAWC,gBAAgBzC,SAAS,KAAK;CAEhD,MAAM0C,aAAa5C,uBAAO,IAAIW,IAAY,CAAC;CAC3C,MAAMkC,oBAAoB7C,uBAAO,IAAIW,IAAY,CAAC;CAClD,MAAMmC,uBAAuB9C,uBAAO,IAAIW,IAAY,CAAC;CACrD,MAAMoC,4BAA4B/C,uBAAO,IAAI4B,IAAwB,CAAC;CAEtE,MAAMoB,mBAAmBhD,OAAOkC,aAAa;CAC7CjC,gBAAgB;EACd+C,iBAAiBC,UAAUf;CAC7B,CAAC;CAED,MAAMgB,gBAAgB7C,cAAc;EAClC,IAAI,CAAC2B,gBAAgB,OAAOmB,KAAAA;EAC5B,IAAI,CAACnB,eAAeoB,YAClB,MAAM,IAAIC,MACR,oFACF;EAEF,OAAOrB,eAAeoB,WAA0BnB,oBAAoB;CACtE,GAAG,CAACD,gBAAgBC,oBAAoB,CAAC;CAEzC,MAAMqB,YAAYJ,iBAAiB,QAAQ,CAACR;CAE5CzC,gBAAgB;EACd,IAAI,CAACiD,iBAAiBf,UAAUc,SAAS,OAAOE,KAAAA;EAEhD,MAAMI,cAAc,YAAY;GAC9B,IAAI;IACF,MAAMC,OAAO,MAAMN,cAAcO,KAAK;IACtC,IAAID,QAAQA,KAAKhD,SAASkD,SAAS,GAAG;KACpC,KAAK,MAAM5C,KAAK0C,KAAKhD,UACnBqC,kBAAkBI,QAAQ7B,IACxBa,qBAAqB0B,MAAM7C,EAAEC,OAAO,CACtC;KAEF,MAAM6C,YAAYtD,4BAA4BC,kBAAkBiD,IAAI;KACpEzB,WAAWkB,QAAQY,OAAOC,OAAOF,SAAS;KAE1C,MAAMG,WAAW,IAAIjE,kBAAkB;KACvCiE,SAASD,OAAOF,SAAS;KACzB,MAAMpD,WAAWuD,SAASC,YAAY;KAEtChB,iBAAiBC,QACfzC,SAASK,QAAQhB,wBAAkC,CACrD;KAEA+C,WAAWK,0BAAU,IAAItC,IAAI;KAC7B,KAAK,MAAMG,OAAK8C,UAAUpD,UAAU;MAClCoC,WAAWK,QAAQ7B,IAAIN,IAAEC,QAAQM,EAAE;MACnC,IAAIE,uBAAuBT,IAAEC,OAAO,GAClC+B,qBAAqBG,QAAQ7B,IAAIN,IAAEC,QAAQM,EAAE;KAEjD;KACA0B,0BAA0BE,UACxBtB,yBACEiC,UAAUpD,SAASqB,KAAKf,QAAMA,IAAEC,OAAO,CACzC;IACJ;GACF,SAASkD,OAAO;IACdjD,QAAQiD,MAAM,mCAAmCA,KAAK;GACxD,UAAU;IACRtB,aAAa,IAAI;GACnB;EACF;EAEAO,cAAcgB,MAAM;EAGpB,IAAI,CADa3B,uBAAuB,CAAC,EAAE6B,SAAS,CAAC,CAACD,UACvC;GACbxB,aAAa,IAAI;GACjB,OAAOL,IAAI+B,gBAAgB;IACzB,IAAI9B,uBAAuB,CAAC,EAAE6B,SAAS,CAAC,CAACD,UACvC9B,cAAciC,MAAMA,IAAI,CAAC;GAE7B,CAAC;EACH;EAEA,MAAMC,cAAcxC,WAAWkB,QAAQY,OAAOO,SAAS;EACvD,IAAIG,YAAYC,aAAaD,YAAY/D,SAASkD,SAAS,GAAG;GAC5DvB,UAAUc,UAAU;GACpBN,aAAa,IAAI;GACjB;EACF;EAEAR,UAAUc,UAAU;EACpB,YAAiB;CAEnB,GAAG;EACDC;EACA3C;EACAwB;EACAQ;EACAD;EACAF;EACAH;CAAoB,CACrB;CAED,MAAMwC,cAAczE,OAAsB,IAAI;CAC9C,MAAM0E,kBAAkB1E,OAA6C,IAAI;CACzE,MAAM6E,qBAAqB7E,OAAsB8E,QAAQC,QAAQ,CAAC;CAClE,MAAMC,oBAAoBhF,OAAiB,CAAA,CAAE;CAC7C,MAAMiF,gBAAgBjF,OAAO,KAAK;CAClC,MAAMkF,mBAAmBlF,OAAO,CAAC;CAEjCC,gBAAgB;EACd,IAAI,CAACiD,eAAe;EACpB,MAAMiC,UAAUjC;EAEhB,MAAMkC,cAAcrD,WAAWkB,QAAQY,OAAOQ,gBAAgB;GAC5D,MAAME,gBAAcxC,WAAWkB,QAAQY,OAAOO,SAAS;GACvD,MAAM,EAAEI,cAAcD;GACtB,MAAMc,aAAaJ,cAAchC;GACjCgC,cAAchC,UAAUuB;GAGxB,IAAIC,YAAYxB,WAAW,MAAM;IAC/B,MAAMqC,UAAUf,cAAY/D,SAAS+E,GAAG,EAAE;IAC1C,IAAID,SAASE,SAAS,aAAa;KACjC,MAAMC,uBAAuBH,QAAQI,QAAQC,QAC1CC,MAAMA,EAAEnE,SAAS,WACpB,CAAC,CAACiC;KACF,OAAOwB,iBAAiBjC,UAAUwC,sBAAsB;MACtDT,kBAAkB/B,QAAQ4C,KAAKC,KAAKC,IAAI,IAAItB,YAAYxB,OAAO;MAC/DiC,iBAAiBjC;KACnB;IACF;GACF;GAEA,IAAIuB,WAAW;IACb,IAAIC,YAAYxB,WAAW,MAAM;KAC/BwB,YAAYxB,UAAU6C,KAAKC,IAAI;KAC/Bf,kBAAkB/B,UAAU,CAAA;KAC5BiC,iBAAiBjC,UAAU;KAC3BkC,QAAQjB,MAAM;IAChB;IAEA,IAAIQ,gBAAgBzB,SAAS;KAC3B+C,aAAatB,gBAAgBzB,OAAO;KACpCyB,gBAAgBzB,UAAU;IAC5B;IACA;GACF;GAGA,IAAI,CAACoC,YAAY;GAGjB,IAAIZ,YAAYxB,WAAW,MACzB+B,kBAAkB/B,QAAQ4C,KAAKC,KAAKC,IAAI,IAAItB,YAAYxB,OAAO;GAIjE,IAAIyB,gBAAgBzB,SAAS+C,aAAatB,gBAAgBzB,OAAO;GACjEyB,gBAAgBzB,UAAU2B,iBAAiB;IACzCqB,eAAe,KAAK;GACtB,GAAG,CAAC;EACN,CAAC;EAED,MAAMC,qBAAqBnE,WAAWkB,QAAQY,OAAOO,SAAS;EAC9Da,cAAchC,UAAUiD,mBAAmB1B;EAC3C,IAAI0B,mBAAmB1B,WACjBC;OAAAA,YAAYxB,WAAW,MAAM;IAC/BwB,YAAYxB,UAAU6C,KAAKC,IAAI;IAC/Bf,kBAAkB/B,UAAU,CAAA;IAC5BiC,iBAAiBjC,UAAU;IAC3BkC,QAAQjB,MAAM;GAChB;SACK,IAAIgC,mBAAmB1F,SAASkD,SAAS,GAC9CuC,eAAe,IAAI;EAGrB,SAASA,eAAeE,eAAwB;GAC9CzB,gBAAgBzB,UAAU;GAC1B,MAAMmD,SAASrE,WAAWkB,QAAQY,OAAOO,SAAS;GAClD,IAAI,CAAC+B,iBAAiBC,OAAO5B,WAAW;GAExC,MAAM6B,aAAarB,kBAAkB/B;GACrC,MAAMqD,aAAaD,WAAW3C,SAAS,IAAI2C,WAAWd,GAAG,EAAE,IAAIpC,KAAAA;GAG/D,IAAIkD,WAAW3C,WAAW,KAAK4C,cAAc,MAAM;IACjD,MAAMC,gBAAgBH,OAAO5F,SAASgG,UACnC1F,QAAMA,IAAE0E,SAAS,WACpB;IACA,IAAIe,eAAe;KACjB,MAAME,UAAUF,cAAcb,QAAQC,QACnCC,QAAMA,IAAEnE,SAAS,WACpB,CAAC,CAACiC;KACF,IAAI+C,UAAU,GAAG;MACf,MAAMC,aAAaD,UAAU;MAC7B,MAAME,UAAUL,aAAaI;MAC7BL,WAAW3C,SAAS;MACpB,KAAK,IAAIkD,IAAI,GAAGA,IAAIF,YAAYE,KAC9BP,WAAWR,KAAKgB,KAAKC,OAAOF,IAAI,KAAKD,OAAO,CAAC;KAEjD;IACF;GACF;GAGA,MAAMI,iBACJV,WAAW3C,SAAS,IAChB2C,WAAWxE,KAAKmF,OAAOJ,SAAO;IAC5BK,UAAUL,QAAM,IAAI,IAAIP,WAAWO,MAAI;IACvCM,QAAQF;GACV,EAAE,IACF7D,KAAAA;GAENsB,YAAYxB,UAAU;GACtB+B,kBAAkB/B,UAAU,CAAA;GAE5B,MAAMkE,mBAAmB;IACvB,GAAIb,cAAc,OAAO,EAAEA,WAAW,IAAInD,KAAAA;IAC1C,GAAI4D,kBAAkB,OAAO,EAAEA,eAAe,IAAI5D,KAAAA;GACpD;GAEA0B,mBAAmB5B,UAAU4B,mBAAmB5B,QAC7CmE,KAAK,YAAY;IAChB,MAAMC,uCAAuB,IAAI1G,IAAY;IAC7C,KAAK,MAAMI,WAAWqF,OAAO5F,UAAU;KACrC,MAAM8G,mBACJzH,yBAAmCkB,OAAO;KAC5C,MAAMwG,WAAWxE,0BAA0BE,QAAQuE,IAAIzG,QAAQM,EAAE;KACjE,IACEkG,aAAapE,KAAAA,KACboE,SAAS7D,WAAW4D,iBAAiB5D,UACrC4D,iBAAiBG,MAAMC,MAAMC,UAAUD,SAASH,SAASI,MAAM,GAE/DN,qBAAqBjG,IAAIL,QAAQM,EAAE;IAEvC;IAEA,MAAM,EAAEb,UAAAA,eAAa4F;IACrB,IAAIwB,qBAAoC;IACxC,MAAMC,kCAAkB,IAAIlH,IAAY;IAExC,MAAMmH,kBAAkBC,SACtBA,KAAKrE,SAAS,IAAIzB,qBAAqB0B,MAAMoE,KAAKxC,GAAG,EAAE,CAAE,IAAI;IAE/D,MAAMyC,gBAAgBD,WACpBA,OAAKlG,KAAKoG,KAAKC,SAAS;KACtBhH,UACEgH,QAAQ,IACJN,qBACA3F,qBAAqB0B,MAAMoE,OAAKG,MAAM,EAAG;KAC/CnH,SAASkH;IACX,EAAE;IAEJ,KAAK,MAAMlH,aAAWP,YAAU;KAC9B,MAAM2H,gBAAgBtI,yBAAmCkB,SAAO;KAEhE,MAAMqH,aACJrH,UAAQS,WAAW2B,KAAAA,KACnBpC,UAAQS,OAAOC,SAAS,cACxBV,UAAQS,OAAOC,SAAS;KAC1B,MAAM4G,sBAAsB9G,uBAAuBR,SAAO;KAM1D,IAAI,EAHFqH,cACCC,uBAAuBlD,QAAQoD,WAAWpF,KAAAA,IAE/B;MACZyE,qBACEE,eAAeK,aAAa,KAAKP;MACnC;KACF;KAEA,MAAMY,qBAAqB5F,WAAWK,QAAQ9B,IAAIJ,UAAQM,EAAE;KAC5D,IAAImH,sBAAsB,CAACnB,qBAAqBlG,IAAIJ,UAAQM,EAAE,GAAG;MAC/DuG,qBACEE,eAAeK,aAAa,KAAKP;MACnC;KACF;KACA,IAAI,CAACY,oBAAoB;MACvB5F,WAAWK,QAAQ7B,IAAIL,UAAQM,EAAE;MACjCyB,qBAAqBG,QAAQ7B,IAAIL,UAAQM,EAAE;KAC7C;KAEA,MAAMoH,aAAaT,aAAaG,aAAa;KAC7C,KAAK,MAAMT,UAAQe,YAAY;MAC7B,MAAMC,UAAUzG,qBAAqB0B,MAAM+D,OAAK3G,OAAO;MACvD,IAAI,CAAC8B,kBAAkBI,QAAQ9B,IAAIuH,OAAO,GAAG;OAC3C,MAAMvD,QAAQwD,OAAOjB,MAAI;OACzB7E,kBAAkBI,QAAQ7B,IAAIsH,OAAO;MACvC,OAAO,IAAIpC,eAAenD,KAAAA,GACxB,IAAI;OACF,MAAMgC,QAAQoD,SAASb,QAAMgB,OAAO;MACtC,QAAQ;OAENb,gBAAgBzG,IAAIL,UAAQM,EAAE;MAChC;KAEJ;KAEAuG,qBACEE,eAAeK,aAAa,KAAKP;KAEnC,IAAI9E,qBAAqBG,QAAQ9B,IAAIJ,UAAQM,EAAE,KAAK+G,YAAY;MAC9DtF,qBAAqBG,QAAQ2F,OAAO7H,UAAQM,EAAE;MAC9C8D,QAAQ0D,kBAAkBJ,YAAYtB,gBAAgB;KACxD;IACF;IAEA,MAAM2B,eAAenH,yBACnByE,OAAO5F,QACT;IACA,KAAK,MAAMa,MAAMwG,iBACfiB,aAAaF,OAAOvH,EAAE;IAExB0B,0BAA0BE,UAAU6F;GACtC,CAAC,CAAC,CACDC,OAAO9E,YAAU;IAChBjD,QAAQiD,MAAM,sCAAsCA,OAAK;GAC3D,CAAC;EACL;EAEA,aAAa;GACXmB,YAAY;GACZ,IAAIV,gBAAgBzB,SAAS;IAC3B+C,aAAatB,gBAAgBzB,OAAO;IACpCyB,gBAAgBzB,UAAU;IAC1BgD,eAAe,KAAK;GACtB;EACF;CACF,GAAG;EAAC/C;EAAejB;EAAsBF;CAAU,CAAC;CA4CpD,OAAO;EAAEuB;EAAW0F,eA1CE5I,YACpB,OAAO6I,cAAsB;GAC3B,MAAMC,iBAAiBhG,eAAe0F,QAAQO,KAAKjG,aAAa;GAChE,IAAI,CAACgG,gBAAgB;GAErB,MAAM1I,aAAWuB,WAAWkB,QAAQY,OAAOO,SAAS,CAAC,CAAC5D;GACtD,MAAM4I,eAAe5I,WAAS6I,WAAWvI,QAAMA,IAAEO,OAAO4H,SAAS;GACjE,IAAIG,iBAAiB,IAAI;GAEzB,MAAME,wBAAwB9I,WAC3B+I,MAAM,GAAGH,YAAY,CAAC,CACtBvI,QAAQhB,wBAAkC;GAC7C,IAAIqB,WAAWoI,sBAAsB/D,GAAG,EAAE,IACtCtD,qBAAqB0B,MAAM2F,sBAAsB/D,GAAG,EAAE,CAAE,IACxD;GACJ,MAAMiE,gBAAgB3J,yBACpBW,WAAS4I,aACX,CAAC,CAACvH,KAAKd,cAAY;IACjB,MAAM2G,SAAO;KAAExG;KAAUH,SAAAA;IAAQ;IACjCG,WAAWe,qBAAqB0B,MAAM5C,SAAO;IAC7C,OAAO2G;GACT,CAAC;GAED,MAAM+B,WAAW5E,mBAAmB5B,QAAQmE,KAAK,YAAY;IAC3D,MAAM8B,eAAeM,aAAa;IAElC5G,WAAWK,QAAQ2F,OAAOK,SAAS;IACnCnG,qBAAqBG,QAAQ2F,OAAOK,SAAS;IAC7ClG,0BAA0BE,QAAQ2F,OAAOK,SAAS;IAClD,KAAK,MAAMvB,UAAQ8B,eACjB3G,kBAAkBI,QAAQ2F,OACxB3G,qBAAqB0B,MAAM+D,OAAK3G,OAAO,CACzC;GAEJ,CAAC;GAED8D,mBAAmB5B,UAAUwG,SAASV,YAAY,CAAC,CAAC;GACpD,MAAMU;EACR,GACA;GAACvG;GAAenB;GAAYE;EAAoB,CAG9B+G;CAAc;AACpC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/ai-sdk",
-  "version": "0.0.1",
+  "version": "0.0.3",
   "description": "Vercel AI SDK adapter for assistant-ui",
   "keywords": [
     "ai-sdk",
@@ -46,14 +46,14 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@ai-sdk/mcp": "^2.0.31",
-    "@ai-sdk/react": "^4.0.65",
-    "@assistant-ui/core": "^0.3.14",
-    "@assistant-ui/store": "^0.3.10",
-    "@assistant-ui/tap": "^0.9.13",
-    "ai": "^7.0.62",
+    "@ai-sdk/mcp": "^2.0.36",
+    "@ai-sdk/react": "^4.0.80",
+    "@assistant-ui/core": "^0.3.16",
+    "@assistant-ui/store": "^0.3.11",
+    "@assistant-ui/tap": "^0.9.15",
+    "ai": "^7.0.77",
     "assistant-cloud": "*",
-    "assistant-stream": "^0.3.38"
+    "assistant-stream": "^0.3.40"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -68,14 +68,14 @@
     }
   },
   "devDependencies": {
+    "@assistant-ui/x-buildutils": "0.0.24",
     "@testing-library/react": "^16.3.2",
     "@types/json-schema": "^7.0.15",
     "@types/react": "^19.2.18",
-    "@types/react-dom": "^19.2.4",
+    "@types/react-dom": "^19.2.5",
     "jsdom": "^30.0.1",
     "react": "^19.2.8",
-    "vitest": "^4.1.10",
-    "@assistant-ui/x-buildutils": "0.0.23"
+    "vitest": "^4.1.11"
   },
   "publishConfig": {
     "access": "public",

package/src/converters/toCreateMessage.ts

@@ -1,8 +1,7 @@
 import type { AppendMessage } from "@assistant-ui/core";
 import {
-  httpUrlPattern,
-  parseDataUrl,
   resolveFileMediaType,
+  resolveFilePartSource,
   resolveImageMediaType,
   toMediaWireUrl,
 } from "@assistant-ui/core/internal";
@@ -72,11 +71,13 @@ export const toCreateMessage = <UI_MESSAGE extends UIMessage = UIMessage>(
         // envelope is rebuilt from the typed format rather than forwarded.
         const mediaType = `audio/${part.audio.format}`;
         const data = part.audio.data;
+        const source = resolveFilePartSource({ data, mimeType: mediaType });
         return {
           type: "file",
-          url: httpUrlPattern.test(data)
-            ? data
-            : `data:${mediaType};base64,${parseDataUrl(data)?.data ?? data}`,
+          url:
+            source.kind === "url"
+              ? source.url
+              : `data:${mediaType};base64,${source.data}`,
           mediaType,
           ...(part.filename && { filename: part.filename }),
         };

package/src/model-context/injectInteractableContext.ts

@@ -23,7 +23,7 @@ import {
  * @example
  * ```ts
  * import { convertToModelMessages, streamText } from "ai";
- * import { unstable_injectInteractableContext } from "@assistant-ui/react-ai-sdk";
+ * import { unstable_injectInteractableContext } from "@assistant-ui/ai-sdk";
  *
  * export async function POST(req: Request) {
  *   const { messages } = await req.json();

package/src/model-context/injectQuoteContext.ts

@@ -22,7 +22,7 @@ const getQuoteText = (metadata: unknown): string | undefined => {
  * @example
  * ```ts
  * import { convertToModelMessages, streamText } from "ai";
- * import { injectQuoteContext } from "@assistant-ui/react-ai-sdk";
+ * import { injectQuoteContext } from "@assistant-ui/ai-sdk";
  *
  * export async function POST(req: Request) {
  *   const { messages } = await req.json();

package/src/runtime/useAISDKRuntime.denied-tool.test.tsx

@@ -0,0 +1,122 @@
+// @vitest-environment jsdom
+
+import { renderHook } from "@testing-library/react";
+import type { ExternalStoreAdapter } from "@assistant-ui/core";
+import { describe, expect, it, vi } from "vitest";
+
+const mocks = vi.hoisted(() => ({
+  adapter: undefined as ExternalStoreAdapter | undefined,
+}));
+
+vi.mock("@assistant-ui/core/react", async (importOriginal) => {
+  const original =
+    await importOriginal<typeof import("@assistant-ui/core/react")>();
+  return {
+    ...original,
+    useExternalStoreRuntime: vi.fn((adapter: ExternalStoreAdapter) => {
+      mocks.adapter = adapter;
+      return {};
+    }),
+    useRuntimeAdapters: vi.fn(() => ({})),
+  };
+});
+
+vi.mock("./useExternalHistory", async (importOriginal) => {
+  const original =
+    await importOriginal<typeof import("./useExternalHistory")>();
+  return {
+    ...original,
+    useExternalHistory: vi.fn(() => ({
+      isLoading: false,
+      deleteMessage: vi.fn().mockResolvedValue(undefined),
+    })),
+  };
+});
+
+import { useAISDKRuntime } from "./useAISDKRuntime";
+
+describe("useAISDKRuntime pending tool completion", () => {
+  const setup = (toolPart: Record<string, unknown>) => {
+    const messages = [
+      {
+        id: "m1",
+        role: "assistant",
+        parts: [toolPart],
+      },
+    ];
+    let updated: any[] | undefined;
+    const setMessages = vi.fn(
+      (updater: (m: typeof messages) => typeof messages) => {
+        updated = typeof updater === "function" ? updater(messages) : updater;
+      },
+    );
+    const chat = {
+      id: "chat-1",
+      status: "ready",
+      error: undefined,
+      messages,
+      setMessages,
+      sendMessage: vi.fn().mockResolvedValue(undefined),
+      regenerate: vi.fn(),
+      addToolOutput: vi.fn(),
+      addToolApprovalResponse: vi.fn(),
+      stop: vi.fn(),
+    };
+
+    renderHook(() => useAISDKRuntime(chat as never));
+
+    return {
+      send: () =>
+        mocks.adapter!.onNew!({
+          role: "user",
+          parentId: "m1",
+          sourceId: null,
+          runConfig: {},
+          content: [{ type: "text", text: "next" }],
+          attachments: [],
+          createdAt: new Date(0),
+          metadata: { custom: {} },
+        } as never),
+      getUpdated: () => updated,
+    };
+  };
+
+  it("preserves a denied tool call and its approval record on the next send", async () => {
+    const { send, getUpdated } = setup({
+      type: "tool-search",
+      toolCallId: "call_1",
+      state: "output-denied",
+      input: { query: "x" },
+      approval: { id: "appr_1", approved: false, reason: "too dangerous" },
+    });
+
+    await send();
+
+    const updated = getUpdated();
+    const part = updated?.at(-1)?.parts?.[0];
+    expect(part?.state).toBe("output-denied");
+    expect(part?.approval).toEqual({
+      id: "appr_1",
+      approved: false,
+      reason: "too dangerous",
+    });
+    expect(part?.errorText).toBeUndefined();
+  });
+
+  it("still cancels a genuinely pending tool call on the next send", async () => {
+    const { send, getUpdated } = setup({
+      type: "tool-search",
+      toolCallId: "call_1",
+      state: "input-available",
+      input: { query: "x" },
+    });
+
+    await send();
+
+    const part = getUpdated()?.at(-1)?.parts?.[0];
+    expect(part?.state).toBe("output-error");
+    expect(part?.errorText).toBe(
+      "User cancelled tool call by sending a new message.",
+    );
+  });
+});

package/src/runtime/useAISDKRuntime.test.ts

@@ -7,13 +7,17 @@ import { validateUIMessages } from "ai";
 // Mock only the sibling module that requires AUI store context (not available
 // in isolation). Every other dependency — useExternalStoreRuntime,
 // useToolInvocations, the message converter — runs for real.
-vi.mock("./useExternalHistory", () => ({
-  useExternalHistory: vi.fn(() => ({
-    isLoading: false,
-    deleteMessage: vi.fn().mockResolvedValue(undefined),
-  })),
-  toExportedMessageRepository: vi.fn(),
-}));
+vi.mock("./useExternalHistory", async (importOriginal) => {
+  const original =
+    await importOriginal<typeof import("./useExternalHistory")>();
+  return {
+    ...original,
+    useExternalHistory: vi.fn(() => ({
+      isLoading: false,
+      deleteMessage: vi.fn().mockResolvedValue(undefined),
+    })),
+  };
+});
 
 import { useExternalHistory } from "./useExternalHistory";
 import { useAISDKRuntime } from "./useAISDKRuntime";
@@ -690,6 +694,186 @@ describe("useAISDKRuntime", () => {
     expect(result.current.thread.getState().suggestions).toEqual(suggestions);
   });
 
+  it("imports a message tree without replacing the chat feed", async () => {
+    const chat = createChatHelpers();
+    const onBranchChange = vi.fn();
+    const messageRepository = {
+      headId: "a2",
+      messages: [
+        {
+          parentId: null,
+          message: {
+            id: "u1",
+            role: "user" as const,
+            parts: [{ type: "text" as const, text: "question" }],
+          },
+        },
+        {
+          parentId: "u1",
+          message: {
+            id: "a1",
+            role: "assistant" as const,
+            parts: [{ type: "text" as const, text: "first" }],
+          },
+        },
+        {
+          parentId: "u1",
+          message: {
+            id: "a2",
+            role: "assistant" as const,
+            parts: [{ type: "text" as const, text: "second" }],
+          },
+        },
+      ],
+    };
+
+    const { result, rerender } = renderHook(() =>
+      useAISDKRuntime(chat, {
+        messageRepository,
+        unstable_onBranchChange: onBranchChange,
+      }),
+    );
+
+    await waitFor(() => {
+      expect(
+        result.current.thread.getState().messages.map((message) => message.id),
+      ).toEqual(["u1", "a2"]);
+    });
+    expect(result.current.thread.getMessageById("a2").getState()).toMatchObject(
+      {
+        branchNumber: 2,
+        branchCount: 2,
+      },
+    );
+    expect(chat.messages.map((message: { id: string }) => message.id)).toEqual([
+      "u1",
+      "a2",
+    ]);
+
+    act(() => {
+      result.current.thread
+        .getMessageById("a2")
+        .switchToBranch({ branchId: "a1" });
+    });
+
+    expect(onBranchChange).toHaveBeenCalledWith({
+      headId: "a1",
+      visibleMessageIds: ["u1", "a1"],
+    });
+    expect(chat.messages.map((message: { id: string }) => message.id)).toEqual([
+      "u1",
+      "a1",
+    ]);
+    expect(chat.messages).not.toEqual([]);
+
+    chat.messages = [
+      { id: "u1", role: "user", parts: [{ type: "text", text: "question" }] },
+      {
+        id: "a1",
+        role: "assistant",
+        parts: [{ type: "text", text: "first token" }],
+      },
+    ];
+    rerender();
+
+    await waitFor(() => {
+      expect(textOf(result.current.thread.getState().messages.at(-1))).toBe(
+        "first token",
+      );
+    });
+    expect(result.current.thread.getMessageById("a1").getState()).toMatchObject(
+      {
+        branchNumber: 1,
+        branchCount: 2,
+      },
+    );
+  });
+
+  it("does not overwrite a nonempty chat with the repository", async () => {
+    const chat = createChatHelpers([
+      { id: "live", role: "user", parts: [{ type: "text", text: "keep me" }] },
+    ]);
+    const messageRepository = {
+      headId: "a1",
+      messages: [
+        {
+          parentId: null,
+          message: {
+            id: "u1",
+            role: "user" as const,
+            parts: [{ type: "text" as const, text: "question" }],
+          },
+        },
+        {
+          parentId: "u1",
+          message: {
+            id: "a1",
+            role: "assistant" as const,
+            parts: [{ type: "text" as const, text: "first" }],
+          },
+        },
+      ],
+    };
+
+    const { result } = renderHook(() =>
+      useAISDKRuntime(chat, { messageRepository }),
+    );
+
+    await waitFor(() => {
+      expect(
+        result.current.thread.getState().messages.map((message) => message.id),
+      ).toEqual(["live"]);
+    });
+    expect(chat.messages.map((message: { id: string }) => message.id)).toEqual([
+      "live",
+    ]);
+  });
+
+  it("does not reseed when the repository object identity changes", async () => {
+    const chat = createChatHelpers();
+    const makeRepository = (text: string) => ({
+      headId: "a1",
+      messages: [
+        {
+          parentId: null,
+          message: {
+            id: "u1",
+            role: "user" as const,
+            parts: [{ type: "text" as const, text: "question" }],
+          },
+        },
+        {
+          parentId: "u1",
+          message: {
+            id: "a1",
+            role: "assistant" as const,
+            parts: [{ type: "text" as const, text }],
+          },
+        },
+      ],
+    });
+
+    const { result, rerender } = renderHook(
+      ({ repository }) =>
+        useAISDKRuntime(chat, { messageRepository: repository }),
+      { initialProps: { repository: makeRepository("first") } },
+    );
+
+    await waitFor(() => {
+      expect(textOf(result.current.thread.getState().messages.at(-1))).toBe(
+        "first",
+      );
+    });
+    const setMessagesCalls = chat.setMessages.mock.calls.length;
+
+    rerender({ repository: makeRepository("second") });
+
+    expect(textOf(result.current.thread.getState().messages.at(-1))).toBe(
+      "first",
+    );
+    expect(chat.setMessages.mock.calls.length).toBe(setMessagesCalls);
+  });
+
   it("calls adapters.suggestion after settle with messages and signal", async () => {
     const generate = vi.fn().mockResolvedValue([{ prompt: "next" }]);
     const chat = createChatHelpers([