@assistant-ui/react-a2a 0.1.0

package/dist/useA2AMessages.js

@@ -0,0 +1,134 @@
+// src/useA2AMessages.ts
+import { useState, useCallback, useRef, useMemo } from "react";
+import { v4 as uuidv4 } from "uuid";
+import { A2AMessageAccumulator } from "./A2AMessageAccumulator.js";
+import {
+  A2AKnownEventTypes
+} from "./types.js";
+var DEFAULT_APPEND_MESSAGE = (_, curr) => curr;
+var useA2AMessages = ({
+  stream,
+  appendMessage = DEFAULT_APPEND_MESSAGE,
+  eventHandlers
+}) => {
+  const [messages, setMessages] = useState([]);
+  const [taskState, setTaskState] = useState();
+  const [artifacts, setArtifacts] = useState([]);
+  const abortControllerRef = useRef(null);
+  const { onTaskUpdate, onArtifacts, onError, onStateUpdate, onCustomEvent } = useMemo(() => eventHandlers ?? {}, [eventHandlers]);
+  const sendMessage = useCallback(
+    async (newMessages, config) => {
+      const newMessagesWithId = newMessages.map(
+        (m) => m.id ? m : { ...m, id: uuidv4() }
+      );
+      const accumulator = new A2AMessageAccumulator({
+        initialMessages: messages,
+        appendMessage
+      });
+      setMessages(accumulator.addMessages(newMessagesWithId));
+      const abortController = new AbortController();
+      abortControllerRef.current = abortController;
+      const response = await stream(newMessagesWithId, {
+        ...config,
+        abortSignal: abortController.signal
+      });
+      for await (const event of response) {
+        switch (event.event) {
+          case A2AKnownEventTypes.TaskUpdate:
+            const taskData = event.data;
+            setTaskState(taskData);
+            onTaskUpdate?.(taskData);
+            break;
+          case A2AKnownEventTypes.TaskComplete:
+            const { messages: taskMessages, artifacts: taskArtifacts } = event.data;
+            if (taskMessages) {
+              setMessages(accumulator.addMessages(taskMessages));
+            }
+            if (taskArtifacts) {
+              setArtifacts(taskArtifacts);
+              onArtifacts?.(taskArtifacts);
+            }
+            setTaskState(void 0);
+            break;
+          case A2AKnownEventTypes.TaskFailed:
+            onError?.(event.data);
+            if (taskState) {
+              setTaskState({
+                ...taskState,
+                state: "failed",
+                message: event.data?.message
+              });
+            }
+            break;
+          case A2AKnownEventTypes.Artifacts:
+            const artifactData = event.data;
+            setArtifacts(artifactData);
+            onArtifacts?.(artifactData);
+            break;
+          case A2AKnownEventTypes.StateUpdate:
+            onStateUpdate?.(event.data);
+            break;
+          case A2AKnownEventTypes.Error:
+            onError?.(event.data);
+            const messages2 = accumulator.getMessages();
+            const lastAssistantMessage = messages2.findLast(
+              (m) => m != null && "role" in m && m.role === "assistant" && m.id != null
+            );
+            if (lastAssistantMessage) {
+              const errorMessage = {
+                ...lastAssistantMessage,
+                status: {
+                  type: "incomplete",
+                  reason: "error",
+                  error: event.data
+                }
+              };
+              setMessages(accumulator.addMessages([errorMessage]));
+            }
+            break;
+          default:
+            if (onCustomEvent) {
+              onCustomEvent(event.event, event.data);
+            } else {
+              console.warn(
+                "Unhandled A2A event received:",
+                event.event,
+                event.data
+              );
+            }
+            break;
+        }
+      }
+    },
+    [
+      messages,
+      appendMessage,
+      stream,
+      onTaskUpdate,
+      onArtifacts,
+      onError,
+      onStateUpdate,
+      onCustomEvent,
+      taskState
+    ]
+  );
+  const cancel = useCallback(() => {
+    if (abortControllerRef.current) {
+      abortControllerRef.current.abort();
+    }
+  }, [abortControllerRef]);
+  return {
+    messages,
+    artifacts,
+    taskState,
+    sendMessage,
+    cancel,
+    setMessages,
+    setArtifacts,
+    setTaskState
+  };
+};
+export {
+  useA2AMessages
+};
+//# sourceMappingURL=useA2AMessages.js.map

package/dist/useA2AMessages.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/useA2AMessages.ts"],"sourcesContent":["import { useState, useCallback, useRef, useMemo } from \"react\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport { A2AMessageAccumulator } from \"./A2AMessageAccumulator\";\nimport {\n  A2AEvent,\n  A2AKnownEventTypes,\n  A2ATaskState,\n  A2AArtifact,\n  A2ASendMessageConfig,\n  A2AStreamCallback,\n  OnTaskUpdateEventCallback,\n  OnArtifactsEventCallback,\n  OnErrorEventCallback,\n  OnStateUpdateEventCallback,\n  OnCustomEventCallback,\n} from \"./types\";\n\nexport type A2AMessagesEvent<_TMessage> = A2AEvent;\n\nconst DEFAULT_APPEND_MESSAGE = <TMessage>(\n  _: TMessage | undefined,\n  curr: TMessage,\n) => curr;\n\nexport const useA2AMessages = <TMessage extends { id?: string }>({\n  stream,\n  appendMessage = DEFAULT_APPEND_MESSAGE,\n  eventHandlers,\n}: {\n  stream: A2AStreamCallback<TMessage>;\n  appendMessage?: (prev: TMessage | undefined, curr: TMessage) => TMessage;\n  eventHandlers?: {\n    onTaskUpdate?: OnTaskUpdateEventCallback;\n    onArtifacts?: OnArtifactsEventCallback;\n    onError?: OnErrorEventCallback;\n    onStateUpdate?: OnStateUpdateEventCallback;\n    onCustomEvent?: OnCustomEventCallback;\n  };\n}) => {\n  const [messages, setMessages] = useState<TMessage[]>([]);\n  const [taskState, setTaskState] = useState<A2ATaskState | undefined>();\n  const [artifacts, setArtifacts] = useState<A2AArtifact[]>([]);\n  const abortControllerRef = useRef<AbortController | null>(null);\n\n  const { onTaskUpdate, onArtifacts, onError, onStateUpdate, onCustomEvent } =\n    useMemo(() => eventHandlers ?? {}, [eventHandlers]);\n\n  const sendMessage = useCallback(\n    async (newMessages: TMessage[], config: A2ASendMessageConfig) => {\n      // ensure all messages have an ID\n      const newMessagesWithId = newMessages.map((m) =>\n        m.id ? m : { ...m, id: uuidv4() },\n      );\n\n      const accumulator = new A2AMessageAccumulator({\n        initialMessages: messages,\n        appendMessage,\n      });\n      setMessages(accumulator.addMessages(newMessagesWithId));\n\n      const abortController = new AbortController();\n      abortControllerRef.current = abortController;\n      const response = await stream(newMessagesWithId, {\n        ...config,\n        abortSignal: abortController.signal,\n      });\n\n      for await (const event of response) {\n        switch (event.event) {\n          case A2AKnownEventTypes.TaskUpdate:\n            const taskData = event.data as A2ATaskState;\n            setTaskState(taskData);\n            onTaskUpdate?.(taskData);\n            break;\n\n          case A2AKnownEventTypes.TaskComplete:\n            // Extract messages and artifacts from completed task\n            const { messages: taskMessages, artifacts: taskArtifacts } =\n              event.data;\n            if (taskMessages) {\n              setMessages(accumulator.addMessages(taskMessages));\n            }\n            if (taskArtifacts) {\n              setArtifacts(taskArtifacts);\n              onArtifacts?.(taskArtifacts);\n            }\n            // Clear task state on completion\n            setTaskState(undefined);\n            break;\n\n          case A2AKnownEventTypes.TaskFailed:\n            onError?.(event.data);\n            // Update task state to failed\n            if (taskState) {\n              setTaskState({\n                ...taskState,\n                state: \"failed\",\n                message: event.data?.message,\n              });\n            }\n            break;\n\n          case A2AKnownEventTypes.Artifacts:\n            const artifactData = event.data as A2AArtifact[];\n            setArtifacts(artifactData);\n            onArtifacts?.(artifactData);\n            break;\n\n          case A2AKnownEventTypes.StateUpdate:\n            onStateUpdate?.(event.data);\n            break;\n\n          case A2AKnownEventTypes.Error:\n            onError?.(event.data);\n            // Update the last assistant message with error status if available\n            const messages = accumulator.getMessages();\n            const lastAssistantMessage = messages.findLast(\n              (m): m is TMessage & { role: string; id: string } =>\n                m != null &&\n                \"role\" in m &&\n                m.role === \"assistant\" &&\n                m.id != null,\n            );\n            if (lastAssistantMessage) {\n              const errorMessage = {\n                ...lastAssistantMessage,\n                status: {\n                  type: \"incomplete\" as const,\n                  reason: \"error\" as const,\n                  error: event.data,\n                },\n              };\n              setMessages(accumulator.addMessages([errorMessage]));\n            }\n            break;\n\n          default:\n            if (onCustomEvent) {\n              onCustomEvent(event.event, event.data);\n            } else {\n              console.warn(\n                \"Unhandled A2A event received:\",\n                event.event,\n                event.data,\n              );\n            }\n            break;\n        }\n      }\n    },\n    [\n      messages,\n      appendMessage,\n      stream,\n      onTaskUpdate,\n      onArtifacts,\n      onError,\n      onStateUpdate,\n      onCustomEvent,\n      taskState,\n    ],\n  );\n\n  const cancel = useCallback(() => {\n    if (abortControllerRef.current) {\n      abortControllerRef.current.abort();\n    }\n  }, [abortControllerRef]);\n\n  return {\n    messages,\n    artifacts,\n    taskState,\n    sendMessage,\n    cancel,\n    setMessages,\n    setArtifacts,\n    setTaskState,\n  };\n};\n"],"mappings":";AAAA,SAAS,UAAU,aAAa,QAAQ,eAAe;AACvD,SAAS,MAAM,cAAc;AAC7B,SAAS,6BAA6B;AACtC;AAAA,EAEE;AAAA,OAUK;AAIP,IAAM,yBAAyB,CAC7B,GACA,SACG;AAEE,IAAM,iBAAiB,CAAmC;AAAA,EAC/D;AAAA,EACA,gBAAgB;AAAA,EAChB;AACF,MAUM;AACJ,QAAM,CAAC,UAAU,WAAW,IAAI,SAAqB,CAAC,CAAC;AACvD,QAAM,CAAC,WAAW,YAAY,IAAI,SAAmC;AACrE,QAAM,CAAC,WAAW,YAAY,IAAI,SAAwB,CAAC,CAAC;AAC5D,QAAM,qBAAqB,OAA+B,IAAI;AAE9D,QAAM,EAAE,cAAc,aAAa,SAAS,eAAe,cAAc,IACvE,QAAQ,MAAM,iBAAiB,CAAC,GAAG,CAAC,aAAa,CAAC;AAEpD,QAAM,cAAc;AAAA,IAClB,OAAO,aAAyB,WAAiC;AAE/D,YAAM,oBAAoB,YAAY;AAAA,QAAI,CAAC,MACzC,EAAE,KAAK,IAAI,EAAE,GAAG,GAAG,IAAI,OAAO,EAAE;AAAA,MAClC;AAEA,YAAM,cAAc,IAAI,sBAAsB;AAAA,QAC5C,iBAAiB;AAAA,QACjB;AAAA,MACF,CAAC;AACD,kBAAY,YAAY,YAAY,iBAAiB,CAAC;AAEtD,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,yBAAmB,UAAU;AAC7B,YAAM,WAAW,MAAM,OAAO,mBAAmB;AAAA,QAC/C,GAAG;AAAA,QACH,aAAa,gBAAgB;AAAA,MAC/B,CAAC;AAED,uBAAiB,SAAS,UAAU;AAClC,gBAAQ,MAAM,OAAO;AAAA,UACnB,KAAK,mBAAmB;AACtB,kBAAM,WAAW,MAAM;AACvB,yBAAa,QAAQ;AACrB,2BAAe,QAAQ;AACvB;AAAA,UAEF,KAAK,mBAAmB;AAEtB,kBAAM,EAAE,UAAU,cAAc,WAAW,cAAc,IACvD,MAAM;AACR,gBAAI,cAAc;AAChB,0BAAY,YAAY,YAAY,YAAY,CAAC;AAAA,YACnD;AACA,gBAAI,eAAe;AACjB,2BAAa,aAAa;AAC1B,4BAAc,aAAa;AAAA,YAC7B;AAEA,yBAAa,MAAS;AACtB;AAAA,UAEF,KAAK,mBAAmB;AACtB,sBAAU,MAAM,IAAI;AAEpB,gBAAI,WAAW;AACb,2BAAa;AAAA,gBACX,GAAG;AAAA,gBACH,OAAO;AAAA,gBACP,SAAS,MAAM,MAAM;AAAA,cACvB,CAAC;AAAA,YACH;AACA;AAAA,UAEF,KAAK,mBAAmB;AACtB,kBAAM,eAAe,MAAM;AAC3B,yBAAa,YAAY;AACzB,0BAAc,YAAY;AAC1B;AAAA,UAEF,KAAK,mBAAmB;AACtB,4BAAgB,MAAM,IAAI;AAC1B;AAAA,UAEF,KAAK,mBAAmB;AACtB,sBAAU,MAAM,IAAI;AAEpB,kBAAMA,YAAW,YAAY,YAAY;AACzC,kBAAM,uBAAuBA,UAAS;AAAA,cACpC,CAAC,MACC,KAAK,QACL,UAAU,KACV,EAAE,SAAS,eACX,EAAE,MAAM;AAAA,YACZ;AACA,gBAAI,sBAAsB;AACxB,oBAAM,eAAe;AAAA,gBACnB,GAAG;AAAA,gBACH,QAAQ;AAAA,kBACN,MAAM;AAAA,kBACN,QAAQ;AAAA,kBACR,OAAO,MAAM;AAAA,gBACf;AAAA,cACF;AACA,0BAAY,YAAY,YAAY,CAAC,YAAY,CAAC,CAAC;AAAA,YACrD;AACA;AAAA,UAEF;AACE,gBAAI,eAAe;AACjB,4BAAc,MAAM,OAAO,MAAM,IAAI;AAAA,YACvC,OAAO;AACL,sBAAQ;AAAA,gBACN;AAAA,gBACA,MAAM;AAAA,gBACN,MAAM;AAAA,cACR;AAAA,YACF;AACA;AAAA,QACJ;AAAA,MACF;AAAA,IACF;AAAA,IACA;AAAA,MACE;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,QAAM,SAAS,YAAY,MAAM;AAC/B,QAAI,mBAAmB,SAAS;AAC9B,yBAAmB,QAAQ,MAAM;AAAA,IACnC;AAAA,EACF,GAAG,CAAC,kBAAkB,CAAC;AAEvB,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;","names":["messages"]}

package/dist/useA2ARuntime.d.ts

@@ -0,0 +1,55 @@
+import { A2AMessage, A2AArtifact, A2ATaskState, A2ASendMessageConfig, A2AStreamCallback, OnTaskUpdateEventCallback, OnArtifactsEventCallback, OnErrorEventCallback, OnStateUpdateEventCallback, OnCustomEventCallback } from "./types";
+import { AttachmentAdapter } from "@assistant-ui/react";
+import { FeedbackAdapter } from "@assistant-ui/react";
+import { SpeechSynthesisAdapter } from "@assistant-ui/react";
+export declare const useA2ATaskState: () => A2ATaskState | undefined;
+export declare const useA2AArtifacts: () => A2AArtifact[];
+export declare const useA2ASend: () => (messages: A2AMessage[], config: A2ASendMessageConfig) => Promise<void>;
+export declare const useA2ARuntime: ({ autoCancelPendingToolCalls, adapters: { attachments, feedback, speech }, unstable_allowCancellation, stream, contextId, onSwitchToNewThread, onSwitchToThread, eventHandlers, }: {
+    /**
+     * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.
+     */
+    contextId?: string | undefined;
+    autoCancelPendingToolCalls?: boolean | undefined;
+    unstable_allowCancellation?: boolean | undefined;
+    stream: A2AStreamCallback<A2AMessage>;
+    /**
+     * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.
+     */
+    onSwitchToNewThread?: () => Promise<void> | void;
+    onSwitchToThread?: (contextId: string) => Promise<{
+        messages: A2AMessage[];
+        artifacts?: A2AArtifact[];
+    }>;
+    adapters?: {
+        attachments?: AttachmentAdapter;
+        speech?: SpeechSynthesisAdapter;
+        feedback?: FeedbackAdapter;
+    } | undefined;
+    /**
+     * Event handlers for various A2A stream events
+     */
+    eventHandlers?: {
+        /**
+         * Called when task updates are received from the A2A stream
+         */
+        onTaskUpdate?: OnTaskUpdateEventCallback;
+        /**
+         * Called when artifacts are received from the A2A stream
+         */
+        onArtifacts?: OnArtifactsEventCallback;
+        /**
+         * Called when errors occur during A2A stream processing
+         */
+        onError?: OnErrorEventCallback;
+        /**
+         * Called when state updates are received from the A2A stream
+         */
+        onStateUpdate?: OnStateUpdateEventCallback;
+        /**
+         * Called when custom events are received from the A2A stream
+         */
+        onCustomEvent?: OnCustomEventCallback;
+    } | undefined;
+}) => import("@assistant-ui/react").AssistantRuntime;
+//# sourceMappingURL=useA2ARuntime.d.ts.map

package/dist/useA2ARuntime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useA2ARuntime.d.ts","sourceRoot":"","sources":["../src/useA2ARuntime.ts"],"names":[],"mappings":"AACA,OAAO,EACL,UAAU,EAEV,WAAW,EACX,YAAY,EACZ,oBAAoB,EACpB,iBAAiB,EACjB,yBAAyB,EACzB,wBAAwB,EACxB,oBAAoB,EACpB,0BAA0B,EAC1B,qBAAqB,EACtB,MAAM,SAAS,CAAC;AASjB,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAGxD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AA4E7D,eAAO,MAAM,eAAe,gCAG3B,CAAC;AAEF,eAAO,MAAM,eAAe,qBAG3B,CAAC;AAEF,eAAO,MAAM,UAAU,mBA5BJ,UAAU,EAAE,UAAU,oBAAoB,KAAK,OAAO,CAAC,IAAI,CA+B7E,CAAC;AAEF,eAAO,MAAM,aAAa,GAAI,mLAS3B;IACD;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC/B,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD,MAAM,EAAE,iBAAiB,CAAC,UAAU,CAAC,CAAC;IACtC;;OAEG;IACH,mBAAmB,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACjD,gBAAgB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC;QAChD,QAAQ,EAAE,UAAU,EAAE,CAAC;QACvB,SAAS,CAAC,EAAE,WAAW,EAAE,CAAC;KAC3B,CAAC,CAAC;IACH,QAAQ,CAAC,EACL;QACE,WAAW,CAAC,EAAE,iBAAiB,CAAC;QAChC,MAAM,CAAC,EAAE,sBAAsB,CAAC;QAChC,QAAQ,CAAC,EAAE,eAAe,CAAC;KAC5B,GACD,SAAS,CAAC;IACd;;OAEG;IACH,aAAa,CAAC,EACV;QACE;;WAEG;QACH,YAAY,CAAC,EAAE,yBAAyB,CAAC;QACzC;;WAEG;QACH,WAAW,CAAC,EAAE,wBAAwB,CAAC;QACvC;;WAEG;QACH,OAAO,CAAC,EAAE,oBAAoB,CAAC;QAC/B;;WAEG;QACH,aAAa,CAAC,EAAE,0BAA0B,CAAC;QAC3C;;WAEG;QACH,aAAa,CAAC,EAAE,qBAAqB,CAAC;KACvC,GACD,SAAS,CAAC;CACf,mDA0JA,CAAC"}

package/dist/useA2ARuntime.js

@@ -0,0 +1,215 @@
+// src/useA2ARuntime.ts
+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";
+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);
+      }
+    }
+    if (message.role === "tool") {
+      pendingToolCalls.delete(message.tool_call_id);
+    }
+  }
+  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}`
+        );
+    }
+  });
+  if (content.length === 1 && content[0]?.type === "text") {
+    return content[0].text ?? "";
+  }
+  return content;
+};
+var symbolA2ARuntimeExtras = 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;
+};
+var useA2ATaskState = () => {
+  const { taskState } = useThread((t) => asA2ARuntimeExtras(t.extras));
+  return taskState;
+};
+var useA2AArtifacts = () => {
+  const { artifacts } = useThread((t) => asA2ARuntimeExtras(t.extras));
+  return artifacts;
+};
+var useA2ASend = () => {
+  const { send } = useThread((t) => asA2ARuntimeExtras(t.extras));
+  return send;
+};
+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 {
+  useA2AArtifacts,
+  useA2ARuntime,
+  useA2ASend,
+  useA2ATaskState
+};
+//# sourceMappingURL=useA2ARuntime.js.map

package/dist/useA2ARuntime.js.map

@@ -0,0 +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    // eslint-disable-next-line react-hooks/exhaustive-deps\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,OAAO,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,EAEF,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"]}

package/package.json

@@ -0,0 +1,68 @@
+{
+  "name": "@assistant-ui/react-a2a",
+  "version": "0.1.0",
+  "license": "MIT",
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist",
+    "src",
+    "README.md"
+  ],
+  "sideEffects": false,
+  "dependencies": {
+    "assistant-stream": "^0.2.23",
+    "uuid": "^11.1.0",
+    "zod": "^4.0.17"
+  },
+  "peerDependencies": {
+    "@assistant-ui/react": "^0.10.50",
+    "@types/react": "*",
+    "react": "^18 || ^19 || ^19.0.0-rc"
+  },
+  "peerDependenciesMeta": {
+    "@types/react": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@testing-library/dom": "^10.4.1",
+    "@testing-library/react": "^16.3.0",
+    "@types/node": "^24.3.0",
+    "@types/react": "^19.1.10",
+    "@types/uuid": "^10.0.0",
+    "eslint": "^9",
+    "eslint-config-next": "15.4.6",
+    "jsdom": "^26.1.0",
+    "react": "19.1.1",
+    "tsx": "^4.20.4",
+    "vitest": "^3.2.4",
+    "@assistant-ui/react": "0.11.39",
+    "@assistant-ui/x-buildutils": "0.0.1"
+  },
+  "publishConfig": {
+    "access": "public",
+    "provenance": true
+  },
+  "homepage": "https://www.assistant-ui.com/",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/assistant-ui/assistant-ui/tree/main/packages/react-a2a"
+  },
+  "bugs": {
+    "url": "https://github.com/assistant-ui/assistant-ui/issues"
+  },
+  "scripts": {
+    "build": "tsx scripts/build.mts",
+    "lint": "eslint .",
+    "test": "vitest run --passWithNoTests",
+    "test:watch": "vitest"
+  }
+}

package/src/A2AMessageAccumulator.ts

@@ -0,0 +1,48 @@
+import { v4 as uuidv4 } from "uuid";
+
+export type A2AStateAccumulatorConfig<TMessage> = {
+  initialMessages?: TMessage[];
+  appendMessage?: (prev: TMessage | undefined, curr: TMessage) => TMessage;
+};
+
+export class A2AMessageAccumulator<TMessage extends { id?: string }> {
+  private messagesMap = new Map<string, TMessage>();
+  private appendMessage: (
+    prev: TMessage | undefined,
+    curr: TMessage,
+  ) => TMessage;
+
+  constructor({
+    initialMessages = [],
+    appendMessage = ((_: TMessage | undefined, curr: TMessage) => curr) as (
+      prev: TMessage | undefined,
+      curr: TMessage,
+    ) => TMessage,
+  }: A2AStateAccumulatorConfig<TMessage> = {}) {
+    this.appendMessage = appendMessage;
+    this.addMessages(initialMessages);
+  }
+
+  private ensureMessageId(message: TMessage): TMessage {
+    return message.id ? message : { ...message, id: uuidv4() };
+  }
+
+  public addMessages(newMessages: TMessage[]) {
+    if (newMessages.length === 0) return this.getMessages();
+
+    for (const message of newMessages.map(this.ensureMessageId)) {
+      const messageId = message.id!; // ensureMessageId guarantees id exists
+      const previous = this.messagesMap.get(messageId);
+      this.messagesMap.set(messageId, this.appendMessage(previous, message));
+    }
+    return this.getMessages();
+  }
+
+  public getMessages(): TMessage[] {
+    return [...this.messagesMap.values()];
+  }
+
+  public clear() {
+    this.messagesMap.clear();
+  }
+}