@assistant-ui/react-a2a 0.2.22 → 0.2.23

package/dist/A2AClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"A2AClient.d.ts","names":[],"sources":["../src/A2AClient.ts"],"mappings":";;KAiBY;EACV;;EAEA;;EAEA;EACA,UACI,gCACO,yBAAyB,QAAQ;;EAG5C;;EAEA,eACI,KAAK;;cAIE,iBAAiB;EAC5B;EACA;EACA;cAEY,MAAM;;cAuHP;UACH;UACA;UACA;UACA;UACA;UAIA;cAII,SAAS;UAkBb;UAIM;UAqBA;UAyBA;EAuBR,aAAa,SAAS,cAAc,QAAQ;EAe5C,qBAAqB,SAAS,cAAc,QAAQ;EASpD,YACJ,SAAS,YACT,gBAAgB,6BAChB,WAAW,yBACX,SAAS,cACR,QAAQ,UAAU;EAuBd,cACL,SAAS,YACT,gBAAgB,6BAChB,WAAW,yBACX,SAAS,cACR,eAAe;EA6BZ,QACJ,gBACA,wBACA,SAAS,cACR,QAAQ;EAaL,UACJ,UAAU,qBACV,SAAS,cACR,QAAQ;EAoBL,WACJ,gBACA,WAAW,yBACX,SAAS,cACR,QAAQ;EAYJ,gBACL,gBACA,SAAS,cACR,eAAe;EAqBZ,iCACJ,QAAQ,+BACR,SAAS,cACR,QAAQ;EAaL,8BACJ,gBACA,kBACA,SAAS,cACR,QAAQ;EAOL,gCACJ,gBACA;IAAY;IAAmB;KAC/B,SAAS,cACR,QAAQ;EAYL,iCACJ,gBACA,kBACA,SAAS,cACR;UAmBY"}
+{"version":3,"file":"A2AClient.d.ts","names":[],"sources":["../src/A2AClient.ts"],"mappings":";;KAkBY;EACV;;EAEA;;EAEA;EACA,UACI,gCACO,yBAAyB,QAAQ;;EAG5C;;EAEA,eACI,KAAK;;cAIE,iBAAiB;EAC5B;EACA;EACA;EAEY,YAAA,MAAM;;cAmOP;UACH;UACA;UACA;UACA;UACA;UAIA;EAII,YAAA,SAAS;UAkBb;UAIM;UAqBA;UAyBA;EAuBR,aAAa,SAAS,cAAc,QAAQ;EAe5C,qBAAqB,SAAS,cAAc,QAAQ;EAUpD,YACJ,SAAS,YACT,gBAAgB,6BAChB,WAAW,yBACX,SAAS,cACR,QAAQ,UAAU;EAmBd,cACL,SAAS,YACT,gBAAgB,6BAChB,WAAW,yBACX,SAAS,cACR,eAAe;EA6BZ,QACJ,gBACA,wBACA,SAAS,cACR,QAAQ;EAaL,UACJ,UAAU,qBACV,SAAS,cACR,QAAQ;EAoBL,WACJ,gBACA,WAAW,yBACX,SAAS,cACR,QAAQ;EAYJ,gBACL,gBACA,SAAS,cACR,eAAe;EAqBZ,iCACJ,QAAQ,+BACR,SAAS,cACR,QAAQ;EAaL,8BACJ,gBACA,kBACA,SAAS,cACR,QAAQ;EAOL,gCACJ,gBACA;IAAY;IAAmB;KAC/B,SAAS,cACR,QAAQ;EAYL,iCACJ,gBACA,kBACA,SAAS,cACR;UAmBY"}

package/dist/A2AClient.js

@@ -1,5 +1,6 @@
 import "./types.js";
 import { SSEEventDecoder } from "assistant-stream/utils";
+import { isRecord } from "@assistant-ui/core/internal";
 //#region src/A2AClient.ts
 var A2AError = class extends Error {
 	code;
@@ -76,6 +77,60 @@ function discriminateStreamResponse(data) {
 	};
 	return null;
 }
+const TASK_STATES = new Set(Object.keys({
+	unspecified: true,
+	submitted: true,
+	working: true,
+	completed: true,
+	failed: true,
+	canceled: true,
+	input_required: true,
+	rejected: true,
+	auth_required: true
+}));
+const isTaskState = (value) => typeof value === "string" && TASK_STATES.has(value);
+const isTask = (value) => isRecord(value) && typeof value.id === "string" && value.id.length > 0 && isRecord(value.status) && isTaskState(value.status.state);
+const isMessage = (value) => isRecord(value) && typeof value.messageId === "string" && value.messageId.length > 0 && typeof value.role === "string" && value.role.length > 0 && Array.isArray(value.parts) && value.parts.every(isRecord);
+const isStringArray = (value) => Array.isArray(value) && value.every((item) => typeof item === "string");
+const invalidAgentCard = () => {
+	throw new Error("Invalid A2A agent card response: expected a valid agent card payload.");
+};
+const parseCardString = (value) => value == null ? "" : typeof value === "string" ? value : invalidAgentCard();
+const parseCardStringArray = (value) => value == null ? [] : isStringArray(value) ? value : invalidAgentCard();
+const parseCardRecordArray = (value) => value == null ? [] : Array.isArray(value) && value.every(isRecord) ? value : invalidAgentCard();
+const parseCardRecord = (value) => value == null ? {} : isRecord(value) ? value : invalidAgentCard();
+const parseAgentCardResponse = (value) => {
+	if (!isRecord(value) || typeof value.name !== "string" || value.name.length === 0) return invalidAgentCard();
+	return {
+		...value,
+		name: value.name,
+		description: parseCardString(value.description),
+		version: parseCardString(value.version),
+		supportedInterfaces: parseCardRecordArray(value.supportedInterfaces).map((entry) => ({
+			...entry,
+			url: parseCardString(entry.url),
+			protocolBinding: parseCardString(entry.protocolBinding),
+			protocolVersion: parseCardString(entry.protocolVersion)
+		})),
+		capabilities: parseCardRecord(value.capabilities),
+		defaultInputModes: parseCardStringArray(value.defaultInputModes),
+		defaultOutputModes: parseCardStringArray(value.defaultOutputModes),
+		skills: parseCardRecordArray(value.skills).map((entry) => ({
+			...entry,
+			id: parseCardString(entry.id),
+			name: parseCardString(entry.name),
+			description: parseCardString(entry.description),
+			tags: parseCardStringArray(entry.tags)
+		}))
+	};
+};
+const parseSendMessageResponse = (value) => {
+	if (isRecord(value)) {
+		const candidate = value.task ?? value.message ?? value;
+		if (isTask(candidate) || isMessage(candidate)) return candidate;
+	}
+	throw new Error("Invalid A2A message:send response: expected a valid task or message payload.");
+};
 function signalInit(signal) {
 	return signal ? { signal } : {};
 }
@@ -151,10 +206,12 @@ var A2AClient = class {
 			...signalInit(signal)
 		});
 		if (!response.ok) await this.throwResponseError(response);
-		return normalizeKeys(await response.json());
+		const json = await response.json();
+		return parseAgentCardResponse(normalizeKeys(json));
 	}
 	async getExtendedAgentCard(signal) {
-		return this.fetchJSON(`${this.getBasePath()}/extendedAgentCard`, signalInit(signal));
+		const result = await this.fetchJSON(`${this.getBasePath()}/extendedAgentCard`, signalInit(signal));
+		return parseAgentCardResponse(result);
 	}
 	async sendMessage(message, configuration, metadata, signal) {
 		const body = { message: toWireMessage(message) };
@@ -165,9 +222,7 @@ var A2AClient = class {
 			body: JSON.stringify(body),
 			...signalInit(signal)
 		});
-		if ("task" in result && result.task) return result.task;
-		if ("message" in result && result.message) return result.message;
-		return result;
+		return parseSendMessageResponse(result);
 	}
 	async *streamMessage(message, configuration, metadata, signal) {
 		const headers = await this.getHeaders(true);

package/dist/A2AClient.js.map

@@ -1 +1 @@
-{"version":3,"file":"A2AClient.js","names":[],"sources":["../src/A2AClient.ts"],"sourcesContent":["import { SSEEventDecoder, type SSEEvent } from \"assistant-stream/utils\";\nimport type {\n  A2AAgentCard,\n  A2AErrorInfo,\n  A2AListTaskPushNotificationConfigsResponse,\n  A2AListTasksRequest,\n  A2AListTasksResponse,\n  A2AMessage,\n  A2ARole,\n  A2ASendMessageConfiguration,\n  A2AStreamEvent,\n  A2ATask,\n  A2ATaskPushNotificationConfig,\n  A2ATaskState,\n} from \"./types\";\nimport { A2A_PROTOCOL_VERSION } from \"./types\";\n\nexport type A2AClientOptions = {\n  baseUrl: string;\n  /** Optional path prefix for all API endpoints (e.g. \"/v1\"). Does not affect agent card discovery. */\n  basePath?: string | undefined;\n  /** Optional tenant ID for multi-tenant servers. */\n  tenant?: string | undefined;\n  headers?:\n    | Record<string, string>\n    | (() => Record<string, string> | Promise<Record<string, string>>)\n    | undefined;\n  /** A2A extension URIs to negotiate. Sent as A2A-Extensions header. */\n  extensions?: string[] | undefined;\n  /** Extra fetch options applied to every request. */\n  fetchOptions?:\n    | Omit<RequestInit, \"headers\" | \"body\" | \"method\" | \"signal\">\n    | undefined;\n};\n\nexport class A2AError extends Error {\n  code: number;\n  status: string;\n  details: unknown[] | undefined;\n\n  constructor(info: A2AErrorInfo) {\n    super(info.message);\n    this.name = \"A2AError\";\n    this.code = info.code;\n    this.status = info.status;\n    this.details = info.details;\n  }\n}\n\n// Incoming key normalization: snake_case → camelCase, plus ProtoJSON enum normalization.\nfunction toCamelCase(key: string): string {\n  return key.replace(/_([a-z])/g, (_, c: string) => c.toUpperCase());\n}\n\n// Fields whose values are opaque user data (google.protobuf.Struct / Value).\n// Keys inside these objects must NOT be camelCased or have enum normalization applied.\nconst OPAQUE_FIELDS = new Set([\n  \"metadata\",\n  \"data\",\n  \"params\",\n  \"forwardedProps\",\n  \"scopes\",\n]);\n\nfunction normalizeKeys(obj: unknown, opaque = false): unknown {\n  if (Array.isArray(obj)) return obj.map((v) => normalizeKeys(v, opaque));\n  if (obj !== null && typeof obj === \"object\") {\n    const result: Record<string, unknown> = {};\n    for (const [key, value] of Object.entries(obj as Record<string, unknown>)) {\n      // Inside opaque fields: preserve keys and values as-is (only recurse arrays/objects structurally)\n      if (opaque) {\n        result[key] =\n          typeof value === \"object\" && value !== null\n            ? normalizeKeys(value, true)\n            : value;\n        continue;\n      }\n\n      const camelKey = toCamelCase(key);\n      const isOpaqueChild = OPAQUE_FIELDS.has(camelKey);\n\n      if (\n        camelKey === \"state\" &&\n        typeof value === \"string\" &&\n        value.startsWith(\"TASK_STATE_\")\n      ) {\n        result[camelKey] = value.slice(11).toLowerCase();\n      } else if (\n        camelKey === \"role\" &&\n        typeof value === \"string\" &&\n        value.startsWith(\"ROLE_\")\n      ) {\n        result[camelKey] = value.slice(5).toLowerCase();\n      } else if (camelKey === \"content\" && Array.isArray(value)) {\n        // v0.3 servers used \"content\" for message/artifact parts; normalize to \"parts\" for backward compat\n        result.parts = normalizeKeys(value, false);\n      } else if (camelKey !== \"parts\" || !(\"parts\" in result)) {\n        // dedup: \"content\" was already mapped to parts above; don't overwrite\n        result[camelKey] = isOpaqueChild ? value : normalizeKeys(value, false);\n      }\n    }\n    return result;\n  }\n  return obj;\n}\n\n// Outgoing enum conversion (v1.0 ProtoJSON format)\nfunction toWireRole(role: A2ARole): string {\n  if (role === \"user\") return \"ROLE_USER\";\n  if (role === \"agent\") return \"ROLE_AGENT\";\n  return \"ROLE_UNSPECIFIED\";\n}\n\nfunction toWireTaskState(state: A2ATaskState): string {\n  return `TASK_STATE_${state.toUpperCase()}`;\n}\n\nfunction toWireMessage(msg: A2AMessage): unknown {\n  return { ...msg, role: toWireRole(msg.role) };\n}\n\nfunction discriminateStreamResponse(\n  data: Record<string, unknown>,\n): A2AStreamEvent | null {\n  if (\"task\" in data && data.task) {\n    return { type: \"task\", task: data.task as A2ATask };\n  }\n  if (\"message\" in data && data.message) {\n    return { type: \"message\", message: data.message as A2AMessage };\n  }\n  if (\"statusUpdate\" in data && data.statusUpdate) {\n    return {\n      type: \"statusUpdate\",\n      event: data.statusUpdate as A2AStreamEvent extends {\n        type: \"statusUpdate\";\n        event: infer E;\n      }\n        ? E\n        : never,\n    };\n  }\n  if (\"artifactUpdate\" in data && data.artifactUpdate) {\n    return {\n      type: \"artifactUpdate\",\n      event: data.artifactUpdate as A2AStreamEvent extends {\n        type: \"artifactUpdate\";\n        event: infer E;\n      }\n        ? E\n        : never,\n    };\n  }\n  return null;\n}\n\nfunction signalInit(signal?: AbortSignal): RequestInit {\n  return signal ? { signal } : {};\n}\n\nexport class A2AClient {\n  private baseUrl: string;\n  private basePath: string;\n  private tenant: string | undefined;\n  private extensionUris: string[] | undefined;\n  private fetchOptions: Omit<\n    RequestInit,\n    \"headers\" | \"body\" | \"method\" | \"signal\"\n  >;\n  private headersFn:\n    | Record<string, string>\n    | (() => Record<string, string> | Promise<Record<string, string>>);\n\n  constructor(options: A2AClientOptions) {\n    this.baseUrl = options.baseUrl.replace(/\\/$/, \"\");\n    this.basePath = options.basePath\n      ? `/${options.basePath.replace(/^\\/|\\/$/g, \"\")}`\n      : \"\";\n    this.tenant = options.tenant;\n    this.extensionUris = options.extensions;\n    const {\n      headers: _h,\n      body: _b,\n      method: _m,\n      signal: _s,\n      ...safeFetchOptions\n    } = (options.fetchOptions ?? {}) as RequestInit;\n    this.fetchOptions = safeFetchOptions;\n    this.headersFn = options.headers ?? {};\n  }\n\n  private getBasePath(): string {\n    return `${this.basePath}${this.tenant ? `/${encodeURIComponent(this.tenant)}` : \"\"}`;\n  }\n\n  private async getHeaders(\n    includeContentType = true,\n  ): Promise<Record<string, string>> {\n    const custom =\n      typeof this.headersFn === \"function\"\n        ? await this.headersFn()\n        : this.headersFn;\n    const headers: Record<string, string> = {\n      Accept: \"application/a2a+json, application/json\",\n      \"A2A-Version\": A2A_PROTOCOL_VERSION,\n      ...custom,\n    };\n    if (includeContentType) {\n      headers[\"Content-Type\"] = \"application/a2a+json\";\n    }\n    if (this.extensionUris?.length) {\n      headers[\"A2A-Extensions\"] = this.extensionUris.join(\", \");\n    }\n    return headers;\n  }\n\n  private async throwResponseError(response: Response): Promise<never> {\n    let errorBody: unknown;\n    try {\n      errorBody = await response.json();\n    } catch {\n      // no parseable body\n    }\n\n    if (errorBody && typeof errorBody === \"object\" && \"error\" in errorBody) {\n      const err = (errorBody as Record<string, any>).error;\n      throw new A2AError({\n        code: err.code ?? response.status,\n        status: err.status ?? response.statusText,\n        message: err.message ?? `A2A request failed: ${response.status}`,\n        details: err.details,\n      });\n    }\n\n    throw new A2AError({\n      code: response.status,\n      status: response.statusText,\n      message: `A2A request failed: ${response.status} ${response.statusText}`,\n    });\n  }\n\n  private async fetchJSON<T>(\n    path: string,\n    options: RequestInit = {},\n  ): Promise<T> {\n    const isGet = !options.method || options.method.toUpperCase() === \"GET\";\n    const headers = await this.getHeaders(!isGet);\n    const response = await fetch(`${this.baseUrl}${path}`, {\n      ...this.fetchOptions,\n      ...options,\n      headers: {\n        ...headers,\n        ...(options.headers as Record<string, string>),\n      },\n    });\n    if (!response.ok) {\n      await this.throwResponseError(response);\n    }\n    const json = await response.json();\n    return normalizeKeys(json) as T;\n  }\n\n  // --- Agent Card ---\n\n  async getAgentCard(signal?: AbortSignal): Promise<A2AAgentCard> {\n    const headers = await this.getHeaders(false); // GET: no Content-Type\n    const url = `${this.baseUrl}/.well-known/agent-card.json`;\n    const response = await fetch(url, {\n      ...this.fetchOptions,\n      headers,\n      ...signalInit(signal),\n    });\n    if (!response.ok) {\n      await this.throwResponseError(response);\n    }\n    const json = await response.json();\n    return normalizeKeys(json) as A2AAgentCard;\n  }\n\n  async getExtendedAgentCard(signal?: AbortSignal): Promise<A2AAgentCard> {\n    return this.fetchJSON<A2AAgentCard>(\n      `${this.getBasePath()}/extendedAgentCard`,\n      signalInit(signal),\n    );\n  }\n\n  // --- Message ---\n\n  async sendMessage(\n    message: A2AMessage,\n    configuration?: A2ASendMessageConfiguration,\n    metadata?: Record<string, unknown>,\n    signal?: AbortSignal,\n  ): Promise<A2ATask | A2AMessage> {\n    const body: Record<string, unknown> = {\n      message: toWireMessage(message),\n    };\n    if (configuration) body.configuration = configuration;\n    if (metadata) body.metadata = metadata;\n\n    const result = await this.fetchJSON<Record<string, unknown>>(\n      `${this.getBasePath()}/message:send`,\n      {\n        method: \"POST\",\n        body: JSON.stringify(body),\n        ...signalInit(signal),\n      },\n    );\n\n    // Unwrap SendMessageResponse: {task: Task} | {message: Message}\n    if (\"task\" in result && result.task) return result.task as A2ATask;\n    if (\"message\" in result && result.message)\n      return result.message as A2AMessage;\n    return result as unknown as A2ATask | A2AMessage;\n  }\n\n  async *streamMessage(\n    message: A2AMessage,\n    configuration?: A2ASendMessageConfiguration,\n    metadata?: Record<string, unknown>,\n    signal?: AbortSignal,\n  ): AsyncGenerator<A2AStreamEvent> {\n    const headers = await this.getHeaders(true);\n    headers.Accept = \"text/event-stream\";\n\n    const body: Record<string, unknown> = {\n      message: toWireMessage(message),\n    };\n    if (configuration) body.configuration = configuration;\n    if (metadata) body.metadata = metadata;\n\n    const response = await fetch(\n      `${this.baseUrl}${this.getBasePath()}/message:stream`,\n      {\n        ...this.fetchOptions,\n        method: \"POST\",\n        headers,\n        body: JSON.stringify(body),\n        ...signalInit(signal),\n      },\n    );\n    if (!response.ok) {\n      await this.throwResponseError(response);\n    }\n\n    yield* this.parseSSE(response);\n  }\n\n  // --- Tasks ---\n\n  async getTask(\n    taskId: string,\n    historyLength?: number,\n    signal?: AbortSignal,\n  ): Promise<A2ATask> {\n    const params = new URLSearchParams();\n    if (historyLength !== undefined) {\n      // Proto field name for HTTP transcoding query params\n      params.set(\"history_length\", String(historyLength));\n    }\n    const qs = params.toString();\n    return this.fetchJSON<A2ATask>(\n      `${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}${qs ? `?${qs}` : \"\"}`,\n      signalInit(signal),\n    );\n  }\n\n  async listTasks(\n    request?: A2AListTasksRequest,\n    signal?: AbortSignal,\n  ): Promise<A2AListTasksResponse> {\n    const params = new URLSearchParams();\n    if (request?.contextId) params.set(\"context_id\", request.contextId);\n    if (request?.status) params.set(\"status\", toWireTaskState(request.status));\n    if (request?.pageSize !== undefined)\n      params.set(\"page_size\", String(request.pageSize));\n    if (request?.pageToken) params.set(\"page_token\", request.pageToken);\n    if (request?.historyLength !== undefined)\n      params.set(\"history_length\", String(request.historyLength));\n    if (request?.statusTimestampAfter)\n      params.set(\"status_timestamp_after\", request.statusTimestampAfter);\n    if (request?.includeArtifacts !== undefined)\n      params.set(\"include_artifacts\", String(request.includeArtifacts));\n    const qs = params.toString();\n    return this.fetchJSON<A2AListTasksResponse>(\n      `${this.getBasePath()}/tasks${qs ? `?${qs}` : \"\"}`,\n      signalInit(signal),\n    );\n  }\n\n  async cancelTask(\n    taskId: string,\n    metadata?: Record<string, unknown>,\n    signal?: AbortSignal,\n  ): Promise<A2ATask> {\n    const body = metadata ? { metadata } : {};\n    return this.fetchJSON<A2ATask>(\n      `${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}:cancel`,\n      {\n        method: \"POST\",\n        body: JSON.stringify(body),\n        ...signalInit(signal),\n      },\n    );\n  }\n\n  async *subscribeToTask(\n    taskId: string,\n    signal?: AbortSignal,\n  ): AsyncGenerator<A2AStreamEvent> {\n    const headers = await this.getHeaders(false); // GET: no Content-Type\n    headers.Accept = \"text/event-stream\";\n\n    const response = await fetch(\n      `${this.baseUrl}${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}:subscribe`,\n      {\n        ...this.fetchOptions,\n        headers,\n        ...signalInit(signal),\n      },\n    );\n    if (!response.ok) {\n      await this.throwResponseError(response);\n    }\n\n    yield* this.parseSSE(response);\n  }\n\n  // --- Push Notification Configs ---\n\n  async createTaskPushNotificationConfig(\n    config: A2ATaskPushNotificationConfig,\n    signal?: AbortSignal,\n  ): Promise<A2ATaskPushNotificationConfig> {\n    const taskId = config.taskId;\n    if (!taskId) throw new Error(\"taskId is required\");\n    return this.fetchJSON<A2ATaskPushNotificationConfig>(\n      `${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}/pushNotificationConfigs`,\n      {\n        method: \"POST\",\n        body: JSON.stringify(config),\n        ...signalInit(signal),\n      },\n    );\n  }\n\n  async getTaskPushNotificationConfig(\n    taskId: string,\n    configId: string,\n    signal?: AbortSignal,\n  ): Promise<A2ATaskPushNotificationConfig> {\n    return this.fetchJSON<A2ATaskPushNotificationConfig>(\n      `${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}/pushNotificationConfigs/${encodeURIComponent(configId)}`,\n      signalInit(signal),\n    );\n  }\n\n  async listTaskPushNotificationConfigs(\n    taskId: string,\n    options?: { pageSize?: number; pageToken?: string },\n    signal?: AbortSignal,\n  ): Promise<A2AListTaskPushNotificationConfigsResponse> {\n    const params = new URLSearchParams();\n    if (options?.pageSize !== undefined)\n      params.set(\"page_size\", String(options.pageSize));\n    if (options?.pageToken) params.set(\"page_token\", options.pageToken);\n    const qs = params.toString();\n    return this.fetchJSON<A2AListTaskPushNotificationConfigsResponse>(\n      `${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}/pushNotificationConfigs${qs ? `?${qs}` : \"\"}`,\n      signalInit(signal),\n    );\n  }\n\n  async deleteTaskPushNotificationConfig(\n    taskId: string,\n    configId: string,\n    signal?: AbortSignal,\n  ): Promise<void> {\n    const isGet = false;\n    const headers = await this.getHeaders(!isGet);\n    const response = await fetch(\n      `${this.baseUrl}${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}/pushNotificationConfigs/${encodeURIComponent(configId)}`,\n      {\n        ...this.fetchOptions,\n        method: \"DELETE\",\n        headers,\n        ...signalInit(signal),\n      },\n    );\n    if (!response.ok) {\n      await this.throwResponseError(response);\n    }\n  }\n\n  // --- SSE Parsing ---\n\n  private async *parseSSE(response: Response): AsyncGenerator<A2AStreamEvent> {\n    const contentType = response.headers.get(\"Content-Type\");\n    const mediaType = contentType?.split(\";\", 1)[0]?.trim().toLowerCase();\n    if (mediaType !== \"text/event-stream\") {\n      const received = contentType\n        ? `\"${contentType}\"`\n        : \"no Content-Type header\";\n      throw new Error(\n        `Expected A2A stream response Content-Type \"text/event-stream\", received ${received}`,\n      );\n    }\n\n    const reader = response.body?.getReader();\n    if (!reader) throw new Error(\"No response body\");\n\n    const decoder = new TextDecoder();\n    const sseDecoder = new SSEEventDecoder();\n\n    const readEvent = (event: SSEEvent): A2AStreamEvent | null => {\n      try {\n        let parsed = JSON.parse(event.data);\n\n        if (\n          parsed &&\n          typeof parsed === \"object\" &&\n          \"jsonrpc\" in parsed &&\n          \"result\" in parsed\n        ) {\n          parsed = parsed.result;\n        }\n\n        const normalized = normalizeKeys(parsed) as Record<string, unknown>;\n        return discriminateStreamResponse(normalized);\n      } catch {\n        return null;\n      }\n    };\n\n    try {\n      while (true) {\n        const { done, value } = await reader.read();\n        if (done) {\n          for (const event of sseDecoder.push(decoder.decode())) {\n            const parsed = readEvent(event);\n            if (parsed) yield parsed;\n          }\n          break;\n        }\n\n        for (const event of sseDecoder.push(\n          decoder.decode(value, { stream: true }),\n        )) {\n          const parsed = readEvent(event);\n          if (parsed) yield parsed;\n        }\n      }\n    } finally {\n      reader.releaseLock();\n    }\n  }\n}\n"],"mappings":";;;AAmCA,IAAa,WAAb,cAA8B,MAAM;CAClC;CACA;CACA;CAEA,YAAY,MAAoB;EAC9B,MAAM,KAAK,OAAO;EAClB,KAAK,OAAO;EACZ,KAAK,OAAO,KAAK;EACjB,KAAK,SAAS,KAAK;EACnB,KAAK,UAAU,KAAK;CACtB;AACF;AAGA,SAAS,YAAY,KAAqB;CACxC,OAAO,IAAI,QAAQ,cAAc,GAAG,MAAc,EAAE,YAAY,CAAC;AACnE;AAIA,MAAM,gCAAgB,IAAI,IAAI;CAC5B;CACA;CACA;CACA;CACA;AACF,CAAC;AAED,SAAS,cAAc,KAAc,SAAS,OAAgB;CAC5D,IAAI,MAAM,QAAQ,GAAG,GAAG,OAAO,IAAI,KAAK,MAAM,cAAc,GAAG,MAAM,CAAC;CACtE,IAAI,QAAQ,QAAQ,OAAO,QAAQ,UAAU;EAC3C,MAAM,SAAkC,CAAC;EACzC,KAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,GAA8B,GAAG;GAEzE,IAAI,QAAQ;IACV,OAAO,OACL,OAAO,UAAU,YAAY,UAAU,OACnC,cAAc,OAAO,IAAI,IACzB;IACN;GACF;GAEA,MAAM,WAAW,YAAY,GAAG;GAChC,MAAM,gBAAgB,cAAc,IAAI,QAAQ;GAEhD,IACE,aAAa,WACb,OAAO,UAAU,YACjB,MAAM,WAAW,aAAa,GAE9B,OAAO,YAAY,MAAM,MAAM,EAAE,CAAC,CAAC,YAAY;QAC1C,IACL,aAAa,UACb,OAAO,UAAU,YACjB,MAAM,WAAW,OAAO,GAExB,OAAO,YAAY,MAAM,MAAM,CAAC,CAAC,CAAC,YAAY;QACzC,IAAI,aAAa,aAAa,MAAM,QAAQ,KAAK,GAEtD,OAAO,QAAQ,cAAc,OAAO,KAAK;QACpC,IAAI,aAAa,WAAW,EAAE,WAAW,SAE9C,OAAO,YAAY,gBAAgB,QAAQ,cAAc,OAAO,KAAK;EAEzE;EACA,OAAO;CACT;CACA,OAAO;AACT;AAGA,SAAS,WAAW,MAAuB;CACzC,IAAI,SAAS,QAAQ,OAAO;CAC5B,IAAI,SAAS,SAAS,OAAO;CAC7B,OAAO;AACT;AAEA,SAAS,gBAAgB,OAA6B;CACpD,OAAO,cAAc,MAAM,YAAY;AACzC;AAEA,SAAS,cAAc,KAA0B;CAC/C,OAAO;EAAE,GAAG;EAAK,MAAM,WAAW,IAAI,IAAI;CAAE;AAC9C;AAEA,SAAS,2BACP,MACuB;CACvB,IAAI,UAAU,QAAQ,KAAK,MACzB,OAAO;EAAE,MAAM;EAAQ,MAAM,KAAK;CAAgB;CAEpD,IAAI,aAAa,QAAQ,KAAK,SAC5B,OAAO;EAAE,MAAM;EAAW,SAAS,KAAK;CAAsB;CAEhE,IAAI,kBAAkB,QAAQ,KAAK,cACjC,OAAO;EACL,MAAM;EACN,OAAO,KAAK;CAMd;CAEF,IAAI,oBAAoB,QAAQ,KAAK,gBACnC,OAAO;EACL,MAAM;EACN,OAAO,KAAK;CAMd;CAEF,OAAO;AACT;AAEA,SAAS,WAAW,QAAmC;CACrD,OAAO,SAAS,EAAE,OAAO,IAAI,CAAC;AAChC;AAEA,IAAa,YAAb,MAAuB;CACrB;CACA;CACA;CACA;CACA;CAIA;CAIA,YAAY,SAA2B;EACrC,KAAK,UAAU,QAAQ,QAAQ,QAAQ,OAAO,EAAE;EAChD,KAAK,WAAW,QAAQ,WACpB,IAAI,QAAQ,SAAS,QAAQ,YAAY,EAAE,MAC3C;EACJ,KAAK,SAAS,QAAQ;EACtB,KAAK,gBAAgB,QAAQ;EAC7B,MAAM,EACJ,SAAS,IACT,MAAM,IACN,QAAQ,IACR,QAAQ,IACR,GAAG,qBACA,QAAQ,gBAAgB,CAAC;EAC9B,KAAK,eAAe;EACpB,KAAK,YAAY,QAAQ,WAAW,CAAC;CACvC;CAEA,cAA8B;EAC5B,OAAO,GAAG,KAAK,WAAW,KAAK,SAAS,IAAI,mBAAmB,KAAK,MAAM,MAAM;CAClF;CAEA,MAAc,WACZ,qBAAqB,MACY;EAKjC,MAAM,UAAkC;GACtC,QAAQ;GACR,eAAA;GACA,GANA,OAAO,KAAK,cAAc,aACtB,MAAM,KAAK,UAAU,IACrB,KAAK;EAKX;EACA,IAAI,oBACF,QAAQ,kBAAkB;EAE5B,IAAI,KAAK,eAAe,QACtB,QAAQ,oBAAoB,KAAK,cAAc,KAAK,IAAI;EAE1D,OAAO;CACT;CAEA,MAAc,mBAAmB,UAAoC;EACnE,IAAI;EACJ,IAAI;GACF,YAAY,MAAM,SAAS,KAAK;EAClC,QAAQ,CAER;EAEA,IAAI,aAAa,OAAO,cAAc,YAAY,WAAW,WAAW;GACtE,MAAM,MAAO,UAAkC;GAC/C,MAAM,IAAI,SAAS;IACjB,MAAM,IAAI,QAAQ,SAAS;IAC3B,QAAQ,IAAI,UAAU,SAAS;IAC/B,SAAS,IAAI,WAAW,uBAAuB,SAAS;IACxD,SAAS,IAAI;GACf,CAAC;EACH;EAEA,MAAM,IAAI,SAAS;GACjB,MAAM,SAAS;GACf,QAAQ,SAAS;GACjB,SAAS,uBAAuB,SAAS,OAAO,GAAG,SAAS;EAC9D,CAAC;CACH;CAEA,MAAc,UACZ,MACA,UAAuB,CAAC,GACZ;EACZ,MAAM,QAAQ,CAAC,QAAQ,UAAU,QAAQ,OAAO,YAAY,MAAM;EAClE,MAAM,UAAU,MAAM,KAAK,WAAW,CAAC,KAAK;EAC5C,MAAM,WAAW,MAAM,MAAM,GAAG,KAAK,UAAU,QAAQ;GACrD,GAAG,KAAK;GACR,GAAG;GACH,SAAS;IACP,GAAG;IACH,GAAI,QAAQ;GACd;EACF,CAAC;EACD,IAAI,CAAC,SAAS,IACZ,MAAM,KAAK,mBAAmB,QAAQ;EAGxC,OAAO,cAAc,MADF,SAAS,KAAK,CACR;CAC3B;CAIA,MAAM,aAAa,QAA6C;EAC9D,MAAM,UAAU,MAAM,KAAK,WAAW,KAAK;EAC3C,MAAM,MAAM,GAAG,KAAK,QAAQ;EAC5B,MAAM,WAAW,MAAM,MAAM,KAAK;GAChC,GAAG,KAAK;GACR;GACA,GAAG,WAAW,MAAM;EACtB,CAAC;EACD,IAAI,CAAC,SAAS,IACZ,MAAM,KAAK,mBAAmB,QAAQ;EAGxC,OAAO,cAAc,MADF,SAAS,KAAK,CACR;CAC3B;CAEA,MAAM,qBAAqB,QAA6C;EACtE,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,qBACtB,WAAW,MAAM,CACnB;CACF;CAIA,MAAM,YACJ,SACA,eACA,UACA,QAC+B;EAC/B,MAAM,OAAgC,EACpC,SAAS,cAAc,OAAO,EAChC;EACA,IAAI,eAAe,KAAK,gBAAgB;EACxC,IAAI,UAAU,KAAK,WAAW;EAE9B,MAAM,SAAS,MAAM,KAAK,UACxB,GAAG,KAAK,YAAY,EAAE,gBACtB;GACE,QAAQ;GACR,MAAM,KAAK,UAAU,IAAI;GACzB,GAAG,WAAW,MAAM;EACtB,CACF;EAGA,IAAI,UAAU,UAAU,OAAO,MAAM,OAAO,OAAO;EACnD,IAAI,aAAa,UAAU,OAAO,SAChC,OAAO,OAAO;EAChB,OAAO;CACT;CAEA,OAAO,cACL,SACA,eACA,UACA,QACgC;EAChC,MAAM,UAAU,MAAM,KAAK,WAAW,IAAI;EAC1C,QAAQ,SAAS;EAEjB,MAAM,OAAgC,EACpC,SAAS,cAAc,OAAO,EAChC;EACA,IAAI,eAAe,KAAK,gBAAgB;EACxC,IAAI,UAAU,KAAK,WAAW;EAE9B,MAAM,WAAW,MAAM,MACrB,GAAG,KAAK,UAAU,KAAK,YAAY,EAAE,kBACrC;GACE,GAAG,KAAK;GACR,QAAQ;GACR;GACA,MAAM,KAAK,UAAU,IAAI;GACzB,GAAG,WAAW,MAAM;EACtB,CACF;EACA,IAAI,CAAC,SAAS,IACZ,MAAM,KAAK,mBAAmB,QAAQ;EAGxC,OAAO,KAAK,SAAS,QAAQ;CAC/B;CAIA,MAAM,QACJ,QACA,eACA,QACkB;EAClB,MAAM,SAAS,IAAI,gBAAgB;EACnC,IAAI,kBAAkB,KAAA,GAEpB,OAAO,IAAI,kBAAkB,OAAO,aAAa,CAAC;EAEpD,MAAM,KAAK,OAAO,SAAS;EAC3B,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,IAAI,KAAK,IAAI,OAAO,MAC5E,WAAW,MAAM,CACnB;CACF;CAEA,MAAM,UACJ,SACA,QAC+B;EAC/B,MAAM,SAAS,IAAI,gBAAgB;EACnC,IAAI,SAAS,WAAW,OAAO,IAAI,cAAc,QAAQ,SAAS;EAClE,IAAI,SAAS,QAAQ,OAAO,IAAI,UAAU,gBAAgB,QAAQ,MAAM,CAAC;EACzE,IAAI,SAAS,aAAa,KAAA,GACxB,OAAO,IAAI,aAAa,OAAO,QAAQ,QAAQ,CAAC;EAClD,IAAI,SAAS,WAAW,OAAO,IAAI,cAAc,QAAQ,SAAS;EAClE,IAAI,SAAS,kBAAkB,KAAA,GAC7B,OAAO,IAAI,kBAAkB,OAAO,QAAQ,aAAa,CAAC;EAC5D,IAAI,SAAS,sBACX,OAAO,IAAI,0BAA0B,QAAQ,oBAAoB;EACnE,IAAI,SAAS,qBAAqB,KAAA,GAChC,OAAO,IAAI,qBAAqB,OAAO,QAAQ,gBAAgB,CAAC;EAClE,MAAM,KAAK,OAAO,SAAS;EAC3B,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,QAAQ,KAAK,IAAI,OAAO,MAC9C,WAAW,MAAM,CACnB;CACF;CAEA,MAAM,WACJ,QACA,UACA,QACkB;EAClB,MAAM,OAAO,WAAW,EAAE,SAAS,IAAI,CAAC;EACxC,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,UAC1D;GACE,QAAQ;GACR,MAAM,KAAK,UAAU,IAAI;GACzB,GAAG,WAAW,MAAM;EACtB,CACF;CACF;CAEA,OAAO,gBACL,QACA,QACgC;EAChC,MAAM,UAAU,MAAM,KAAK,WAAW,KAAK;EAC3C,QAAQ,SAAS;EAEjB,MAAM,WAAW,MAAM,MACrB,GAAG,KAAK,UAAU,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,aACzE;GACE,GAAG,KAAK;GACR;GACA,GAAG,WAAW,MAAM;EACtB,CACF;EACA,IAAI,CAAC,SAAS,IACZ,MAAM,KAAK,mBAAmB,QAAQ;EAGxC,OAAO,KAAK,SAAS,QAAQ;CAC/B;CAIA,MAAM,iCACJ,QACA,QACwC;EACxC,MAAM,SAAS,OAAO;EACtB,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,oBAAoB;EACjD,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,2BAC1D;GACE,QAAQ;GACR,MAAM,KAAK,UAAU,MAAM;GAC3B,GAAG,WAAW,MAAM;EACtB,CACF;CACF;CAEA,MAAM,8BACJ,QACA,UACA,QACwC;EACxC,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,2BAA2B,mBAAmB,QAAQ,KAChH,WAAW,MAAM,CACnB;CACF;CAEA,MAAM,gCACJ,QACA,SACA,QACqD;EACrD,MAAM,SAAS,IAAI,gBAAgB;EACnC,IAAI,SAAS,aAAa,KAAA,GACxB,OAAO,IAAI,aAAa,OAAO,QAAQ,QAAQ,CAAC;EAClD,IAAI,SAAS,WAAW,OAAO,IAAI,cAAc,QAAQ,SAAS;EAClE,MAAM,KAAK,OAAO,SAAS;EAC3B,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,0BAA0B,KAAK,IAAI,OAAO,MACpG,WAAW,MAAM,CACnB;CACF;CAEA,MAAM,iCACJ,QACA,UACA,QACe;EAEf,MAAM,UAAU,MAAM,KAAK,WAAW,IAAM;EAC5C,MAAM,WAAW,MAAM,MACrB,GAAG,KAAK,UAAU,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,2BAA2B,mBAAmB,QAAQ,KAC/H;GACE,GAAG,KAAK;GACR,QAAQ;GACR;GACA,GAAG,WAAW,MAAM;EACtB,CACF;EACA,IAAI,CAAC,SAAS,IACZ,MAAM,KAAK,mBAAmB,QAAQ;CAE1C;CAIA,OAAe,SAAS,UAAoD;EAC1E,MAAM,cAAc,SAAS,QAAQ,IAAI,cAAc;EAEvD,IADkB,aAAa,MAAM,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,YAAY,MAClD,qBAAqB;GACrC,MAAM,WAAW,cACb,IAAI,YAAY,KAChB;GACJ,MAAM,IAAI,MACR,2EAA2E,UAC7E;EACF;EAEA,MAAM,SAAS,SAAS,MAAM,UAAU;EACxC,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,kBAAkB;EAE/C,MAAM,UAAU,IAAI,YAAY;EAChC,MAAM,aAAa,IAAI,gBAAgB;EAEvC,MAAM,aAAa,UAA2C;GAC5D,IAAI;IACF,IAAI,SAAS,KAAK,MAAM,MAAM,IAAI;IAElC,IACE,UACA,OAAO,WAAW,YAClB,aAAa,UACb,YAAY,QAEZ,SAAS,OAAO;IAIlB,OAAO,2BADY,cAAc,MACU,CAAC;GAC9C,QAAQ;IACN,OAAO;GACT;EACF;EAEA,IAAI;GACF,OAAO,MAAM;IACX,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,KAAK;IAC1C,IAAI,MAAM;KACR,KAAK,MAAM,SAAS,WAAW,KAAK,QAAQ,OAAO,CAAC,GAAG;MACrD,MAAM,SAAS,UAAU,KAAK;MAC9B,IAAI,QAAQ,MAAM;KACpB;KACA;IACF;IAEA,KAAK,MAAM,SAAS,WAAW,KAC7B,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC,CACxC,GAAG;KACD,MAAM,SAAS,UAAU,KAAK;KAC9B,IAAI,QAAQ,MAAM;IACpB;GACF;EACF,UAAU;GACR,OAAO,YAAY;EACrB;CACF;AACF"}
+{"version":3,"file":"A2AClient.js","names":[],"sources":["../src/A2AClient.ts"],"sourcesContent":["import { SSEEventDecoder, type SSEEvent } from \"assistant-stream/utils\";\nimport { isRecord } from \"@assistant-ui/core/internal\";\nimport type {\n  A2AAgentCard,\n  A2AErrorInfo,\n  A2AListTaskPushNotificationConfigsResponse,\n  A2AListTasksRequest,\n  A2AListTasksResponse,\n  A2AMessage,\n  A2ARole,\n  A2ASendMessageConfiguration,\n  A2AStreamEvent,\n  A2ATask,\n  A2ATaskPushNotificationConfig,\n  A2ATaskState,\n} from \"./types\";\nimport { A2A_PROTOCOL_VERSION } from \"./types\";\n\nexport type A2AClientOptions = {\n  baseUrl: string;\n  /** Optional path prefix for all API endpoints (e.g. \"/v1\"). Does not affect agent card discovery. */\n  basePath?: string | undefined;\n  /** Optional tenant ID for multi-tenant servers. */\n  tenant?: string | undefined;\n  headers?:\n    | Record<string, string>\n    | (() => Record<string, string> | Promise<Record<string, string>>)\n    | undefined;\n  /** A2A extension URIs to negotiate. Sent as A2A-Extensions header. */\n  extensions?: string[] | undefined;\n  /** Extra fetch options applied to every request. */\n  fetchOptions?:\n    | Omit<RequestInit, \"headers\" | \"body\" | \"method\" | \"signal\">\n    | undefined;\n};\n\nexport class A2AError extends Error {\n  code: number;\n  status: string;\n  details: unknown[] | undefined;\n\n  constructor(info: A2AErrorInfo) {\n    super(info.message);\n    this.name = \"A2AError\";\n    this.code = info.code;\n    this.status = info.status;\n    this.details = info.details;\n  }\n}\n\n// Incoming key normalization: snake_case → camelCase, plus ProtoJSON enum normalization.\nfunction toCamelCase(key: string): string {\n  return key.replace(/_([a-z])/g, (_, c: string) => c.toUpperCase());\n}\n\n// Fields whose values are opaque user data (google.protobuf.Struct / Value).\n// Keys inside these objects must NOT be camelCased or have enum normalization applied.\nconst OPAQUE_FIELDS = new Set([\n  \"metadata\",\n  \"data\",\n  \"params\",\n  \"forwardedProps\",\n  \"scopes\",\n]);\n\nfunction normalizeKeys(obj: unknown, opaque = false): unknown {\n  if (Array.isArray(obj)) return obj.map((v) => normalizeKeys(v, opaque));\n  if (obj !== null && typeof obj === \"object\") {\n    const result: Record<string, unknown> = {};\n    for (const [key, value] of Object.entries(obj as Record<string, unknown>)) {\n      // Inside opaque fields: preserve keys and values as-is (only recurse arrays/objects structurally)\n      if (opaque) {\n        result[key] =\n          typeof value === \"object\" && value !== null\n            ? normalizeKeys(value, true)\n            : value;\n        continue;\n      }\n\n      const camelKey = toCamelCase(key);\n      const isOpaqueChild = OPAQUE_FIELDS.has(camelKey);\n\n      if (\n        camelKey === \"state\" &&\n        typeof value === \"string\" &&\n        value.startsWith(\"TASK_STATE_\")\n      ) {\n        result[camelKey] = value.slice(11).toLowerCase();\n      } else if (\n        camelKey === \"role\" &&\n        typeof value === \"string\" &&\n        value.startsWith(\"ROLE_\")\n      ) {\n        result[camelKey] = value.slice(5).toLowerCase();\n      } else if (camelKey === \"content\" && Array.isArray(value)) {\n        // v0.3 servers used \"content\" for message/artifact parts; normalize to \"parts\" for backward compat\n        result.parts = normalizeKeys(value, false);\n      } else if (camelKey !== \"parts\" || !(\"parts\" in result)) {\n        // dedup: \"content\" was already mapped to parts above; don't overwrite\n        result[camelKey] = isOpaqueChild ? value : normalizeKeys(value, false);\n      }\n    }\n    return result;\n  }\n  return obj;\n}\n\n// Outgoing enum conversion (v1.0 ProtoJSON format)\nfunction toWireRole(role: A2ARole): string {\n  if (role === \"user\") return \"ROLE_USER\";\n  if (role === \"agent\") return \"ROLE_AGENT\";\n  return \"ROLE_UNSPECIFIED\";\n}\n\nfunction toWireTaskState(state: A2ATaskState): string {\n  return `TASK_STATE_${state.toUpperCase()}`;\n}\n\nfunction toWireMessage(msg: A2AMessage): unknown {\n  return { ...msg, role: toWireRole(msg.role) };\n}\n\nfunction discriminateStreamResponse(\n  data: Record<string, unknown>,\n): A2AStreamEvent | null {\n  if (\"task\" in data && data.task) {\n    return { type: \"task\", task: data.task as A2ATask };\n  }\n  if (\"message\" in data && data.message) {\n    return { type: \"message\", message: data.message as A2AMessage };\n  }\n  if (\"statusUpdate\" in data && data.statusUpdate) {\n    return {\n      type: \"statusUpdate\",\n      event: data.statusUpdate as A2AStreamEvent extends {\n        type: \"statusUpdate\";\n        event: infer E;\n      }\n        ? E\n        : never,\n    };\n  }\n  if (\"artifactUpdate\" in data && data.artifactUpdate) {\n    return {\n      type: \"artifactUpdate\",\n      event: data.artifactUpdate as A2AStreamEvent extends {\n        type: \"artifactUpdate\";\n        event: infer E;\n      }\n        ? E\n        : never,\n    };\n  }\n  return null;\n}\n\nconst TASK_STATES: ReadonlySet<string> = new Set(\n  Object.keys({\n    unspecified: true,\n    submitted: true,\n    working: true,\n    completed: true,\n    failed: true,\n    canceled: true,\n    input_required: true,\n    rejected: true,\n    auth_required: true,\n  } satisfies Record<A2ATaskState, true>),\n);\n\nconst isTaskState = (value: unknown): value is A2ATaskState =>\n  typeof value === \"string\" && TASK_STATES.has(value);\n\nconst isTask = (value: unknown): value is A2ATask =>\n  isRecord(value) &&\n  typeof value.id === \"string\" &&\n  value.id.length > 0 &&\n  isRecord(value.status) &&\n  isTaskState(value.status.state);\n\nconst isMessage = (value: unknown): value is A2AMessage =>\n  isRecord(value) &&\n  typeof value.messageId === \"string\" &&\n  value.messageId.length > 0 &&\n  typeof value.role === \"string\" &&\n  value.role.length > 0 &&\n  Array.isArray(value.parts) &&\n  value.parts.every(isRecord);\n\nconst isStringArray = (value: unknown): value is string[] =>\n  Array.isArray(value) && value.every((item) => typeof item === \"string\");\n\nconst invalidAgentCard = (): never => {\n  throw new Error(\n    \"Invalid A2A agent card response: expected a valid agent card payload.\",\n  );\n};\n\nconst parseCardString = (value: unknown): string =>\n  value == null ? \"\" : typeof value === \"string\" ? value : invalidAgentCard();\n\nconst parseCardStringArray = (value: unknown): string[] =>\n  value == null ? [] : isStringArray(value) ? value : invalidAgentCard();\n\nconst parseCardRecordArray = (value: unknown): Record<string, unknown>[] =>\n  value == null\n    ? []\n    : Array.isArray(value) && value.every(isRecord)\n      ? (value as Record<string, unknown>[])\n      : invalidAgentCard();\n\nconst parseCardRecord = (value: unknown): Record<string, unknown> =>\n  value == null ? {} : isRecord(value) ? value : invalidAgentCard();\n\n// Proto3 JSON parsing treats omitted and null fields as defaults, so a valid\n// card may arrive without its empty lists, strings, or capabilities. Fill\n// those per the proto3 JSON mapping rules; a payload without a name or with a\n// present field of the wrong type rejects.\nconst parseAgentCardResponse = (value: unknown): A2AAgentCard => {\n  if (\n    !isRecord(value) ||\n    typeof value.name !== \"string\" ||\n    value.name.length === 0\n  ) {\n    return invalidAgentCard();\n  }\n\n  return {\n    ...value,\n    name: value.name,\n    description: parseCardString(value.description),\n    version: parseCardString(value.version),\n    supportedInterfaces: parseCardRecordArray(value.supportedInterfaces).map(\n      (entry) => ({\n        ...entry,\n        url: parseCardString(entry.url),\n        protocolBinding: parseCardString(entry.protocolBinding),\n        protocolVersion: parseCardString(entry.protocolVersion),\n      }),\n    ),\n    capabilities: parseCardRecord(value.capabilities),\n    defaultInputModes: parseCardStringArray(value.defaultInputModes),\n    defaultOutputModes: parseCardStringArray(value.defaultOutputModes),\n    skills: parseCardRecordArray(value.skills).map((entry) => ({\n      ...entry,\n      id: parseCardString(entry.id),\n      name: parseCardString(entry.name),\n      description: parseCardString(entry.description),\n      tags: parseCardStringArray(entry.tags),\n    })),\n  } as A2AAgentCard;\n};\n\nconst parseSendMessageResponse = (value: unknown): A2ATask | A2AMessage => {\n  if (isRecord(value)) {\n    const candidate = value.task ?? value.message ?? value;\n    if (isTask(candidate) || isMessage(candidate)) return candidate;\n  }\n\n  throw new Error(\n    \"Invalid A2A message:send response: expected a valid task or message payload.\",\n  );\n};\n\nfunction signalInit(signal?: AbortSignal): RequestInit {\n  return signal ? { signal } : {};\n}\n\nexport class A2AClient {\n  private baseUrl: string;\n  private basePath: string;\n  private tenant: string | undefined;\n  private extensionUris: string[] | undefined;\n  private fetchOptions: Omit<\n    RequestInit,\n    \"headers\" | \"body\" | \"method\" | \"signal\"\n  >;\n  private headersFn:\n    | Record<string, string>\n    | (() => Record<string, string> | Promise<Record<string, string>>);\n\n  constructor(options: A2AClientOptions) {\n    this.baseUrl = options.baseUrl.replace(/\\/$/, \"\");\n    this.basePath = options.basePath\n      ? `/${options.basePath.replace(/^\\/|\\/$/g, \"\")}`\n      : \"\";\n    this.tenant = options.tenant;\n    this.extensionUris = options.extensions;\n    const {\n      headers: _h,\n      body: _b,\n      method: _m,\n      signal: _s,\n      ...safeFetchOptions\n    } = (options.fetchOptions ?? {}) as RequestInit;\n    this.fetchOptions = safeFetchOptions;\n    this.headersFn = options.headers ?? {};\n  }\n\n  private getBasePath(): string {\n    return `${this.basePath}${this.tenant ? `/${encodeURIComponent(this.tenant)}` : \"\"}`;\n  }\n\n  private async getHeaders(\n    includeContentType = true,\n  ): Promise<Record<string, string>> {\n    const custom =\n      typeof this.headersFn === \"function\"\n        ? await this.headersFn()\n        : this.headersFn;\n    const headers: Record<string, string> = {\n      Accept: \"application/a2a+json, application/json\",\n      \"A2A-Version\": A2A_PROTOCOL_VERSION,\n      ...custom,\n    };\n    if (includeContentType) {\n      headers[\"Content-Type\"] = \"application/a2a+json\";\n    }\n    if (this.extensionUris?.length) {\n      headers[\"A2A-Extensions\"] = this.extensionUris.join(\", \");\n    }\n    return headers;\n  }\n\n  private async throwResponseError(response: Response): Promise<never> {\n    let errorBody: unknown;\n    try {\n      errorBody = await response.json();\n    } catch {\n      // no parseable body\n    }\n\n    if (errorBody && typeof errorBody === \"object\" && \"error\" in errorBody) {\n      const err = (errorBody as Record<string, any>).error;\n      throw new A2AError({\n        code: err.code ?? response.status,\n        status: err.status ?? response.statusText,\n        message: err.message ?? `A2A request failed: ${response.status}`,\n        details: err.details,\n      });\n    }\n\n    throw new A2AError({\n      code: response.status,\n      status: response.statusText,\n      message: `A2A request failed: ${response.status} ${response.statusText}`,\n    });\n  }\n\n  private async fetchJSON<T>(\n    path: string,\n    options: RequestInit = {},\n  ): Promise<T> {\n    const isGet = !options.method || options.method.toUpperCase() === \"GET\";\n    const headers = await this.getHeaders(!isGet);\n    const response = await fetch(`${this.baseUrl}${path}`, {\n      ...this.fetchOptions,\n      ...options,\n      headers: {\n        ...headers,\n        ...(options.headers as Record<string, string>),\n      },\n    });\n    if (!response.ok) {\n      await this.throwResponseError(response);\n    }\n    const json = await response.json();\n    return normalizeKeys(json) as T;\n  }\n\n  // --- Agent Card ---\n\n  async getAgentCard(signal?: AbortSignal): Promise<A2AAgentCard> {\n    const headers = await this.getHeaders(false); // GET: no Content-Type\n    const url = `${this.baseUrl}/.well-known/agent-card.json`;\n    const response = await fetch(url, {\n      ...this.fetchOptions,\n      headers,\n      ...signalInit(signal),\n    });\n    if (!response.ok) {\n      await this.throwResponseError(response);\n    }\n    const json = await response.json();\n    return parseAgentCardResponse(normalizeKeys(json));\n  }\n\n  async getExtendedAgentCard(signal?: AbortSignal): Promise<A2AAgentCard> {\n    const result = await this.fetchJSON<unknown>(\n      `${this.getBasePath()}/extendedAgentCard`,\n      signalInit(signal),\n    );\n    return parseAgentCardResponse(result);\n  }\n\n  // --- Message ---\n\n  async sendMessage(\n    message: A2AMessage,\n    configuration?: A2ASendMessageConfiguration,\n    metadata?: Record<string, unknown>,\n    signal?: AbortSignal,\n  ): Promise<A2ATask | A2AMessage> {\n    const body: Record<string, unknown> = {\n      message: toWireMessage(message),\n    };\n    if (configuration) body.configuration = configuration;\n    if (metadata) body.metadata = metadata;\n\n    const result = await this.fetchJSON<unknown>(\n      `${this.getBasePath()}/message:send`,\n      {\n        method: \"POST\",\n        body: JSON.stringify(body),\n        ...signalInit(signal),\n      },\n    );\n\n    return parseSendMessageResponse(result);\n  }\n\n  async *streamMessage(\n    message: A2AMessage,\n    configuration?: A2ASendMessageConfiguration,\n    metadata?: Record<string, unknown>,\n    signal?: AbortSignal,\n  ): AsyncGenerator<A2AStreamEvent> {\n    const headers = await this.getHeaders(true);\n    headers.Accept = \"text/event-stream\";\n\n    const body: Record<string, unknown> = {\n      message: toWireMessage(message),\n    };\n    if (configuration) body.configuration = configuration;\n    if (metadata) body.metadata = metadata;\n\n    const response = await fetch(\n      `${this.baseUrl}${this.getBasePath()}/message:stream`,\n      {\n        ...this.fetchOptions,\n        method: \"POST\",\n        headers,\n        body: JSON.stringify(body),\n        ...signalInit(signal),\n      },\n    );\n    if (!response.ok) {\n      await this.throwResponseError(response);\n    }\n\n    yield* this.parseSSE(response);\n  }\n\n  // --- Tasks ---\n\n  async getTask(\n    taskId: string,\n    historyLength?: number,\n    signal?: AbortSignal,\n  ): Promise<A2ATask> {\n    const params = new URLSearchParams();\n    if (historyLength !== undefined) {\n      // Proto field name for HTTP transcoding query params\n      params.set(\"history_length\", String(historyLength));\n    }\n    const qs = params.toString();\n    return this.fetchJSON<A2ATask>(\n      `${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}${qs ? `?${qs}` : \"\"}`,\n      signalInit(signal),\n    );\n  }\n\n  async listTasks(\n    request?: A2AListTasksRequest,\n    signal?: AbortSignal,\n  ): Promise<A2AListTasksResponse> {\n    const params = new URLSearchParams();\n    if (request?.contextId) params.set(\"context_id\", request.contextId);\n    if (request?.status) params.set(\"status\", toWireTaskState(request.status));\n    if (request?.pageSize !== undefined)\n      params.set(\"page_size\", String(request.pageSize));\n    if (request?.pageToken) params.set(\"page_token\", request.pageToken);\n    if (request?.historyLength !== undefined)\n      params.set(\"history_length\", String(request.historyLength));\n    if (request?.statusTimestampAfter)\n      params.set(\"status_timestamp_after\", request.statusTimestampAfter);\n    if (request?.includeArtifacts !== undefined)\n      params.set(\"include_artifacts\", String(request.includeArtifacts));\n    const qs = params.toString();\n    return this.fetchJSON<A2AListTasksResponse>(\n      `${this.getBasePath()}/tasks${qs ? `?${qs}` : \"\"}`,\n      signalInit(signal),\n    );\n  }\n\n  async cancelTask(\n    taskId: string,\n    metadata?: Record<string, unknown>,\n    signal?: AbortSignal,\n  ): Promise<A2ATask> {\n    const body = metadata ? { metadata } : {};\n    return this.fetchJSON<A2ATask>(\n      `${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}:cancel`,\n      {\n        method: \"POST\",\n        body: JSON.stringify(body),\n        ...signalInit(signal),\n      },\n    );\n  }\n\n  async *subscribeToTask(\n    taskId: string,\n    signal?: AbortSignal,\n  ): AsyncGenerator<A2AStreamEvent> {\n    const headers = await this.getHeaders(false); // GET: no Content-Type\n    headers.Accept = \"text/event-stream\";\n\n    const response = await fetch(\n      `${this.baseUrl}${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}:subscribe`,\n      {\n        ...this.fetchOptions,\n        headers,\n        ...signalInit(signal),\n      },\n    );\n    if (!response.ok) {\n      await this.throwResponseError(response);\n    }\n\n    yield* this.parseSSE(response);\n  }\n\n  // --- Push Notification Configs ---\n\n  async createTaskPushNotificationConfig(\n    config: A2ATaskPushNotificationConfig,\n    signal?: AbortSignal,\n  ): Promise<A2ATaskPushNotificationConfig> {\n    const taskId = config.taskId;\n    if (!taskId) throw new Error(\"taskId is required\");\n    return this.fetchJSON<A2ATaskPushNotificationConfig>(\n      `${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}/pushNotificationConfigs`,\n      {\n        method: \"POST\",\n        body: JSON.stringify(config),\n        ...signalInit(signal),\n      },\n    );\n  }\n\n  async getTaskPushNotificationConfig(\n    taskId: string,\n    configId: string,\n    signal?: AbortSignal,\n  ): Promise<A2ATaskPushNotificationConfig> {\n    return this.fetchJSON<A2ATaskPushNotificationConfig>(\n      `${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}/pushNotificationConfigs/${encodeURIComponent(configId)}`,\n      signalInit(signal),\n    );\n  }\n\n  async listTaskPushNotificationConfigs(\n    taskId: string,\n    options?: { pageSize?: number; pageToken?: string },\n    signal?: AbortSignal,\n  ): Promise<A2AListTaskPushNotificationConfigsResponse> {\n    const params = new URLSearchParams();\n    if (options?.pageSize !== undefined)\n      params.set(\"page_size\", String(options.pageSize));\n    if (options?.pageToken) params.set(\"page_token\", options.pageToken);\n    const qs = params.toString();\n    return this.fetchJSON<A2AListTaskPushNotificationConfigsResponse>(\n      `${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}/pushNotificationConfigs${qs ? `?${qs}` : \"\"}`,\n      signalInit(signal),\n    );\n  }\n\n  async deleteTaskPushNotificationConfig(\n    taskId: string,\n    configId: string,\n    signal?: AbortSignal,\n  ): Promise<void> {\n    const isGet = false;\n    const headers = await this.getHeaders(!isGet);\n    const response = await fetch(\n      `${this.baseUrl}${this.getBasePath()}/tasks/${encodeURIComponent(taskId)}/pushNotificationConfigs/${encodeURIComponent(configId)}`,\n      {\n        ...this.fetchOptions,\n        method: \"DELETE\",\n        headers,\n        ...signalInit(signal),\n      },\n    );\n    if (!response.ok) {\n      await this.throwResponseError(response);\n    }\n  }\n\n  // --- SSE Parsing ---\n\n  private async *parseSSE(response: Response): AsyncGenerator<A2AStreamEvent> {\n    const contentType = response.headers.get(\"Content-Type\");\n    const mediaType = contentType?.split(\";\", 1)[0]?.trim().toLowerCase();\n    if (mediaType !== \"text/event-stream\") {\n      const received = contentType\n        ? `\"${contentType}\"`\n        : \"no Content-Type header\";\n      throw new Error(\n        `Expected A2A stream response Content-Type \"text/event-stream\", received ${received}`,\n      );\n    }\n\n    const reader = response.body?.getReader();\n    if (!reader) throw new Error(\"No response body\");\n\n    const decoder = new TextDecoder();\n    const sseDecoder = new SSEEventDecoder();\n\n    const readEvent = (event: SSEEvent): A2AStreamEvent | null => {\n      try {\n        let parsed = JSON.parse(event.data);\n\n        if (\n          parsed &&\n          typeof parsed === \"object\" &&\n          \"jsonrpc\" in parsed &&\n          \"result\" in parsed\n        ) {\n          parsed = parsed.result;\n        }\n\n        const normalized = normalizeKeys(parsed) as Record<string, unknown>;\n        return discriminateStreamResponse(normalized);\n      } catch {\n        return null;\n      }\n    };\n\n    try {\n      while (true) {\n        const { done, value } = await reader.read();\n        if (done) {\n          for (const event of sseDecoder.push(decoder.decode())) {\n            const parsed = readEvent(event);\n            if (parsed) yield parsed;\n          }\n          break;\n        }\n\n        for (const event of sseDecoder.push(\n          decoder.decode(value, { stream: true }),\n        )) {\n          const parsed = readEvent(event);\n          if (parsed) yield parsed;\n        }\n      }\n    } finally {\n      reader.releaseLock();\n    }\n  }\n}\n"],"mappings":";;;;AAoCA,IAAa,WAAb,cAA8B,MAAM;CAClC;CACA;CACA;CAEA,YAAY,MAAoB;EAC9B,MAAM,KAAK,OAAO;EAClB,KAAK,OAAO;EACZ,KAAK,OAAO,KAAK;EACjB,KAAK,SAAS,KAAK;EACnB,KAAK,UAAU,KAAK;CACtB;AACF;AAGA,SAAS,YAAY,KAAqB;CACxC,OAAO,IAAI,QAAQ,cAAc,GAAG,MAAc,EAAE,YAAY,CAAC;AACnE;AAIA,MAAM,gCAAgB,IAAI,IAAI;CAC5B;CACA;CACA;CACA;CACA;AACF,CAAC;AAED,SAAS,cAAc,KAAc,SAAS,OAAgB;CAC5D,IAAI,MAAM,QAAQ,GAAG,GAAG,OAAO,IAAI,KAAK,MAAM,cAAc,GAAG,MAAM,CAAC;CACtE,IAAI,QAAQ,QAAQ,OAAO,QAAQ,UAAU;EAC3C,MAAM,SAAkC,CAAC;EACzC,KAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,GAA8B,GAAG;GAEzE,IAAI,QAAQ;IACV,OAAO,OACL,OAAO,UAAU,YAAY,UAAU,OACnC,cAAc,OAAO,IAAI,IACzB;IACN;GACF;GAEA,MAAM,WAAW,YAAY,GAAG;GAChC,MAAM,gBAAgB,cAAc,IAAI,QAAQ;GAEhD,IACE,aAAa,WACb,OAAO,UAAU,YACjB,MAAM,WAAW,aAAa,GAE9B,OAAO,YAAY,MAAM,MAAM,EAAE,CAAC,CAAC,YAAY;QAC1C,IACL,aAAa,UACb,OAAO,UAAU,YACjB,MAAM,WAAW,OAAO,GAExB,OAAO,YAAY,MAAM,MAAM,CAAC,CAAC,CAAC,YAAY;QACzC,IAAI,aAAa,aAAa,MAAM,QAAQ,KAAK,GAEtD,OAAO,QAAQ,cAAc,OAAO,KAAK;QACpC,IAAI,aAAa,WAAW,EAAE,WAAW,SAE9C,OAAO,YAAY,gBAAgB,QAAQ,cAAc,OAAO,KAAK;EAEzE;EACA,OAAO;CACT;CACA,OAAO;AACT;AAGA,SAAS,WAAW,MAAuB;CACzC,IAAI,SAAS,QAAQ,OAAO;CAC5B,IAAI,SAAS,SAAS,OAAO;CAC7B,OAAO;AACT;AAEA,SAAS,gBAAgB,OAA6B;CACpD,OAAO,cAAc,MAAM,YAAY;AACzC;AAEA,SAAS,cAAc,KAA0B;CAC/C,OAAO;EAAE,GAAG;EAAK,MAAM,WAAW,IAAI,IAAI;CAAE;AAC9C;AAEA,SAAS,2BACP,MACuB;CACvB,IAAI,UAAU,QAAQ,KAAK,MACzB,OAAO;EAAE,MAAM;EAAQ,MAAM,KAAK;CAAgB;CAEpD,IAAI,aAAa,QAAQ,KAAK,SAC5B,OAAO;EAAE,MAAM;EAAW,SAAS,KAAK;CAAsB;CAEhE,IAAI,kBAAkB,QAAQ,KAAK,cACjC,OAAO;EACL,MAAM;EACN,OAAO,KAAK;CAMd;CAEF,IAAI,oBAAoB,QAAQ,KAAK,gBACnC,OAAO;EACL,MAAM;EACN,OAAO,KAAK;CAMd;CAEF,OAAO;AACT;AAEA,MAAM,cAAmC,IAAI,IAC3C,OAAO,KAAK;CACV,aAAa;CACb,WAAW;CACX,SAAS;CACT,WAAW;CACX,QAAQ;CACR,UAAU;CACV,gBAAgB;CAChB,UAAU;CACV,eAAe;AACjB,CAAsC,CACxC;AAEA,MAAM,eAAe,UACnB,OAAO,UAAU,YAAY,YAAY,IAAI,KAAK;AAEpD,MAAM,UAAU,UACd,SAAS,KAAK,KACd,OAAO,MAAM,OAAO,YACpB,MAAM,GAAG,SAAS,KAClB,SAAS,MAAM,MAAM,KACrB,YAAY,MAAM,OAAO,KAAK;AAEhC,MAAM,aAAa,UACjB,SAAS,KAAK,KACd,OAAO,MAAM,cAAc,YAC3B,MAAM,UAAU,SAAS,KACzB,OAAO,MAAM,SAAS,YACtB,MAAM,KAAK,SAAS,KACpB,MAAM,QAAQ,MAAM,KAAK,KACzB,MAAM,MAAM,MAAM,QAAQ;AAE5B,MAAM,iBAAiB,UACrB,MAAM,QAAQ,KAAK,KAAK,MAAM,OAAO,SAAS,OAAO,SAAS,QAAQ;AAExE,MAAM,yBAAgC;CACpC,MAAM,IAAI,MACR,uEACF;AACF;AAEA,MAAM,mBAAmB,UACvB,SAAS,OAAO,KAAK,OAAO,UAAU,WAAW,QAAQ,iBAAiB;AAE5E,MAAM,wBAAwB,UAC5B,SAAS,OAAO,CAAC,IAAI,cAAc,KAAK,IAAI,QAAQ,iBAAiB;AAEvE,MAAM,wBAAwB,UAC5B,SAAS,OACL,CAAC,IACD,MAAM,QAAQ,KAAK,KAAK,MAAM,MAAM,QAAQ,IACzC,QACD,iBAAiB;AAEzB,MAAM,mBAAmB,UACvB,SAAS,OAAO,CAAC,IAAI,SAAS,KAAK,IAAI,QAAQ,iBAAiB;AAMlE,MAAM,0BAA0B,UAAiC;CAC/D,IACE,CAAC,SAAS,KAAK,KACf,OAAO,MAAM,SAAS,YACtB,MAAM,KAAK,WAAW,GAEtB,OAAO,iBAAiB;CAG1B,OAAO;EACL,GAAG;EACH,MAAM,MAAM;EACZ,aAAa,gBAAgB,MAAM,WAAW;EAC9C,SAAS,gBAAgB,MAAM,OAAO;EACtC,qBAAqB,qBAAqB,MAAM,mBAAmB,CAAC,CAAC,KAClE,WAAW;GACV,GAAG;GACH,KAAK,gBAAgB,MAAM,GAAG;GAC9B,iBAAiB,gBAAgB,MAAM,eAAe;GACtD,iBAAiB,gBAAgB,MAAM,eAAe;EACxD,EACF;EACA,cAAc,gBAAgB,MAAM,YAAY;EAChD,mBAAmB,qBAAqB,MAAM,iBAAiB;EAC/D,oBAAoB,qBAAqB,MAAM,kBAAkB;EACjE,QAAQ,qBAAqB,MAAM,MAAM,CAAC,CAAC,KAAK,WAAW;GACzD,GAAG;GACH,IAAI,gBAAgB,MAAM,EAAE;GAC5B,MAAM,gBAAgB,MAAM,IAAI;GAChC,aAAa,gBAAgB,MAAM,WAAW;GAC9C,MAAM,qBAAqB,MAAM,IAAI;EACvC,EAAE;CACJ;AACF;AAEA,MAAM,4BAA4B,UAAyC;CACzE,IAAI,SAAS,KAAK,GAAG;EACnB,MAAM,YAAY,MAAM,QAAQ,MAAM,WAAW;EACjD,IAAI,OAAO,SAAS,KAAK,UAAU,SAAS,GAAG,OAAO;CACxD;CAEA,MAAM,IAAI,MACR,8EACF;AACF;AAEA,SAAS,WAAW,QAAmC;CACrD,OAAO,SAAS,EAAE,OAAO,IAAI,CAAC;AAChC;AAEA,IAAa,YAAb,MAAuB;CACrB;CACA;CACA;CACA;CACA;CAIA;CAIA,YAAY,SAA2B;EACrC,KAAK,UAAU,QAAQ,QAAQ,QAAQ,OAAO,EAAE;EAChD,KAAK,WAAW,QAAQ,WACpB,IAAI,QAAQ,SAAS,QAAQ,YAAY,EAAE,MAC3C;EACJ,KAAK,SAAS,QAAQ;EACtB,KAAK,gBAAgB,QAAQ;EAC7B,MAAM,EACJ,SAAS,IACT,MAAM,IACN,QAAQ,IACR,QAAQ,IACR,GAAG,qBACA,QAAQ,gBAAgB,CAAC;EAC9B,KAAK,eAAe;EACpB,KAAK,YAAY,QAAQ,WAAW,CAAC;CACvC;CAEA,cAA8B;EAC5B,OAAO,GAAG,KAAK,WAAW,KAAK,SAAS,IAAI,mBAAmB,KAAK,MAAM,MAAM;CAClF;CAEA,MAAc,WACZ,qBAAqB,MACY;EAKjC,MAAM,UAAkC;GACtC,QAAQ;GACR,eAAA;GACA,GANA,OAAO,KAAK,cAAc,aACtB,MAAM,KAAK,UAAU,IACrB,KAAK;EAKX;EACA,IAAI,oBACF,QAAQ,kBAAkB;EAE5B,IAAI,KAAK,eAAe,QACtB,QAAQ,oBAAoB,KAAK,cAAc,KAAK,IAAI;EAE1D,OAAO;CACT;CAEA,MAAc,mBAAmB,UAAoC;EACnE,IAAI;EACJ,IAAI;GACF,YAAY,MAAM,SAAS,KAAK;EAClC,QAAQ,CAER;EAEA,IAAI,aAAa,OAAO,cAAc,YAAY,WAAW,WAAW;GACtE,MAAM,MAAO,UAAkC;GAC/C,MAAM,IAAI,SAAS;IACjB,MAAM,IAAI,QAAQ,SAAS;IAC3B,QAAQ,IAAI,UAAU,SAAS;IAC/B,SAAS,IAAI,WAAW,uBAAuB,SAAS;IACxD,SAAS,IAAI;GACf,CAAC;EACH;EAEA,MAAM,IAAI,SAAS;GACjB,MAAM,SAAS;GACf,QAAQ,SAAS;GACjB,SAAS,uBAAuB,SAAS,OAAO,GAAG,SAAS;EAC9D,CAAC;CACH;CAEA,MAAc,UACZ,MACA,UAAuB,CAAC,GACZ;EACZ,MAAM,QAAQ,CAAC,QAAQ,UAAU,QAAQ,OAAO,YAAY,MAAM;EAClE,MAAM,UAAU,MAAM,KAAK,WAAW,CAAC,KAAK;EAC5C,MAAM,WAAW,MAAM,MAAM,GAAG,KAAK,UAAU,QAAQ;GACrD,GAAG,KAAK;GACR,GAAG;GACH,SAAS;IACP,GAAG;IACH,GAAI,QAAQ;GACd;EACF,CAAC;EACD,IAAI,CAAC,SAAS,IACZ,MAAM,KAAK,mBAAmB,QAAQ;EAGxC,OAAO,cAAc,MADF,SAAS,KAAK,CACR;CAC3B;CAIA,MAAM,aAAa,QAA6C;EAC9D,MAAM,UAAU,MAAM,KAAK,WAAW,KAAK;EAC3C,MAAM,MAAM,GAAG,KAAK,QAAQ;EAC5B,MAAM,WAAW,MAAM,MAAM,KAAK;GAChC,GAAG,KAAK;GACR;GACA,GAAG,WAAW,MAAM;EACtB,CAAC;EACD,IAAI,CAAC,SAAS,IACZ,MAAM,KAAK,mBAAmB,QAAQ;EAExC,MAAM,OAAO,MAAM,SAAS,KAAK;EACjC,OAAO,uBAAuB,cAAc,IAAI,CAAC;CACnD;CAEA,MAAM,qBAAqB,QAA6C;EACtE,MAAM,SAAS,MAAM,KAAK,UACxB,GAAG,KAAK,YAAY,EAAE,qBACtB,WAAW,MAAM,CACnB;EACA,OAAO,uBAAuB,MAAM;CACtC;CAIA,MAAM,YACJ,SACA,eACA,UACA,QAC+B;EAC/B,MAAM,OAAgC,EACpC,SAAS,cAAc,OAAO,EAChC;EACA,IAAI,eAAe,KAAK,gBAAgB;EACxC,IAAI,UAAU,KAAK,WAAW;EAE9B,MAAM,SAAS,MAAM,KAAK,UACxB,GAAG,KAAK,YAAY,EAAE,gBACtB;GACE,QAAQ;GACR,MAAM,KAAK,UAAU,IAAI;GACzB,GAAG,WAAW,MAAM;EACtB,CACF;EAEA,OAAO,yBAAyB,MAAM;CACxC;CAEA,OAAO,cACL,SACA,eACA,UACA,QACgC;EAChC,MAAM,UAAU,MAAM,KAAK,WAAW,IAAI;EAC1C,QAAQ,SAAS;EAEjB,MAAM,OAAgC,EACpC,SAAS,cAAc,OAAO,EAChC;EACA,IAAI,eAAe,KAAK,gBAAgB;EACxC,IAAI,UAAU,KAAK,WAAW;EAE9B,MAAM,WAAW,MAAM,MACrB,GAAG,KAAK,UAAU,KAAK,YAAY,EAAE,kBACrC;GACE,GAAG,KAAK;GACR,QAAQ;GACR;GACA,MAAM,KAAK,UAAU,IAAI;GACzB,GAAG,WAAW,MAAM;EACtB,CACF;EACA,IAAI,CAAC,SAAS,IACZ,MAAM,KAAK,mBAAmB,QAAQ;EAGxC,OAAO,KAAK,SAAS,QAAQ;CAC/B;CAIA,MAAM,QACJ,QACA,eACA,QACkB;EAClB,MAAM,SAAS,IAAI,gBAAgB;EACnC,IAAI,kBAAkB,KAAA,GAEpB,OAAO,IAAI,kBAAkB,OAAO,aAAa,CAAC;EAEpD,MAAM,KAAK,OAAO,SAAS;EAC3B,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,IAAI,KAAK,IAAI,OAAO,MAC5E,WAAW,MAAM,CACnB;CACF;CAEA,MAAM,UACJ,SACA,QAC+B;EAC/B,MAAM,SAAS,IAAI,gBAAgB;EACnC,IAAI,SAAS,WAAW,OAAO,IAAI,cAAc,QAAQ,SAAS;EAClE,IAAI,SAAS,QAAQ,OAAO,IAAI,UAAU,gBAAgB,QAAQ,MAAM,CAAC;EACzE,IAAI,SAAS,aAAa,KAAA,GACxB,OAAO,IAAI,aAAa,OAAO,QAAQ,QAAQ,CAAC;EAClD,IAAI,SAAS,WAAW,OAAO,IAAI,cAAc,QAAQ,SAAS;EAClE,IAAI,SAAS,kBAAkB,KAAA,GAC7B,OAAO,IAAI,kBAAkB,OAAO,QAAQ,aAAa,CAAC;EAC5D,IAAI,SAAS,sBACX,OAAO,IAAI,0BAA0B,QAAQ,oBAAoB;EACnE,IAAI,SAAS,qBAAqB,KAAA,GAChC,OAAO,IAAI,qBAAqB,OAAO,QAAQ,gBAAgB,CAAC;EAClE,MAAM,KAAK,OAAO,SAAS;EAC3B,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,QAAQ,KAAK,IAAI,OAAO,MAC9C,WAAW,MAAM,CACnB;CACF;CAEA,MAAM,WACJ,QACA,UACA,QACkB;EAClB,MAAM,OAAO,WAAW,EAAE,SAAS,IAAI,CAAC;EACxC,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,UAC1D;GACE,QAAQ;GACR,MAAM,KAAK,UAAU,IAAI;GACzB,GAAG,WAAW,MAAM;EACtB,CACF;CACF;CAEA,OAAO,gBACL,QACA,QACgC;EAChC,MAAM,UAAU,MAAM,KAAK,WAAW,KAAK;EAC3C,QAAQ,SAAS;EAEjB,MAAM,WAAW,MAAM,MACrB,GAAG,KAAK,UAAU,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,aACzE;GACE,GAAG,KAAK;GACR;GACA,GAAG,WAAW,MAAM;EACtB,CACF;EACA,IAAI,CAAC,SAAS,IACZ,MAAM,KAAK,mBAAmB,QAAQ;EAGxC,OAAO,KAAK,SAAS,QAAQ;CAC/B;CAIA,MAAM,iCACJ,QACA,QACwC;EACxC,MAAM,SAAS,OAAO;EACtB,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,oBAAoB;EACjD,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,2BAC1D;GACE,QAAQ;GACR,MAAM,KAAK,UAAU,MAAM;GAC3B,GAAG,WAAW,MAAM;EACtB,CACF;CACF;CAEA,MAAM,8BACJ,QACA,UACA,QACwC;EACxC,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,2BAA2B,mBAAmB,QAAQ,KAChH,WAAW,MAAM,CACnB;CACF;CAEA,MAAM,gCACJ,QACA,SACA,QACqD;EACrD,MAAM,SAAS,IAAI,gBAAgB;EACnC,IAAI,SAAS,aAAa,KAAA,GACxB,OAAO,IAAI,aAAa,OAAO,QAAQ,QAAQ,CAAC;EAClD,IAAI,SAAS,WAAW,OAAO,IAAI,cAAc,QAAQ,SAAS;EAClE,MAAM,KAAK,OAAO,SAAS;EAC3B,OAAO,KAAK,UACV,GAAG,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,0BAA0B,KAAK,IAAI,OAAO,MACpG,WAAW,MAAM,CACnB;CACF;CAEA,MAAM,iCACJ,QACA,UACA,QACe;EAEf,MAAM,UAAU,MAAM,KAAK,WAAW,IAAM;EAC5C,MAAM,WAAW,MAAM,MACrB,GAAG,KAAK,UAAU,KAAK,YAAY,EAAE,SAAS,mBAAmB,MAAM,EAAE,2BAA2B,mBAAmB,QAAQ,KAC/H;GACE,GAAG,KAAK;GACR,QAAQ;GACR;GACA,GAAG,WAAW,MAAM;EACtB,CACF;EACA,IAAI,CAAC,SAAS,IACZ,MAAM,KAAK,mBAAmB,QAAQ;CAE1C;CAIA,OAAe,SAAS,UAAoD;EAC1E,MAAM,cAAc,SAAS,QAAQ,IAAI,cAAc;EAEvD,IADkB,aAAa,MAAM,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,YAAY,MAClD,qBAAqB;GACrC,MAAM,WAAW,cACb,IAAI,YAAY,KAChB;GACJ,MAAM,IAAI,MACR,2EAA2E,UAC7E;EACF;EAEA,MAAM,SAAS,SAAS,MAAM,UAAU;EACxC,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,kBAAkB;EAE/C,MAAM,UAAU,IAAI,YAAY;EAChC,MAAM,aAAa,IAAI,gBAAgB;EAEvC,MAAM,aAAa,UAA2C;GAC5D,IAAI;IACF,IAAI,SAAS,KAAK,MAAM,MAAM,IAAI;IAElC,IACE,UACA,OAAO,WAAW,YAClB,aAAa,UACb,YAAY,QAEZ,SAAS,OAAO;IAIlB,OAAO,2BADY,cAAc,MACU,CAAC;GAC9C,QAAQ;IACN,OAAO;GACT;EACF;EAEA,IAAI;GACF,OAAO,MAAM;IACX,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,KAAK;IAC1C,IAAI,MAAM;KACR,KAAK,MAAM,SAAS,WAAW,KAAK,QAAQ,OAAO,CAAC,GAAG;MACrD,MAAM,SAAS,UAAU,KAAK;MAC9B,IAAI,QAAQ,MAAM;KACpB;KACA;IACF;IAEA,KAAK,MAAM,SAAS,WAAW,KAC7B,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC,CACxC,GAAG;KACD,MAAM,SAAS,UAAU,KAAK;KAC9B,IAAI,QAAQ,MAAM;IACpB;GACF;EACF,UAAU;GACR,OAAO,YAAY;EACrB;CACF;AACF"}

package/dist/A2AThreadRuntimeCore.d.ts

@@ -1,6 +1,6 @@
 import { A2AAgentCard, A2AArtifact, A2ASendMessageConfiguration, A2ATask } from "./types.js";
 import { A2AClient } from "./A2AClient.js";
-import { AppendMessage, AssistantRuntime, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
+import { AppendMessage, AssistantRuntime, ExportedMessageRepository, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
 //#region src/A2AThreadRuntimeCore.d.ts
 type A2AThreadRuntimeCoreOptions = {
   client: A2AClient;
@@ -22,7 +22,8 @@ declare class A2AThreadRuntimeCore {
   private history;
   private readonly notifyUpdate;
   private runtime;
-  private messages;
+  private readonly repository;
+  private exportedRepository;
   private isRunningFlag;
   private abortController;
   private pendingError;
@@ -39,6 +40,15 @@ declare class A2AThreadRuntimeCore {
   detachRuntime(): void;
   getRuntime(): AssistantRuntime | undefined;
   getMessages(): readonly ThreadMessage[];
+  getMessageRepository(): ExportedMessageRepository;
+  private tryGetMessage;
+  private tryGetMessages;
+  private hasMessage;
+  private addOrUpdateMessage;
+  private switchToBranch;
+  private resetRepositoryHead;
+  private clearRepository;
+  private updateMessage;
   getTask(): A2ATask | undefined;
   getArtifacts(): readonly A2AArtifact[];
   getAgentCard(): A2AAgentCard | undefined;
@@ -51,7 +61,10 @@ declare class A2AThreadRuntimeCore {
     runConfig?: Record<string, unknown>;
   }): Promise<void>;
   cancel(): Promise<void>;
+  private appendLinearChain;
+  private finalizeExternalApply;
   applyExternalMessages(messages: readonly ThreadMessage[]): void;
+  private applyExternalMessageRepository;
   private startRun;
   private runStreaming;
   private runSync;
@@ -67,7 +80,6 @@ declare class A2AThreadRuntimeCore {
   private getAssistantStatus;
   private setRunning;
   private finishRun;
-  private resetHead;
   private recordHistoryEntry;
   private markPendingAssistantHistory;
   private persistAssistantHistory;

package/dist/A2AThreadRuntimeCore.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"A2AThreadRuntimeCore.d.ts","names":[],"sources":["../src/A2AThreadRuntimeCore.ts"],"mappings":";;;;KA6BY;EACV,QAAQ;EACR;EACA,gBAAgB;EAChB,YAAY,OAAO;EACnB;EACA,uBAAuB,UAAU;EACjC,UAAU;EACV;;cAQW;UACH;UACA;UACA;UACA;UACA;UACA;UACA;mBACS;UAET;UACA;UACA;UACA;UACA;UAGA;UACA;UACA;mBAGS;mBACA;UACT;UACA;cAEI,SAAS;EAWrB,cAAc,SAAS,KAAK;EAU5B,cAAc,SAAS;EAIvB;EASA,cAAc;EAId,wBAAwB;EAIxB,WAAW;EAIX,yBAAyB;EAIzB,gBAAgB;EAIhB;MAII;EAIJ,mBAAmB;EAgCb,OAAO,SAAS,gBAAgB;EAsBhC,KAAK,SAAS,gBAAgB;EAI9B,OACJ,yBACA;IAAW,YAAY;MACtB;EAaG,UAAU;EAiBhB,sBAAsB,mBAAmB;UAe3B;UAwEA;UA4BA;UA4BN;UAiBA;UA6BA;UAqCA;UAQA;UAuBA;UAiCA;UAqBA;UAWA;UAgBA;UASA;UAKA;UAOA;UAcA;UAIA;UAQA;UAeA"}
+{"version":3,"file":"A2AThreadRuntimeCore.d.ts","names":[],"sources":["../src/A2AThreadRuntimeCore.ts"],"mappings":";;;;KAgCY;EACV,QAAQ;EACR;EACA,gBAAgB;EAChB,YAAY,OAAO;EACnB;EACA,uBAAuB,UAAU;EACjC,UAAU;EACV;;cAQW;UACH;UACA;UACA;UACA;UACA;UACA;UACA;mBACS;UAET;mBACS;UACT;UACA;UACA;UACA;UAGA;UACA;UACA;mBAGS;mBACA;UACT;UACA;EAEI,YAAA,SAAS;EAWrB,cAAc,SAAS,KAAK;EAU5B,cAAc,SAAS;EAIvB;EASA,cAAc;EAId,wBAAwB;EAIxB,wBAAwB;UAKhB;UAQA;UAUA;UAIA;UAQA;UAKA;UAKA;UAKA;EAYR,WAAW;EAIX,yBAAyB;EAIzB,gBAAgB;EAIhB;MAII;EAIJ,mBAAmB;EA+Bb,OAAO,SAAS,gBAAgB;EAuBhC,KAAK,SAAS,gBAAgB;EAI9B,OACJ,yBACA;IAAW,YAAY;MACtB;EAaG,UAAU;UAiBR;UAgBA;EAWR,sBAAsB,mBAAmB;UAiCjC;UA6DM;UAwEA;UA4BA;UA4BN;UAiBA;UA6BA;UAqCA;UAQA;UAuBA;UAmCA;UAsBA;UAUA;UAaA;UAQA;UAKA;UASA;UAIA;UAQA;UAeA"}

package/dist/A2AThreadRuntimeCore.js

@@ -1,5 +1,6 @@
 "use client";
-import { a2aMessageToContent, isTerminalTaskState, taskStateToMessageStatus } from "./conversions.js";
+import { a2aMessageToContent, contentPartsToA2AParts, isTerminalTaskState, taskStateToMessageStatus } from "./conversions.js";
+import { MessageRepository } from "@assistant-ui/core/internal";
 import { fromThreadMessageLike, generateId } from "@assistant-ui/core";
 //#region src/A2AThreadRuntimeCore.ts
 const FALLBACK_USER_STATUS = {
@@ -16,7 +17,8 @@ var A2AThreadRuntimeCore = class {
 	history;
 	notifyUpdate;
 	runtime;
-	messages = [];
+	repository = new MessageRepository();
+	exportedRepository;
 	isRunningFlag = false;
 	abortController = null;
 	pendingError = null;
@@ -60,7 +62,52 @@ var A2AThreadRuntimeCore = class {
 		return this.runtime;
 	}
 	getMessages() {
-		return this.messages;
+		return this.repository.getMessages();
+	}
+	getMessageRepository() {
+		this.exportedRepository ??= this.repository.export();
+		return this.exportedRepository;
+	}
+	tryGetMessage(messageId) {
+		try {
+			return this.repository.getMessage(messageId);
+		} catch {
+			return;
+		}
+	}
+	tryGetMessages(messageId) {
+		try {
+			return this.repository.getMessages(messageId);
+		} catch {
+			return;
+		}
+	}
+	hasMessage(messageId) {
+		return this.tryGetMessage(messageId) !== void 0;
+	}
+	addOrUpdateMessage(parentId, message) {
+		this.repository.addOrUpdateMessage(parentId, message);
+		this.exportedRepository = void 0;
+	}
+	switchToBranch(messageId) {
+		this.repository.switchToBranch(messageId);
+		this.exportedRepository = void 0;
+	}
+	resetRepositoryHead(messageId) {
+		this.repository.resetHead(messageId);
+		this.exportedRepository = void 0;
+	}
+	clearRepository() {
+		this.repository.clear();
+		this.exportedRepository = void 0;
+	}
+	updateMessage(messageId, updater) {
+		const item = this.tryGetMessage(messageId);
+		if (!item) return false;
+		const message = updater(item.message);
+		if (message === item.message) return false;
+		this.addOrUpdateMessage(item.parentId, message);
+		return true;
 	}
 	getTask() {
 		return this.currentTask;
@@ -84,10 +131,7 @@ var A2AThreadRuntimeCore = class {
 		const agentCardPromise = this.client.getAgentCard().catch(() => void 0);
 		this._loadPromise = Promise.all([historyPromise, agentCardPromise]).then(([repo, agentCard]) => {
 			if (agentCard) this.agentCardValue = agentCard;
-			if (repo) {
-				const messages = repo.messages.map((item) => item.message);
-				this.applyExternalMessages(messages);
-			}
+			if (repo) this.applyExternalMessageRepository(repo);
 		}).catch((error) => {
 			this.onError?.(error instanceof Error ? error : new Error(String(error)));
 		}).finally(() => {
@@ -99,12 +143,12 @@ var A2AThreadRuntimeCore = class {
 	}
 	async append(message) {
 		const startRun = message.startRun ?? message.role === "user";
-		if (message.sourceId) this.messages = this.messages.filter((entry) => entry.id !== message.sourceId);
-		this.resetHead(message.parentId);
 		const threadMessage = fromThreadMessageLike(message, generateId(), FALLBACK_USER_STATUS);
-		this.messages = [...this.messages, threadMessage];
+		const parentId = message.parentId === null ? null : message.parentId && this.hasMessage(message.parentId) ? message.parentId : this.repository.headId;
+		this.addOrUpdateMessage(parentId, threadMessage);
+		this.switchToBranch(threadMessage.id);
 		this.notifyUpdate();
-		this.recordHistoryEntry(message.parentId ?? null, threadMessage);
+		this.recordHistoryEntry(parentId, threadMessage);
 		if (!startRun) return;
 		await this.startRun(threadMessage);
 	}
@@ -112,10 +156,9 @@ var A2AThreadRuntimeCore = class {
 		await this.append(message);
 	}
 	async reload(parentId, _config = {}) {
-		this.resetHead(parentId);
-		this.notifyUpdate();
-		for (let i = this.messages.length - 1; i >= 0; i--) if (this.messages[i].role === "user") {
-			await this.startRun(this.messages[i]);
+		const messages = parentId === null ? [] : this.tryGetMessages(parentId) ?? this.getMessages();
+		for (let i = messages.length - 1; i >= 0; i--) if (messages[i].role === "user") {
+			await this.startRun(messages[i]);
 			return;
 		}
 	}
@@ -127,15 +170,101 @@ var A2AThreadRuntimeCore = class {
 			this.currentTask = updated;
 		} catch {}
 	}
-	applyExternalMessages(messages) {
+	appendLinearChain(messages) {
+		let parentId = null;
+		let lastId = null;
+		const seen = /* @__PURE__ */ new Set();
+		for (const message of messages) {
+			if (seen.has(message.id)) continue;
+			seen.add(message.id);
+			this.addOrUpdateMessage(parentId, message);
+			parentId = message.id;
+			lastId = message.id;
+		}
+		return lastId;
+	}
+	finalizeExternalApply() {
 		this.assistantHistoryParents.clear();
-		this.messages = [...messages];
 		this.recordedHistoryIds.clear();
-		for (const message of this.messages) this.recordedHistoryIds.add(message.id);
+		for (const { message } of this.getMessageRepository().messages) this.recordedHistoryIds.add(message.id);
 		this.currentTask = void 0;
 		this.currentArtifacts = [];
 		this.notifyUpdate();
 	}
+	applyExternalMessages(messages) {
+		if (messages.length === 0) this.clearRepository();
+		else {
+			let expectedParentId = null;
+			let lastAppliedId = null;
+			let hardReplace = false;
+			const seen = /* @__PURE__ */ new Set();
+			for (const message of messages) {
+				if (seen.has(message.id)) continue;
+				seen.add(message.id);
+				const existing = this.tryGetMessage(message.id);
+				if (existing && existing.parentId !== expectedParentId) {
+					hardReplace = true;
+					break;
+				}
+				this.addOrUpdateMessage(expectedParentId, message);
+				expectedParentId = message.id;
+				lastAppliedId = message.id;
+			}
+			if (hardReplace) {
+				this.clearRepository();
+				lastAppliedId = this.appendLinearChain(messages);
+			}
+			this.resetRepositoryHead(lastAppliedId);
+		}
+		this.finalizeExternalApply();
+	}
+	applyExternalMessageRepository(loaded) {
+		const headId = loaded.headId ?? loaded.messages.at(-1)?.message.id ?? null;
+		const ids = /* @__PURE__ */ new Set();
+		let degenerate = false;
+		for (const { message } of loaded.messages) {
+			if (ids.has(message.id)) {
+				degenerate = true;
+				break;
+			}
+			ids.add(message.id);
+		}
+		if (headId !== null && !ids.has(headId)) degenerate = true;
+		if (!degenerate) {
+			this.clearRepository();
+			let pending = [...loaded.messages];
+			const importedIds = /* @__PURE__ */ new Set();
+			while (pending.length > 0) {
+				const unresolved = [];
+				let progressed = false;
+				for (const item of pending) {
+					if (item.parentId !== null && !importedIds.has(item.parentId)) {
+						unresolved.push(item);
+						continue;
+					}
+					this.addOrUpdateMessage(item.parentId, item.message);
+					importedIds.add(item.message.id);
+					progressed = true;
+				}
+				if (!progressed) {
+					degenerate = true;
+					break;
+				}
+				pending = unresolved;
+			}
+		}
+		if (degenerate) {
+			this.clearRepository();
+			let previousId = null;
+			for (const { message } of loaded.messages) {
+				const existing = this.tryGetMessage(message.id);
+				this.addOrUpdateMessage(existing ? existing.parentId : previousId, message);
+				previousId = message.id;
+			}
+			this.resetRepositoryHead(previousId);
+		} else this.resetRepositoryHead(headId);
+		this.finalizeExternalApply();
+	}
 	async startRun(userThreadMessage) {
 		if (this.abortController) {
 			this.abortController.abort();
@@ -145,7 +274,7 @@ var A2AThreadRuntimeCore = class {
 		if (this.currentTask && isTerminalTaskState(this.currentTask.status.state)) this.currentTask = void 0;
 		this.currentArtifacts = [];
 		const assistantParentId = userThreadMessage.id;
-		const assistantId = this.insertAssistantPlaceholder();
+		const assistantId = this.insertAssistantPlaceholder(assistantParentId);
 		this.markPendingAssistantHistory(assistantId, assistantParentId);
 		const abortController = new AbortController();
 		this.abortController = abortController;
@@ -291,11 +420,8 @@ var A2AThreadRuntimeCore = class {
 	threadMessageToA2AMessage(message) {
 		const parts = [];
 		if (message.role === "user") {
-			for (const part of message.content) if (part.type === "text") parts.push({ text: part.text });
-			else if (part.type === "image") parts.push({
-				url: part.image,
-				mediaType: "image/*"
-			});
+			parts.push(...contentPartsToA2AParts(message.content));
+			for (const attachment of message.attachments ?? []) parts.push(...contentPartsToA2AParts(attachment.content ?? [], attachment.contentType));
 		}
 		const a2aMsg = {
 			messageId: message.id,
@@ -306,7 +432,7 @@ var A2AThreadRuntimeCore = class {
 		if (this.currentTask?.id && !isTerminalTaskState(this.currentTask.status.state)) a2aMsg.taskId = this.currentTask.id;
 		return a2aMsg;
 	}
-	insertAssistantPlaceholder() {
+	insertAssistantPlaceholder(parentId) {
 		const id = generateId();
 		const assistant = {
 			id,
@@ -322,13 +448,14 @@ var A2AThreadRuntimeCore = class {
 				custom: {}
 			}
 		};
-		this.messages = [...this.messages, assistant];
+		this.addOrUpdateMessage(parentId, assistant);
+		this.switchToBranch(id);
 		this.notifyUpdate();
 		return id;
 	}
 	updateAssistantContent(messageId, content) {
-		this.messages = this.messages.map((message) => {
-			if (message.id !== messageId || message.role !== "assistant") return message;
+		this.updateMessage(messageId, (message) => {
+			if (message.role !== "assistant") return message;
 			return {
 				...message,
 				content
@@ -336,22 +463,21 @@ var A2AThreadRuntimeCore = class {
 		});
 	}
 	updateAssistantStatus(messageId, status) {
-		let touched = false;
-		this.messages = this.messages.map((message) => {
-			if (message.id !== messageId || message.role !== "assistant") return message;
-			touched = true;
+		if (this.updateMessage(messageId, (message) => {
+			if (message.role !== "assistant") return message;
 			return {
 				...message,
 				status
 			};
-		});
-		if (touched) {
+		})) {
 			this.notifyUpdate();
 			if (status.type === "complete" || status.type === "incomplete") this.persistAssistantHistory(messageId);
 		}
 	}
 	getAssistantStatus(messageId) {
-		return this.messages.find((m) => m.id === messageId && m.role === "assistant")?.status;
+		const msg = this.tryGetMessage(messageId)?.message;
+		if (msg?.role !== "assistant") return void 0;
+		return msg.status;
 	}
 	setRunning(running) {
 		this.isRunningFlag = running;
@@ -361,15 +487,6 @@ var A2AThreadRuntimeCore = class {
 		if (this.abortController === controller) this.abortController = null;
 		this.setRunning(false);
 	}
-	resetHead(parentId) {
-		if (!parentId) {
-			if (this.messages.length) this.messages = [];
-			return;
-		}
-		const idx = this.messages.findIndex((message) => message.id === parentId);
-		if (idx === -1) return;
-		this.messages = this.messages.slice(0, idx + 1);
-	}
 	recordHistoryEntry(parentId, message) {
 		this.appendHistoryItem(parentId, message);
 	}
@@ -381,7 +498,7 @@ var A2AThreadRuntimeCore = class {
 		if (!this.history) return;
 		const parentId = this.assistantHistoryParents.get(messageId);
 		if (parentId === void 0) return;
-		const message = this.messages.find((m) => m.id === messageId);
+		const message = this.tryGetMessage(messageId)?.message;
 		if (!message || message.role !== "assistant") return;
 		if (message.status?.type !== "complete" && message.status?.type !== "incomplete") return;
 		this.assistantHistoryParents.delete(messageId);