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

package/dist/useAdkMessages.d.ts

@@ -1,29 +1,36 @@
 /// <reference types="@assistant-ui/core/store" />
-import type { AdkMessage, AdkMessageMetadata, AdkSendMessageConfig, AdkStreamCallback, AdkToolConfirmation, AdkAuthRequest, OnAdkErrorCallback, OnAdkCustomEventCallback, OnAdkAgentTransferCallback } from "./types.js";
-export type UseAdkMessagesOptions = {
-    stream: AdkStreamCallback;
-    eventHandlers?: {
-        onError?: OnAdkErrorCallback;
-        onCustomEvent?: OnAdkCustomEventCallback;
-        onAgentTransfer?: OnAdkAgentTransferCallback;
-    };
+import { AdkAuthRequest, AdkMessage, AdkMessageMetadata, AdkSendMessageConfig, AdkStreamCallback, AdkToolConfirmation, OnAdkAgentTransferCallback, OnAdkCustomEventCallback, OnAdkErrorCallback } from "./types.js";
+
+//#region src/useAdkMessages.d.ts
+type UseAdkMessagesOptions = {
+  stream: AdkStreamCallback;
+  eventHandlers?: {
+    onError?: OnAdkErrorCallback;
+    onCustomEvent?: OnAdkCustomEventCallback;
+    onAgentTransfer?: OnAdkAgentTransferCallback;
+  };
 };
-export declare const useAdkMessages: ({ stream, eventHandlers, }: UseAdkMessagesOptions) => {
-    messages: AdkMessage[];
-    stateDelta: Record<string, unknown>;
-    agentInfo: {
-        name?: string | undefined;
-        branch?: string | undefined;
-    };
-    longRunningToolIds: string[];
-    artifactDelta: Record<string, number>;
-    toolConfirmations: AdkToolConfirmation[];
-    authRequests: AdkAuthRequest[];
-    escalated: boolean;
-    messageMetadata: Map<string, AdkMessageMetadata>;
-    sendMessage: (newMessages: AdkMessage[], config: AdkSendMessageConfig) => Promise<void>;
-    cancel: () => void;
-    setMessages: (msgs: AdkMessage[]) => void;
-    replaceMessages: (msgs: AdkMessage[]) => void;
+declare const useAdkMessages: ({
+  stream,
+  eventHandlers
+}: UseAdkMessagesOptions) => {
+  messages: AdkMessage[];
+  stateDelta: Record<string, unknown>;
+  agentInfo: {
+    name?: string | undefined;
+    branch?: string | undefined;
+  };
+  longRunningToolIds: string[];
+  artifactDelta: Record<string, number>;
+  toolConfirmations: AdkToolConfirmation[];
+  authRequests: AdkAuthRequest[];
+  escalated: boolean;
+  messageMetadata: Map<string, AdkMessageMetadata>;
+  sendMessage: (newMessages: AdkMessage[], config: AdkSendMessageConfig) => Promise<void>;
+  cancel: () => void;
+  setMessages: (msgs: AdkMessage[]) => void;
+  replaceMessages: (msgs: AdkMessage[]) => void;
 };
+//#endregion
+export { UseAdkMessagesOptions, useAdkMessages };
 //# sourceMappingURL=useAdkMessages.d.ts.map

package/dist/useAdkMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAdkMessages.d.ts","sourceRoot":"","sources":["../src/useAdkMessages.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAEV,UAAU,EACV,kBAAkB,EAClB,oBAAoB,EACpB,iBAAiB,EACjB,mBAAmB,EACnB,cAAc,EACd,kBAAkB,EAClB,wBAAwB,EACxB,0BAA0B,EAC3B,mBAAgB;AAEjB,MAAM,MAAM,qBAAqB,GAAG;IAClC,MAAM,EAAE,iBAAiB,CAAC;IAC1B,aAAa,CAAC,EAAE;QACd,OAAO,CAAC,EAAE,kBAAkB,CAAC;QAC7B,aAAa,CAAC,EAAE,wBAAwB,CAAC;QACzC,eAAe,CAAC,EAAE,0BAA0B,CAAC;KAC9C,CAAC;CACH,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,4BAG5B,qBAAqB;;;;eAIb,MAAM,GAAG,SAAS;iBAChB,MAAM,GAAG,SAAS;;;;;;;;+BAsDP,UAAU,EAAE,UAAU,oBAAoB;;wBA9BhB,UAAU,EAAE;4BAUnD,UAAU,EAAE;CAiItB,CAAC"}
+{"version":3,"file":"useAdkMessages.d.ts","names":[],"sources":["../src/useAdkMessages.ts"],"mappings":";;;KAmBY,qBAAA;EACV,MAAA,EAAQ,iBAAA;EACR,aAAA;IACE,OAAA,GAAU,kBAAA;IACV,aAAA,GAAgB,wBAAA;IAChB,eAAA,GAAkB,0BAAA;EAAA;AAAA;AAAA,cAIT,cAAA;EAAkB,MAAA;EAAA;AAAA,GAG5B,qBAAA;;;;;;;;;;;;;6BA2DqB,UAAA,IAAY,MAAA,EAAU,oBAAA,KAAoB,OAAA;;sBA9BhB,UAAA;0BAUvC,UAAA;AAAA"}

package/dist/useAdkMessages.js

@@ -1,187 +1,160 @@
-/// <reference types="@assistant-ui/core/store" />
-import { useState, useCallback, useRef, useMemo } from "react";
-import { v4 as uuidv4 } from "uuid";
-import { useAui } from "@assistant-ui/store";
-import { AdkEventAccumulator } from "./AdkEventAccumulator.js";
 import { contentToParts } from "./contentToParts.js";
-export const useAdkMessages = ({ stream, eventHandlers, }) => {
-    const [messages, _setMessages] = useState([]);
-    const [stateDelta, setStateDelta] = useState({});
-    const [agentInfo, setAgentInfo] = useState({});
-    const [longRunningToolIds, setLongRunningToolIds] = useState([]);
-    const [artifactDelta, setArtifactDelta] = useState({});
-    const [toolConfirmations, setToolConfirmations] = useState([]);
-    const [authRequests, setAuthRequests] = useState([]);
-    const [escalated, setEscalated] = useState(false);
-    const [messageMetadata, setMessageMetadata] = useState(new Map());
-    const lastTransferToAgentRef = useRef(undefined);
-    const messagesRef = useRef(messages);
-    messagesRef.current = messages;
-    const stateDeltaRef = useRef(stateDelta);
-    stateDeltaRef.current = stateDelta;
-    const artifactDeltaRef = useRef(artifactDelta);
-    artifactDeltaRef.current = artifactDelta;
-    const messageMetadataRef = useRef(messageMetadata);
-    messageMetadataRef.current = messageMetadata;
-    const setMessagesImmediate = useCallback((msgs) => {
-        messagesRef.current = msgs;
-        _setMessages(msgs);
-    }, []);
-    // Replace the message list AND reset derived per-turn HITL state.
-    // Used by truncation paths (edit, reload, load) so that stale interrupt
-    // markers and per-message metadata from the removed messages don't leak
-    // into the next turn.
-    const replaceMessages = useCallback((msgs) => {
-        setMessagesImmediate(msgs);
-        setLongRunningToolIds([]);
-        setToolConfirmations([]);
-        setAuthRequests([]);
-        setEscalated(false);
-        setMessageMetadata(new Map());
-    }, [setMessagesImmediate]);
-    const abortControllerRef = useRef(null);
-    const { onError, onCustomEvent, onAgentTransfer } = useMemo(() => eventHandlers ?? {}, [eventHandlers]);
-    const aui = useAui();
-    const sendMessage = useCallback(async (newMessages, config) => {
-        const newMessagesWithId = newMessages.map((m) => m.id ? m : { ...m, id: uuidv4() });
-        const accumulator = new AdkEventAccumulator(messagesRef.current);
-        for (const msg of newMessagesWithId) {
-            accumulator.processEvent(messageToEvent(msg));
-        }
-        setMessagesImmediate(accumulator.getMessages());
-        const abortController = new AbortController();
-        abortControllerRef.current = abortController;
-        try {
-            const response = await stream(newMessagesWithId, {
-                ...config,
-                abortSignal: abortController.signal,
-                initialize: async () => {
-                    return await aui.threadListItem().initialize();
-                },
-            });
-            for await (const event of response) {
-                const updatedMessages = accumulator.processEvent(event);
-                setMessagesImmediate(updatedMessages);
-                setStateDelta({
-                    ...stateDeltaRef.current,
-                    ...accumulator.getStateDelta(),
-                });
-                setAgentInfo(accumulator.getAgentInfo());
-                setLongRunningToolIds(accumulator.getLongRunningToolIds());
-                setArtifactDelta({
-                    ...artifactDeltaRef.current,
-                    ...accumulator.getArtifactDelta(),
-                });
-                setToolConfirmations(accumulator.getToolConfirmations());
-                setAuthRequests(accumulator.getAuthRequests());
-                setEscalated(accumulator.isEscalated());
-                {
-                    const newMeta = accumulator.getMessageMetadata();
-                    if (newMeta.size > 0) {
-                        setMessageMetadata(new Map([...messageMetadataRef.current, ...newMeta]));
-                    }
-                }
-                const transfer = accumulator.getLastTransferToAgent();
-                if (transfer && transfer !== lastTransferToAgentRef.current) {
-                    lastTransferToAgentRef.current = transfer;
-                    onAgentTransfer?.(transfer);
-                }
-                // Fire custom event callback for events with customMetadata
-                if (event.customMetadata && onCustomEvent) {
-                    for (const [key, value] of Object.entries(event.customMetadata)) {
-                        onCustomEvent(key, value);
-                    }
-                }
-                if (event.errorCode || event.errorMessage) {
-                    onError?.(event.errorMessage ?? event.errorCode);
-                }
-            }
-        }
-        catch (error) {
-            if (!abortController.signal.aborted &&
-                !(error instanceof Error && error.name === "AbortError")) {
-                throw error;
-            }
-        }
-        finally {
-            if (abortControllerRef.current === abortController) {
-                abortControllerRef.current = null;
-            }
-        }
-    }, [
-        aui,
-        setMessagesImmediate,
-        stream,
-        onError,
-        onCustomEvent,
-        onAgentTransfer,
-    ]);
-    const cancel = useCallback(() => {
-        if (abortControllerRef.current) {
-            abortControllerRef.current.abort();
-        }
-    }, []);
-    return {
-        messages,
-        stateDelta,
-        agentInfo,
-        longRunningToolIds,
-        artifactDelta,
-        toolConfirmations,
-        authRequests,
-        escalated,
-        messageMetadata,
-        sendMessage,
-        cancel,
-        setMessages: setMessagesImmediate,
-        replaceMessages,
-    };
+import { AdkEventAccumulator } from "./AdkEventAccumulator.js";
+import { v4 } from "uuid";
+import { useAui } from "@assistant-ui/store";
+import { useCallback, useMemo, useRef, useState } from "react";
+//#region src/useAdkMessages.ts
+const useAdkMessages = ({ stream, eventHandlers }) => {
+	const [messages, _setMessages] = useState([]);
+	const [stateDelta, setStateDelta] = useState({});
+	const [agentInfo, setAgentInfo] = useState({});
+	const [longRunningToolIds, setLongRunningToolIds] = useState([]);
+	const [artifactDelta, setArtifactDelta] = useState({});
+	const [toolConfirmations, setToolConfirmations] = useState([]);
+	const [authRequests, setAuthRequests] = useState([]);
+	const [escalated, setEscalated] = useState(false);
+	const [messageMetadata, setMessageMetadata] = useState(/* @__PURE__ */ new Map());
+	const lastTransferToAgentRef = useRef(void 0);
+	const messagesRef = useRef(messages);
+	messagesRef.current = messages;
+	const stateDeltaRef = useRef(stateDelta);
+	stateDeltaRef.current = stateDelta;
+	const artifactDeltaRef = useRef(artifactDelta);
+	artifactDeltaRef.current = artifactDelta;
+	const messageMetadataRef = useRef(messageMetadata);
+	messageMetadataRef.current = messageMetadata;
+	const setMessagesImmediate = useCallback((msgs) => {
+		messagesRef.current = msgs;
+		_setMessages(msgs);
+	}, []);
+	const replaceMessages = useCallback((msgs) => {
+		setMessagesImmediate(msgs);
+		setLongRunningToolIds([]);
+		setToolConfirmations([]);
+		setAuthRequests([]);
+		setEscalated(false);
+		setMessageMetadata(/* @__PURE__ */ new Map());
+	}, [setMessagesImmediate]);
+	const abortControllerRef = useRef(null);
+	const { onError, onCustomEvent, onAgentTransfer } = useMemo(() => eventHandlers ?? {}, [eventHandlers]);
+	const aui = useAui();
+	return {
+		messages,
+		stateDelta,
+		agentInfo,
+		longRunningToolIds,
+		artifactDelta,
+		toolConfirmations,
+		authRequests,
+		escalated,
+		messageMetadata,
+		sendMessage: useCallback(async (newMessages, config) => {
+			const newMessagesWithId = newMessages.map((m) => m.id ? m : {
+				...m,
+				id: v4()
+			});
+			const accumulator = new AdkEventAccumulator(messagesRef.current);
+			for (const msg of newMessagesWithId) accumulator.processEvent(messageToEvent(msg));
+			setMessagesImmediate(accumulator.getMessages());
+			const abortController = new AbortController();
+			abortControllerRef.current = abortController;
+			try {
+				const response = await stream(newMessagesWithId, {
+					...config,
+					abortSignal: abortController.signal,
+					initialize: async () => {
+						return await aui.threadListItem().initialize();
+					}
+				});
+				for await (const event of response) {
+					setMessagesImmediate(accumulator.processEvent(event));
+					setStateDelta({
+						...stateDeltaRef.current,
+						...accumulator.getStateDelta()
+					});
+					setAgentInfo(accumulator.getAgentInfo());
+					setLongRunningToolIds(accumulator.getLongRunningToolIds());
+					setArtifactDelta({
+						...artifactDeltaRef.current,
+						...accumulator.getArtifactDelta()
+					});
+					setToolConfirmations(accumulator.getToolConfirmations());
+					setAuthRequests(accumulator.getAuthRequests());
+					setEscalated(accumulator.isEscalated());
+					{
+						const newMeta = accumulator.getMessageMetadata();
+						if (newMeta.size > 0) setMessageMetadata(new Map([...messageMetadataRef.current, ...newMeta]));
+					}
+					const transfer = accumulator.getLastTransferToAgent();
+					if (transfer && transfer !== lastTransferToAgentRef.current) {
+						lastTransferToAgentRef.current = transfer;
+						onAgentTransfer?.(transfer);
+					}
+					if (event.customMetadata && onCustomEvent) for (const [key, value] of Object.entries(event.customMetadata)) onCustomEvent(key, value);
+					if (event.errorCode || event.errorMessage) onError?.(event.errorMessage ?? event.errorCode);
+				}
+			} catch (error) {
+				if (!abortController.signal.aborted && !(error instanceof Error && error.name === "AbortError")) throw error;
+			} finally {
+				if (abortControllerRef.current === abortController) abortControllerRef.current = null;
+			}
+		}, [
+			aui,
+			setMessagesImmediate,
+			stream,
+			onError,
+			onCustomEvent,
+			onAgentTransfer
+		]),
+		cancel: useCallback(() => {
+			if (abortControllerRef.current) abortControllerRef.current.abort();
+		}, []),
+		setMessages: setMessagesImmediate,
+		replaceMessages
+	};
 };
 /** @internal — exported for unit tests. */
-export const messageToEvent = (msg) => {
-    if (msg.type === "human") {
-        return {
-            id: msg.id ?? uuidv4(),
-            author: "user",
-            content: { role: "user", parts: contentToParts(msg.content) },
-        };
-    }
-    if (msg.type === "tool") {
-        let response;
-        try {
-            response = JSON.parse(msg.content);
-        }
-        catch {
-            response = msg.content;
-        }
-        return {
-            id: msg.id ?? uuidv4(),
-            content: {
-                role: "user",
-                parts: [
-                    {
-                        functionResponse: {
-                            name: msg.name,
-                            id: msg.tool_call_id,
-                            response,
-                        },
-                    },
-                ],
-            },
-        };
-    }
-    const result = { id: msg.id ?? uuidv4() };
-    if (msg.author != null)
-        result.author = msg.author;
-    result.content = {
-        role: "model",
-        parts: [
-            ...contentToParts(msg.content),
-            ...(msg.tool_calls?.map((tc) => ({
-                functionCall: { name: tc.name, id: tc.id, args: { ...tc.args } },
-            })) ?? []),
-        ],
-    };
-    return result;
+const messageToEvent = (msg) => {
+	if (msg.type === "human") return {
+		id: msg.id ?? v4(),
+		author: "user",
+		content: {
+			role: "user",
+			parts: contentToParts(msg.content)
+		}
+	};
+	if (msg.type === "tool") {
+		let response;
+		try {
+			response = JSON.parse(msg.content);
+		} catch {
+			response = msg.content;
+		}
+		return {
+			id: msg.id ?? v4(),
+			content: {
+				role: "user",
+				parts: [{ functionResponse: {
+					name: msg.name,
+					id: msg.tool_call_id,
+					response
+				} }]
+			}
+		};
+	}
+	const result = { id: msg.id ?? v4() };
+	if (msg.author != null) result.author = msg.author;
+	result.content = {
+		role: "model",
+		parts: [...contentToParts(msg.content), ...msg.tool_calls?.map((tc) => ({ functionCall: {
+			name: tc.name,
+			id: tc.id,
+			args: { ...tc.args }
+		} })) ?? []]
+	};
+	return result;
 };
+//#endregion
+export { messageToEvent, useAdkMessages };
+
 //# sourceMappingURL=useAdkMessages.js.map

package/dist/useAdkMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAdkMessages.js","sourceRoot":"","sources":["../src/useAdkMessages.ts"],"names":[],"mappings":"AAAA,kDAAkD;AAClD,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAC/D,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AACpC,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,mBAAmB,EAAE,iCAA8B;AAC5D,OAAO,EAAE,cAAc,EAAE,4BAAyB;AAuBlD,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,EAC7B,MAAM,EACN,aAAa,GACS,EAAE,EAAE;IAC1B,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAe,EAAE,CAAC,CAAC;IAC5D,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,QAAQ,CAA0B,EAAE,CAAC,CAAC;IAC1E,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAGvC,EAAE,CAAC,CAAC;IACP,MAAM,CAAC,kBAAkB,EAAE,qBAAqB,CAAC,GAAG,QAAQ,CAAW,EAAE,CAAC,CAAC;IAC3E,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAChD,EAAE,CACH,CAAC;IACF,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,GAAG,QAAQ,CAExD,EAAE,CAAC,CAAC;IACN,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAmB,EAAE,CAAC,CAAC;IACvE,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAClD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CAEpD,IAAI,GAAG,EAAE,CAAC,CAAC;IACb,MAAM,sBAAsB,GAAG,MAAM,CAAqB,SAAS,CAAC,CAAC;IACrE,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;IACrC,WAAW,CAAC,OAAO,GAAG,QAAQ,CAAC;IAC/B,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;IACzC,aAAa,CAAC,OAAO,GAAG,UAAU,CAAC;IACnC,MAAM,gBAAgB,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC;IAC/C,gBAAgB,CAAC,OAAO,GAAG,aAAa,CAAC;IACzC,MAAM,kBAAkB,GAAG,MAAM,CAAC,eAAe,CAAC,CAAC;IACnD,kBAAkB,CAAC,OAAO,GAAG,eAAe,CAAC;IAE7C,MAAM,oBAAoB,GAAG,WAAW,CAAC,CAAC,IAAkB,EAAE,EAAE;QAC9D,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;QAC3B,YAAY,CAAC,IAAI,CAAC,CAAC;IACrB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,kEAAkE;IAClE,wEAAwE;IACxE,wEAAwE;IACxE,sBAAsB;IACtB,MAAM,eAAe,GAAG,WAAW,CACjC,CAAC,IAAkB,EAAE,EAAE;QACrB,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC3B,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAC1B,oBAAoB,CAAC,EAAE,CAAC,CAAC;QACzB,eAAe,CAAC,EAAE,CAAC,CAAC;QACpB,YAAY,CAAC,KAAK,CAAC,CAAC;QACpB,kBAAkB,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;IAChC,CAAC,EACD,CAAC,oBAAoB,CAAC,CACvB,CAAC;IAEF,MAAM,kBAAkB,GAAG,MAAM,CAAyB,IAAI,CAAC,CAAC;IAEhE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,GAAG,OAAO,CACzD,GAAG,EAAE,CAAC,aAAa,IAAI,EAAE,EACzB,CAAC,aAAa,CAAC,CAChB,CAAC;IAEF,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IACrB,MAAM,WAAW,GAAG,WAAW,CAC7B,KAAK,EAAE,WAAyB,EAAE,MAA4B,EAAE,EAAE;QAChE,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAC9C,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,CAClB,CAAC;QAElB,MAAM,WAAW,GAAG,IAAI,mBAAmB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACjE,KAAK,MAAM,GAAG,IAAI,iBAAiB,EAAE,CAAC;YACpC,WAAW,CAAC,YAAY,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC;QAChD,CAAC;QACD,oBAAoB,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC,CAAC;QAEhD,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,kBAAkB,CAAC,OAAO,GAAG,eAAe,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,iBAAiB,EAAE;gBAC/C,GAAG,MAAM;gBACT,WAAW,EAAE,eAAe,CAAC,MAAM;gBACnC,UAAU,EAAE,KAAK,IAAI,EAAE;oBACrB,OAAO,MAAM,GAAG,CAAC,cAAc,EAAE,CAAC,UAAU,EAAE,CAAC;gBACjD,CAAC;aACF,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;gBACnC,MAAM,eAAe,GAAG,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBACxD,oBAAoB,CAAC,eAAe,CAAC,CAAC;gBACtC,aAAa,CAAC;oBACZ,GAAG,aAAa,CAAC,OAAO;oBACxB,GAAG,WAAW,CAAC,aAAa,EAAE;iBAC/B,CAAC,CAAC;gBACH,YAAY,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC,CAAC;gBACzC,qBAAqB,CAAC,WAAW,CAAC,qBAAqB,EAAE,CAAC,CAAC;gBAC3D,gBAAgB,CAAC;oBACf,GAAG,gBAAgB,CAAC,OAAO;oBAC3B,GAAG,WAAW,CAAC,gBAAgB,EAAE;iBAClC,CAAC,CAAC;gBACH,oBAAoB,CAAC,WAAW,CAAC,oBAAoB,EAAE,CAAC,CAAC;gBACzD,eAAe,CAAC,WAAW,CAAC,eAAe,EAAE,CAAC,CAAC;gBAC/C,YAAY,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC,CAAC;gBACxC,CAAC;oBACC,MAAM,OAAO,GAAG,WAAW,CAAC,kBAAkB,EAAE,CAAC;oBACjD,IAAI,OAAO,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;wBACrB,kBAAkB,CAChB,IAAI,GAAG,CAAC,CAAC,GAAG,kBAAkB,CAAC,OAAO,EAAE,GAAG,OAAO,CAAC,CAAC,CACrD,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,MAAM,QAAQ,GAAG,WAAW,CAAC,sBAAsB,EAAE,CAAC;gBACtD,IAAI,QAAQ,IAAI,QAAQ,KAAK,sBAAsB,CAAC,OAAO,EAAE,CAAC;oBAC5D,sBAAsB,CAAC,OAAO,GAAG,QAAQ,CAAC;oBAC1C,eAAe,EAAE,CAAC,QAAQ,CAAC,CAAC;gBAC9B,CAAC;gBAED,4DAA4D;gBAC5D,IAAI,KAAK,CAAC,cAAc,IAAI,aAAa,EAAE,CAAC;oBAC1C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,CAAC;wBAChE,aAAa,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;oBAC5B,CAAC;gBACH,CAAC;gBAED,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;oBAC1C,OAAO,EAAE,CAAC,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IACE,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO;gBAC/B,CAAC,CAAC,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,EACxD,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,kBAAkB,CAAC,OAAO,KAAK,eAAe,EAAE,CAAC;gBACnD,kBAAkB,CAAC,OAAO,GAAG,IAAI,CAAC;YACpC,CAAC;QACH,CAAC;IACH,CAAC,EACD;QACE,GAAG;QACH,oBAAoB;QACpB,MAAM;QACN,OAAO;QACP,aAAa;QACb,eAAe;KAChB,CACF,CAAC;IAEF,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,EAAE;QAC9B,IAAI,kBAAkB,CAAC,OAAO,EAAE,CAAC;YAC/B,kBAAkB,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrC,CAAC;IACH,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO;QACL,QAAQ;QACR,UAAU;QACV,SAAS;QACT,kBAAkB;QAClB,aAAa;QACb,iBAAiB;QACjB,YAAY;QACZ,SAAS;QACT,eAAe;QACf,WAAW;QACX,MAAM;QACN,WAAW,EAAE,oBAAoB;QACjC,eAAe;KAChB,CAAC;AACJ,CAAC,CAAC;AAEF,2CAA2C;AAC3C,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,GAAe,EAAY,EAAE;IAC1D,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QACzB,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE,IAAI,MAAM,EAAE;YACtB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;SAC9D,CAAC;IACJ,CAAC;IAED,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACxB,IAAI,QAAiB,CAAC;QACtB,IAAI,CAAC;YACH,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACrC,CAAC;QAAC,MAAM,CAAC;YACP,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC;QACzB,CAAC;QACD,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE,IAAI,MAAM,EAAE;YACtB,OAAO,EAAE;gBACP,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACL;wBACE,gBAAgB,EAAE;4BAChB,IAAI,EAAE,GAAG,CAAC,IAAI;4BACd,EAAE,EAAE,GAAG,CAAC,YAAY;4BACpB,QAAQ;yBACT;qBACF;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAa,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,IAAI,MAAM,EAAE,EAAE,CAAC;IACpD,IAAI,GAAG,CAAC,MAAM,IAAI,IAAI;QAAE,MAAM,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC;IACnD,MAAM,CAAC,OAAO,GAAG;QACf,IAAI,EAAE,OAAO;QACb,KAAK,EAAE;YACL,GAAG,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC;YAC9B,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC/B,YAAY,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE;aACjE,CAAC,CAAC,IAAI,EAAE,CAAC;SACX;KACF,CAAC;IACF,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC"}
+{"version":3,"file":"useAdkMessages.js","names":["uuidv4"],"sources":["../src/useAdkMessages.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/store\" />\nimport { useState, useCallback, useRef, useMemo } from \"react\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport { useAui } from \"@assistant-ui/store\";\nimport { AdkEventAccumulator } from \"./AdkEventAccumulator\";\nimport { contentToParts } from \"./contentToParts\";\nimport type {\n  AdkEvent,\n  AdkMessage,\n  AdkMessageMetadata,\n  AdkSendMessageConfig,\n  AdkStreamCallback,\n  AdkToolConfirmation,\n  AdkAuthRequest,\n  OnAdkErrorCallback,\n  OnAdkCustomEventCallback,\n  OnAdkAgentTransferCallback,\n} from \"./types\";\n\nexport type UseAdkMessagesOptions = {\n  stream: AdkStreamCallback;\n  eventHandlers?: {\n    onError?: OnAdkErrorCallback;\n    onCustomEvent?: OnAdkCustomEventCallback;\n    onAgentTransfer?: OnAdkAgentTransferCallback;\n  };\n};\n\nexport const useAdkMessages = ({\n  stream,\n  eventHandlers,\n}: UseAdkMessagesOptions) => {\n  const [messages, _setMessages] = useState<AdkMessage[]>([]);\n  const [stateDelta, setStateDelta] = useState<Record<string, unknown>>({});\n  const [agentInfo, setAgentInfo] = useState<{\n    name?: string | undefined;\n    branch?: string | undefined;\n  }>({});\n  const [longRunningToolIds, setLongRunningToolIds] = useState<string[]>([]);\n  const [artifactDelta, setArtifactDelta] = useState<Record<string, number>>(\n    {},\n  );\n  const [toolConfirmations, setToolConfirmations] = useState<\n    AdkToolConfirmation[]\n  >([]);\n  const [authRequests, setAuthRequests] = useState<AdkAuthRequest[]>([]);\n  const [escalated, setEscalated] = useState(false);\n  const [messageMetadata, setMessageMetadata] = useState<\n    Map<string, AdkMessageMetadata>\n  >(new Map());\n  const lastTransferToAgentRef = useRef<string | undefined>(undefined);\n  const messagesRef = useRef(messages);\n  messagesRef.current = messages;\n  const stateDeltaRef = useRef(stateDelta);\n  stateDeltaRef.current = stateDelta;\n  const artifactDeltaRef = useRef(artifactDelta);\n  artifactDeltaRef.current = artifactDelta;\n  const messageMetadataRef = useRef(messageMetadata);\n  messageMetadataRef.current = messageMetadata;\n\n  const setMessagesImmediate = useCallback((msgs: AdkMessage[]) => {\n    messagesRef.current = msgs;\n    _setMessages(msgs);\n  }, []);\n\n  // Replace the message list AND reset derived per-turn HITL state.\n  // Used by truncation paths (edit, reload, load) so that stale interrupt\n  // markers and per-message metadata from the removed messages don't leak\n  // into the next turn.\n  const replaceMessages = useCallback(\n    (msgs: AdkMessage[]) => {\n      setMessagesImmediate(msgs);\n      setLongRunningToolIds([]);\n      setToolConfirmations([]);\n      setAuthRequests([]);\n      setEscalated(false);\n      setMessageMetadata(new Map());\n    },\n    [setMessagesImmediate],\n  );\n\n  const abortControllerRef = useRef<AbortController | null>(null);\n\n  const { onError, onCustomEvent, onAgentTransfer } = useMemo(\n    () => eventHandlers ?? {},\n    [eventHandlers],\n  );\n\n  const aui = useAui();\n  const sendMessage = useCallback(\n    async (newMessages: AdkMessage[], config: AdkSendMessageConfig) => {\n      const newMessagesWithId = newMessages.map((m) =>\n        m.id ? m : { ...m, id: uuidv4() },\n      ) as AdkMessage[];\n\n      const accumulator = new AdkEventAccumulator(messagesRef.current);\n      for (const msg of newMessagesWithId) {\n        accumulator.processEvent(messageToEvent(msg));\n      }\n      setMessagesImmediate(accumulator.getMessages());\n\n      const abortController = new AbortController();\n      abortControllerRef.current = abortController;\n\n      try {\n        const response = await stream(newMessagesWithId, {\n          ...config,\n          abortSignal: abortController.signal,\n          initialize: async () => {\n            return await aui.threadListItem().initialize();\n          },\n        });\n\n        for await (const event of response) {\n          const updatedMessages = accumulator.processEvent(event);\n          setMessagesImmediate(updatedMessages);\n          setStateDelta({\n            ...stateDeltaRef.current,\n            ...accumulator.getStateDelta(),\n          });\n          setAgentInfo(accumulator.getAgentInfo());\n          setLongRunningToolIds(accumulator.getLongRunningToolIds());\n          setArtifactDelta({\n            ...artifactDeltaRef.current,\n            ...accumulator.getArtifactDelta(),\n          });\n          setToolConfirmations(accumulator.getToolConfirmations());\n          setAuthRequests(accumulator.getAuthRequests());\n          setEscalated(accumulator.isEscalated());\n          {\n            const newMeta = accumulator.getMessageMetadata();\n            if (newMeta.size > 0) {\n              setMessageMetadata(\n                new Map([...messageMetadataRef.current, ...newMeta]),\n              );\n            }\n          }\n\n          const transfer = accumulator.getLastTransferToAgent();\n          if (transfer && transfer !== lastTransferToAgentRef.current) {\n            lastTransferToAgentRef.current = transfer;\n            onAgentTransfer?.(transfer);\n          }\n\n          // Fire custom event callback for events with customMetadata\n          if (event.customMetadata && onCustomEvent) {\n            for (const [key, value] of Object.entries(event.customMetadata)) {\n              onCustomEvent(key, value);\n            }\n          }\n\n          if (event.errorCode || event.errorMessage) {\n            onError?.(event.errorMessage ?? event.errorCode);\n          }\n        }\n      } catch (error) {\n        if (\n          !abortController.signal.aborted &&\n          !(error instanceof Error && error.name === \"AbortError\")\n        ) {\n          throw error;\n        }\n      } finally {\n        if (abortControllerRef.current === abortController) {\n          abortControllerRef.current = null;\n        }\n      }\n    },\n    [\n      aui,\n      setMessagesImmediate,\n      stream,\n      onError,\n      onCustomEvent,\n      onAgentTransfer,\n    ],\n  );\n\n  const cancel = useCallback(() => {\n    if (abortControllerRef.current) {\n      abortControllerRef.current.abort();\n    }\n  }, []);\n\n  return {\n    messages,\n    stateDelta,\n    agentInfo,\n    longRunningToolIds,\n    artifactDelta,\n    toolConfirmations,\n    authRequests,\n    escalated,\n    messageMetadata,\n    sendMessage,\n    cancel,\n    setMessages: setMessagesImmediate,\n    replaceMessages,\n  };\n};\n\n/** @internal — exported for unit tests. */\nexport const messageToEvent = (msg: AdkMessage): AdkEvent => {\n  if (msg.type === \"human\") {\n    return {\n      id: msg.id ?? uuidv4(),\n      author: \"user\",\n      content: { role: \"user\", parts: contentToParts(msg.content) },\n    };\n  }\n\n  if (msg.type === \"tool\") {\n    let response: unknown;\n    try {\n      response = JSON.parse(msg.content);\n    } catch {\n      response = msg.content;\n    }\n    return {\n      id: msg.id ?? uuidv4(),\n      content: {\n        role: \"user\",\n        parts: [\n          {\n            functionResponse: {\n              name: msg.name,\n              id: msg.tool_call_id,\n              response,\n            },\n          },\n        ],\n      },\n    };\n  }\n\n  const result: AdkEvent = { id: msg.id ?? uuidv4() };\n  if (msg.author != null) result.author = msg.author;\n  result.content = {\n    role: \"model\",\n    parts: [\n      ...contentToParts(msg.content),\n      ...(msg.tool_calls?.map((tc) => ({\n        functionCall: { name: tc.name, id: tc.id, args: { ...tc.args } },\n      })) ?? []),\n    ],\n  };\n  return result;\n};\n"],"mappings":";;;;;;AA4BA,MAAa,kBAAkB,EAC7B,QACA,oBAC2B;CAC3B,MAAM,CAAC,UAAU,gBAAgB,SAAuB,CAAC,CAAC;CAC1D,MAAM,CAAC,YAAY,iBAAiB,SAAkC,CAAC,CAAC;CACxE,MAAM,CAAC,WAAW,gBAAgB,SAG/B,CAAC,CAAC;CACL,MAAM,CAAC,oBAAoB,yBAAyB,SAAmB,CAAC,CAAC;CACzE,MAAM,CAAC,eAAe,oBAAoB,SACxC,CAAC,CACH;CACA,MAAM,CAAC,mBAAmB,wBAAwB,SAEhD,CAAC,CAAC;CACJ,MAAM,CAAC,cAAc,mBAAmB,SAA2B,CAAC,CAAC;CACrE,MAAM,CAAC,WAAW,gBAAgB,SAAS,KAAK;CAChD,MAAM,CAAC,iBAAiB,sBAAsB,yBAE5C,IAAI,IAAI,CAAC;CACX,MAAM,yBAAyB,OAA2B,KAAA,CAAS;CACnE,MAAM,cAAc,OAAO,QAAQ;CACnC,YAAY,UAAU;CACtB,MAAM,gBAAgB,OAAO,UAAU;CACvC,cAAc,UAAU;CACxB,MAAM,mBAAmB,OAAO,aAAa;CAC7C,iBAAiB,UAAU;CAC3B,MAAM,qBAAqB,OAAO,eAAe;CACjD,mBAAmB,UAAU;CAE7B,MAAM,uBAAuB,aAAa,SAAuB;EAC/D,YAAY,UAAU;EACtB,aAAa,IAAI;CACnB,GAAG,CAAC,CAAC;CAML,MAAM,kBAAkB,aACrB,SAAuB;EACtB,qBAAqB,IAAI;EACzB,sBAAsB,CAAC,CAAC;EACxB,qBAAqB,CAAC,CAAC;EACvB,gBAAgB,CAAC,CAAC;EAClB,aAAa,KAAK;EAClB,mCAAmB,IAAI,IAAI,CAAC;CAC9B,GACA,CAAC,oBAAoB,CACvB;CAEA,MAAM,qBAAqB,OAA+B,IAAI;CAE9D,MAAM,EAAE,SAAS,eAAe,oBAAoB,cAC5C,iBAAiB,CAAC,GACxB,CAAC,aAAa,CAChB;CAEA,MAAM,MAAM,OAAO;CAgGnB,OAAO;EACL;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA,aAzGkB,YAClB,OAAO,aAA2B,WAAiC;GACjE,MAAM,oBAAoB,YAAY,KAAK,MACzC,EAAE,KAAK,IAAI;IAAE,GAAG;IAAG,IAAIA,GAAO;GAAE,CAClC;GAEA,MAAM,cAAc,IAAI,oBAAoB,YAAY,OAAO;GAC/D,KAAK,MAAM,OAAO,mBAChB,YAAY,aAAa,eAAe,GAAG,CAAC;GAE9C,qBAAqB,YAAY,YAAY,CAAC;GAE9C,MAAM,kBAAkB,IAAI,gBAAgB;GAC5C,mBAAmB,UAAU;GAE7B,IAAI;IACF,MAAM,WAAW,MAAM,OAAO,mBAAmB;KAC/C,GAAG;KACH,aAAa,gBAAgB;KAC7B,YAAY,YAAY;MACtB,OAAO,MAAM,IAAI,eAAe,EAAE,WAAW;KAC/C;IACF,CAAC;IAED,WAAW,MAAM,SAAS,UAAU;KAElC,qBADwB,YAAY,aAAa,KACd,CAAC;KACpC,cAAc;MACZ,GAAG,cAAc;MACjB,GAAG,YAAY,cAAc;KAC/B,CAAC;KACD,aAAa,YAAY,aAAa,CAAC;KACvC,sBAAsB,YAAY,sBAAsB,CAAC;KACzD,iBAAiB;MACf,GAAG,iBAAiB;MACpB,GAAG,YAAY,iBAAiB;KAClC,CAAC;KACD,qBAAqB,YAAY,qBAAqB,CAAC;KACvD,gBAAgB,YAAY,gBAAgB,CAAC;KAC7C,aAAa,YAAY,YAAY,CAAC;KACtC;MACE,MAAM,UAAU,YAAY,mBAAmB;MAC/C,IAAI,QAAQ,OAAO,GACjB,mBACE,IAAI,IAAI,CAAC,GAAG,mBAAmB,SAAS,GAAG,OAAO,CAAC,CACrD;KAEJ;KAEA,MAAM,WAAW,YAAY,uBAAuB;KACpD,IAAI,YAAY,aAAa,uBAAuB,SAAS;MAC3D,uBAAuB,UAAU;MACjC,kBAAkB,QAAQ;KAC5B;KAGA,IAAI,MAAM,kBAAkB,eAC1B,KAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,MAAM,cAAc,GAC5D,cAAc,KAAK,KAAK;KAI5B,IAAI,MAAM,aAAa,MAAM,cAC3B,UAAU,MAAM,gBAAgB,MAAM,SAAS;IAEnD;GACF,SAAS,OAAO;IACd,IACE,CAAC,gBAAgB,OAAO,WACxB,EAAE,iBAAiB,SAAS,MAAM,SAAS,eAE3C,MAAM;GAEV,UAAU;IACR,IAAI,mBAAmB,YAAY,iBACjC,mBAAmB,UAAU;GAEjC;EACF,GACA;GACE;GACA;GACA;GACA;GACA;GACA;EACF,CAmBU;EACV,QAjBa,kBAAkB;GAC/B,IAAI,mBAAmB,SACrB,mBAAmB,QAAQ,MAAM;EAErC,GAAG,CAAC,CAaG;EACL,aAAa;EACb;CACF;AACF;;AAGA,MAAa,kBAAkB,QAA8B;CAC3D,IAAI,IAAI,SAAS,SACf,OAAO;EACL,IAAI,IAAI,MAAMA,GAAO;EACrB,QAAQ;EACR,SAAS;GAAE,MAAM;GAAQ,OAAO,eAAe,IAAI,OAAO;EAAE;CAC9D;CAGF,IAAI,IAAI,SAAS,QAAQ;EACvB,IAAI;EACJ,IAAI;GACF,WAAW,KAAK,MAAM,IAAI,OAAO;EACnC,QAAQ;GACN,WAAW,IAAI;EACjB;EACA,OAAO;GACL,IAAI,IAAI,MAAMA,GAAO;GACrB,SAAS;IACP,MAAM;IACN,OAAO,CACL,EACE,kBAAkB;KAChB,MAAM,IAAI;KACV,IAAI,IAAI;KACR;IACF,EACF,CACF;GACF;EACF;CACF;CAEA,MAAM,SAAmB,EAAE,IAAI,IAAI,MAAMA,GAAO,EAAE;CAClD,IAAI,IAAI,UAAU,MAAM,OAAO,SAAS,IAAI;CAC5C,OAAO,UAAU;EACf,MAAM;EACN,OAAO,CACL,GAAG,eAAe,IAAI,OAAO,GAC7B,GAAI,IAAI,YAAY,KAAK,QAAQ,EAC/B,cAAc;GAAE,MAAM,GAAG;GAAM,IAAI,GAAG;GAAI,MAAM,EAAE,GAAG,GAAG,KAAK;EAAE,EACjE,EAAE,KAAK,CAAC,CACV;CACF;CACA,OAAO;AACT"}

package/dist/useAdkRuntime.d.ts

@@ -1,37 +1,46 @@
 /// <reference types="@assistant-ui/core/store" />
-import { type AttachmentAdapter, type DictationAdapter, type FeedbackAdapter, type SpeechSynthesisAdapter } from "@assistant-ui/core";
-import type { AssistantCloud } from "assistant-cloud";
-import type { RemoteThreadListAdapter } from "@assistant-ui/core";
-import type { AdkMessage, AdkStreamCallback, OnAdkErrorCallback, OnAdkCustomEventCallback, OnAdkAgentTransferCallback } from "./types.js";
-export type UseAdkRuntimeOptions = {
-    stream: AdkStreamCallback;
-    autoCancelPendingToolCalls?: boolean | undefined;
-    unstable_allowCancellation?: boolean | undefined;
-    getCheckpointId?: (threadId: string, parentMessages: AdkMessage[]) => Promise<string | null>;
-    load?: (threadId: string) => Promise<{
-        messages: AdkMessage[];
-    }>;
-    create?: () => Promise<{
-        externalId: string;
-    }>;
-    delete?: (threadId: string) => Promise<void>;
-    adapters?: {
-        attachments?: AttachmentAdapter;
-        speech?: SpeechSynthesisAdapter;
-        dictation?: DictationAdapter;
-        feedback?: FeedbackAdapter;
-    } | undefined;
-    eventHandlers?: {
-        onError?: OnAdkErrorCallback;
-        onCustomEvent?: OnAdkCustomEventCallback;
-        onAgentTransfer?: OnAdkAgentTransferCallback;
-    } | undefined;
-    cloud?: AssistantCloud | undefined;
-    /**
-     * A `RemoteThreadListAdapter` to use instead of the cloud adapter.
-     * Use with `createAdkSessionAdapter` for ADK session-backed persistence.
-     */
-    sessionAdapter?: RemoteThreadListAdapter | undefined;
+import { AdkMessage, AdkStreamCallback, OnAdkAgentTransferCallback, OnAdkCustomEventCallback, OnAdkErrorCallback } from "./types.js";
+import { AttachmentAdapter, DictationAdapter, FeedbackAdapter, RemoteThreadListAdapter, SpeechSynthesisAdapter } from "@assistant-ui/core";
+import { AssistantCloud } from "assistant-cloud";
+
+//#region src/useAdkRuntime.d.ts
+type UseAdkRuntimeOptions = {
+  stream: AdkStreamCallback;
+  autoCancelPendingToolCalls?: boolean | undefined;
+  unstable_allowCancellation?: boolean | undefined;
+  getCheckpointId?: (threadId: string, parentMessages: AdkMessage[]) => Promise<string | null>;
+  load?: (threadId: string) => Promise<{
+    messages: AdkMessage[];
+  }>;
+  create?: () => Promise<{
+    externalId: string;
+  }>;
+  delete?: (threadId: string) => Promise<void>;
+  adapters?: {
+    attachments?: AttachmentAdapter;
+    speech?: SpeechSynthesisAdapter;
+    dictation?: DictationAdapter;
+    feedback?: FeedbackAdapter;
+  } | undefined;
+  eventHandlers?: {
+    onError?: OnAdkErrorCallback;
+    onCustomEvent?: OnAdkCustomEventCallback;
+    onAgentTransfer?: OnAdkAgentTransferCallback;
+  } | undefined;
+  cloud?: AssistantCloud | undefined;
+  /**
+   * A `RemoteThreadListAdapter` to use instead of the cloud adapter.
+   * Use with `createAdkSessionAdapter` for ADK session-backed persistence.
+   */
+  sessionAdapter?: RemoteThreadListAdapter | undefined;
 };
-export declare const useAdkRuntime: ({ cloud, sessionAdapter, create, delete: deleteFn, ...options }: UseAdkRuntimeOptions) => import("@assistant-ui/core").AssistantRuntime;
+declare const useAdkRuntime: ({
+  cloud,
+  sessionAdapter,
+  create,
+  delete: deleteFn,
+  ...options
+}: UseAdkRuntimeOptions) => import("@assistant-ui/core").AssistantRuntime;
+//#endregion
+export { UseAdkRuntimeOptions, useAdkRuntime };
 //# sourceMappingURL=useAdkRuntime.d.ts.map

package/dist/useAdkRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAdkRuntime.d.ts","sourceRoot":"","sources":["../src/useAdkRuntime.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,KAAK,iBAAiB,EACtB,KAAK,gBAAgB,EACrB,KAAK,eAAe,EACpB,KAAK,sBAAsB,EAG5B,MAAM,oBAAoB,CAAC;AAU5B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAClE,OAAO,KAAK,EACV,UAAU,EAEV,iBAAiB,EACjB,kBAAkB,EAClB,wBAAwB,EACxB,0BAA0B,EAC3B,mBAAgB;AA0GjB,MAAM,MAAM,oBAAoB,GAAG;IACjC,MAAM,EAAE,iBAAiB,CAAC;IAC1B,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD,eAAe,CAAC,EAAE,CAChB,QAAQ,EAAE,MAAM,EAChB,cAAc,EAAE,UAAU,EAAE,KACzB,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAC5B,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC;QAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;KAAE,CAAC,CAAC;IACjE,MAAM,CAAC,EAAE,MAAM,OAAO,CAAC;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC/C,MAAM,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,QAAQ,CAAC,EACL;QACE,WAAW,CAAC,EAAE,iBAAiB,CAAC;QAChC,MAAM,CAAC,EAAE,sBAAsB,CAAC;QAChC,SAAS,CAAC,EAAE,gBAAgB,CAAC;QAC7B,QAAQ,CAAC,EAAE,eAAe,CAAC;KAC5B,GACD,SAAS,CAAC;IACd,aAAa,CAAC,EACV;QACE,OAAO,CAAC,EAAE,kBAAkB,CAAC;QAC7B,aAAa,CAAC,EAAE,wBAAwB,CAAC;QACzC,eAAe,CAAC,EAAE,0BAA0B,CAAC;KAC9C,GACD,SAAS,CAAC;IACd,KAAK,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACnC;;;OAGG;IACH,cAAc,CAAC,EAAE,uBAAuB,GAAG,SAAS,CAAC;CACtD,CAAC;AAiPF,eAAO,MAAM,aAAa,GAAI,iEAM3B,oBAAoB,kDAsBtB,CAAC"}
+{"version":3,"file":"useAdkRuntime.d.ts","names":[],"sources":["../src/useAdkRuntime.ts"],"mappings":";;;;;KA2IY,oBAAA;EACV,MAAA,EAAQ,iBAAA;EACR,0BAAA;EACA,0BAAA;EACA,eAAA,IACE,QAAA,UACA,cAAA,EAAgB,UAAA,OACb,OAAA;EACL,IAAA,IAAQ,QAAA,aAAqB,OAAA;IAAU,QAAA,EAAU,UAAA;EAAA;EACjD,MAAA,SAAe,OAAA;IAAU,UAAA;EAAA;EACzB,MAAA,IAAU,QAAA,aAAqB,OAAA;EAC/B,QAAA;IAEM,WAAA,GAAc,iBAAA;IACd,MAAA,GAAS,sBAAA;IACT,SAAA,GAAY,gBAAA;IACZ,QAAA,GAAW,eAAA;EAAA;EAGjB,aAAA;IAEM,OAAA,GAAU,kBAAA;IACV,aAAA,GAAgB,wBAAA;IAChB,eAAA,GAAkB,0BAAA;EAAA;EAGxB,KAAA,GAAQ,cAAA;EAxBR;;;;EA6BA,cAAA,GAAiB,uBAAA;AAAA;AAAA,cAkPN,aAAA;EAAiB,KAAA;EAAA,cAAA;EAAA,MAAA;EAAA,MAAA,EAAA,QAAA;EAAA,GAAA;AAAA,GAM3B,oBAAA,kCAAoB,gBAAA"}