@assistant-ui/ai-sdk 0.0.2 → 0.0.4

package/dist/runtime/useAISDKRuntime.d.ts

@@ -1,14 +1,16 @@
 import { JoinStrategy } from "@assistant-ui/core/react";
-import { AppendMessage, AssistantRuntime, ExternalStoreAdapter, ExternalStoreSharedOptions, SuggestionAdapter, ThreadHistoryAdapter } from "@assistant-ui/core";
+import { AppendMessage, AssistantRuntime, ExternalStoreAdapter, ExternalStoreSharedOptions, MessageFormatRepository, SuggestionAdapter, ThreadHistoryAdapter } from "@assistant-ui/core";
+import { MessageRepository } from "@assistant-ui/core/internal";
 import { CreateUIMessage, UIMessage, useChat } from "@ai-sdk/react";
 //#region src/runtime/useAISDKRuntime.d.ts
 type CustomToCreateMessageFunction = <UI_MESSAGE extends UIMessage = UIMessage>(message: AppendMessage) => CreateUIMessage<UI_MESSAGE>;
-type AISDKRuntimeAdapter = ExternalStoreSharedOptions & {
+type AISDKRuntimeAdapter<UI_MESSAGE extends UIMessage = UIMessage> = ExternalStoreSharedOptions & {
   adapters?: (NonNullable<ExternalStoreAdapter["adapters"]> & {
     history?: ThreadHistoryAdapter | undefined;
     suggestion?: SuggestionAdapter | undefined;
   }) | undefined;
   toCreateMessage?: CustomToCreateMessageFunction;
+  unstable_messageRepositoryInstance?: MessageRepository | undefined;
   /**
    * Whether to automatically cancel pending interactive tool calls when the user sends a new message.
    *
@@ -42,8 +44,21 @@ type AISDKRuntimeAdapter = ExternalStoreSharedOptions & {
    * as separate entries.
    */
   joinStrategy?: JoinStrategy | undefined;
+  /**
+   * A branch-aware AI SDK message tree seeded once when `useChat` is empty.
+   * After that seed, live updates come only from `useChat`. A later empty
+   * chat or a new object identity does not reload the tree.
+   */
+  messageRepository?: MessageFormatRepository<UI_MESSAGE>;
+  /**
+   * Called after an explicit `switchToBranch` (for example a BranchPicker
+   * click). Complements `setMessages` and does not enable switching by itself.
+   *
+   * @deprecated This API is still under active development and might change without notice.
+   */
+  unstable_onBranchChange?: ExternalStoreAdapter["unstable_onBranchChange"];
 };
-declare const useAISDKRuntime: <UI_MESSAGE extends UIMessage = UIMessage>(chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>, adapter?: AISDKRuntimeAdapter) => AssistantRuntime;
+declare const useAISDKRuntime: <UI_MESSAGE extends UIMessage = UIMessage>(chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>, adapter?: AISDKRuntimeAdapter<UI_MESSAGE>) => AssistantRuntime;
 //#endregion
 export { AISDKRuntimeAdapter, CustomToCreateMessageFunction, useAISDKRuntime };
 //# sourceMappingURL=useAISDKRuntime.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"useAISDKRuntime.d.ts","names":[],"sources":["../../src/runtime/useAISDKRuntime.ts"],"mappings":";;;;KAwDY,iCACV,mBAAmB,YAAY,WAE/B,SAAS,kBACN,gBAAgB;KAYT,sBAAsB;EAChC,YACK,YAAY;IACX,UAAU;IACV,aAAa;;EAGnB,kBAAkB;;;;;;;;;EASlB;;;;;;;;EAQA,WAAW;;;;;;;EAOX,mBAAmB;;;;;;;;;EASnB,eAAe;;cA8EJ,kBAAmB,mBAAmB,YAAY,WAC7D,aAAa,kBAAkB,QAAQ,cACvC,UAAS,wBAAwB"}
+{"version":3,"file":"useAISDKRuntime.d.ts","names":[],"sources":["../../src/runtime/useAISDKRuntime.ts"],"mappings":";;;;;KAkEY,iCACV,mBAAmB,YAAY,WAE/B,SAAS,kBACN,gBAAgB;KAYT,oBAAoB,mBAAmB,YAAY,aAC7D;EACE,YACK,YAAY;IACX,UAAU;IACV,aAAa;;EAGnB,kBAAkB;EAClB,qCAAqC;;;;;;;;;EASrC;;;;;;;;EAQA,WAAW;;;;;;;EAOX,mBAAmB;;;;;;;;;EASnB,eAAe;;;;;;EAMf,oBAAoB,wBAAwB;;;;;;;EAO5C,0BAA0B;;cAoFjB,kBAAmB,mBAAmB,YAAY,WAC7D,aAAa,kBAAkB,QAAQ,cACvC,UAAS,oBAAoB,gBAAgB"}

package/dist/runtime/useAISDKRuntime.js

@@ -10,10 +10,11 @@ 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 { consumeSuggestionResult } from "@assistant-ui/core/internal";
+import { MessageRepository, consumeSuggestionResult } from "@assistant-ui/core/internal";
 //#region src/runtime/useAISDKRuntime.ts
 const toUIMessage = (createMessage, fallbackRole) => ({
 	...createMessage,
@@ -22,73 +23,153 @@ 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 } = 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());
 	const lastRunConfigRef = useRef(void 0);
 	const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
-	const isRunning = chatHelpers.status === "submitted" || chatHelpers.status === "streaming" || hasExecutingTools;
+	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,
@@ -100,14 +181,21 @@ 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(() => {
+		if (!messageRepository) return void 0;
+		const converted = toExportedMessageRepository(AISDKMessageConverter.toThreadMessages, messageRepository);
+		return converted.messages.length > 0 ? converted : void 0;
+	}, [messageRepository]);
 	const generatedSuggestions = useGeneratedSuggestions(suggestionAdapter, messages, isRunning);
 	const [runtimeRef] = useState(() => ({ get current() {
 		return runtime;
@@ -115,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
@@ -133,7 +221,7 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 			let hasChanges = false;
 			const parts = lastMessage_0.parts?.map((part) => {
 				if (!isToolUIPart(part)) return part;
-				if (part.state === "output-available" || part.state === "output-error") return part;
+				if (part.state === "output-available" || part.state === "output-error" || part.state === "output-denied") return part;
 				hasChanges = true;
 				const { approval: _approval, ...rest } = part;
 				return {
@@ -149,9 +237,11 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 			}];
 		});
 	};
+	const hasSeededRepositoryRef = useRef(false);
+	const shouldFeedRepository = exportedMessageRepository != null && !hasSeededRepositoryRef.current && messages.length === 0;
 	const runtime = useExternalStoreRuntime({
-		isRunning,
-		messages,
+		isRunning: providerIsRunning,
+		...shouldFeedRepository ? { messageRepository: exportedMessageRepository } : { messages },
 		unstable_enableToolInvocations: true,
 		setToolStatuses,
 		setMessages: (messages_2) => chatHelpers.setMessages(messages_2.map(getVercelAIMessages).filter(Boolean).flat()),
@@ -181,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;
@@ -247,9 +352,11 @@ 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 },
+		...unstable_onBranchChange && { unstable_onBranchChange },
 		adapters: {
 			attachments: vercelAttachmentAdapter,
 			...contextAdapters,
@@ -258,8 +365,27 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 		extras,
 		isLoading
 	});
+	const setMessagesRef = useRef(chatHelpers.setMessages);
+	useInsertionEffect(() => {
+		setMessagesRef.current = chatHelpers.setMessages;
+	}, [chatHelpers.setMessages]);
+	useEffect(() => {
+		if (hasSeededRepositoryRef.current) return;
+		if (!exportedMessageRepository) return;
+		if (chatHelpers.messages.length > 0) {
+			hasSeededRepositoryRef.current = true;
+			return;
+		}
+		const tempRepo = new MessageRepository();
+		tempRepo.import(exportedMessageRepository);
+		setMessagesRef.current(tempRepo.getMessages().flatMap(getExternalStoreMessages));
+		hasSeededRepositoryRef.current = true;
+	}, [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","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","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","messageTiming","lastMessage","optimisticMessageId","useThreadMessages","metadata","toolArgsKeyOrderCache","toolLastInputCache","mcpAppMetadataCache","error","generatedSuggestions","runtimeRef","runtime","isLoading","deleteMessage","deleteHistoryMessage","toThreadMessages","setMessages","chatId","chatMessages","chatStatus","extras","provide","chat","completePendingToolCalls","hasChanges","parts","map","part","state","approval","_approval","rest","const","errorText","slice","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"],"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 { consumeSuggestionResult } 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 = 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\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 = {},\n) => {\n  const {\n    adapters,\n    toCreateMessage: customToCreateMessage,\n    cancelPendingToolCallsOnSend = true,\n    onResume,\n    onResumeToolCall,\n    joinStrategy,\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 isRunning =\n    chatHelpers.status === \"submitted\" ||\n    chatHelpers.status === \"streaming\" ||\n    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 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 (part.state === \"output-available\" || part.state === \"output-error\")\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 runtime = useExternalStoreRuntime({\n    isRunning,\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    adapters: {\n      attachments: vercelAttachmentAdapter,\n      ...contextAdapters,\n      ...adapters,\n    },\n    extras,\n    isLoading,\n  });\n\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;;;;AA8DA,MAAM8C,eACJC,eACAC,kBAEC;CACC,GAAGD;CACHE,IAAIF,cAAcE,MAAMxC,WAAW;CACnCyC,MAAMH,cAAcG,QAAQF;AAC9B;AA6CF,MAAMY,oBAAiD,CAAA;AAEvD,MAAMC,2BACJC,mBACAC,UACAC,cACgC;CAChC,MAAM,CAACC,aAAaC,kBAClB/D,SAAsCyD,iBAAiB;CACzD,MAAMO,gBAAgBjE,OAA+B,IAAI;CACzD,MAAMmE,gBAAgBnE,OAAO,KAAK;CAClC,MAAMoE,cAAcpE,OAAO6D,QAAQ;CACnCO,YAAYC,UAAUR;CACtB,MAAMS,aAAatE,OAAO4D,iBAAiB;CAC3CU,WAAWD,UAAUT;CAGrB9D,gBAAgB;EACd,MAAM0E,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,MAAM9B,SAAS,aAAa;EAChC,IAAI8B,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,MAAMxD,wBAAwByD,oBAAoB;KAChDD;KACAG,UAAUtB;IACZ,CAAC;GACH,QAAQ,CAAC;EACX,EAAA,CAAG;CACL,GAAG,CAjDgBJ,qBAAqB,MAiDxBE,SAAS,CAAC;CAE1BhE,gBAAgB;EACd,aAAa;GACXmE,cAAcI,SAASI,MAAM;EAC/B;CACF,GAAG,CAAA,CAAE;CAEL,OAAOV;AACT;AAEA,MAAawB,mBACXC,aACAZ,UAA+B,CAAC,MAC7B;CACH,MAAM,EACJ1B,UACApB,iBAAiB4D,uBACjBpC,+BAA+B,MAC/BC,UACAC,kBACAC,iBACEmB;CACJ,MAAMhB,oBAAoBV,UAAUG;CACpC,MAAMsC,kBAAkBlF,mBAAmB;CAC3C,MAAM,CAACmF,cAAcC,mBAAmB5F,SAEtC,CAAC,CAAC;CACJ,MAAM8F,2BAA2B/F,uBAC/B,IAAIgG,IAAI,CACV;CACA,MAAMC,wBAAwBjG,uBAC5B,IAAIgG,IAAI,CACV;CACA,MAAME,yBAAyBlG,uBAAoC,IAAIgG,IAAI,CAAC;CAC5E,MAAMG,mBAAmBnG,OAA8BoG,KAAAA,CAAS;CAEhE,MAAMC,oBAAoBC,OAAOC,OAAOX,YAAY,CAAC,CAACY,MACnDC,MAAMA,GAAGxB,SAAS,WACrB;CACA,MAAMnB,YACJ0B,YAAYR,WAAW,eACvBQ,YAAYR,WAAW,eACvBqB;CAEF,MAAMK,gBAAgBnE,qBAAmBiD,YAAY3B,UAAUC,SAAS;CAIxE,MAAM6C,cAAcnB,YAAY3B,SAASkB,GAAG,EAAE;CAC9C,MAAM6B,sBACJ9C,aAAa6C,aAAa3D,SAAS,cAAc2D,YAAY5D,KAAKqD,KAAAA;CAEpE,MAAMvC,WAAW5B,sBAAsB4E,kBAAkB;EACvD/C;EACAD,UAAU2B,YAAY3B;EACtBJ;EACAqD,UAAU/G,eACD;GACL6F;GACAc;GACAK,uBAAuBhB,yBAAyB1B;GAChD2C,oBAAoBf,sBAAsB5B;GAC1C4C,qBAAqBf,uBAAuB7B;GAC5C,GAAIuC,uBAAuB,EAAEA,oBAAoB;GACjD,GAAIpB,YAAY0B,SAAS,EAAEA,OAAO1B,YAAY0B,MAAMxE,QAAQ;EAC9D,IACA;GAACkD;GAAcc;GAAeE;GAAqBpB,YAAY0B;EAAK,CACtE;CACF,CAAC;CAED,MAAMC,uBAAuBxD,wBAC3BC,mBACAC,UACAC,SACF;CAEA,MAAM,CAACsD,cAAcnH,gBAAgB,EACnC,IAAIoE,UAA4B;EAC9B,OAAOgD;CACT,EACF,EAAE;CAEF,MAAM,EAAEC,WAAWC,eAAeC,yBAAyBnF,mBACzD+E,YACAlE,UAAUE,WAAWuC,iBAAiBvC,SACtCnB,sBAAsBwF,kBAGtBrF,uBAICyB,eAAa;EACZ2B,YAAYkC,YAAY7D,UAAQ;CAClC,CACF;CAEA,MAAM,EACJd,IAAI4E,QACJ9D,UAAU+D,cACV5C,QAAQ6C,YACRX,UACE1B;CACJ,MAAMsC,SAAS/H,cAEXyC,YAAYuF,QAAQ;EAClBC,MAAMxC;EACN0B;CACF,CAAC,GAEH;EAACS;EAAQC;EAAcC;EAAYX;CAAK,CAC1C;CAEA,MAAMe,2BAA2B,YAAY;EAC3C,IAAI,CAAC3E,8BAA8B;EAOnCkC,YAAYkC,aAAa7D,eAAa;GACpC,MAAM8C,gBAAc9C,WAASkB,GAAG,EAAE;GAClC,IAAI4B,eAAa3D,SAAS,aAAa,OAAOa;GAE9C,IAAIqE,aAAa;GACjB,MAAMC,QAAQxB,cAAYwB,OAAOC,KAAKC,SAAS;IAC7C,IAAI,CAAC/H,aAAa+H,IAAI,GAAG,OAAOA;IAChC,IAAIA,KAAKC,UAAU,sBAAsBD,KAAKC,UAAU,gBACtD,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,OAAOrE;GACxB,OAAO,CAAC,GAAGA,WAAS+E,MAAM,GAAG,EAAE,GAAG;IAAE,GAAGjC;IAAawB;GAAM,CAAC;EAC7D,CAAC;CACH;CAEA,MAAMd,UAAU7G,wBAAwB;EACtCsD;EACAD;EACAgF,gCAAgC;EAChChD;EACA6B,cAAc7D,eACZ2B,YAAYkC,YACV7D,WACGuE,IAAIpG,mBAA+B,CAAC,CACpC8G,OAAOC,OAAO,CAAC,CACfC,KAAK,CACV;EACFC,WAAWpF,eACT2B,YAAYkC,YACV7D,WACGuE,IAAIpG,mBAA+B,CAAC,CACpC8G,OAAOC,OAAO,CAAC,CACfC,KAAK,CACV;EACFE,6BAAkE;GAChE,MAAMC,WAAW/B,WAAW/C,QAAQ+E,OAAOC,OAAO;GAElD,MAAMC,mBAAoD,CAAA;GAC1D,MAAMC,iCAAiB,IAAIvD,IAAoB;GAE/C,KAAK,MAAMwD,QAAQL,SAAStF,UAAU;IACpC,MAAM4F,gBAAgBhI,yBACpB+H,KAAK9G,OACP;IACA,IAAIgH,WACFF,KAAKE,YAAY,OACZH,eAAeI,IAAIH,KAAKE,QAAQ,KAAKF,KAAKE,WAC3C;IACN,KAAK,MAAME,gBAAgBH,eAAe;KACxCH,iBAAiBO,KAAK;MAAEH;MAAUhH,SAASkH;KAAa,CAAC;KACzDF,WAAWtH,qBAAqB0H,MAAMF,YAAyB;IACjE;IACA,IAAIH,cAAc9E,SAAS,GACzB4E,eAAeQ,IACbP,KAAK9G,QAAQK,IACbX,qBAAqB0H,MACnBL,cAAcA,cAAc9E,SAAS,EACvC,CACF;GAEJ;GAEA,MAAMqF,SAA8C,EAClDnG,UAAUyF,iBACZ;GAEA,IAAIH,SAASc,UAAU,MACrBD,OAAOC,SAASV,eAAeI,IAAIR,SAASc,MAAM,KAAKd,SAASc;GAGlE,OAAOD;EACT;EACAE,sBAAsBC,SAA8C;GAElE,MAAMC,eAAe9H,4BACnBL,sBAAsBwF,kBACtB0C,IACF;GAGA/C,WAAW/C,QAAQ+E,OAAOiB,OAAOD,YAAY;EAC/C;EACAE,UAAU,YAAY;GACpB,IAAI;IACF,MAAM9E,YAAY+E,KAAK;GACzB,SAASrD,SAAO;IACd,IAAI,EAAEA,mBAAiBsD,SAAStD,QAAMuD,SAAS,eAC7C,MAAMvD;GAEV;EACF;EACAwD,OAAO,OAAOhI,YAAY;GACxB,MAAMG,iBACJ6C,yBAAyB5D,gBAAAA,CACbY,OAAO;GAErB,IAAI,EAAEA,QAAQiI,YAAYjI,QAAQM,SAAS,SAAS;IAClDwC,YAAYkC,aAAarD,YAAY,CACnC,GAAGA,SACHzB,YAAwBC,eAAeH,QAAQM,IAAI,CAAC,CACrD;IACD;GACF;GAEAmD,iBAAiB9B,UAAU3B,QAAQkI;GACnC,MAAM3C,yBAAyB;GAC/B,MAAMzC,YAAYqF,YAAYhI,eAAe,EAC3CiE,UAAUpE,QAAQkI,UACpB,CAAC;EACH;EACAE,QAAQ,OAAOpI,cAAY;GACzB,MAAMG,mBACJ6C,yBAAyB5D,gBAAAA,CACbY,SAAO;GAErB,IAAI,EAAEA,UAAQiI,YAAYjI,UAAQM,SAAS,SAAS;IAClDwC,YAAYkC,aAAarD,cAAY,CACnC,GAAGxC,mBAAmBwC,WAAS3B,UAAQgH,QAAQ,GAC/C9G,YAAwBC,iBAAeH,UAAQM,IAAI,CAAC,CACrD;IACD;GACF;GAEAmD,iBAAiB9B,UAAU3B,UAAQkI;GACnCpF,YAAYkC,aAAarD,cACvBxC,mBAAmBwC,WAAS3B,UAAQgH,QAAQ,CAC9C;GACA,MAAMlE,YAAYqF,YAAYhI,iBAAe,EAC3CiE,UAAUpE,UAAQkI,UACpB,CAAC;EACH;EACAG,UAAU,OAAOC,cAAc;GAC7B,MAAMC,iBAAiB7D,WAAW/C,QAAQ+E,OAAO8B,SAAS,CAAC,CAACrH;GAC5D,MAAMsH,eAAeF,eAAeG,WACjC1I,cAAYA,UAAQK,OAAOiI,SAC9B;GACA,IAAIG,iBAAiB,IAAI;GAEzB,MAAM3D,qBAAqBwD,SAAS;GAEpC,MAAMK,YAAY,IAAIC,IACpB7J,yBAAqCwJ,eAAeE,aAAc,CAAC,CAAC/C,KACjE1F,cAAYA,UAAQK,EACvB,CACF;GACAyC,YAAYkC,aAAarD,cACvBA,UAAQyE,QAAQpG,cAAY,CAAC2I,UAAUE,IAAI7I,UAAQK,EAAE,CAAC,CACxD;EACF;EACAyI,UAAU,OAAO9B,YAAyB+B,WAAW;GACnDtF,iBAAiB9B,UAAUoH,OAAOb;GAClC,MAAMc,cAAc7J,mBAAmB2D,YAAY3B,UAAU6F,UAAQ;GACrElE,YAAYkC,YAAYgE,WAAW;GAEnC,MAAMlG,YAAYmG,WAAW,EAAE7E,UAAU2E,OAAOb,UAAU,CAAC;EAC7D;EACAgB,kBAAkB,EAChBC,YACAC,UACA9B,QAAAA,UACA+B,SACAC,mBACI;GACJ,MAAMC,UAAU,EAAEnF,UAAUX,iBAAiB9B,QAAQ;GACrD,IAAI0H,SACF,OAAOG,QAAQC,QACb3G,YAAY4G,cAAc;IACxB9D,OAAO;IACP+D,MAAMP,YAAYD;IAClBA;IACAlD,WACE,OAAOqB,aAAW,WAAWA,WAASsC,KAAKC,UAAUvC,QAAM;IAC7DiC;GACF,CAAC,CACH;QACK;IACL,MAAMO,SACJR,iBAAiB5F,KAAAA,IACblE,yBAAyB8H,UAAQgC,YAAY,IAC7ChC;IACN,OAAOkC,QAAQC,QACb3G,YAAY4G,cAAc;KACxBC,MAAMP;KACND;KACAW;KACAP;IACF,CAAC,CACH;GACF;EACF;EACAQ,0BAA0B,EAAEC,YAAYC,UAAUC,aAChDV,QAAQC,QACN3G,YAAYqH,wBAAwB;GAClC9J,IAAI2J;GACJC;GACA,GAAIC,UAAU,QAAQ,EAAEA,OAAO;GAC/BX,SAAS,EAAEnF,UAAUX,iBAAiB9B,QAAQ;EAChD,CAAC,CACH;EACF,GAAG3C,+BAA+BkD,OAAO;EACzC,GAAIhB,oBAAoB,EAAEG,aAAaoD,qBAAqB,IAAI,CAAC;EACjE,GAAI5D,YAAY,EAAEA,SAAS;EAC3B,GAAIC,oBAAoB,EAAEA,iBAAiB;EAC3CN,UAAU;GACR4J,aAAa/K;GACb,GAAG4D;GACH,GAAGzC;EACL;EACA4E;EACAR;CACF,CAAC;CAED,OAAOD;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);