@assistant-ui/ai-sdk 0.0.3 → 0.0.4

package/dist/runtime/useAISDKRuntime.js

@@ -10,7 +10,8 @@ import { aiSDKV6FormatAdapter } from "../adapters/aiSDKFormatAdapter.js";
 import { toExportedMessageRepository, useExternalHistory } from "./useExternalHistory.js";
 import { useStreamingTiming as useStreamingTiming$1 } from "./useStreamingTiming.js";
 import { useExternalStoreRuntime, useRuntimeAdapters } from "@assistant-ui/core/react";
-import { useEffect, useMemo, useRef, useState } from "@assistant-ui/tap/react-shim";
+import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useCallback, useEffect, useInsertionEffect, useMemo, useRef, useState } from "@assistant-ui/tap/react-shim";
 import { generateId, isToolUIPart } from "ai";
 import { getExternalStoreMessages, pickExternalStoreSharedOptions } from "@assistant-ui/core";
 import { MessageRepository, consumeSuggestionResult } from "@assistant-ui/core/internal";
@@ -22,64 +23,118 @@ const toUIMessage = (createMessage, fallbackRole) => ({
 });
 const EMPTY_SUGGESTIONS = [];
 const useGeneratedSuggestions = (suggestionAdapter, messages, isRunning) => {
+	const $ = c(13);
 	const [suggestions, setSuggestions] = useState(EMPTY_SUGGESTIONS);
 	const controllerRef = useRef(null);
 	const wasRunningRef = useRef(false);
 	const messagesRef = useRef(messages);
-	messagesRef.current = messages;
+	let t0;
+	let t1;
+	if ($[0] !== messages) {
+		t0 = () => {
+			messagesRef.current = messages;
+		};
+		t1 = [messages];
+		$[0] = messages;
+		$[1] = t0;
+		$[2] = t1;
+	} else {
+		t0 = $[1];
+		t1 = $[2];
+	}
+	useInsertionEffect(t0, t1);
 	const adapterRef = useRef(suggestionAdapter);
-	adapterRef.current = suggestionAdapter;
-	useEffect(() => {
-		const clearSuggestions = () => {
-			controllerRef.current?.abort();
-			controllerRef.current = null;
-			setSuggestions((prev) => prev.length === 0 ? prev : EMPTY_SUGGESTIONS);
+	let t2;
+	let t3;
+	if ($[3] !== suggestionAdapter) {
+		t2 = () => {
+			adapterRef.current = suggestionAdapter;
 		};
-		const adapter = adapterRef.current;
-		if (!adapter) {
-			clearSuggestions();
-			wasRunningRef.current = isRunning;
-			return;
-		}
-		if (isRunning) {
-			if (!wasRunningRef.current) clearSuggestions();
-			wasRunningRef.current = true;
-			return;
-		}
-		if (!wasRunningRef.current) return;
-		wasRunningRef.current = false;
-		const currentMessages = messagesRef.current;
-		const last = currentMessages.at(-1);
-		if (last?.role !== "assistant") return;
-		if (last.status?.type === "requires-action") return;
-		const controller = new AbortController();
-		controllerRef.current = controller;
-		const { signal } = controller;
-		(async () => {
-			try {
-				const promiseOrGenerator = adapter.generate({
-					messages: currentMessages,
-					signal
-				});
-				await consumeSuggestionResult(promiseOrGenerator, {
-					signal,
-					onUpdate: setSuggestions
-				});
-			} catch {}
-		})();
-	}, [suggestionAdapter != null, isRunning]);
-	useEffect(() => {
-		return () => {
+		t3 = [suggestionAdapter];
+		$[3] = suggestionAdapter;
+		$[4] = t2;
+		$[5] = t3;
+	} else {
+		t2 = $[4];
+		t3 = $[5];
+	}
+	useInsertionEffect(t2, t3);
+	const hasAdapter = suggestionAdapter != null;
+	let t4;
+	if ($[6] !== isRunning) {
+		t4 = () => {
+			const clearSuggestions = () => {
+				controllerRef.current?.abort();
+				controllerRef.current = null;
+				setSuggestions(_temp);
+			};
+			const adapter = adapterRef.current;
+			if (!adapter) {
+				clearSuggestions();
+				wasRunningRef.current = isRunning;
+				return;
+			}
+			if (isRunning) {
+				if (!wasRunningRef.current) clearSuggestions();
+				wasRunningRef.current = true;
+				return;
+			}
+			if (!wasRunningRef.current) return;
+			wasRunningRef.current = false;
+			const currentMessages = messagesRef.current;
+			const last = currentMessages.at(-1);
+			if (last?.role !== "assistant") return;
+			if (last.status?.type === "requires-action") return;
+			const controller = new AbortController();
+			controllerRef.current = controller;
+			const { signal } = controller;
+			(async () => {
+				try {
+					const promiseOrGenerator = adapter.generate({
+						messages: currentMessages,
+						signal
+					});
+					await consumeSuggestionResult(promiseOrGenerator, {
+						signal,
+						onUpdate: setSuggestions
+					});
+				} catch {}
+			})();
+		};
+		$[6] = isRunning;
+		$[7] = t4;
+	} else t4 = $[7];
+	let t5;
+	if ($[8] !== hasAdapter || $[9] !== isRunning) {
+		t5 = [hasAdapter, isRunning];
+		$[8] = hasAdapter;
+		$[9] = isRunning;
+		$[10] = t5;
+	} else t5 = $[10];
+	useEffect(t4, t5);
+	let t6;
+	let t7;
+	if ($[11] === Symbol.for("react.memo_cache_sentinel")) {
+		t6 = () => () => {
 			controllerRef.current?.abort();
 		};
-	}, []);
+		t7 = [];
+		$[11] = t6;
+		$[12] = t7;
+	} else {
+		t6 = $[11];
+		t7 = $[12];
+	}
+	useEffect(t6, t7);
 	return suggestions;
 };
+const NO_CANCELLED_MESSAGE_IDS = /* @__PURE__ */ new Set();
 const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 	const { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, onResume, onResumeToolCall, joinStrategy, messageRepository, unstable_onBranchChange } = adapter;
 	const suggestionAdapter = adapters?.suggestion;
 	const contextAdapters = useRuntimeAdapters();
 	const [toolStatuses, setToolStatuses] = useState({});
+	const [cancelledMessages, setCancelledMessages] = useState(null);
 	const toolArgsKeyOrderCacheRef = useRef(/* @__PURE__ */ new Map());
 	const toolLastInputCacheRef = useRef(/* @__PURE__ */ new Map());
 	const mcpAppMetadataCacheRef = useRef(/* @__PURE__ */ new Map());
@@ -87,9 +142,34 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 	const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
 	const providerIsRunning = chatHelpers.status === "submitted" || chatHelpers.status === "streaming";
 	const isRunning = providerIsRunning || hasExecutingTools;
+	const wasProviderRunningRef = useRef(providerIsRunning);
 	const messageTiming = useStreamingTiming$1(chatHelpers.messages, isRunning);
 	const lastMessage = chatHelpers.messages.at(-1);
 	const optimisticMessageId = isRunning && lastMessage?.role === "assistant" ? lastMessage.id : void 0;
+	const cancelledMessageIds = cancelledMessages?.chatId === chatHelpers.id ? cancelledMessages.ids : NO_CANCELLED_MESSAGE_IDS;
+	const retractCancellation = useCallback((chatId, messageId) => {
+		setCancelledMessages((prev) => {
+			if (prev?.chatId !== chatId || !prev.ids.has(messageId)) return prev;
+			const ids = new Set(prev.ids);
+			ids.delete(messageId);
+			return {
+				chatId,
+				ids
+			};
+		});
+	}, []);
+	const resumedMessageId = providerIsRunning && lastMessage?.role === "assistant" ? lastMessage.id : void 0;
+	useEffect(() => {
+		const wasProviderRunning = wasProviderRunningRef.current;
+		wasProviderRunningRef.current = providerIsRunning;
+		if (wasProviderRunning || !resumedMessageId) return;
+		retractCancellation(chatHelpers.id, resumedMessageId);
+	}, [
+		providerIsRunning,
+		resumedMessageId,
+		chatHelpers.id,
+		retractCancellation
+	]);
 	const messages = AISDKMessageConverter.useThreadMessages({
 		isRunning,
 		messages: chatHelpers.messages,
@@ -101,12 +181,14 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 			toolLastInputCache: toolLastInputCacheRef.current,
 			mcpAppMetadataCache: mcpAppMetadataCacheRef.current,
 			...optimisticMessageId && { optimisticMessageId },
-			...chatHelpers.error && { error: chatHelpers.error.message }
+			...chatHelpers.error && { error: chatHelpers.error.message },
+			...cancelledMessageIds.size > 0 && { cancelledMessageIds }
 		}), [
 			toolStatuses,
 			messageTiming,
 			optimisticMessageId,
-			chatHelpers.error
+			chatHelpers.error,
+			cancelledMessageIds
 		])
 	});
 	const exportedMessageRepository = useMemo(() => {
@@ -121,12 +203,12 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 	const { isLoading, deleteMessage: deleteHistoryMessage } = useExternalHistory(runtimeRef, adapters?.history ?? contextAdapters?.history, AISDKMessageConverter.toThreadMessages, aiSDKV6FormatAdapter, (messages_0) => {
 		chatHelpers.setMessages(messages_0);
 	});
-	const { id: chatId, messages: chatMessages, status: chatStatus, error } = chatHelpers;
+	const { id: chatId_0, messages: chatMessages, status: chatStatus, error } = chatHelpers;
 	const extras = useMemo(() => aiSDKExtras.provide({
 		chat: chatHelpers,
 		error
 	}), [
-		chatId,
+		chatId_0,
 		chatMessages,
 		chatStatus,
 		error
@@ -189,39 +271,54 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 			runtimeRef.current.thread.import(exportedRepo);
 		},
 		onCancel: async () => {
+			const message = chatHelpers.messages.at(-1);
+			const cancelledId = isRunning && message?.role === "assistant" ? message.id : void 0;
+			if (cancelledId) {
+				const liveIds = new Set(chatHelpers.messages.map((m) => m.id));
+				setCancelledMessages((prev_0) => {
+					const kept = prev_0?.chatId === chatHelpers.id ? [...prev_0.ids].filter((id) => liveIds.has(id)) : [];
+					return {
+						chatId: chatHelpers.id,
+						ids: /* @__PURE__ */ new Set([...kept, cancelledId])
+					};
+				});
+			}
 			try {
 				await chatHelpers.stop();
 			} catch (error_0) {
-				if (!(error_0 instanceof Error && error_0.name === "AbortError")) throw error_0;
+				if (!(error_0 instanceof Error && error_0.name === "AbortError")) {
+					if (cancelledId) retractCancellation(chatHelpers.id, cancelledId);
+					throw error_0;
+				}
 			}
 		},
-		onNew: async (message) => {
-			const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
-			if (!(message.startRun ?? message.role === "user")) {
-				chatHelpers.setMessages((current) => [...current, toUIMessage(createMessage, message.role)]);
+		onNew: async (message_0) => {
+			const createMessage = (customToCreateMessage ?? toCreateMessage)(message_0);
+			if (!(message_0.startRun ?? message_0.role === "user")) {
+				chatHelpers.setMessages((current) => [...current, toUIMessage(createMessage, message_0.role)]);
 				return;
 			}
-			lastRunConfigRef.current = message.runConfig;
+			lastRunConfigRef.current = message_0.runConfig;
 			await completePendingToolCalls();
-			await chatHelpers.sendMessage(createMessage, { metadata: message.runConfig });
+			await chatHelpers.sendMessage(createMessage, { metadata: message_0.runConfig });
 		},
-		onEdit: async (message_0) => {
-			const createMessage_0 = (customToCreateMessage ?? toCreateMessage)(message_0);
-			if (!(message_0.startRun ?? message_0.role === "user")) {
-				chatHelpers.setMessages((current_0) => [...sliceMessagesUntil(current_0, message_0.parentId), toUIMessage(createMessage_0, message_0.role)]);
+		onEdit: async (message_1) => {
+			const createMessage_0 = (customToCreateMessage ?? toCreateMessage)(message_1);
+			if (!(message_1.startRun ?? message_1.role === "user")) {
+				chatHelpers.setMessages((current_0) => [...sliceMessagesUntil(current_0, message_1.parentId), toUIMessage(createMessage_0, message_1.role)]);
 				return;
 			}
-			lastRunConfigRef.current = message_0.runConfig;
-			chatHelpers.setMessages((current_1) => sliceMessagesUntil(current_1, message_0.parentId));
-			await chatHelpers.sendMessage(createMessage_0, { metadata: message_0.runConfig });
+			lastRunConfigRef.current = message_1.runConfig;
+			chatHelpers.setMessages((current_1) => sliceMessagesUntil(current_1, message_1.parentId));
+			await chatHelpers.sendMessage(createMessage_0, { metadata: message_1.runConfig });
 		},
-		onDelete: async (messageId) => {
+		onDelete: async (messageId_0) => {
 			const threadMessages = runtimeRef.current.thread.getState().messages;
-			const messageIndex = threadMessages.findIndex((message_1) => message_1.id === messageId);
+			const messageIndex = threadMessages.findIndex((message_2) => message_2.id === messageId_0);
 			if (messageIndex === -1) return;
-			await deleteHistoryMessage(messageId);
-			const deleteIds = new Set(getExternalStoreMessages(threadMessages[messageIndex]).map((message_2) => message_2.id));
-			chatHelpers.setMessages((current_2) => current_2.filter((message_3) => !deleteIds.has(message_3.id)));
+			await deleteHistoryMessage(messageId_0);
+			const deleteIds = new Set(getExternalStoreMessages(threadMessages[messageIndex]).map((message_3) => message_3.id));
+			chatHelpers.setMessages((current_2) => current_2.filter((message_4) => !deleteIds.has(message_4.id)));
 		},
 		onReload: async (parentId_0, config) => {
 			lastRunConfigRef.current = config.runConfig;
@@ -255,6 +352,7 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 			options: { metadata: lastRunConfigRef.current }
 		})),
 		...pickExternalStoreSharedOptions(adapter),
+		...adapter.unstable_messageRepositoryInstance && { unstable_messageRepositoryInstance: adapter.unstable_messageRepositoryInstance },
 		...suggestionAdapter ? { suggestions: generatedSuggestions } : {},
 		...onResume && { onResume },
 		...onResumeToolCall && { onResumeToolCall },
@@ -268,7 +366,9 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 		isLoading
 	});
 	const setMessagesRef = useRef(chatHelpers.setMessages);
-	setMessagesRef.current = chatHelpers.setMessages;
+	useInsertionEffect(() => {
+		setMessagesRef.current = chatHelpers.setMessages;
+	}, [chatHelpers.setMessages]);
 	useEffect(() => {
 		if (hasSeededRepositoryRef.current) return;
 		if (!exportedMessageRepository) return;
@@ -283,6 +383,9 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 	}, [exportedMessageRepository, chatHelpers.messages.length]);
 	return runtime;
 };
+function _temp(prev) {
+	return prev.length === 0 ? prev : EMPTY_SUGGESTIONS;
+}
 //#endregion
 export { useAISDKRuntime };
 

package/dist/runtime/useAISDKRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAISDKRuntime.js","names":["useEffect","useMemo","useRef","useState","UIMessage","useChat","CreateUIMessage","UseChatHelpers","isToolUIPart","generateId","useExternalStoreRuntime","useRuntimeAdapters","JoinStrategy","SuggestionAdapter","ThreadSuggestion","ToolExecutionStatus","ExternalStoreAdapter","ExternalStoreSharedOptions","ThreadHistoryAdapter","AssistantRuntime","ThreadMessage","MessageFormatAdapter","MessageFormatItem","MessageFormatRepository","AppendMessage","RunConfig","McpAppMetadata","getExternalStoreMessages","pickExternalStoreSharedOptions","consumeSuggestionResult","MessageRepository","ReadonlyJSONObject","sliceMessagesUntil","toCreateMessage","vercelAttachmentAdapter","getVercelAIMessages","AISDKMessageConverter","wrapModelContentEnvelope","AISDKStorageFormat","aiSDKV6FormatAdapter","useExternalHistory","toExportedMessageRepository","useStreamingTiming","aiSDKExtras","CustomToCreateMessageFunction","message","UI_MESSAGE","toUIMessage","createMessage","fallbackRole","id","role","AISDKRuntimeAdapter","adapters","NonNullable","history","suggestion","cancelPendingToolCallsOnSend","onResume","onResumeToolCall","joinStrategy","messageRepository","unstable_onBranchChange","EMPTY_SUGGESTIONS","useGeneratedSuggestions","suggestionAdapter","messages","isRunning","suggestions","setSuggestions","controllerRef","AbortController","wasRunningRef","messagesRef","current","adapterRef","hasAdapter","clearSuggestions","abort","prev","length","adapter","currentMessages","last","at","status","type","controller","signal","promiseOrGenerator","generate","onUpdate","useAISDKRuntime","chatHelpers","ReturnType","customToCreateMessage","contextAdapters","toolStatuses","setToolStatuses","Record","toolArgsKeyOrderCacheRef","Map","toolLastInputCacheRef","mcpAppMetadataCacheRef","lastRunConfigRef","undefined","hasExecutingTools","Object","values","some","s","providerIsRunning","messageTiming","lastMessage","optimisticMessageId","useThreadMessages","metadata","toolArgsKeyOrderCache","toolLastInputCache","mcpAppMetadataCache","error","exportedMessageRepository","converted","toThreadMessages","generatedSuggestions","runtimeRef","runtime","isLoading","deleteMessage","deleteHistoryMessage","setMessages","chatId","chatMessages","chatStatus","extras","provide","chat","completePendingToolCalls","hasChanges","parts","map","part","state","approval","_approval","rest","const","errorText","slice","hasSeededRepositoryRef","shouldFeedRepository","unstable_enableToolInvocations","filter","Boolean","flat","onImport","onExportExternalState","exported","thread","export","expandedMessages","lastInnerIdMap","item","innerMessages","parentId","get","innerMessage","push","getId","set","result","headId","onLoadExternalState","repo","exportedRepo","import","onCancel","stop","Error","name","onNew","startRun","runConfig","sendMessage","onEdit","onDelete","messageId","threadMessages","getState","messageIndex","findIndex","deleteIds","Set","has","onReload","config","newMessages","regenerate","onAddToolResult","toolCallId","toolName","isError","modelContent","options","Promise","resolve","addToolOutput","tool","JSON","stringify","output","onRespondToToolApproval","approvalId","approved","reason","addToolApprovalResponse","attachments","setMessagesRef","tempRepo","getMessages","flatMap"],"sources":["../../src/runtime/useAISDKRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useEffect, useMemo, useRef, useState } from \"react\";\nimport type {\n  UIMessage,\n  useChat,\n  CreateUIMessage,\n  UseChatHelpers,\n} from \"@ai-sdk/react\";\nimport { isToolUIPart, generateId } from \"ai\";\nimport {\n  useExternalStoreRuntime,\n  useRuntimeAdapters,\n  type JoinStrategy,\n} from \"@assistant-ui/core/react\";\nimport type {\n  SuggestionAdapter,\n  ThreadSuggestion,\n  ToolExecutionStatus,\n} from \"@assistant-ui/core\";\nimport type {\n  ExternalStoreAdapter,\n  ExternalStoreSharedOptions,\n  ThreadHistoryAdapter,\n  AssistantRuntime,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatItem,\n  MessageFormatRepository,\n  AppendMessage,\n  RunConfig,\n  McpAppMetadata,\n} from \"@assistant-ui/core\";\nimport {\n  getExternalStoreMessages,\n  pickExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport {\n  consumeSuggestionResult,\n  MessageRepository,\n} from \"@assistant-ui/core/internal\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../converters/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../adapters/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../utils/getVercelAIMessages\";\nimport { AISDKMessageConverter } from \"../converters/convertMessage\";\nimport { wrapModelContentEnvelope } from \"../converters/modelContentEnvelope\";\nimport {\n  type AISDKStorageFormat,\n  aiSDKV6FormatAdapter,\n} from \"../adapters/aiSDKFormatAdapter\";\nimport {\n  useExternalHistory,\n  toExportedMessageRepository,\n} from \"./useExternalHistory\";\nimport { useStreamingTiming } from \"./useStreamingTiming\";\nimport { aiSDKExtras } from \"../aiSDKExtras\";\n\nexport type CustomToCreateMessageFunction = <\n  UI_MESSAGE extends UIMessage = UIMessage,\n>(\n  message: AppendMessage,\n) => CreateUIMessage<UI_MESSAGE>;\n\nconst toUIMessage = <UI_MESSAGE extends UIMessage>(\n  createMessage: CreateUIMessage<UI_MESSAGE>,\n  fallbackRole: UI_MESSAGE[\"role\"],\n): UI_MESSAGE =>\n  ({\n    ...createMessage,\n    id: createMessage.id ?? generateId(),\n    role: createMessage.role ?? fallbackRole,\n  }) as UI_MESSAGE;\n\nexport type AISDKRuntimeAdapter<UI_MESSAGE extends UIMessage = UIMessage> =\n  ExternalStoreSharedOptions & {\n    adapters?:\n      | (NonNullable<ExternalStoreAdapter[\"adapters\"]> & {\n          history?: ThreadHistoryAdapter | undefined;\n          suggestion?: SuggestionAdapter | undefined;\n        })\n      | undefined;\n    toCreateMessage?: CustomToCreateMessageFunction;\n    /**\n     * Whether to automatically cancel pending interactive tool calls when the user sends a new message.\n     *\n     * When enabled (default), the pending tool calls will be marked as failed with an error message\n     * indicating the user cancelled the tool call by sending a new message.\n     *\n     * @default true\n     */\n    cancelPendingToolCallsOnSend?: boolean | undefined;\n    /**\n     * Called when `runtime.thread.resumeRun(config)` is invoked.\n     *\n     * When omitted, `resumeRun` throws `\"Runtime does not support resuming runs.\"`.\n     * Provide this to bridge resume invocations into a custom replay channel\n     * (for example, an SSE reconnect endpoint keyed by turn id).\n     */\n    onResume?: ExternalStoreAdapter[\"onResume\"];\n    /**\n     * Called when `runtime.thread.resumeToolCall(options)` is invoked for a tool call the in-process tracker does not own.\n     *\n     * When omitted, `resumeToolCall` throws `\"Tool call ${toolCallId} is not waiting for resume.\"`.\n     * Provide this to bridge resume-tool-call invocations into a custom handler.\n     */\n    onResumeToolCall?: ExternalStoreAdapter[\"onResumeToolCall\"];\n    /**\n     * How consecutive assistant messages are rendered.\n     *\n     * `\"concat-content\"` (the default) merges them into a single thread message.\n     * `\"none\"` keeps each assistant message as its own thread message, which is\n     * useful when a backend persists proactive or consecutive assistant messages\n     * as separate entries.\n     */\n    joinStrategy?: JoinStrategy | undefined;\n    /**\n     * A branch-aware AI SDK message tree seeded once when `useChat` is empty.\n     * After that seed, live updates come only from `useChat`. A later empty\n     * chat or a new object identity does not reload the tree.\n     */\n    messageRepository?: MessageFormatRepository<UI_MESSAGE>;\n    /**\n     * Called after an explicit `switchToBranch` (for example a BranchPicker\n     * click). Complements `setMessages` and does not enable switching by itself.\n     *\n     * @deprecated This API is still under active development and might change without notice.\n     */\n    unstable_onBranchChange?: ExternalStoreAdapter[\"unstable_onBranchChange\"];\n  };\n\nconst EMPTY_SUGGESTIONS: readonly ThreadSuggestion[] = [];\n\nconst useGeneratedSuggestions = (\n  suggestionAdapter: SuggestionAdapter | undefined,\n  messages: readonly ThreadMessage[],\n  isRunning: boolean,\n): readonly ThreadSuggestion[] => {\n  const [suggestions, setSuggestions] =\n    useState<readonly ThreadSuggestion[]>(EMPTY_SUGGESTIONS);\n  const controllerRef = useRef<AbortController | null>(null);\n  const wasRunningRef = useRef(false);\n  const messagesRef = useRef(messages);\n  messagesRef.current = messages;\n  const adapterRef = useRef(suggestionAdapter);\n  adapterRef.current = suggestionAdapter;\n  const hasAdapter = suggestionAdapter != null;\n\n  useEffect(() => {\n    const clearSuggestions = () => {\n      controllerRef.current?.abort();\n      controllerRef.current = null;\n      setSuggestions((prev) => (prev.length === 0 ? prev : EMPTY_SUGGESTIONS));\n    };\n\n    const adapter = adapterRef.current;\n    if (!adapter) {\n      clearSuggestions();\n      wasRunningRef.current = isRunning;\n      return;\n    }\n\n    if (isRunning) {\n      if (!wasRunningRef.current) {\n        clearSuggestions();\n      }\n      wasRunningRef.current = true;\n      return;\n    }\n\n    if (!wasRunningRef.current) return;\n    wasRunningRef.current = false;\n\n    const currentMessages = messagesRef.current;\n    const last = currentMessages.at(-1);\n    if (last?.role !== \"assistant\") return;\n    if (last.status?.type === \"requires-action\") return;\n\n    const controller = new AbortController();\n    controllerRef.current = controller;\n    const { signal } = controller;\n\n    void (async () => {\n      try {\n        const promiseOrGenerator = adapter.generate({\n          messages: currentMessages,\n          signal,\n        });\n\n        await consumeSuggestionResult(promiseOrGenerator, {\n          signal,\n          onUpdate: setSuggestions,\n        });\n      } catch {}\n    })();\n  }, [hasAdapter, isRunning]);\n\n  useEffect(() => {\n    return () => {\n      controllerRef.current?.abort();\n    };\n  }, []);\n\n  return suggestions;\n};\n\nexport const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>,\n  adapter: AISDKRuntimeAdapter<UI_MESSAGE> = {},\n) => {\n  const {\n    adapters,\n    toCreateMessage: customToCreateMessage,\n    cancelPendingToolCallsOnSend = true,\n    onResume,\n    onResumeToolCall,\n    joinStrategy,\n    messageRepository,\n    unstable_onBranchChange,\n  } = adapter;\n  const suggestionAdapter = adapters?.suggestion;\n  const contextAdapters = useRuntimeAdapters();\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const toolArgsKeyOrderCacheRef = useRef<Map<string, Map<string, string[]>>>(\n    new Map(),\n  );\n  const toolLastInputCacheRef = useRef<Map<string, ReadonlyJSONObject>>(\n    new Map(),\n  );\n  const mcpAppMetadataCacheRef = useRef<Map<string, McpAppMetadata>>(new Map());\n  const lastRunConfigRef = useRef<RunConfig | undefined>(undefined);\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const providerIsRunning =\n    chatHelpers.status === \"submitted\" || chatHelpers.status === \"streaming\";\n  const isRunning = providerIsRunning || hasExecutingTools;\n\n  const messageTiming = useStreamingTiming(chatHelpers.messages, isRunning);\n\n  // Flag the streaming message optimistic: its id can be swapped for a server\n  // id mid-run, and the repository then drops the orphaned pre-swap id (#4037).\n  const lastMessage = chatHelpers.messages.at(-1);\n  const optimisticMessageId =\n    isRunning && lastMessage?.role === \"assistant\" ? lastMessage.id : undefined;\n\n  const messages = AISDKMessageConverter.useThreadMessages({\n    isRunning,\n    messages: chatHelpers.messages,\n    joinStrategy,\n    metadata: useMemo(\n      () => ({\n        toolStatuses,\n        messageTiming,\n        toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,\n        toolLastInputCache: toolLastInputCacheRef.current,\n        mcpAppMetadataCache: mcpAppMetadataCacheRef.current,\n        ...(optimisticMessageId && { optimisticMessageId }),\n        ...(chatHelpers.error && { error: chatHelpers.error.message }),\n      }),\n      [toolStatuses, messageTiming, optimisticMessageId, chatHelpers.error],\n    ),\n  });\n\n  const exportedMessageRepository = useMemo(() => {\n    if (!messageRepository) return undefined;\n    const converted = toExportedMessageRepository(\n      AISDKMessageConverter.toThreadMessages as (\n        messages: UI_MESSAGE[],\n      ) => ThreadMessage[],\n      messageRepository,\n    );\n    return converted.messages.length > 0 ? converted : undefined;\n  }, [messageRepository]);\n\n  const generatedSuggestions = useGeneratedSuggestions(\n    suggestionAdapter,\n    messages,\n    isRunning,\n  );\n\n  const [runtimeRef] = useState(() => ({\n    get current(): AssistantRuntime {\n      return runtime;\n    },\n  }));\n\n  const { isLoading, deleteMessage: deleteHistoryMessage } = useExternalHistory(\n    runtimeRef,\n    adapters?.history ?? contextAdapters?.history,\n    AISDKMessageConverter.toThreadMessages as (\n      messages: UI_MESSAGE[],\n    ) => ThreadMessage[],\n    aiSDKV6FormatAdapter as MessageFormatAdapter<\n      UI_MESSAGE,\n      AISDKStorageFormat\n    >,\n    (messages) => {\n      chatHelpers.setMessages(messages);\n    },\n  );\n\n  const {\n    id: chatId,\n    messages: chatMessages,\n    status: chatStatus,\n    error,\n  } = chatHelpers;\n  const extras = useMemo(\n    () =>\n      aiSDKExtras.provide({\n        chat: chatHelpers as unknown as UseChatHelpers<UIMessage>,\n        error,\n      }),\n    // oxlint-disable-next-line react/exhaustive-deps -- keyed on the chat's identity and reactive snapshots; useChat re-mints the helpers object every render while its remaining fields are instance-bound methods, and a render-stable extras identity is what lets the external-store core dedupe adapter updates\n    [chatId, chatMessages, chatStatus, error],\n  );\n\n  const completePendingToolCalls = async () => {\n    if (!cancelPendingToolCallsOnSend) return;\n\n    // The runtime auto-aborts in-flight tool invocations when a new run\n    // is dispatched (append() / startRun()). All we need to do here is\n    // mark any tool without a result as cancelled in the UI message list.\n\n    // Mark any tool without a result as cancelled (uses setMessages to avoid triggering sendAutomaticallyWhen)\n    chatHelpers.setMessages((messages) => {\n      const lastMessage = messages.at(-1);\n      if (lastMessage?.role !== \"assistant\") return messages;\n\n      let hasChanges = false;\n      const parts = lastMessage.parts?.map((part) => {\n        if (!isToolUIPart(part)) return part;\n        if (\n          part.state === \"output-available\" ||\n          part.state === \"output-error\" ||\n          part.state === \"output-denied\"\n        )\n          return part;\n\n        hasChanges = true;\n        const { approval: _approval, ...rest } = part;\n        return {\n          ...rest,\n          state: \"output-error\" as const,\n          errorText: \"User cancelled tool call by sending a new message.\",\n        };\n      });\n\n      if (!hasChanges) return messages;\n      return [...messages.slice(0, -1), { ...lastMessage, parts }];\n    });\n  };\n\n  const hasSeededRepositoryRef = useRef(false);\n  const shouldFeedRepository =\n    exportedMessageRepository != null &&\n    !hasSeededRepositoryRef.current &&\n    messages.length === 0;\n\n  const runtime = useExternalStoreRuntime({\n    isRunning: providerIsRunning,\n    ...(shouldFeedRepository\n      ? { messageRepository: exportedMessageRepository }\n      : { messages }),\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onImport: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onExportExternalState: (): MessageFormatRepository<UI_MESSAGE> => {\n      const exported = runtimeRef.current.thread.export();\n\n      const expandedMessages: MessageFormatItem<UI_MESSAGE>[] = [];\n      const lastInnerIdMap = new Map<string, string>();\n\n      for (const item of exported.messages) {\n        const innerMessages = getExternalStoreMessages<UI_MESSAGE>(\n          item.message,\n        );\n        let parentId =\n          item.parentId != null\n            ? (lastInnerIdMap.get(item.parentId) ?? item.parentId)\n            : null;\n        for (const innerMessage of innerMessages) {\n          expandedMessages.push({ parentId, message: innerMessage });\n          parentId = aiSDKV6FormatAdapter.getId(innerMessage as UIMessage);\n        }\n        if (innerMessages.length > 0) {\n          lastInnerIdMap.set(\n            item.message.id,\n            aiSDKV6FormatAdapter.getId(\n              innerMessages[innerMessages.length - 1]! as UIMessage,\n            ),\n          );\n        }\n      }\n\n      const result: MessageFormatRepository<UI_MESSAGE> = {\n        messages: expandedMessages,\n      };\n\n      if (exported.headId != null) {\n        result.headId = lastInnerIdMap.get(exported.headId) ?? exported.headId;\n      }\n\n      return result;\n    },\n    onLoadExternalState: (repo: MessageFormatRepository<UI_MESSAGE>) => {\n      // Convert MessageFormatRepository to ExportedMessageRepository\n      const exportedRepo = toExportedMessageRepository(\n        AISDKMessageConverter.toThreadMessages,\n        repo,\n      );\n\n      // Import into the thread's MessageRepository\n      runtimeRef.current.thread.import(exportedRepo);\n    },\n    onCancel: async () => {\n      try {\n        await chatHelpers.stop();\n      } catch (error) {\n        if (!(error instanceof Error && error.name === \"AbortError\")) {\n          throw error;\n        }\n      }\n    },\n    onNew: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...current,\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      await completePendingToolCalls();\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onEdit: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...sliceMessagesUntil(current, message.parentId),\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      chatHelpers.setMessages((current) =>\n        sliceMessagesUntil(current, message.parentId),\n      );\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onDelete: async (messageId) => {\n      const threadMessages = runtimeRef.current.thread.getState().messages;\n      const messageIndex = threadMessages.findIndex(\n        (message) => message.id === messageId,\n      );\n      if (messageIndex === -1) return;\n\n      await deleteHistoryMessage(messageId);\n\n      const deleteIds = new Set(\n        getExternalStoreMessages<UI_MESSAGE>(threadMessages[messageIndex]!).map(\n          (message) => message.id,\n        ),\n      );\n      chatHelpers.setMessages((current) =>\n        current.filter((message) => !deleteIds.has(message.id)),\n      );\n    },\n    onReload: async (parentId: string | null, config) => {\n      lastRunConfigRef.current = config.runConfig;\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.regenerate({ metadata: config.runConfig });\n    },\n    onAddToolResult: ({\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      modelContent,\n    }) => {\n      const options = { metadata: lastRunConfigRef.current };\n      if (isError) {\n        return Promise.resolve(\n          chatHelpers.addToolOutput({\n            state: \"output-error\",\n            tool: toolName ?? toolCallId,\n            toolCallId,\n            errorText:\n              typeof result === \"string\" ? result : JSON.stringify(result),\n            options,\n          }),\n        );\n      } else {\n        const output =\n          modelContent !== undefined\n            ? wrapModelContentEnvelope(result, modelContent)\n            : result;\n        return Promise.resolve(\n          chatHelpers.addToolOutput({\n            tool: toolName,\n            toolCallId,\n            output,\n            options,\n          }),\n        );\n      }\n    },\n    onRespondToToolApproval: ({ approvalId, approved, reason }) =>\n      Promise.resolve(\n        chatHelpers.addToolApprovalResponse({\n          id: approvalId,\n          approved,\n          ...(reason != null && { reason }),\n          options: { metadata: lastRunConfigRef.current },\n        }),\n      ),\n    ...pickExternalStoreSharedOptions(adapter),\n    ...(suggestionAdapter ? { suggestions: generatedSuggestions } : {}),\n    ...(onResume && { onResume }),\n    ...(onResumeToolCall && { onResumeToolCall }),\n    ...(unstable_onBranchChange && { unstable_onBranchChange }),\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n      ...contextAdapters,\n      ...adapters,\n    },\n    extras,\n    isLoading,\n  });\n\n  const setMessagesRef = useRef(chatHelpers.setMessages);\n  setMessagesRef.current = chatHelpers.setMessages;\n\n  useEffect(() => {\n    if (hasSeededRepositoryRef.current) return;\n    if (!exportedMessageRepository) return;\n    if (chatHelpers.messages.length > 0) {\n      hasSeededRepositoryRef.current = true;\n      return;\n    }\n    const tempRepo = new MessageRepository();\n    tempRepo.import(exportedMessageRepository);\n    setMessagesRef.current(\n      tempRepo.getMessages().flatMap(getExternalStoreMessages<UI_MESSAGE>),\n    );\n    hasSeededRepositoryRef.current = true;\n  }, [exportedMessageRepository, chatHelpers.messages.length]);\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;;;;AAiEA,MAAM+C,eACJC,eACAC,kBAEC;CACC,GAAGD;CACHE,IAAIF,cAAcE,MAAMzC,WAAW;CACnC0C,MAAMH,cAAcG,QAAQF;AAC9B;AA2DF,MAAMc,oBAAiD,CAAA;AAEvD,MAAMC,2BACJC,mBACAC,UACAC,cACgC;CAChC,MAAM,CAACC,aAAaC,kBAClBlE,SAAsC4D,iBAAiB;CACzD,MAAMO,gBAAgBpE,OAA+B,IAAI;CACzD,MAAMsE,gBAAgBtE,OAAO,KAAK;CAClC,MAAMuE,cAAcvE,OAAOgE,QAAQ;CACnCO,YAAYC,UAAUR;CACtB,MAAMS,aAAazE,OAAO+D,iBAAiB;CAC3CU,WAAWD,UAAUT;CAGrBjE,gBAAgB;EACd,MAAM6E,yBAAyB;GAC7BP,cAAcI,SAASI,MAAM;GAC7BR,cAAcI,UAAU;GACxBL,gBAAgBU,SAAUA,KAAKC,WAAW,IAAID,OAAOhB,iBAAkB;EACzE;EAEA,MAAMkB,UAAUN,WAAWD;EAC3B,IAAI,CAACO,SAAS;GACZJ,iBAAiB;GACjBL,cAAcE,UAAUP;GACxB;EACF;EAEA,IAAIA,WAAW;GACb,IAAI,CAACK,cAAcE,SACjBG,iBAAiB;GAEnBL,cAAcE,UAAU;GACxB;EACF;EAEA,IAAI,CAACF,cAAcE,SAAS;EAC5BF,cAAcE,UAAU;EAExB,MAAMQ,kBAAkBT,YAAYC;EACpC,MAAMS,OAAOD,gBAAgBE,GAAG,EAAE;EAClC,IAAID,MAAMhC,SAAS,aAAa;EAChC,IAAIgC,KAAKE,QAAQC,SAAS,mBAAmB;EAE7C,MAAMC,aAAa,IAAIhB,gBAAgB;EACvCD,cAAcI,UAAUa;EACxB,MAAM,EAAEC,WAAWD;EAEnB,CAAM,YAAY;GAChB,IAAI;IACF,MAAME,qBAAqBR,QAAQS,SAAS;KAC1CxB,UAAUgB;KACVM;IACF,CAAC;IAED,MAAM3D,wBAAwB4D,oBAAoB;KAChDD;KACAG,UAAUtB;IACZ,CAAC;GACH,QAAQ,CAAC;EACX,EAAA,CAAG;CACL,GAAG,CAjDgBJ,qBAAqB,MAiDxBE,SAAS,CAAC;CAE1BnE,gBAAgB;EACd,aAAa;GACXsE,cAAcI,SAASI,MAAM;EAC/B;CACF,GAAG,CAAA,CAAE;CAEL,OAAOV;AACT;AAEA,MAAawB,mBACXC,aACAZ,UAA2C,CAAC,MACzC;CACH,MAAM,EACJ5B,UACApB,iBAAiB8D,uBACjBtC,+BAA+B,MAC/BC,UACAC,kBACAC,cACAC,mBACAC,4BACEmB;CACJ,MAAMhB,oBAAoBZ,UAAUG;CACpC,MAAMwC,kBAAkBrF,mBAAmB;CAC3C,MAAM,CAACsF,cAAcC,mBAAmB/F,SAEtC,CAAC,CAAC;CACJ,MAAMiG,2BAA2BlG,uBAC/B,IAAImG,IAAI,CACV;CACA,MAAMC,wBAAwBpG,uBAC5B,IAAImG,IAAI,CACV;CACA,MAAME,yBAAyBrG,uBAAoC,IAAImG,IAAI,CAAC;CAC5E,MAAMG,mBAAmBtG,OAA8BuG,KAAAA,CAAS;CAEhE,MAAMC,oBAAoBC,OAAOC,OAAOX,YAAY,CAAC,CAACY,MACnDC,MAAMA,GAAGxB,SAAS,WACrB;CACA,MAAMyB,oBACJlB,YAAYR,WAAW,eAAeQ,YAAYR,WAAW;CAC/D,MAAMlB,YAAY4C,qBAAqBL;CAEvC,MAAMM,gBAAgBtE,qBAAmBmD,YAAY3B,UAAUC,SAAS;CAIxE,MAAM8C,cAAcpB,YAAY3B,SAASkB,GAAG,EAAE;CAC9C,MAAM8B,sBACJ/C,aAAa8C,aAAa9D,SAAS,cAAc8D,YAAY/D,KAAKuD,KAAAA;CAEpE,MAAMvC,WAAW9B,sBAAsB+E,kBAAkB;EACvDhD;EACAD,UAAU2B,YAAY3B;EACtBN;EACAwD,UAAUnH,eACD;GACLgG;GACAe;GACAK,uBAAuBjB,yBAAyB1B;GAChD4C,oBAAoBhB,sBAAsB5B;GAC1C6C,qBAAqBhB,uBAAuB7B;GAC5C,GAAIwC,uBAAuB,EAAEA,oBAAoB;GACjD,GAAIrB,YAAY2B,SAAS,EAAEA,OAAO3B,YAAY2B,MAAM3E,QAAQ;EAC9D,IACA;GAACoD;GAAce;GAAeE;GAAqBrB,YAAY2B;EAAK,CACtE;CACF,CAAC;CAED,MAAMC,4BAA4BxH,cAAc;EAC9C,IAAI,CAAC4D,mBAAmB,OAAO4C,KAAAA;EAC/B,MAAMiB,YAAYjF,4BAChBL,sBAAsBuF,kBAGtB9D,iBACF;EACA,OAAO6D,UAAUxD,SAASc,SAAS,IAAI0C,YAAYjB,KAAAA;CACrD,GAAG,CAAC5C,iBAAiB,CAAC;CAEtB,MAAM+D,uBAAuB5D,wBAC3BC,mBACAC,UACAC,SACF;CAEA,MAAM,CAAC0D,cAAc1H,gBAAgB,EACnC,IAAIuE,UAA4B;EAC9B,OAAOoD;CACT,EACF,EAAE;CAEF,MAAM,EAAEC,WAAWC,eAAeC,yBAAyBzF,mBACzDqF,YACAxE,UAAUE,WAAWyC,iBAAiBzC,SACtCnB,sBAAsBuF,kBAGtBpF,uBAIC2B,eAAa;EACZ2B,YAAYqC,YAAYhE,UAAQ;CAClC,CACF;CAEA,MAAM,EACJhB,IAAIiF,QACJjE,UAAUkE,cACV/C,QAAQgD,YACRb,UACE3B;CACJ,MAAMyC,SAASrI,cAEX0C,YAAY4F,QAAQ;EAClBC,MAAM3C;EACN2B;CACF,CAAC,GAEH;EAACW;EAAQC;EAAcC;EAAYb;CAAK,CAC1C;CAEA,MAAMiB,2BAA2B,YAAY;EAC3C,IAAI,CAAChF,8BAA8B;EAOnCoC,YAAYqC,aAAahE,eAAa;GACpC,MAAM+C,gBAAc/C,WAASkB,GAAG,EAAE;GAClC,IAAI6B,eAAa9D,SAAS,aAAa,OAAOe;GAE9C,IAAIwE,aAAa;GACjB,MAAMC,QAAQ1B,cAAY0B,OAAOC,KAAKC,SAAS;IAC7C,IAAI,CAACrI,aAAaqI,IAAI,GAAG,OAAOA;IAChC,IACEA,KAAKC,UAAU,sBACfD,KAAKC,UAAU,kBACfD,KAAKC,UAAU,iBAEf,OAAOD;IAETH,aAAa;IACb,MAAM,EAAEK,UAAUC,WAAW,GAAGC,SAASJ;IACzC,OAAO;KACL,GAAGI;KACHH,OAAO;KACPK,WAAW;IACb;GACF,CAAC;GAED,IAAI,CAACT,YAAY,OAAOxE;GACxB,OAAO,CAAC,GAAGA,WAASkF,MAAM,GAAG,EAAE,GAAG;IAAE,GAAGnC;IAAa0B;GAAM,CAAC;EAC7D,CAAC;CACH;CAEA,MAAMU,yBAAyBnJ,OAAO,KAAK;CAC3C,MAAMoJ,uBACJ7B,6BAA6B,QAC7B,CAAC4B,uBAAuB3E,WACxBR,SAASc,WAAW;CAEtB,MAAM8C,UAAUpH,wBAAwB;EACtCyD,WAAW4C;EACX,GAAIuC,uBACA,EAAEzF,mBAAmB4D,0BAA0B,IAC/C,EAAEvD,SAAS;EACfqF,gCAAgC;EAChCrD;EACAgC,cAAchE,eACZ2B,YAAYqC,YACVhE,WACG0E,IAAIzG,mBAA+B,CAAC,CACpCqH,OAAOC,OAAO,CAAC,CACfC,KAAK,CACV;EACFC,WAAWzF,eACT2B,YAAYqC,YACVhE,WACG0E,IAAIzG,mBAA+B,CAAC,CACpCqH,OAAOC,OAAO,CAAC,CACfC,KAAK,CACV;EACFE,6BAAkE;GAChE,MAAMC,WAAWhC,WAAWnD,QAAQoF,OAAOC,OAAO;GAElD,MAAMC,mBAAoD,CAAA;GAC1D,MAAMC,iCAAiB,IAAI5D,IAAoB;GAE/C,KAAK,MAAM6D,QAAQL,SAAS3F,UAAU;IACpC,MAAMiG,gBAAgBxI,yBACpBuI,KAAKrH,OACP;IACA,IAAIuH,WACFF,KAAKE,YAAY,OACZH,eAAeI,IAAIH,KAAKE,QAAQ,KAAKF,KAAKE,WAC3C;IACN,KAAK,MAAME,gBAAgBH,eAAe;KACxCH,iBAAiBO,KAAK;MAAEH;MAAUvH,SAASyH;KAAa,CAAC;KACzDF,WAAW7H,qBAAqBiI,MAAMF,YAAyB;IACjE;IACA,IAAIH,cAAcnF,SAAS,GACzBiF,eAAeQ,IACbP,KAAKrH,QAAQK,IACbX,qBAAqBiI,MACnBL,cAAcA,cAAcnF,SAAS,EACvC,CACF;GAEJ;GAEA,MAAM0F,SAA8C,EAClDxG,UAAU8F,iBACZ;GAEA,IAAIH,SAASc,UAAU,MACrBD,OAAOC,SAASV,eAAeI,IAAIR,SAASc,MAAM,KAAKd,SAASc;GAGlE,OAAOD;EACT;EACAE,sBAAsBC,SAA8C;GAElE,MAAMC,eAAerI,4BACnBL,sBAAsBuF,kBACtBkD,IACF;GAGAhD,WAAWnD,QAAQoF,OAAOiB,OAAOD,YAAY;EAC/C;EACAE,UAAU,YAAY;GACpB,IAAI;IACF,MAAMnF,YAAYoF,KAAK;GACzB,SAASzD,SAAO;IACd,IAAI,EAAEA,mBAAiB0D,SAAS1D,QAAM2D,SAAS,eAC7C,MAAM3D;GAEV;EACF;EACA4D,OAAO,OAAOvI,YAAY;GACxB,MAAMG,iBACJ+C,yBAAyB9D,gBAAAA,CACbY,OAAO;GAErB,IAAI,EAAEA,QAAQwI,YAAYxI,QAAQM,SAAS,SAAS;IAClD0C,YAAYqC,aAAaxD,YAAY,CACnC,GAAGA,SACH3B,YAAwBC,eAAeH,QAAQM,IAAI,CAAC,CACrD;IACD;GACF;GAEAqD,iBAAiB9B,UAAU7B,QAAQyI;GACnC,MAAM7C,yBAAyB;GAC/B,MAAM5C,YAAY0F,YAAYvI,eAAe,EAC3CoE,UAAUvE,QAAQyI,UACpB,CAAC;EACH;EACAE,QAAQ,OAAO3I,cAAY;GACzB,MAAMG,mBACJ+C,yBAAyB9D,gBAAAA,CACbY,SAAO;GAErB,IAAI,EAAEA,UAAQwI,YAAYxI,UAAQM,SAAS,SAAS;IAClD0C,YAAYqC,aAAaxD,cAAY,CACnC,GAAG1C,mBAAmB0C,WAAS7B,UAAQuH,QAAQ,GAC/CrH,YAAwBC,iBAAeH,UAAQM,IAAI,CAAC,CACrD;IACD;GACF;GAEAqD,iBAAiB9B,UAAU7B,UAAQyI;GACnCzF,YAAYqC,aAAaxD,cACvB1C,mBAAmB0C,WAAS7B,UAAQuH,QAAQ,CAC9C;GACA,MAAMvE,YAAY0F,YAAYvI,iBAAe,EAC3CoE,UAAUvE,UAAQyI,UACpB,CAAC;EACH;EACAG,UAAU,OAAOC,cAAc;GAC7B,MAAMC,iBAAiB9D,WAAWnD,QAAQoF,OAAO8B,SAAS,CAAC,CAAC1H;GAC5D,MAAM2H,eAAeF,eAAeG,WACjCjJ,cAAYA,UAAQK,OAAOwI,SAC9B;GACA,IAAIG,iBAAiB,IAAI;GAEzB,MAAM5D,qBAAqByD,SAAS;GAEpC,MAAMK,YAAY,IAAIC,IACpBrK,yBAAqCgK,eAAeE,aAAc,CAAC,CAACjD,KACjE/F,cAAYA,UAAQK,EACvB,CACF;GACA2C,YAAYqC,aAAaxD,cACvBA,UAAQ8E,QAAQ3G,cAAY,CAACkJ,UAAUE,IAAIpJ,UAAQK,EAAE,CAAC,CACxD;EACF;EACAgJ,UAAU,OAAO9B,YAAyB+B,WAAW;GACnD3F,iBAAiB9B,UAAUyH,OAAOb;GAClC,MAAMc,cAAcpK,mBAAmB6D,YAAY3B,UAAUkG,UAAQ;GACrEvE,YAAYqC,YAAYkE,WAAW;GAEnC,MAAMvG,YAAYwG,WAAW,EAAEjF,UAAU+E,OAAOb,UAAU,CAAC;EAC7D;EACAgB,kBAAkB,EAChBC,YACAC,UACA9B,QAAAA,UACA+B,SACAC,mBACI;GACJ,MAAMC,UAAU,EAAEvF,UAAUZ,iBAAiB9B,QAAQ;GACrD,IAAI+H,SACF,OAAOG,QAAQC,QACbhH,YAAYiH,cAAc;IACxBhE,OAAO;IACPiE,MAAMP,YAAYD;IAClBA;IACApD,WACE,OAAOuB,aAAW,WAAWA,WAASsC,KAAKC,UAAUvC,QAAM;IAC7DiC;GACF,CAAC,CACH;QACK;IACL,MAAMO,SACJR,iBAAiBjG,KAAAA,IACbpE,yBAAyBqI,UAAQgC,YAAY,IAC7ChC;IACN,OAAOkC,QAAQC,QACbhH,YAAYiH,cAAc;KACxBC,MAAMP;KACND;KACAW;KACAP;IACF,CAAC,CACH;GACF;EACF;EACAQ,0BAA0B,EAAEC,YAAYC,UAAUC,aAChDV,QAAQC,QACNhH,YAAY0H,wBAAwB;GAClCrK,IAAIkK;GACJC;GACA,GAAIC,UAAU,QAAQ,EAAEA,OAAO;GAC/BX,SAAS,EAAEvF,UAAUZ,iBAAiB9B,QAAQ;EAChD,CAAC,CACH;EACF,GAAG9C,+BAA+BqD,OAAO;EACzC,GAAIhB,oBAAoB,EAAEG,aAAawD,qBAAqB,IAAI,CAAC;EACjE,GAAIlE,YAAY,EAAEA,SAAS;EAC3B,GAAIC,oBAAoB,EAAEA,iBAAiB;EAC3C,GAAIG,2BAA2B,EAAEA,wBAAwB;EACzDT,UAAU;GACRmK,aAAatL;GACb,GAAG8D;GACH,GAAG3C;EACL;EACAiF;EACAP;CACF,CAAC;CAED,MAAM0F,iBAAiBvN,OAAO2F,YAAYqC,WAAW;CACrDuF,eAAe/I,UAAUmB,YAAYqC;CAErClI,gBAAgB;EACd,IAAIqJ,uBAAuB3E,SAAS;EACpC,IAAI,CAAC+C,2BAA2B;EAChC,IAAI5B,YAAY3B,SAASc,SAAS,GAAG;GACnCqE,uBAAuB3E,UAAU;GACjC;EACF;EACA,MAAMgJ,WAAW,IAAI5L,kBAAkB;EACvC4L,SAAS3C,OAAOtD,yBAAyB;EACzCgG,eAAe/I,QACbgJ,SAASC,YAAY,CAAC,CAACC,QAAQjM,wBAAoC,CACrE;EACA0H,uBAAuB3E,UAAU;CACnC,GAAG,CAAC+C,2BAA2B5B,YAAY3B,SAASc,MAAM,CAAC;CAC3D,OAAO8C;AACT"}
+{"version":3,"file":"useAISDKRuntime.js","names":["c","_c","useCallback","useEffect","useInsertionEffect","useMemo","useRef","useState","UIMessage","useChat","CreateUIMessage","UseChatHelpers","isToolUIPart","generateId","useExternalStoreRuntime","useRuntimeAdapters","JoinStrategy","SuggestionAdapter","ThreadSuggestion","ToolExecutionStatus","ExternalStoreAdapter","ExternalStoreSharedOptions","ThreadHistoryAdapter","AssistantRuntime","ThreadMessage","MessageFormatAdapter","MessageFormatItem","MessageFormatRepository","AppendMessage","RunConfig","McpAppMetadata","getExternalStoreMessages","pickExternalStoreSharedOptions","consumeSuggestionResult","MessageRepository","ReadonlyJSONObject","sliceMessagesUntil","toCreateMessage","vercelAttachmentAdapter","getVercelAIMessages","AISDKMessageConverter","wrapModelContentEnvelope","AISDKStorageFormat","aiSDKV6FormatAdapter","useExternalHistory","toExportedMessageRepository","useStreamingTiming","aiSDKExtras","CustomToCreateMessageFunction","message","UI_MESSAGE","toUIMessage","createMessage","fallbackRole","id","role","AISDKRuntimeAdapter","adapters","NonNullable","history","suggestion","unstable_messageRepositoryInstance","cancelPendingToolCallsOnSend","onResume","onResumeToolCall","joinStrategy","messageRepository","unstable_onBranchChange","EMPTY_SUGGESTIONS","useGeneratedSuggestions","suggestionAdapter","messages","isRunning","$","suggestions","setSuggestions","controllerRef","wasRunningRef","messagesRef","t0","t1","current","adapterRef","t2","t3","hasAdapter","t4","clearSuggestions","abort","_temp","adapter","currentMessages","last","at","status","type","controller","AbortController","signal","promiseOrGenerator","generate","onUpdate","t5","t6","t7","Symbol","for","NO_CANCELLED_MESSAGE_IDS","ReadonlySet","Set","useAISDKRuntime","chatHelpers","ReturnType","customToCreateMessage","contextAdapters","toolStatuses","setToolStatuses","Record","cancelledMessages","setCancelledMessages","chatId","ids","toolArgsKeyOrderCacheRef","Map","toolLastInputCacheRef","mcpAppMetadataCacheRef","lastRunConfigRef","undefined","hasExecutingTools","Object","values","some","s","providerIsRunning","wasProviderRunningRef","messageTiming","lastMessage","optimisticMessageId","cancelledMessageIds","retractCancellation","messageId","prev","has","delete","resumedMessageId","wasProviderRunning","useThreadMessages","metadata","toolArgsKeyOrderCache","toolLastInputCache","mcpAppMetadataCache","error","size","exportedMessageRepository","converted","toThreadMessages","length","generatedSuggestions","runtimeRef","runtime","isLoading","deleteMessage","deleteHistoryMessage","setMessages","chatMessages","chatStatus","extras","provide","chat","completePendingToolCalls","hasChanges","parts","map","part","state","approval","_approval","rest","const","errorText","slice","hasSeededRepositoryRef","shouldFeedRepository","unstable_enableToolInvocations","filter","Boolean","flat","onImport","onExportExternalState","exported","thread","export","expandedMessages","lastInnerIdMap","item","innerMessages","parentId","get","innerMessage","push","getId","set","result","headId","onLoadExternalState","repo","exportedRepo","import","onCancel","cancelledId","liveIds","m","kept","stop","Error","name","onNew","startRun","runConfig","sendMessage","onEdit","onDelete","threadMessages","getState","messageIndex","findIndex","deleteIds","onReload","config","newMessages","regenerate","onAddToolResult","toolCallId","toolName","isError","modelContent","options","Promise","resolve","addToolOutput","tool","JSON","stringify","output","onRespondToToolApproval","approvalId","approved","reason","addToolApprovalResponse","attachments","setMessagesRef","tempRepo","getMessages","flatMap"],"sources":["../../src/runtime/useAISDKRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  useCallback,\n  useEffect,\n  useInsertionEffect,\n  useMemo,\n  useRef,\n  useState,\n} from \"react\";\nimport type {\n  UIMessage,\n  useChat,\n  CreateUIMessage,\n  UseChatHelpers,\n} from \"@ai-sdk/react\";\nimport { isToolUIPart, generateId } from \"ai\";\nimport {\n  useExternalStoreRuntime,\n  useRuntimeAdapters,\n  type JoinStrategy,\n} from \"@assistant-ui/core/react\";\nimport type {\n  SuggestionAdapter,\n  ThreadSuggestion,\n  ToolExecutionStatus,\n} from \"@assistant-ui/core\";\nimport type {\n  ExternalStoreAdapter,\n  ExternalStoreSharedOptions,\n  ThreadHistoryAdapter,\n  AssistantRuntime,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatItem,\n  MessageFormatRepository,\n  AppendMessage,\n  RunConfig,\n  McpAppMetadata,\n} from \"@assistant-ui/core\";\nimport {\n  getExternalStoreMessages,\n  pickExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport {\n  consumeSuggestionResult,\n  MessageRepository,\n} from \"@assistant-ui/core/internal\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../converters/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../adapters/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../utils/getVercelAIMessages\";\nimport { AISDKMessageConverter } from \"../converters/convertMessage\";\nimport { wrapModelContentEnvelope } from \"../converters/modelContentEnvelope\";\nimport {\n  type AISDKStorageFormat,\n  aiSDKV6FormatAdapter,\n} from \"../adapters/aiSDKFormatAdapter\";\nimport {\n  useExternalHistory,\n  toExportedMessageRepository,\n} from \"./useExternalHistory\";\nimport { useStreamingTiming } from \"./useStreamingTiming\";\nimport { aiSDKExtras } from \"../aiSDKExtras\";\n\nexport type CustomToCreateMessageFunction = <\n  UI_MESSAGE extends UIMessage = UIMessage,\n>(\n  message: AppendMessage,\n) => CreateUIMessage<UI_MESSAGE>;\n\nconst toUIMessage = <UI_MESSAGE extends UIMessage>(\n  createMessage: CreateUIMessage<UI_MESSAGE>,\n  fallbackRole: UI_MESSAGE[\"role\"],\n): UI_MESSAGE =>\n  ({\n    ...createMessage,\n    id: createMessage.id ?? generateId(),\n    role: createMessage.role ?? fallbackRole,\n  }) as UI_MESSAGE;\n\nexport type AISDKRuntimeAdapter<UI_MESSAGE extends UIMessage = UIMessage> =\n  ExternalStoreSharedOptions & {\n    adapters?:\n      | (NonNullable<ExternalStoreAdapter[\"adapters\"]> & {\n          history?: ThreadHistoryAdapter | undefined;\n          suggestion?: SuggestionAdapter | undefined;\n        })\n      | undefined;\n    toCreateMessage?: CustomToCreateMessageFunction;\n    unstable_messageRepositoryInstance?: MessageRepository | undefined;\n    /**\n     * Whether to automatically cancel pending interactive tool calls when the user sends a new message.\n     *\n     * When enabled (default), the pending tool calls will be marked as failed with an error message\n     * indicating the user cancelled the tool call by sending a new message.\n     *\n     * @default true\n     */\n    cancelPendingToolCallsOnSend?: boolean | undefined;\n    /**\n     * Called when `runtime.thread.resumeRun(config)` is invoked.\n     *\n     * When omitted, `resumeRun` throws `\"Runtime does not support resuming runs.\"`.\n     * Provide this to bridge resume invocations into a custom replay channel\n     * (for example, an SSE reconnect endpoint keyed by turn id).\n     */\n    onResume?: ExternalStoreAdapter[\"onResume\"];\n    /**\n     * Called when `runtime.thread.resumeToolCall(options)` is invoked for a tool call the in-process tracker does not own.\n     *\n     * When omitted, `resumeToolCall` throws `\"Tool call ${toolCallId} is not waiting for resume.\"`.\n     * Provide this to bridge resume-tool-call invocations into a custom handler.\n     */\n    onResumeToolCall?: ExternalStoreAdapter[\"onResumeToolCall\"];\n    /**\n     * How consecutive assistant messages are rendered.\n     *\n     * `\"concat-content\"` (the default) merges them into a single thread message.\n     * `\"none\"` keeps each assistant message as its own thread message, which is\n     * useful when a backend persists proactive or consecutive assistant messages\n     * as separate entries.\n     */\n    joinStrategy?: JoinStrategy | undefined;\n    /**\n     * A branch-aware AI SDK message tree seeded once when `useChat` is empty.\n     * After that seed, live updates come only from `useChat`. A later empty\n     * chat or a new object identity does not reload the tree.\n     */\n    messageRepository?: MessageFormatRepository<UI_MESSAGE>;\n    /**\n     * Called after an explicit `switchToBranch` (for example a BranchPicker\n     * click). Complements `setMessages` and does not enable switching by itself.\n     *\n     * @deprecated This API is still under active development and might change without notice.\n     */\n    unstable_onBranchChange?: ExternalStoreAdapter[\"unstable_onBranchChange\"];\n  };\n\nconst EMPTY_SUGGESTIONS: readonly ThreadSuggestion[] = [];\n\nconst useGeneratedSuggestions = (\n  suggestionAdapter: SuggestionAdapter | undefined,\n  messages: readonly ThreadMessage[],\n  isRunning: boolean,\n): readonly ThreadSuggestion[] => {\n  const [suggestions, setSuggestions] =\n    useState<readonly ThreadSuggestion[]>(EMPTY_SUGGESTIONS);\n  const controllerRef = useRef<AbortController | null>(null);\n  const wasRunningRef = useRef(false);\n  const messagesRef = useRef(messages);\n  useInsertionEffect(() => {\n    messagesRef.current = messages;\n  }, [messages]);\n  const adapterRef = useRef(suggestionAdapter);\n  useInsertionEffect(() => {\n    adapterRef.current = suggestionAdapter;\n  }, [suggestionAdapter]);\n  const hasAdapter = suggestionAdapter != null;\n\n  useEffect(() => {\n    const clearSuggestions = () => {\n      controllerRef.current?.abort();\n      controllerRef.current = null;\n      setSuggestions((prev) => (prev.length === 0 ? prev : EMPTY_SUGGESTIONS));\n    };\n\n    const adapter = adapterRef.current;\n    if (!adapter) {\n      clearSuggestions();\n      wasRunningRef.current = isRunning;\n      return;\n    }\n\n    if (isRunning) {\n      if (!wasRunningRef.current) {\n        clearSuggestions();\n      }\n      wasRunningRef.current = true;\n      return;\n    }\n\n    if (!wasRunningRef.current) return;\n    wasRunningRef.current = false;\n\n    const currentMessages = messagesRef.current;\n    const last = currentMessages.at(-1);\n    if (last?.role !== \"assistant\") return;\n    if (last.status?.type === \"requires-action\") return;\n\n    const controller = new AbortController();\n    controllerRef.current = controller;\n    const { signal } = controller;\n\n    void (async () => {\n      try {\n        const promiseOrGenerator = adapter.generate({\n          messages: currentMessages,\n          signal,\n        });\n\n        await consumeSuggestionResult(promiseOrGenerator, {\n          signal,\n          onUpdate: setSuggestions,\n        });\n      } catch {}\n    })();\n  }, [hasAdapter, isRunning]);\n\n  useEffect(() => {\n    return () => {\n      controllerRef.current?.abort();\n    };\n  }, []);\n\n  return suggestions;\n};\n\nconst NO_CANCELLED_MESSAGE_IDS: ReadonlySet<string> = new Set();\n\nexport const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>,\n  adapter: AISDKRuntimeAdapter<UI_MESSAGE> = {},\n) => {\n  const {\n    adapters,\n    toCreateMessage: customToCreateMessage,\n    cancelPendingToolCallsOnSend = true,\n    onResume,\n    onResumeToolCall,\n    joinStrategy,\n    messageRepository,\n    unstable_onBranchChange,\n  } = adapter;\n  const suggestionAdapter = adapters?.suggestion;\n  const contextAdapters = useRuntimeAdapters();\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const [cancelledMessages, setCancelledMessages] = useState<{\n    chatId: string;\n    ids: ReadonlySet<string>;\n  } | null>(null);\n  const toolArgsKeyOrderCacheRef = useRef<Map<string, Map<string, string[]>>>(\n    new Map(),\n  );\n  const toolLastInputCacheRef = useRef<Map<string, ReadonlyJSONObject>>(\n    new Map(),\n  );\n  const mcpAppMetadataCacheRef = useRef<Map<string, McpAppMetadata>>(new Map());\n  const lastRunConfigRef = useRef<RunConfig | undefined>(undefined);\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const providerIsRunning =\n    chatHelpers.status === \"submitted\" || chatHelpers.status === \"streaming\";\n  const isRunning = providerIsRunning || hasExecutingTools;\n  const wasProviderRunningRef = useRef(providerIsRunning);\n\n  const messageTiming = useStreamingTiming(chatHelpers.messages, isRunning);\n\n  // Flag the streaming message optimistic: its id can be swapped for a server\n  // id mid-run, and the repository then drops the orphaned pre-swap id (#4037).\n  const lastMessage = chatHelpers.messages.at(-1);\n  const optimisticMessageId =\n    isRunning && lastMessage?.role === \"assistant\" ? lastMessage.id : undefined;\n\n  const cancelledMessageIds =\n    cancelledMessages?.chatId === chatHelpers.id\n      ? cancelledMessages.ids\n      : NO_CANCELLED_MESSAGE_IDS;\n\n  const retractCancellation = useCallback(\n    (chatId: string, messageId: string) => {\n      setCancelledMessages((prev) => {\n        if (prev?.chatId !== chatId || !prev.ids.has(messageId)) return prev;\n        const ids = new Set(prev.ids);\n        ids.delete(messageId);\n        return { chatId, ids };\n      });\n    },\n    [],\n  );\n\n  // A provider run that resumes the stopped response retracts its cancellation;\n  // a run that starts a new response leaves the stopped one marked.\n  const resumedMessageId =\n    providerIsRunning && lastMessage?.role === \"assistant\"\n      ? lastMessage.id\n      : undefined;\n\n  useEffect(() => {\n    const wasProviderRunning = wasProviderRunningRef.current;\n    wasProviderRunningRef.current = providerIsRunning;\n    if (wasProviderRunning || !resumedMessageId) return;\n    retractCancellation(chatHelpers.id, resumedMessageId);\n  }, [\n    providerIsRunning,\n    resumedMessageId,\n    chatHelpers.id,\n    retractCancellation,\n  ]);\n\n  const messages = AISDKMessageConverter.useThreadMessages({\n    isRunning,\n    messages: chatHelpers.messages,\n    joinStrategy,\n    metadata: useMemo(\n      () => ({\n        toolStatuses,\n        messageTiming,\n        toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,\n        toolLastInputCache: toolLastInputCacheRef.current,\n        mcpAppMetadataCache: mcpAppMetadataCacheRef.current,\n        ...(optimisticMessageId && { optimisticMessageId }),\n        ...(chatHelpers.error && { error: chatHelpers.error.message }),\n        ...(cancelledMessageIds.size > 0 && { cancelledMessageIds }),\n      }),\n      [\n        toolStatuses,\n        messageTiming,\n        optimisticMessageId,\n        chatHelpers.error,\n        cancelledMessageIds,\n      ],\n    ),\n  });\n\n  const exportedMessageRepository = useMemo(() => {\n    if (!messageRepository) return undefined;\n    const converted = toExportedMessageRepository(\n      AISDKMessageConverter.toThreadMessages as (\n        messages: UI_MESSAGE[],\n      ) => ThreadMessage[],\n      messageRepository,\n    );\n    return converted.messages.length > 0 ? converted : undefined;\n  }, [messageRepository]);\n\n  const generatedSuggestions = useGeneratedSuggestions(\n    suggestionAdapter,\n    messages,\n    isRunning,\n  );\n\n  const [runtimeRef] = useState(() => ({\n    get current(): AssistantRuntime {\n      return runtime;\n    },\n  }));\n\n  const { isLoading, deleteMessage: deleteHistoryMessage } = useExternalHistory(\n    runtimeRef,\n    adapters?.history ?? contextAdapters?.history,\n    AISDKMessageConverter.toThreadMessages as (\n      messages: UI_MESSAGE[],\n    ) => ThreadMessage[],\n    aiSDKV6FormatAdapter as MessageFormatAdapter<\n      UI_MESSAGE,\n      AISDKStorageFormat\n    >,\n    (messages) => {\n      chatHelpers.setMessages(messages);\n    },\n  );\n\n  const {\n    id: chatId,\n    messages: chatMessages,\n    status: chatStatus,\n    error,\n  } = chatHelpers;\n  const extras = useMemo(\n    () =>\n      aiSDKExtras.provide({\n        chat: chatHelpers as unknown as UseChatHelpers<UIMessage>,\n        error,\n      }),\n    // oxlint-disable-next-line react/exhaustive-deps -- keyed on the chat's identity and reactive snapshots; useChat re-mints the helpers object every render while its remaining fields are instance-bound methods, and a render-stable extras identity is what lets the external-store core dedupe adapter updates\n    [chatId, chatMessages, chatStatus, error],\n  );\n\n  const completePendingToolCalls = async () => {\n    if (!cancelPendingToolCallsOnSend) return;\n\n    // The runtime auto-aborts in-flight tool invocations when a new run\n    // is dispatched (append() / startRun()). All we need to do here is\n    // mark any tool without a result as cancelled in the UI message list.\n\n    // Mark any tool without a result as cancelled (uses setMessages to avoid triggering sendAutomaticallyWhen)\n    chatHelpers.setMessages((messages) => {\n      const lastMessage = messages.at(-1);\n      if (lastMessage?.role !== \"assistant\") return messages;\n\n      let hasChanges = false;\n      const parts = lastMessage.parts?.map((part) => {\n        if (!isToolUIPart(part)) return part;\n        if (\n          part.state === \"output-available\" ||\n          part.state === \"output-error\" ||\n          part.state === \"output-denied\"\n        )\n          return part;\n\n        hasChanges = true;\n        const { approval: _approval, ...rest } = part;\n        return {\n          ...rest,\n          state: \"output-error\" as const,\n          errorText: \"User cancelled tool call by sending a new message.\",\n        };\n      });\n\n      if (!hasChanges) return messages;\n      return [...messages.slice(0, -1), { ...lastMessage, parts }];\n    });\n  };\n\n  const hasSeededRepositoryRef = useRef(false);\n  const shouldFeedRepository =\n    exportedMessageRepository != null &&\n    !hasSeededRepositoryRef.current &&\n    messages.length === 0;\n\n  const runtime = useExternalStoreRuntime({\n    isRunning: providerIsRunning,\n    ...(shouldFeedRepository\n      ? { messageRepository: exportedMessageRepository }\n      : { messages }),\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onImport: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onExportExternalState: (): MessageFormatRepository<UI_MESSAGE> => {\n      const exported = runtimeRef.current.thread.export();\n\n      const expandedMessages: MessageFormatItem<UI_MESSAGE>[] = [];\n      const lastInnerIdMap = new Map<string, string>();\n\n      for (const item of exported.messages) {\n        const innerMessages = getExternalStoreMessages<UI_MESSAGE>(\n          item.message,\n        );\n        let parentId =\n          item.parentId != null\n            ? (lastInnerIdMap.get(item.parentId) ?? item.parentId)\n            : null;\n        for (const innerMessage of innerMessages) {\n          expandedMessages.push({ parentId, message: innerMessage });\n          parentId = aiSDKV6FormatAdapter.getId(innerMessage as UIMessage);\n        }\n        if (innerMessages.length > 0) {\n          lastInnerIdMap.set(\n            item.message.id,\n            aiSDKV6FormatAdapter.getId(\n              innerMessages[innerMessages.length - 1]! as UIMessage,\n            ),\n          );\n        }\n      }\n\n      const result: MessageFormatRepository<UI_MESSAGE> = {\n        messages: expandedMessages,\n      };\n\n      if (exported.headId != null) {\n        result.headId = lastInnerIdMap.get(exported.headId) ?? exported.headId;\n      }\n\n      return result;\n    },\n    onLoadExternalState: (repo: MessageFormatRepository<UI_MESSAGE>) => {\n      // Convert MessageFormatRepository to ExportedMessageRepository\n      const exportedRepo = toExportedMessageRepository(\n        AISDKMessageConverter.toThreadMessages,\n        repo,\n      );\n\n      // Import into the thread's MessageRepository\n      runtimeRef.current.thread.import(exportedRepo);\n    },\n    onCancel: async () => {\n      const message = chatHelpers.messages.at(-1);\n      const cancelledId =\n        isRunning && message?.role === \"assistant\" ? message.id : undefined;\n      if (cancelledId) {\n        const liveIds = new Set(chatHelpers.messages.map((m) => m.id));\n        setCancelledMessages((prev) => {\n          const kept =\n            prev?.chatId === chatHelpers.id\n              ? [...prev.ids].filter((id) => liveIds.has(id))\n              : [];\n          return {\n            chatId: chatHelpers.id,\n            ids: new Set([...kept, cancelledId]),\n          };\n        });\n      }\n      try {\n        await chatHelpers.stop();\n      } catch (error) {\n        if (!(error instanceof Error && error.name === \"AbortError\")) {\n          if (cancelledId) retractCancellation(chatHelpers.id, cancelledId);\n          throw error;\n        }\n      }\n    },\n    onNew: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...current,\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      await completePendingToolCalls();\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onEdit: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...sliceMessagesUntil(current, message.parentId),\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      chatHelpers.setMessages((current) =>\n        sliceMessagesUntil(current, message.parentId),\n      );\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onDelete: async (messageId) => {\n      const threadMessages = runtimeRef.current.thread.getState().messages;\n      const messageIndex = threadMessages.findIndex(\n        (message) => message.id === messageId,\n      );\n      if (messageIndex === -1) return;\n\n      await deleteHistoryMessage(messageId);\n\n      const deleteIds = new Set(\n        getExternalStoreMessages<UI_MESSAGE>(threadMessages[messageIndex]!).map(\n          (message) => message.id,\n        ),\n      );\n      chatHelpers.setMessages((current) =>\n        current.filter((message) => !deleteIds.has(message.id)),\n      );\n    },\n    onReload: async (parentId: string | null, config) => {\n      lastRunConfigRef.current = config.runConfig;\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.regenerate({ metadata: config.runConfig });\n    },\n    onAddToolResult: ({\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      modelContent,\n    }) => {\n      const options = { metadata: lastRunConfigRef.current };\n      if (isError) {\n        return Promise.resolve(\n          chatHelpers.addToolOutput({\n            state: \"output-error\",\n            tool: toolName ?? toolCallId,\n            toolCallId,\n            errorText:\n              typeof result === \"string\" ? result : JSON.stringify(result),\n            options,\n          }),\n        );\n      } else {\n        const output =\n          modelContent !== undefined\n            ? wrapModelContentEnvelope(result, modelContent)\n            : result;\n        return Promise.resolve(\n          chatHelpers.addToolOutput({\n            tool: toolName,\n            toolCallId,\n            output,\n            options,\n          }),\n        );\n      }\n    },\n    onRespondToToolApproval: ({ approvalId, approved, reason }) =>\n      Promise.resolve(\n        chatHelpers.addToolApprovalResponse({\n          id: approvalId,\n          approved,\n          ...(reason != null && { reason }),\n          options: { metadata: lastRunConfigRef.current },\n        }),\n      ),\n    ...pickExternalStoreSharedOptions(adapter),\n    ...(adapter.unstable_messageRepositoryInstance && {\n      unstable_messageRepositoryInstance:\n        adapter.unstable_messageRepositoryInstance,\n    }),\n    ...(suggestionAdapter ? { suggestions: generatedSuggestions } : {}),\n    ...(onResume && { onResume }),\n    ...(onResumeToolCall && { onResumeToolCall }),\n    ...(unstable_onBranchChange && { unstable_onBranchChange }),\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n      ...contextAdapters,\n      ...adapters,\n    },\n    extras,\n    isLoading,\n  });\n\n  const setMessagesRef = useRef(chatHelpers.setMessages);\n  useInsertionEffect(() => {\n    setMessagesRef.current = chatHelpers.setMessages;\n  }, [chatHelpers.setMessages]);\n\n  useEffect(() => {\n    if (hasSeededRepositoryRef.current) return;\n    if (!exportedMessageRepository) return;\n    if (chatHelpers.messages.length > 0) {\n      hasSeededRepositoryRef.current = true;\n      return;\n    }\n    const tempRepo = new MessageRepository();\n    tempRepo.import(exportedMessageRepository);\n    setMessagesRef.current(\n      tempRepo.getMessages().flatMap(getExternalStoreMessages<UI_MESSAGE>),\n    );\n    hasSeededRepositoryRef.current = true;\n  }, [exportedMessageRepository, chatHelpers.messages.length]);\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;AAwEA,MAAMmD,eACJC,eACAC,kBAEC;CACC,GAAGD;CACHE,IAAIF,cAAcE,MAAMzC,WAAW;CACnC0C,MAAMH,cAAcG,QAAQF;AAC9B;AA4DF,MAAMe,oBAAiD,CAAA;AAEvD,MAAMC,2BAA0BC,mBAAAC,UAAAC,cAAA;CAAA,MAAAC,IAAAxE,EAAA,EAAA;CAK9B,MAAA,CAAAyE,aAAAC,kBACEpE,SAAsC6D,iBAAiB;CACzD,MAAAQ,gBAAsBtE,OAA+B,IAAI;CACzD,MAAAuE,gBAAsBvE,OAAO,KAAK;CAClC,MAAAwE,cAAoBxE,OAAOiE,QAAQ;CAAE,IAAAQ;CAAA,IAAAC;CAAA,IAAAP,EAAA,OAAAF,UAAA;EAClBQ,WAAA;GACjBD,YAAWG,UAAWV;EAAH;EAClBS,KAAA,CAACT,QAAQ;EAACE,EAAA,KAAAF;EAAAE,EAAA,KAAAM;EAAAN,EAAA,KAAAO;CAAA,OAAA;EAAAD,KAAAN,EAAA;EAAAO,KAAAP,EAAA;CAAA;CAFbrE,mBAAmB2E,IAEhBC,EAAU;CACb,MAAAE,aAAmB5E,OAAOgE,iBAAiB;CAAE,IAAAa;CAAA,IAAAC;CAAA,IAAAX,EAAA,OAAAH,mBAAA;EAC1Ba,WAAA;GACjBD,WAAUD,UAAWX;EAAH;EACjBc,KAAA,CAACd,iBAAiB;EAACG,EAAA,KAAAH;EAAAG,EAAA,KAAAU;EAAAV,EAAA,KAAAW;CAAA,OAAA;EAAAD,KAAAV,EAAA;EAAAW,KAAAX,EAAA;CAAA;CAFtBrE,mBAAmB+E,IAEhBC,EAAmB;CACtB,MAAAC,aAAmBf,qBAAqB;CAAK,IAAAgB;CAAA,IAAAb,EAAA,OAAAD,WAAA;EAEnCc,WAAA;GACR,MAAAC,yBAAyB;IACvBX,cAAaK,SAAeO,MAAC;IAC7BZ,cAAaK,UAAW;IACxBN,eAAec,KAAwD;GAAC;GAG1E,MAAAC,UAAgBR,WAAUD;GAC1B,IAAI,CAACS,SAAO;IACVH,iBAAiB;IACjBV,cAAaI,UAAWT;IAAH;GAAA;GAIvB,IAAIA,WAAS;IACX,IAAI,CAACK,cAAaI,SAChBM,iBAAiB;IAEnBV,cAAaI,UAAW;IAAH;GAAA;GAIvB,IAAI,CAACJ,cAAaI,SAAQ;GAC1BJ,cAAaI,UAAW;GAExB,MAAAU,kBAAwBb,YAAWG;GACnC,MAAAW,OAAaD,gBAAeE,GAAI,EAAE;GAClC,IAAID,MAAIrC,SAAW,aAAW;GAC9B,IAAIqC,KAAIE,QAAaC,SAAK,mBAAiB;GAE3C,MAAAC,aAAmB,IAAIC,gBAAgB;GACvCrB,cAAaK,UAAWe;GACxB,MAAA,EAAAE,WAAmBF;GAEd,CAAC,YAAA;IACJ,IAAA;KACE,MAAAG,qBAA2BT,QAAOU,SAAU;MAAA7B,UAChCoB;MAAeO;KAE3B,CAAC;KAED,MAAMjE,wBAAwBkE,oBAAoB;MAAAD;MAAAG,UAEtC1B;KACZ,CAAC;IAAC,QAAA,CAAA;GACM,EAAA,CACT;EAAC;EACLF,EAAA,KAAAD;EAAAC,EAAA,KAAAa;CAAA,OAAAA,KAAAb,EAAA;CAAA,IAAA6B;CAAA,IAAA7B,EAAA,OAAAY,cAAAZ,EAAA,OAAAD,WAAA;EAAE8B,KAAA,CAACjB,YAAYb,SAAS;EAACC,EAAA,KAAAY;EAAAZ,EAAA,KAAAD;EAAAC,EAAA,MAAA6B;CAAA,OAAAA,KAAA7B,EAAA;CA/C1BtE,UAAUmF,IA+CPgB,EAAuB;CAAC,IAAAC;CAAA,IAAAC;CAAA,IAAA/B,EAAA,QAAAgC,OAAAC,IAAA,2BAAA,GAAA;EAEjBH,iBACD;GACL3B,cAAaK,SAAeO,MAAC;EAAC;EAE/BgB,KAAA,CAAA;EAAE/B,EAAA,MAAA8B;EAAA9B,EAAA,MAAA+B;CAAA,OAAA;EAAAD,KAAA9B,EAAA;EAAA+B,KAAA/B,EAAA;CAAA;CAJLtE,UAAUoG,IAIPC,EAAE;CAAC,OAEC9B;AAAW;AAGpB,MAAMiC,2CAAgD,IAAIE,IAAI;AAE9D,MAAaC,mBACXC,aACArB,UAA2C,CAAC,MACzC;CACH,MAAM,EACJjC,UACApB,iBAAiB4E,uBACjBnD,+BAA+B,MAC/BC,UACAC,kBACAC,cACAC,mBACAC,4BACEuB;CACJ,MAAMpB,oBAAoBb,UAAUG;CACpC,MAAMsD,kBAAkBnG,mBAAmB;CAC3C,MAAM,CAACoG,cAAcC,mBAAmB7G,SAEtC,CAAC,CAAC;CACJ,MAAM,CAAC+G,mBAAmBC,wBAAwBhH,SAGxC,IAAI;CACd,MAAMmH,2BAA2BpH,uBAC/B,IAAIqH,IAAI,CACV;CACA,MAAMC,wBAAwBtH,uBAC5B,IAAIqH,IAAI,CACV;CACA,MAAME,yBAAyBvH,uBAAoC,IAAIqH,IAAI,CAAC;CAC5E,MAAMG,mBAAmBxH,OAA8ByH,KAAAA,CAAS;CAEhE,MAAMC,oBAAoBC,OAAOC,OAAOf,YAAY,CAAC,CAACgB,MACnDC,MAAMA,GAAGrC,SAAS,WACrB;CACA,MAAMsC,oBACJtB,YAAYjB,WAAW,eAAeiB,YAAYjB,WAAW;CAC/D,MAAMtB,YAAY6D,qBAAqBL;CACvC,MAAMM,wBAAwBhI,OAAO+H,iBAAiB;CAEtD,MAAME,gBAAgBzF,qBAAmBiE,YAAYxC,UAAUC,SAAS;CAIxE,MAAMgE,cAAczB,YAAYxC,SAASsB,GAAG,EAAE;CAC9C,MAAM4C,sBACJjE,aAAagE,aAAajF,SAAS,cAAciF,YAAYlF,KAAKyE,KAAAA;CAEpE,MAAMW,sBACJpB,mBAAmBE,WAAWT,YAAYzD,KACtCgE,kBAAkBG,MAClBd;CAEN,MAAMgC,sBAAsBzI,aACzBsH,QAAgBoB,cAAsB;EACrCrB,sBAAsBsB,SAAS;GAC7B,IAAIA,MAAMrB,WAAWA,UAAU,CAACqB,KAAKpB,IAAIqB,IAAIF,SAAS,GAAG,OAAOC;GAChE,MAAMpB,MAAM,IAAIZ,IAAIgC,KAAKpB,GAAG;GAC5BA,IAAIsB,OAAOH,SAAS;GACpB,OAAO;IAAEpB;IAAQC;GAAI;EACvB,CAAC;CACH,GACA,CAAA,CACF;CAIA,MAAMuB,mBACJX,qBAAqBG,aAAajF,SAAS,cACvCiF,YAAYlF,KACZyE,KAAAA;CAEN5H,gBAAgB;EACd,MAAM8I,qBAAqBX,sBAAsBrD;EACjDqD,sBAAsBrD,UAAUoD;EAChC,IAAIY,sBAAsB,CAACD,kBAAkB;EAC7CL,oBAAoB5B,YAAYzD,IAAI0F,gBAAgB;CACtD,GAAG;EACDX;EACAW;EACAjC,YAAYzD;EACZqF;CAAmB,CACpB;CAED,MAAMpE,WAAW/B,sBAAsB0G,kBAAkB;EACvD1E;EACAD,UAAUwC,YAAYxC;EACtBN;EACAkF,UAAU9I,eACD;GACL8G;GACAoB;GACAa,uBAAuB1B,yBAAyBzC;GAChDoE,oBAAoBzB,sBAAsB3C;GAC1CqE,qBAAqBzB,uBAAuB5C;GAC5C,GAAIwD,uBAAuB,EAAEA,oBAAoB;GACjD,GAAI1B,YAAYwC,SAAS,EAAEA,OAAOxC,YAAYwC,MAAMtG,QAAQ;GAC5D,GAAIyF,oBAAoBc,OAAO,KAAK,EAAEd,oBAAoB;EAC5D,IACA;GACEvB;GACAoB;GACAE;GACA1B,YAAYwC;GACZb;EAAmB,CAEvB;CACF,CAAC;CAED,MAAMe,4BAA4BpJ,cAAc;EAC9C,IAAI,CAAC6D,mBAAmB,OAAO6D,KAAAA;EAC/B,MAAM2B,YAAY7G,4BAChBL,sBAAsBmH,kBAGtBzF,iBACF;EACA,OAAOwF,UAAUnF,SAASqF,SAAS,IAAIF,YAAY3B,KAAAA;CACrD,GAAG,CAAC7D,iBAAiB,CAAC;CAEtB,MAAM2F,uBAAuBxF,wBAC3BC,mBACAC,UACAC,SACF;CAEA,MAAM,CAACsF,cAAcvJ,gBAAgB,EACnC,IAAI0E,UAA4B;EAC9B,OAAO8E;CACT,EACF,EAAE;CAEF,MAAM,EAAEC,WAAWC,eAAeC,yBAAyBtH,mBACzDkH,YACArG,UAAUE,WAAWuD,iBAAiBvD,SACtCnB,sBAAsBmH,kBAGtBhH,uBAIC4B,eAAa;EACZwC,YAAYoD,YAAY5F,UAAQ;CAClC,CACF;CAEA,MAAM,EACJjB,IAAIkE,UACJjD,UAAU6F,cACVtE,QAAQuE,YACRd,UACExC;CACJ,MAAMuD,SAASjK,cAEX0C,YAAYwH,QAAQ;EAClBC,MAAMzD;EACNwC;CACF,CAAC,GAEH;EAAC/B;EAAQ4C;EAAcC;EAAYd;CAAK,CAC1C;CAEA,MAAMkB,2BAA2B,YAAY;EAC3C,IAAI,CAAC3G,8BAA8B;EAOnCiD,YAAYoD,aAAa5F,eAAa;GACpC,MAAMiE,gBAAcjE,WAASsB,GAAG,EAAE;GAClC,IAAI2C,eAAajF,SAAS,aAAa,OAAOgB;GAE9C,IAAImG,aAAa;GACjB,MAAMC,QAAQnC,cAAYmC,OAAOC,KAAKC,SAAS;IAC7C,IAAI,CAACjK,aAAaiK,IAAI,GAAG,OAAOA;IAChC,IACEA,KAAKC,UAAU,sBACfD,KAAKC,UAAU,kBACfD,KAAKC,UAAU,iBAEf,OAAOD;IAETH,aAAa;IACb,MAAM,EAAEK,UAAUC,WAAW,GAAGC,SAASJ;IACzC,OAAO;KACL,GAAGI;KACHH,OAAO;KACPK,WAAW;IACb;GACF,CAAC;GAED,IAAI,CAACT,YAAY,OAAOnG;GACxB,OAAO,CAAC,GAAGA,WAAS6G,MAAM,GAAG,EAAE,GAAG;IAAE,GAAG5C;IAAamC;GAAM,CAAC;EAC7D,CAAC;CACH;CAEA,MAAMU,yBAAyB/K,OAAO,KAAK;CAC3C,MAAMgL,uBACJ7B,6BAA6B,QAC7B,CAAC4B,uBAAuBpG,WACxBV,SAASqF,WAAW;CAEtB,MAAMG,UAAUjJ,wBAAwB;EACtC0D,WAAW6D;EACX,GAAIiD,uBACA,EAAEpH,mBAAmBuF,0BAA0B,IAC/C,EAAElF,SAAS;EACfgH,gCAAgC;EAChCnE;EACA+C,cAAc5F,eACZwC,YAAYoD,YACV5F,WACGqG,IAAIrI,mBAA+B,CAAC,CACpCiJ,OAAOC,OAAO,CAAC,CACfC,KAAK,CACV;EACFC,WAAWpH,eACTwC,YAAYoD,YACV5F,WACGqG,IAAIrI,mBAA+B,CAAC,CACpCiJ,OAAOC,OAAO,CAAC,CACfC,KAAK,CACV;EACFE,6BAAkE;GAChE,MAAMC,WAAW/B,WAAW7E,QAAQ6G,OAAOC,OAAO;GAElD,MAAMC,mBAAoD,CAAA;GAC1D,MAAMC,iCAAiB,IAAItE,IAAoB;GAE/C,KAAK,MAAMuE,QAAQL,SAAStH,UAAU;IACpC,MAAM4H,gBAAgBpK,yBACpBmK,KAAKjJ,OACP;IACA,IAAImJ,WACFF,KAAKE,YAAY,OACZH,eAAeI,IAAIH,KAAKE,QAAQ,KAAKF,KAAKE,WAC3C;IACN,KAAK,MAAME,gBAAgBH,eAAe;KACxCH,iBAAiBO,KAAK;MAAEH;MAAUnJ,SAASqJ;KAAa,CAAC;KACzDF,WAAWzJ,qBAAqB6J,MAAMF,YAAyB;IACjE;IACA,IAAIH,cAAcvC,SAAS,GACzBqC,eAAeQ,IACbP,KAAKjJ,QAAQK,IACbX,qBAAqB6J,MACnBL,cAAcA,cAAcvC,SAAS,EACvC,CACF;GAEJ;GAEA,MAAM8C,SAA8C,EAClDnI,UAAUyH,iBACZ;GAEA,IAAIH,SAASc,UAAU,MACrBD,OAAOC,SAASV,eAAeI,IAAIR,SAASc,MAAM,KAAKd,SAASc;GAGlE,OAAOD;EACT;EACAE,sBAAsBC,SAA8C;GAElE,MAAMC,eAAejK,4BACnBL,sBAAsBmH,kBACtBkD,IACF;GAGA/C,WAAW7E,QAAQ6G,OAAOiB,OAAOD,YAAY;EAC/C;EACAE,UAAU,YAAY;GACpB,MAAM/J,UAAU8D,YAAYxC,SAASsB,GAAG,EAAE;GAC1C,MAAMoH,cACJzI,aAAavB,SAASM,SAAS,cAAcN,QAAQK,KAAKyE,KAAAA;GAC5D,IAAIkF,aAAa;IACf,MAAMC,UAAU,IAAIrG,IAAIE,YAAYxC,SAASqG,KAAKuC,MAAMA,EAAE7J,EAAE,CAAC;IAC7DiE,sBAAsBsB,WAAS;KAC7B,MAAMuE,OACJvE,QAAMrB,WAAWT,YAAYzD,KACzB,CAAC,GAAGuF,OAAKpB,GAAG,CAAC,CAAC+D,QAAQlI,OAAO4J,QAAQpE,IAAIxF,EAAE,CAAC,IAC5C,CAAA;KACN,OAAO;MACLkE,QAAQT,YAAYzD;MACpBmE,qBAAK,IAAIZ,IAAI,CAAC,GAAGuG,MAAMH,WAAW,CAAC;KACrC;IACF,CAAC;GACH;GACA,IAAI;IACF,MAAMlG,YAAYsG,KAAK;GACzB,SAAS9D,SAAO;IACd,IAAI,EAAEA,mBAAiB+D,SAAS/D,QAAMgE,SAAS,eAAe;KAC5D,IAAIN,aAAatE,oBAAoB5B,YAAYzD,IAAI2J,WAAW;KAChE,MAAM1D;IACR;GACF;EACF;EACAiE,OAAO,OAAOvK,cAAY;GACxB,MAAMG,iBACJ6D,yBAAyB5E,gBAAAA,CACbY,SAAO;GAErB,IAAI,EAAEA,UAAQwK,YAAYxK,UAAQM,SAAS,SAAS;IAClDwD,YAAYoD,aAAalF,YAAY,CACnC,GAAGA,SACH9B,YAAwBC,eAAeH,UAAQM,IAAI,CAAC,CACrD;IACD;GACF;GAEAuE,iBAAiB7C,UAAUhC,UAAQyK;GACnC,MAAMjD,yBAAyB;GAC/B,MAAM1D,YAAY4G,YAAYvK,eAAe,EAC3C+F,UAAUlG,UAAQyK,UACpB,CAAC;EACH;EACAE,QAAQ,OAAO3K,cAAY;GACzB,MAAMG,mBACJ6D,yBAAyB5E,gBAAAA,CACbY,SAAO;GAErB,IAAI,EAAEA,UAAQwK,YAAYxK,UAAQM,SAAS,SAAS;IAClDwD,YAAYoD,aAAalF,cAAY,CACnC,GAAG7C,mBAAmB6C,WAAShC,UAAQmJ,QAAQ,GAC/CjJ,YAAwBC,iBAAeH,UAAQM,IAAI,CAAC,CACrD;IACD;GACF;GAEAuE,iBAAiB7C,UAAUhC,UAAQyK;GACnC3G,YAAYoD,aAAalF,cACvB7C,mBAAmB6C,WAAShC,UAAQmJ,QAAQ,CAC9C;GACA,MAAMrF,YAAY4G,YAAYvK,iBAAe,EAC3C+F,UAAUlG,UAAQyK,UACpB,CAAC;EACH;EACAG,UAAU,OAAOjF,gBAAc;GAC7B,MAAMkF,iBAAiBhE,WAAW7E,QAAQ6G,OAAOiC,SAAS,CAAC,CAACxJ;GAC5D,MAAMyJ,eAAeF,eAAeG,WACjChL,cAAYA,UAAQK,OAAOsF,WAC9B;GACA,IAAIoF,iBAAiB,IAAI;GAEzB,MAAM9D,qBAAqBtB,WAAS;GAEpC,MAAMsF,YAAY,IAAIrH,IACpB9E,yBAAqC+L,eAAeE,aAAc,CAAC,CAACpD,KACjE3H,cAAYA,UAAQK,EACvB,CACF;GACAyD,YAAYoD,aAAalF,cACvBA,UAAQuG,QAAQvI,cAAY,CAACiL,UAAUpF,IAAI7F,UAAQK,EAAE,CAAC,CACxD;EACF;EACA6K,UAAU,OAAO/B,YAAyBgC,WAAW;GACnDtG,iBAAiB7C,UAAUmJ,OAAOV;GAClC,MAAMW,cAAcjM,mBAAmB2E,YAAYxC,UAAU6H,UAAQ;GACrErF,YAAYoD,YAAYkE,WAAW;GAEnC,MAAMtH,YAAYuH,WAAW,EAAEnF,UAAUiF,OAAOV,UAAU,CAAC;EAC7D;EACAa,kBAAkB,EAChBC,YACAC,UACA/B,QAAAA,UACAgC,SACAC,mBACI;GACJ,MAAMC,UAAU,EAAEzF,UAAUrB,iBAAiB7C,QAAQ;GACrD,IAAIyJ,SACF,OAAOG,QAAQC,QACb/H,YAAYgI,cAAc;IACxBjE,OAAO;IACPkE,MAAMP,YAAYD;IAClBA;IACArD,WACE,OAAOuB,aAAW,WAAWA,WAASuC,KAAKC,UAAUxC,QAAM;IAC7DkC;GACF,CAAC,CACH;QACK;IACL,MAAMO,SACJR,iBAAiB5G,KAAAA,IACbtF,yBAAyBiK,UAAQiC,YAAY,IAC7CjC;IACN,OAAOmC,QAAQC,QACb/H,YAAYgI,cAAc;KACxBC,MAAMP;KACND;KACAW;KACAP;IACF,CAAC,CACH;GACF;EACF;EACAQ,0BAA0B,EAAEC,YAAYC,UAAUC,aAChDV,QAAQC,QACN/H,YAAYyI,wBAAwB;GAClClM,IAAI+L;GACJC;GACA,GAAIC,UAAU,QAAQ,EAAEA,OAAO;GAC/BX,SAAS,EAAEzF,UAAUrB,iBAAiB7C,QAAQ;EAChD,CAAC,CACH;EACF,GAAGjD,+BAA+B0D,OAAO;EACzC,GAAIA,QAAQ7B,sCAAsC,EAChDA,oCACE6B,QAAQ7B,mCACZ;EACA,GAAIS,oBAAoB,EAAEI,aAAamF,qBAAqB,IAAI,CAAC;EACjE,GAAI9F,YAAY,EAAEA,SAAS;EAC3B,GAAIC,oBAAoB,EAAEA,iBAAiB;EAC3C,GAAIG,2BAA2B,EAAEA,wBAAwB;EACzDV,UAAU;GACRgM,aAAanN;GACb,GAAG4E;GACH,GAAGzD;EACL;EACA6G;EACAN;CACF,CAAC;CAED,MAAM0F,iBAAiBpP,OAAOyG,YAAYoD,WAAW;CACrD/J,yBAAyB;EACvBsP,eAAezK,UAAU8B,YAAYoD;CACvC,GAAG,CAACpD,YAAYoD,WAAW,CAAC;CAE5BhK,gBAAgB;EACd,IAAIkL,uBAAuBpG,SAAS;EACpC,IAAI,CAACwE,2BAA2B;EAChC,IAAI1C,YAAYxC,SAASqF,SAAS,GAAG;GACnCyB,uBAAuBpG,UAAU;GACjC;EACF;EACA,MAAM0K,WAAW,IAAIzN,kBAAkB;EACvCyN,SAAS5C,OAAOtD,yBAAyB;EACzCiG,eAAezK,QACb0K,SAASC,YAAY,CAAC,CAACC,QAAQ9N,wBAAoC,CACrE;EACAsJ,uBAAuBpG,UAAU;CACnC,GAAG,CAACwE,2BAA2B1C,YAAYxC,SAASqF,MAAM,CAAC;CAC3D,OAAOG;AACT;AA7gBgC,SAAAtE,MAAAoD,MAAA;CAAA,OAuBAA,KAAIe,WAAY,IAAhBf,OAAAzE;AAA4C"}

package/dist/runtime/useChatRuntime.js

@@ -1,8 +1,8 @@
 "use client";
 import { useChatThread } from "./useChatThread.js";
 import { useCloudThreadListAdapter, useRemoteThreadListRuntime } from "@assistant-ui/core/react";
-import { useAui, useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAui, useAuiState } from "@assistant-ui/store";
 //#region src/runtime/useChatRuntime.ts
 const useChatThreadRuntime = (options) => {
 	const $ = c(6);

package/dist/runtime/useChatThread.d.ts

@@ -2,6 +2,7 @@ import { AISDKRuntimeAdapter, CustomToCreateMessageFunction } from "./useAISDKRu
 import { InitializableThreadListItem } from "../transport/AssistantChatTransport.js";
 import { ChatInit, ChatTransport } from "ai";
 import { AssistantRuntime, ExternalStoreSharedOptions } from "@assistant-ui/core";
+import { MessageRepository } from "@assistant-ui/core/internal";
 import { Chat, UIMessage as UIMessage$1 } from "@ai-sdk/react";
 //#region src/runtime/useChatThread.d.ts
 type ChatThreadOptions<UI_MESSAGE extends UIMessage$1 = UIMessage$1> = ChatInit<UI_MESSAGE> & ExternalStoreSharedOptions & {
@@ -32,6 +33,12 @@ type ChatThreadEnvironment<UI_MESSAGE extends UIMessage$1 = UIMessage$1> = {
    * from the instance.
    */
   chat?: Chat<UI_MESSAGE> | undefined;
+  /**
+   * An externally owned per-thread message repository. Hosts that route
+   * multiple threads through one mounting pass a distinct instance per
+   * thread so histories and branches stay isolated.
+   */
+  messageRepositoryInstance?: MessageRepository | undefined;
 };
 /**
  * Splits the combined options into the assistant-ui side and the `ChatInit`

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

@@ -1 +1 @@
-{"version":3,"file":"useChatThread.d.ts","names":[],"sources":["../../src/runtime/useChatThread.ts"],"mappings":";;;;;;KA+BY,kBAAkB,mBAAmB,cAAY,eAC3D,SAAS,cACP;EACE;EACA,WAAW;EACX,kBAAkB;EAClB,WAAW;EACX,mBAAmB;;;;;;;EAOnB,kBAAkB;EAClB,eAAe;EACf,oBAAoB,oBAAoB;EACxC,0BAA0B;;KAGpB,sBAAsB,mBAAmB,cAAY;EAC/D;EACA;EACA,yBAAyB;EACzB;;;;;;EAMA,OAAO,KAAK;;;;;;;cA4DD,yBAA0B,mBAAmB,aACxD,SAAS,kBAAkB;;;;;;;;;;;;;;;;;;;;EA7EE,iBAAA;;;;;;;;;;;;;;;;;;;cAoHlB,gBAAiB,mBAAmB,cAAY,aAC3D,SAAS,kBAAkB,yBAC3B,KAAK,sBAAsB,gBAC1B"}
+{"version":3,"file":"useChatThread.d.ts","names":[],"sources":["../../src/runtime/useChatThread.ts"],"mappings":";;;;;;;KAgCY,kBAAkB,mBAAmB,cAAY,eAC3D,SAAS,cACP;EACE;EACA,WAAW;EACX,kBAAkB;EAClB,WAAW;EACX,mBAAmB;;;;;;;EAOnB,kBAAkB;EAClB,eAAe;EACf,oBAAoB,oBAAoB;EACxC,0BAA0B;;KAGpB,sBAAsB,mBAAmB,cAAY;EAC/D;EACA;EACA,yBAAyB;EACzB;;;;;;EAMA,OAAO,KAAK;;;;;;EAMZ,4BAA4B;;;;;;;cA4DjB,yBAA0B,mBAAmB,aACxD,SAAS,kBAAkB;;;;;;;;;;;;;;;;;;;;EAnFE,iBAAA;;;;;;;;;;;;;;;;;;;cA0HlB,gBAAiB,mBAAmB,cAAY,aAC3D,SAAS,kBAAkB,yBAC3B,KAAK,sBAAsB,gBAC1B"}

package/dist/runtime/useChatThread.js

@@ -56,7 +56,7 @@ const splitChatThreadOptions = (options) => {
 };
 const useChatThread = (options, env) => {
 	const { adapters, transport: transportOptions, throttle, toCreateMessage, onResume, onResumeToolCall, onResumeError, joinStrategy, messageRepository, unstable_onBranchChange, chatInit: chatOptions } = splitChatThreadOptions(options);
-	const { id, isMainThread, getThreadListItem, stopOnClientDestroy = false, chat: externalChat } = env;
+	const { id, isMainThread, getThreadListItem, stopOnClientDestroy = false, chat: externalChat, messageRepositoryInstance } = env;
 	const defaultTransport = useMemo(() => new AssistantChatTransport(), []);
 	const sourceTransport = transportOptions ?? defaultTransport;
 	const transport = useDynamicChatTransport(sourceTransport);
@@ -78,6 +78,7 @@ const useChatThread = (options, env) => {
 		...onResumeToolCall && { onResumeToolCall },
 		...joinStrategy && { joinStrategy },
 		...messageRepository && { messageRepository },
+		...messageRepositoryInstance && { unstable_messageRepositoryInstance: messageRepositoryInstance },
 		...unstable_onBranchChange && { unstable_onBranchChange }
 	});
 	if (sourceTransport instanceof AssistantChatTransport) {