@assistant-ui/react-a2a 0.2.15 → 0.2.17

package/src/A2AClient.ts

@@ -26,6 +26,10 @@ export type A2AClientOptions = {
     | undefined;
   /** A2A extension URIs to negotiate. Sent as A2A-Extensions header. */
   extensions?: string[] | undefined;
+  /** Extra fetch options applied to every request. */
+  fetchOptions?:
+    | Omit<RequestInit, "headers" | "body" | "method" | "signal">
+    | undefined;
 };
 
 export class A2AError extends Error {
@@ -87,10 +91,10 @@ function normalizeKeys(obj: unknown, opaque = false): unknown {
       ) {
         result[camelKey] = value.slice(5).toLowerCase();
       } else if (camelKey === "content" && Array.isArray(value)) {
-        // v1.0 proto uses "content" for message/artifact parts; map to internal "parts"
+        // v0.3 servers used "content" for message/artifact parts; normalize to "parts" for backward compat
         result.parts = normalizeKeys(value, false);
       } else if (camelKey !== "parts" || !("parts" in result)) {
-        // skip "parts" if "content" already mapped it (prefer content over parts)
+        // dedup: "content" was already mapped to parts above; don't overwrite
         result[camelKey] = isOpaqueChild ? value : normalizeKeys(value, false);
       }
     }
@@ -111,8 +115,7 @@ function toWireTaskState(state: A2ATaskState): string {
 }
 
 function toWireMessage(msg: A2AMessage): unknown {
-  const { parts, ...rest } = msg;
-  return { ...rest, role: toWireRole(msg.role), content: parts };
+  return { ...msg, role: toWireRole(msg.role) };
 }
 
 function discriminateStreamResponse(
@@ -158,6 +161,10 @@ export class A2AClient {
   private basePath: string;
   private tenant: string | undefined;
   private extensionUris: string[] | undefined;
+  private fetchOptions: Omit<
+    RequestInit,
+    "headers" | "body" | "method" | "signal"
+  >;
   private headersFn:
     | Record<string, string>
     | (() => Record<string, string> | Promise<Record<string, string>>);
@@ -169,6 +176,14 @@ export class A2AClient {
       : "";
     this.tenant = options.tenant;
     this.extensionUris = options.extensions;
+    const {
+      headers: _h,
+      body: _b,
+      method: _m,
+      signal: _s,
+      ...safeFetchOptions
+    } = (options.fetchOptions ?? {}) as RequestInit;
+    this.fetchOptions = safeFetchOptions;
     this.headersFn = options.headers ?? {};
   }
 
@@ -229,6 +244,7 @@ export class A2AClient {
     const isGet = !options.method || options.method.toUpperCase() === "GET";
     const headers = await this.getHeaders(!isGet);
     const response = await fetch(`${this.baseUrl}${path}`, {
+      ...this.fetchOptions,
       ...options,
       headers: {
         ...headers,
@@ -247,7 +263,11 @@ export class A2AClient {
   async getAgentCard(signal?: AbortSignal): Promise<A2AAgentCard> {
     const headers = await this.getHeaders(false); // GET: no Content-Type
     const url = `${this.baseUrl}/.well-known/agent-card.json`;
-    const response = await fetch(url, { headers, ...signalInit(signal) });
+    const response = await fetch(url, {
+      ...this.fetchOptions,
+      headers,
+      ...signalInit(signal),
+    });
     if (!response.ok) {
       await this.throwResponseError(response);
     }
@@ -310,6 +330,7 @@ export class A2AClient {
     const response = await fetch(
       `${this.baseUrl}${this.getBasePath()}/message:stream`,
       {
+        ...this.fetchOptions,
         method: "POST",
         headers,
         body: JSON.stringify(body),
@@ -390,7 +411,11 @@ export class A2AClient {
 
     const response = await fetch(
       `${this.baseUrl}${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}:subscribe`,
-      { headers, ...signalInit(signal) },
+      {
+        ...this.fetchOptions,
+        headers,
+        ...signalInit(signal),
+      },
     );
     if (!response.ok) {
       await this.throwResponseError(response);
@@ -453,7 +478,12 @@ export class A2AClient {
     const headers = await this.getHeaders(!isGet);
     const response = await fetch(
       `${this.baseUrl}${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}/pushNotificationConfigs/${encodeURIComponent(configId)}`,
-      { method: "DELETE", headers, ...signalInit(signal) },
+      {
+        ...this.fetchOptions,
+        method: "DELETE",
+        headers,
+        ...signalInit(signal),
+      },
     );
     if (!response.ok) {
       await this.throwResponseError(response);

package/src/A2AThreadRuntimeCore.test.ts

@@ -37,7 +37,7 @@ function createUserAppendMessage(text: string): AppendMessage {
     parentId: null,
     role: "user",
     content: [{ type: "text", text }],
-  };
+  } as unknown as AppendMessage;
 }
 
 function statusUpdateEvent(state: string, text?: string): A2AStreamEvent {
@@ -90,7 +90,7 @@ describe("A2AThreadRuntimeCore", () => {
   ) {
     return new A2AThreadRuntimeCore({
       client: createMockClient(clientOverrides),
-      notifyUpdate,
+      notifyUpdate: notifyUpdate as unknown as () => void,
       ...coreOverrides,
     });
   }
@@ -369,7 +369,7 @@ describe("A2AThreadRuntimeCore", () => {
           }),
         }),
         onArtifactComplete,
-        notifyUpdate,
+        notifyUpdate: notifyUpdate as unknown as () => void,
       });
 
       await core.append(createUserAppendMessage("Go"));
@@ -588,7 +588,7 @@ describe("A2AThreadRuntimeCore", () => {
           })),
         }),
         onError,
-        notifyUpdate,
+        notifyUpdate: notifyUpdate as unknown as () => void,
       });
 
       await expect(core.append(createUserAppendMessage("Go"))).rejects.toThrow(
@@ -680,7 +680,7 @@ describe("A2AThreadRuntimeCore", () => {
           createdAt: new Date(),
           content: [{ type: "text", text: "External" }],
           status: { type: "complete", reason: "stop" },
-        } as ThreadMessage,
+        } as unknown as ThreadMessage,
       ];
 
       core.applyExternalMessages(msgs);

package/src/conversions.test.ts

@@ -200,14 +200,12 @@ describe("taskStateToMessageStatus", () => {
 });
 
 describe("isTerminalTaskState", () => {
-  it.each([
-    "completed",
-    "failed",
-    "canceled",
-    "rejected",
-  ] as A2ATaskState[])("returns true for %s", (state) => {
-    expect(isTerminalTaskState(state)).toBe(true);
-  });
+  it.each(["completed", "failed", "canceled", "rejected"] as A2ATaskState[])(
+    "returns true for %s",
+    (state) => {
+      expect(isTerminalTaskState(state)).toBe(true);
+    },
+  );
 
   it.each([
     "submitted",
@@ -221,12 +219,12 @@ describe("isTerminalTaskState", () => {
 });
 
 describe("isInterruptedTaskState", () => {
-  it.each([
-    "input_required",
-    "auth_required",
-  ] as A2ATaskState[])("returns true for %s", (state) => {
-    expect(isInterruptedTaskState(state)).toBe(true);
-  });
+  it.each(["input_required", "auth_required"] as A2ATaskState[])(
+    "returns true for %s",
+    (state) => {
+      expect(isInterruptedTaskState(state)).toBe(true);
+    },
+  );
 
   it.each([
     "submitted",

package/src/useA2ARuntime.ts

@@ -1,17 +1,20 @@
-/// <reference types="@assistant-ui/core/store" />
 "use client";
 
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import {
   useExternalStoreRuntime,
+  useExternalStoreSharedOptions,
   useRuntimeAdapters,
 } from "@assistant-ui/core/react";
 import type {
   AssistantRuntime,
   AppendMessage,
   AttachmentAdapter,
+  DictationAdapter,
   ExternalStoreAdapter,
+  ExternalStoreSharedOptions,
   FeedbackAdapter,
+  RealtimeVoiceAdapter,
   SpeechSynthesisAdapter,
   ThreadHistoryAdapter,
   ThreadMessage,
@@ -76,7 +79,7 @@ export type UseA2AThreadListAdapter = {
 
 // --- Options ---
 
-export type UseA2ARuntimeOptions = {
+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. */
@@ -89,6 +92,8 @@ export type UseA2ARuntimeOptions = {
   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;
@@ -105,6 +110,8 @@ export type UseA2ARuntimeOptions = {
   adapters?: {
     attachments?: AttachmentAdapter;
     speech?: SpeechSynthesisAdapter;
+    dictation?: DictationAdapter;
+    voice?: RealtimeVoiceAdapter;
     feedback?: FeedbackAdapter;
     history?: ThreadHistoryAdapter;
     threadList?: UseA2AThreadListAdapter;
@@ -132,6 +139,7 @@ export function useA2ARuntime(options: UseA2ARuntimeOptions): AssistantRuntime {
         tenant: options.tenant,
         headers: options.headers,
         extensions: options.extensions,
+        fetchOptions: options.fetchOptions,
       });
     } else {
       throw new Error("useA2ARuntime requires either `client` or `baseUrl`");
@@ -198,6 +206,8 @@ export function useA2ARuntime(options: UseA2ARuntimeOptions): AssistantRuntime {
     () => ({
       attachments: adapters?.attachments ?? runtimeAdapters?.attachments,
       speech: adapters?.speech,
+      dictation: adapters?.dictation,
+      voice: adapters?.voice,
       feedback: adapters?.feedback,
       threadList,
     }),
@@ -205,10 +215,12 @@ export function useA2ARuntime(options: UseA2ARuntimeOptions): AssistantRuntime {
   );
 
   // Build store adapter
+  const shared = useExternalStoreSharedOptions(options);
   const store = useMemo(() => {
     void _version;
 
     return {
+      ...shared,
       isLoading: core.isLoading,
       messages: core.getMessages(),
       isRunning: core.isRunning(),
@@ -228,7 +240,7 @@ export function useA2ARuntime(options: UseA2ARuntimeOptions): AssistantRuntime {
         core.applyExternalMessages(messages),
       adapters: adapterAdapters,
     } satisfies ExternalStoreAdapter<ThreadMessage>;
-  }, [adapterAdapters, core, _version]);
+  }, [adapterAdapters, core, _version, shared]);
 
   const runtime = useExternalStoreRuntime(store);
 

package/dist/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,aAAa,EACb,UAAU,EACV,eAAe,EACf,eAAe,GAChB,2BAAwB;AACzB,YAAY,EACV,oBAAoB,EACpB,uBAAuB,GACxB,2BAAwB;AAGzB,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,uBAAoB;AAClD,YAAY,EAAE,gBAAgB,EAAE,uBAAoB;AAGpD,YAAY,EACV,OAAO,EACP,OAAO,EACP,UAAU,EACV,YAAY,EACZ,aAAa,EACb,WAAW,EACX,OAAO,EACP,cAAc,EACd,wBAAwB,EACxB,0BAA0B,EAC1B,qBAAqB,EACrB,6BAA6B,EAC7B,0CAA0C,EAC1C,2BAA2B,EAC3B,mBAAmB,EACnB,oBAAoB,EACpB,YAAY,EACZ,uBAAuB,EACvB,yBAAyB,EACzB,6BAA6B,EAC7B,6BAA6B,EAC7B,sBAAsB,EACtB,oBAAoB,EACpB,oBAAoB,EACpB,aAAa,EACb,uBAAuB,EACvB,8BAA8B,EAC9B,0BAA0B,EAC1B,iBAAiB,EACjB,sBAAsB,EACtB,qBAAqB,EACrB,aAAa,EACb,oBAAoB,EACpB,iBAAiB,EACjB,YAAY,GACb,mBAAgB;AACjB,OAAO,EAAE,oBAAoB,EAAE,mBAAgB;AAG/C,OAAO,EACL,gBAAgB,EAChB,iBAAiB,EACjB,mBAAmB,EACnB,wBAAwB,EACxB,sBAAsB,EACtB,mBAAmB,EACnB,sBAAsB,GACvB,yBAAsB"}

package/dist/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,YAAY;AACZ,OAAO,EACL,aAAa,EACb,UAAU,EACV,eAAe,EACf,eAAe,GAChB,2BAAwB;AAMzB,SAAS;AACT,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,uBAAoB;AAyClD,OAAO,EAAE,oBAAoB,EAAE,mBAAgB;AAE/C,4CAA4C;AAC5C,OAAO,EACL,gBAAgB,EAChB,iBAAiB,EACjB,mBAAmB,EACnB,wBAAwB,EACxB,sBAAsB,EACtB,mBAAmB,EACnB,sBAAsB,GACvB,yBAAsB"}