npm - @assistant-ui/react-ai-sdk - Versions diffs - 1.3.25 → 1.3.27 - Mend

@assistant-ui/react-ai-sdk 1.3.25 → 1.3.27

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (104) hide show

package/dist/ui/use-chat/useAISDKRuntime.js CHANGED Viewed

@@ -1,215 +1,187 @@
 "use client";
-import { useState, useMemo, useRef } from "react";
-import { isToolUIPart, generateId } from "ai";
-import { useExternalStoreRuntime, useRuntimeAdapters, useToolInvocations, } from "@assistant-ui/core/react";
-import { getExternalStoreMessages } from "@assistant-ui/core";
+import { wrapModelContentEnvelope } from "../../modelContentEnvelope.js";
 import { sliceMessagesUntil } from "../utils/sliceMessagesUntil.js";
 import { toCreateMessage } from "../utils/toCreateMessage.js";
 import { vercelAttachmentAdapter } from "../utils/vercelAttachmentAdapter.js";
 import { getVercelAIMessages } from "../getVercelAIMessages.js";
 import { AISDKMessageConverter } from "../utils/convertMessage.js";
-import { wrapModelContentEnvelope } from "../../modelContentEnvelope.js";
-import { aiSDKV6FormatAdapter, } from "../adapters/aiSDKFormatAdapter.js";
-import { useExternalHistory, toExportedMessageRepository, } from "./useExternalHistory.js";
+import { aiSDKV6FormatAdapter } from "../adapters/aiSDKFormatAdapter.js";
+import { toExportedMessageRepository, useExternalHistory } from "./useExternalHistory.js";
 import { useStreamingTiming } from "./useStreamingTiming.js";
+import { generateId, isToolUIPart } from "ai";
+import { useMemo, useRef, useState } from "react";
+import { useExternalStoreRuntime, useRuntimeAdapters, useToolInvocations } from "@assistant-ui/core/react";
+import { getExternalStoreMessages } from "@assistant-ui/core";
+//#region src/ui/use-chat/useAISDKRuntime.ts
 const toUIMessage = (createMessage, fallbackRole) => ({
-    ...createMessage,
-    id: createMessage.id ?? generateId(),
-    role: createMessage.role ?? fallbackRole,
+	...createMessage,
+	id: createMessage.id ?? generateId(),
+	role: createMessage.role ?? fallbackRole
 });
-export const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, onResume, suggestions, } = {}) => {
-    const contextAdapters = useRuntimeAdapters();
-    const [toolStatuses, setToolStatuses] = useState({});
-    const toolArgsKeyOrderCacheRef = useRef(new Map());
-    const toolLastInputCacheRef = useRef(new Map());
-    const mcpAppMetadataCacheRef = useRef(new Map());
-    const lastRunConfigRef = useRef(undefined);
-    const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
-    const isRunning = chatHelpers.status === "submitted" ||
-        chatHelpers.status === "streaming" ||
-        hasExecutingTools;
-    const messageTiming = useStreamingTiming(chatHelpers.messages, isRunning);
-    const messages = AISDKMessageConverter.useThreadMessages({
-        isRunning,
-        messages: chatHelpers.messages,
-        metadata: useMemo(() => ({
-            toolStatuses,
-            messageTiming,
-            toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,
-            toolLastInputCache: toolLastInputCacheRef.current,
-            mcpAppMetadataCache: mcpAppMetadataCacheRef.current,
-            ...(chatHelpers.error && { error: chatHelpers.error.message }),
-        }), [toolStatuses, messageTiming, chatHelpers.error]),
-    });
-    const [runtimeRef] = useState(() => ({
-        get current() {
-            return runtime;
-        },
-    }));
-    const toolInvocations = useToolInvocations({
-        state: {
-            messages,
-            isRunning,
-        },
-        getTools: () => runtimeRef.current.thread.getModelContext().tools,
-        onResult: (command) => {
-            if (command.type === "add-tool-result") {
-                const output = command.modelContent !== undefined
-                    ? wrapModelContentEnvelope(command.result, command.modelContent)
-                    : command.result;
-                chatHelpers.addToolResult({
-                    tool: command.toolName,
-                    toolCallId: command.toolCallId,
-                    output,
-                    options: { metadata: lastRunConfigRef.current },
-                });
-            }
-        },
-        setToolStatuses,
-    });
-    const isLoading = useExternalHistory(runtimeRef, adapters?.history ?? contextAdapters?.history, AISDKMessageConverter.toThreadMessages, aiSDKV6FormatAdapter, (messages) => {
-        chatHelpers.setMessages(messages);
-    });
-    const completePendingToolCalls = async () => {
-        if (!cancelPendingToolCallsOnSend)
-            return;
-        await toolInvocations.abort();
-        // Mark any tool without a result as cancelled (uses setMessages to avoid triggering sendAutomaticallyWhen)
-        chatHelpers.setMessages((messages) => {
-            const lastMessage = messages.at(-1);
-            if (lastMessage?.role !== "assistant")
-                return messages;
-            let hasChanges = false;
-            const parts = lastMessage.parts?.map((part) => {
-                if (!isToolUIPart(part))
-                    return part;
-                if (part.state === "output-available" || part.state === "output-error")
-                    return part;
-                hasChanges = true;
-                return {
-                    ...part,
-                    state: "output-error",
-                    errorText: "User cancelled tool call by sending a new message.",
-                };
-            });
-            if (!hasChanges)
-                return messages;
-            return [...messages.slice(0, -1), { ...lastMessage, parts }];
-        });
-    };
-    const runtime = useExternalStoreRuntime({
-        isRunning,
-        messages,
-        setMessages: (messages) => chatHelpers.setMessages(messages
-            .map((getVercelAIMessages))
-            .filter(Boolean)
-            .flat()),
-        onImport: (messages) => chatHelpers.setMessages(messages
-            .map((getVercelAIMessages))
-            .filter(Boolean)
-            .flat()),
-        onExportExternalState: () => {
-            const exported = runtimeRef.current.thread.export();
-            const expandedMessages = [];
-            const lastInnerIdMap = new Map();
-            for (const item of exported.messages) {
-                const innerMessages = getExternalStoreMessages(item.message);
-                let parentId = item.parentId != null
-                    ? (lastInnerIdMap.get(item.parentId) ?? item.parentId)
-                    : null;
-                for (const innerMessage of innerMessages) {
-                    expandedMessages.push({ parentId, message: innerMessage });
-                    parentId = aiSDKV6FormatAdapter.getId(innerMessage);
-                }
-                if (innerMessages.length > 0) {
-                    lastInnerIdMap.set(item.message.id, aiSDKV6FormatAdapter.getId(innerMessages[innerMessages.length - 1]));
-                }
-            }
-            const result = {
-                messages: expandedMessages,
-            };
-            if (exported.headId != null) {
-                result.headId = lastInnerIdMap.get(exported.headId) ?? exported.headId;
-            }
-            return result;
-        },
-        onLoadExternalState: (repo) => {
-            // Convert MessageFormatRepository to ExportedMessageRepository
-            const exportedRepo = toExportedMessageRepository(AISDKMessageConverter.toThreadMessages, repo);
-            // Import into the thread's MessageRepository
-            runtimeRef.current.thread.import(exportedRepo);
-        },
-        onCancel: async () => {
-            chatHelpers.stop();
-            await toolInvocations.abort();
-        },
-        onNew: async (message) => {
-            const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
-            if (!(message.startRun ?? message.role === "user")) {
-                chatHelpers.setMessages((current) => [
-                    ...current,
-                    toUIMessage(createMessage, message.role),
-                ]);
-                return;
-            }
-            lastRunConfigRef.current = message.runConfig;
-            await completePendingToolCalls();
-            await chatHelpers.sendMessage(createMessage, {
-                metadata: message.runConfig,
-            });
-        },
-        onEdit: async (message) => {
-            const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
-            if (!(message.startRun ?? message.role === "user")) {
-                chatHelpers.setMessages((current) => [
-                    ...sliceMessagesUntil(current, message.parentId),
-                    toUIMessage(createMessage, message.role),
-                ]);
-                return;
-            }
-            lastRunConfigRef.current = message.runConfig;
-            chatHelpers.setMessages((current) => sliceMessagesUntil(current, message.parentId));
-            await chatHelpers.sendMessage(createMessage, {
-                metadata: message.runConfig,
-            });
-        },
-        onReload: async (parentId, config) => {
-            lastRunConfigRef.current = config.runConfig;
-            const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);
-            chatHelpers.setMessages(newMessages);
-            await chatHelpers.regenerate({ metadata: config.runConfig });
-        },
-        onAddToolResult: ({ toolCallId, result, isError }) => {
-            const options = { metadata: lastRunConfigRef.current };
-            if (isError) {
-                chatHelpers.addToolOutput({
-                    state: "output-error",
-                    tool: toolCallId,
-                    toolCallId,
-                    errorText: typeof result === "string" ? result : JSON.stringify(result),
-                    options,
-                });
-            }
-            else {
-                chatHelpers.addToolOutput({
-                    state: "output-available",
-                    tool: toolCallId,
-                    toolCallId,
-                    output: result,
-                    options,
-                });
-            }
-        },
-        onResumeToolCall: (options) => toolInvocations.resume(options.toolCallId, options.payload),
-        ...(onResume && { onResume }),
-        ...(suggestions && { suggestions }),
-        adapters: {
-            attachments: vercelAttachmentAdapter,
-            ...contextAdapters,
-            ...adapters,
-        },
-        isLoading,
-    });
-    return runtime;
+const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, onResume, suggestions } = {}) => {
+	const contextAdapters = useRuntimeAdapters();
+	const [toolStatuses, setToolStatuses] = useState({});
+	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 messageTiming = useStreamingTiming(chatHelpers.messages, isRunning);
+	const messages = AISDKMessageConverter.useThreadMessages({
+		isRunning,
+		messages: chatHelpers.messages,
+		metadata: useMemo(() => ({
+			toolStatuses,
+			messageTiming,
+			toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,
+			toolLastInputCache: toolLastInputCacheRef.current,
+			mcpAppMetadataCache: mcpAppMetadataCacheRef.current,
+			...chatHelpers.error && { error: chatHelpers.error.message }
+		}), [
+			toolStatuses,
+			messageTiming,
+			chatHelpers.error
+		])
+	});
+	const [runtimeRef] = useState(() => ({ get current() {
+		return runtime;
+	} }));
+	const toolInvocations = useToolInvocations({
+		state: {
+			messages,
+			isRunning
+		},
+		getTools: () => runtimeRef.current.thread.getModelContext().tools,
+		onResult: (command) => {
+			if (command.type === "add-tool-result") {
+				const output = command.modelContent !== void 0 ? wrapModelContentEnvelope(command.result, command.modelContent) : command.result;
+				chatHelpers.addToolResult({
+					tool: command.toolName,
+					toolCallId: command.toolCallId,
+					output,
+					options: { metadata: lastRunConfigRef.current }
+				});
+			}
+		},
+		setToolStatuses
+	});
+	const isLoading = useExternalHistory(runtimeRef, adapters?.history ?? contextAdapters?.history, AISDKMessageConverter.toThreadMessages, aiSDKV6FormatAdapter, (messages) => {
+		chatHelpers.setMessages(messages);
+	});
+	const completePendingToolCalls = async () => {
+		if (!cancelPendingToolCallsOnSend) return;
+		await toolInvocations.abort();
+		chatHelpers.setMessages((messages) => {
+			const lastMessage = messages.at(-1);
+			if (lastMessage?.role !== "assistant") return messages;
+			let hasChanges = false;
+			const parts = lastMessage.parts?.map((part) => {
+				if (!isToolUIPart(part)) return part;
+				if (part.state === "output-available" || part.state === "output-error") return part;
+				hasChanges = true;
+				return {
+					...part,
+					state: "output-error",
+					errorText: "User cancelled tool call by sending a new message."
+				};
+			});
+			if (!hasChanges) return messages;
+			return [...messages.slice(0, -1), {
+				...lastMessage,
+				parts
+			}];
+		});
+	};
+	const runtime = useExternalStoreRuntime({
+		isRunning,
+		messages,
+		setMessages: (messages) => chatHelpers.setMessages(messages.map(getVercelAIMessages).filter(Boolean).flat()),
+		onImport: (messages) => chatHelpers.setMessages(messages.map(getVercelAIMessages).filter(Boolean).flat()),
+		onExportExternalState: () => {
+			const exported = runtimeRef.current.thread.export();
+			const expandedMessages = [];
+			const lastInnerIdMap = /* @__PURE__ */ new Map();
+			for (const item of exported.messages) {
+				const innerMessages = getExternalStoreMessages(item.message);
+				let parentId = item.parentId != null ? lastInnerIdMap.get(item.parentId) ?? item.parentId : null;
+				for (const innerMessage of innerMessages) {
+					expandedMessages.push({
+						parentId,
+						message: innerMessage
+					});
+					parentId = aiSDKV6FormatAdapter.getId(innerMessage);
+				}
+				if (innerMessages.length > 0) lastInnerIdMap.set(item.message.id, aiSDKV6FormatAdapter.getId(innerMessages[innerMessages.length - 1]));
+			}
+			const result = { messages: expandedMessages };
+			if (exported.headId != null) result.headId = lastInnerIdMap.get(exported.headId) ?? exported.headId;
+			return result;
+		},
+		onLoadExternalState: (repo) => {
+			const exportedRepo = toExportedMessageRepository(AISDKMessageConverter.toThreadMessages, repo);
+			runtimeRef.current.thread.import(exportedRepo);
+		},
+		onCancel: async () => {
+			chatHelpers.stop();
+			await toolInvocations.abort();
+		},
+		onNew: async (message) => {
+			const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
+			if (!(message.startRun ?? message.role === "user")) {
+				chatHelpers.setMessages((current) => [...current, toUIMessage(createMessage, message.role)]);
+				return;
+			}
+			lastRunConfigRef.current = message.runConfig;
+			await completePendingToolCalls();
+			await chatHelpers.sendMessage(createMessage, { metadata: message.runConfig });
+		},
+		onEdit: async (message) => {
+			const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
+			if (!(message.startRun ?? message.role === "user")) {
+				chatHelpers.setMessages((current) => [...sliceMessagesUntil(current, message.parentId), toUIMessage(createMessage, message.role)]);
+				return;
+			}
+			lastRunConfigRef.current = message.runConfig;
+			chatHelpers.setMessages((current) => sliceMessagesUntil(current, message.parentId));
+			await chatHelpers.sendMessage(createMessage, { metadata: message.runConfig });
+		},
+		onReload: async (parentId, config) => {
+			lastRunConfigRef.current = config.runConfig;
+			const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);
+			chatHelpers.setMessages(newMessages);
+			await chatHelpers.regenerate({ metadata: config.runConfig });
+		},
+		onAddToolResult: ({ toolCallId, result, isError }) => {
+			const options = { metadata: lastRunConfigRef.current };
+			if (isError) chatHelpers.addToolOutput({
+				state: "output-error",
+				tool: toolCallId,
+				toolCallId,
+				errorText: typeof result === "string" ? result : JSON.stringify(result),
+				options
+			});
+			else chatHelpers.addToolOutput({
+				state: "output-available",
+				tool: toolCallId,
+				toolCallId,
+				output: result,
+				options
+			});
+		},
+		onResumeToolCall: (options) => toolInvocations.resume(options.toolCallId, options.payload),
+		...onResume && { onResume },
+		...suggestions && { suggestions },
+		adapters: {
+			attachments: vercelAttachmentAdapter,
+			...contextAdapters,
+			...adapters
+		},
+		isLoading
+	});
+	return runtime;
 };
+//#endregion
+export { useAISDKRuntime };
 //# sourceMappingURL=useAISDKRuntime.js.map

package/dist/ui/use-chat/useAISDKRuntime.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"useAISDKRuntime.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAElD,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,EACL,uBAAuB,EACvB,kBAAkB,EAClB,kBAAkB,GAEnB,MAAM,0BAA0B,CAAC;AAclC,OAAO,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAE9D,OAAO,EAAE,kBAAkB,EAAE,uCAAoC;AACjE,OAAO,EAAE,eAAe,EAAE,oCAAiC;AAC3D,OAAO,EAAE,uBAAuB,EAAE,4CAAyC;AAC3E,OAAO,EAAE,mBAAmB,EAAE,kCAA+B;AAC7D,OAAO,EAAE,qBAAqB,EAAE,mCAAgC;AAChE,OAAO,EAAE,wBAAwB,EAAE,sCAAmC;AACtE,OAAO,EAEL,oBAAoB,GACrB,0CAAuC;AACxC,OAAO,EACL,kBAAkB,EAClB,2BAA2B,GAC5B,gCAA6B;AAC9B,OAAO,EAAE,kBAAkB,EAAE,gCAA6B;AAQ1D,MAAM,WAAW,GAAG,CAClB,aAA0C,EAC1C,YAAgC,EACpB,EAAE,CACd,CAAC;IACC,GAAG,aAAa;IAChB,EAAE,EAAE,aAAa,CAAC,EAAE,IAAI,UAAU,EAAE;IACpC,IAAI,EAAE,aAAa,CAAC,IAAI,IAAI,YAAY;CACzC,CAAe,CAAC;AAmCnB,MAAM,CAAC,MAAM,eAAe,GAAG,CAC7B,WAAmD,EACnD,EACE,QAAQ,EACR,eAAe,EAAE,qBAAqB,EACtC,4BAA4B,GAAG,IAAI,EACnC,QAAQ,EACR,WAAW,MACY,EAAE,EAC3B,EAAE;IACF,MAAM,eAAe,GAAG,kBAAkB,EAAE,CAAC;IAC7C,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAE9C,EAAE,CAAC,CAAC;IACN,MAAM,wBAAwB,GAAG,MAAM,CACrC,IAAI,GAAG,EAAE,CACV,CAAC;IACF,MAAM,qBAAqB,GAAG,MAAM,CAClC,IAAI,GAAG,EAAE,CACV,CAAC;IACF,MAAM,sBAAsB,GAAG,MAAM,CAA8B,IAAI,GAAG,EAAE,CAAC,CAAC;IAC9E,MAAM,gBAAgB,GAAG,MAAM,CAAwB,SAAS,CAAC,CAAC;IAElE,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CACxD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,KAAK,WAAW,CAC/B,CAAC;IACF,MAAM,SAAS,GACb,WAAW,CAAC,MAAM,KAAK,WAAW;QAClC,WAAW,CAAC,MAAM,KAAK,WAAW;QAClC,iBAAiB,CAAC;IAEpB,MAAM,aAAa,GAAG,kBAAkB,CAAC,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE1E,MAAM,QAAQ,GAAG,qBAAqB,CAAC,iBAAiB,CAAC;QACvD,SAAS;QACT,QAAQ,EAAE,WAAW,CAAC,QAAQ;QAC9B,QAAQ,EAAE,OAAO,CACf,GAAG,EAAE,CAAC,CAAC;YACL,YAAY;YACZ,aAAa;YACb,qBAAqB,EAAE,wBAAwB,CAAC,OAAO;YACvD,kBAAkB,EAAE,qBAAqB,CAAC,OAAO;YACjD,mBAAmB,EAAE,sBAAsB,CAAC,OAAO;YACnD,GAAG,CAAC,WAAW,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;SAC/D,CAAC,EACF,CAAC,YAAY,EAAE,aAAa,EAAE,WAAW,CAAC,KAAK,CAAC,CACjD;KACF,CAAC,CAAC;IAEH,MAAM,CAAC,UAAU,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC;QACnC,IAAI,OAAO;YACT,OAAO,OAAO,CAAC;QACjB,CAAC;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,eAAe,GAAG,kBAAkB,CAAC;QACzC,KAAK,EAAE;YACL,QAAQ;YACR,SAAS;SACV;QACD,QAAQ,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,KAAK;QACjE,QAAQ,EAAE,CAAC,OAAO,EAAE,EAAE;YACpB,IAAI,OAAO,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;gBACvC,MAAM,MAAM,GACV,OAAO,CAAC,YAAY,KAAK,SAAS;oBAChC,CAAC,CAAC,wBAAwB,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,YAAY,CAAC;oBAChE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;gBACrB,WAAW,CAAC,aAAa,CAAC;oBACxB,IAAI,EAAE,OAAO,CAAC,QAAQ;oBACtB,UAAU,EAAE,OAAO,CAAC,UAAU;oBAC9B,MAAM;oBACN,OAAO,EAAE,EAAE,QAAQ,EAAE,gBAAgB,CAAC,OAAO,EAAE;iBAChD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,eAAe;KAChB,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,kBAAkB,CAClC,UAAU,EACV,QAAQ,EAAE,OAAO,IAAI,eAAe,EAAE,OAAO,EAC7C,qBAAqB,CAAC,gBAEF,EACpB,oBAGC,EACD,CAAC,QAAQ,EAAE,EAAE;QACX,WAAW,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC,CACF,CAAC;IAEF,MAAM,wBAAwB,GAAG,KAAK,IAAI,EAAE;QAC1C,IAAI,CAAC,4BAA4B;YAAE,OAAO;QAE1C,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;QAE9B,2GAA2G;QAC3G,WAAW,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACpC,IAAI,WAAW,EAAE,IAAI,KAAK,WAAW;gBAAE,OAAO,QAAQ,CAAC;YAEvD,IAAI,UAAU,GAAG,KAAK,CAAC;YACvB,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC5C,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;oBAAE,OAAO,IAAI,CAAC;gBACrC,IAAI,IAAI,CAAC,KAAK,KAAK,kBAAkB,IAAI,IAAI,CAAC,KAAK,KAAK,cAAc;oBACpE,OAAO,IAAI,CAAC;gBAEd,UAAU,GAAG,IAAI,CAAC;gBAClB,OAAO;oBACL,GAAG,IAAI;oBACP,KAAK,EAAE,cAAuB;oBAC9B,SAAS,EAAE,oDAAoD;iBAChE,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,UAAU;gBAAE,OAAO,QAAQ,CAAC;YACjC,OAAO,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,uBAAuB,CAAC;QACtC,SAAS;QACT,QAAQ;QACR,WAAW,EAAE,CAAC,QAAQ,EAAE,EAAE,CACxB,WAAW,CAAC,WAAW,CACrB,QAAQ;aACL,GAAG,CAAC,CAAA,mBAA+B,CAAA,CAAC;aACpC,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,EAAE,CACV;QACH,QAAQ,EAAE,CAAC,QAAQ,EAAE,EAAE,CACrB,WAAW,CAAC,WAAW,CACrB,QAAQ;aACL,GAAG,CAAC,CAAA,mBAA+B,CAAA,CAAC;aACpC,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,EAAE,CACV;QACH,qBAAqB,EAAE,GAAwC,EAAE;YAC/D,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YAEpD,MAAM,gBAAgB,GAAoC,EAAE,CAAC;YAC7D,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;YAEjD,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACrC,MAAM,aAAa,GAAG,wBAAwB,CAC5C,IAAI,CAAC,OAAO,CACb,CAAC;gBACF,IAAI,QAAQ,GACV,IAAI,CAAC,QAAQ,IAAI,IAAI;oBACnB,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC;oBACtD,CAAC,CAAC,IAAI,CAAC;gBACX,KAAK,MAAM,YAAY,IAAI,aAAa,EAAE,CAAC;oBACzC,gBAAgB,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;oBAC3D,QAAQ,GAAG,oBAAoB,CAAC,KAAK,CAAC,YAAyB,CAAC,CAAC;gBACnE,CAAC;gBACD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC7B,cAAc,CAAC,GAAG,CAChB,IAAI,CAAC,OAAO,CAAC,EAAE,EACf,oBAAoB,CAAC,KAAK,CACxB,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAe,CACtD,CACF,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,MAAM,GAAwC;gBAClD,QAAQ,EAAE,gBAAgB;aAC3B,CAAC;YAEF,IAAI,QAAQ,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;gBAC5B,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC;YACzE,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,mBAAmB,EAAE,CAAC,IAAyC,EAAE,EAAE;YACjE,+DAA+D;YAC/D,MAAM,YAAY,GAAG,2BAA2B,CAC9C,qBAAqB,CAAC,gBAAgB,EACtC,IAAI,CACL,CAAC;YAEF,6CAA6C;YAC7C,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACjD,CAAC;QACD,QAAQ,EAAE,KAAK,IAAI,EAAE;YACnB,WAAW,CAAC,IAAI,EAAE,CAAC;YACnB,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;QAChC,CAAC;QACD,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACvB,MAAM,aAAa,GAAG,CACpB,qBAAqB,IAAI,eAAe,CACzC,CAAa,OAAO,CAAC,CAAC;YAEvB,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,CAAC;gBACnD,WAAW,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;oBACnC,GAAG,OAAO;oBACV,WAAW,CAAa,aAAa,EAAE,OAAO,CAAC,IAAI,CAAC;iBACrD,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,gBAAgB,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC;YAC7C,MAAM,wBAAwB,EAAE,CAAC;YACjC,MAAM,WAAW,CAAC,WAAW,CAAC,aAAa,EAAE;gBAC3C,QAAQ,EAAE,OAAO,CAAC,SAAS;aAC5B,CAAC,CAAC;QACL,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACxB,MAAM,aAAa,GAAG,CACpB,qBAAqB,IAAI,eAAe,CACzC,CAAa,OAAO,CAAC,CAAC;YAEvB,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,CAAC;gBACnD,WAAW,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;oBACnC,GAAG,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC;oBAChD,WAAW,CAAa,aAAa,EAAE,OAAO,CAAC,IAAI,CAAC;iBACrD,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,gBAAgB,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC;YAC7C,WAAW,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE,EAAE,CAClC,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,CAC9C,CAAC;YACF,MAAM,WAAW,CAAC,WAAW,CAAC,aAAa,EAAE;gBAC3C,QAAQ,EAAE,OAAO,CAAC,SAAS;aAC5B,CAAC,CAAC;QACL,CAAC;QACD,QAAQ,EAAE,KAAK,EAAE,QAAuB,EAAE,MAAM,EAAE,EAAE;YAClD,gBAAgB,CAAC,OAAO,GAAG,MAAM,CAAC,SAAS,CAAC;YAC5C,MAAM,WAAW,GAAG,kBAAkB,CAAC,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACvE,WAAW,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;YAErC,MAAM,WAAW,CAAC,UAAU,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;QAC/D,CAAC;QACD,eAAe,EAAE,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE;YACnD,MAAM,OAAO,GAAG,EAAE,QAAQ,EAAE,gBAAgB,CAAC,OAAO,EAAE,CAAC;YACvD,IAAI,OAAO,EAAE,CAAC;gBACZ,WAAW,CAAC,aAAa,CAAC;oBACxB,KAAK,EAAE,cAAc;oBACrB,IAAI,EAAE,UAAU;oBAChB,UAAU;oBACV,SAAS,EACP,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC9D,OAAO;iBACR,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,aAAa,CAAC;oBACxB,KAAK,EAAE,kBAAkB;oBACzB,IAAI,EAAE,UAAU;oBAChB,UAAU;oBACV,MAAM,EAAE,MAAM;oBACd,OAAO;iBACR,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,gBAAgB,EAAE,CAAC,OAAO,EAAE,EAAE,CAC5B,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC;QAC7D,GAAG,CAAC,QAAQ,IAAI,EAAE,QAAQ,EAAE,CAAC;QAC7B,GAAG,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,CAAC;QACnC,QAAQ,EAAE;YACR,WAAW,EAAE,uBAAuB;YACpC,GAAG,eAAe;YAClB,GAAG,QAAQ;SACZ;QACD,SAAS;KACV,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC"}
1	+ {"version":3,"file":"useAISDKRuntime.js","names":[],"sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useState, useMemo, useRef } from \"react\";\nimport type { UIMessage, useChat, CreateUIMessage } from \"@ai-sdk/react\";\nimport { isToolUIPart, generateId } from \"ai\";\nimport {\n useExternalStoreRuntime,\n useRuntimeAdapters,\n useToolInvocations,\n type ToolExecutionStatus,\n} from \"@assistant-ui/core/react\";\nimport type {\n ExternalStoreAdapter,\n ThreadHistoryAdapter,\n AssistantRuntime,\n ThreadMessage,\n ThreadSuggestion,\n MessageFormatAdapter,\n MessageFormatItem,\n MessageFormatRepository,\n AppendMessage,\n RunConfig,\n McpAppMetadata,\n} from \"@assistant-ui/core\";\nimport { getExternalStoreMessages } from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../getVercelAIMessages\";\nimport { AISDKMessageConverter } from \"../utils/convertMessage\";\nimport { wrapModelContentEnvelope } from \"../../modelContentEnvelope\";\nimport {\n type AISDKStorageFormat,\n aiSDKV6FormatAdapter,\n} from \"../adapters/aiSDKFormatAdapter\";\nimport {\n useExternalHistory,\n toExportedMessageRepository,\n} from \"./useExternalHistory\";\nimport { useStreamingTiming } from \"./useStreamingTiming\";\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 = {\n adapters?:\n \| (NonNullable<ExternalStoreAdapter[\"adapters\"]> & {\n history?: ThreadHistoryAdapter \| 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 Follow up suggestions to surface on the thread. Use this to drive\n * dynamic suggestions from application state, tool results, or backend\n * responses; flows into `thread.suggestions` and is rendered by\n * components that read it (such as the shadcn `ThreadFollowupSuggestions`).\n */\n suggestions?: readonly ThreadSuggestion[] \| undefined;\n};\n\nexport const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>,\n {\n adapters,\n toCreateMessage: customToCreateMessage,\n cancelPendingToolCallsOnSend = true,\n onResume,\n suggestions,\n }: AISDKRuntimeAdapter = {},\n) => {\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 const messages = AISDKMessageConverter.useThreadMessages({\n isRunning,\n messages: chatHelpers.messages,\n metadata: useMemo(\n () => ({\n toolStatuses,\n messageTiming,\n toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,\n toolLastInputCache: toolLastInputCacheRef.current,\n mcpAppMetadataCache: mcpAppMetadataCacheRef.current,\n ...(chatHelpers.error && { error: chatHelpers.error.message }),\n }),\n [toolStatuses, messageTiming, chatHelpers.error],\n ),\n });\n\n const [runtimeRef] = useState(() => ({\n get current(): AssistantRuntime {\n return runtime;\n },\n }));\n\n const toolInvocations = useToolInvocations({\n state: {\n messages,\n isRunning,\n },\n getTools: () => runtimeRef.current.thread.getModelContext().tools,\n onResult: (command) => {\n if (command.type === \"add-tool-result\") {\n const output =\n command.modelContent !== undefined\n ? wrapModelContentEnvelope(command.result, command.modelContent)\n : command.result;\n chatHelpers.addToolResult({\n tool: command.toolName,\n toolCallId: command.toolCallId,\n output,\n options: { metadata: lastRunConfigRef.current },\n });\n }\n },\n setToolStatuses,\n });\n\n const isLoading = 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 completePendingToolCalls = async () => {\n if (!cancelPendingToolCallsOnSend) return;\n\n await toolInvocations.abort();\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 return {\n ...part,\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 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 chatHelpers.stop();\n await toolInvocations.abort();\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 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: ({ toolCallId, result, isError }) => {\n const options = { metadata: lastRunConfigRef.current };\n if (isError) {\n chatHelpers.addToolOutput({\n state: \"output-error\",\n tool: toolCallId,\n toolCallId,\n errorText:\n typeof result === \"string\" ? result : JSON.stringify(result),\n options,\n });\n } else {\n chatHelpers.addToolOutput({\n state: \"output-available\",\n tool: toolCallId,\n toolCallId,\n output: result,\n options,\n });\n }\n },\n onResumeToolCall: (options) =>\n toolInvocations.resume(options.toolCallId, options.payload),\n ...(onResume && { onResume }),\n ...(suggestions && { suggestions }),\n adapters: {\n attachments: vercelAttachmentAdapter,\n ...contextAdapters,\n ...adapters,\n },\n isLoading,\n });\n\n return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;;AAgDA,MAAM,eACJ,eACA,kBAEC;CACC,GAAG;CACH,IAAI,cAAc,MAAM,WAAW;CACnC,MAAM,cAAc,QAAQ;AAC9B;AAmCF,MAAa,mBACX,aACA,EACE,UACA,iBAAiB,uBACjB,+BAA+B,MAC/B,UACA,gBACuB,CAAC,MACvB;CACH,MAAM,kBAAkB,mBAAmB;CAC3C,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,2BAA2B,uBAC/B,IAAI,IAAI,CACV;CACA,MAAM,wBAAwB,uBAC5B,IAAI,IAAI,CACV;CACA,MAAM,yBAAyB,uBAAoC,IAAI,IAAI,CAAC;CAC5E,MAAM,mBAAmB,OAA8B,KAAA,CAAS;CAEhE,MAAM,oBAAoB,OAAO,OAAO,YAAY,EAAE,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,YACJ,YAAY,WAAW,eACvB,YAAY,WAAW,eACvB;CAEF,MAAM,gBAAgB,mBAAmB,YAAY,UAAU,SAAS;CAExE,MAAM,WAAW,sBAAsB,kBAAkB;EACvD;EACA,UAAU,YAAY;EACtB,UAAU,eACD;GACL;GACA;GACA,uBAAuB,yBAAyB;GAChD,oBAAoB,sBAAsB;GAC1C,qBAAqB,uBAAuB;GAC5C,GAAI,YAAY,SAAS,EAAE,OAAO,YAAY,MAAM,QAAQ;EAC9D,IACA;GAAC;GAAc;GAAe,YAAY;EAAK,CACjD;CACF,CAAC;CAED,MAAM,CAAC,cAAc,gBAAgB,EACnC,IAAI,UAA4B;EAC9B,OAAO;CACT,EACF,EAAE;CAEF,MAAM,kBAAkB,mBAAmB;EACzC,OAAO;GACL;GACA;EACF;EACA,gBAAgB,WAAW,QAAQ,OAAO,gBAAgB,EAAE;EAC5D,WAAW,YAAY;GACrB,IAAI,QAAQ,SAAS,mBAAmB;IACtC,MAAM,SACJ,QAAQ,iBAAiB,KAAA,IACrB,yBAAyB,QAAQ,QAAQ,QAAQ,YAAY,IAC7D,QAAQ;IACd,YAAY,cAAc;KACxB,MAAM,QAAQ;KACd,YAAY,QAAQ;KACpB;KACA,SAAS,EAAE,UAAU,iBAAiB,QAAQ;IAChD,CAAC;GACH;EACF;EACA;CACF,CAAC;CAED,MAAM,YAAY,mBAChB,YACA,UAAU,WAAW,iBAAiB,SACtC,sBAAsB,kBAGtB,uBAIC,aAAa;EACZ,YAAY,YAAY,QAAQ;CAClC,CACF;CAEA,MAAM,2BAA2B,YAAY;EAC3C,IAAI,CAAC,8BAA8B;EAEnC,MAAM,gBAAgB,MAAM;EAG5B,YAAY,aAAa,aAAa;GACpC,MAAM,cAAc,SAAS,GAAG,EAAE;GAClC,IAAI,aAAa,SAAS,aAAa,OAAO;GAE9C,IAAI,aAAa;GACjB,MAAM,QAAQ,YAAY,OAAO,KAAK,SAAS;IAC7C,IAAI,CAAC,aAAa,IAAI,GAAG,OAAO;IAChC,IAAI,KAAK,UAAU,sBAAsB,KAAK,UAAU,gBACtD,OAAO;IAET,aAAa;IACb,OAAO;KACL,GAAG;KACH,OAAO;KACP,WAAW;IACb;GACF,CAAC;GAED,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,CAAC,GAAG,SAAS,MAAM,GAAG,EAAE,GAAG;IAAE,GAAG;IAAa;GAAM,CAAC;EAC7D,CAAC;CACH;CAEA,MAAM,UAAU,wBAAwB;EACtC;EACA;EACA,cAAc,aACZ,YAAY,YACV,SACG,IAAI,mBAA+B,EACnC,OAAO,OAAO,EACd,KAAK,CACV;EACF,WAAW,aACT,YAAY,YACV,SACG,IAAI,mBAA+B,EACnC,OAAO,OAAO,EACd,KAAK,CACV;EACF,6BAAkE;GAChE,MAAM,WAAW,WAAW,QAAQ,OAAO,OAAO;GAElD,MAAM,mBAAoD,CAAC;GAC3D,MAAM,iCAAiB,IAAI,IAAoB;GAE/C,KAAK,MAAM,QAAQ,SAAS,UAAU;IACpC,MAAM,gBAAgB,yBACpB,KAAK,OACP;IACA,IAAI,WACF,KAAK,YAAY,OACZ,eAAe,IAAI,KAAK,QAAQ,KAAK,KAAK,WAC3C;IACN,KAAK,MAAM,gBAAgB,eAAe;KACxC,iBAAiB,KAAK;MAAE;MAAU,SAAS;KAAa,CAAC;KACzD,WAAW,qBAAqB,MAAM,YAAyB;IACjE;IACA,IAAI,cAAc,SAAS,GACzB,eAAe,IACb,KAAK,QAAQ,IACb,qBAAqB,MACnB,cAAc,cAAc,SAAS,EACvC,CACF;GAEJ;GAEA,MAAM,SAA8C,EAClD,UAAU,iBACZ;GAEA,IAAI,SAAS,UAAU,MACrB,OAAO,SAAS,eAAe,IAAI,SAAS,MAAM,KAAK,SAAS;GAGlE,OAAO;EACT;EACA,sBAAsB,SAA8C;GAElE,MAAM,eAAe,4BACnB,sBAAsB,kBACtB,IACF;GAGA,WAAW,QAAQ,OAAO,OAAO,YAAY;EAC/C;EACA,UAAU,YAAY;GACpB,YAAY,KAAK;GACjB,MAAM,gBAAgB,MAAM;EAC9B;EACA,OAAO,OAAO,YAAY;GACxB,MAAM,iBACJ,yBAAyB,iBACb,OAAO;GAErB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,YAAY,aAAa,YAAY,CACnC,GAAG,SACH,YAAwB,eAAe,QAAQ,IAAI,CACrD,CAAC;IACD;GACF;GAEA,iBAAiB,UAAU,QAAQ;GACnC,MAAM,yBAAyB;GAC/B,MAAM,YAAY,YAAY,eAAe,EAC3C,UAAU,QAAQ,UACpB,CAAC;EACH;EACA,QAAQ,OAAO,YAAY;GACzB,MAAM,iBACJ,yBAAyB,iBACb,OAAO;GAErB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,YAAY,aAAa,YAAY,CACnC,GAAG,mBAAmB,SAAS,QAAQ,QAAQ,GAC/C,YAAwB,eAAe,QAAQ,IAAI,CACrD,CAAC;IACD;GACF;GAEA,iBAAiB,UAAU,QAAQ;GACnC,YAAY,aAAa,YACvB,mBAAmB,SAAS,QAAQ,QAAQ,CAC9C;GACA,MAAM,YAAY,YAAY,eAAe,EAC3C,UAAU,QAAQ,UACpB,CAAC;EACH;EACA,UAAU,OAAO,UAAyB,WAAW;GACnD,iBAAiB,UAAU,OAAO;GAClC,MAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;GACrE,YAAY,YAAY,WAAW;GAEnC,MAAM,YAAY,WAAW,EAAE,UAAU,OAAO,UAAU,CAAC;EAC7D;EACA,kBAAkB,EAAE,YAAY,QAAQ,cAAc;GACpD,MAAM,UAAU,EAAE,UAAU,iBAAiB,QAAQ;GACrD,IAAI,SACF,YAAY,cAAc;IACxB,OAAO;IACP,MAAM;IACN;IACA,WACE,OAAO,WAAW,WAAW,SAAS,KAAK,UAAU,MAAM;IAC7D;GACF,CAAC;QAED,YAAY,cAAc;IACxB,OAAO;IACP,MAAM;IACN;IACA,QAAQ;IACR;GACF,CAAC;EAEL;EACA,mBAAmB,YACjB,gBAAgB,OAAO,QAAQ,YAAY,QAAQ,OAAO;EAC5D,GAAI,YAAY,EAAE,SAAS;EAC3B,GAAI,eAAe,EAAE,YAAY;EACjC,UAAU;GACR,aAAa;GACb,GAAG;GACH,GAAG;EACL;EACA;CACF,CAAC;CAED,OAAO;AACT"}

package/dist/ui/use-chat/useChatRuntime.d.ts CHANGED Viewed

@@ -1,14 +1,21 @@
-import { type UIMessage } from "@ai-sdk/react";
-import type { AssistantCloud } from "assistant-cloud";
-import type { AssistantRuntime } from "@assistant-ui/core";
-import { type AISDKRuntimeAdapter, type CustomToCreateMessageFunction } from "./useAISDKRuntime.js";
-import type { ChatInit } from "ai";
-export type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage = UIMessage> = ChatInit<UI_MESSAGE> & {
-    cloud?: AssistantCloud | undefined;
-    adapters?: AISDKRuntimeAdapter["adapters"] | undefined;
-    toCreateMessage?: CustomToCreateMessageFunction;
-    onResume?: AISDKRuntimeAdapter["onResume"];
-    suggestions?: AISDKRuntimeAdapter["suggestions"];
+import { AISDKRuntimeAdapter, CustomToCreateMessageFunction } from "./useAISDKRuntime.js";
+import { ChatInit } from "ai";
+import { AssistantRuntime } from "@assistant-ui/core";
+import { UIMessage as UIMessage$1 } from "@ai-sdk/react";
+import { AssistantCloud } from "assistant-cloud";
+//#region src/ui/use-chat/useChatRuntime.d.ts
+type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage$1 = UIMessage$1> = ChatInit<UI_MESSAGE> & {
+  cloud?: AssistantCloud | undefined;
+  adapters?: AISDKRuntimeAdapter["adapters"] | undefined;
+  toCreateMessage?: CustomToCreateMessageFunction;
+  onResume?: AISDKRuntimeAdapter["onResume"];
+  suggestions?: AISDKRuntimeAdapter["suggestions"];
 };
-export declare const useChatRuntime: <UI_MESSAGE extends UIMessage = UIMessage>({ cloud, ...options }?: UseChatRuntimeOptions<UI_MESSAGE>) => AssistantRuntime;
+declare const useChatRuntime: <UI_MESSAGE extends UIMessage$1 = UIMessage$1>({
+  cloud,
+  ...options
+}?: UseChatRuntimeOptions<UI_MESSAGE>) => AssistantRuntime;
+//#endregion
+export { UseChatRuntimeOptions, useChatRuntime };
 //# sourceMappingURL=useChatRuntime.d.ts.map

package/dist/ui/use-chat/useChatRuntime.d.ts.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"useChatRuntime.d.ts","~~sourceRoot~~":"","sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"~~names":[],"~~mappings":"~~AAEA~~,~~OAAO~~,~~EAAW~~,~~KAAK~~,~~SAAS~~,~~EAAE~~,~~MAAM~~,~~eAAe~~,~~CAAC;AACxD~~,~~OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC~~;~~AACtD~~,~~OAAO~~,~~KAAK~~,~~EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC~~;~~AAM3D~~,~~OAAO~~,~~EAEL~~,~~KAAK,mBAAmB,EACxB,KAAK,6BAA6B,EACnC,6BAA0B~~;~~AAC3B~~,~~OAAO~~,~~KAAK~~,~~EAAE,QAAQ,EAAiB,MAAM,IAAI,CAAC~~;~~AAKlD~~,~~MAAM~~,~~MAAM~~,~~qBAAqB,CAAC,UAAU,SAAS,SAAS,GAAG,SAAS,IACxE,QAAQ,CAAC,UAAU,CAAC,GAAG~~;~~IACrB~~,~~KAAK~~,~~CAAC~~,~~EAAE,cAAc,GAAG,SAAS,CAAC~~;~~IACnC,QAAQ,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,GAAG,SAAS,CAAC~~;~~IACvD~~,~~eAAe~~,~~CAAC~~,~~EAAE~~,~~6BAA6B~~,~~CAAC;IAChD~~,~~QAAQ,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,CAAC~~;~~IAC3C~~,~~WAAW,CAAC,EAAE,mBAAmB,CAAC,aAAa,CAAC,CAAC~~;~~CAClD~~,~~CAAC~~;~~AAyGJ~~,~~eAAO~~,~~MAAM~~,~~cAAc~~,~~GAAI~~,~~UAAU~~,~~SAAS,SAAS,GAAG,SAAS,EAAE,wBAGtE,qBAAqB,CAAC,UAAU,CAAM,KAAG,gBAU3C,CAAC~~"}
1	+ {"version":3,"file":"useChatRuntime.d.ts","names":[],"sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"mappings":";;;;;;;KAoBY,qBAAA,oBAAyC,WAAA,GAAY,WAAA,IAC/D,QAAA,CAAS,UAAA;EACP,KAAA,GAAQ,cAAA;EACR,QAAA,GAAW,mBAAA;EACX,eAAA,GAAkB,6BAAA;EAClB,QAAA,GAAW,mBAAA;EACX,WAAA,GAAc,mBAAA;AAAA;AAAA,cA0GL,cAAA,sBAAqC,WAAA,GAAY,WAAA;EAAW,KAAA;EAAA,GAAA;AAAA,IAGtE,qBAAA,CAAsB,UAAA,MAAmB,gBAAA"}