@assistant-ui/react-a2a 0.2.20 → 0.2.22

package/dist/useA2ARuntime.js

@@ -1,24 +1,10 @@
 "use client";
 import { A2AClient } from "./A2AClient.js";
 import { A2AThreadRuntimeCore } from "./A2AThreadRuntimeCore.js";
+import { a2aExtras } from "./a2aExtras.js";
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { useExternalStoreRuntime, useExternalStoreSharedOptions, useRuntimeAdapters } from "@assistant-ui/core/react";
-import { useAuiState } from "@assistant-ui/store";
 //#region src/useA2ARuntime.ts
-const symbolA2AExtras = Symbol("a2a-extras");
-const asA2AExtras = (extras) => {
-	if (typeof extras !== "object" || extras == null || !(symbolA2AExtras in extras)) throw new Error("This hook can only be used inside a useA2ARuntime provider");
-	return extras;
-};
-const useA2ATask = () => {
-	return useAuiState((s) => asA2AExtras(s.thread.extras).task);
-};
-const useA2AArtifacts = () => {
-	return useAuiState((s) => asA2AExtras(s.thread.extras).artifacts);
-};
-const useA2AAgentCard = () => {
-	return useAuiState((s) => asA2AExtras(s.thread.extras).agentCard);
-};
 function useA2ARuntime(options) {
 	const [_version, setVersion] = useState(0);
 	const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);
@@ -93,12 +79,11 @@ function useA2ARuntime(options) {
 			isLoading: core.isLoading,
 			messages: core.getMessages(),
 			isRunning: core.isRunning(),
-			extras: {
-				[symbolA2AExtras]: true,
+			extras: a2aExtras.provide({
 				task: core.getTask(),
 				artifacts: core.getArtifacts(),
 				agentCard: core.getAgentCard()
-			},
+			}),
 			onNew: (message) => core.append(message),
 			onEdit: (message) => core.edit(message),
 			onReload: (parentId) => core.reload(parentId),
@@ -125,6 +110,6 @@ function useA2ARuntime(options) {
 	return runtime;
 }
 //#endregion
-export { useA2AAgentCard, useA2AArtifacts, useA2ARuntime, useA2ATask };
+export { useA2ARuntime };
 
 //# sourceMappingURL=useA2ARuntime.js.map

package/dist/useA2ARuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useA2ARuntime.js","names":[],"sources":["../src/useA2ARuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useCallback, useEffect, useMemo, useRef, useState } from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport type {\n  AssistantRuntime,\n  AppendMessage,\n  AttachmentAdapter,\n  DictationAdapter,\n  ExternalStoreAdapter,\n  ExternalStoreSharedOptions,\n  FeedbackAdapter,\n  RealtimeVoiceAdapter,\n  SpeechSynthesisAdapter,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { A2AClient } from \"./A2AClient\";\nimport type { A2AClientOptions } from \"./A2AClient\";\nimport { A2AThreadRuntimeCore } from \"./A2AThreadRuntimeCore\";\nimport type {\n  A2AArtifact,\n  A2AAgentCard,\n  A2ASendMessageConfiguration,\n  A2ATask,\n} from \"./types\";\n\n// --- Extras symbol for A2A-specific state ---\n\nconst symbolA2AExtras = Symbol(\"a2a-extras\");\n\ntype A2AExtras = {\n  [symbolA2AExtras]: true;\n  task: A2ATask | undefined;\n  artifacts: readonly A2AArtifact[];\n  agentCard: A2AAgentCard | undefined;\n};\n\nconst asA2AExtras = (extras: unknown): A2AExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolA2AExtras in extras)\n  )\n    throw new Error(\n      \"This hook can only be used inside a useA2ARuntime provider\",\n    );\n  return extras as A2AExtras;\n};\n\n// --- Public hooks for A2A state ---\n\nexport const useA2ATask = () => {\n  return useAuiState((s) => asA2AExtras(s.thread.extras).task);\n};\n\nexport const useA2AArtifacts = () => {\n  return useAuiState((s) => asA2AExtras(s.thread.extras).artifacts);\n};\n\nexport const useA2AAgentCard = () => {\n  return useAuiState((s) => asA2AExtras(s.thread.extras).agentCard);\n};\n\n// --- Thread list adapter type ---\n\nexport type UseA2AThreadListAdapter = {\n  threadId?: string;\n  onSwitchToNewThread?: () => Promise<void> | void;\n  onSwitchToThread?: (threadId: string) => Promise<{\n    messages: readonly ThreadMessage[];\n  }>;\n};\n\n// --- Options ---\n\nexport type UseA2ARuntimeOptions = ExternalStoreSharedOptions & {\n  /** Pre-built A2A client instance. Provide this OR baseUrl. */\n  client?: A2AClient;\n  /** Base URL of the A2A server. Used to create a client if `client` is not provided. */\n  baseUrl?: string;\n  /** Optional path prefix for all API endpoints (e.g. \"/v1\"). Does not affect agent card discovery. Only used with baseUrl. */\n  basePath?: string;\n  /** Optional tenant ID for multi-tenant servers. Only used with baseUrl. */\n  tenant?: string;\n  /** Headers for the A2A client (only used with baseUrl). */\n  headers?: A2AClientOptions[\"headers\"];\n  /** A2A extension URIs to negotiate. Only used with baseUrl. */\n  extensions?: string[];\n  /** Extra fetch options (e.g. `{ credentials: 'include' }`). Only used with `baseUrl`. */\n  fetchOptions?: A2AClientOptions[\"fetchOptions\"];\n\n  /** Initial context ID for the conversation. */\n  contextId?: string;\n  /** Default send message configuration. */\n  configuration?: A2ASendMessageConfiguration;\n\n  /** Called when an error occurs. */\n  onError?: (error: Error) => void;\n  /** Called when a run is cancelled. */\n  onCancel?: () => void;\n  /** Called when an artifact is fully received (lastChunk). */\n  onArtifactComplete?: (artifact: import(\"./types\").A2AArtifact) => void;\n\n  adapters?: {\n    attachments?: AttachmentAdapter;\n    speech?: SpeechSynthesisAdapter;\n    dictation?: DictationAdapter;\n    voice?: RealtimeVoiceAdapter;\n    feedback?: FeedbackAdapter;\n    history?: ThreadHistoryAdapter;\n    threadList?: UseA2AThreadListAdapter;\n  };\n};\n\n// --- Main hook ---\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  // Create or reuse client\n  const clientRef = useRef<A2AClient | null>(null);\n  if (!clientRef.current) {\n    if (options.client) {\n      clientRef.current = options.client;\n    } else if (options.baseUrl) {\n      clientRef.current = new A2AClient({\n        baseUrl: options.baseUrl,\n        basePath: options.basePath,\n        tenant: options.tenant,\n        headers: options.headers,\n        extensions: options.extensions,\n        fetchOptions: options.fetchOptions,\n      });\n    } else {\n      throw new Error(\"useA2ARuntime requires either `client` or `baseUrl`\");\n    }\n  }\n  const client = clientRef.current;\n\n  // Create or reuse core\n  const coreRef = useRef<A2AThreadRuntimeCore | null>(null);\n  if (!coreRef.current) {\n    coreRef.current = new A2AThreadRuntimeCore({\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      notifyUpdate,\n    });\n  }\n\n  const core = coreRef.current;\n  core.updateOptions({\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\n  // Thread list\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            await onSwitchToNewThread();\n            core.applyExternalMessages([]);\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const result = await onSwitchToThread(threadId);\n            core.applyExternalMessages(result.messages);\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      messages: core.getMessages(),\n      isRunning: core.isRunning(),\n      extras: {\n        [symbolA2AExtras]: true,\n        task: core.getTask(),\n        artifacts: core.getArtifacts(),\n        agentCard: core.getAgentCard(),\n      } satisfies A2AExtras,\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":";;;;;;;AAkCA,MAAM,kBAAkB,OAAO,YAAY;AAS3C,MAAM,eAAe,WAA+B;CAClD,IACE,OAAO,WAAW,YAClB,UAAU,QACV,EAAE,mBAAmB,SAErB,MAAM,IAAI,MACR,4DACF;CACF,OAAO;AACT;AAIA,MAAa,mBAAmB;CAC9B,OAAO,aAAa,MAAM,YAAY,EAAE,OAAO,MAAM,CAAC,CAAC,IAAI;AAC7D;AAEA,MAAa,wBAAwB;CACnC,OAAO,aAAa,MAAM,YAAY,EAAE,OAAO,MAAM,CAAC,CAAC,SAAS;AAClE;AAEA,MAAa,wBAAwB;CACnC,OAAO,aAAa,MAAM,YAAY,EAAE,OAAO,MAAM,CAAC,CAAC,SAAS;AAClE;AAuDA,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;CAG5C,MAAM,YAAY,OAAyB,IAAI;CAC/C,IAAI,CAAC,UAAU,SACb,IAAI,QAAQ,QACV,UAAU,UAAU,QAAQ;MACvB,IAAI,QAAQ,SACjB,UAAU,UAAU,IAAI,UAAU;EAChC,SAAS,QAAQ;EACjB,UAAU,QAAQ;EAClB,QAAQ,QAAQ;EAChB,SAAS,QAAQ;EACjB,YAAY,QAAQ;EACpB,cAAc,QAAQ;CACxB,CAAC;MAED,MAAM,IAAI,MAAM,qDAAqD;CAGzE,MAAM,SAAS,UAAU;CAGzB,MAAM,UAAU,OAAoC,IAAI;CACxD,IAAI,CAAC,QAAQ,SACX,QAAQ,UAAU,IAAI,qBAAqB;EACzC;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;EAChD;CACF,CAAC;CAGH,MAAM,OAAO,QAAQ;CACrB,KAAK,cAAc;EACjB;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,CAAC;CAGD,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,oBAAoB;IAC1B,KAAK,sBAAsB,CAAC,CAAC;GAC/B,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,SAAS,MAAM,iBAAiB,QAAQ;IAC9C,KAAK,sBAAsB,OAAO,QAAQ;GAC5C,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;CA2BpD,MAAM,UAAU,wBA1BF,cAAc;EAG1B,OAAO;GACL,GAAG;GACH,WAAW,KAAK;GAChB,UAAU,KAAK,YAAY;GAC3B,WAAW,KAAK,UAAU;GAC1B,QAAQ;KACL,kBAAkB;IACnB,MAAM,KAAK,QAAQ;IACnB,WAAW,KAAK,aAAa;IAC7B,WAAW,KAAK,aAAa;GAC/B;GACA,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,CAEC,CAAC;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 { useCallback, useEffect, useMemo, useRef, useState } 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 } from \"./A2AClient\";\nimport { A2AThreadRuntimeCore } from \"./A2AThreadRuntimeCore\";\nimport { a2aExtras } from \"./a2aExtras\";\nimport type { UseA2ARuntimeOptions } from \"./types\";\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  // Create or reuse client\n  const clientRef = useRef<A2AClient | null>(null);\n  if (!clientRef.current) {\n    if (options.client) {\n      clientRef.current = options.client;\n    } else if (options.baseUrl) {\n      clientRef.current = new A2AClient({\n        baseUrl: options.baseUrl,\n        basePath: options.basePath,\n        tenant: options.tenant,\n        headers: options.headers,\n        extensions: options.extensions,\n        fetchOptions: options.fetchOptions,\n      });\n    } else {\n      throw new Error(\"useA2ARuntime requires either `client` or `baseUrl`\");\n    }\n  }\n  const client = clientRef.current;\n\n  // Create or reuse core\n  const coreRef = useRef<A2AThreadRuntimeCore | null>(null);\n  if (!coreRef.current) {\n    coreRef.current = new A2AThreadRuntimeCore({\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      notifyUpdate,\n    });\n  }\n\n  const core = coreRef.current;\n  core.updateOptions({\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\n  // Thread list\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            await onSwitchToNewThread();\n            core.applyExternalMessages([]);\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const result = await onSwitchToThread(threadId);\n            core.applyExternalMessages(result.messages);\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      messages: core.getMessages(),\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":";;;;;;;AAmBA,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;CAG5C,MAAM,YAAY,OAAyB,IAAI;CAC/C,IAAI,CAAC,UAAU,SACb,IAAI,QAAQ,QACV,UAAU,UAAU,QAAQ;MACvB,IAAI,QAAQ,SACjB,UAAU,UAAU,IAAI,UAAU;EAChC,SAAS,QAAQ;EACjB,UAAU,QAAQ;EAClB,QAAQ,QAAQ;EAChB,SAAS,QAAQ;EACjB,YAAY,QAAQ;EACpB,cAAc,QAAQ;CACxB,CAAC;MAED,MAAM,IAAI,MAAM,qDAAqD;CAGzE,MAAM,SAAS,UAAU;CAGzB,MAAM,UAAU,OAAoC,IAAI;CACxD,IAAI,CAAC,QAAQ,SACX,QAAQ,UAAU,IAAI,qBAAqB;EACzC;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;EAChD;CACF,CAAC;CAGH,MAAM,OAAO,QAAQ;CACrB,KAAK,cAAc;EACjB;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,CAAC;CAGD,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,oBAAoB;IAC1B,KAAK,sBAAsB,CAAC,CAAC;GAC/B,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,SAAS,MAAM,iBAAiB,QAAQ;IAC9C,KAAK,sBAAsB,OAAO,QAAQ;GAC5C,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;CA0BpD,MAAM,UAAU,wBAzBF,cAAc;EAG1B,OAAO;GACL,GAAG;GACH,WAAW,KAAK;GAChB,UAAU,KAAK,YAAY;GAC3B,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,CAEC,CAAC;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.20",
+  "version": "0.2.22",
   "description": "A2A (Agent-to-Agent) v1.0 protocol adapter for assistant-ui",
   "keywords": [
     "a2a",
@@ -29,8 +29,9 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.2.17",
-    "@assistant-ui/store": "^0.2.18"
+    "@assistant-ui/core": "^0.2.21",
+    "@assistant-ui/store": "^0.2.20",
+    "assistant-stream": "^0.3.26"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -44,8 +45,8 @@
   "devDependencies": {
     "@types/react": "^19.2.17",
     "react": "^19.2.7",
-    "vitest": "^4.1.8",
-    "@assistant-ui/x-buildutils": "0.0.15"
+    "vitest": "^4.1.10",
+    "@assistant-ui/x-buildutils": "0.0.18"
   },
   "publishConfig": {
     "access": "public",

package/src/A2AClient.test.ts

@@ -14,17 +14,23 @@ function mockFetchResponse(body: unknown, ok = true, status = 200): Response {
   } as unknown as Response;
 }
 
-function mockSSEResponse(lines: string[]): Response {
-  const text = lines.join("\n");
+function mockSSETextResponse(
+  text: string | string[],
+  contentType: string | null = "text/event-stream",
+): Response {
   const encoder = new TextEncoder();
-  const chunks = [encoder.encode(text)];
+  const chunks = (Array.isArray(text) ? text : [text]).map((chunk) =>
+    encoder.encode(chunk),
+  );
   let index = 0;
+  const headers = new Headers();
+  if (contentType !== null) headers.set("content-type", contentType);
 
   return {
     ok: true,
     status: 200,
     statusText: "OK",
-    headers: new Headers({ "content-type": "text/event-stream" }),
+    headers,
     body: {
       getReader: () => ({
         read: vi.fn().mockImplementation(() => {
@@ -42,6 +48,13 @@ function mockSSEResponse(lines: string[]): Response {
   } as unknown as Response;
 }
 
+function mockSSEResponse(
+  lines: string[],
+  contentType: string | null = "text/event-stream",
+): Response {
+  return mockSSETextResponse(lines.join("\n"), contentType);
+}
+
 const userMessage: A2AMessage = {
   messageId: "msg-1",
   role: "user",
@@ -799,6 +812,111 @@ describe("A2AClient", () => {
       expect(evt.event.status.message?.role).toBe("agent");
     });
 
+    it("parses CRLF-delimited SSE events", async () => {
+      const sseData = JSON.stringify({
+        status_update: {
+          task_id: "t1",
+          context_id: "ctx-1",
+          status: { state: "TASK_STATE_WORKING" },
+        },
+      });
+
+      fetchMock.mockResolvedValue(
+        mockSSETextResponse(`data: ${sseData}\r\n\r\n`),
+      );
+
+      const events: A2AStreamEvent[] = [];
+      for await (const event of client.streamMessage(userMessage)) {
+        events.push(event);
+      }
+
+      expect(events).toHaveLength(1);
+      expect(events[0]!.type).toBe("statusUpdate");
+    });
+
+    it("parses CR-delimited SSE events", async () => {
+      const sseData = JSON.stringify({
+        status_update: {
+          task_id: "t1",
+          context_id: "ctx-1",
+          status: { state: "TASK_STATE_WORKING" },
+        },
+      });
+
+      fetchMock.mockResolvedValue(mockSSETextResponse(`data: ${sseData}\r\r`));
+
+      const events: A2AStreamEvent[] = [];
+      for await (const event of client.streamMessage(userMessage)) {
+        events.push(event);
+      }
+
+      expect(events).toHaveLength(1);
+      expect(events[0]!.type).toBe("statusUpdate");
+    });
+
+    it("parses CRLF delimiters split across chunks", async () => {
+      const sseData = JSON.stringify({
+        status_update: {
+          task_id: "t1",
+          context_id: "ctx-1",
+          status: { state: "TASK_STATE_WORKING" },
+        },
+      });
+
+      fetchMock.mockResolvedValue(
+        mockSSETextResponse([`data: ${sseData}\r`, "\n\r", "\n"]),
+      );
+
+      const events: A2AStreamEvent[] = [];
+      for await (const event of client.streamMessage(userMessage)) {
+        events.push(event);
+      }
+
+      expect(events).toHaveLength(1);
+      expect(events[0]!.type).toBe("statusUpdate");
+    });
+
+    it("parses CR-delimited events split across chunks", async () => {
+      const sseData = JSON.stringify({
+        status_update: {
+          task_id: "t1",
+          context_id: "ctx-1",
+          status: { state: "TASK_STATE_WORKING" },
+        },
+      });
+
+      fetchMock.mockResolvedValue(
+        mockSSETextResponse([`data: ${sseData}\r`, "\r"]),
+      );
+
+      const events: A2AStreamEvent[] = [];
+      for await (const event of client.streamMessage(userMessage)) {
+        events.push(event);
+      }
+
+      expect(events).toHaveLength(1);
+      expect(events[0]!.type).toBe("statusUpdate");
+    });
+
+    it("does not dispatch unterminated SSE events", async () => {
+      const sseData = JSON.stringify({
+        status_update: {
+          task_id: "t1",
+          context_id: "ctx-1",
+          status: { state: "TASK_STATE_WORKING" },
+        },
+      });
+
+      fetchMock.mockResolvedValue(mockSSETextResponse(`data: ${sseData}`));
+
+      const events: A2AStreamEvent[] = [];
+      for await (const event of client.streamMessage(userMessage)) {
+        events.push(event);
+      }
+
+      expect(events).toHaveLength(0);
+    });
+
     it("parses SSE artifact update events", async () => {
       const sseData = JSON.stringify({
         artifact_update: {
@@ -887,6 +1005,63 @@ describe("A2AClient", () => {
       expect(events).toHaveLength(2);
     });
 
+    it("accepts parameterized event-stream content types", async () => {
+      const sseData = JSON.stringify({
+        status_update: {
+          task_id: "t1",
+          context_id: "ctx-1",
+          status: { state: "TASK_STATE_COMPLETED" },
+        },
+      });
+
+      fetchMock.mockResolvedValue(
+        mockSSEResponse(
+          [`data: ${sseData}`, "", ""],
+          "text/event-stream; charset=utf-8",
+        ),
+      );
+
+      const events: A2AStreamEvent[] = [];
+      for await (const event of client.streamMessage(userMessage)) {
+        events.push(event);
+      }
+
+      expect(events).toHaveLength(1);
+    });
+
+    it("rejects successful responses that are not event streams", async () => {
+      fetchMock.mockResolvedValue(
+        mockSSETextResponse(
+          "<html><body>Please sign in</body></html>",
+          "text/html; charset=utf-8",
+        ),
+      );
+
+      const consumeStream = async () => {
+        for await (const event of client.streamMessage(userMessage)) {
+          void event;
+        }
+      };
+
+      await expect(consumeStream()).rejects.toThrow(
+        'Expected A2A stream response Content-Type "text/event-stream", received "text/html; charset=utf-8"',
+      );
+    });
+
+    it("rejects task subscriptions without a content type", async () => {
+      fetchMock.mockResolvedValue(mockSSETextResponse("", null));
+
+      const consumeStream = async () => {
+        for await (const event of client.subscribeToTask("t1")) {
+          void event;
+        }
+      };
+
+      await expect(consumeStream()).rejects.toThrow(
+        'Expected A2A stream response Content-Type "text/event-stream", received no Content-Type header',
+      );
+    });
+
     it("normalizes 'content' array from v0.3 server response to 'parts' in SSE artifact update events", async () => {
       const sseData = JSON.stringify({
         artifact_update: {

package/src/A2AClient.ts

@@ -1,3 +1,4 @@
+import { SSEEventDecoder, type SSEEvent } from "assistant-stream/utils";
 import type {
   A2AAgentCard,
   A2AErrorInfo,
@@ -493,57 +494,59 @@ export class A2AClient {
   // --- SSE Parsing ---
 
   private async *parseSSE(response: Response): AsyncGenerator<A2AStreamEvent> {
+    const contentType = response.headers.get("Content-Type");
+    const mediaType = contentType?.split(";", 1)[0]?.trim().toLowerCase();
+    if (mediaType !== "text/event-stream") {
+      const received = contentType
+        ? `"${contentType}"`
+        : "no Content-Type header";
+      throw new Error(
+        `Expected A2A stream response Content-Type "text/event-stream", received ${received}`,
+      );
+    }
+
     const reader = response.body?.getReader();
     if (!reader) throw new Error("No response body");
 
     const decoder = new TextDecoder();
-    let buffer = "";
+    const sseDecoder = new SSEEventDecoder();
+
+    const readEvent = (event: SSEEvent): A2AStreamEvent | null => {
+      try {
+        let parsed = JSON.parse(event.data);
+
+        if (
+          parsed &&
+          typeof parsed === "object" &&
+          "jsonrpc" in parsed &&
+          "result" in parsed
+        ) {
+          parsed = parsed.result;
+        }
+
+        const normalized = normalizeKeys(parsed) as Record<string, unknown>;
+        return discriminateStreamResponse(normalized);
+      } catch {
+        return null;
+      }
+    };
 
     try {
       while (true) {
         const { done, value } = await reader.read();
-        if (done) break;
-
-        buffer += decoder.decode(value, { stream: true });
-
-        let eventEnd: number = buffer.indexOf("\n\n");
-        while (eventEnd !== -1) {
-          const eventText = buffer.slice(0, eventEnd);
-          buffer = buffer.slice(eventEnd + 2);
-
-          const dataLines: string[] = [];
-
-          for (const line of eventText.split("\n")) {
-            const trimmed = line.replace(/\r$/, "");
-            if (trimmed.startsWith("data:")) {
-              dataLines.push(trimmed.slice(5).trim());
-            }
-            // event:, id:, retry: lines are parsed but not used —
-            // we discriminate event type from the JSON payload.
+        if (done) {
+          for (const event of sseDecoder.push(decoder.decode())) {
+            const parsed = readEvent(event);
+            if (parsed) yield parsed;
           }
+          break;
+        }
 
-          if (dataLines.length === 0) continue;
-
-          try {
-            let parsed = JSON.parse(dataLines.join("\n"));
-
-            // Unwrap JSON-RPC envelope if present
-            if (
-              parsed &&
-              typeof parsed === "object" &&
-              "jsonrpc" in parsed &&
-              "result" in parsed
-            ) {
-              parsed = parsed.result;
-            }
-
-            const normalized = normalizeKeys(parsed) as Record<string, unknown>;
-            const event = discriminateStreamResponse(normalized);
-            if (event) yield event;
-          } catch {
-            // Skip malformed events
-          }
-          eventEnd = buffer.indexOf("\n\n");
+        for (const event of sseDecoder.push(
+          decoder.decode(value, { stream: true }),
+        )) {
+          const parsed = readEvent(event);
+          if (parsed) yield parsed;
         }
       }
     } finally {

package/src/a2aExtras.ts

@@ -0,0 +1,4 @@
+import { createRuntimeExtras } from "@assistant-ui/core/internal";
+import type { A2AExtras } from "./types";
+
+export const a2aExtras = createRuntimeExtras<A2AExtras>("useA2ARuntime");

package/src/hooks.ts

@@ -0,0 +1,12 @@
+"use client";
+
+import { a2aExtras } from "./a2aExtras";
+
+/** The current A2A task, if a run is in progress. */
+export const useA2ATask = () => a2aExtras.use((e) => e.task);
+
+/** The artifacts accumulated for the current A2A run. */
+export const useA2AArtifacts = () => a2aExtras.use((e) => e.artifacts);
+
+/** The discovered A2A agent card. */
+export const useA2AAgentCard = () => a2aExtras.use((e) => e.agentCard);

package/src/index.ts

@@ -1,14 +1,7 @@
 // Main hook
-export {
-  useA2ARuntime,
-  useA2ATask,
-  useA2AArtifacts,
-  useA2AAgentCard,
-} from "./useA2ARuntime";
-export type {
-  UseA2ARuntimeOptions,
-  UseA2AThreadListAdapter,
-} from "./useA2ARuntime";
+export { useA2ARuntime } from "./useA2ARuntime";
+export { useA2ATask, useA2AArtifacts, useA2AAgentCard } from "./hooks";
+export type { UseA2ARuntimeOptions, UseA2AThreadListAdapter } from "./types";
 
 // Client
 export { A2AClient, A2AError } from "./A2AClient";

package/src/types.ts

@@ -2,6 +2,18 @@
 // Enum values use lowercase internally; normalized from ProtoJSON SCREAMING_SNAKE_CASE on read.
 // Wire format uses ROLE_USER/ROLE_AGENT and TASK_STATE_* per ADR-001.
 
+import type {
+  AttachmentAdapter,
+  DictationAdapter,
+  ExternalStoreSharedOptions,
+  FeedbackAdapter,
+  RealtimeVoiceAdapter,
+  SpeechSynthesisAdapter,
+  ThreadHistoryAdapter,
+  ThreadMessage,
+} from "@assistant-ui/core";
+import type { A2AClient, A2AClientOptions } from "./A2AClient";
+
 export const A2A_PROTOCOL_VERSION = "1.0";
 
 export type A2ARole = "unspecified" | "user" | "agent";
@@ -281,3 +293,57 @@ export type A2AAgentCard = {
   signatures?: A2AAgentCardSignature[] | undefined;
   iconUrl?: string | undefined;
 };
+
+/** Private state `useA2ARuntime` exposes through `thread.extras`. */
+export type A2AExtras = {
+  task: A2ATask | undefined;
+  artifacts: readonly A2AArtifact[];
+  agentCard: A2AAgentCard | undefined;
+};
+
+export type UseA2AThreadListAdapter = {
+  threadId?: string;
+  onSwitchToNewThread?: () => Promise<void> | void;
+  onSwitchToThread?: (threadId: string) => Promise<{
+    messages: readonly ThreadMessage[];
+  }>;
+};
+
+export type UseA2ARuntimeOptions = ExternalStoreSharedOptions & {
+  /** Pre-built A2A client instance. Provide this OR baseUrl. */
+  client?: A2AClient;
+  /** Base URL of the A2A server. Used to create a client if `client` is not provided. */
+  baseUrl?: string;
+  /** Optional path prefix for all API endpoints (e.g. "/v1"). Does not affect agent card discovery. Only used with baseUrl. */
+  basePath?: string;
+  /** Optional tenant ID for multi-tenant servers. Only used with baseUrl. */
+  tenant?: string;
+  /** Headers for the A2A client (only used with baseUrl). */
+  headers?: A2AClientOptions["headers"];
+  /** A2A extension URIs to negotiate. Only used with baseUrl. */
+  extensions?: string[];
+  /** Extra fetch options (e.g. `{ credentials: 'include' }`). Only used with `baseUrl`. */
+  fetchOptions?: A2AClientOptions["fetchOptions"];
+
+  /** Initial context ID for the conversation. */
+  contextId?: string;
+  /** Default send message configuration. */
+  configuration?: A2ASendMessageConfiguration;
+
+  /** Called when an error occurs. */
+  onError?: (error: Error) => void;
+  /** Called when a run is cancelled. */
+  onCancel?: () => void;
+  /** Called when an artifact is fully received (lastChunk). */
+  onArtifactComplete?: (artifact: A2AArtifact) => void;
+
+  adapters?: {
+    attachments?: AttachmentAdapter;
+    speech?: SpeechSynthesisAdapter;
+    dictation?: DictationAdapter;
+    voice?: RealtimeVoiceAdapter;
+    feedback?: FeedbackAdapter;
+    history?: ThreadHistoryAdapter;
+    threadList?: UseA2AThreadListAdapter;
+  };
+};