@assistant-ui/react-a2a 0.2.35 → 0.2.37

package/dist/useA2ARuntime.js

@@ -72,6 +72,8 @@ function useA2ARuntime(options) {
 			threadId: threadListAdapter.threadId,
 			onSwitchToNewThread: onSwitchToNewThread ? async () => {
 				const generation = ++threadSwitchGenerationRef.current;
+				core.applyExternalMessages([]);
+				core.resetContext();
 				await onSwitchToNewThread();
 				if (generation !== threadSwitchGenerationRef.current) return;
 				core.applyExternalMessages([]);
@@ -79,8 +81,11 @@ function useA2ARuntime(options) {
 			} : void 0,
 			onSwitchToThread: onSwitchToThread ? async (threadId) => {
 				const generation = ++threadSwitchGenerationRef.current;
+				core.applyExternalMessages([]);
+				core.resetContext();
 				const result = await onSwitchToThread(threadId);
 				if (generation !== threadSwitchGenerationRef.current) return;
+				core.applyExternalMessages([]);
 				core.applyExternalMessages(result.messages);
 				core.resetContext();
 			} : void 0
@@ -106,15 +111,18 @@ function useA2ARuntime(options) {
 			isLoading: core.isLoading,
 			messageRepository: core.getMessageRepository(),
 			isRunning: core.isRunning(),
+			unstable_persistsHistory: true,
 			extras: a2aExtras.provide({
 				task: core.getTask(),
 				artifacts: core.getArtifacts(),
 				agentCard: core.getAgentCard()
 			}),
 			onNew: (message) => core.append(message),
+			onVoiceTranscript: (message) => core.appendVoiceTranscript(message),
 			onEdit: (message) => core.edit(message),
 			onReload: (parentId) => core.reload(parentId),
 			onCancel: () => core.cancel(),
+			unstable_onRecordToolInteraction: (options) => core.recordToolInteraction(options),
 			setMessages: (messages) => core.applyExternalMessages(messages),
 			onImport: (messages) => core.applyExternalMessages(messages),
 			adapters: adapterAdapters

package/dist/useA2ARuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useA2ARuntime.js","names":[],"sources":["../src/useA2ARuntime.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  useCallback,\n  useEffect,\n  useInsertionEffect,\n  useMemo,\n  useRef,\n  useState,\n} from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport type {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport { A2AClient, type A2AClientOptions } from \"./A2AClient\";\nimport { A2AThreadRuntimeCore } from \"./A2AThreadRuntimeCore\";\nimport { a2aExtras } from \"./a2aExtras\";\nimport type { UseA2ARuntimeOptions } from \"./types\";\n\ntype ManagedA2AClientOptions = Omit<A2AClientOptions, \"headers\">;\n\nconst serializeManagedClientOptions = (\n  options: ManagedA2AClientOptions,\n): string => {\n  const fetchOptions = Object.fromEntries(\n    Object.entries(options.fetchOptions ?? {}).sort(([a], [b]) =>\n      a.localeCompare(b),\n    ),\n  );\n\n  return JSON.stringify({\n    ...options,\n    fetchOptions,\n  });\n};\n\nexport function useA2ARuntime(options: UseA2ARuntimeOptions): AssistantRuntime {\n  const [_version, setVersion] = useState(0);\n  const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);\n  const runtimeAdapters = useRuntimeAdapters();\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  const headersRef = useRef(options.headers);\n  useInsertionEffect(() => {\n    headersRef.current = options.headers;\n  });\n  const resolveHeaders = useCallback(() => {\n    const headers = headersRef.current;\n    return typeof headers === \"function\" ? headers() : (headers ?? {});\n  }, []);\n\n  const managedClientOptionsKey = options.client\n    ? null\n    : options.baseUrl\n      ? serializeManagedClientOptions({\n          baseUrl: options.baseUrl,\n          basePath: options.basePath,\n          tenant: options.tenant,\n          extensions: options.extensions,\n          fetchOptions: options.fetchOptions,\n        })\n      : null;\n\n  const client = useMemo(() => {\n    if (options.client) return options.client;\n    if (!managedClientOptionsKey) {\n      throw new Error(\"useA2ARuntime requires either `client` or `baseUrl`\");\n    }\n\n    return new A2AClient({\n      ...(JSON.parse(managedClientOptionsKey) as ManagedA2AClientOptions),\n      headers: resolveHeaders,\n    });\n  }, [managedClientOptionsKey, options.client, resolveHeaders]);\n\n  const coreOptions = {\n    client,\n    contextId: options.contextId,\n    configuration: options.configuration,\n    ...(options.onError && { onError: options.onError }),\n    ...(options.onCancel && { onCancel: options.onCancel }),\n    ...(options.onArtifactComplete && {\n      onArtifactComplete: options.onArtifactComplete,\n    }),\n    ...(historyAdapter && { history: historyAdapter }),\n  };\n  const coreOptionsRef = useRef(coreOptions);\n  coreOptionsRef.current = coreOptions;\n\n  const core = useMemo(\n    () =>\n      new A2AThreadRuntimeCore({\n        ...coreOptionsRef.current,\n        client,\n        notifyUpdate,\n      }),\n    [client, notifyUpdate],\n  );\n\n  useEffect(() => {\n    core.updateOptions(coreOptions);\n  });\n\n  // Thread list\n  const threadSwitchGenerationRef = useRef(0);\n  const threadList = useMemo(() => {\n    if (!threadListAdapter) return undefined;\n\n    const { onSwitchToNewThread, onSwitchToThread } = threadListAdapter;\n\n    return {\n      threadId: threadListAdapter.threadId,\n      onSwitchToNewThread: onSwitchToNewThread\n        ? async () => {\n            const generation = ++threadSwitchGenerationRef.current;\n            await onSwitchToNewThread();\n            if (generation !== threadSwitchGenerationRef.current) return;\n            // Apply first so the abort inside resetContext finds an already\n            // cleared repository and cannot persist the old thread's partial\n            // assistant message.\n            core.applyExternalMessages([]);\n            core.resetContext();\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const generation = ++threadSwitchGenerationRef.current;\n            const result = await onSwitchToThread(threadId);\n            if (generation !== threadSwitchGenerationRef.current) return;\n            core.applyExternalMessages(result.messages);\n            core.resetContext();\n          }\n        : undefined,\n    };\n  }, [threadListAdapter, core]);\n\n  // Adapters\n  const adapters = options.adapters;\n  const adapterAdapters = useMemo(\n    () => ({\n      attachments: adapters?.attachments ?? runtimeAdapters?.attachments,\n      speech: adapters?.speech,\n      dictation: adapters?.dictation,\n      voice: adapters?.voice,\n      feedback: adapters?.feedback,\n      threadList,\n    }),\n    [adapters, runtimeAdapters, threadList],\n  );\n\n  // Build store adapter\n  const shared = useExternalStoreSharedOptions(options);\n  const store = useMemo(() => {\n    void _version;\n\n    return {\n      ...shared,\n      isLoading: core.isLoading,\n      messageRepository: core.getMessageRepository(),\n      isRunning: core.isRunning(),\n      extras: a2aExtras.provide({\n        task: core.getTask(),\n        artifacts: core.getArtifacts(),\n        agentCard: core.getAgentCard(),\n      }),\n      onNew: (message: AppendMessage) => core.append(message),\n      onEdit: (message: AppendMessage) => core.edit(message),\n      onReload: (parentId: string | null) => core.reload(parentId),\n      onCancel: () => core.cancel(),\n      setMessages: (messages: readonly ThreadMessage[]) =>\n        core.applyExternalMessages(messages),\n      onImport: (messages: readonly ThreadMessage[]) =>\n        core.applyExternalMessages(messages),\n      adapters: adapterAdapters,\n    } satisfies ExternalStoreAdapter<ThreadMessage>;\n  }, [adapterAdapters, core, _version, shared]);\n\n  const runtime = useExternalStoreRuntime(store);\n\n  useEffect(() => {\n    core.attachRuntime(runtime);\n    return () => {\n      core.detachRuntime();\n    };\n  }, [core, runtime]);\n\n  useEffect(() => {\n    core.__internal_load();\n  }, [core]);\n\n  return runtime;\n}\n"],"mappings":";;;;;;;AA4BA,MAAM,iCACJ,YACW;CACX,MAAM,eAAe,OAAO,YAC1B,OAAO,QAAQ,QAAQ,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OACrD,EAAE,cAAc,CAAC,CACnB,CACF;CAEA,OAAO,KAAK,UAAU;EACpB,GAAG;EACH;CACF,CAAC;AACH;AAEA,SAAgB,cAAc,SAAiD;CAC7E,MAAM,CAAC,UAAU,cAAc,SAAS,CAAC;CACzC,MAAM,eAAe,kBAAkB,YAAY,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;CACnE,MAAM,kBAAkB,mBAAmB;CAC3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,MAAM,aAAa,OAAO,QAAQ,OAAO;CACzC,yBAAyB;EACvB,WAAW,UAAU,QAAQ;CAC/B,CAAC;CACD,MAAM,iBAAiB,kBAAkB;EACvC,MAAM,UAAU,WAAW;EAC3B,OAAO,OAAO,YAAY,aAAa,QAAQ,IAAK,WAAW,CAAC;CAClE,GAAG,CAAC,CAAC;CAEL,MAAM,0BAA0B,QAAQ,SACpC,OACA,QAAQ,UACN,8BAA8B;EAC5B,SAAS,QAAQ;EACjB,UAAU,QAAQ;EAClB,QAAQ,QAAQ;EAChB,YAAY,QAAQ;EACpB,cAAc,QAAQ;CACxB,CAAC,IACD;CAEN,MAAM,SAAS,cAAc;EAC3B,IAAI,QAAQ,QAAQ,OAAO,QAAQ;EACnC,IAAI,CAAC,yBACH,MAAM,IAAI,MAAM,qDAAqD;EAGvE,OAAO,IAAI,UAAU;GACnB,GAAI,KAAK,MAAM,uBAAuB;GACtC,SAAS;EACX,CAAC;CACH,GAAG;EAAC;EAAyB,QAAQ;EAAQ;CAAc,CAAC;CAE5D,MAAM,cAAc;EAClB;EACA,WAAW,QAAQ;EACnB,eAAe,QAAQ;EACvB,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,QAAQ,sBAAsB,EAChC,oBAAoB,QAAQ,mBAC9B;EACA,GAAI,kBAAkB,EAAE,SAAS,eAAe;CAClD;CACA,MAAM,iBAAiB,OAAO,WAAW;CACzC,eAAe,UAAU;CAEzB,MAAM,OAAO,cAET,IAAI,qBAAqB;EACvB,GAAG,eAAe;EAClB;EACA;CACF,CAAC,GACH,CAAC,QAAQ,YAAY,CACvB;CAEA,gBAAgB;EACd,KAAK,cAAc,WAAW;CAChC,CAAC;CAGD,MAAM,4BAA4B,OAAO,CAAC;CAC1C,MAAM,aAAa,cAAc;EAC/B,IAAI,CAAC,mBAAmB,OAAO,KAAA;EAE/B,MAAM,EAAE,qBAAqB,qBAAqB;EAElD,OAAO;GACL,UAAU,kBAAkB;GAC5B,qBAAqB,sBACjB,YAAY;IACV,MAAM,aAAa,EAAE,0BAA0B;IAC/C,MAAM,oBAAoB;IAC1B,IAAI,eAAe,0BAA0B,SAAS;IAItD,KAAK,sBAAsB,CAAC,CAAC;IAC7B,KAAK,aAAa;GACpB,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,aAAa,EAAE,0BAA0B;IAC/C,MAAM,SAAS,MAAM,iBAAiB,QAAQ;IAC9C,IAAI,eAAe,0BAA0B,SAAS;IACtD,KAAK,sBAAsB,OAAO,QAAQ;IAC1C,KAAK,aAAa;GACpB,IACA,KAAA;EACN;CACF,GAAG,CAAC,mBAAmB,IAAI,CAAC;CAG5B,MAAM,WAAW,QAAQ;CACzB,MAAM,kBAAkB,eACf;EACL,aAAa,UAAU,eAAe,iBAAiB;EACvD,QAAQ,UAAU;EAClB,WAAW,UAAU;EACrB,OAAO,UAAU;EACjB,UAAU,UAAU;EACpB;CACF,IACA;EAAC;EAAU;EAAiB;CAAU,CACxC;CAGA,MAAM,SAAS,8BAA8B,OAAO;CACpD,MAAM,QAAQ,cAAc;EAG1B,OAAO;GACL,GAAG;GACH,WAAW,KAAK;GAChB,mBAAmB,KAAK,qBAAqB;GAC7C,WAAW,KAAK,UAAU;GAC1B,QAAQ,UAAU,QAAQ;IACxB,MAAM,KAAK,QAAQ;IACnB,WAAW,KAAK,aAAa;IAC7B,WAAW,KAAK,aAAa;GAC/B,CAAC;GACD,QAAQ,YAA2B,KAAK,OAAO,OAAO;GACtD,SAAS,YAA2B,KAAK,KAAK,OAAO;GACrD,WAAW,aAA4B,KAAK,OAAO,QAAQ;GAC3D,gBAAgB,KAAK,OAAO;GAC5B,cAAc,aACZ,KAAK,sBAAsB,QAAQ;GACrC,WAAW,aACT,KAAK,sBAAsB,QAAQ;GACrC,UAAU;EACZ;CACF,GAAG;EAAC;EAAiB;EAAM;EAAU;CAAM,CAAC;CAE5C,MAAM,UAAU,wBAAwB,KAAK;CAE7C,gBAAgB;EACd,KAAK,cAAc,OAAO;EAC1B,aAAa;GACX,KAAK,cAAc;EACrB;CACF,GAAG,CAAC,MAAM,OAAO,CAAC;CAElB,gBAAgB;EACd,KAAK,gBAAgB;CACvB,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;AACT"}
+{"version":3,"file":"useA2ARuntime.js","names":[],"sources":["../src/useA2ARuntime.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  useCallback,\n  useEffect,\n  useInsertionEffect,\n  useMemo,\n  useRef,\n  useState,\n} from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport type {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport { A2AClient, type A2AClientOptions } from \"./A2AClient\";\nimport { A2AThreadRuntimeCore } from \"./A2AThreadRuntimeCore\";\nimport { a2aExtras } from \"./a2aExtras\";\nimport type { UseA2ARuntimeOptions } from \"./types\";\n\ntype ManagedA2AClientOptions = Omit<A2AClientOptions, \"headers\">;\n\nconst serializeManagedClientOptions = (\n  options: ManagedA2AClientOptions,\n): string => {\n  const fetchOptions = Object.fromEntries(\n    Object.entries(options.fetchOptions ?? {}).sort(([a], [b]) =>\n      a.localeCompare(b),\n    ),\n  );\n\n  return JSON.stringify({\n    ...options,\n    fetchOptions,\n  });\n};\n\nexport function useA2ARuntime(options: UseA2ARuntimeOptions): AssistantRuntime {\n  const [_version, setVersion] = useState(0);\n  const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);\n  const runtimeAdapters = useRuntimeAdapters();\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  const headersRef = useRef(options.headers);\n  useInsertionEffect(() => {\n    headersRef.current = options.headers;\n  });\n  const resolveHeaders = useCallback(() => {\n    const headers = headersRef.current;\n    return typeof headers === \"function\" ? headers() : (headers ?? {});\n  }, []);\n\n  const managedClientOptionsKey = options.client\n    ? null\n    : options.baseUrl\n      ? serializeManagedClientOptions({\n          baseUrl: options.baseUrl,\n          basePath: options.basePath,\n          tenant: options.tenant,\n          extensions: options.extensions,\n          fetchOptions: options.fetchOptions,\n        })\n      : null;\n\n  const client = useMemo(() => {\n    if (options.client) return options.client;\n    if (!managedClientOptionsKey) {\n      throw new Error(\"useA2ARuntime requires either `client` or `baseUrl`\");\n    }\n\n    return new A2AClient({\n      ...(JSON.parse(managedClientOptionsKey) as ManagedA2AClientOptions),\n      headers: resolveHeaders,\n    });\n  }, [managedClientOptionsKey, options.client, resolveHeaders]);\n\n  const coreOptions = {\n    client,\n    contextId: options.contextId,\n    configuration: options.configuration,\n    ...(options.onError && { onError: options.onError }),\n    ...(options.onCancel && { onCancel: options.onCancel }),\n    ...(options.onArtifactComplete && {\n      onArtifactComplete: options.onArtifactComplete,\n    }),\n    ...(historyAdapter && { history: historyAdapter }),\n  };\n  const coreOptionsRef = useRef(coreOptions);\n  coreOptionsRef.current = coreOptions;\n\n  const core = useMemo(\n    () =>\n      new A2AThreadRuntimeCore({\n        ...coreOptionsRef.current,\n        client,\n        notifyUpdate,\n      }),\n    [client, notifyUpdate],\n  );\n\n  useEffect(() => {\n    core.updateOptions(coreOptions);\n  });\n\n  // Thread list\n  const threadSwitchGenerationRef = useRef(0);\n  const threadList = useMemo(() => {\n    if (!threadListAdapter) return undefined;\n\n    const { onSwitchToNewThread, onSwitchToThread } = threadListAdapter;\n\n    return {\n      threadId: threadListAdapter.threadId,\n      onSwitchToNewThread: onSwitchToNewThread\n        ? async () => {\n            const generation = ++threadSwitchGenerationRef.current;\n            // Clear before the thread id flips, or the old messages leak\n            // into the new thread as a sibling branch.\n            core.applyExternalMessages([]);\n            core.resetContext();\n            await onSwitchToNewThread();\n            if (generation !== threadSwitchGenerationRef.current) return;\n            // Apply first so the abort inside resetContext finds an already\n            // cleared repository and cannot persist the old thread's partial\n            // assistant message.\n            core.applyExternalMessages([]);\n            core.resetContext();\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const generation = ++threadSwitchGenerationRef.current;\n            // Clear before the thread id flips, or the old messages leak\n            // into the new thread as a sibling branch.\n            core.applyExternalMessages([]);\n            core.resetContext();\n            const result = await onSwitchToThread(threadId);\n            if (generation !== threadSwitchGenerationRef.current) return;\n            core.applyExternalMessages([]);\n            core.applyExternalMessages(result.messages);\n            core.resetContext();\n          }\n        : undefined,\n    };\n  }, [threadListAdapter, core]);\n\n  // Adapters\n  const adapters = options.adapters;\n  const adapterAdapters = useMemo(\n    () => ({\n      attachments: adapters?.attachments ?? runtimeAdapters?.attachments,\n      speech: adapters?.speech,\n      dictation: adapters?.dictation,\n      voice: adapters?.voice,\n      feedback: adapters?.feedback,\n      threadList,\n    }),\n    [adapters, runtimeAdapters, threadList],\n  );\n\n  // Build store adapter\n  const shared = useExternalStoreSharedOptions(options);\n  const store = useMemo(() => {\n    void _version;\n\n    return {\n      ...shared,\n      isLoading: core.isLoading,\n      messageRepository: core.getMessageRepository(),\n      isRunning: core.isRunning(),\n      unstable_persistsHistory: true,\n      extras: a2aExtras.provide({\n        task: core.getTask(),\n        artifacts: core.getArtifacts(),\n        agentCard: core.getAgentCard(),\n      }),\n      onNew: (message: AppendMessage) => core.append(message),\n      onVoiceTranscript: (message: ThreadMessage) =>\n        core.appendVoiceTranscript(message),\n      onEdit: (message: AppendMessage) => core.edit(message),\n      onReload: (parentId: string | null) => core.reload(parentId),\n      onCancel: () => core.cancel(),\n      unstable_onRecordToolInteraction: (options) =>\n        core.recordToolInteraction(options),\n      setMessages: (messages: readonly ThreadMessage[]) =>\n        core.applyExternalMessages(messages),\n      onImport: (messages: readonly ThreadMessage[]) =>\n        core.applyExternalMessages(messages),\n      adapters: adapterAdapters,\n    } satisfies ExternalStoreAdapter<ThreadMessage>;\n  }, [adapterAdapters, core, _version, shared]);\n\n  const runtime = useExternalStoreRuntime(store);\n\n  useEffect(() => {\n    core.attachRuntime(runtime);\n    return () => {\n      core.detachRuntime();\n    };\n  }, [core, runtime]);\n\n  useEffect(() => {\n    core.__internal_load();\n  }, [core]);\n\n  return runtime;\n}\n"],"mappings":";;;;;;;AA4BA,MAAM,iCACJ,YACW;CACX,MAAM,eAAe,OAAO,YAC1B,OAAO,QAAQ,QAAQ,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OACrD,EAAE,cAAc,CAAC,CACnB,CACF;CAEA,OAAO,KAAK,UAAU;EACpB,GAAG;EACH;CACF,CAAC;AACH;AAEA,SAAgB,cAAc,SAAiD;CAC7E,MAAM,CAAC,UAAU,cAAc,SAAS,CAAC;CACzC,MAAM,eAAe,kBAAkB,YAAY,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;CACnE,MAAM,kBAAkB,mBAAmB;CAC3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,MAAM,aAAa,OAAO,QAAQ,OAAO;CACzC,yBAAyB;EACvB,WAAW,UAAU,QAAQ;CAC/B,CAAC;CACD,MAAM,iBAAiB,kBAAkB;EACvC,MAAM,UAAU,WAAW;EAC3B,OAAO,OAAO,YAAY,aAAa,QAAQ,IAAK,WAAW,CAAC;CAClE,GAAG,CAAC,CAAC;CAEL,MAAM,0BAA0B,QAAQ,SACpC,OACA,QAAQ,UACN,8BAA8B;EAC5B,SAAS,QAAQ;EACjB,UAAU,QAAQ;EAClB,QAAQ,QAAQ;EAChB,YAAY,QAAQ;EACpB,cAAc,QAAQ;CACxB,CAAC,IACD;CAEN,MAAM,SAAS,cAAc;EAC3B,IAAI,QAAQ,QAAQ,OAAO,QAAQ;EACnC,IAAI,CAAC,yBACH,MAAM,IAAI,MAAM,qDAAqD;EAGvE,OAAO,IAAI,UAAU;GACnB,GAAI,KAAK,MAAM,uBAAuB;GACtC,SAAS;EACX,CAAC;CACH,GAAG;EAAC;EAAyB,QAAQ;EAAQ;CAAc,CAAC;CAE5D,MAAM,cAAc;EAClB;EACA,WAAW,QAAQ;EACnB,eAAe,QAAQ;EACvB,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,QAAQ,sBAAsB,EAChC,oBAAoB,QAAQ,mBAC9B;EACA,GAAI,kBAAkB,EAAE,SAAS,eAAe;CAClD;CACA,MAAM,iBAAiB,OAAO,WAAW;CACzC,eAAe,UAAU;CAEzB,MAAM,OAAO,cAET,IAAI,qBAAqB;EACvB,GAAG,eAAe;EAClB;EACA;CACF,CAAC,GACH,CAAC,QAAQ,YAAY,CACvB;CAEA,gBAAgB;EACd,KAAK,cAAc,WAAW;CAChC,CAAC;CAGD,MAAM,4BAA4B,OAAO,CAAC;CAC1C,MAAM,aAAa,cAAc;EAC/B,IAAI,CAAC,mBAAmB,OAAO,KAAA;EAE/B,MAAM,EAAE,qBAAqB,qBAAqB;EAElD,OAAO;GACL,UAAU,kBAAkB;GAC5B,qBAAqB,sBACjB,YAAY;IACV,MAAM,aAAa,EAAE,0BAA0B;IAG/C,KAAK,sBAAsB,CAAC,CAAC;IAC7B,KAAK,aAAa;IAClB,MAAM,oBAAoB;IAC1B,IAAI,eAAe,0BAA0B,SAAS;IAItD,KAAK,sBAAsB,CAAC,CAAC;IAC7B,KAAK,aAAa;GACpB,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,aAAa,EAAE,0BAA0B;IAG/C,KAAK,sBAAsB,CAAC,CAAC;IAC7B,KAAK,aAAa;IAClB,MAAM,SAAS,MAAM,iBAAiB,QAAQ;IAC9C,IAAI,eAAe,0BAA0B,SAAS;IACtD,KAAK,sBAAsB,CAAC,CAAC;IAC7B,KAAK,sBAAsB,OAAO,QAAQ;IAC1C,KAAK,aAAa;GACpB,IACA,KAAA;EACN;CACF,GAAG,CAAC,mBAAmB,IAAI,CAAC;CAG5B,MAAM,WAAW,QAAQ;CACzB,MAAM,kBAAkB,eACf;EACL,aAAa,UAAU,eAAe,iBAAiB;EACvD,QAAQ,UAAU;EAClB,WAAW,UAAU;EACrB,OAAO,UAAU;EACjB,UAAU,UAAU;EACpB;CACF,IACA;EAAC;EAAU;EAAiB;CAAU,CACxC;CAGA,MAAM,SAAS,8BAA8B,OAAO;CACpD,MAAM,QAAQ,cAAc;EAG1B,OAAO;GACL,GAAG;GACH,WAAW,KAAK;GAChB,mBAAmB,KAAK,qBAAqB;GAC7C,WAAW,KAAK,UAAU;GAC1B,0BAA0B;GAC1B,QAAQ,UAAU,QAAQ;IACxB,MAAM,KAAK,QAAQ;IACnB,WAAW,KAAK,aAAa;IAC7B,WAAW,KAAK,aAAa;GAC/B,CAAC;GACD,QAAQ,YAA2B,KAAK,OAAO,OAAO;GACtD,oBAAoB,YAClB,KAAK,sBAAsB,OAAO;GACpC,SAAS,YAA2B,KAAK,KAAK,OAAO;GACrD,WAAW,aAA4B,KAAK,OAAO,QAAQ;GAC3D,gBAAgB,KAAK,OAAO;GAC5B,mCAAmC,YACjC,KAAK,sBAAsB,OAAO;GACpC,cAAc,aACZ,KAAK,sBAAsB,QAAQ;GACrC,WAAW,aACT,KAAK,sBAAsB,QAAQ;GACrC,UAAU;EACZ;CACF,GAAG;EAAC;EAAiB;EAAM;EAAU;CAAM,CAAC;CAE5C,MAAM,UAAU,wBAAwB,KAAK;CAE7C,gBAAgB;EACd,KAAK,cAAc,OAAO;EAC1B,aAAa;GACX,KAAK,cAAc;EACrB;CACF,GAAG,CAAC,MAAM,OAAO,CAAC;CAElB,gBAAgB;EACd,KAAK,gBAAgB;CACvB,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;AACT"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-a2a",
-  "version": "0.2.35",
+  "version": "0.2.37",
   "description": "A2A (Agent-to-Agent) v1.0 protocol adapter for assistant-ui",
   "keywords": [
     "a2a",
@@ -29,9 +29,10 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.3.19",
-    "@assistant-ui/store": "^0.3.13",
-    "assistant-stream": "^0.3.43"
+    "@assistant-ui/core": "^0.3.21",
+    "@assistant-ui/react-generative-ui": "^0.0.20",
+    "@assistant-ui/store": "^0.3.15",
+    "assistant-stream": "^0.3.45"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -43,14 +44,14 @@
     }
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.27",
-    "@testing-library/dom": "^10.4.1",
+    "@assistant-ui/x-buildutils": "0.0.29",
+    "@testing-library/dom": "^10.4.2",
     "@testing-library/react": "^16.3.3",
-    "@types/react": "^19.2.18",
+    "@types/react": "^19.3.0",
     "jsdom": "^30.0.1",
-    "react": "^19.2.8",
-    "react-dom": "^19.2.8",
-    "vitest": "^5.0.0"
+    "react": "^19.3.0",
+    "react-dom": "^19.3.0",
+    "vitest": "^5.0.1"
   },
   "publishConfig": {
     "access": "public",
@@ -68,6 +69,7 @@
   "scripts": {
     "build": "aui-build",
     "test": "vitest run",
+    "typecheck": "tsc --noEmit",
     "test:watch": "vitest"
   }
 }

package/src/A2AClient.test.ts

@@ -160,6 +160,47 @@ describe("A2AClient", () => {
       const [, init] = fetchMock.mock.calls[0]!;
       expect(init.headers.Authorization).toBe("Bearer tok123");
     });
+
+    it("aborts a request while dynamic headers are pending", async () => {
+      const dynamicClient = new A2AClient({
+        baseUrl: "https://agent.test",
+        headers: () => new Promise<Record<string, string>>(() => {}),
+      });
+      const controller = new AbortController();
+      const reason = new Error("cancelled");
+
+      const request = dynamicClient.sendMessage(
+        userMessage,
+        undefined,
+        undefined,
+        controller.signal,
+      );
+      controller.abort(reason);
+
+      await expect(request).rejects.toBe(reason);
+      expect(fetchMock).not.toHaveBeenCalled();
+    });
+
+    it("aborts a stream while dynamic headers are pending", async () => {
+      const dynamicClient = new A2AClient({
+        baseUrl: "https://agent.test",
+        headers: () => new Promise<Record<string, string>>(() => {}),
+      });
+      const controller = new AbortController();
+      const reason = new Error("cancelled");
+      const stream = dynamicClient.streamMessage(
+        userMessage,
+        undefined,
+        undefined,
+        controller.signal,
+      );
+
+      const next = stream.next();
+      controller.abort(reason);
+
+      await expect(next).rejects.toBe(reason);
+      expect(fetchMock).not.toHaveBeenCalled();
+    });
   });
 
   describe("fetchOptions", () => {

package/src/A2AClient.ts

@@ -680,6 +680,49 @@ function signalInit(signal?: AbortSignal): RequestInit {
   return signal ? { signal } : {};
 }
 
+const getAbortReason = (signal: AbortSignal): unknown => {
+  if (signal.reason !== undefined) return signal.reason;
+  const error = new Error("The operation was aborted");
+  error.name = "AbortError";
+  return error;
+};
+
+const raceWithAbortSignal = <T>(
+  signal: AbortSignal | undefined,
+  operation: () => T | PromiseLike<T>,
+): Promise<T> => {
+  if (!signal) return Promise.resolve().then(operation);
+  if (signal.aborted) return Promise.reject(getAbortReason(signal));
+
+  return new Promise<T>((resolve, reject) => {
+    let settled = false;
+    const cleanup = () => signal.removeEventListener("abort", handleAbort);
+    const resolveOnce = (value: T) => {
+      if (settled) return;
+      settled = true;
+      cleanup();
+      resolve(value);
+    };
+    const rejectOnce = (error: unknown) => {
+      if (settled) return;
+      settled = true;
+      cleanup();
+      reject(error);
+    };
+    const handleAbort = () => rejectOnce(getAbortReason(signal));
+
+    signal.addEventListener("abort", handleAbort, { once: true });
+    let result: T | PromiseLike<T>;
+    try {
+      result = operation();
+    } catch (error) {
+      rejectOnce(error);
+      return;
+    }
+    Promise.resolve(result).then(resolveOnce, rejectOnce);
+  });
+};
+
 const SKIPPED_FRAME_SNIPPET_LENGTH = 120;
 
 function describeSkippedFrame(data: string, reason: string): string {
@@ -728,10 +771,11 @@ export class A2AClient {
 
   private async getHeaders(
     includeContentType = true,
+    signal?: AbortSignal,
   ): Promise<Record<string, string>> {
     const custom =
       typeof this.headersFn === "function"
-        ? await this.headersFn()
+        ? await raceWithAbortSignal(signal, this.headersFn)
         : this.headersFn;
     const headers: Record<string, string> = {
       Accept: "application/a2a+json, application/json",
@@ -777,7 +821,7 @@ export class A2AClient {
     options: RequestInit = {},
   ): Promise<T> {
     const isGet = !options.method || options.method.toUpperCase() === "GET";
-    const headers = await this.getHeaders(!isGet);
+    const headers = await this.getHeaders(!isGet, options.signal ?? undefined);
     const response = await fetch(`${this.baseUrl}${path}`, {
       ...this.fetchOptions,
       ...options,
@@ -809,7 +853,7 @@ export class A2AClient {
   // --- Agent Card ---
 
   async getAgentCard(signal?: AbortSignal): Promise<A2AAgentCard> {
-    const headers = await this.getHeaders(false); // GET: no Content-Type
+    const headers = await this.getHeaders(false, signal); // GET: no Content-Type
     const url = `${this.baseUrl}/.well-known/agent-card.json`;
     const response = await fetch(url, {
       ...this.fetchOptions,
@@ -863,7 +907,7 @@ export class A2AClient {
     metadata?: Record<string, unknown>,
     signal?: AbortSignal,
   ): AsyncGenerator<A2AStreamEvent> {
-    const headers = await this.getHeaders(true);
+    const headers = await this.getHeaders(true, signal);
     headers.Accept = "text/event-stream";
 
     const body: Record<string, unknown> = {
@@ -954,7 +998,7 @@ export class A2AClient {
     taskId: string,
     signal?: AbortSignal,
   ): AsyncGenerator<A2AStreamEvent> {
-    const headers = await this.getHeaders(false); // GET: no Content-Type
+    const headers = await this.getHeaders(false, signal); // GET: no Content-Type
     headers.Accept = "text/event-stream";
 
     const response = await fetch(
@@ -1032,7 +1076,7 @@ export class A2AClient {
     signal?: AbortSignal,
   ): Promise<void> {
     const isGet = false;
-    const headers = await this.getHeaders(!isGet);
+    const headers = await this.getHeaders(!isGet, signal);
     const response = await fetch(
       `${this.baseUrl}${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}/pushNotificationConfigs/${encodeURIComponent(configId)}`,
       {