@assistant-ui/ai-sdk 0.0.5 → 0.0.7

package/dist/runtime/useExternalHistory.js

@@ -25,6 +25,7 @@ const toExportedMessageRepository = (toThreadMessages, messages) => {
 	};
 };
 const isAwaitingToolApproval = (message) => message.status?.type === "requires-action" && message.status.reason === "tool-calls";
+const isTerminalMessage = (message) => message.status === void 0 || message.status.type === "complete" || message.status.type === "incomplete";
 const encodeContent = (storageFormatAdapter, item) => JSON.stringify(storageFormatAdapter.encode(item));
 const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storageFormatAdapter, onSetMessages) => {
 	const loadedRef = useRef(false);
@@ -106,6 +107,8 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 	useEffect(() => {
 		if (!formatAdapter) return;
 		const adapter = formatAdapter;
+		const idleScheduledInnerIds = /* @__PURE__ */ new Set();
+		const unpersistedInnerIds = (message) => isTerminalMessage(message) ? getExternalStoreMessages(message).map((innerMessage) => storageFormatAdapter.getId(innerMessage)).filter((innerId) => !persistedInnerMessages.current.has(innerId)) : [];
 		const unsubscribe = runtimeRef.current.thread.subscribe(() => {
 			const threadState_0 = runtimeRef.current.thread.getState();
 			const { isRunning } = threadState_0;
@@ -134,8 +137,13 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 				}
 				return;
 			}
-			if (!wasRunning) return;
-			if (runStartRef.current != null) stepBoundariesRef.current.push(Date.now() - runStartRef.current);
+			if (wasRunning) {
+				if (runStartRef.current != null) stepBoundariesRef.current.push(Date.now() - runStartRef.current);
+			} else {
+				const pending = threadState_0.messages.flatMap(unpersistedInnerIds);
+				if (pending.every((innerId_0) => idleScheduledInnerIds.has(innerId_0))) return;
+				for (const innerId_1 of pending) idleScheduledInnerIds.add(innerId_1);
+			}
 			if (persistTimerRef.current) clearTimeout(persistTimerRef.current);
 			persistTimerRef.current = setTimeout(() => {
 				persistSettled(false);
@@ -187,25 +195,25 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 					parentId: idx === 0 ? lastInnerMessageId : storageFormatAdapter.getId(msgs_0[idx - 1]),
 					message: msg
 				}));
-				for (const message of messages_0) {
-					const innerMessages = getExternalStoreMessages(message);
-					const isTerminal = message.status === void 0 || message.status.type === "complete" || message.status.type === "incomplete";
-					const isAwaitingToolCalls = isAwaitingToolApproval(message);
+				for (const message_0 of messages_0) {
+					const innerMessages = getExternalStoreMessages(message_0);
+					const isTerminal = isTerminalMessage(message_0);
+					const isAwaitingToolCalls = isAwaitingToolApproval(message_0);
 					if (!(isTerminal || isAwaitingToolCalls && adapter.update !== void 0)) {
 						lastInnerMessageId = getLastInnerId(innerMessages) ?? lastInnerMessageId;
 						continue;
 					}
-					if (!historyIds.current.has(message.id)) {
-						historyIds.current.add(message.id);
-						deferredTelemetryIds.current.add(message.id);
+					if (!historyIds.current.has(message_0.id)) {
+						historyIds.current.add(message_0.id);
+						deferredTelemetryIds.current.add(message_0.id);
 					}
 					const batchItems = toBatchItems(innerMessages);
 					for (const item of batchItems) {
-						const innerId = storageFormatAdapter.getId(item.message);
-						const persisted = persistedInnerMessages.current.get(innerId);
+						const innerId_2 = storageFormatAdapter.getId(item.message);
+						const persisted = persistedInnerMessages.current.get(innerId_2);
 						if (!persisted) {
 							await adapter.append(item);
-							persistedInnerMessages.current.set(innerId, {
+							persistedInnerMessages.current.set(innerId_2, {
 								source: item.message,
 								content: encodeContent(storageFormatAdapter, item)
 							});
@@ -213,8 +221,8 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 							const content = encodeContent(storageFormatAdapter, item);
 							if (content === persisted.content) persisted.source = item.message;
 							else try {
-								await adapter.update(item, innerId);
-								persistedInnerMessages.current.set(innerId, {
+								await adapter.update(item, innerId_2);
+								persistedInnerMessages.current.set(innerId_2, {
 									source: item.message,
 									content
 								});
@@ -222,9 +230,12 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 						}
 					}
 					lastInnerMessageId = getLastInnerId(innerMessages) ?? lastInnerMessageId;
-					if (deferredTelemetryIds.current.has(message.id) && isTerminal) {
-						deferredTelemetryIds.current.delete(message.id);
-						adapter.reportTelemetry?.(batchItems, telemetryOptions);
+					if (deferredTelemetryIds.current.has(message_0.id) && isTerminal) {
+						deferredTelemetryIds.current.delete(message_0.id);
+						adapter.reportTelemetry?.(batchItems, {
+							...telemetryOptions,
+							message: message_0
+						});
 					}
 				}
 			}).catch((error_0) => {
@@ -254,12 +265,12 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 			if (messageIndex === -1) return;
 			const previousInnerMessages = messages_1.slice(0, messageIndex).flatMap(getExternalStoreMessages);
 			let parentId = previousInnerMessages.at(-1) ? storageFormatAdapter.getId(previousInnerMessages.at(-1)) : null;
-			const itemsToDelete = getExternalStoreMessages(messages_1[messageIndex]).map((message_0) => {
+			const itemsToDelete = getExternalStoreMessages(messages_1[messageIndex]).map((message_1) => {
 				const item_0 = {
 					parentId,
-					message: message_0
+					message: message_1
 				};
-				parentId = storageFormatAdapter.getId(message_0);
+				parentId = storageFormatAdapter.getId(message_1);
 				return item_0;
 			});
 			const deletion = persistInFlightRef.current.then(async () => {

package/dist/runtime/useExternalHistory.js.map

@@ -1 +1 @@
-{"version":3,"file":"useExternalHistory.js","names":["AssistantRuntime","ThreadHistoryAdapter","ThreadMessage","MessageFormatAdapter","MessageFormatItem","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","encodeContent","storageFormatAdapter","item","JSON","stringify","encode","useExternalHistory","runtimeRef","historyAdapter","onSetMessages","loadedRef","itemEpoch","setItemEpoch","aui","optionalThreadListItem","threadListItem","source","hasLoaded","setHasLoaded","historyIds","deferredTelemetryIds","persistedInnerMessages","Map","content","onSetMessagesRef","current","formatAdapter","undefined","withFormat","Error","isLoading","loadHistory","repo","load","length","set","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","filter","p","push","Date","now","clearTimeout","persistSettled","initialThreadState","ignoreRunning","latest","boundaries","durationMs","lastAssistant","findLast","tcCount","totalSteps","stepDur","i","Math","round","stepTimestamps","map","endMs","start_ms","end_ms","telemetryOptions","then","lastInnerMessageId","getLastInnerId","msgs","toBatchItems","msg","idx","innerMessages","isTerminal","isAwaitingToolCalls","isReady","update","batchItems","innerId","persisted","get","append","delete","reportTelemetry","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  MessageFormatItem,\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 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) => {\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        if (repo && repo.messages.length > 0) {\n          for (const m of repo.messages) {\n            persistedInnerMessages.current.set(\n              storageFormatAdapter.getId(m.message),\n              {\n                source: m.message,\n                content: encodeContent(storageFormatAdapter, m),\n              },\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        }\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  ]);\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 { 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 = (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            if (!historyIds.current.has(message.id)) {\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              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?.(batchItems, telemetryOptions);\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  }, [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        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 { isLoading, deleteMessage };\n};\n"],"mappings":";;;;;;AAuBA,MAAagB,+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,iBACJC,sBACAC,SACGC,KAAKC,UAAUH,qBAAqBI,OAAOH,IAAI,CAAC;AAErD,MAAaI,sBACXC,YACAC,gBACA5B,kBACAqB,sBACAQ,kBACG;CACH,MAAMC,YAAYrC,OAAO,KAAK;CAC9B,MAAM,CAACsC,WAAWC,gBAAgBrC,SAAS,CAAC;CAE5C,MAAMsC,MAAMzC,OAAO;CACnB,MAAM0C,yBAAyBrC,kBACtBoC,IAAIE,eAAeC,SAASH,IAAIE,iBAAiB,MACxD,CAACF,GAAG,CACN;CAEA,MAAM,CAACI,WAAWC,gBAAgB3C,SAAS,KAAK;CAEhD,MAAM4C,aAAa9C,uBAAO,IAAIW,IAAY,CAAC;CAC3C,MAAMoC,uBAAuB/C,uBAAO,IAAIW,IAAY,CAAC;CAErD,MAAMqC,yBAAyBhD,uBAC7B,IAAIiD,IAAmD,CACzD;CAEA,MAAME,mBAAmBnD,OAAOoC,aAAa;CAC7CnC,gBAAgB;EACdkD,iBAAiBC,UAAUhB;CAC7B,CAAC;CAED,MAAMiB,gBAAgBhD,cAAc;EAClC,IAAI,CAAC8B,gBAAgB,OAAOmB,KAAAA;EAC5B,IAAI,CAACnB,eAAeoB,YAClB,MAAM,IAAIC,MACR,oFACF;EAEF,OAAOrB,eAAeoB,WAA0B3B,oBAAoB;CACtE,GAAG,CAACO,gBAAgBP,oBAAoB,CAAC;CAEzC,MAAM6B,YAAYJ,iBAAiB,QAAQ,CAACT;CAE5C3C,gBAAgB;EACd,IAAI,CAACoD,iBAAiBhB,UAAUe,SAAS,OAAOE,KAAAA;EAEhD,MAAMI,cAAc,YAAY;GAC9B,IAAI;IACF,MAAMC,OAAO,MAAMN,cAAcO,KAAK;IACtC,IAAID,QAAQA,KAAKnD,SAASqD,SAAS,GAAG;KACpC,KAAK,MAAM/C,KAAK6C,KAAKnD,UACnBwC,uBAAuBI,QAAQU,IAC7BlC,qBAAqBmC,MAAMjD,EAAEC,OAAO,GACpC;MACE4B,QAAQ7B,EAAEC;MACVmC,SAASvB,cAAcC,sBAAsBd,CAAC;KAChD,CACF;KAEF,MAAMkD,YAAY1D,4BAA4BC,kBAAkBoD,IAAI;KACpEzB,WAAWkB,QAAQa,OAAOC,OAAOF,SAAS;KAE1C,MAAMG,WAAW,IAAIrE,kBAAkB;KACvCqE,SAASD,OAAOF,SAAS;KACzB,MAAMxD,WAAW2D,SAASC,YAAY;KAEtCjB,iBAAiBC,QACf5C,SAASK,QAAQhB,wBAAkC,CACrD;KAEAiD,WAAWM,0BAAU,IAAIzC,IAAI;KAC7B,KAAK,MAAMG,OAAKkD,UAAUxD,UAAU;MAClCsC,WAAWM,QAAQhC,IAAIN,IAAEC,QAAQM,EAAE;MACnC,IAAIE,uBAAuBT,IAAEC,OAAO,GAClCgC,qBAAqBK,QAAQhC,IAAIN,IAAEC,QAAQM,EAAE;KAEjD;IACF;GACF,SAASgD,OAAO;IACdrD,QAAQqD,MAAM,mCAAmCA,KAAK;GACxD,UAAU;IACRxB,aAAa,IAAI;GACnB;EACF;EAEAQ,cAAciB,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,cAAczC,WAAWkB,QAAQa,OAAOO,SAAS;EACvD,IAAIG,YAAYC,aAAaD,YAAYnE,SAASqD,SAAS,GAAG;GAC5DxB,UAAUe,UAAU;GACpBP,aAAa,IAAI;GACjB;EACF;EAEAR,UAAUe,UAAU;EACpB,YAAiB;CAEnB,GAAG;EACDC;EACA9C;EACA2B;EACAO;EACAD;EACAF;EACAV;CAAoB,CACrB;CAED,MAAMiD,cAAc7E,OAAsB,IAAI;CAC9C,MAAM8E,kBAAkB9E,OAA6C,IAAI;CACzE,MAAMiF,qBAAqBjF,OAAsBkF,QAAQC,QAAQ,CAAC;CAClE,MAAMC,oBAAoBpF,OAAiB,CAAA,CAAE;CAC7C,MAAMqF,gBAAgBrF,OAAO,KAAK;CAClC,MAAMsF,mBAAmBtF,OAAO,CAAC;CAEjCC,gBAAgB;EACd,IAAI,CAACoD,eAAe;EACpB,MAAMkC,UAAUlC;EAEhB,MAAMmC,cAActD,WAAWkB,QAAQa,OAAOQ,gBAAgB;GAC5D,MAAME,gBAAczC,WAAWkB,QAAQa,OAAOO,SAAS;GACvD,MAAM,EAAEI,cAAcD;GACtB,MAAMc,aAAaJ,cAAcjC;GACjCiC,cAAcjC,UAAUwB;GAGxB,IAAIC,YAAYzB,WAAW,MAAM;IAC/B,MAAMsC,UAAUf,cAAYnE,SAASmF,GAAG,EAAE;IAC1C,IAAID,SAASE,SAAS,aAAa;KACjC,MAAMC,uBAAuBH,QAAQxC,QAAQ4C,QAC1CC,MAAMA,EAAEtE,SAAS,WACpB,CAAC,CAACoC;KACF,OAAOyB,iBAAiBlC,UAAUyC,sBAAsB;MACtDT,kBAAkBhC,QAAQ4C,KAAKC,KAAKC,IAAI,IAAIrB,YAAYzB,OAAO;MAC/DkC,iBAAiBlC;KACnB;IACF;GACF;GAEA,IAAIwB,WAAW;IACb,IAAIC,YAAYzB,WAAW,MAAM;KAC/ByB,YAAYzB,UAAU6C,KAAKC,IAAI;KAC/Bd,kBAAkBhC,UAAU,CAAA;KAC5BkC,iBAAiBlC,UAAU;KAC3BmC,QAAQjB,MAAM;IAChB;IAEA,IAAIQ,gBAAgB1B,SAAS;KAC3B+C,aAAarB,gBAAgB1B,OAAO;KACpC0B,gBAAgB1B,UAAU;IAC5B;IACA;GACF;GAGA,IAAI,CAACqC,YAAY;GAGjB,IAAIZ,YAAYzB,WAAW,MACzBgC,kBAAkBhC,QAAQ4C,KAAKC,KAAKC,IAAI,IAAIrB,YAAYzB,OAAO;GAIjE,IAAI0B,gBAAgB1B,SAAS+C,aAAarB,gBAAgB1B,OAAO;GACjE0B,gBAAgB1B,UAAU4B,iBAAiB;IACzCoB,eAAe,KAAK;GACtB,GAAG,CAAC;EACN,CAAC;EAED,MAAMC,qBAAqBnE,WAAWkB,QAAQa,OAAOO,SAAS;EAC9Da,cAAcjC,UAAUiD,mBAAmBzB;EAC3C,IAAIyB,mBAAmBzB,WACjBC;OAAAA,YAAYzB,WAAW,MAAM;IAC/ByB,YAAYzB,UAAU6C,KAAKC,IAAI;IAC/Bd,kBAAkBhC,UAAU,CAAA;IAC5BkC,iBAAiBlC,UAAU;IAC3BmC,QAAQjB,MAAM;GAChB;SACK,IAAI+B,mBAAmB7F,SAASqD,SAAS,GAC9CuC,eAAe,IAAI;EAGrB,SAASA,eAAeE,eAAwB;GAC9CxB,gBAAgB1B,UAAU;GAC1B,MAAMmD,SAASrE,WAAWkB,QAAQa,OAAOO,SAAS;GAClD,IAAI,CAAC8B,iBAAiBC,OAAO3B,WAAW;GAExC,MAAM4B,aAAapB,kBAAkBhC;GACrC,MAAMqD,aAAaD,WAAW3C,SAAS,IAAI2C,WAAWb,GAAG,EAAE,IAAIrC,KAAAA;GAG/D,IAAIkD,WAAW3C,WAAW,KAAK4C,cAAc,MAAM;IACjD,MAAMC,gBAAgBH,OAAO/F,SAASmG,UACnC7F,QAAMA,IAAE8E,SAAS,WACpB;IACA,IAAIc,eAAe;KACjB,MAAME,UAAUF,cAAcxD,QAAQ4C,QACnCC,QAAMA,IAAEtE,SAAS,WACpB,CAAC,CAACoC;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,WAAWW,KAAKC,OAAOL,SAAO;IAC5BM,UAAUN,QAAM,IAAI,IAAIP,WAAWO,MAAI;IACvCO,QAAQF;GACV,EAAE,IACF9D,KAAAA;GAENuB,YAAYzB,UAAU;GACtBgC,kBAAkBhC,UAAU,CAAA;GAE5B,MAAMmE,mBAAmB;IACvB,GAAId,cAAc,OAAO,EAAEA,WAAW,IAAInD,KAAAA;IAC1C,GAAI4D,kBAAkB,OAAO,EAAEA,eAAe,IAAI5D,KAAAA;GACpD;GAEA2B,mBAAmB7B,UAAU6B,mBAAmB7B,QAC7CoE,KAAK,YAAY;IAChB,MAAM,EAAEhH,UAAAA,eAAa+F;IACrB,IAAIkB,qBAAoC;IAExC,MAAMC,kBAAkBC,SACtBA,KAAK9D,SAAS,IAAIjC,qBAAqBmC,MAAM4D,KAAKhC,GAAG,EAAE,CAAE,IAAI;IAE/D,MAAMiC,gBAAgBD,WACpBA,OAAKR,KAAKU,KAAKC,SAAS;KACtB5G,UACE4G,QAAQ,IACJL,qBACA7F,qBAAqBmC,MAAM4D,OAAKG,MAAM,EAAG;KAC/C/G,SAAS8G;IACX,EAAE;IAEJ,KAAK,MAAM9G,WAAWP,YAAU;KAC9B,MAAMuH,gBAAgBlI,yBAAmCkB,OAAO;KAEhE,MAAMiH,aACJjH,QAAQS,WAAW8B,KAAAA,KACnBvC,QAAQS,OAAOC,SAAS,cACxBV,QAAQS,OAAOC,SAAS;KAC1B,MAAMwG,sBAAsB1G,uBAAuBR,OAAO;KAM1D,IAAI,EAHFiH,cACCC,uBAAuB1C,QAAQ4C,WAAW7E,KAAAA,IAE/B;MACZmE,qBACEC,eAAeK,aAAa,KAAKN;MACnC;KACF;KAEA,IAAI,CAAC3E,WAAWM,QAAQjC,IAAIJ,QAAQM,EAAE,GAAG;MACvCyB,WAAWM,QAAQhC,IAAIL,QAAQM,EAAE;MACjC0B,qBAAqBK,QAAQhC,IAAIL,QAAQM,EAAE;KAC7C;KAEA,MAAM+G,aAAaR,aAAaG,aAAa;KAC7C,KAAK,MAAMlG,QAAQuG,YAAY;MAC7B,MAAMC,UAAUzG,qBAAqBmC,MAAMlC,KAAKd,OAAO;MACvD,MAAMuH,YAAYtF,uBAAuBI,QAAQmF,IAAIF,OAAO;MAC5D,IAAI,CAACC,WAAW;OACd,MAAM/C,QAAQiD,OAAO3G,IAAI;OACzBmB,uBAAuBI,QAAQU,IAAIuE,SAAS;QAC1C1F,QAAQd,KAAKd;QACbmC,SAASvB,cAAcC,sBAAsBC,IAAI;OACnD,CAAC;MACH,OAAO,IACLyG,UAAU3F,WAAWd,KAAKd,WAC1B0F,eAAenD,KAAAA,KACfiC,QAAQ4C,QACR;OACA,MAAMjF,UAAUvB,cAAcC,sBAAsBC,IAAI;OACxD,IAAIqB,YAAYoF,UAAUpF,SACxBoF,UAAU3F,SAASd,KAAKd;YAExB,IAAI;QACF,MAAMwE,QAAQ4C,OAAOtG,MAAMwG,OAAO;QAClCrF,uBAAuBI,QAAQU,IAAIuE,SAAS;SAC1C1F,QAAQd,KAAKd;SACbmC;QACF,CAAC;OACH,QAAQ,CACN;MAGN;KACF;KAEAuE,qBACEC,eAAeK,aAAa,KAAKN;KAEnC,IAAI1E,qBAAqBK,QAAQjC,IAAIJ,QAAQM,EAAE,KAAK2G,YAAY;MAC9DjF,qBAAqBK,QAAQqF,OAAO1H,QAAQM,EAAE;MAC9CkE,QAAQmD,kBAAkBN,YAAYb,gBAAgB;KACxD;IACF;GACF,CAAC,CAAC,CACDoB,OAAOtE,YAAU;IAChBrD,QAAQqD,MAAM,sCAAsCA,OAAK;GAC3D,CAAC;EACL;EAEA,aAAa;GACXmB,YAAY;GACZ,IAAIV,gBAAgB1B,SAAS;IAC3B+C,aAAarB,gBAAgB1B,OAAO;IACpC0B,gBAAgB1B,UAAU;IAC1BgD,eAAe,KAAK;GACtB;EACF;CACF,GAAG;EAAC/C;EAAezB;EAAsBM;CAAU,CAAC;CA2CpD,OAAO;EAAEuB;EAAWmF,eAzCExI,YACpB,OAAOyI,cAAsB;GAC3B,MAAMC,iBAAiBzF,eAAeoF,QAAQM,KAAK1F,aAAa;GAChE,IAAI,CAACyF,gBAAgB;GAErB,MAAMtI,aAAW0B,WAAWkB,QAAQa,OAAOO,SAAS,CAAC,CAAChE;GACtD,MAAMwI,eAAexI,WAASyI,WAAWnI,QAAMA,IAAEO,OAAOwH,SAAS;GACjE,IAAIG,iBAAiB,IAAI;GAEzB,MAAME,wBAAwB1I,WAC3B2I,MAAM,GAAGH,YAAY,CAAC,CACtBnI,QAAQhB,wBAAkC;GAC7C,IAAIqB,WAAWgI,sBAAsBvD,GAAG,EAAE,IACtC/D,qBAAqBmC,MAAMmF,sBAAsBvD,GAAG,EAAE,CAAE,IACxD;GACJ,MAAMyD,gBAAgBvJ,yBACpBW,WAASwI,aACX,CAAC,CAAC7B,KAAKpG,cAAY;IACjB,MAAMc,SAAO;KAAEX;KAAUH,SAAAA;IAAQ;IACjCG,WAAWU,qBAAqBmC,MAAMhD,SAAO;IAC7C,OAAOc;GACT,CAAC;GAED,MAAMwH,WAAWpE,mBAAmB7B,QAAQoE,KAAK,YAAY;IAC3D,MAAMsB,eAAeM,aAAa;IAElCtG,WAAWM,QAAQqF,OAAOI,SAAS;IACnC9F,qBAAqBK,QAAQqF,OAAOI,SAAS;IAC7C,KAAK,MAAMhH,UAAQuH,eACjBpG,uBAAuBI,QAAQqF,OAC7B7G,qBAAqBmC,MAAMlC,OAAKd,OAAO,CACzC;GAEJ,CAAC;GAEDkE,mBAAmB7B,UAAUiG,SAASV,YAAY,CAAC,CAAC;GACpD,MAAMU;EACR,GACA;GAAChG;GAAenB;GAAYN;EAAoB,CAG9BgH;CAAc;AACpC"}
+{"version":3,"file":"useExternalHistory.js","names":["AssistantRuntime","ThreadHistoryAdapter","ThreadMessage","MessageFormatAdapter","MessageFormatItem","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","isTerminalMessage","undefined","encodeContent","storageFormatAdapter","item","JSON","stringify","encode","useExternalHistory","runtimeRef","historyAdapter","onSetMessages","loadedRef","itemEpoch","setItemEpoch","aui","optionalThreadListItem","threadListItem","source","hasLoaded","setHasLoaded","historyIds","deferredTelemetryIds","persistedInnerMessages","Map","content","onSetMessagesRef","current","formatAdapter","withFormat","Error","isLoading","loadHistory","repo","load","length","set","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","idleScheduledInnerIds","unpersistedInnerIds","map","innerMessage","filter","innerId","unsubscribe","wasRunning","lastMsg","at","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","then","lastInnerMessageId","getLastInnerId","msgs","toBatchItems","msg","idx","innerMessages","isTerminal","isAwaitingToolCalls","isReady","update","batchItems","persisted","get","append","delete","reportTelemetry","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  MessageFormatItem,\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 isTerminalMessage = (message: ThreadMessage) =>\n  message.status === undefined ||\n  message.status.type === \"complete\" ||\n  message.status.type === \"incomplete\";\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) => {\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        if (repo && repo.messages.length > 0) {\n          for (const m of repo.messages) {\n            persistedInnerMessages.current.set(\n              storageFormatAdapter.getId(m.message),\n              {\n                source: m.message,\n                content: encodeContent(storageFormatAdapter, m),\n              },\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        }\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  ]);\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 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 = (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 = 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(innerMessages);\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?.(batchItems, {\n                ...telemetryOptions,\n                message,\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  }, [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        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 { isLoading, deleteMessage };\n};\n"],"mappings":";;;;;;AAuBA,MAAagB,+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,qBAAqBZ,YACzBA,QAAQS,WAAWI,KAAAA,KACnBb,QAAQS,OAAOC,SAAS,cACxBV,QAAQS,OAAOC,SAAS;AAE1B,MAAMI,iBACJC,sBACAC,SACGC,KAAKC,UAAUH,qBAAqBI,OAAOH,IAAI,CAAC;AAErD,MAAaI,sBACXC,YACAC,gBACA9B,kBACAuB,sBACAQ,kBACG;CACH,MAAMC,YAAYvC,OAAO,KAAK;CAC9B,MAAM,CAACwC,WAAWC,gBAAgBvC,SAAS,CAAC;CAE5C,MAAMwC,MAAM3C,OAAO;CACnB,MAAM4C,yBAAyBvC,kBACtBsC,IAAIE,eAAeC,SAASH,IAAIE,iBAAiB,MACxD,CAACF,GAAG,CACN;CAEA,MAAM,CAACI,WAAWC,gBAAgB7C,SAAS,KAAK;CAEhD,MAAM8C,aAAahD,uBAAO,IAAIW,IAAY,CAAC;CAC3C,MAAMsC,uBAAuBjD,uBAAO,IAAIW,IAAY,CAAC;CAErD,MAAMuC,yBAAyBlD,uBAC7B,IAAImD,IAAmD,CACzD;CAEA,MAAME,mBAAmBrD,OAAOsC,aAAa;CAC7CrC,gBAAgB;EACdoD,iBAAiBC,UAAUhB;CAC7B,CAAC;CAED,MAAMiB,gBAAgBlD,cAAc;EAClC,IAAI,CAACgC,gBAAgB,OAAOT,KAAAA;EAC5B,IAAI,CAACS,eAAemB,YAClB,MAAM,IAAIC,MACR,oFACF;EAEF,OAAOpB,eAAemB,WAA0B1B,oBAAoB;CACtE,GAAG,CAACO,gBAAgBP,oBAAoB,CAAC;CAEzC,MAAM4B,YAAYH,iBAAiB,QAAQ,CAACT;CAE5C7C,gBAAgB;EACd,IAAI,CAACsD,iBAAiBhB,UAAUe,SAAS,OAAO1B,KAAAA;EAEhD,MAAM+B,cAAc,YAAY;GAC9B,IAAI;IACF,MAAMC,OAAO,MAAML,cAAcM,KAAK;IACtC,IAAID,QAAQA,KAAKpD,SAASsD,SAAS,GAAG;KACpC,KAAK,MAAMhD,KAAK8C,KAAKpD,UACnB0C,uBAAuBI,QAAQS,IAC7BjC,qBAAqBkC,MAAMlD,EAAEC,OAAO,GACpC;MACE8B,QAAQ/B,EAAEC;MACVqC,SAASvB,cAAcC,sBAAsBhB,CAAC;KAChD,CACF;KAEF,MAAMmD,YAAY3D,4BAA4BC,kBAAkBqD,IAAI;KACpExB,WAAWkB,QAAQY,OAAOC,OAAOF,SAAS;KAE1C,MAAMG,WAAW,IAAItE,kBAAkB;KACvCsE,SAASD,OAAOF,SAAS;KACzB,MAAMzD,WAAW4D,SAASC,YAAY;KAEtChB,iBAAiBC,QACf9C,SAASK,QAAQhB,wBAAkC,CACrD;KAEAmD,WAAWM,0BAAU,IAAI3C,IAAI;KAC7B,KAAK,MAAMG,OAAKmD,UAAUzD,UAAU;MAClCwC,WAAWM,QAAQlC,IAAIN,IAAEC,QAAQM,EAAE;MACnC,IAAIE,uBAAuBT,IAAEC,OAAO,GAClCkC,qBAAqBK,QAAQlC,IAAIN,IAAEC,QAAQM,EAAE;KAEjD;IACF;GACF,SAASiD,OAAO;IACdtD,QAAQsD,MAAM,mCAAmCA,KAAK;GACxD,UAAU;IACRvB,aAAa,IAAI;GACnB;EACF;EAEAQ,cAAcgB,MAAM;EAGpB,IAAI,CADa5B,uBAAuB,CAAC,EAAE8B,SAAS,CAAC,CAACD,UACvC;GAIbzB,aAAa,IAAI;GACjB,OAAOL,IAAIgC,gBAAgB;IACzB,IAAI/B,uBAAuB,CAAC,EAAE8B,SAAS,CAAC,CAACD,UACvC/B,cAAckC,MAAMA,IAAI,CAAC;GAE7B,CAAC;EACH;EAEA,MAAMC,cAAcxC,WAAWkB,QAAQY,OAAOO,SAAS;EACvD,IAAIG,YAAYC,aAAaD,YAAYpE,SAASsD,SAAS,GAAG;GAC5DvB,UAAUe,UAAU;GACpBP,aAAa,IAAI;GACjB;EACF;EAEAR,UAAUe,UAAU;EACpB,YAAiB;CAEnB,GAAG;EACDC;EACAhD;EACA6B;EACAO;EACAD;EACAF;EACAV;CAAoB,CACrB;CAED,MAAMgD,cAAc9E,OAAsB,IAAI;CAC9C,MAAM+E,kBAAkB/E,OAA6C,IAAI;CACzE,MAAMkF,qBAAqBlF,OAAsBmF,QAAQC,QAAQ,CAAC;CAClE,MAAMC,oBAAoBrF,OAAiB,CAAA,CAAE;CAC7C,MAAMsF,gBAAgBtF,OAAO,KAAK;CAClC,MAAMuF,mBAAmBvF,OAAO,CAAC;CAEjCC,gBAAgB;EACd,IAAI,CAACsD,eAAe;EACpB,MAAMiC,UAAUjC;EAEhB,MAAMkC,wCAAwB,IAAI9E,IAAY;EAC9C,MAAM+E,uBAAuB3E,YAC3BY,kBAAkBZ,OAAO,IACrBlB,yBAAmCkB,OAAO,CAAC,CACxC4E,KAAKC,iBAAiB9D,qBAAqBkC,MAAM4B,YAAY,CAAC,CAAC,CAC/DC,QAAQC,YAAY,CAAC5C,uBAAuBI,QAAQnC,IAAI2E,OAAO,CAAC,IACnE,CAAA;EAEN,MAAMC,cAAc3D,WAAWkB,QAAQY,OAAOQ,gBAAgB;GAC5D,MAAME,gBAAcxC,WAAWkB,QAAQY,OAAOO,SAAS;GACvD,MAAM,EAAEI,cAAcD;GACtB,MAAMoB,aAAaV,cAAchC;GACjCgC,cAAchC,UAAUuB;GAGxB,IAAIC,YAAYxB,WAAW,MAAM;IAC/B,MAAM2C,UAAUrB,cAAYpE,SAAS0F,GAAG,EAAE;IAC1C,IAAID,SAASE,SAAS,aAAa;KACjC,MAAMC,uBAAuBH,QAAQ7C,QAAQyC,QAC1CQ,MAAMA,EAAE5E,SAAS,WACpB,CAAC,CAACqC;KACF,OAAOyB,iBAAiBjC,UAAU8C,sBAAsB;MACtDf,kBAAkB/B,QAAQgD,KAAKC,KAAKC,IAAI,IAAI1B,YAAYxB,OAAO;MAC/DiC,iBAAiBjC;KACnB;IACF;GACF;GAEA,IAAIuB,WAAW;IACb,IAAIC,YAAYxB,WAAW,MAAM;KAC/BwB,YAAYxB,UAAUiD,KAAKC,IAAI;KAC/BnB,kBAAkB/B,UAAU,CAAA;KAC5BiC,iBAAiBjC,UAAU;KAC3BkC,QAAQjB,MAAM;IAChB;IAEA,IAAIQ,gBAAgBzB,SAAS;KAC3BmD,aAAa1B,gBAAgBzB,OAAO;KACpCyB,gBAAgBzB,UAAU;IAC5B;IACA;GACF;GAEA,IAAI0C,YAEElB;QAAAA,YAAYxB,WAAW,MACzB+B,kBAAkB/B,QAAQgD,KAAKC,KAAKC,IAAI,IAAI1B,YAAYxB,OAAO;GAAA,OAE5D;IACL,MAAMoD,UAAU9B,cAAYpE,SAASK,QAAQ6E,mBAAmB;IAChE,IAAIgB,QAAQC,OAAOb,cAAYL,sBAAsBtE,IAAI2E,SAAO,CAAC,GAC/D;IACF,KAAK,MAAMA,aAAWY,SAASjB,sBAAsBrE,IAAI0E,SAAO;GAClE;GAGA,IAAIf,gBAAgBzB,SAASmD,aAAa1B,gBAAgBzB,OAAO;GACjEyB,gBAAgBzB,UAAU2B,iBAAiB;IACzC2B,eAAe,KAAK;GACtB,GAAG,CAAC;EACN,CAAC;EAED,MAAMC,qBAAqBzE,WAAWkB,QAAQY,OAAOO,SAAS;EAC9Da,cAAchC,UAAUuD,mBAAmBhC;EAC3C,IAAIgC,mBAAmBhC,WACjBC;OAAAA,YAAYxB,WAAW,MAAM;IAC/BwB,YAAYxB,UAAUiD,KAAKC,IAAI;IAC/BnB,kBAAkB/B,UAAU,CAAA;IAC5BiC,iBAAiBjC,UAAU;IAC3BkC,QAAQjB,MAAM;GAChB;SACK,IAAIsC,mBAAmBrG,SAASsD,SAAS,GAC9C8C,eAAe,IAAI;EAGrB,SAASA,eAAeE,eAAwB;GAC9C/B,gBAAgBzB,UAAU;GAC1B,MAAMyD,SAAS3E,WAAWkB,QAAQY,OAAOO,SAAS;GAClD,IAAI,CAACqC,iBAAiBC,OAAOlC,WAAW;GAExC,MAAMmC,aAAa3B,kBAAkB/B;GACrC,MAAM2D,aAAaD,WAAWlD,SAAS,IAAIkD,WAAWd,GAAG,EAAE,IAAItE,KAAAA;GAG/D,IAAIoF,WAAWlD,WAAW,KAAKmD,cAAc,MAAM;IACjD,MAAMC,gBAAgBH,OAAOvG,SAAS2G,UACnCrG,QAAMA,IAAEqF,SAAS,WACpB;IACA,IAAIe,eAAe;KACjB,MAAME,UAAUF,cAAc9D,QAAQyC,QACnCQ,QAAMA,IAAE5E,SAAS,WACpB,CAAC,CAACqC;KACF,IAAIsD,UAAU,GAAG;MACf,MAAMC,aAAaD,UAAU;MAC7B,MAAME,UAAUL,aAAaI;MAC7BL,WAAWlD,SAAS;MACpB,KAAK,IAAIyD,IAAI,GAAGA,IAAIF,YAAYE,KAC9BP,WAAWV,KAAKkB,KAAKC,OAAOF,IAAI,KAAKD,OAAO,CAAC;KAEjD;IACF;GACF;GAGA,MAAMI,iBACJV,WAAWlD,SAAS,IAChBkD,WAAWrB,KAAKgC,OAAOJ,SAAO;IAC5BK,UAAUL,QAAM,IAAI,IAAIP,WAAWO,MAAI;IACvCM,QAAQF;GACV,EAAE,IACF/F,KAAAA;GAENkD,YAAYxB,UAAU;GACtB+B,kBAAkB/B,UAAU,CAAA;GAE5B,MAAMwE,mBAAmB;IACvB,GAAIb,cAAc,OAAO,EAAEA,WAAW,IAAIrF,KAAAA;IAC1C,GAAI8F,kBAAkB,OAAO,EAAEA,eAAe,IAAI9F,KAAAA;GACpD;GAEAsD,mBAAmB5B,UAAU4B,mBAAmB5B,QAC7CyE,KAAK,YAAY;IAChB,MAAM,EAAEvH,UAAAA,eAAauG;IACrB,IAAIiB,qBAAoC;IAExC,MAAMC,kBAAkBC,SACtBA,KAAKpE,SAAS,IAAIhC,qBAAqBkC,MAAMkE,KAAKhC,GAAG,EAAE,CAAE,IAAI;IAE/D,MAAMiC,gBAAgBD,WACpBA,OAAKvC,KAAKyC,KAAKC,SAAS;KACtBnH,UACEmH,QAAQ,IACJL,qBACAlG,qBAAqBkC,MAAMkE,OAAKG,MAAM,EAAG;KAC/CtH,SAASqH;IACX,EAAE;IAEJ,KAAK,MAAMrH,aAAWP,YAAU;KAC9B,MAAM8H,gBAAgBzI,yBAAmCkB,SAAO;KAEhE,MAAMwH,aAAa5G,kBAAkBZ,SAAO;KAC5C,MAAMyH,sBAAsBjH,uBAAuBR,SAAO;KAM1D,IAAI,EAHFwH,cACCC,uBAAuBhD,QAAQkD,WAAW9G,KAAAA,IAE/B;MACZoG,qBACEC,eAAeK,aAAa,KAAKN;MACnC;KACF;KAEA,IAAI,CAAChF,WAAWM,QAAQnC,IAAIJ,UAAQM,EAAE,GAAG;MACvC2B,WAAWM,QAAQlC,IAAIL,UAAQM,EAAE;MACjC4B,qBAAqBK,QAAQlC,IAAIL,UAAQM,EAAE;KAC7C;KAEA,MAAMsH,aAAaR,aAAaG,aAAa;KAC7C,KAAK,MAAMvG,QAAQ4G,YAAY;MAC7B,MAAM7C,YAAUhE,qBAAqBkC,MAAMjC,KAAKhB,OAAO;MACvD,MAAM6H,YAAY1F,uBAAuBI,QAAQuF,IAAI/C,SAAO;MAC5D,IAAI,CAAC8C,WAAW;OACd,MAAMpD,QAAQsD,OAAO/G,IAAI;OACzBmB,uBAAuBI,QAAQS,IAAI+B,WAAS;QAC1CjD,QAAQd,KAAKhB;QACbqC,SAASvB,cAAcC,sBAAsBC,IAAI;OACnD,CAAC;MACH,OAAO,IACL6G,UAAU/F,WAAWd,KAAKhB,WAC1BkG,eAAerF,KAAAA,KACf4D,QAAQkD,QACR;OACA,MAAMtF,UAAUvB,cAAcC,sBAAsBC,IAAI;OACxD,IAAIqB,YAAYwF,UAAUxF,SACxBwF,UAAU/F,SAASd,KAAKhB;YAExB,IAAI;QACF,MAAMyE,QAAQkD,OAAO3G,MAAM+D,SAAO;QAClC5C,uBAAuBI,QAAQS,IAAI+B,WAAS;SAC1CjD,QAAQd,KAAKhB;SACbqC;QACF,CAAC;OACH,QAAQ,CACN;MAGN;KACF;KAEA4E,qBACEC,eAAeK,aAAa,KAAKN;KAEnC,IAAI/E,qBAAqBK,QAAQnC,IAAIJ,UAAQM,EAAE,KAAKkH,YAAY;MAC9DtF,qBAAqBK,QAAQyF,OAAOhI,UAAQM,EAAE;MAC9CmE,QAAQwD,kBAAkBL,YAAY;OACpC,GAAGb;OACH/G,SAAAA;MACF,CAAC;KACH;IACF;GACF,CAAC,CAAC,CACDkI,OAAO3E,YAAU;IAChBtD,QAAQsD,MAAM,sCAAsCA,OAAK;GAC3D,CAAC;EACL;EAEA,aAAa;GACXyB,YAAY;GACZ,IAAIhB,gBAAgBzB,SAAS;IAC3BmD,aAAa1B,gBAAgBzB,OAAO;IACpCyB,gBAAgBzB,UAAU;IAC1BsD,eAAe,KAAK;GACtB;EACF;CACF,GAAG;EAACrD;EAAezB;EAAsBM;CAAU,CAAC;CA2CpD,OAAO;EAAEsB;EAAWwF,eAzCE9I,YACpB,OAAO+I,cAAsB;GAC3B,MAAMC,iBAAiB7F,eAAewF,QAAQM,KAAK9F,aAAa;GAChE,IAAI,CAAC6F,gBAAgB;GAErB,MAAM5I,aAAW4B,WAAWkB,QAAQY,OAAOO,SAAS,CAAC,CAACjE;GACtD,MAAM8I,eAAe9I,WAAS+I,WAAWzI,QAAMA,IAAEO,OAAO8H,SAAS;GACjE,IAAIG,iBAAiB,IAAI;GAEzB,MAAME,wBAAwBhJ,WAC3BiJ,MAAM,GAAGH,YAAY,CAAC,CACtBzI,QAAQhB,wBAAkC;GAC7C,IAAIqB,WAAWsI,sBAAsBtD,GAAG,EAAE,IACtCpE,qBAAqBkC,MAAMwF,sBAAsBtD,GAAG,EAAE,CAAE,IACxD;GACJ,MAAMwD,gBAAgB7J,yBACpBW,WAAS8I,aACX,CAAC,CAAC3D,KAAK5E,cAAY;IACjB,MAAMgB,SAAO;KAAEb;KAAUH,SAAAA;IAAQ;IACjCG,WAAWY,qBAAqBkC,MAAMjD,SAAO;IAC7C,OAAOgB;GACT,CAAC;GAED,MAAM4H,WAAWzE,mBAAmB5B,QAAQyE,KAAK,YAAY;IAC3D,MAAMqB,eAAeM,aAAa;IAElC1G,WAAWM,QAAQyF,OAAOI,SAAS;IACnClG,qBAAqBK,QAAQyF,OAAOI,SAAS;IAC7C,KAAK,MAAMpH,UAAQ2H,eACjBxG,uBAAuBI,QAAQyF,OAC7BjH,qBAAqBkC,MAAMjC,OAAKhB,OAAO,CACzC;GAEJ,CAAC;GAEDmE,mBAAmB5B,UAAUqG,SAASV,YAAY,CAAC,CAAC;GACpD,MAAMU;EACR,GACA;GAACpG;GAAenB;GAAYN;EAAoB,CAG9BoH;CAAc;AACpC"}

package/dist/runtime/useResourceCleanup.d.ts

@@ -1,5 +1,4 @@
 //#region src/runtime/useResourceCleanup.d.ts
-declare const useResourceCleanup: (enabled: boolean, cleanup: () => void) => void;
+export declare const useResourceCleanup: (enabled: boolean, cleanup: () => void) => void;
 //#endregion
-export { useResourceCleanup };
 //# sourceMappingURL=useResourceCleanup.d.ts.map

package/dist/runtime/useResourceCleanup.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useResourceCleanup.d.ts","names":[],"sources":["../../src/runtime/useResourceCleanup.ts"],"mappings":";cAGa,qBAAsB,kBAAkB"}
+{"version":3,"file":"useResourceCleanup.d.ts","names":[],"sources":["../../src/runtime/useResourceCleanup.ts"],"mappings":";qBAGa,qBAAsB,kBAAkB"}

package/dist/runtime/useStreamingTiming.d.ts

@@ -2,14 +2,13 @@ import { StreamingTimingAccessors } from "@assistant-ui/core/react";
 import { MessageTiming } from "@assistant-ui/core";
 import { UIMessage } from "@ai-sdk/react";
 //#region src/runtime/useStreamingTiming.d.ts
-declare const aiSdkStreamingTimingAccessors: StreamingTimingAccessors<UIMessage>;
+export declare const aiSdkStreamingTimingAccessors: StreamingTimingAccessors<UIMessage>;
 /**
  * Tracks streaming timing for AI SDK messages client-side. Delegates to the
  * shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,
  * adapted to the `UIMessage` shape. Timing is finalized when streaming ends
  * and stored per message id.
  */
-declare const useStreamingTiming: (messages: UIMessage[], isRunning: boolean) => Record<string, MessageTiming>;
+export declare const useStreamingTiming: (messages: UIMessage[], isRunning: boolean) => Record<string, MessageTiming>;
 //#endregion
-export { aiSdkStreamingTimingAccessors, useStreamingTiming };
 //# sourceMappingURL=useStreamingTiming.d.ts.map

package/dist/runtime/useStreamingTiming.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamingTiming.d.ts","names":[],"sources":["../../src/runtime/useStreamingTiming.ts"],"mappings":";;;;cA8Ca,+BAA+B,yBAAyB;;;;;;;cAaxD,qBACX,UAAU,aACV,uBACC,eAAe"}
+{"version":3,"file":"useStreamingTiming.d.ts","names":[],"sources":["../../src/runtime/useStreamingTiming.ts"],"mappings":";;;;qBA8Ca,+BAA+B,yBAAyB;;;;;;;qBAaxD,qBACX,UAAU,aACV,uBACC,eAAe"}

package/dist/tools/frontendTools.d.ts

@@ -2,8 +2,8 @@ import { ToolSet } from "ai";
 import { ToolJSONSchema } from "assistant-stream";
 //#region src/tools/frontendTools.d.ts
 /** Frontend tool definitions uploaded by AssistantChatTransport. */
-type FrontendTools = Record<string, ToolJSONSchema>;
-declare const defaultToModelOutput: ({ output }: {
+export type FrontendTools = Record<string, ToolJSONSchema>;
+export declare const defaultToModelOutput: ({ output }: {
   output: unknown;
 }) => {
   type: "content";
@@ -27,7 +27,6 @@ declare const defaultToModelOutput: ({ output }: {
   type: "json";
   value: import("ai").JSONValue;
 };
-declare const frontendTools: (tools: FrontendTools) => ToolSet;
+export declare const frontendTools: (tools: FrontendTools) => ToolSet;
 //#endregion
-export { FrontendTools, defaultToModelOutput, frontendTools };
 //# sourceMappingURL=frontendTools.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"frontendTools.d.ts","names":[],"sources":["../../src/tools/frontendTools.ts"],"mappings":";;;;KASY,gBAAgB,eAAe;cAE9B,yBAAwB;EAAc;;;;;;;;;;;;;;;;;;;;;;;cAuDtC,gBAAiB,OAAO,kBAAgB"}
+{"version":3,"file":"frontendTools.d.ts","names":[],"sources":["../../src/tools/frontendTools.ts"],"mappings":";;;;YASY,gBAAgB,eAAe;qBAE9B,yBAAwB;EAAc;;;;;;;;;;;;;;;;;;;;;;;qBAuDtC,gBAAiB,OAAO,kBAAgB"}

package/dist/tools/generativeTools.d.ts

@@ -7,7 +7,7 @@ import { ToolSet } from "ai";
  * {@link AISDKToolkit} with {@link AISDKToolkitOptions} /
  * {@link AISDKToolkitToolsOptions} instead.
  */
-interface GenerativeToolsOptions {
+export interface GenerativeToolsOptions {
   /**
    * The server build of a generative toolkit (schema + server `execute`). Typed
    * as the canonical {@link Toolkit} so callers don't need to cast; the server
@@ -21,10 +21,10 @@ interface GenerativeToolsOptions {
    */
   frontendTools?: FrontendTools;
 }
-type AISDKToolkitOptions = {
+export type AISDKToolkitOptions = {
   toolkit: Toolkit;
 };
-type AISDKToolkitToolsOptions = {
+export type AISDKToolkitToolsOptions = {
   /**
    * Tools uploaded by the frontend request body.
    */
@@ -61,13 +61,12 @@ type AISDKToolkitToolsOptions = {
  * });
  * ```
  */
-declare const generativeTools: (options: GenerativeToolsOptions) => ToolSet;
-declare class AISDKToolkit {
+export declare const generativeTools: (options: GenerativeToolsOptions) => ToolSet;
+export declare class AISDKToolkit {
   #private;
   constructor(options: AISDKToolkitOptions);
   tools(options?: AISDKToolkitToolsOptions): Promise<ToolSet>;
   close(): Promise<void>;
 }
 //#endregion
-export { AISDKToolkit, AISDKToolkitOptions, AISDKToolkitToolsOptions, GenerativeToolsOptions, generativeTools };
 //# sourceMappingURL=generativeTools.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"generativeTools.d.ts","names":[],"sources":["../../src/tools/generativeTools.ts"],"mappings":";;;;;;;;;UAsFiB;;;;;;EAMf,SAAS;;;;;;EAMT,gBAAgB;;KAGN;EACV,SAAS;;KAGC;;;;EAIV,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAkCA,kBAAmB,SAAS,2BAAyB;cAYrD;;EAIC,YAAA,SAAS;EAIf,MAAM,UAAS,2BAAgC,QAAQ;EAsBvD,SAAS"}
+{"version":3,"file":"generativeTools.d.ts","names":[],"sources":["../../src/tools/generativeTools.ts"],"mappings":";;;;;;;;;iBAsFiB;;;;;;EAMf,SAAS;;;;;;EAMT,gBAAgB;;YAGN;EACV,SAAS;;YAGC;;;;EAIV,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qBAkCA,kBAAmB,SAAS,2BAAyB;qBAYrD;;EAIC,YAAA,SAAS;EAIf,MAAM,UAAS,2BAAgC,QAAQ;EAsBvD,SAAS"}

package/dist/tools/mcp-stdio.unsupported.d.ts

@@ -1,6 +1,5 @@
 import { Experimental_StdioMCPTransport as Experimental_StdioMCPTransport$1 } from "@ai-sdk/mcp/mcp-stdio";
 //#region src/tools/mcp-stdio.unsupported.d.ts
-declare const Experimental_StdioMCPTransport: typeof Experimental_StdioMCPTransport$1;
+export declare const Experimental_StdioMCPTransport: typeof Experimental_StdioMCPTransport$1;
 //#endregion
-export { Experimental_StdioMCPTransport };
 //# sourceMappingURL=mcp-stdio.unsupported.d.ts.map

package/dist/tools/mcp-stdio.unsupported.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mcp-stdio.unsupported.d.ts","names":[],"sources":["../../src/tools/mcp-stdio.unsupported.ts"],"mappings":";;cAUa,uCACuC"}
+{"version":3,"file":"mcp-stdio.unsupported.d.ts","names":[],"sources":["../../src/tools/mcp-stdio.unsupported.ts"],"mappings":";;qBAUa,uCACuC"}

package/dist/transport/AssistantChatTransport.d.ts

@@ -2,11 +2,11 @@ import { AssistantChatResumableOptions } from "./resumable.js";
 import { DefaultChatTransport, HttpChatTransportInitOptions, UIMessage } from "ai";
 import { AssistantRuntime, ThreadListItemRuntime } from "@assistant-ui/core";
 //#region src/transport/AssistantChatTransport.d.ts
-type InitializableThreadListItem = Pick<ThreadListItemRuntime, "initialize">;
-type AssistantChatTransportInitOptions<UI_MESSAGE extends UIMessage> = HttpChatTransportInitOptions<UI_MESSAGE> & {
+export type InitializableThreadListItem = Pick<ThreadListItemRuntime, "initialize">;
+export type AssistantChatTransportInitOptions<UI_MESSAGE extends UIMessage> = HttpChatTransportInitOptions<UI_MESSAGE> & {
   resumable?: AssistantChatResumableOptions;
 };
-declare class AssistantChatTransport<UI_MESSAGE extends UIMessage> extends DefaultChatTransport<UI_MESSAGE> {
+export declare class AssistantChatTransport<UI_MESSAGE extends UIMessage> extends DefaultChatTransport<UI_MESSAGE> {
   private runtime;
   private getThreadListItem;
   private readonly resumable;
@@ -19,5 +19,4 @@ declare class AssistantChatTransport<UI_MESSAGE extends UIMessage> extends Defau
   __internal_setGetThreadListItem(getter: () => InitializableThreadListItem | undefined): void;
 }
 //#endregion
-export { AssistantChatTransport, AssistantChatTransportInitOptions, InitializableThreadListItem };
 //# sourceMappingURL=AssistantChatTransport.d.ts.map

package/dist/transport/AssistantChatTransport.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantChatTransport.d.ts","names":[],"sources":["../../src/transport/AssistantChatTransport.ts"],"mappings":";;;;KAeY,8BAA8B,KACxC;KAaU,kCAAkC,mBAAmB,aAC/D,6BAA6B;EAC3B,YAAY;;cAGH,uBACX,mBAAmB,mBACX,qBAAqB;UACrB;UACA;mBAGS;mBAEA;;EAKV,oBAAoB,uBAAuB;EAOtC,YAAA,cAAc,kCAAkC;EA0D5D,WAAW,SAAS;EAIpB,uBAAuB;EAIvB,gCACE,cAAc"}
+{"version":3,"file":"AssistantChatTransport.d.ts","names":[],"sources":["../../src/transport/AssistantChatTransport.ts"],"mappings":";;;;YAeY,8BAA8B,KACxC;YAaU,kCAAkC,mBAAmB,aAC/D,6BAA6B;EAC3B,YAAY;;qBAGH,uBACX,mBAAmB,mBACX,qBAAqB;UACrB;UACA;mBAGS;mBAEA;;EAKV,oBAAoB,uBAAuB;EAOtC,YAAA,cAAc,kCAAkC;EA0D5D,WAAW,SAAS;EAIpB,uBAAuB;EAIvB,gCACE,cAAc"}

package/dist/transport/resumable.d.ts

@@ -1,7 +1,7 @@
 //#region src/transport/resumable.d.ts
 /** Response header used by the [Resumable Streams](/docs/guides/resumable-streams) server and client wiring. */
-declare const RESUMABLE_STREAM_ID_HEADER = "x-resumable-stream-id";
-type ResumableClientStorage = {
+export declare const RESUMABLE_STREAM_ID_HEADER = "x-resumable-stream-id";
+export type ResumableClientStorage = {
   getStreamId(threadId?: string): string | null;
   setStreamId(id: string, threadId?: string): void;
   clear(threadId?: string): void;
@@ -9,7 +9,7 @@ type ResumableClientStorage = {
   subscribe?(listener: () => void, threadId?: string): () => void;
 };
 /** `sessionStorage`-backed storage for the pending resumable stream id. See the [Resumable Streams](/docs/guides/resumable-streams) guide for end-to-end wiring. */
-declare function createResumableSessionStorage(options?: {
+export declare function createResumableSessionStorage(options?: {
   /**
    * Storage key for the pending stream id. A static string namespaces per route
    * or chat surface. A getter is read lazily on every access, so the key can be
@@ -24,7 +24,7 @@ declare function createResumableSessionStorage(options?: {
    */
   key?: string | (() => string | undefined);
 }): ResumableClientStorage;
-type AssistantChatResumableOptions = {
+export type AssistantChatResumableOptions = {
   storage: ResumableClientStorage;
   resumeApi: string | ((streamId: string) => string);
   /**
@@ -34,5 +34,4 @@ type AssistantChatResumableOptions = {
   isFinishEvent?: (chunk: Uint8Array, accumulator: string) => boolean;
 };
 //#endregion
-export { AssistantChatResumableOptions, RESUMABLE_STREAM_ID_HEADER, ResumableClientStorage, createResumableSessionStorage };
 //# sourceMappingURL=resumable.d.ts.map

package/dist/transport/resumable.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"resumable.d.ts","names":[],"sources":["../../src/transport/resumable.ts"],"mappings":";;cAKa;KAID;EACV,YAAY;EACZ,YAAY,YAAY;EACxB,MAAM;;EAEN,WAAW,sBAAsB;;;iBAkBnB,8BAA8B;;;;;;;;;;;;;EAa5C;IACE;KA0GQ;EACV,SAAS;EACT,sBAAsB;;;;;EAKtB,iBAAiB,OAAO,YAAY"}
+{"version":3,"file":"resumable.d.ts","names":[],"sources":["../../src/transport/resumable.ts"],"mappings":";;qBAKa;YAID;EACV,YAAY;EACZ,YAAY,YAAY;EACxB,MAAM;;EAEN,WAAW,sBAAsB;;;wBAkBnB,8BAA8B;;;;;;;;;;;;;EAa5C;IACE;YA0GQ;EACV,SAAS;EACT,sBAAsB;;;;;EAKtB,iBAAiB,OAAO,YAAY"}

package/dist/usage.d.ts

@@ -1,17 +1,17 @@
 /// <reference types="@assistant-ui/core/react" />
 //#region src/usage.d.ts
-type ThreadTokenUsage = {
+export type ThreadTokenUsage = {
   totalTokens?: number;
   inputTokens?: number;
   outputTokens?: number;
   reasoningTokens?: number;
   cachedInputTokens?: number;
 };
-interface TokenUsageExtractableMessage {
+export interface TokenUsageExtractableMessage {
   role?: string;
   metadata?: unknown;
 }
-declare function getThreadMessageTokenUsage(message: TokenUsageExtractableMessage | undefined): ThreadTokenUsage | undefined;
+export declare function getThreadMessageTokenUsage(message: TokenUsageExtractableMessage | undefined): ThreadTokenUsage | undefined;
 /**
  * Reads token usage from the newest assistant message that reports any.
  *
@@ -20,7 +20,6 @@ declare function getThreadMessageTokenUsage(message: TokenUsageExtractableMessag
  * other key the route returns into `metadata.custom`, which is where this hook
  * looks.
  */
-declare function useThreadTokenUsage(): ThreadTokenUsage | undefined;
+export declare function useThreadTokenUsage(): ThreadTokenUsage | undefined;
 //#endregion
-export { ThreadTokenUsage, TokenUsageExtractableMessage, getThreadMessageTokenUsage, useThreadTokenUsage };
 //# sourceMappingURL=usage.d.ts.map

package/dist/usage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"usage.d.ts","names":[],"sources":["../src/usage.ts"],"mappings":";KAIY;EACV;EACA;EACA;EACA;EACA;;UAGe;EACf;EACA;;iBAuHc,2BACd,SAAS,2CACR;;;;;;;;;iBAsCa,uBAAuB"}
+{"version":3,"file":"usage.d.ts","names":[],"sources":["../src/usage.ts"],"mappings":";YAIY;EACV;EACA;EACA;EACA;EACA;;iBAGe;EACf;EACA;;wBAuHc,2BACd,SAAS,2CACR;;;;;;;;;wBAsCa,uBAAuB"}

package/dist/utils/getVercelAIMessages.d.ts

@@ -1,7 +1,6 @@
 import { UIMessage } from "ai";
 import { ThreadMessage } from "@assistant-ui/core";
 //#region src/utils/getVercelAIMessages.d.ts
-declare const getVercelAIMessages: <UI_MESSAGE extends UIMessage = UIMessage>(message: ThreadMessage) => UI_MESSAGE[];
+export declare const getVercelAIMessages: <UI_MESSAGE extends UIMessage = UIMessage>(message: ThreadMessage) => UI_MESSAGE[];
 //#endregion
-export { getVercelAIMessages };
 //# sourceMappingURL=getVercelAIMessages.d.ts.map

package/dist/utils/getVercelAIMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"getVercelAIMessages.d.ts","names":[],"sources":["../../src/utils/getVercelAIMessages.ts"],"mappings":";;;cAMa,sBAAuB,mBAAmB,YAAY,WACjE,SAAS,kBAEmC"}
+{"version":3,"file":"getVercelAIMessages.d.ts","names":[],"sources":["../../src/utils/getVercelAIMessages.ts"],"mappings":";;;qBAMa,sBAAuB,mBAAmB,YAAY,WACjE,SAAS,kBAEmC"}

package/dist/utils/sliceMessagesUntil.d.ts

@@ -1,6 +1,5 @@
 import { UIMessage } from "ai";
 //#region src/utils/sliceMessagesUntil.d.ts
-declare const sliceMessagesUntil: <UI_MESSAGE extends UIMessage = UIMessage>(messages: UI_MESSAGE[], messageId: string | null) => UI_MESSAGE[];
+export declare const sliceMessagesUntil: <UI_MESSAGE extends UIMessage = UIMessage>(messages: UI_MESSAGE[], messageId: string | null) => UI_MESSAGE[];
 //#endregion
-export { sliceMessagesUntil };
 //# sourceMappingURL=sliceMessagesUntil.d.ts.map

package/dist/utils/sliceMessagesUntil.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sliceMessagesUntil.d.ts","names":[],"sources":["../../src/utils/sliceMessagesUntil.ts"],"mappings":";;cAEa,qBAAsB,mBAAmB,YAAY,WAChE,UAAU,cACV,6BAAwB"}
+{"version":3,"file":"sliceMessagesUntil.d.ts","names":[],"sources":["../../src/utils/sliceMessagesUntil.ts"],"mappings":";;qBAEa,qBAAsB,mBAAmB,YAAY,WAChE,UAAU,cACV,6BAAwB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/ai-sdk",
-  "version": "0.0.5",
+  "version": "0.0.7",
   "description": "Vercel AI SDK adapter for assistant-ui",
   "keywords": [
     "ai-sdk",
@@ -46,14 +46,14 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@ai-sdk/mcp": "^2.0.45",
-    "@ai-sdk/react": "^4.0.96",
-    "@assistant-ui/core": "^0.3.18",
-    "@assistant-ui/store": "^0.3.13",
-    "@assistant-ui/tap": "^0.9.17",
-    "ai": "^7.0.93",
-    "assistant-cloud": "*",
-    "assistant-stream": "^0.3.42"
+    "@ai-sdk/mcp": "^2.0.49",
+    "@ai-sdk/react": "^4.0.104",
+    "@assistant-ui/core": "^0.3.20",
+    "@assistant-ui/store": "^0.3.14",
+    "@assistant-ui/tap": "^0.9.18",
+    "ai": "^7.0.101",
+    "assistant-cloud": "^0.2.2",
+    "assistant-stream": "^0.3.44"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -68,14 +68,14 @@
     }
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.26",
+    "@assistant-ui/x-buildutils": "0.0.28",
     "@testing-library/react": "^16.3.3",
     "@types/json-schema": "^7.0.15",
-    "@types/react": "^19.2.18",
-    "@types/react-dom": "^19.2.7",
+    "@types/react": "^19.3.0",
+    "@types/react-dom": "^19.3.0",
     "jsdom": "^30.0.1",
-    "react": "^19.2.8",
-    "vitest": "^4.1.11"
+    "react": "^19.3.0",
+    "vitest": "^5.0.1"
   },
   "publishConfig": {
     "access": "public",
@@ -93,6 +93,7 @@
   "scripts": {
     "build": "aui-build",
     "test": "vitest run",
+    "typecheck": "tsc --noEmit && tsc --noEmit -p tsconfig.test.json",
     "test:react-compiler": "vitest run --config vitest.react-compiler.config.ts",
     "test:standalone": "vitest run --project standalone",
     "test:watch": "vitest"

package/src/adapters/aiSDKFormatAdapter.ts

@@ -1,41 +1,4 @@
-import type { UIMessage } from "ai";
-import type {
-  MessageFormatAdapter,
-  MessageFormatItem,
-  MessageStorageEntry,
-} from "@assistant-ui/core";
-
-// Storage format for AI SDK messages - just the UIMessage
-export type AISDKStorageFormat = Omit<UIMessage, "id">;
-
-export const aiSDKV6FormatAdapter: MessageFormatAdapter<
-  UIMessage,
-  AISDKStorageFormat
-> = {
-  format: "ai-sdk/v6",
-
-  encode({
-    message: { id, parts, ...message },
-  }: MessageFormatItem<UIMessage>): AISDKStorageFormat {
-    return {
-      ...message,
-      parts,
-    };
-  },
-
-  decode(
-    stored: MessageStorageEntry<AISDKStorageFormat>,
-  ): MessageFormatItem<UIMessage> {
-    return {
-      parentId: stored.parent_id,
-      message: {
-        id: stored.id,
-        ...stored.content,
-      },
-    };
-  },
-
-  getId(message: UIMessage): string {
-    return message.id;
-  },
-};
+export {
+  aiSDKV6FormatAdapter,
+  type AISDKStorageFormat,
+} from "assistant-cloud/ai-sdk";