@assistant-ui/react-a2a 0.1.4 → 0.1.6

package/dist/useA2ARuntime.js

@@ -1,215 +1,188 @@
-// src/useA2ARuntime.ts
 import { useEffect, useRef, useState } from "react";
-import {
-  useExternalMessageConverter,
-  useExternalStoreRuntime,
-  useThread,
-  useThreadListItemRuntime
-} from "@assistant-ui/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";
-var getPendingToolCalls = (messages) => {
-  const pendingToolCalls = /* @__PURE__ */ new Map();
-  for (const message of messages) {
-    if (message.role === "assistant") {
-      for (const toolCall of message.tool_calls ?? []) {
-        pendingToolCalls.set(toolCall.id, toolCall);
-      }
+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);
+        }
     }
-    if (message.role === "tool") {
-      pendingToolCalls.delete(message.tool_call_id);
-    }
-  }
-  return [...pendingToolCalls.values()];
+    return [...pendingToolCalls.values()];
 };
-var 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}`
-        );
+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 ?? "";
     }
-  });
-  if (content.length === 1 && content[0]?.type === "text") {
-    return content[0].text ?? "";
-  }
-  return content;
-};
-var symbolA2ARuntimeExtras = /* @__PURE__ */ Symbol("a2a-runtime-extras");
-var asA2ARuntimeExtras = (extras) => {
-  if (typeof extras !== "object" || extras == null || !(symbolA2ARuntimeExtras in extras))
-    throw new Error(
-      "This method can only be called when you are using useA2ARuntime"
-    );
-  return extras;
+    return content;
 };
-var useA2ATaskState = () => {
-  const { taskState } = useThread((t) => asA2ARuntimeExtras(t.extras));
-  return taskState;
+const symbolA2ARuntimeExtras = Symbol("a2a-runtime-extras");
+const asA2ARuntimeExtras = (extras) => {
+    if (typeof extras !== "object" ||
+        extras == null ||
+        !(symbolA2ARuntimeExtras in extras))
+        throw new Error("This method can only be called when you are using useA2ARuntime");
+    return extras;
 };
-var useA2AArtifacts = () => {
-  const { artifacts } = useThread((t) => asA2ARuntimeExtras(t.extras));
-  return artifacts;
+export const useA2ATaskState = () => {
+    const { taskState } = useThread((t) => asA2ARuntimeExtras(t.extras));
+    return taskState;
 };
-var useA2ASend = () => {
-  const { send } = useThread((t) => asA2ARuntimeExtras(t.extras));
-  return send;
+export const useA2AArtifacts = () => {
+    const { artifacts } = useThread((t) => asA2ARuntimeExtras(t.extras));
+    return artifacts;
 };
-var 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 (messages2, config) => {
-    try {
-      setIsRunning(true);
-      await sendMessage(messages2, config);
-    } catch (error) {
-      console.error("Error streaming A2A messages:", error);
-    } finally {
-      setIsRunning(false);
-    }
-  };
-  const threadMessages = useExternalMessageConverter({
-    callback: convertA2AMessage,
-    messages,
-    isRunning
-  });
-  const switchToThread = !onSwitchToThread ? void 0 : async (externalId) => {
-    const { messages: messages2, artifacts: artifacts2 } = await onSwitchToThread(externalId);
-    setMessages(messages2);
-    if (artifacts2) {
-      setArtifacts(artifacts2);
-    }
-  };
-  const threadList = {
-    threadId: contextId,
-    onSwitchToNewThread: !onSwitchToNewThread ? void 0 : async () => {
-      await onSwitchToNewThread();
-      setMessages([]);
-      setArtifacts([]);
-    },
-    onSwitchToThread: switchToThread
-  };
-  const loadingRef = useRef(false);
-  const threadListItemRuntime = useThreadListItemRuntime({ optional: true });
-  useEffect(() => {
-    if (!threadListItemRuntime || !switchToThread || loadingRef.current) return;
-    const externalId = threadListItemRuntime.getState().externalId;
-    if (externalId) {
-      loadingRef.current = true;
-      switchToThread(externalId).finally(() => {
-        loadingRef.current = false;
-      });
-    }
-  }, []);
-  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 !== void 0) config.contextId = contextId;
-      if (msg.runConfig !== void 0) config.runConfig = msg.runConfig;
-      return handleSendMessage(
-        [
-          ...cancellations,
-          {
-            role: "user",
-            content: getMessageContent(msg)
-          }
-        ],
-        config
-      );
-    },
-    onAddToolResult: async ({
-      toolCallId,
-      toolName: _toolName,
-      result,
-      isError,
-      artifact
-    }) => {
-      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 !== void 0) config.contextId = contextId;
-      await handleSendMessage(
-        [message],
-        // TODO reuse runconfig here!
-        config
-      );
-    },
-    onCancel: unstable_allowCancellation ? async () => {
-      cancel();
-    } : void 0
-  });
+export const useA2ASend = () => {
+    const { send } = useThread((t) => asA2ARuntimeExtras(t.extras));
+    return send;
 };
-export {
-  useA2AArtifacts,
-  useA2ARuntime,
-  useA2ASend,
-  useA2ATaskState
+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);
+        }
+        catch (error) {
+            console.error("Error streaming A2A messages:", error);
+        }
+        finally {
+            setIsRunning(false);
+        }
+    };
+    const threadMessages = useExternalMessageConverter({
+        callback: convertA2AMessage,
+        messages,
+        isRunning,
+    });
+    const switchToThread = !onSwitchToThread
+        ? undefined
+        : async (externalId) => {
+            const { messages, artifacts } = await onSwitchToThread(externalId);
+            setMessages(messages);
+            if (artifacts) {
+                setArtifacts(artifacts);
+            }
+        };
+    const threadList = {
+        threadId: contextId,
+        onSwitchToNewThread: !onSwitchToNewThread
+            ? undefined
+            : async () => {
+                await onSwitchToNewThread();
+                setMessages([]);
+                setArtifacts([]);
+            },
+        onSwitchToThread: switchToThread,
+    };
+    const loadingRef = useRef(false);
+    const threadListItemRuntime = useThreadListItemRuntime({ optional: true });
+    useEffect(() => {
+        if (!threadListItemRuntime || !switchToThread || loadingRef.current)
+            return;
+        const externalId = threadListItemRuntime.getState().externalId;
+        if (externalId) {
+            loadingRef.current = true;
+            switchToThread(externalId).finally(() => {
+                loadingRef.current = false;
+            });
+        }
+    }, []);
+    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,
+    });
 };
 //# sourceMappingURL=useA2ARuntime.js.map

package/dist/useA2ARuntime.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/useA2ARuntime.ts"],"sourcesContent":["import { useEffect, useRef, useState } from \"react\";\nimport {\n  A2AMessage,\n  A2AToolCall,\n  A2AArtifact,\n  A2ATaskState,\n  A2ASendMessageConfig,\n  A2AStreamCallback,\n  OnTaskUpdateEventCallback,\n  OnArtifactsEventCallback,\n  OnErrorEventCallback,\n  OnStateUpdateEventCallback,\n  OnCustomEventCallback,\n} from \"./types\";\nimport {\n  useExternalMessageConverter,\n  useExternalStoreRuntime,\n  useThread,\n  useThreadListItemRuntime,\n} from \"@assistant-ui/react\";\nimport { convertA2AMessage } from \"./convertA2AMessages\";\nimport { useA2AMessages } from \"./useA2AMessages\";\nimport { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { AppendMessage } from \"@assistant-ui/react\";\nimport { ExternalStoreAdapter } from \"@assistant-ui/react\";\nimport { FeedbackAdapter } from \"@assistant-ui/react\";\nimport { SpeechSynthesisAdapter } from \"@assistant-ui/react\";\nimport { appendA2AChunk } from \"./appendA2AChunk\";\n\nconst getPendingToolCalls = (messages: A2AMessage[]) => {\n  const pendingToolCalls = new Map<string, A2AToolCall>();\n  for (const message of messages) {\n    if (message.role === \"assistant\") {\n      for (const toolCall of message.tool_calls ?? []) {\n        pendingToolCalls.set(toolCall.id, toolCall);\n      }\n    }\n    if (message.role === \"tool\") {\n      pendingToolCalls.delete(message.tool_call_id!);\n    }\n  }\n\n  return [...pendingToolCalls.values()];\n};\n\nconst 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, image_url: { url: part.image } };\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          | \"file\"\n          | \"audio\"\n          | \"data\" = type;\n        throw new Error(\n          `Unsupported append message part type: ${_exhaustiveCheck}`,\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\nconst symbolA2ARuntimeExtras = Symbol(\"a2a-runtime-extras\");\ntype A2ARuntimeExtras = {\n  [symbolA2ARuntimeExtras]: true;\n  send: (messages: A2AMessage[], config: A2ASendMessageConfig) => Promise<void>;\n  taskState: A2ATaskState | undefined;\n  artifacts: A2AArtifact[];\n};\n\nconst asA2ARuntimeExtras = (extras: unknown): A2ARuntimeExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolA2ARuntimeExtras in extras)\n  )\n    throw new Error(\n      \"This method can only be called when you are using useA2ARuntime\",\n    );\n\n  return extras as A2ARuntimeExtras;\n};\n\nexport const useA2ATaskState = () => {\n  const { taskState } = useThread((t) => asA2ARuntimeExtras(t.extras));\n  return taskState;\n};\n\nexport const useA2AArtifacts = () => {\n  const { artifacts } = useThread((t) => asA2ARuntimeExtras(t.extras));\n  return artifacts;\n};\n\nexport const useA2ASend = () => {\n  const { send } = useThread((t) => asA2ARuntimeExtras(t.extras));\n  return send;\n};\n\nexport const useA2ARuntime = ({\n  autoCancelPendingToolCalls,\n  adapters: { attachments, feedback, speech } = {},\n  unstable_allowCancellation,\n  stream,\n  contextId,\n  onSwitchToNewThread,\n  onSwitchToThread,\n  eventHandlers,\n}: {\n  /**\n   * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.\n   */\n  contextId?: string | undefined;\n  autoCancelPendingToolCalls?: boolean | undefined;\n  unstable_allowCancellation?: boolean | undefined;\n  stream: A2AStreamCallback<A2AMessage>;\n  /**\n   * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.\n   */\n  onSwitchToNewThread?: () => Promise<void> | void;\n  onSwitchToThread?: (contextId: string) => Promise<{\n    messages: A2AMessage[];\n    artifacts?: A2AArtifact[];\n  }>;\n  adapters?:\n    | {\n        attachments?: AttachmentAdapter;\n        speech?: SpeechSynthesisAdapter;\n        feedback?: FeedbackAdapter;\n      }\n    | undefined;\n  /**\n   * Event handlers for various A2A stream events\n   */\n  eventHandlers?:\n    | {\n        /**\n         * Called when task updates are received from the A2A stream\n         */\n        onTaskUpdate?: OnTaskUpdateEventCallback;\n        /**\n         * Called when artifacts are received from the A2A stream\n         */\n        onArtifacts?: OnArtifactsEventCallback;\n        /**\n         * Called when errors occur during A2A stream processing\n         */\n        onError?: OnErrorEventCallback;\n        /**\n         * Called when state updates are received from the A2A stream\n         */\n        onStateUpdate?: OnStateUpdateEventCallback;\n        /**\n         * Called when custom events are received from the A2A stream\n         */\n        onCustomEvent?: OnCustomEventCallback;\n      }\n    | undefined;\n}) => {\n  const {\n    taskState,\n    artifacts,\n    setArtifacts,\n    messages,\n    sendMessage,\n    cancel,\n    setMessages,\n  } = useA2AMessages({\n    appendMessage: appendA2AChunk,\n    stream,\n    ...(eventHandlers && { eventHandlers }),\n  });\n\n  const [isRunning, setIsRunning] = useState(false);\n  const handleSendMessage = async (\n    messages: A2AMessage[],\n    config: A2ASendMessageConfig,\n  ) => {\n    try {\n      setIsRunning(true);\n      await sendMessage(messages, config);\n    } catch (error) {\n      console.error(\"Error streaming A2A messages:\", error);\n    } finally {\n      setIsRunning(false);\n    }\n  };\n\n  const threadMessages = useExternalMessageConverter({\n    callback: convertA2AMessage,\n    messages,\n    isRunning,\n  });\n\n  const switchToThread = !onSwitchToThread\n    ? undefined\n    : async (externalId: string) => {\n        const { messages, artifacts } = await onSwitchToThread(externalId);\n        setMessages(messages);\n        if (artifacts) {\n          setArtifacts(artifacts);\n        }\n      };\n\n  const threadList: NonNullable<\n    ExternalStoreAdapter[\"adapters\"]\n  >[\"threadList\"] = {\n    threadId: contextId,\n    onSwitchToNewThread: !onSwitchToNewThread\n      ? undefined\n      : async () => {\n          await onSwitchToNewThread();\n          setMessages([]);\n          setArtifacts([]);\n        },\n    onSwitchToThread: switchToThread,\n  };\n\n  const loadingRef = useRef(false);\n  const threadListItemRuntime = useThreadListItemRuntime({ optional: true });\n  useEffect(() => {\n    if (!threadListItemRuntime || !switchToThread || loadingRef.current) return;\n\n    const externalId = threadListItemRuntime.getState().externalId;\n    if (externalId) {\n      loadingRef.current = true;\n      switchToThread(externalId).finally(() => {\n        loadingRef.current = false;\n      });\n    }\n  }, []);\n\n  return useExternalStoreRuntime({\n    isRunning,\n    messages: threadMessages,\n    adapters: {\n      attachments,\n      feedback,\n      speech,\n      threadList,\n    },\n    extras: {\n      [symbolA2ARuntimeExtras]: true,\n      taskState,\n      artifacts,\n      send: handleSendMessage,\n    } satisfies A2ARuntimeExtras,\n    onNew: (msg) => {\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(messages).map(\n              (t) =>\n                ({\n                  role: \"tool\",\n                  tool_call_id: t.id,\n                  content: JSON.stringify({ cancelled: true }),\n                  status: {\n                    type: \"incomplete\",\n                    reason: \"cancelled\",\n                  },\n                }) satisfies A2AMessage & { role: \"tool\" },\n            )\n          : [];\n\n      const config: A2ASendMessageConfig = {};\n      if (contextId !== undefined) config.contextId = contextId;\n      if (msg.runConfig !== undefined) config.runConfig = msg.runConfig;\n      return handleSendMessage(\n        [\n          ...cancellations,\n          {\n            role: \"user\",\n            content: getMessageContent(msg),\n          },\n        ],\n        config,\n      );\n    },\n    onAddToolResult: async ({\n      toolCallId,\n      toolName: _toolName,\n      result,\n      isError,\n      artifact,\n    }) => {\n      // TODO parallel human in the loop calls\n      const message: A2AMessage = {\n        role: \"tool\",\n        tool_call_id: toolCallId,\n        content: JSON.stringify(result),\n        status: isError\n          ? { type: \"incomplete\", reason: \"error\" }\n          : { type: \"complete\", reason: \"stop\" },\n      };\n      if (artifact) {\n        message.artifacts = [artifact] as A2AArtifact[];\n      }\n      const config: A2ASendMessageConfig = {};\n      if (contextId !== undefined) config.contextId = contextId;\n      await handleSendMessage(\n        [message],\n        // TODO reuse runconfig here!\n        config,\n      );\n    },\n    onCancel: unstable_allowCancellation\n      ? async () => {\n          cancel();\n        }\n      : undefined,\n  });\n};\n"],"mappings":";AAAA,SAAS,WAAW,QAAQ,gBAAgB;AAc5C;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,yBAAyB;AAClC,SAAS,sBAAsB;AAM/B,SAAS,sBAAsB;AAE/B,IAAM,sBAAsB,CAAC,aAA2B;AACtD,QAAM,mBAAmB,oBAAI,IAAyB;AACtD,aAAW,WAAW,UAAU;AAC9B,QAAI,QAAQ,SAAS,aAAa;AAChC,iBAAW,YAAY,QAAQ,cAAc,CAAC,GAAG;AAC/C,yBAAiB,IAAI,SAAS,IAAI,QAAQ;AAAA,MAC5C;AAAA,IACF;AACA,QAAI,QAAQ,SAAS,QAAQ;AAC3B,uBAAiB,OAAO,QAAQ,YAAa;AAAA,IAC/C;AAAA,EACF;AAEA,SAAO,CAAC,GAAG,iBAAiB,OAAO,CAAC;AACtC;AAEA,IAAM,oBAAoB,CAAC,QAAuB;AAChD,QAAM,aAAa;AAAA,IACjB,GAAG,IAAI;AAAA,IACP,GAAI,IAAI,aAAa,QAAQ,CAAC,MAAM,EAAE,OAAO,KAAK,CAAC;AAAA,EACrD;AACA,QAAM,UAAU,WAAW,IAAI,CAAC,SAAS;AACvC,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAiB,MAAM,KAAK,KAAK;AAAA,MAClD,KAAK;AACH,eAAO,EAAE,MAAM,aAAsB,WAAW,EAAE,KAAK,KAAK,MAAM,EAAE;AAAA,MAEtE,KAAK;AACH,cAAM,IAAI,MAAM,sCAAsC;AAAA,MAExD;AACE,cAAM,mBAKO;AACb,cAAM,IAAI;AAAA,UACR,yCAAyC,gBAAgB;AAAA,QAC3D;AAAA,IACJ;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,WAAW,KAAK,QAAQ,CAAC,GAAG,SAAS,QAAQ;AACvD,WAAO,QAAQ,CAAC,EAAE,QAAQ;AAAA,EAC5B;AAEA,SAAO;AACT;AAEA,IAAM,yBAAyB,uBAAO,oBAAoB;AAQ1D,IAAM,qBAAqB,CAAC,WAAsC;AAChE,MACE,OAAO,WAAW,YAClB,UAAU,QACV,EAAE,0BAA0B;AAE5B,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAEF,SAAO;AACT;AAEO,IAAM,kBAAkB,MAAM;AACnC,QAAM,EAAE,UAAU,IAAI,UAAU,CAAC,MAAM,mBAAmB,EAAE,MAAM,CAAC;AACnE,SAAO;AACT;AAEO,IAAM,kBAAkB,MAAM;AACnC,QAAM,EAAE,UAAU,IAAI,UAAU,CAAC,MAAM,mBAAmB,EAAE,MAAM,CAAC;AACnE,SAAO;AACT;AAEO,IAAM,aAAa,MAAM;AAC9B,QAAM,EAAE,KAAK,IAAI,UAAU,CAAC,MAAM,mBAAmB,EAAE,MAAM,CAAC;AAC9D,SAAO;AACT;AAEO,IAAM,gBAAgB,CAAC;AAAA,EAC5B;AAAA,EACA,UAAU,EAAE,aAAa,UAAU,OAAO,IAAI,CAAC;AAAA,EAC/C;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAkDM;AACJ,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,IAAI,eAAe;AAAA,IACjB,eAAe;AAAA,IACf;AAAA,IACA,GAAI,iBAAiB,EAAE,cAAc;AAAA,EACvC,CAAC;AAED,QAAM,CAAC,WAAW,YAAY,IAAI,SAAS,KAAK;AAChD,QAAM,oBAAoB,OACxBA,WACA,WACG;AACH,QAAI;AACF,mBAAa,IAAI;AACjB,YAAM,YAAYA,WAAU,MAAM;AAAA,IACpC,SAAS,OAAO;AACd,cAAQ,MAAM,iCAAiC,KAAK;AAAA,IACtD,UAAE;AACA,mBAAa,KAAK;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,iBAAiB,4BAA4B;AAAA,IACjD,UAAU;AAAA,IACV;AAAA,IACA;AAAA,EACF,CAAC;AAED,QAAM,iBAAiB,CAAC,mBACpB,SACA,OAAO,eAAuB;AAC5B,UAAM,EAAE,UAAAA,WAAU,WAAAC,WAAU,IAAI,MAAM,iBAAiB,UAAU;AACjE,gBAAYD,SAAQ;AACpB,QAAIC,YAAW;AACb,mBAAaA,UAAS;AAAA,IACxB;AAAA,EACF;AAEJ,QAAM,aAEY;AAAA,IAChB,UAAU;AAAA,IACV,qBAAqB,CAAC,sBAClB,SACA,YAAY;AACV,YAAM,oBAAoB;AAC1B,kBAAY,CAAC,CAAC;AACd,mBAAa,CAAC,CAAC;AAAA,IACjB;AAAA,IACJ,kBAAkB;AAAA,EACpB;AAEA,QAAM,aAAa,OAAO,KAAK;AAC/B,QAAM,wBAAwB,yBAAyB,EAAE,UAAU,KAAK,CAAC;AACzE,YAAU,MAAM;AACd,QAAI,CAAC,yBAAyB,CAAC,kBAAkB,WAAW,QAAS;AAErE,UAAM,aAAa,sBAAsB,SAAS,EAAE;AACpD,QAAI,YAAY;AACd,iBAAW,UAAU;AACrB,qBAAe,UAAU,EAAE,QAAQ,MAAM;AACvC,mBAAW,UAAU;AAAA,MACvB,CAAC;AAAA,IACH;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,SAAO,wBAAwB;AAAA,IAC7B;AAAA,IACA,UAAU;AAAA,IACV,UAAU;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,IACA,QAAQ;AAAA,MACN,CAAC,sBAAsB,GAAG;AAAA,MAC1B;AAAA,MACA;AAAA,MACA,MAAM;AAAA,IACR;AAAA,IACA,OAAO,CAAC,QAAQ;AACd,YAAM,gBACJ,+BAA+B,QAC3B,oBAAoB,QAAQ,EAAE;AAAA,QAC5B,CAAC,OACE;AAAA,UACC,MAAM;AAAA,UACN,cAAc,EAAE;AAAA,UAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;AAAA,UAC3C,QAAQ;AAAA,YACN,MAAM;AAAA,YACN,QAAQ;AAAA,UACV;AAAA,QACF;AAAA,MACJ,IACA,CAAC;AAEP,YAAM,SAA+B,CAAC;AACtC,UAAI,cAAc,OAAW,QAAO,YAAY;AAChD,UAAI,IAAI,cAAc,OAAW,QAAO,YAAY,IAAI;AACxD,aAAO;AAAA,QACL;AAAA,UACE,GAAG;AAAA,UACH;AAAA,YACE,MAAM;AAAA,YACN,SAAS,kBAAkB,GAAG;AAAA,UAChC;AAAA,QACF;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,IACA,iBAAiB,OAAO;AAAA,MACtB;AAAA,MACA,UAAU;AAAA,MACV;AAAA,MACA;AAAA,MACA;AAAA,IACF,MAAM;AAEJ,YAAM,UAAsB;AAAA,QAC1B,MAAM;AAAA,QACN,cAAc;AAAA,QACd,SAAS,KAAK,UAAU,MAAM;AAAA,QAC9B,QAAQ,UACJ,EAAE,MAAM,cAAc,QAAQ,QAAQ,IACtC,EAAE,MAAM,YAAY,QAAQ,OAAO;AAAA,MACzC;AACA,UAAI,UAAU;AACZ,gBAAQ,YAAY,CAAC,QAAQ;AAAA,MAC/B;AACA,YAAM,SAA+B,CAAC;AACtC,UAAI,cAAc,OAAW,QAAO,YAAY;AAChD,YAAM;AAAA,QACJ,CAAC,OAAO;AAAA;AAAA,QAER;AAAA,MACF;AAAA,IACF;AAAA,IACA,UAAU,6BACN,YAAY;AACV,aAAO;AAAA,IACT,IACA;AAAA,EACN,CAAC;AACH;","names":["messages","artifacts"]}
+{"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,EAAE,CAAC,CAAC;IAEP,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"}

package/package.json

@@ -1,10 +1,22 @@
 {
   "name": "@assistant-ui/react-a2a",
-  "version": "0.1.4",
+  "version": "0.1.6",
+  "description": "A2A protocol adapter for assistant-ui",
+  "keywords": [
+    "a2a",
+    "agent-to-agent",
+    "assistant-ui",
+    "react",
+    "ai",
+    "chat",
+    "protocol"
+  ],
+  "author": "AgentbaseAI Inc.",
   "license": "MIT",
   "type": "module",
   "exports": {
     ".": {
+      "aui-source": "./src/index.ts",
       "types": "./dist/index.d.ts",
       "default": "./dist/index.js"
     }
@@ -18,14 +30,13 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-stream": "^0.2.45",
-    "uuid": "^13.0.0",
-    "zod": "^4.1.13"
+    "assistant-stream": "^0.2.47",
+    "uuid": "^13.0.0"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.11.50",
+    "@assistant-ui/react": "^0.11.58",
     "@types/react": "*",
-    "react": "^18 || ^19 || ^19.0.0-rc"
+    "react": "^18 || ^19"
   },
   "peerDependenciesMeta": {
     "@types/react": {
@@ -33,16 +44,10 @@
     }
   },
   "devDependencies": {
-    "@testing-library/dom": "^10.4.1",
-    "@testing-library/react": "^16.3.0",
-    "@types/node": "^25.0.0",
-    "@types/react": "^19.2.7",
+    "@types/react": "^19.2.9",
     "@types/uuid": "^11.0.0",
-    "jsdom": "^27.3.0",
-    "react": "19.2.3",
-    "tsx": "^4.21.0",
-    "vitest": "^4.0.15",
-    "@assistant-ui/react": "0.11.50",
+    "react": "^19.2.3",
+    "@assistant-ui/react": "0.11.58",
     "@assistant-ui/x-buildutils": "0.0.1"
   },
   "publishConfig": {
@@ -52,14 +57,13 @@
   "homepage": "https://www.assistant-ui.com/",
   "repository": {
     "type": "git",
-    "url": "https://github.com/assistant-ui/assistant-ui/tree/main/packages/react-a2a"
+    "url": "git+https://github.com/assistant-ui/assistant-ui.git",
+    "directory": "packages/react-a2a"
   },
   "bugs": {
     "url": "https://github.com/assistant-ui/assistant-ui/issues"
   },
   "scripts": {
-    "build": "tsx scripts/build.mts",
-    "test": "vitest run --passWithNoTests",
-    "test:watch": "vitest"
+    "build": "aui-build"
   }
 }