@assistant-ui/react-a2a 0.2.5 → 0.2.7

package/dist/useA2ARuntime.js

@@ -1,188 +1,142 @@
-import { useEffect, useRef, useState } from "react";
-import { useExternalMessageConverter, useExternalStoreRuntime, useThread, useThreadListItemRuntime, } from "@assistant-ui/react";
-import { convertA2AMessage } from "./convertA2AMessages.js";
-import { useA2AMessages } from "./useA2AMessages.js";
-import { appendA2AChunk } from "./appendA2AChunk.js";
-const getPendingToolCalls = (messages) => {
-    const pendingToolCalls = new Map();
-    for (const message of messages) {
-        if (message.role === "assistant") {
-            for (const toolCall of message.tool_calls ?? []) {
-                pendingToolCalls.set(toolCall.id, toolCall);
-            }
-        }
-        if (message.role === "tool") {
-            pendingToolCalls.delete(message.tool_call_id);
-        }
-    }
-    return [...pendingToolCalls.values()];
-};
-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", image_url: { url: part.image } };
-            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 symbolA2ARuntimeExtras = Symbol("a2a-runtime-extras");
-const asA2ARuntimeExtras = (extras) => {
+/// <reference types="@assistant-ui/core/store" />
+"use client";
+import { useCallback, useEffect, useMemo, useRef, useState } from "react";
+import { useExternalStoreRuntime, useRuntimeAdapters, } from "@assistant-ui/core/react";
+import { useAuiState } from "@assistant-ui/store";
+import { A2AClient } from "./A2AClient.js";
+import { A2AThreadRuntimeCore } from "./A2AThreadRuntimeCore.js";
+// --- Extras symbol for A2A-specific state ---
+const symbolA2AExtras = Symbol("a2a-extras");
+const asA2AExtras = (extras) => {
     if (typeof extras !== "object" ||
         extras == null ||
-        !(symbolA2ARuntimeExtras in extras))
-        throw new Error("This method can only be called when you are using useA2ARuntime");
+        !(symbolA2AExtras in extras))
+        throw new Error("This hook can only be used inside a useA2ARuntime provider");
     return extras;
 };
-export const useA2ATaskState = () => {
-    const { taskState } = useThread((t) => asA2ARuntimeExtras(t.extras));
-    return taskState;
+// --- Public hooks for A2A state ---
+export const useA2ATask = () => {
+    return useAuiState((s) => asA2AExtras(s.thread.extras).task);
 };
 export const useA2AArtifacts = () => {
-    const { artifacts } = useThread((t) => asA2ARuntimeExtras(t.extras));
-    return artifacts;
+    return useAuiState((s) => asA2AExtras(s.thread.extras).artifacts);
 };
-export const useA2ASend = () => {
-    const { send } = useThread((t) => asA2ARuntimeExtras(t.extras));
-    return send;
+export const useA2AAgentCard = () => {
+    return useAuiState((s) => asA2AExtras(s.thread.extras).agentCard);
 };
-export const useA2ARuntime = ({ autoCancelPendingToolCalls, adapters: { attachments, feedback, speech } = {}, unstable_allowCancellation, stream, contextId, onSwitchToNewThread, onSwitchToThread, eventHandlers, }) => {
-    const { taskState, artifacts, setArtifacts, messages, sendMessage, cancel, setMessages, } = useA2AMessages({
-        appendMessage: appendA2AChunk,
-        stream,
-        ...(eventHandlers && { eventHandlers }),
-    });
-    const [isRunning, setIsRunning] = useState(false);
-    const handleSendMessage = async (messages, config) => {
-        try {
-            setIsRunning(true);
-            await sendMessage(messages, config);
+// --- Main hook ---
+export function useA2ARuntime(options) {
+    const [_version, setVersion] = useState(0);
+    const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);
+    const runtimeAdapters = useRuntimeAdapters();
+    const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;
+    const threadListAdapter = options.adapters?.threadList;
+    // Create or reuse client
+    const clientRef = useRef(null);
+    if (!clientRef.current) {
+        if (options.client) {
+            clientRef.current = options.client;
         }
-        catch (error) {
-            console.error("Error streaming A2A messages:", error);
+        else if (options.baseUrl) {
+            clientRef.current = new A2AClient({
+                baseUrl: options.baseUrl,
+                headers: options.headers,
+            });
         }
-        finally {
-            setIsRunning(false);
+        else {
+            throw new Error("useA2ARuntime requires either `client` or `baseUrl`");
         }
-    };
-    const threadMessages = useExternalMessageConverter({
-        callback: convertA2AMessage,
-        messages,
-        isRunning,
+    }
+    const client = clientRef.current;
+    // Create or reuse core
+    const coreRef = useRef(null);
+    if (!coreRef.current) {
+        coreRef.current = new A2AThreadRuntimeCore({
+            client,
+            contextId: options.contextId,
+            configuration: options.configuration,
+            ...(options.onError && { onError: options.onError }),
+            ...(options.onCancel && { onCancel: options.onCancel }),
+            ...(options.onArtifactComplete && {
+                onArtifactComplete: options.onArtifactComplete,
+            }),
+            ...(historyAdapter && { history: historyAdapter }),
+            notifyUpdate,
+        });
+    }
+    const core = coreRef.current;
+    core.updateOptions({
+        client,
+        contextId: options.contextId,
+        configuration: options.configuration,
+        ...(options.onError && { onError: options.onError }),
+        ...(options.onCancel && { onCancel: options.onCancel }),
+        ...(options.onArtifactComplete && {
+            onArtifactComplete: options.onArtifactComplete,
+        }),
+        ...(historyAdapter && { history: historyAdapter }),
     });
-    const switchToThread = !onSwitchToThread
-        ? undefined
-        : async (externalId) => {
-            const { messages, artifacts } = await onSwitchToThread(externalId);
-            setMessages(messages);
-            if (artifacts) {
-                setArtifacts(artifacts);
-            }
+    // Thread list
+    const threadList = useMemo(() => {
+        if (!threadListAdapter)
+            return undefined;
+        const { onSwitchToNewThread, onSwitchToThread } = threadListAdapter;
+        return {
+            threadId: threadListAdapter.threadId,
+            onSwitchToNewThread: onSwitchToNewThread
+                ? async () => {
+                    await onSwitchToNewThread();
+                    core.applyExternalMessages([]);
+                }
+                : undefined,
+            onSwitchToThread: onSwitchToThread
+                ? async (threadId) => {
+                    const result = await onSwitchToThread(threadId);
+                    core.applyExternalMessages(result.messages);
+                }
+                : undefined,
         };
-    const threadList = {
-        threadId: contextId,
-        onSwitchToNewThread: !onSwitchToNewThread
-            ? undefined
-            : async () => {
-                await onSwitchToNewThread();
-                setMessages([]);
-                setArtifacts([]);
+    }, [threadListAdapter, core]);
+    // Adapters
+    const adapters = options.adapters;
+    const adapterAdapters = useMemo(() => ({
+        attachments: adapters?.attachments ?? runtimeAdapters?.attachments,
+        speech: adapters?.speech,
+        feedback: adapters?.feedback,
+        threadList,
+    }), [adapters, runtimeAdapters, threadList]);
+    // Build store adapter
+    const store = useMemo(() => {
+        void _version;
+        return {
+            isLoading: core.isLoading,
+            messages: core.getMessages(),
+            isRunning: core.isRunning(),
+            extras: {
+                [symbolA2AExtras]: true,
+                task: core.getTask(),
+                artifacts: core.getArtifacts(),
+                agentCard: core.getAgentCard(),
             },
-        onSwitchToThread: switchToThread,
-    };
-    const loadingRef = useRef(false);
-    const threadListItemRuntime = useThreadListItemRuntime({ optional: true });
+            onNew: (message) => core.append(message),
+            onEdit: (message) => core.edit(message),
+            onReload: (parentId) => core.reload(parentId),
+            onCancel: () => core.cancel(),
+            setMessages: (messages) => core.applyExternalMessages(messages),
+            onImport: (messages) => core.applyExternalMessages(messages),
+            adapters: adapterAdapters,
+        };
+    }, [adapterAdapters, core, _version]);
+    const runtime = useExternalStoreRuntime(store);
     useEffect(() => {
-        if (!threadListItemRuntime || !switchToThread || loadingRef.current)
-            return;
-        const externalId = threadListItemRuntime.getState().externalId;
-        if (externalId) {
-            loadingRef.current = true;
-            switchToThread(externalId).finally(() => {
-                loadingRef.current = false;
-            });
-        }
-    }, [switchToThread, threadListItemRuntime]);
-    return useExternalStoreRuntime({
-        isRunning,
-        messages: threadMessages,
-        adapters: {
-            attachments,
-            feedback,
-            speech,
-            threadList,
-        },
-        extras: {
-            [symbolA2ARuntimeExtras]: true,
-            taskState,
-            artifacts,
-            send: handleSendMessage,
-        },
-        onNew: (msg) => {
-            const cancellations = autoCancelPendingToolCalls !== false
-                ? getPendingToolCalls(messages).map((t) => ({
-                    role: "tool",
-                    tool_call_id: t.id,
-                    content: JSON.stringify({ cancelled: true }),
-                    status: {
-                        type: "incomplete",
-                        reason: "cancelled",
-                    },
-                }))
-                : [];
-            const config = {};
-            if (contextId !== undefined)
-                config.contextId = contextId;
-            if (msg.runConfig !== undefined)
-                config.runConfig = msg.runConfig;
-            return handleSendMessage([
-                ...cancellations,
-                {
-                    role: "user",
-                    content: getMessageContent(msg),
-                },
-            ], config);
-        },
-        onAddToolResult: async ({ toolCallId, toolName: _toolName, result, isError, artifact, }) => {
-            // TODO parallel human in the loop calls
-            const message = {
-                role: "tool",
-                tool_call_id: toolCallId,
-                content: JSON.stringify(result),
-                status: isError
-                    ? { type: "incomplete", reason: "error" }
-                    : { type: "complete", reason: "stop" },
-            };
-            if (artifact) {
-                message.artifacts = [artifact];
-            }
-            const config = {};
-            if (contextId !== undefined)
-                config.contextId = contextId;
-            await handleSendMessage([message], 
-            // TODO reuse runconfig here!
-            config);
-        },
-        onCancel: unstable_allowCancellation
-            ? async () => {
-                cancel();
-            }
-            : undefined,
-    });
-};
+        core.attachRuntime(runtime);
+        return () => {
+            core.detachRuntime();
+        };
+    }, [core, runtime]);
+    useEffect(() => {
+        core.__internal_load();
+    }, [core]);
+    return runtime;
+}
 //# sourceMappingURL=useA2ARuntime.js.map

package/dist/useA2ARuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useA2ARuntime.js","sourceRoot":"","sources":["../src/useA2ARuntime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAcpD,OAAO,EACL,2BAA2B,EAC3B,uBAAuB,EACvB,SAAS,EACT,wBAAwB,GACzB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,iBAAiB,EAAE,gCAA6B;AACzD,OAAO,EAAE,cAAc,EAAE,4BAAyB;AAMlD,OAAO,EAAE,cAAc,EAAE,4BAAyB;AAElD,MAAM,mBAAmB,GAAG,CAAC,QAAsB,EAAE,EAAE;IACrD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAuB,CAAC;IACxD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACjC,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,UAAU,IAAI,EAAE,EAAE,CAAC;gBAChD,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,YAAa,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC;AACxC,CAAC,CAAC;AAEF,MAAM,iBAAiB,GAAG,CAAC,GAAkB,EAAE,EAAE;IAC/C,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,SAAS,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC;YAExE,KAAK,WAAW;gBACd,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAE1D;gBACE,MAAM,gBAAgB,GAKT,IAAI,CAAC;gBAClB,MAAM,IAAI,KAAK,CACb,yCAAyC,gBAAgB,EAAE,CAC5D,CAAC;QACN,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,MAAM,sBAAsB,GAAG,MAAM,CAAC,oBAAoB,CAAC,CAAC;AAQ5D,MAAM,kBAAkB,GAAG,CAAC,MAAe,EAAoB,EAAE;IAC/D,IACE,OAAO,MAAM,KAAK,QAAQ;QAC1B,MAAM,IAAI,IAAI;QACd,CAAC,CAAC,sBAAsB,IAAI,MAAM,CAAC;QAEnC,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;IAEJ,OAAO,MAA0B,CAAC;AACpC,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAG,GAAG,EAAE;IAClC,MAAM,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;IACrE,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAG,GAAG,EAAE;IAClC,MAAM,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;IACrE,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,UAAU,GAAG,GAAG,EAAE;IAC7B,MAAM,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;IAChE,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,EAC5B,0BAA0B,EAC1B,QAAQ,EAAE,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,EAChD,0BAA0B,EAC1B,MAAM,EACN,SAAS,EACT,mBAAmB,EACnB,gBAAgB,EAChB,aAAa,GAmDd,EAAE,EAAE;IACH,MAAM,EACJ,SAAS,EACT,SAAS,EACT,YAAY,EACZ,QAAQ,EACR,WAAW,EACX,MAAM,EACN,WAAW,GACZ,GAAG,cAAc,CAAC;QACjB,aAAa,EAAE,cAAc;QAC7B,MAAM;QACN,GAAG,CAAC,aAAa,IAAI,EAAE,aAAa,EAAE,CAAC;KACxC,CAAC,CAAC;IAEH,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAClD,MAAM,iBAAiB,GAAG,KAAK,EAC7B,QAAsB,EACtB,MAA4B,EAC5B,EAAE;QACF,IAAI,CAAC;YACH,YAAY,CAAC,IAAI,CAAC,CAAC;YACnB,MAAM,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;QACxD,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,cAAc,GAAG,2BAA2B,CAAC;QACjD,QAAQ,EAAE,iBAAiB;QAC3B,QAAQ;QACR,SAAS;KACV,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,CAAC,gBAAgB;QACtC,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,KAAK,EAAE,UAAkB,EAAE,EAAE;YAC3B,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,MAAM,gBAAgB,CAAC,UAAU,CAAC,CAAC;YACnE,WAAW,CAAC,QAAQ,CAAC,CAAC;YACtB,IAAI,SAAS,EAAE,CAAC;gBACd,YAAY,CAAC,SAAS,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC,CAAC;IAEN,MAAM,UAAU,GAEE;QAChB,QAAQ,EAAE,SAAS;QACnB,mBAAmB,EAAE,CAAC,mBAAmB;YACvC,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,KAAK,IAAI,EAAE;gBACT,MAAM,mBAAmB,EAAE,CAAC;gBAC5B,WAAW,CAAC,EAAE,CAAC,CAAC;gBAChB,YAAY,CAAC,EAAE,CAAC,CAAC;YACnB,CAAC;QACL,gBAAgB,EAAE,cAAc;KACjC,CAAC;IAEF,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IACjC,MAAM,qBAAqB,GAAG,wBAAwB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3E,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,qBAAqB,IAAI,CAAC,cAAc,IAAI,UAAU,CAAC,OAAO;YAAE,OAAO;QAE5E,MAAM,UAAU,GAAG,qBAAqB,CAAC,QAAQ,EAAE,CAAC,UAAU,CAAC;QAC/D,IAAI,UAAU,EAAE,CAAC;YACf,UAAU,CAAC,OAAO,GAAG,IAAI,CAAC;YAC1B,cAAc,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;gBACtC,UAAU,CAAC,OAAO,GAAG,KAAK,CAAC;YAC7B,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EAAE,CAAC,cAAc,EAAE,qBAAqB,CAAC,CAAC,CAAC;IAE5C,OAAO,uBAAuB,CAAC;QAC7B,SAAS;QACT,QAAQ,EAAE,cAAc;QACxB,QAAQ,EAAE;YACR,WAAW;YACX,QAAQ;YACR,MAAM;YACN,UAAU;SACX;QACD,MAAM,EAAE;YACN,CAAC,sBAAsB,CAAC,EAAE,IAAI;YAC9B,SAAS;YACT,SAAS;YACT,IAAI,EAAE,iBAAiB;SACG;QAC5B,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE;YACb,MAAM,aAAa,GACjB,0BAA0B,KAAK,KAAK;gBAClC,CAAC,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,GAAG,CAC/B,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC;oBACC,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,CAAC,CAAC,EAAE;oBAClB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;oBAC5C,MAAM,EAAE;wBACN,IAAI,EAAE,YAAY;wBAClB,MAAM,EAAE,WAAW;qBACpB;iBACF,CAAyC,CAC7C;gBACH,CAAC,CAAC,EAAE,CAAC;YAET,MAAM,MAAM,GAAyB,EAAE,CAAC;YACxC,IAAI,SAAS,KAAK,SAAS;gBAAE,MAAM,CAAC,SAAS,GAAG,SAAS,CAAC;YAC1D,IAAI,GAAG,CAAC,SAAS,KAAK,SAAS;gBAAE,MAAM,CAAC,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC;YAClE,OAAO,iBAAiB,CACtB;gBACE,GAAG,aAAa;gBAChB;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,iBAAiB,CAAC,GAAG,CAAC;iBAChC;aACF,EACD,MAAM,CACP,CAAC;QACJ,CAAC;QACD,eAAe,EAAE,KAAK,EAAE,EACtB,UAAU,EACV,QAAQ,EAAE,SAAS,EACnB,MAAM,EACN,OAAO,EACP,QAAQ,GACT,EAAE,EAAE;YACH,wCAAwC;YACxC,MAAM,OAAO,GAAe;gBAC1B,IAAI,EAAE,MAAM;gBACZ,YAAY,EAAE,UAAU;gBACxB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;gBAC/B,MAAM,EAAE,OAAO;oBACb,CAAC,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE;oBACzC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE;aACzC,CAAC;YACF,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,CAAC,SAAS,GAAG,CAAC,QAAQ,CAAkB,CAAC;YAClD,CAAC;YACD,MAAM,MAAM,GAAyB,EAAE,CAAC;YACxC,IAAI,SAAS,KAAK,SAAS;gBAAE,MAAM,CAAC,SAAS,GAAG,SAAS,CAAC;YAC1D,MAAM,iBAAiB,CACrB,CAAC,OAAO,CAAC;YACT,6BAA6B;YAC7B,MAAM,CACP,CAAC;QACJ,CAAC;QACD,QAAQ,EAAE,0BAA0B;YAClC,CAAC,CAAC,KAAK,IAAI,EAAE;gBACT,MAAM,EAAE,CAAC;YACX,CAAC;YACH,CAAC,CAAC,SAAS;KACd,CAAC,CAAC;AACL,CAAC,CAAC"}
+{"version":3,"file":"useA2ARuntime.js","sourceRoot":"","sources":["../src/useA2ARuntime.ts"],"names":[],"mappings":"AAAA,kDAAkD;AAClD,YAAY,CAAC;AAEb,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC1E,OAAO,EACL,uBAAuB,EACvB,kBAAkB,GACnB,MAAM,0BAA0B,CAAC;AAWlC,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,uBAAoB;AAExC,OAAO,EAAE,oBAAoB,EAAE,kCAA+B;AAQ9D,+CAA+C;AAE/C,MAAM,eAAe,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;AAS7C,MAAM,WAAW,GAAG,CAAC,MAAe,EAAa,EAAE;IACjD,IACE,OAAO,MAAM,KAAK,QAAQ;QAC1B,MAAM,IAAI,IAAI;QACd,CAAC,CAAC,eAAe,IAAI,MAAM,CAAC;QAE5B,MAAM,IAAI,KAAK,CACb,4DAA4D,CAC7D,CAAC;IACJ,OAAO,MAAmB,CAAC;AAC7B,CAAC,CAAC;AAEF,qCAAqC;AAErC,MAAM,CAAC,MAAM,UAAU,GAAG,GAAG,EAAE;IAC7B,OAAO,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAG,GAAG,EAAE;IAClC,OAAO,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC;AACpE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAG,GAAG,EAAE;IAClC,OAAO,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC;AACpE,CAAC,CAAC;AA2CF,oBAAoB;AAEpB,MAAM,UAAU,aAAa,CAAC,OAA6B;IACzD,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;IAC3C,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACrE,MAAM,eAAe,GAAG,kBAAkB,EAAE,CAAC;IAC7C,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,EAAE,OAAO,IAAI,eAAe,EAAE,OAAO,CAAC;IAC7E,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,EAAE,UAAU,CAAC;IAEvD,yBAAyB;IACzB,MAAM,SAAS,GAAG,MAAM,CAAmB,IAAI,CAAC,CAAC;IACjD,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QACvB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,SAAS,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;QACrC,CAAC;aAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YAC3B,SAAS,CAAC,OAAO,GAAG,IAAI,SAAS,CAAC;gBAChC,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;aACzB,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;IACD,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC;IAEjC,uBAAuB;IACvB,MAAM,OAAO,GAAG,MAAM,CAA8B,IAAI,CAAC,CAAC;IAC1D,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,CAAC,OAAO,GAAG,IAAI,oBAAoB,CAAC;YACzC,MAAM;YACN,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;YACpD,GAAG,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC;YACvD,GAAG,CAAC,OAAO,CAAC,kBAAkB,IAAI;gBAChC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;aAC/C,CAAC;YACF,GAAG,CAAC,cAAc,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;YAClD,YAAY;SACb,CAAC,CAAC;IACL,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;IAC7B,IAAI,CAAC,aAAa,CAAC;QACjB,MAAM;QACN,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,aAAa,EAAE,OAAO,CAAC,aAAa;QACpC,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;QACpD,GAAG,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC;QACvD,GAAG,CAAC,OAAO,CAAC,kBAAkB,IAAI;YAChC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;SAC/C,CAAC;QACF,GAAG,CAAC,cAAc,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;KACnD,CAAC,CAAC;IAEH,cAAc;IACd,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,EAAE;QAC9B,IAAI,CAAC,iBAAiB;YAAE,OAAO,SAAS,CAAC;QAEzC,MAAM,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,GAAG,iBAAiB,CAAC;QAEpE,OAAO;YACL,QAAQ,EAAE,iBAAiB,CAAC,QAAQ;YACpC,mBAAmB,EAAE,mBAAmB;gBACtC,CAAC,CAAC,KAAK,IAAI,EAAE;oBACT,MAAM,mBAAmB,EAAE,CAAC;oBAC5B,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;gBACjC,CAAC;gBACH,CAAC,CAAC,SAAS;YACb,gBAAgB,EAAE,gBAAgB;gBAChC,CAAC,CAAC,KAAK,EAAE,QAAgB,EAAE,EAAE;oBACzB,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,QAAQ,CAAC,CAAC;oBAChD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBAC9C,CAAC;gBACH,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC,EAAE,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC,CAAC;IAE9B,WAAW;IACX,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IAClC,MAAM,eAAe,GAAG,OAAO,CAC7B,GAAG,EAAE,CAAC,CAAC;QACL,WAAW,EAAE,QAAQ,EAAE,WAAW,IAAI,eAAe,EAAE,WAAW;QAClE,MAAM,EAAE,QAAQ,EAAE,MAAM;QACxB,QAAQ,EAAE,QAAQ,EAAE,QAAQ;QAC5B,UAAU;KACX,CAAC,EACF,CAAC,QAAQ,EAAE,eAAe,EAAE,UAAU,CAAC,CACxC,CAAC;IAEF,sBAAsB;IACtB,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,EAAE;QACzB,KAAK,QAAQ,CAAC;QAEd,OAAO;YACL,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE;YAC5B,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE;YAC3B,MAAM,EAAE;gBACN,CAAC,eAAe,CAAC,EAAE,IAAI;gBACvB,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE;gBACpB,SAAS,EAAE,IAAI,CAAC,YAAY,EAAE;gBAC9B,SAAS,EAAE,IAAI,CAAC,YAAY,EAAE;aACX;YACrB,KAAK,EAAE,CAAC,OAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YACvD,MAAM,EAAE,CAAC,OAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC;YACtD,QAAQ,EAAE,CAAC,QAAuB,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;YAC5D,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE;YAC7B,WAAW,EAAE,CAAC,QAAkC,EAAE,EAAE,CAClD,IAAI,CAAC,qBAAqB,CAAC,QAAQ,CAAC;YACtC,QAAQ,EAAE,CAAC,QAAkC,EAAE,EAAE,CAC/C,IAAI,CAAC,qBAAqB,CAAC,QAAQ,CAAC;YACtC,QAAQ,EAAE,eAAe;SACoB,CAAC;IAClD,CAAC,EAAE,CAAC,eAAe,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAEtC,MAAM,OAAO,GAAG,uBAAuB,CAAC,KAAK,CAAC,CAAC;IAE/C,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC5B,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,aAAa,EAAE,CAAC;QACvB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC;IAEpB,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;IAEX,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@assistant-ui/react-a2a",
-  "version": "0.2.5",
-  "description": "A2A protocol adapter for assistant-ui",
+  "version": "0.2.7",
+  "description": "A2A (Agent-to-Agent) v1.0 protocol adapter for assistant-ui",
   "keywords": [
     "a2a",
     "agent-to-agent",
@@ -30,11 +30,10 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-stream": "^0.3.4",
-    "uuid": "^13.0.0"
+    "@assistant-ui/core": "^0.1.7",
+    "@assistant-ui/store": "^0.2.3"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.12.12",
     "@types/react": "*",
     "react": "^18 || ^19"
   },
@@ -45,10 +44,9 @@
   },
   "devDependencies": {
     "@types/react": "^19.2.14",
-    "@types/uuid": "^11.0.0",
     "react": "^19.2.4",
-    "@assistant-ui/react": "0.12.12",
-    "@assistant-ui/x-buildutils": "0.0.1"
+    "vitest": "^4.1.0",
+    "@assistant-ui/x-buildutils": "0.0.3"
   },
   "publishConfig": {
     "access": "public",
@@ -64,6 +62,8 @@
     "url": "https://github.com/assistant-ui/assistant-ui/issues"
   },
   "scripts": {
-    "build": "aui-build"
+    "build": "aui-build",
+    "test": "vitest run",
+    "test:watch": "vitest"
   }
 }