@assistant-ui/react-google-adk 0.0.8 → 0.0.11

package/dist/useAdkRuntime.js

@@ -1,286 +1,232 @@
-/// <reference types="@assistant-ui/core/store" />
-import { useEffect, useRef, useState } from "react";
-import { getExternalStoreMessages, } from "@assistant-ui/core";
-import { useCloudThreadListAdapter, useRemoteThreadListRuntime, useExternalMessageConverter, useExternalStoreRuntime, useToolInvocations, } from "@assistant-ui/core/react";
-import { useAui } from "@assistant-ui/store";
-import { useAdkMessages } from "./useAdkMessages.js";
 import { convertAdkMessage } from "./convertAdkMessages.js";
 import { symbolAdkRuntimeExtras } from "./hooks.js";
-import { v4 as uuidv4 } from "uuid";
+import { useAdkMessages } from "./useAdkMessages.js";
+import { v4 } from "uuid";
+import { useAui } from "@assistant-ui/store";
+import { useEffect, useRef, useState } from "react";
+import { getExternalStoreMessages } from "@assistant-ui/core";
+import { useCloudThreadListAdapter, useExternalMessageConverter, useExternalStoreRuntime, useRemoteThreadListRuntime, useToolInvocations } from "@assistant-ui/core/react";
+//#region src/useAdkRuntime.ts
 /** @internal — exported for unit tests. */
-export const getMessageContent = (msg) => {
-    const allContent = [
-        ...msg.content,
-        ...(msg.attachments?.flatMap((a) => a.content) ?? []),
-    ];
-    const content = allContent.map((part) => {
-        const type = part.type;
-        switch (type) {
-            case "text":
-                return { type: "text", text: part.text };
-            case "image":
-                return { type: "image_url", url: part.image };
-            case "file":
-                return {
-                    type: "file",
-                    mimeType: part.mimeType,
-                    data: part.data,
-                    ...(part.filename != null && { filename: part.filename }),
-                };
-            case "tool-call":
-                throw new Error("Tool call appends are not supported.");
-            default: {
-                const _exhaustiveCheck = type;
-                throw new Error(`Unsupported append message part type: ${_exhaustiveCheck}`);
-            }
-        }
-    });
-    if (content.length === 1 && content[0]?.type === "text") {
-        return content[0].text ?? "";
-    }
-    return content;
+const getMessageContent = (msg) => {
+	const content = [...msg.content, ...msg.attachments?.flatMap((a) => a.content) ?? []].map((part) => {
+		const type = part.type;
+		switch (type) {
+			case "text": return {
+				type: "text",
+				text: part.text
+			};
+			case "image": return {
+				type: "image_url",
+				url: part.image
+			};
+			case "file": return {
+				type: "file",
+				mimeType: part.mimeType,
+				data: part.data,
+				...part.filename != null && { filename: part.filename }
+			};
+			case "tool-call": throw new Error("Tool call appends are not supported.");
+			default: throw new Error(`Unsupported append message part type: ${type}`);
+		}
+	});
+	if (content.length === 1 && content[0]?.type === "text") return content[0].text ?? "";
+	return content;
 };
 /** @internal — exported for unit tests. */
-export const getPendingToolCalls = (messages) => {
-    const pending = new Map();
-    for (const msg of messages) {
-        if (msg.type === "ai" && msg.tool_calls) {
-            for (const tc of msg.tool_calls) {
-                pending.set(tc.id, tc);
-            }
-        }
-        if (msg.type === "tool") {
-            pending.delete(msg.tool_call_id);
-        }
-    }
-    return [...pending.values()];
+const getPendingToolCalls = (messages) => {
+	const pending = /* @__PURE__ */ new Map();
+	for (const msg of messages) {
+		if (msg.type === "ai" && msg.tool_calls) for (const tc of msg.tool_calls) pending.set(tc.id, tc);
+		if (msg.type === "tool") pending.delete(msg.tool_call_id);
+	}
+	return [...pending.values()];
 };
 /**
- * @internal — exported for unit tests.
- *
- * Returns `{cancelled: true}` tool responses for pending tool calls when the
- * user sends a new turn, EXCEPT for HITL interrupts marked via
- * `long_running_tool_ids` (`adk_request_input`, `adk_request_confirmation`,
- * `adk_request_credential`). Those must be answered through a dedicated tool
- * UI + submit helper, not auto-cancelled.
- */
-export const getPendingCancellations = (messages, longRunningToolIds) => {
-    const longRunningSet = new Set(longRunningToolIds);
-    return getPendingToolCalls(messages)
-        .filter((t) => !longRunningSet.has(t.id))
-        .map((t) => ({
-        id: uuidv4(),
-        type: "tool",
-        name: t.name,
-        tool_call_id: t.id,
-        content: JSON.stringify({ cancelled: true }),
-        status: "error",
-    }));
+* @internal — exported for unit tests.
+*
+* Returns `{cancelled: true}` tool responses for pending tool calls when the
+* user sends a new turn, EXCEPT for HITL interrupts marked via
+* `long_running_tool_ids` (`adk_request_input`, `adk_request_confirmation`,
+* `adk_request_credential`). Those must be answered through a dedicated tool
+* UI + submit helper, not auto-cancelled.
+*/
+const getPendingCancellations = (messages, longRunningToolIds) => {
+	const longRunningSet = new Set(longRunningToolIds);
+	return getPendingToolCalls(messages).filter((t) => !longRunningSet.has(t.id)).map((t) => ({
+		id: v4(),
+		type: "tool",
+		name: t.name,
+		tool_call_id: t.id,
+		content: JSON.stringify({ cancelled: true }),
+		status: "error"
+	}));
 };
 const truncateAdkMessages = (threadMessages, parentId) => {
-    if (parentId === null)
-        return [];
-    const parentIndex = threadMessages.findIndex((m) => m.id === parentId);
-    if (parentIndex === -1)
-        return [];
-    const truncated = [];
-    for (let i = 0; i <= parentIndex && i < threadMessages.length; i++) {
-        truncated.push(...getExternalStoreMessages(threadMessages[i]));
-    }
-    return truncated;
+	if (parentId === null) return [];
+	const parentIndex = threadMessages.findIndex((m) => m.id === parentId);
+	if (parentIndex === -1) return [];
+	const truncated = [];
+	for (let i = 0; i <= parentIndex && i < threadMessages.length; i++) truncated.push(...getExternalStoreMessages(threadMessages[i]));
+	return truncated;
 };
-const useAdkRuntimeImpl = ({ autoCancelPendingToolCalls, adapters: { attachments, dictation, feedback, speech } = {}, unstable_allowCancellation, stream, load, getCheckpointId, eventHandlers, }) => {
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const aui = useAui();
-    const { messages, stateDelta, agentInfo, longRunningToolIds, artifactDelta, toolConfirmations, authRequests, escalated, messageMetadata, sendMessage, cancel, replaceMessages,
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-     } = useAdkMessages({
-        stream,
-        ...(eventHandlers && { eventHandlers }),
-    });
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const [isRunning, setIsRunning] = useState(false);
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const [toolStatuses, setToolStatuses] = useState({});
-    const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
-    const effectiveIsRunning = isRunning || hasExecutingTools;
-    const handleSendMessage = async (msgs, config) => {
-        try {
-            setIsRunning(true);
-            await sendMessage(msgs, config);
-        }
-        finally {
-            setIsRunning(false);
-        }
-    };
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const threadMessages = useExternalMessageConverter({
-        callback: convertAdkMessage,
-        messages,
-        isRunning: effectiveIsRunning,
-    });
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const threadMessagesRef = useRef(threadMessages);
-    threadMessagesRef.current = threadMessages;
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const [runtimeRef] = useState(() => ({
-        get current() {
-            return runtime;
-        },
-    }));
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const toolInvocations = useToolInvocations({
-        state: { messages: threadMessages, isRunning: effectiveIsRunning },
-        getTools: () => runtimeRef.current.thread.getModelContext().tools,
-        onResult: (command) => {
-            if (command.type === "add-tool-result") {
-                void handleSendMessage([
-                    {
-                        id: uuidv4(),
-                        type: "tool",
-                        name: command.toolName,
-                        tool_call_id: command.toolCallId,
-                        content: JSON.stringify(command.result),
-                        status: command.isError ? "error" : "success",
-                    },
-                ], {});
-            }
-        },
-        setToolStatuses,
-    });
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const runtime = useExternalStoreRuntime({
-        isRunning: effectiveIsRunning,
-        messages: threadMessages,
-        adapters: { attachments, dictation, feedback, speech },
-        extras: {
-            [symbolAdkRuntimeExtras]: true,
-            agentInfo,
-            stateDelta,
-            artifactDelta,
-            longRunningToolIds,
-            toolConfirmations,
-            authRequests,
-            escalated,
-            messageMetadata,
-            send: handleSendMessage,
-        },
-        onNew: async (msg) => {
-            await toolInvocations.abort();
-            const cancellations = autoCancelPendingToolCalls !== false
-                ? getPendingCancellations(messages, longRunningToolIds)
-                : [];
-            return handleSendMessage([
-                ...cancellations,
-                {
-                    id: uuidv4(),
-                    type: "human",
-                    content: getMessageContent(msg),
-                },
-            ], { runConfig: msg.runConfig });
-        },
-        onEdit: getCheckpointId
-            ? async (msg) => {
-                await toolInvocations.abort();
-                const truncated = truncateAdkMessages(threadMessagesRef.current, msg.parentId);
-                replaceMessages(truncated);
-                const externalId = aui.threadListItem().getState().externalId;
-                const checkpointId = externalId
-                    ? await getCheckpointId(externalId, truncated)
-                    : null;
-                return handleSendMessage([
-                    {
-                        id: uuidv4(),
-                        type: "human",
-                        content: getMessageContent(msg),
-                    },
-                ], {
-                    runConfig: msg.runConfig,
-                    ...(checkpointId && { checkpointId }),
-                });
-            }
-            : undefined,
-        onReload: getCheckpointId
-            ? async (parentId, config) => {
-                await toolInvocations.abort();
-                const truncated = truncateAdkMessages(threadMessagesRef.current, parentId);
-                replaceMessages(truncated);
-                const externalId = aui.threadListItem().getState().externalId;
-                const checkpointId = externalId
-                    ? await getCheckpointId(externalId, truncated)
-                    : null;
-                return handleSendMessage([], {
-                    runConfig: config.runConfig,
-                    ...(checkpointId && { checkpointId }),
-                });
-            }
-            : undefined,
-        onAddToolResult: async ({ toolCallId, toolName, result, isError, artifact, }) => {
-            await handleSendMessage([
-                {
-                    id: uuidv4(),
-                    type: "tool",
-                    name: toolName,
-                    tool_call_id: toolCallId,
-                    content: JSON.stringify(result),
-                    artifact,
-                    status: isError ? "error" : "success",
-                },
-            ], {});
-        },
-        onResumeToolCall: (options) => toolInvocations.resume(options.toolCallId, options.payload),
-        onCancel: unstable_allowCancellation
-            ? async () => {
-                cancel();
-                await toolInvocations.abort();
-            }
-            : undefined,
-    });
-    {
-        // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-        const loadRef = useRef(load);
-        // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-        useEffect(() => {
-            loadRef.current = load;
-        });
-        // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-        useEffect(() => {
-            const loadFn = loadRef.current;
-            if (!loadFn)
-                return;
-            const externalId = aui.threadListItem().getState().externalId;
-            if (externalId == null)
-                return;
-            loadFn(externalId).then(({ messages: msgs }) => {
-                replaceMessages(msgs);
-            }, (e) => {
-                console.warn("Failed to load ADK session:", e);
-            });
-        }, [aui, replaceMessages]);
-    }
-    return runtime;
+const useAdkRuntimeImpl = ({ autoCancelPendingToolCalls, adapters: { attachments, dictation, feedback, speech } = {}, unstable_allowCancellation, stream, load, getCheckpointId, eventHandlers }) => {
+	const aui = useAui();
+	const { messages, stateDelta, agentInfo, longRunningToolIds, artifactDelta, toolConfirmations, authRequests, escalated, messageMetadata, sendMessage, cancel, replaceMessages } = useAdkMessages({
+		stream,
+		...eventHandlers && { eventHandlers }
+	});
+	const [isRunning, setIsRunning] = useState(false);
+	const [toolStatuses, setToolStatuses] = useState({});
+	const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
+	const effectiveIsRunning = isRunning || hasExecutingTools;
+	const handleSendMessage = async (msgs, config) => {
+		try {
+			setIsRunning(true);
+			await sendMessage(msgs, config);
+		} finally {
+			setIsRunning(false);
+		}
+	};
+	const threadMessages = useExternalMessageConverter({
+		callback: convertAdkMessage,
+		messages,
+		isRunning: effectiveIsRunning
+	});
+	const threadMessagesRef = useRef(threadMessages);
+	threadMessagesRef.current = threadMessages;
+	const [runtimeRef] = useState(() => ({ get current() {
+		return runtime;
+	} }));
+	const toolInvocations = useToolInvocations({
+		state: {
+			messages: threadMessages,
+			isRunning: effectiveIsRunning
+		},
+		getTools: () => runtimeRef.current.thread.getModelContext().tools,
+		onResult: (command) => {
+			if (command.type === "add-tool-result") handleSendMessage([{
+				id: v4(),
+				type: "tool",
+				name: command.toolName,
+				tool_call_id: command.toolCallId,
+				content: JSON.stringify(command.result),
+				status: command.isError ? "error" : "success"
+			}], {});
+		},
+		setToolStatuses
+	});
+	const runtime = useExternalStoreRuntime({
+		isRunning: effectiveIsRunning,
+		messages: threadMessages,
+		adapters: {
+			attachments,
+			dictation,
+			feedback,
+			speech
+		},
+		extras: {
+			[symbolAdkRuntimeExtras]: true,
+			agentInfo,
+			stateDelta,
+			artifactDelta,
+			longRunningToolIds,
+			toolConfirmations,
+			authRequests,
+			escalated,
+			messageMetadata,
+			send: handleSendMessage
+		},
+		onNew: async (msg) => {
+			await toolInvocations.abort();
+			return handleSendMessage([...autoCancelPendingToolCalls !== false ? getPendingCancellations(messages, longRunningToolIds) : [], {
+				id: v4(),
+				type: "human",
+				content: getMessageContent(msg)
+			}], { runConfig: msg.runConfig });
+		},
+		onEdit: getCheckpointId ? async (msg) => {
+			await toolInvocations.abort();
+			const truncated = truncateAdkMessages(threadMessagesRef.current, msg.parentId);
+			replaceMessages(truncated);
+			const externalId = aui.threadListItem().getState().externalId;
+			const checkpointId = externalId ? await getCheckpointId(externalId, truncated) : null;
+			return handleSendMessage([{
+				id: v4(),
+				type: "human",
+				content: getMessageContent(msg)
+			}], {
+				runConfig: msg.runConfig,
+				...checkpointId && { checkpointId }
+			});
+		} : void 0,
+		onReload: getCheckpointId ? async (parentId, config) => {
+			await toolInvocations.abort();
+			const truncated = truncateAdkMessages(threadMessagesRef.current, parentId);
+			replaceMessages(truncated);
+			const externalId = aui.threadListItem().getState().externalId;
+			const checkpointId = externalId ? await getCheckpointId(externalId, truncated) : null;
+			return handleSendMessage([], {
+				runConfig: config.runConfig,
+				...checkpointId && { checkpointId }
+			});
+		} : void 0,
+		onAddToolResult: async ({ toolCallId, toolName, result, isError, artifact }) => {
+			await handleSendMessage([{
+				id: v4(),
+				type: "tool",
+				name: toolName,
+				tool_call_id: toolCallId,
+				content: JSON.stringify(result),
+				artifact,
+				status: isError ? "error" : "success"
+			}], {});
+		},
+		onResumeToolCall: (options) => toolInvocations.resume(options.toolCallId, options.payload),
+		onCancel: unstable_allowCancellation ? async () => {
+			cancel();
+			await toolInvocations.abort();
+		} : void 0
+	});
+	{
+		const loadRef = useRef(load);
+		useEffect(() => {
+			loadRef.current = load;
+		});
+		useEffect(() => {
+			const loadFn = loadRef.current;
+			if (!loadFn) return;
+			const externalId = aui.threadListItem().getState().externalId;
+			if (externalId == null) return;
+			loadFn(externalId).then(({ messages: msgs }) => {
+				replaceMessages(msgs);
+			}, (e) => {
+				console.warn("Failed to load ADK session:", e);
+			});
+		}, [aui, replaceMessages]);
+	}
+	return runtime;
 };
-export const useAdkRuntime = ({ cloud, sessionAdapter, create, delete: deleteFn, ...options }) => {
-    const aui = useAui();
-    const cloudAdapter = useCloudThreadListAdapter({
-        cloud,
-        create: async () => {
-            if (create)
-                return create();
-            if (aui.threadListItem.source)
-                return aui.threadListItem().initialize();
-            return { externalId: undefined };
-        },
-        delete: deleteFn,
-    });
-    const adapter = sessionAdapter ?? cloudAdapter;
-    return useRemoteThreadListRuntime({
-        runtimeHook: function RuntimeHook() {
-            // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-            return useAdkRuntimeImpl(options);
-        },
-        adapter,
-        allowNesting: true,
-    });
+const useAdkRuntime = ({ cloud, sessionAdapter, create, delete: deleteFn, ...options }) => {
+	const aui = useAui();
+	const cloudAdapter = useCloudThreadListAdapter({
+		cloud,
+		create: async () => {
+			if (create) return create();
+			if (aui.threadListItem.source) return aui.threadListItem().initialize();
+			return { externalId: void 0 };
+		},
+		delete: deleteFn
+	});
+	return useRemoteThreadListRuntime({
+		runtimeHook: function RuntimeHook() {
+			return useAdkRuntimeImpl(options);
+		},
+		adapter: sessionAdapter ?? cloudAdapter,
+		allowNesting: true
+	});
 };
+//#endregion
+export { getMessageContent, getPendingCancellations, getPendingToolCalls, useAdkRuntime };
+
 //# sourceMappingURL=useAdkRuntime.js.map

package/dist/useAdkRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAdkRuntime.js","sourceRoot":"","sources":["../src/useAdkRuntime.ts"],"names":[],"mappings":"AAAA,kDAAkD;AAClD,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACpD,OAAO,EACL,wBAAwB,GAOzB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAEL,yBAAyB,EACzB,0BAA0B,EAC1B,2BAA2B,EAC3B,uBAAuB,EACvB,kBAAkB,GACnB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAW7C,OAAO,EAAE,cAAc,EAAE,4BAAyB;AAClD,OAAO,EAAE,iBAAiB,EAAE,gCAA6B;AACzD,OAAO,EAAE,sBAAsB,EAAyB,mBAAgB;AACxE,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AAEpC,2CAA2C;AAC3C,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,GAAkB,EAAE,EAAE;IACtD,MAAM,UAAU,GAAG;QACjB,GAAG,GAAG,CAAC,OAAO;QACd,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;KACtD,CAAC;IACF,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACtC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACvB,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,MAAM;gBACT,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACpD,KAAK,OAAO;gBACV,OAAO,EAAE,IAAI,EAAE,WAAoB,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;YACzD,KAAK,MAAM;gBACT,OAAO;oBACL,IAAI,EAAE,MAAe;oBACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;iBAC1D,CAAC;YAEJ,KAAK,WAAW;gBACd,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAE1D,OAAO,CAAC,CAAC,CAAC;gBACR,MAAM,gBAAgB,GACpB,IAAI,CAAC;gBACP,MAAM,IAAI,KAAK,CACb,yCAAyC,gBAAgB,EAAE,CAC5D,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;QACxD,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;IAC/B,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,2CAA2C;AAC3C,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,QAAsB,EAAE,EAAE;IAC5D,MAAM,OAAO,GAAG,IAAI,GAAG,EAAwC,CAAC;IAChE,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,IAAI,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;YACxC,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;gBAChC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;QACD,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACxB,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IACD,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;AAC/B,CAAC,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,CACrC,QAAsB,EACtB,kBAAqC,EACC,EAAE;IACxC,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,kBAAkB,CAAC,CAAC;IACnD,OAAO,mBAAmB,CAAC,QAAQ,CAAC;SACjC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;SACxC,GAAG,CACF,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC;QACC,EAAE,EAAE,MAAM,EAAE;QACZ,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,YAAY,EAAE,CAAC,CAAC,EAAE;QAClB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QAC5C,MAAM,EAAE,OAAO;KAChB,CAAyC,CAC7C,CAAC;AACN,CAAC,CAAC;AAEF,MAAM,mBAAmB,GAAG,CAC1B,cAAwC,EACxC,QAAuB,EACT,EAAE;IAChB,IAAI,QAAQ,KAAK,IAAI;QAAE,OAAO,EAAE,CAAC;IACjC,MAAM,WAAW,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,QAAQ,CAAC,CAAC;IACvE,IAAI,WAAW,KAAK,CAAC,CAAC;QAAE,OAAO,EAAE,CAAC;IAClC,MAAM,SAAS,GAAiB,EAAE,CAAC;IACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,WAAW,IAAI,CAAC,GAAG,cAAc,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACnE,SAAS,CAAC,IAAI,CAAC,GAAG,wBAAwB,CAAa,cAAc,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC;IAC9E,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC;AAoCF,MAAM,iBAAiB,GAAG,CAAC,EACzB,0BAA0B,EAC1B,QAAQ,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,EAC3D,0BAA0B,EAC1B,MAAM,EACN,IAAI,EACJ,eAAe,EACf,aAAa,GACQ,EAAE,EAAE;IACzB,6FAA6F;IAC7F,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IACrB,MAAM,EACJ,QAAQ,EACR,UAAU,EACV,SAAS,EACT,kBAAkB,EAClB,aAAa,EACb,iBAAiB,EACjB,YAAY,EACZ,SAAS,EACT,eAAe,EACf,WAAW,EACX,MAAM,EACN,eAAe;IACf,6FAA6F;MAC9F,GAAG,cAAc,CAAC;QACjB,MAAM;QACN,GAAG,CAAC,aAAa,IAAI,EAAE,aAAa,EAAE,CAAC;KACxC,CAAC,CAAC;IAEH,6FAA6F;IAC7F,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAClD,6FAA6F;IAC7F,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAE9C,EAAE,CAAC,CAAC;IACN,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,kBAAkB,GAAG,SAAS,IAAI,iBAAiB,CAAC;IAE1D,MAAM,iBAAiB,GAAG,KAAK,EAC7B,IAAkB,EAClB,MAA4B,EAC5B,EAAE;QACF,IAAI,CAAC;YACH,YAAY,CAAC,IAAI,CAAC,CAAC;YACnB,MAAM,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAClC,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC,CAAC;IAEF,6FAA6F;IAC7F,MAAM,cAAc,GAAG,2BAA2B,CAAC;QACjD,QAAQ,EAAE,iBAAiB;QAC3B,QAAQ;QACR,SAAS,EAAE,kBAAkB;KAC9B,CAAC,CAAC;IAEH,6FAA6F;IAC7F,MAAM,iBAAiB,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;IACjD,iBAAiB,CAAC,OAAO,GAAG,cAAc,CAAC;IAE3C,6FAA6F;IAC7F,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,6FAA6F;IAC7F,MAAM,eAAe,GAAG,kBAAkB,CAAC;QACzC,KAAK,EAAE,EAAE,QAAQ,EAAE,cAAc,EAAE,SAAS,EAAE,kBAAkB,EAAE;QAClE,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,KAAK,iBAAiB,CACpB;oBACE;wBACE,EAAE,EAAE,MAAM,EAAE;wBACZ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,OAAO,CAAC,QAAQ;wBACtB,YAAY,EAAE,OAAO,CAAC,UAAU;wBAChC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC;wBACvC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;qBAC9C;iBACF,EACD,EAAE,CACH,CAAC;YACJ,CAAC;QACH,CAAC;QACD,eAAe;KAChB,CAAC,CAAC;IAEH,6FAA6F;IAC7F,MAAM,OAAO,GAAG,uBAAuB,CAAC;QACtC,SAAS,EAAE,kBAAkB;QAC7B,QAAQ,EAAE,cAAc;QACxB,QAAQ,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE;QACtD,MAAM,EAAE;YACN,CAAC,sBAAsB,CAAC,EAAE,IAAI;YAC9B,SAAS;YACT,UAAU;YACV,aAAa;YACb,kBAAkB;YAClB,iBAAiB;YACjB,YAAY;YACZ,SAAS;YACT,eAAe;YACf,IAAI,EAAE,iBAAiB;SACG;QAC5B,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACnB,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;YAE9B,MAAM,aAAa,GACjB,0BAA0B,KAAK,KAAK;gBAClC,CAAC,CAAC,uBAAuB,CAAC,QAAQ,EAAE,kBAAkB,CAAC;gBACvD,CAAC,CAAC,EAAE,CAAC;YAET,OAAO,iBAAiB,CACtB;gBACE,GAAG,aAAa;gBAChB;oBACE,EAAE,EAAE,MAAM,EAAE;oBACZ,IAAI,EAAE,OAAO;oBACb,OAAO,EAAE,iBAAiB,CAAC,GAAG,CAAC;iBAChC;aACF,EACD,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,CAC7B,CAAC;QACJ,CAAC;QACD,MAAM,EAAE,eAAe;YACrB,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;gBACZ,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;gBAC9B,MAAM,SAAS,GAAG,mBAAmB,CACnC,iBAAiB,CAAC,OAAO,EACzB,GAAG,CAAC,QAAQ,CACb,CAAC;gBACF,eAAe,CAAC,SAAS,CAAC,CAAC;gBAC3B,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC,QAAQ,EAAE,CAAC,UAAU,CAAC;gBAC9D,MAAM,YAAY,GAAG,UAAU;oBAC7B,CAAC,CAAC,MAAM,eAAe,CAAC,UAAU,EAAE,SAAS,CAAC;oBAC9C,CAAC,CAAC,IAAI,CAAC;gBACT,OAAO,iBAAiB,CACtB;oBACE;wBACE,EAAE,EAAE,MAAM,EAAE;wBACZ,IAAI,EAAE,OAAO;wBACb,OAAO,EAAE,iBAAiB,CAAC,GAAG,CAAC;qBAChC;iBACF,EACD;oBACE,SAAS,EAAE,GAAG,CAAC,SAAS;oBACxB,GAAG,CAAC,YAAY,IAAI,EAAE,YAAY,EAAE,CAAC;iBACtC,CACF,CAAC;YACJ,CAAC;YACH,CAAC,CAAC,SAAS;QACb,QAAQ,EAAE,eAAe;YACvB,CAAC,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE;gBACzB,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;gBAC9B,MAAM,SAAS,GAAG,mBAAmB,CACnC,iBAAiB,CAAC,OAAO,EACzB,QAAQ,CACT,CAAC;gBACF,eAAe,CAAC,SAAS,CAAC,CAAC;gBAC3B,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC,QAAQ,EAAE,CAAC,UAAU,CAAC;gBAC9D,MAAM,YAAY,GAAG,UAAU;oBAC7B,CAAC,CAAC,MAAM,eAAe,CAAC,UAAU,EAAE,SAAS,CAAC;oBAC9C,CAAC,CAAC,IAAI,CAAC;gBACT,OAAO,iBAAiB,CAAC,EAAE,EAAE;oBAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;oBAC3B,GAAG,CAAC,YAAY,IAAI,EAAE,YAAY,EAAE,CAAC;iBACtC,CAAC,CAAC;YACL,CAAC;YACH,CAAC,CAAC,SAAS;QACb,eAAe,EAAE,KAAK,EAAE,EACtB,UAAU,EACV,QAAQ,EACR,MAAM,EACN,OAAO,EACP,QAAQ,GACT,EAAE,EAAE;YACH,MAAM,iBAAiB,CACrB;gBACE;oBACE,EAAE,EAAE,MAAM,EAAE;oBACZ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ;oBACd,YAAY,EAAE,UAAU;oBACxB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC/B,QAAQ;oBACR,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;iBACtC;aACF,EACD,EAAE,CACH,CAAC;QACJ,CAAC;QACD,gBAAgB,EAAE,CAAC,OAAO,EAAE,EAAE,CAC5B,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC;QAC7D,QAAQ,EAAE,0BAA0B;YAClC,CAAC,CAAC,KAAK,IAAI,EAAE;gBACT,MAAM,EAAE,CAAC;gBACT,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;YAChC,CAAC;YACH,CAAC,CAAC,SAAS;KACd,CAAC,CAAC;IAEH,CAAC;QACC,6FAA6F;QAC7F,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;QAC7B,6FAA6F;QAC7F,SAAS,CAAC,GAAG,EAAE;YACb,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,6FAA6F;QAC7F,SAAS,CAAC,GAAG,EAAE;YACb,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;YAC/B,IAAI,CAAC,MAAM;gBAAE,OAAO;YAEpB,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC,QAAQ,EAAE,CAAC,UAAU,CAAC;YAC9D,IAAI,UAAU,IAAI,IAAI;gBAAE,OAAO;YAE/B,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CACrB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE;gBACrB,eAAe,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC,EACD,CAAC,CAAC,EAAE,EAAE;gBACJ,OAAO,CAAC,IAAI,CAAC,6BAA6B,EAAE,CAAC,CAAC,CAAC;YACjD,CAAC,CACF,CAAC;QACJ,CAAC,EAAE,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,EAC5B,KAAK,EACL,cAAc,EACd,MAAM,EACN,MAAM,EAAE,QAAQ,EAChB,GAAG,OAAO,EACW,EAAE,EAAE;IACzB,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IACrB,MAAM,YAAY,GAAG,yBAAyB,CAAC;QAC7C,KAAK;QACL,MAAM,EAAE,KAAK,IAAI,EAAE;YACjB,IAAI,MAAM;gBAAE,OAAO,MAAM,EAAE,CAAC;YAC5B,IAAI,GAAG,CAAC,cAAc,CAAC,MAAM;gBAAE,OAAO,GAAG,CAAC,cAAc,EAAE,CAAC,UAAU,EAAE,CAAC;YACxE,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;QACnC,CAAC;QACD,MAAM,EAAE,QAAQ;KACjB,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,cAAc,IAAI,YAAY,CAAC;IAE/C,OAAO,0BAA0B,CAAC;QAChC,WAAW,EAAE,SAAS,WAAW;YAC/B,6FAA6F;YAC7F,OAAO,iBAAiB,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC;QACD,OAAO;QACP,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;AACL,CAAC,CAAC"}
+{"version":3,"file":"useAdkRuntime.js","names":["_exhaustiveCheck","uuidv4"],"sources":["../src/useAdkRuntime.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/store\" />\nimport { useEffect, useRef, useState } from \"react\";\nimport {\n  getExternalStoreMessages,\n  type AttachmentAdapter,\n  type DictationAdapter,\n  type FeedbackAdapter,\n  type SpeechSynthesisAdapter,\n  type AppendMessage,\n  type ThreadMessage,\n} from \"@assistant-ui/core\";\nimport {\n  type ToolExecutionStatus,\n  useCloudThreadListAdapter,\n  useRemoteThreadListRuntime,\n  useExternalMessageConverter,\n  useExternalStoreRuntime,\n  useToolInvocations,\n} from \"@assistant-ui/core/react\";\nimport { useAui } from \"@assistant-ui/store\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport type { RemoteThreadListAdapter } from \"@assistant-ui/core\";\nimport type {\n  AdkMessage,\n  AdkSendMessageConfig,\n  AdkStreamCallback,\n  OnAdkErrorCallback,\n  OnAdkCustomEventCallback,\n  OnAdkAgentTransferCallback,\n} from \"./types\";\nimport { useAdkMessages } from \"./useAdkMessages\";\nimport { convertAdkMessage } from \"./convertAdkMessages\";\nimport { symbolAdkRuntimeExtras, type AdkRuntimeExtras } from \"./hooks\";\nimport { v4 as uuidv4 } from \"uuid\";\n\n/** @internal — exported for unit tests. */\nexport const getMessageContent = (msg: AppendMessage) => {\n  const allContent = [\n    ...msg.content,\n    ...(msg.attachments?.flatMap((a) => a.content) ?? []),\n  ];\n  const content = allContent.map((part) => {\n    const type = part.type;\n    switch (type) {\n      case \"text\":\n        return { type: \"text\" as const, text: part.text };\n      case \"image\":\n        return { type: \"image_url\" as const, url: part.image };\n      case \"file\":\n        return {\n          type: \"file\" as const,\n          mimeType: part.mimeType,\n          data: part.data,\n          ...(part.filename != null && { filename: part.filename }),\n        };\n\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported.\");\n\n      default: {\n        const _exhaustiveCheck:\n          | \"reasoning\"\n          | \"source\"\n          | \"audio\"\n          | \"data\"\n          | \"generative-ui\" = type;\n        throw new Error(\n          `Unsupported append message part type: ${_exhaustiveCheck}`,\n        );\n      }\n    }\n  });\n\n  if (content.length === 1 && content[0]?.type === \"text\") {\n    return content[0].text ?? \"\";\n  }\n\n  return content;\n};\n\n/** @internal — exported for unit tests. */\nexport const getPendingToolCalls = (messages: AdkMessage[]) => {\n  const pending = new Map<string, { id: string; name: string }>();\n  for (const msg of messages) {\n    if (msg.type === \"ai\" && msg.tool_calls) {\n      for (const tc of msg.tool_calls) {\n        pending.set(tc.id, tc);\n      }\n    }\n    if (msg.type === \"tool\") {\n      pending.delete(msg.tool_call_id);\n    }\n  }\n  return [...pending.values()];\n};\n\n/**\n * @internal — exported for unit tests.\n *\n * Returns `{cancelled: true}` tool responses for pending tool calls when the\n * user sends a new turn, EXCEPT for HITL interrupts marked via\n * `long_running_tool_ids` (`adk_request_input`, `adk_request_confirmation`,\n * `adk_request_credential`). Those must be answered through a dedicated tool\n * UI + submit helper, not auto-cancelled.\n */\nexport const getPendingCancellations = (\n  messages: AdkMessage[],\n  longRunningToolIds: readonly string[],\n): Array<AdkMessage & { type: \"tool\" }> => {\n  const longRunningSet = new Set(longRunningToolIds);\n  return getPendingToolCalls(messages)\n    .filter((t) => !longRunningSet.has(t.id))\n    .map(\n      (t) =>\n        ({\n          id: uuidv4(),\n          type: \"tool\",\n          name: t.name,\n          tool_call_id: t.id,\n          content: JSON.stringify({ cancelled: true }),\n          status: \"error\",\n        }) satisfies AdkMessage & { type: \"tool\" },\n    );\n};\n\nconst truncateAdkMessages = (\n  threadMessages: readonly ThreadMessage[],\n  parentId: string | null,\n): AdkMessage[] => {\n  if (parentId === null) return [];\n  const parentIndex = threadMessages.findIndex((m) => m.id === parentId);\n  if (parentIndex === -1) return [];\n  const truncated: AdkMessage[] = [];\n  for (let i = 0; i <= parentIndex && i < threadMessages.length; i++) {\n    truncated.push(...getExternalStoreMessages<AdkMessage>(threadMessages[i]!));\n  }\n  return truncated;\n};\n\nexport type UseAdkRuntimeOptions = {\n  stream: AdkStreamCallback;\n  autoCancelPendingToolCalls?: boolean | undefined;\n  unstable_allowCancellation?: boolean | undefined;\n  getCheckpointId?: (\n    threadId: string,\n    parentMessages: AdkMessage[],\n  ) => Promise<string | null>;\n  load?: (threadId: string) => Promise<{ messages: AdkMessage[] }>;\n  create?: () => Promise<{ externalId: string }>;\n  delete?: (threadId: string) => Promise<void>;\n  adapters?:\n    | {\n        attachments?: AttachmentAdapter;\n        speech?: SpeechSynthesisAdapter;\n        dictation?: DictationAdapter;\n        feedback?: FeedbackAdapter;\n      }\n    | undefined;\n  eventHandlers?:\n    | {\n        onError?: OnAdkErrorCallback;\n        onCustomEvent?: OnAdkCustomEventCallback;\n        onAgentTransfer?: OnAdkAgentTransferCallback;\n      }\n    | undefined;\n  cloud?: AssistantCloud | undefined;\n  /**\n   * A `RemoteThreadListAdapter` to use instead of the cloud adapter.\n   * Use with `createAdkSessionAdapter` for ADK session-backed persistence.\n   */\n  sessionAdapter?: RemoteThreadListAdapter | undefined;\n};\n\nconst useAdkRuntimeImpl = ({\n  autoCancelPendingToolCalls,\n  adapters: { attachments, dictation, feedback, speech } = {},\n  unstable_allowCancellation,\n  stream,\n  load,\n  getCheckpointId,\n  eventHandlers,\n}: UseAdkRuntimeOptions) => {\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const aui = useAui();\n  const {\n    messages,\n    stateDelta,\n    agentInfo,\n    longRunningToolIds,\n    artifactDelta,\n    toolConfirmations,\n    authRequests,\n    escalated,\n    messageMetadata,\n    sendMessage,\n    cancel,\n    replaceMessages,\n    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  } = useAdkMessages({\n    stream,\n    ...(eventHandlers && { eventHandlers }),\n  });\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const [isRunning, setIsRunning] = useState(false);\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const effectiveIsRunning = isRunning || hasExecutingTools;\n\n  const handleSendMessage = async (\n    msgs: AdkMessage[],\n    config: AdkSendMessageConfig,\n  ) => {\n    try {\n      setIsRunning(true);\n      await sendMessage(msgs, config);\n    } finally {\n      setIsRunning(false);\n    }\n  };\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const threadMessages = useExternalMessageConverter({\n    callback: convertAdkMessage,\n    messages,\n    isRunning: effectiveIsRunning,\n  });\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const threadMessagesRef = useRef(threadMessages);\n  threadMessagesRef.current = threadMessages;\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const [runtimeRef] = useState(() => ({\n    get current() {\n      return runtime;\n    },\n  }));\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const toolInvocations = useToolInvocations({\n    state: { messages: threadMessages, isRunning: effectiveIsRunning },\n    getTools: () => runtimeRef.current.thread.getModelContext().tools,\n    onResult: (command) => {\n      if (command.type === \"add-tool-result\") {\n        void handleSendMessage(\n          [\n            {\n              id: uuidv4(),\n              type: \"tool\",\n              name: command.toolName,\n              tool_call_id: command.toolCallId,\n              content: JSON.stringify(command.result),\n              status: command.isError ? \"error\" : \"success\",\n            },\n          ],\n          {},\n        );\n      }\n    },\n    setToolStatuses,\n  });\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const runtime = useExternalStoreRuntime({\n    isRunning: effectiveIsRunning,\n    messages: threadMessages,\n    adapters: { attachments, dictation, feedback, speech },\n    extras: {\n      [symbolAdkRuntimeExtras]: true,\n      agentInfo,\n      stateDelta,\n      artifactDelta,\n      longRunningToolIds,\n      toolConfirmations,\n      authRequests,\n      escalated,\n      messageMetadata,\n      send: handleSendMessage,\n    } satisfies AdkRuntimeExtras,\n    onNew: async (msg) => {\n      await toolInvocations.abort();\n\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingCancellations(messages, longRunningToolIds)\n          : [];\n\n      return handleSendMessage(\n        [\n          ...cancellations,\n          {\n            id: uuidv4(),\n            type: \"human\",\n            content: getMessageContent(msg),\n          },\n        ],\n        { runConfig: msg.runConfig },\n      );\n    },\n    onEdit: getCheckpointId\n      ? async (msg) => {\n          await toolInvocations.abort();\n          const truncated = truncateAdkMessages(\n            threadMessagesRef.current,\n            msg.parentId,\n          );\n          replaceMessages(truncated);\n          const externalId = aui.threadListItem().getState().externalId;\n          const checkpointId = externalId\n            ? await getCheckpointId(externalId, truncated)\n            : null;\n          return handleSendMessage(\n            [\n              {\n                id: uuidv4(),\n                type: \"human\",\n                content: getMessageContent(msg),\n              },\n            ],\n            {\n              runConfig: msg.runConfig,\n              ...(checkpointId && { checkpointId }),\n            },\n          );\n        }\n      : undefined,\n    onReload: getCheckpointId\n      ? async (parentId, config) => {\n          await toolInvocations.abort();\n          const truncated = truncateAdkMessages(\n            threadMessagesRef.current,\n            parentId,\n          );\n          replaceMessages(truncated);\n          const externalId = aui.threadListItem().getState().externalId;\n          const checkpointId = externalId\n            ? await getCheckpointId(externalId, truncated)\n            : null;\n          return handleSendMessage([], {\n            runConfig: config.runConfig,\n            ...(checkpointId && { checkpointId }),\n          });\n        }\n      : undefined,\n    onAddToolResult: async ({\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      artifact,\n    }) => {\n      await handleSendMessage(\n        [\n          {\n            id: uuidv4(),\n            type: \"tool\",\n            name: toolName,\n            tool_call_id: toolCallId,\n            content: JSON.stringify(result),\n            artifact,\n            status: isError ? \"error\" : \"success\",\n          },\n        ],\n        {},\n      );\n    },\n    onResumeToolCall: (options) =>\n      toolInvocations.resume(options.toolCallId, options.payload),\n    onCancel: unstable_allowCancellation\n      ? async () => {\n          cancel();\n          await toolInvocations.abort();\n        }\n      : undefined,\n  });\n\n  {\n    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n    const loadRef = useRef(load);\n    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n    useEffect(() => {\n      loadRef.current = load;\n    });\n\n    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n    useEffect(() => {\n      const loadFn = loadRef.current;\n      if (!loadFn) return;\n\n      const externalId = aui.threadListItem().getState().externalId;\n      if (externalId == null) return;\n\n      loadFn(externalId).then(\n        ({ messages: msgs }) => {\n          replaceMessages(msgs);\n        },\n        (e) => {\n          console.warn(\"Failed to load ADK session:\", e);\n        },\n      );\n    }, [aui, replaceMessages]);\n  }\n\n  return runtime;\n};\n\nexport const useAdkRuntime = ({\n  cloud,\n  sessionAdapter,\n  create,\n  delete: deleteFn,\n  ...options\n}: UseAdkRuntimeOptions) => {\n  const aui = useAui();\n  const cloudAdapter = useCloudThreadListAdapter({\n    cloud,\n    create: async () => {\n      if (create) return create();\n      if (aui.threadListItem.source) return aui.threadListItem().initialize();\n      return { externalId: undefined };\n    },\n    delete: deleteFn,\n  });\n\n  const adapter = sessionAdapter ?? cloudAdapter;\n\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n      return useAdkRuntimeImpl(options);\n    },\n    adapter,\n    allowNesting: true,\n  });\n};\n"],"mappings":";;;;;;;;;;AAoCA,MAAa,qBAAqB,QAAuB;CAKvD,MAAM,UAAU,CAHd,GAAG,IAAI,SACP,GAAI,IAAI,aAAa,SAAS,MAAM,EAAE,OAAO,KAAK,CAAC,CAE5B,EAAE,KAAK,SAAS;EACvC,MAAM,OAAO,KAAK;EAClB,QAAQ,MAAR;GACE,KAAK,QACH,OAAO;IAAE,MAAM;IAAiB,MAAM,KAAK;GAAK;GAClD,KAAK,SACH,OAAO;IAAE,MAAM;IAAsB,KAAK,KAAK;GAAM;GACvD,KAAK,QACH,OAAO;IACL,MAAM;IACN,UAAU,KAAK;IACf,MAAM,KAAK;IACX,GAAI,KAAK,YAAY,QAAQ,EAAE,UAAU,KAAK,SAAS;GACzD;GAEF,KAAK,aACH,MAAM,IAAI,MAAM,sCAAsC;GAExD,SAOE,MAAM,IAAI,MACR,yCAAyCA,MAC3C;EAEJ;CACF,CAAC;CAED,IAAI,QAAQ,WAAW,KAAK,QAAQ,IAAI,SAAS,QAC/C,OAAO,QAAQ,GAAG,QAAQ;CAG5B,OAAO;AACT;;AAGA,MAAa,uBAAuB,aAA2B;CAC7D,MAAM,0BAAU,IAAI,IAA0C;CAC9D,KAAK,MAAM,OAAO,UAAU;EAC1B,IAAI,IAAI,SAAS,QAAQ,IAAI,YAC3B,KAAK,MAAM,MAAM,IAAI,YACnB,QAAQ,IAAI,GAAG,IAAI,EAAE;EAGzB,IAAI,IAAI,SAAS,QACf,QAAQ,OAAO,IAAI,YAAY;CAEnC;CACA,OAAO,CAAC,GAAG,QAAQ,OAAO,CAAC;AAC7B;;;;;;;;;;AAWA,MAAa,2BACX,UACA,uBACyC;CACzC,MAAM,iBAAiB,IAAI,IAAI,kBAAkB;CACjD,OAAO,oBAAoB,QAAQ,EAChC,QAAQ,MAAM,CAAC,eAAe,IAAI,EAAE,EAAE,CAAC,EACvC,KACE,OACE;EACC,IAAIC,GAAO;EACX,MAAM;EACN,MAAM,EAAE;EACR,cAAc,EAAE;EAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;EAC3C,QAAQ;CACV,EACJ;AACJ;AAEA,MAAM,uBACJ,gBACA,aACiB;CACjB,IAAI,aAAa,MAAM,OAAO,CAAC;CAC/B,MAAM,cAAc,eAAe,WAAW,MAAM,EAAE,OAAO,QAAQ;CACrE,IAAI,gBAAgB,IAAI,OAAO,CAAC;CAChC,MAAM,YAA0B,CAAC;CACjC,KAAK,IAAI,IAAI,GAAG,KAAK,eAAe,IAAI,eAAe,QAAQ,KAC7D,UAAU,KAAK,GAAG,yBAAqC,eAAe,EAAG,CAAC;CAE5E,OAAO;AACT;AAoCA,MAAM,qBAAqB,EACzB,4BACA,UAAU,EAAE,aAAa,WAAW,UAAU,WAAW,CAAC,GAC1D,4BACA,QACA,MACA,iBACA,oBAC0B;CAE1B,MAAM,MAAM,OAAO;CACnB,MAAM,EACJ,UACA,YACA,WACA,oBACA,eACA,mBACA,cACA,WACA,iBACA,aACA,QACA,oBAEE,eAAe;EACjB;EACA,GAAI,iBAAiB,EAAE,cAAc;CACvC,CAAC;CAGD,MAAM,CAAC,WAAW,gBAAgB,SAAS,KAAK;CAEhD,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,EAAE,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,qBAAqB,aAAa;CAExC,MAAM,oBAAoB,OACxB,MACA,WACG;EACH,IAAI;GACF,aAAa,IAAI;GACjB,MAAM,YAAY,MAAM,MAAM;EAChC,UAAU;GACR,aAAa,KAAK;EACpB;CACF;CAGA,MAAM,iBAAiB,4BAA4B;EACjD,UAAU;EACV;EACA,WAAW;CACb,CAAC;CAGD,MAAM,oBAAoB,OAAO,cAAc;CAC/C,kBAAkB,UAAU;CAG5B,MAAM,CAAC,cAAc,gBAAgB,EACnC,IAAI,UAAU;EACZ,OAAO;CACT,EACF,EAAE;CAGF,MAAM,kBAAkB,mBAAmB;EACzC,OAAO;GAAE,UAAU;GAAgB,WAAW;EAAmB;EACjE,gBAAgB,WAAW,QAAQ,OAAO,gBAAgB,EAAE;EAC5D,WAAW,YAAY;GACrB,IAAI,QAAQ,SAAS,mBACnB,kBACE,CACE;IACE,IAAIA,GAAO;IACX,MAAM;IACN,MAAM,QAAQ;IACd,cAAc,QAAQ;IACtB,SAAS,KAAK,UAAU,QAAQ,MAAM;IACtC,QAAQ,QAAQ,UAAU,UAAU;GACtC,CACF,GACA,CAAC,CACH;EAEJ;EACA;CACF,CAAC;CAGD,MAAM,UAAU,wBAAwB;EACtC,WAAW;EACX,UAAU;EACV,UAAU;GAAE;GAAa;GAAW;GAAU;EAAO;EACrD,QAAQ;IACL,yBAAyB;GAC1B;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA,MAAM;EACR;EACA,OAAO,OAAO,QAAQ;GACpB,MAAM,gBAAgB,MAAM;GAO5B,OAAO,kBACL,CACE,GANF,+BAA+B,QAC3B,wBAAwB,UAAU,kBAAkB,IACpD,CAAC,GAKH;IACE,IAAIA,GAAO;IACX,MAAM;IACN,SAAS,kBAAkB,GAAG;GAChC,CACF,GACA,EAAE,WAAW,IAAI,UAAU,CAC7B;EACF;EACA,QAAQ,kBACJ,OAAO,QAAQ;GACb,MAAM,gBAAgB,MAAM;GAC5B,MAAM,YAAY,oBAChB,kBAAkB,SAClB,IAAI,QACN;GACA,gBAAgB,SAAS;GACzB,MAAM,aAAa,IAAI,eAAe,EAAE,SAAS,EAAE;GACnD,MAAM,eAAe,aACjB,MAAM,gBAAgB,YAAY,SAAS,IAC3C;GACJ,OAAO,kBACL,CACE;IACE,IAAIA,GAAO;IACX,MAAM;IACN,SAAS,kBAAkB,GAAG;GAChC,CACF,GACA;IACE,WAAW,IAAI;IACf,GAAI,gBAAgB,EAAE,aAAa;GACrC,CACF;EACF,IACA,KAAA;EACJ,UAAU,kBACN,OAAO,UAAU,WAAW;GAC1B,MAAM,gBAAgB,MAAM;GAC5B,MAAM,YAAY,oBAChB,kBAAkB,SAClB,QACF;GACA,gBAAgB,SAAS;GACzB,MAAM,aAAa,IAAI,eAAe,EAAE,SAAS,EAAE;GACnD,MAAM,eAAe,aACjB,MAAM,gBAAgB,YAAY,SAAS,IAC3C;GACJ,OAAO,kBAAkB,CAAC,GAAG;IAC3B,WAAW,OAAO;IAClB,GAAI,gBAAgB,EAAE,aAAa;GACrC,CAAC;EACH,IACA,KAAA;EACJ,iBAAiB,OAAO,EACtB,YACA,UACA,QACA,SACA,eACI;GACJ,MAAM,kBACJ,CACE;IACE,IAAIA,GAAO;IACX,MAAM;IACN,MAAM;IACN,cAAc;IACd,SAAS,KAAK,UAAU,MAAM;IAC9B;IACA,QAAQ,UAAU,UAAU;GAC9B,CACF,GACA,CAAC,CACH;EACF;EACA,mBAAmB,YACjB,gBAAgB,OAAO,QAAQ,YAAY,QAAQ,OAAO;EAC5D,UAAU,6BACN,YAAY;GACV,OAAO;GACP,MAAM,gBAAgB,MAAM;EAC9B,IACA,KAAA;CACN,CAAC;CAED;EAEE,MAAM,UAAU,OAAO,IAAI;EAE3B,gBAAgB;GACd,QAAQ,UAAU;EACpB,CAAC;EAGD,gBAAgB;GACd,MAAM,SAAS,QAAQ;GACvB,IAAI,CAAC,QAAQ;GAEb,MAAM,aAAa,IAAI,eAAe,EAAE,SAAS,EAAE;GACnD,IAAI,cAAc,MAAM;GAExB,OAAO,UAAU,EAAE,MAChB,EAAE,UAAU,WAAW;IACtB,gBAAgB,IAAI;GACtB,IACC,MAAM;IACL,QAAQ,KAAK,+BAA+B,CAAC;GAC/C,CACF;EACF,GAAG,CAAC,KAAK,eAAe,CAAC;CAC3B;CAEA,OAAO;AACT;AAEA,MAAa,iBAAiB,EAC5B,OACA,gBACA,QACA,QAAQ,UACR,GAAG,cACuB;CAC1B,MAAM,MAAM,OAAO;CACnB,MAAM,eAAe,0BAA0B;EAC7C;EACA,QAAQ,YAAY;GAClB,IAAI,QAAQ,OAAO,OAAO;GAC1B,IAAI,IAAI,eAAe,QAAQ,OAAO,IAAI,eAAe,EAAE,WAAW;GACtE,OAAO,EAAE,YAAY,KAAA,EAAU;EACjC;EACA,QAAQ;CACV,CAAC;CAID,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAElC,OAAO,kBAAkB,OAAO;EAClC;EACA,SAPc,kBAAkB;EAQhC,cAAc;CAChB,CAAC;AACH"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-google-adk",
-  "version": "0.0.8",
+  "version": "0.0.11",
   "description": "Google ADK adapter for assistant-ui",
   "keywords": [
     "google-adk",
@@ -34,10 +34,10 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.1.16",
-    "@assistant-ui/store": "^0.2.9",
+    "@assistant-ui/core": "^0.2.5",
+    "@assistant-ui/store": "^0.2.12",
     "assistant-cloud": "*",
-    "assistant-stream": "^0.3.12",
+    "assistant-stream": "^0.3.16",
     "uuid": "^14.0.0"
   },
   "peerDependencies": {
@@ -54,11 +54,11 @@
     }
   },
   "devDependencies": {
-    "@types/react": "^19.2.14",
+    "@types/react": "^19.2.15",
     "@types/uuid": "^11.0.0",
-    "react": "^19.2.5",
-    "vitest": "^4.1.5",
-    "@assistant-ui/x-buildutils": "0.0.6"
+    "react": "^19.2.6",
+    "vitest": "^4.1.7",
+    "@assistant-ui/x-buildutils": "0.0.9"
   },
   "publishConfig": {
     "access": "public",

package/src/useAdkRuntime.ts

@@ -58,8 +58,12 @@ export const getMessageContent = (msg: AppendMessage) => {
         throw new Error("Tool call appends are not supported.");
 
       default: {
-        const _exhaustiveCheck: "reasoning" | "source" | "audio" | "data" =
-          type;
+        const _exhaustiveCheck:
+          | "reasoning"
+          | "source"
+          | "audio"
+          | "data"
+          | "generative-ui" = type;
         throw new Error(
           `Unsupported append message part type: ${_exhaustiveCheck}`,
         );

package/dist/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,EACb,KAAK,oBAAoB,GAC1B,2BAAwB;AAEzB,OAAO,EACL,cAAc,EACd,KAAK,qBAAqB,GAC3B,4BAAyB;AAE1B,OAAO,EAAE,iBAAiB,EAAE,gCAA6B;AAEzD,YAAY,EACV,QAAQ,EACR,YAAY,EACZ,eAAe,EACf,UAAU,EACV,qBAAqB,EACrB,WAAW,EACX,mBAAmB,EACnB,iBAAiB,EACjB,qBAAqB,EACrB,cAAc,EACd,kBAAkB,EAClB,YAAY,EACZ,oBAAoB,EACpB,iBAAiB,EACjB,kBAAkB,EAClB,kBAAkB,EAClB,wBAAwB,EACxB,0BAA0B,GAC3B,mBAAgB;AAEjB,OAAO,EAAE,YAAY,EAAE,mBAAgB;AAEvC,OAAO,EAAE,qBAAqB,EAAE,8BAA2B;AAE3D,OAAO,EAAE,mBAAmB,EAAE,iCAA8B;AAE5D,OAAO,EACL,eAAe,EACf,KAAK,sBAAsB,GAC5B,uBAAoB;AAErB,OAAO,EACL,uBAAuB,EACvB,KAAK,wBAAwB,EAC7B,KAAK,eAAe,GACrB,+BAA4B;AAE7B,OAAO,EACL,eAAe,EACf,kBAAkB,EAClB,UAAU,EACV,wBAAwB,EACxB,uBAAuB,EACvB,kBAAkB,EAClB,eAAe,EACf,gBAAgB,EAChB,qBAAqB,EACrB,iBAAiB,EACjB,gBAAgB,EAChB,iBAAiB,EACjB,cAAc,EACd,eAAe,EACf,eAAe,GAChB,mBAAgB"}

package/dist/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,GAEd,2BAAwB;AAEzB,OAAO,EACL,cAAc,GAEf,4BAAyB;AAE1B,OAAO,EAAE,iBAAiB,EAAE,gCAA6B;AAuBzD,OAAO,EAAE,YAAY,EAAE,mBAAgB;AAEvC,OAAO,EAAE,qBAAqB,EAAE,8BAA2B;AAE3D,OAAO,EAAE,mBAAmB,EAAE,iCAA8B;AAE5D,OAAO,EACL,eAAe,GAEhB,uBAAoB;AAErB,OAAO,EACL,uBAAuB,GAGxB,+BAA4B;AAE7B,OAAO,EACL,eAAe,EACf,kBAAkB,EAClB,UAAU,EACV,wBAAwB,EACxB,uBAAuB,EACvB,kBAAkB,EAClB,eAAe,EACf,gBAAgB,EAChB,qBAAqB,EACrB,iBAAiB,EACjB,gBAAgB,EAChB,iBAAiB,EACjB,cAAc,EACd,eAAe,EACf,eAAe,GAChB,mBAAgB"}