@assistant-ui/react-a2a 0.2.35 → 0.2.36

package/dist/A2AClient.d.ts

@@ -1,6 +1,6 @@
 import { A2AAgentCard, A2AErrorInfo, A2AListTaskPushNotificationConfigsResponse, A2AListTasksRequest, A2AListTasksResponse, A2AMessage, A2ASendMessageConfiguration, A2AStreamEvent, A2ATask, A2ATaskPushNotificationConfig } from "./types.js";
 //#region src/A2AClient.d.ts
-type A2AClientOptions = {
+export type A2AClientOptions = {
   baseUrl: string;
   /** Optional path prefix for all API endpoints (e.g. "/v1"). Does not affect agent card discovery. */
   basePath?: string | undefined;
@@ -12,13 +12,13 @@ type A2AClientOptions = {
   /** Extra fetch options applied to every request. */
   fetchOptions?: Omit<RequestInit, "headers" | "body" | "method" | "signal"> | undefined;
 };
-declare class A2AError extends Error {
+export declare class A2AError extends Error {
   code: number;
   status: string;
   details: unknown[] | undefined;
   constructor(info: A2AErrorInfo);
 }
-declare class A2AClient {
+export declare class A2AClient {
   private baseUrl;
   private basePath;
   private tenant;
@@ -48,5 +48,4 @@ declare class A2AClient {
   private parseSSE;
 }
 //#endregion
-export { A2AClient, A2AClientOptions, A2AError };
 //# sourceMappingURL=A2AClient.d.ts.map

package/dist/A2AClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"A2AClient.d.ts","names":[],"sources":["../src/A2AClient.ts"],"mappings":";;KAmBY;EACV;;EAEA;;EAEA;EACA,UACI,gCACO,yBAAyB,QAAQ;;EAG5C;;EAEA,eACI,KAAK;;cAIE,iBAAiB;EAC5B;EACA;EACA;EAEY,YAAA,MAAM;;cA2oBP;UACH;UACA;UACA;UACA;UACA;UAIA;EAII,YAAA,SAAS;UAkBb;UAIM;UAqBA;UAyBA;EAoCR,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;EAcL,UACJ,UAAU,qBACV,SAAS,cACR,QAAQ;EAqBL,WACJ,gBACA,WAAW,yBACX,SAAS,cACR,QAAQ;EAaJ,gBACL,gBACA,SAAS,cACR,eAAe;EAqBZ,iCACJ,QAAQ,+BACR,SAAS,cACR,QAAQ;EAiBL,8BACJ,gBACA,kBACA,SAAS,cACR,QAAQ;EAWL,gCACJ,gBACA;IAAY;IAAmB;KAC/B,SAAS,cACR,QAAQ;EAaL,iCACJ,gBACA,kBACA,SAAS,cACR;UAmBY"}
+{"version":3,"file":"A2AClient.d.ts","names":[],"sources":["../src/A2AClient.ts"],"mappings":";;YAmBY;EACV;;EAEA;;EAEA;EACA,UACI,gCACO,yBAAyB,QAAQ;;EAG5C;;EAEA,eACI,KAAK;;qBAIE,iBAAiB;EAC5B;EACA;EACA;EAEY,YAAA,MAAM;;qBA2oBP;UACH;UACA;UACA;UACA;UACA;UAIA;EAII,YAAA,SAAS;UAkBb;UAIM;UAqBA;UAyBA;EAoCR,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;EAcL,UACJ,UAAU,qBACV,SAAS,cACR,QAAQ;EAqBL,WACJ,gBACA,WAAW,yBACX,SAAS,cACR,QAAQ;EAaJ,gBACL,gBACA,SAAS,cACR,eAAe;EAqBZ,iCACJ,QAAQ,+BACR,SAAS,cACR,QAAQ;EAiBL,8BACJ,gBACA,kBACA,SAAS,cACR,QAAQ;EAWL,gCACJ,gBACA;IAAY;IAAmB;KAC/B,SAAS,cACR,QAAQ;EAaL,iCACJ,gBACA,kBACA,SAAS,cACR;UAmBY"}

package/dist/A2AThreadRuntimeCore.d.ts

@@ -2,7 +2,7 @@ import { A2AAgentCard, A2AArtifact, A2ASendMessageConfiguration, A2ATask } from
 import { A2AClient } from "./A2AClient.js";
 import { AppendMessage, AssistantRuntime, ExportedMessageRepository, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
 //#region src/A2AThreadRuntimeCore.d.ts
-type A2AThreadRuntimeCoreOptions = {
+export type A2AThreadRuntimeCoreOptions = {
   client: A2AClient;
   contextId?: string | undefined;
   configuration?: A2ASendMessageConfiguration | undefined;
@@ -12,7 +12,7 @@ type A2AThreadRuntimeCoreOptions = {
   history?: ThreadHistoryAdapter | undefined;
   notifyUpdate: () => void;
 };
-declare class A2AThreadRuntimeCore {
+export declare class A2AThreadRuntimeCore {
   private client;
   private contextId;
   private configuration;
@@ -25,6 +25,7 @@ declare class A2AThreadRuntimeCore {
   private readonly session;
   private isRunningFlag;
   private abortController;
+  private runGeneration;
   private pendingError;
   private currentTask;
   private currentArtifacts;
@@ -33,6 +34,7 @@ declare class A2AThreadRuntimeCore {
   private readonly recordedHistoryIds;
   private _isLoading;
   private _loadPromise;
+  private _historyLoadGeneration;
   private _loadRequested;
   private _agentCardPromise;
   private _agentCardRetryAfter;
@@ -86,5 +88,4 @@ declare class A2AThreadRuntimeCore {
   private appendHistoryItem;
 }
 //#endregion
-export { A2AThreadRuntimeCore, A2AThreadRuntimeCoreOptions };
 //# sourceMappingURL=A2AThreadRuntimeCore.d.ts.map

package/dist/A2AThreadRuntimeCore.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"A2AThreadRuntimeCore.d.ts","names":[],"sources":["../src/A2AThreadRuntimeCore.ts"],"mappings":";;;;KAsCY;EACV,QAAQ;EACR;EACA,gBAAgB;EAChB,YAAY,OAAO;EACnB;EACA,uBAAuB,UAAU;EACjC,UAAU;EACV;;cAyBW;UACH;UACA;UACA;UACA;UACA;UACA;UACA;mBACS;UAET;mBACS;UACT;UACA;UACA;UAGA;UACA;UACA;mBAGS;mBACA;UACT;UACA;UACA;UACA;UACA;UACA;UACA;UAEA;EAEI,YAAA,SAAS;EAYrB,cAAc,SAAS,KAAK;;;EA8B5B;EAcA,cAAc,SAAS;EAIvB;EASA,cAAc;EAId,wBAAwB;EAIxB,wBAAwB;EAIxB,WAAW;EAIX,yBAAyB;EAIzB,gBAAgB;EAIhB;MAII;UAII;UAwBM;EAoBd,mBAAmB;EAkCb,OAAO,SAAS,gBAAgB;EAuBhC,KAAK,SAAS,gBAAgB;EAI9B,OACJ,yBACA;IAAW,YAAY;MACtB;EAaG,UAAU;UAiBR;UAgBA;EAWR,sBAAsB,mBAAmB;UAmC3B;UA+EA;UA4CA;UA4BN;UAiBA;UA6BA;UA0CA;UAQA;UAuCA;UAsBA;UAUA;UAaA;UAQA;UAKA;UAQA;UAIA;UAQA;UAeA"}
+{"version":3,"file":"A2AThreadRuntimeCore.d.ts","names":[],"sources":["../src/A2AThreadRuntimeCore.ts"],"mappings":";;;;YAsCY;EACV,QAAQ;EACR;EACA,gBAAgB;EAChB,YAAY,OAAO;EACnB;EACA,uBAAuB,UAAU;EACjC,UAAU;EACV;;qBAyBW;UACH;UACA;UACA;UACA;UACA;UACA;UACA;mBACS;UAET;mBACS;UACT;UACA;UACA;UACA;UAGA;UACA;UACA;mBAGS;mBACA;UACT;UACA;UACA;UACA;UACA;UACA;UACA;UACA;UAEA;EAEI,YAAA,SAAS;EAYrB,cAAc,SAAS,KAAK;;;EA8B5B;EAgBA,cAAc,SAAS;EAIvB;EASA,cAAc;EAId,wBAAwB;EAIxB,wBAAwB;EAIxB,WAAW;EAIX,yBAAyB;EAIzB,gBAAgB;EAIhB;MAII;UAII;UAwBM;EAoBd,mBAAmB;EAsCb,OAAO,SAAS,gBAAgB;EAuBhC,KAAK,SAAS,gBAAgB;EAI9B,OACJ,yBACA;IAAW,YAAY;MACtB;EAaG,UAAU;UA6BR;UAgBA;EAWR,sBAAsB,mBAAmB;UAmC3B;UAiFA;UA4CA;UA4BN;UAiBA;UA6BA;UA0CA;UAQA;UAuCA;UAsBA;UAUA;UAaA;UAQA;UAKA;UAQA;UAIA;UAQA;UAeA"}

package/dist/A2AThreadRuntimeCore.js

@@ -31,6 +31,7 @@ var A2AThreadRuntimeCore = class {
 	session = createMessageRepositorySession();
 	isRunningFlag = false;
 	abortController = null;
+	runGeneration = 0;
 	pendingError = null;
 	currentTask;
 	currentArtifacts = [];
@@ -39,6 +40,7 @@ var A2AThreadRuntimeCore = class {
 	recordedHistoryIds = /* @__PURE__ */ new Set();
 	_isLoading = false;
 	_loadPromise;
+	_historyLoadGeneration = 0;
 	_loadRequested = false;
 	_agentCardPromise;
 	_agentCardRetryAfter = 0;
@@ -73,6 +75,8 @@ var A2AThreadRuntimeCore = class {
 	/** Thread-boundary reset: applyExternalMessages alone also serves branch
 	* switches, deletes, and cancel resyncs, which must keep the live context. */
 	resetContext() {
+		this._historyLoadGeneration++;
+		this._isLoading = false;
 		const controller = this.abortController;
 		this.contextId = this.lastOptionsContextId;
 		if (controller) {
@@ -153,15 +157,19 @@ var A2AThreadRuntimeCore = class {
 		if (this._loadPromise) return this._loadPromise;
 		if (!this.history) return agentCardPromise;
 		this._isLoading = true;
+		const generation = this._historyLoadGeneration;
 		const historyPromise = this.history.load();
 		this._loadPromise = Promise.all([historyPromise, agentCardPromise]).then(([repo]) => {
+			if (generation !== this._historyLoadGeneration) return;
 			if (repo) {
 				this.session.applyExternalMessageRepository(repo);
 				this.finalizeExternalApply();
 			}
 		}).catch((error) => {
+			if (generation !== this._historyLoadGeneration) return;
 			invokeRuntimeCallback("onError", this.onError, error instanceof Error ? error : new Error(String(error)));
 		}).finally(() => {
+			if (generation !== this._historyLoadGeneration) return;
 			this._isLoading = false;
 			this.notifyUpdate();
 		});
@@ -191,10 +199,12 @@ var A2AThreadRuntimeCore = class {
 	}
 	async cancel() {
 		if (!this.abortController) return;
+		const task = this.currentTask;
+		const generation = this.runGeneration;
 		this.abortController.abort();
-		if (this.currentTask?.id) try {
-			const updated = await this.client.cancelTask(this.currentTask.id);
-			this.currentTask = updated;
+		if (task?.id) try {
+			const updated = await this.client.cancelTask(task.id);
+			if (this.currentTask === task && this.runGeneration === generation) this.currentTask = updated;
 		} catch {}
 	}
 	appendLinearChain(messages) {
@@ -246,6 +256,7 @@ var A2AThreadRuntimeCore = class {
 		this.finalizeExternalApply();
 	}
 	async startRun(userThreadMessage) {
+		this.runGeneration++;
 		if (this.abortController) {
 			this.abortController.abort();
 			this.abortController = null;

package/dist/A2AThreadRuntimeCore.js.map

@@ -1 +1 @@
-{"version":3,"file":"A2AThreadRuntimeCore.js","names":[],"sources":["../src/A2AThreadRuntimeCore.ts"],"sourcesContent":["\"use client\";\n\nimport { generateId, fromThreadMessageLike } from \"@assistant-ui/core\";\nimport type {\n  AppendMessage,\n  AssistantRuntime,\n  ExportedMessageRepository,\n  MessageStatus,\n  ThreadAssistantMessage,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport {\n  createMessageRepositorySession,\n  invokeUserCallback,\n} from \"@assistant-ui/core/internal\";\nimport type { A2AClient } from \"./A2AClient\";\nimport type {\n  A2AArtifact,\n  A2AAgentCard,\n  A2AMessage,\n  A2ASendMessageConfiguration,\n  A2AStreamEvent,\n  A2ATask,\n  A2ATaskArtifactUpdateEvent,\n  A2ATaskStatusUpdateEvent,\n} from \"./types\";\n\nimport {\n  a2aMessageToContent,\n  isTerminalTaskState,\n  threadMessageToA2AMessage,\n  taskStateToMessageStatus,\n} from \"./conversions\";\n\nconst INITIAL_AGENT_CARD_RETRY_DELAY_MS = 5_000;\nconst MAX_AGENT_CARD_RETRY_DELAY_MS = 5 * 60_000;\n\nexport type A2AThreadRuntimeCoreOptions = {\n  client: A2AClient;\n  contextId?: string | undefined;\n  configuration?: A2ASendMessageConfiguration | undefined;\n  onError?: ((error: Error) => void) | undefined;\n  onCancel?: (() => void) | undefined;\n  onArtifactComplete?: ((artifact: A2AArtifact) => void) | undefined;\n  history?: ThreadHistoryAdapter | undefined;\n  notifyUpdate: () => void;\n};\n\nconst FALLBACK_USER_STATUS = {\n  type: \"complete\",\n  reason: \"unknown\",\n} as const;\n\ntype A2ARuntimeCallbackName = \"onError\" | \"onCancel\" | \"onArtifactComplete\";\n\nconst invokeRuntimeCallback = <TArgs extends readonly unknown[]>(\n  name: A2ARuntimeCallbackName,\n  callback: ((...args: TArgs) => unknown) | undefined,\n  ...args: TArgs\n): void => {\n  void invokeUserCallback(\"react-a2a\", name, callback, ...args);\n};\n\nfunction normalizeArtifact(artifact: A2AArtifact): A2AArtifact {\n  return {\n    ...artifact,\n    parts: Array.isArray(artifact.parts) ? artifact.parts : [],\n  };\n}\n\nexport class A2AThreadRuntimeCore {\n  private client: A2AClient;\n  private contextId: string | undefined;\n  private configuration: A2ASendMessageConfiguration | undefined;\n  private onError: ((error: Error) => void) | undefined;\n  private onCancel: (() => void) | undefined;\n  private onArtifactComplete: ((artifact: A2AArtifact) => void) | undefined;\n  private history: ThreadHistoryAdapter | undefined;\n  private readonly notifyUpdate: () => void;\n\n  private runtime: AssistantRuntime | undefined;\n  private readonly session = createMessageRepositorySession();\n  private isRunningFlag = false;\n  private abortController: AbortController | null = null;\n  private pendingError: Error | null = null;\n\n  // A2A-specific state\n  private currentTask: A2ATask | undefined;\n  private currentArtifacts: A2AArtifact[] = [];\n  private agentCardValue: A2AAgentCard | undefined;\n\n  // History tracking\n  private readonly assistantHistoryParents = new Map<string, string | null>();\n  private readonly recordedHistoryIds = new Set<string>();\n  private _isLoading = false;\n  private _loadPromise: Promise<void> | undefined;\n  private _loadRequested = false;\n  private _agentCardPromise: Promise<void> | undefined;\n  private _agentCardRetryAfter = 0;\n  private _agentCardRetryDelay = INITIAL_AGENT_CARD_RETRY_DELAY_MS;\n  private _agentCardDiscoveryFailed = false;\n\n  private lastOptionsContextId: string | undefined;\n\n  constructor(options: A2AThreadRuntimeCoreOptions) {\n    this.client = options.client;\n    this.contextId = options.contextId;\n    this.lastOptionsContextId = options.contextId;\n    this.configuration = options.configuration;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.onArtifactComplete = options.onArtifactComplete;\n    this.history = options.history;\n    this.notifyUpdate = options.notifyUpdate;\n  }\n\n  updateOptions(options: Omit<A2AThreadRuntimeCoreOptions, \"notifyUpdate\">) {\n    this.client = options.client;\n    // The hook re-applies options on every render, including renders caused\n    // by this core's own notifyUpdate. The option only seeds the context: a\n    // re-render with the same value must not clobber a server-assigned\n    // contextId learned from the stream.\n    if (options.contextId !== this.lastOptionsContextId) {\n      this.contextId = options.contextId;\n      this.lastOptionsContextId = options.contextId;\n    }\n    this.configuration = options.configuration;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.onArtifactComplete = options.onArtifactComplete;\n    const previousHistory = this.history;\n    this.history = options.history;\n\n    if (\n      this._loadRequested &&\n      !this._loadPromise &&\n      !previousHistory &&\n      options.history &&\n      this.session.getMessages().length === 0\n    ) {\n      void this.__internal_load();\n    }\n  }\n\n  /** Thread-boundary reset: applyExternalMessages alone also serves branch\n   * switches, deletes, and cancel resyncs, which must keep the live context. */\n  resetContext(): void {\n    // Restore the seed before aborting: an onCancel callback that starts a\n    // new run must not pick up the old thread's context, and its controller\n    // must not be discarded.\n    const controller = this.abortController;\n    this.contextId = this.lastOptionsContextId;\n    if (controller) {\n      controller.abort();\n      if (this.abortController === controller) {\n        this.abortController = null;\n      }\n    }\n  }\n\n  attachRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  detachRuntime() {\n    this.runtime = undefined;\n    // Abort in-flight requests on unmount\n    if (this.abortController) {\n      this.abortController.abort();\n      this.abortController = null;\n    }\n  }\n\n  getRuntime(): AssistantRuntime | undefined {\n    return this.runtime;\n  }\n\n  getMessages(): readonly ThreadMessage[] {\n    return this.session.getMessages();\n  }\n\n  getMessageRepository(): ExportedMessageRepository {\n    return this.session.export();\n  }\n\n  getTask(): A2ATask | undefined {\n    return this.currentTask;\n  }\n\n  getArtifacts(): readonly A2AArtifact[] {\n    return this.currentArtifacts;\n  }\n\n  getAgentCard(): A2AAgentCard | undefined {\n    return this.agentCardValue;\n  }\n\n  isRunning(): boolean {\n    return this.isRunningFlag;\n  }\n\n  get isLoading(): boolean {\n    return this._isLoading;\n  }\n\n  private loadAgentCard(): Promise<void> {\n    if (Date.now() < this._agentCardRetryAfter) return Promise.resolve();\n\n    this._agentCardPromise ??= this.client.getAgentCard().then(\n      (agentCard) => {\n        this.agentCardValue = agentCard;\n        this._agentCardRetryAfter = 0;\n        this._agentCardRetryDelay = INITIAL_AGENT_CARD_RETRY_DELAY_MS;\n        this._agentCardDiscoveryFailed = false;\n        this.notifyUpdate();\n      },\n      () => {\n        this._agentCardDiscoveryFailed = true;\n        this._agentCardRetryAfter = Date.now() + this._agentCardRetryDelay;\n        this._agentCardRetryDelay = Math.min(\n          this._agentCardRetryDelay * 2,\n          MAX_AGENT_CARD_RETRY_DELAY_MS,\n        );\n        this._agentCardPromise = undefined;\n      },\n    );\n    return this._agentCardPromise;\n  }\n\n  private async waitForAgentCard(signal: AbortSignal): Promise<boolean> {\n    const shouldWait = !this._agentCardDiscoveryFailed;\n    const load = this.loadAgentCard();\n    if (signal.aborted) return false;\n    if (!shouldWait) return true;\n\n    let onAbort!: () => void;\n    const abort = new Promise<void>((resolve) => {\n      onAbort = resolve;\n      signal.addEventListener(\"abort\", onAbort, { once: true });\n    });\n\n    try {\n      await Promise.race([load, abort]);\n    } finally {\n      signal.removeEventListener(\"abort\", onAbort);\n    }\n    return !signal.aborted;\n  }\n\n  __internal_load(): Promise<void> {\n    this._loadRequested = true;\n    const agentCardPromise = this.loadAgentCard();\n\n    if (this._loadPromise) return this._loadPromise;\n    if (!this.history) return agentCardPromise;\n\n    this._isLoading = true;\n\n    const historyPromise = this.history.load();\n\n    this._loadPromise = Promise.all([historyPromise, agentCardPromise])\n      .then(([repo]) => {\n        if (repo) {\n          this.session.applyExternalMessageRepository(repo);\n          this.finalizeExternalApply();\n        }\n      })\n      .catch((error) => {\n        invokeRuntimeCallback(\n          \"onError\",\n          this.onError,\n          error instanceof Error ? error : new Error(String(error)),\n        );\n      })\n      .finally(() => {\n        this._isLoading = false;\n        this.notifyUpdate();\n      });\n\n    this.notifyUpdate();\n    return this._loadPromise;\n  }\n\n  async append(message: AppendMessage): Promise<void> {\n    const startRun = message.startRun ?? message.role === \"user\";\n\n    const threadMessage = fromThreadMessageLike(\n      message as any,\n      generateId(),\n      FALLBACK_USER_STATUS,\n    );\n    const parentId =\n      message.parentId === null\n        ? null\n        : message.parentId && this.session.hasMessage(message.parentId)\n          ? message.parentId\n          : this.session.headId;\n    this.session.addOrUpdateMessage(parentId, threadMessage);\n    this.session.switchToBranch(threadMessage.id);\n    this.notifyUpdate();\n    this.recordHistoryEntry(parentId, threadMessage);\n\n    if (!startRun) return;\n    await this.startRun(threadMessage);\n  }\n\n  async edit(message: AppendMessage): Promise<void> {\n    await this.append(message);\n  }\n\n  async reload(\n    parentId: string | null,\n    _config: { runConfig?: Record<string, unknown> } = {},\n  ): Promise<void> {\n    const messages =\n      parentId === null\n        ? []\n        : (this.session.tryGetMessages(parentId) ?? this.getMessages());\n    for (let i = messages.length - 1; i >= 0; i--) {\n      if (messages[i]!.role === \"user\") {\n        await this.startRun(messages[i]!);\n        return;\n      }\n    }\n  }\n\n  async cancel(): Promise<void> {\n    if (!this.abortController) return;\n\n    // Abort locally first so the stream stops immediately\n    this.abortController.abort();\n\n    // Then try to cancel the task on the server\n    if (this.currentTask?.id) {\n      try {\n        const updated = await this.client.cancelTask(this.currentTask.id);\n        this.currentTask = updated;\n      } catch {\n        // Server cancel failed; local abort already handled\n      }\n    }\n  }\n\n  private appendLinearChain(messages: readonly ThreadMessage[]): string | null {\n    let parentId: string | null = null;\n    let lastId: string | null = null;\n    const seen = new Set<string>();\n\n    for (const message of messages) {\n      if (seen.has(message.id)) continue;\n      seen.add(message.id);\n      this.session.addOrUpdateMessage(parentId, message);\n      parentId = message.id;\n      lastId = message.id;\n    }\n\n    return lastId;\n  }\n\n  private finalizeExternalApply(): void {\n    this.assistantHistoryParents.clear();\n    this.recordedHistoryIds.clear();\n    for (const { message } of this.getMessageRepository().messages) {\n      this.recordedHistoryIds.add(message.id);\n    }\n    this.currentTask = undefined;\n    this.currentArtifacts = [];\n    this.notifyUpdate();\n  }\n\n  applyExternalMessages(messages: readonly ThreadMessage[]): void {\n    if (messages.length === 0) {\n      this.session.clear();\n    } else {\n      let expectedParentId: string | null = null;\n      let lastAppliedId: string | null = null;\n      let hardReplace = false;\n      const seen = new Set<string>();\n\n      for (const message of messages) {\n        if (seen.has(message.id)) continue;\n        seen.add(message.id);\n        const existing = this.session.tryGetMessage(message.id);\n        if (existing && existing.parentId !== expectedParentId) {\n          hardReplace = true;\n          break;\n        }\n        this.session.addOrUpdateMessage(expectedParentId, message);\n        expectedParentId = message.id;\n        lastAppliedId = message.id;\n      }\n\n      if (hardReplace) {\n        this.session.clear();\n        lastAppliedId = this.appendLinearChain(messages);\n      }\n\n      this.session.resetHead(lastAppliedId);\n    }\n\n    this.finalizeExternalApply();\n  }\n\n  // --- Run logic ---\n\n  private async startRun(userThreadMessage: ThreadMessage): Promise<void> {\n    // Cancel any in-progress run before starting a new one\n    if (this.abortController) {\n      this.abortController.abort();\n      this.abortController = null;\n    }\n\n    const a2aMessage = threadMessageToA2AMessage(userThreadMessage, {\n      contextId: this.contextId,\n      taskId:\n        this.currentTask?.id &&\n        !isTerminalTaskState(this.currentTask.status.state)\n          ? this.currentTask.id\n          : undefined,\n    });\n\n    // Clear task if previous task reached terminal state\n    if (\n      this.currentTask &&\n      isTerminalTaskState(this.currentTask.status.state)\n    ) {\n      this.currentTask = undefined;\n    }\n\n    this.currentArtifacts = [];\n\n    const assistantParentId = userThreadMessage.id;\n    const assistantId = this.insertAssistantPlaceholder(assistantParentId);\n    this.markPendingAssistantHistory(assistantId, assistantParentId);\n\n    const abortController = new AbortController();\n    this.abortController = abortController;\n\n    abortController.signal.addEventListener(\n      \"abort\",\n      () => {\n        this.updateAssistantStatus(assistantId, {\n          type: \"incomplete\",\n          reason: \"cancelled\",\n        });\n        this.finishRun(abortController);\n        invokeRuntimeCallback(\"onCancel\", this.onCancel);\n      },\n      { once: true },\n    );\n\n    this.setRunning(true);\n\n    try {\n      if (!(await this.waitForAgentCard(abortController.signal))) return;\n\n      const supportsStreaming =\n        this.agentCardValue?.capabilities?.streaming !== false;\n      if (supportsStreaming) {\n        await this.runStreaming(a2aMessage, assistantId, abortController);\n      } else {\n        await this.runSync(a2aMessage, assistantId, abortController);\n      }\n    } catch (error) {\n      if (!abortController.signal.aborted) {\n        const err = error instanceof Error ? error : new Error(String(error));\n        this.updateAssistantStatus(assistantId, {\n          type: \"incomplete\",\n          reason: \"error\",\n        });\n        invokeRuntimeCallback(\"onError\", this.onError, err);\n        this.pendingError = this.pendingError ?? err;\n      }\n    } finally {\n      this.finishRun(abortController);\n    }\n\n    if (this.pendingError) {\n      const err = this.pendingError;\n      this.pendingError = null;\n      throw err;\n    }\n  }\n\n  private async runStreaming(\n    a2aMessage: A2AMessage,\n    assistantId: string,\n    abortController: AbortController,\n  ): Promise<void> {\n    const stream = this.client.streamMessage(\n      a2aMessage,\n      this.configuration,\n      undefined, // metadata\n      abortController.signal,\n    );\n\n    let receivedEvent = false;\n    let skipReason: string | undefined;\n    const diagnosedStream = (async function* () {\n      const reason = yield* stream;\n      if (typeof reason === \"string\") skipReason = reason;\n    })();\n\n    for await (const event of diagnosedStream) {\n      if (abortController.signal.aborted) break;\n      receivedEvent = true;\n      this.handleStreamEvent(assistantId, event);\n    }\n\n    if (!abortController.signal.aborted) {\n      if (!receivedEvent) {\n        throw new Error(\n          skipReason\n            ? `A2A message stream ended without any events. First skipped frame: ${skipReason}`\n            : \"A2A message stream ended without any events.\",\n        );\n      }\n\n      const lastStatus = this.getAssistantStatus(assistantId);\n      if (lastStatus?.type === \"running\") {\n        this.updateAssistantStatus(assistantId, {\n          type: \"complete\",\n          reason: \"stop\",\n        });\n      }\n    }\n  }\n\n  private async runSync(\n    a2aMessage: A2AMessage,\n    assistantId: string,\n    abortController: AbortController,\n  ): Promise<void> {\n    const result = await this.client.sendMessage(\n      a2aMessage,\n      this.configuration,\n      undefined, // metadata\n      abortController.signal,\n    );\n\n    if (abortController.signal.aborted) return;\n\n    // Result is either A2ATask or A2AMessage\n    if (\"id\" in result && \"status\" in result) {\n      // It's a Task\n      this.handleTaskSnapshot(assistantId, result as A2ATask);\n    } else if (\"messageId\" in result && \"parts\" in result) {\n      // It's a Message\n      this.handleMessage(assistantId, result as A2AMessage);\n      this.updateAssistantStatus(assistantId, {\n        type: \"complete\",\n        reason: \"stop\",\n      });\n    }\n  }\n\n  private handleStreamEvent(assistantId: string, event: A2AStreamEvent) {\n    switch (event.type) {\n      case \"statusUpdate\":\n        this.handleStatusUpdate(assistantId, event.event);\n        break;\n      case \"artifactUpdate\":\n        this.handleArtifactUpdate(event.event);\n        break;\n      case \"message\":\n        this.handleMessage(assistantId, event.message);\n        break;\n      case \"task\":\n        this.handleTaskSnapshot(assistantId, event.task);\n        break;\n    }\n  }\n\n  private handleStatusUpdate(\n    assistantId: string,\n    event: A2ATaskStatusUpdateEvent,\n  ) {\n    if (!this.currentTask) {\n      this.currentTask = {\n        id: event.taskId,\n        contextId: event.contextId,\n        status: event.status,\n      };\n    } else {\n      this.currentTask = { ...this.currentTask, status: event.status };\n    }\n\n    if (event.contextId) {\n      this.contextId = event.contextId;\n    }\n\n    if (event.status.message) {\n      const content = a2aMessageToContent(event.status.message);\n      this.updateAssistantContent(assistantId, content);\n    }\n\n    const status = taskStateToMessageStatus(event.status.state);\n    this.updateAssistantStatus(assistantId, status);\n\n    this.notifyUpdate();\n  }\n\n  private handleArtifactUpdate(event: A2ATaskArtifactUpdateEvent) {\n    const { append, lastChunk } = event;\n    const artifact = normalizeArtifact(event.artifact);\n    const existingIdx = this.currentArtifacts.findIndex(\n      (a) => a.artifactId === artifact.artifactId,\n    );\n\n    let updated: A2AArtifact;\n    if (existingIdx >= 0 && append) {\n      const existing = this.currentArtifacts[existingIdx]!;\n      updated = {\n        ...existing,\n        parts: [...existing.parts, ...artifact.parts],\n      };\n      this.currentArtifacts = [\n        ...this.currentArtifacts.slice(0, existingIdx),\n        updated,\n        ...this.currentArtifacts.slice(existingIdx + 1),\n      ];\n    } else if (existingIdx >= 0) {\n      updated = artifact;\n      this.currentArtifacts = [\n        ...this.currentArtifacts.slice(0, existingIdx),\n        updated,\n        ...this.currentArtifacts.slice(existingIdx + 1),\n      ];\n    } else {\n      updated = artifact;\n      this.currentArtifacts = [...this.currentArtifacts, updated];\n    }\n\n    if (lastChunk) {\n      invokeRuntimeCallback(\n        \"onArtifactComplete\",\n        this.onArtifactComplete,\n        updated,\n      );\n    }\n\n    this.notifyUpdate();\n  }\n\n  private handleMessage(assistantId: string, message: A2AMessage) {\n    if (message.role !== \"agent\") return;\n\n    const content = a2aMessageToContent(message);\n    this.updateAssistantContent(assistantId, content);\n    this.notifyUpdate();\n  }\n\n  private handleTaskSnapshot(assistantId: string, task: A2ATask) {\n    const artifacts =\n      task.artifacts === undefined\n        ? undefined\n        : Array.isArray(task.artifacts)\n          ? task.artifacts.map(normalizeArtifact)\n          : [];\n    const history =\n      task.history === undefined\n        ? undefined\n        : Array.isArray(task.history)\n          ? task.history\n          : [];\n    this.currentTask = {\n      ...task,\n      ...(artifacts === undefined ? {} : { artifacts }),\n      ...(history === undefined ? {} : { history }),\n    };\n\n    if (task.contextId) {\n      this.contextId = task.contextId;\n    }\n    if (artifacts) {\n      this.currentArtifacts = artifacts;\n    }\n\n    if (task.status.message) {\n      const content = a2aMessageToContent(task.status.message);\n      this.updateAssistantContent(assistantId, content);\n    }\n\n    const status = taskStateToMessageStatus(task.status.state);\n    this.updateAssistantStatus(assistantId, status);\n\n    this.notifyUpdate();\n  }\n\n  // --- Message helpers ---\n\n  private insertAssistantPlaceholder(parentId: string): string {\n    const id = generateId();\n    const assistant: ThreadAssistantMessage = {\n      id,\n      role: \"assistant\",\n      createdAt: new Date(),\n      status: { type: \"running\" },\n      content: [],\n      metadata: {\n        unstable_state: null,\n        unstable_annotations: [],\n        unstable_data: [],\n        steps: [],\n        custom: {},\n      },\n    };\n    this.session.addOrUpdateMessage(parentId, assistant);\n    this.session.switchToBranch(id);\n    this.notifyUpdate();\n    return id;\n  }\n\n  private updateAssistantContent(\n    messageId: string,\n    content: ThreadAssistantMessage[\"content\"],\n  ) {\n    this.session.updateMessage(messageId, (message) => {\n      if (message.role !== \"assistant\") return message;\n      return { ...message, content };\n    });\n  }\n\n  private updateAssistantStatus(messageId: string, status: MessageStatus) {\n    const touched = this.session.updateMessage(messageId, (message) => {\n      if (message.role !== \"assistant\") return message;\n      return { ...message, status };\n    });\n    if (touched) {\n      this.notifyUpdate();\n      if (status.type === \"complete\" || status.type === \"incomplete\") {\n        this.persistAssistantHistory(messageId);\n      }\n    }\n  }\n\n  private getAssistantStatus(messageId: string): MessageStatus | undefined {\n    const msg = this.session.tryGetMessage(messageId)?.message;\n    if (msg?.role !== \"assistant\") return undefined;\n    return msg.status;\n  }\n\n  // --- Lifecycle helpers ---\n\n  private setRunning(running: boolean) {\n    this.isRunningFlag = running;\n    this.notifyUpdate();\n  }\n\n  private finishRun(controller: AbortController | null) {\n    if (this.abortController !== controller) return;\n    this.abortController = null;\n    this.setRunning(false);\n  }\n\n  // --- History persistence ---\n\n  private recordHistoryEntry(parentId: string | null, message: ThreadMessage) {\n    this.appendHistoryItem(parentId, message);\n  }\n\n  private markPendingAssistantHistory(\n    messageId: string,\n    parentId: string | null,\n  ) {\n    if (!this.history) return;\n    this.assistantHistoryParents.set(messageId, parentId);\n  }\n\n  private persistAssistantHistory(messageId: string) {\n    if (!this.history) return;\n    const parentId = this.assistantHistoryParents.get(messageId);\n    if (parentId === undefined) return;\n    const message = this.session.tryGetMessage(messageId)?.message;\n    if (!message || message.role !== \"assistant\") return;\n    if (\n      message.status?.type !== \"complete\" &&\n      message.status?.type !== \"incomplete\"\n    )\n      return;\n    this.assistantHistoryParents.delete(messageId);\n    this.appendHistoryItem(parentId, message);\n  }\n\n  private appendHistoryItem(parentId: string | null, message: ThreadMessage) {\n    if (!this.history || this.recordedHistoryIds.has(message.id)) return;\n    this.recordedHistoryIds.add(message.id);\n    void this.history.append({ parentId, message }).catch(() => {\n      this.recordedHistoryIds.delete(message.id);\n    });\n  }\n}\n"],"mappings":";;;;;AAmCA,MAAM,oCAAoC;AAC1C,MAAM,gCAAgC;AAatC,MAAM,uBAAuB;CAC3B,MAAM;CACN,QAAQ;AACV;AAIA,MAAM,yBACJ,MACA,UACA,GAAG,SACM;CACT,mBAAwB,aAAa,MAAM,UAAU,GAAG,IAAI;AAC9D;AAEA,SAAS,kBAAkB,UAAoC;CAC7D,OAAO;EACL,GAAG;EACH,OAAO,MAAM,QAAQ,SAAS,KAAK,IAAI,SAAS,QAAQ,CAAC;CAC3D;AACF;AAEA,IAAa,uBAAb,MAAkC;CAChC;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CAEA;CACA,UAA2B,+BAA+B;CAC1D,gBAAwB;CACxB,kBAAkD;CAClD,eAAqC;CAGrC;CACA,mBAA0C,CAAC;CAC3C;CAGA,0CAA2C,IAAI,IAA2B;CAC1E,qCAAsC,IAAI,IAAY;CACtD,aAAqB;CACrB;CACA,iBAAyB;CACzB;CACA,uBAA+B;CAC/B,uBAA+B;CAC/B,4BAAoC;CAEpC;CAEA,YAAY,SAAsC;EAChD,KAAK,SAAS,QAAQ;EACtB,KAAK,YAAY,QAAQ;EACzB,KAAK,uBAAuB,QAAQ;EACpC,KAAK,gBAAgB,QAAQ;EAC7B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,qBAAqB,QAAQ;EAClC,KAAK,UAAU,QAAQ;EACvB,KAAK,eAAe,QAAQ;CAC9B;CAEA,cAAc,SAA4D;EACxE,KAAK,SAAS,QAAQ;EAKtB,IAAI,QAAQ,cAAc,KAAK,sBAAsB;GACnD,KAAK,YAAY,QAAQ;GACzB,KAAK,uBAAuB,QAAQ;EACtC;EACA,KAAK,gBAAgB,QAAQ;EAC7B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,qBAAqB,QAAQ;EAClC,MAAM,kBAAkB,KAAK;EAC7B,KAAK,UAAU,QAAQ;EAEvB,IACE,KAAK,kBACL,CAAC,KAAK,gBACN,CAAC,mBACD,QAAQ,WACR,KAAK,QAAQ,YAAY,CAAC,CAAC,WAAW,GAEtC,KAAU,gBAAgB;CAE9B;;;CAIA,eAAqB;EAInB,MAAM,aAAa,KAAK;EACxB,KAAK,YAAY,KAAK;EACtB,IAAI,YAAY;GACd,WAAW,MAAM;GACjB,IAAI,KAAK,oBAAoB,YAC3B,KAAK,kBAAkB;EAE3B;CACF;CAEA,cAAc,SAA2B;EACvC,KAAK,UAAU;CACjB;CAEA,gBAAgB;EACd,KAAK,UAAU,KAAA;EAEf,IAAI,KAAK,iBAAiB;GACxB,KAAK,gBAAgB,MAAM;GAC3B,KAAK,kBAAkB;EACzB;CACF;CAEA,aAA2C;EACzC,OAAO,KAAK;CACd;CAEA,cAAwC;EACtC,OAAO,KAAK,QAAQ,YAAY;CAClC;CAEA,uBAAkD;EAChD,OAAO,KAAK,QAAQ,OAAO;CAC7B;CAEA,UAA+B;EAC7B,OAAO,KAAK;CACd;CAEA,eAAuC;EACrC,OAAO,KAAK;CACd;CAEA,eAAyC;EACvC,OAAO,KAAK;CACd;CAEA,YAAqB;EACnB,OAAO,KAAK;CACd;CAEA,IAAI,YAAqB;EACvB,OAAO,KAAK;CACd;CAEA,gBAAuC;EACrC,IAAI,KAAK,IAAI,IAAI,KAAK,sBAAsB,OAAO,QAAQ,QAAQ;EAEnE,KAAK,sBAAsB,KAAK,OAAO,aAAa,CAAC,CAAC,MACnD,cAAc;GACb,KAAK,iBAAiB;GACtB,KAAK,uBAAuB;GAC5B,KAAK,uBAAuB;GAC5B,KAAK,4BAA4B;GACjC,KAAK,aAAa;EACpB,SACM;GACJ,KAAK,4BAA4B;GACjC,KAAK,uBAAuB,KAAK,IAAI,IAAI,KAAK;GAC9C,KAAK,uBAAuB,KAAK,IAC/B,KAAK,uBAAuB,GAC5B,6BACF;GACA,KAAK,oBAAoB,KAAA;EAC3B,CACF;EACA,OAAO,KAAK;CACd;CAEA,MAAc,iBAAiB,QAAuC;EACpE,MAAM,aAAa,CAAC,KAAK;EACzB,MAAM,OAAO,KAAK,cAAc;EAChC,IAAI,OAAO,SAAS,OAAO;EAC3B,IAAI,CAAC,YAAY,OAAO;EAExB,IAAI;EACJ,MAAM,QAAQ,IAAI,SAAe,YAAY;GAC3C,UAAU;GACV,OAAO,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;EAC1D,CAAC;EAED,IAAI;GACF,MAAM,QAAQ,KAAK,CAAC,MAAM,KAAK,CAAC;EAClC,UAAU;GACR,OAAO,oBAAoB,SAAS,OAAO;EAC7C;EACA,OAAO,CAAC,OAAO;CACjB;CAEA,kBAAiC;EAC/B,KAAK,iBAAiB;EACtB,MAAM,mBAAmB,KAAK,cAAc;EAE5C,IAAI,KAAK,cAAc,OAAO,KAAK;EACnC,IAAI,CAAC,KAAK,SAAS,OAAO;EAE1B,KAAK,aAAa;EAElB,MAAM,iBAAiB,KAAK,QAAQ,KAAK;EAEzC,KAAK,eAAe,QAAQ,IAAI,CAAC,gBAAgB,gBAAgB,CAAC,CAAC,CAChE,MAAM,CAAC,UAAU;GAChB,IAAI,MAAM;IACR,KAAK,QAAQ,+BAA+B,IAAI;IAChD,KAAK,sBAAsB;GAC7B;EACF,CAAC,CAAC,CACD,OAAO,UAAU;GAChB,sBACE,WACA,KAAK,SACL,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAC1D;EACF,CAAC,CAAC,CACD,cAAc;GACb,KAAK,aAAa;GAClB,KAAK,aAAa;EACpB,CAAC;EAEH,KAAK,aAAa;EAClB,OAAO,KAAK;CACd;CAEA,MAAM,OAAO,SAAuC;EAClD,MAAM,WAAW,QAAQ,YAAY,QAAQ,SAAS;EAEtD,MAAM,gBAAgB,sBACpB,SACA,WAAW,GACX,oBACF;EACA,MAAM,WACJ,QAAQ,aAAa,OACjB,OACA,QAAQ,YAAY,KAAK,QAAQ,WAAW,QAAQ,QAAQ,IAC1D,QAAQ,WACR,KAAK,QAAQ;EACrB,KAAK,QAAQ,mBAAmB,UAAU,aAAa;EACvD,KAAK,QAAQ,eAAe,cAAc,EAAE;EAC5C,KAAK,aAAa;EAClB,KAAK,mBAAmB,UAAU,aAAa;EAE/C,IAAI,CAAC,UAAU;EACf,MAAM,KAAK,SAAS,aAAa;CACnC;CAEA,MAAM,KAAK,SAAuC;EAChD,MAAM,KAAK,OAAO,OAAO;CAC3B;CAEA,MAAM,OACJ,UACA,UAAmD,CAAC,GACrC;EACf,MAAM,WACJ,aAAa,OACT,CAAC,IACA,KAAK,QAAQ,eAAe,QAAQ,KAAK,KAAK,YAAY;EACjE,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KACxC,IAAI,SAAS,EAAE,CAAE,SAAS,QAAQ;GAChC,MAAM,KAAK,SAAS,SAAS,EAAG;GAChC;EACF;CAEJ;CAEA,MAAM,SAAwB;EAC5B,IAAI,CAAC,KAAK,iBAAiB;EAG3B,KAAK,gBAAgB,MAAM;EAG3B,IAAI,KAAK,aAAa,IACpB,IAAI;GACF,MAAM,UAAU,MAAM,KAAK,OAAO,WAAW,KAAK,YAAY,EAAE;GAChE,KAAK,cAAc;EACrB,QAAQ,CAER;CAEJ;CAEA,kBAA0B,UAAmD;EAC3E,IAAI,WAA0B;EAC9B,IAAI,SAAwB;EAC5B,MAAM,uBAAO,IAAI,IAAY;EAE7B,KAAK,MAAM,WAAW,UAAU;GAC9B,IAAI,KAAK,IAAI,QAAQ,EAAE,GAAG;GAC1B,KAAK,IAAI,QAAQ,EAAE;GACnB,KAAK,QAAQ,mBAAmB,UAAU,OAAO;GACjD,WAAW,QAAQ;GACnB,SAAS,QAAQ;EACnB;EAEA,OAAO;CACT;CAEA,wBAAsC;EACpC,KAAK,wBAAwB,MAAM;EACnC,KAAK,mBAAmB,MAAM;EAC9B,KAAK,MAAM,EAAE,aAAa,KAAK,qBAAqB,CAAC,CAAC,UACpD,KAAK,mBAAmB,IAAI,QAAQ,EAAE;EAExC,KAAK,cAAc,KAAA;EACnB,KAAK,mBAAmB,CAAC;EACzB,KAAK,aAAa;CACpB;CAEA,sBAAsB,UAA0C;EAC9D,IAAI,SAAS,WAAW,GACtB,KAAK,QAAQ,MAAM;OACd;GACL,IAAI,mBAAkC;GACtC,IAAI,gBAA+B;GACnC,IAAI,cAAc;GAClB,MAAM,uBAAO,IAAI,IAAY;GAE7B,KAAK,MAAM,WAAW,UAAU;IAC9B,IAAI,KAAK,IAAI,QAAQ,EAAE,GAAG;IAC1B,KAAK,IAAI,QAAQ,EAAE;IACnB,MAAM,WAAW,KAAK,QAAQ,cAAc,QAAQ,EAAE;IACtD,IAAI,YAAY,SAAS,aAAa,kBAAkB;KACtD,cAAc;KACd;IACF;IACA,KAAK,QAAQ,mBAAmB,kBAAkB,OAAO;IACzD,mBAAmB,QAAQ;IAC3B,gBAAgB,QAAQ;GAC1B;GAEA,IAAI,aAAa;IACf,KAAK,QAAQ,MAAM;IACnB,gBAAgB,KAAK,kBAAkB,QAAQ;GACjD;GAEA,KAAK,QAAQ,UAAU,aAAa;EACtC;EAEA,KAAK,sBAAsB;CAC7B;CAIA,MAAc,SAAS,mBAAiD;EAEtE,IAAI,KAAK,iBAAiB;GACxB,KAAK,gBAAgB,MAAM;GAC3B,KAAK,kBAAkB;EACzB;EAEA,MAAM,aAAa,0BAA0B,mBAAmB;GAC9D,WAAW,KAAK;GAChB,QACE,KAAK,aAAa,MAClB,CAAC,oBAAoB,KAAK,YAAY,OAAO,KAAK,IAC9C,KAAK,YAAY,KACjB,KAAA;EACR,CAAC;EAGD,IACE,KAAK,eACL,oBAAoB,KAAK,YAAY,OAAO,KAAK,GAEjD,KAAK,cAAc,KAAA;EAGrB,KAAK,mBAAmB,CAAC;EAEzB,MAAM,oBAAoB,kBAAkB;EAC5C,MAAM,cAAc,KAAK,2BAA2B,iBAAiB;EACrE,KAAK,4BAA4B,aAAa,iBAAiB;EAE/D,MAAM,kBAAkB,IAAI,gBAAgB;EAC5C,KAAK,kBAAkB;EAEvB,gBAAgB,OAAO,iBACrB,eACM;GACJ,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;GACD,KAAK,UAAU,eAAe;GAC9B,sBAAsB,YAAY,KAAK,QAAQ;EACjD,GACA,EAAE,MAAM,KAAK,CACf;EAEA,KAAK,WAAW,IAAI;EAEpB,IAAI;GACF,IAAI,CAAE,MAAM,KAAK,iBAAiB,gBAAgB,MAAM,GAAI;GAI5D,IADE,KAAK,gBAAgB,cAAc,cAAc,OAEjD,MAAM,KAAK,aAAa,YAAY,aAAa,eAAe;QAEhE,MAAM,KAAK,QAAQ,YAAY,aAAa,eAAe;EAE/D,SAAS,OAAO;GACd,IAAI,CAAC,gBAAgB,OAAO,SAAS;IACnC,MAAM,MAAM,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;IACpE,KAAK,sBAAsB,aAAa;KACtC,MAAM;KACN,QAAQ;IACV,CAAC;IACD,sBAAsB,WAAW,KAAK,SAAS,GAAG;IAClD,KAAK,eAAe,KAAK,gBAAgB;GAC3C;EACF,UAAU;GACR,KAAK,UAAU,eAAe;EAChC;EAEA,IAAI,KAAK,cAAc;GACrB,MAAM,MAAM,KAAK;GACjB,KAAK,eAAe;GACpB,MAAM;EACR;CACF;CAEA,MAAc,aACZ,YACA,aACA,iBACe;EACf,MAAM,SAAS,KAAK,OAAO,cACzB,YACA,KAAK,eACL,KAAA,GACA,gBAAgB,MAClB;EAEA,IAAI,gBAAgB;EACpB,IAAI;EACJ,MAAM,mBAAmB,mBAAmB;GAC1C,MAAM,SAAS,OAAO;GACtB,IAAI,OAAO,WAAW,UAAU,aAAa;EAC/C,EAAA,CAAG;EAEH,WAAW,MAAM,SAAS,iBAAiB;GACzC,IAAI,gBAAgB,OAAO,SAAS;GACpC,gBAAgB;GAChB,KAAK,kBAAkB,aAAa,KAAK;EAC3C;EAEA,IAAI,CAAC,gBAAgB,OAAO,SAAS;GACnC,IAAI,CAAC,eACH,MAAM,IAAI,MACR,aACI,qEAAqE,eACrE,8CACN;GAIF,IADmB,KAAK,mBAAmB,WAC9B,CAAC,EAAE,SAAS,WACvB,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;EAEL;CACF;CAEA,MAAc,QACZ,YACA,aACA,iBACe;EACf,MAAM,SAAS,MAAM,KAAK,OAAO,YAC/B,YACA,KAAK,eACL,KAAA,GACA,gBAAgB,MAClB;EAEA,IAAI,gBAAgB,OAAO,SAAS;EAGpC,IAAI,QAAQ,UAAU,YAAY,QAEhC,KAAK,mBAAmB,aAAa,MAAiB;OACjD,IAAI,eAAe,UAAU,WAAW,QAAQ;GAErD,KAAK,cAAc,aAAa,MAAoB;GACpD,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;EACH;CACF;CAEA,kBAA0B,aAAqB,OAAuB;EACpE,QAAQ,MAAM,MAAd;GACE,KAAK;IACH,KAAK,mBAAmB,aAAa,MAAM,KAAK;IAChD;GACF,KAAK;IACH,KAAK,qBAAqB,MAAM,KAAK;IACrC;GACF,KAAK;IACH,KAAK,cAAc,aAAa,MAAM,OAAO;IAC7C;GACF,KAAK,QACH,KAAK,mBAAmB,aAAa,MAAM,IAAI;EAEnD;CACF;CAEA,mBACE,aACA,OACA;EACA,IAAI,CAAC,KAAK,aACR,KAAK,cAAc;GACjB,IAAI,MAAM;GACV,WAAW,MAAM;GACjB,QAAQ,MAAM;EAChB;OAEA,KAAK,cAAc;GAAE,GAAG,KAAK;GAAa,QAAQ,MAAM;EAAO;EAGjE,IAAI,MAAM,WACR,KAAK,YAAY,MAAM;EAGzB,IAAI,MAAM,OAAO,SAAS;GACxB,MAAM,UAAU,oBAAoB,MAAM,OAAO,OAAO;GACxD,KAAK,uBAAuB,aAAa,OAAO;EAClD;EAEA,MAAM,SAAS,yBAAyB,MAAM,OAAO,KAAK;EAC1D,KAAK,sBAAsB,aAAa,MAAM;EAE9C,KAAK,aAAa;CACpB;CAEA,qBAA6B,OAAmC;EAC9D,MAAM,EAAE,QAAQ,cAAc;EAC9B,MAAM,WAAW,kBAAkB,MAAM,QAAQ;EACjD,MAAM,cAAc,KAAK,iBAAiB,WACvC,MAAM,EAAE,eAAe,SAAS,UACnC;EAEA,IAAI;EACJ,IAAI,eAAe,KAAK,QAAQ;GAC9B,MAAM,WAAW,KAAK,iBAAiB;GACvC,UAAU;IACR,GAAG;IACH,OAAO,CAAC,GAAG,SAAS,OAAO,GAAG,SAAS,KAAK;GAC9C;GACA,KAAK,mBAAmB;IACtB,GAAG,KAAK,iBAAiB,MAAM,GAAG,WAAW;IAC7C;IACA,GAAG,KAAK,iBAAiB,MAAM,cAAc,CAAC;GAChD;EACF,OAAO,IAAI,eAAe,GAAG;GAC3B,UAAU;GACV,KAAK,mBAAmB;IACtB,GAAG,KAAK,iBAAiB,MAAM,GAAG,WAAW;IAC7C;IACA,GAAG,KAAK,iBAAiB,MAAM,cAAc,CAAC;GAChD;EACF,OAAO;GACL,UAAU;GACV,KAAK,mBAAmB,CAAC,GAAG,KAAK,kBAAkB,OAAO;EAC5D;EAEA,IAAI,WACF,sBACE,sBACA,KAAK,oBACL,OACF;EAGF,KAAK,aAAa;CACpB;CAEA,cAAsB,aAAqB,SAAqB;EAC9D,IAAI,QAAQ,SAAS,SAAS;EAE9B,MAAM,UAAU,oBAAoB,OAAO;EAC3C,KAAK,uBAAuB,aAAa,OAAO;EAChD,KAAK,aAAa;CACpB;CAEA,mBAA2B,aAAqB,MAAe;EAC7D,MAAM,YACJ,KAAK,cAAc,KAAA,IACf,KAAA,IACA,MAAM,QAAQ,KAAK,SAAS,IAC1B,KAAK,UAAU,IAAI,iBAAiB,IACpC,CAAC;EACT,MAAM,UACJ,KAAK,YAAY,KAAA,IACb,KAAA,IACA,MAAM,QAAQ,KAAK,OAAO,IACxB,KAAK,UACL,CAAC;EACT,KAAK,cAAc;GACjB,GAAG;GACH,GAAI,cAAc,KAAA,IAAY,CAAC,IAAI,EAAE,UAAU;GAC/C,GAAI,YAAY,KAAA,IAAY,CAAC,IAAI,EAAE,QAAQ;EAC7C;EAEA,IAAI,KAAK,WACP,KAAK,YAAY,KAAK;EAExB,IAAI,WACF,KAAK,mBAAmB;EAG1B,IAAI,KAAK,OAAO,SAAS;GACvB,MAAM,UAAU,oBAAoB,KAAK,OAAO,OAAO;GACvD,KAAK,uBAAuB,aAAa,OAAO;EAClD;EAEA,MAAM,SAAS,yBAAyB,KAAK,OAAO,KAAK;EACzD,KAAK,sBAAsB,aAAa,MAAM;EAE9C,KAAK,aAAa;CACpB;CAIA,2BAAmC,UAA0B;EAC3D,MAAM,KAAK,WAAW;EACtB,MAAM,YAAoC;GACxC;GACA,MAAM;GACN,2BAAW,IAAI,KAAK;GACpB,QAAQ,EAAE,MAAM,UAAU;GAC1B,SAAS,CAAC;GACV,UAAU;IACR,gBAAgB;IAChB,sBAAsB,CAAC;IACvB,eAAe,CAAC;IAChB,OAAO,CAAC;IACR,QAAQ,CAAC;GACX;EACF;EACA,KAAK,QAAQ,mBAAmB,UAAU,SAAS;EACnD,KAAK,QAAQ,eAAe,EAAE;EAC9B,KAAK,aAAa;EAClB,OAAO;CACT;CAEA,uBACE,WACA,SACA;EACA,KAAK,QAAQ,cAAc,YAAY,YAAY;GACjD,IAAI,QAAQ,SAAS,aAAa,OAAO;GACzC,OAAO;IAAE,GAAG;IAAS;GAAQ;EAC/B,CAAC;CACH;CAEA,sBAA8B,WAAmB,QAAuB;EAKtE,IAJgB,KAAK,QAAQ,cAAc,YAAY,YAAY;GACjE,IAAI,QAAQ,SAAS,aAAa,OAAO;GACzC,OAAO;IAAE,GAAG;IAAS;GAAO;EAC9B,CACU,GAAG;GACX,KAAK,aAAa;GAClB,IAAI,OAAO,SAAS,cAAc,OAAO,SAAS,cAChD,KAAK,wBAAwB,SAAS;EAE1C;CACF;CAEA,mBAA2B,WAA8C;EACvE,MAAM,MAAM,KAAK,QAAQ,cAAc,SAAS,CAAC,EAAE;EACnD,IAAI,KAAK,SAAS,aAAa,OAAO,KAAA;EACtC,OAAO,IAAI;CACb;CAIA,WAAmB,SAAkB;EACnC,KAAK,gBAAgB;EACrB,KAAK,aAAa;CACpB;CAEA,UAAkB,YAAoC;EACpD,IAAI,KAAK,oBAAoB,YAAY;EACzC,KAAK,kBAAkB;EACvB,KAAK,WAAW,KAAK;CACvB;CAIA,mBAA2B,UAAyB,SAAwB;EAC1E,KAAK,kBAAkB,UAAU,OAAO;CAC1C;CAEA,4BACE,WACA,UACA;EACA,IAAI,CAAC,KAAK,SAAS;EACnB,KAAK,wBAAwB,IAAI,WAAW,QAAQ;CACtD;CAEA,wBAAgC,WAAmB;EACjD,IAAI,CAAC,KAAK,SAAS;EACnB,MAAM,WAAW,KAAK,wBAAwB,IAAI,SAAS;EAC3D,IAAI,aAAa,KAAA,GAAW;EAC5B,MAAM,UAAU,KAAK,QAAQ,cAAc,SAAS,CAAC,EAAE;EACvD,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;EAC9C,IACE,QAAQ,QAAQ,SAAS,cACzB,QAAQ,QAAQ,SAAS,cAEzB;EACF,KAAK,wBAAwB,OAAO,SAAS;EAC7C,KAAK,kBAAkB,UAAU,OAAO;CAC1C;CAEA,kBAA0B,UAAyB,SAAwB;EACzE,IAAI,CAAC,KAAK,WAAW,KAAK,mBAAmB,IAAI,QAAQ,EAAE,GAAG;EAC9D,KAAK,mBAAmB,IAAI,QAAQ,EAAE;EACtC,KAAU,QAAQ,OAAO;GAAE;GAAU;EAAQ,CAAC,CAAC,CAAC,YAAY;GAC1D,KAAK,mBAAmB,OAAO,QAAQ,EAAE;EAC3C,CAAC;CACH;AACF"}
+{"version":3,"file":"A2AThreadRuntimeCore.js","names":[],"sources":["../src/A2AThreadRuntimeCore.ts"],"sourcesContent":["\"use client\";\n\nimport { generateId, fromThreadMessageLike } from \"@assistant-ui/core\";\nimport type {\n  AppendMessage,\n  AssistantRuntime,\n  ExportedMessageRepository,\n  MessageStatus,\n  ThreadAssistantMessage,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport {\n  createMessageRepositorySession,\n  invokeUserCallback,\n} from \"@assistant-ui/core/internal\";\nimport type { A2AClient } from \"./A2AClient\";\nimport type {\n  A2AArtifact,\n  A2AAgentCard,\n  A2AMessage,\n  A2ASendMessageConfiguration,\n  A2AStreamEvent,\n  A2ATask,\n  A2ATaskArtifactUpdateEvent,\n  A2ATaskStatusUpdateEvent,\n} from \"./types\";\n\nimport {\n  a2aMessageToContent,\n  isTerminalTaskState,\n  threadMessageToA2AMessage,\n  taskStateToMessageStatus,\n} from \"./conversions\";\n\nconst INITIAL_AGENT_CARD_RETRY_DELAY_MS = 5_000;\nconst MAX_AGENT_CARD_RETRY_DELAY_MS = 5 * 60_000;\n\nexport type A2AThreadRuntimeCoreOptions = {\n  client: A2AClient;\n  contextId?: string | undefined;\n  configuration?: A2ASendMessageConfiguration | undefined;\n  onError?: ((error: Error) => void) | undefined;\n  onCancel?: (() => void) | undefined;\n  onArtifactComplete?: ((artifact: A2AArtifact) => void) | undefined;\n  history?: ThreadHistoryAdapter | undefined;\n  notifyUpdate: () => void;\n};\n\nconst FALLBACK_USER_STATUS = {\n  type: \"complete\",\n  reason: \"unknown\",\n} as const;\n\ntype A2ARuntimeCallbackName = \"onError\" | \"onCancel\" | \"onArtifactComplete\";\n\nconst invokeRuntimeCallback = <TArgs extends readonly unknown[]>(\n  name: A2ARuntimeCallbackName,\n  callback: ((...args: TArgs) => unknown) | undefined,\n  ...args: TArgs\n): void => {\n  void invokeUserCallback(\"react-a2a\", name, callback, ...args);\n};\n\nfunction normalizeArtifact(artifact: A2AArtifact): A2AArtifact {\n  return {\n    ...artifact,\n    parts: Array.isArray(artifact.parts) ? artifact.parts : [],\n  };\n}\n\nexport class A2AThreadRuntimeCore {\n  private client: A2AClient;\n  private contextId: string | undefined;\n  private configuration: A2ASendMessageConfiguration | undefined;\n  private onError: ((error: Error) => void) | undefined;\n  private onCancel: (() => void) | undefined;\n  private onArtifactComplete: ((artifact: A2AArtifact) => void) | undefined;\n  private history: ThreadHistoryAdapter | undefined;\n  private readonly notifyUpdate: () => void;\n\n  private runtime: AssistantRuntime | undefined;\n  private readonly session = createMessageRepositorySession();\n  private isRunningFlag = false;\n  private abortController: AbortController | null = null;\n  private runGeneration = 0;\n  private pendingError: Error | null = null;\n\n  // A2A-specific state\n  private currentTask: A2ATask | undefined;\n  private currentArtifacts: A2AArtifact[] = [];\n  private agentCardValue: A2AAgentCard | undefined;\n\n  // History tracking\n  private readonly assistantHistoryParents = new Map<string, string | null>();\n  private readonly recordedHistoryIds = new Set<string>();\n  private _isLoading = false;\n  private _loadPromise: Promise<void> | undefined;\n  private _historyLoadGeneration = 0;\n  private _loadRequested = false;\n  private _agentCardPromise: Promise<void> | undefined;\n  private _agentCardRetryAfter = 0;\n  private _agentCardRetryDelay = INITIAL_AGENT_CARD_RETRY_DELAY_MS;\n  private _agentCardDiscoveryFailed = false;\n\n  private lastOptionsContextId: string | undefined;\n\n  constructor(options: A2AThreadRuntimeCoreOptions) {\n    this.client = options.client;\n    this.contextId = options.contextId;\n    this.lastOptionsContextId = options.contextId;\n    this.configuration = options.configuration;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.onArtifactComplete = options.onArtifactComplete;\n    this.history = options.history;\n    this.notifyUpdate = options.notifyUpdate;\n  }\n\n  updateOptions(options: Omit<A2AThreadRuntimeCoreOptions, \"notifyUpdate\">) {\n    this.client = options.client;\n    // The hook re-applies options on every render, including renders caused\n    // by this core's own notifyUpdate. The option only seeds the context: a\n    // re-render with the same value must not clobber a server-assigned\n    // contextId learned from the stream.\n    if (options.contextId !== this.lastOptionsContextId) {\n      this.contextId = options.contextId;\n      this.lastOptionsContextId = options.contextId;\n    }\n    this.configuration = options.configuration;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.onArtifactComplete = options.onArtifactComplete;\n    const previousHistory = this.history;\n    this.history = options.history;\n\n    if (\n      this._loadRequested &&\n      !this._loadPromise &&\n      !previousHistory &&\n      options.history &&\n      this.session.getMessages().length === 0\n    ) {\n      void this.__internal_load();\n    }\n  }\n\n  /** Thread-boundary reset: applyExternalMessages alone also serves branch\n   * switches, deletes, and cancel resyncs, which must keep the live context. */\n  resetContext(): void {\n    this._historyLoadGeneration++;\n    this._isLoading = false;\n    // Restore the seed before aborting: an onCancel callback that starts a\n    // new run must not pick up the old thread's context, and its controller\n    // must not be discarded.\n    const controller = this.abortController;\n    this.contextId = this.lastOptionsContextId;\n    if (controller) {\n      controller.abort();\n      if (this.abortController === controller) {\n        this.abortController = null;\n      }\n    }\n  }\n\n  attachRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  detachRuntime() {\n    this.runtime = undefined;\n    // Abort in-flight requests on unmount\n    if (this.abortController) {\n      this.abortController.abort();\n      this.abortController = null;\n    }\n  }\n\n  getRuntime(): AssistantRuntime | undefined {\n    return this.runtime;\n  }\n\n  getMessages(): readonly ThreadMessage[] {\n    return this.session.getMessages();\n  }\n\n  getMessageRepository(): ExportedMessageRepository {\n    return this.session.export();\n  }\n\n  getTask(): A2ATask | undefined {\n    return this.currentTask;\n  }\n\n  getArtifacts(): readonly A2AArtifact[] {\n    return this.currentArtifacts;\n  }\n\n  getAgentCard(): A2AAgentCard | undefined {\n    return this.agentCardValue;\n  }\n\n  isRunning(): boolean {\n    return this.isRunningFlag;\n  }\n\n  get isLoading(): boolean {\n    return this._isLoading;\n  }\n\n  private loadAgentCard(): Promise<void> {\n    if (Date.now() < this._agentCardRetryAfter) return Promise.resolve();\n\n    this._agentCardPromise ??= this.client.getAgentCard().then(\n      (agentCard) => {\n        this.agentCardValue = agentCard;\n        this._agentCardRetryAfter = 0;\n        this._agentCardRetryDelay = INITIAL_AGENT_CARD_RETRY_DELAY_MS;\n        this._agentCardDiscoveryFailed = false;\n        this.notifyUpdate();\n      },\n      () => {\n        this._agentCardDiscoveryFailed = true;\n        this._agentCardRetryAfter = Date.now() + this._agentCardRetryDelay;\n        this._agentCardRetryDelay = Math.min(\n          this._agentCardRetryDelay * 2,\n          MAX_AGENT_CARD_RETRY_DELAY_MS,\n        );\n        this._agentCardPromise = undefined;\n      },\n    );\n    return this._agentCardPromise;\n  }\n\n  private async waitForAgentCard(signal: AbortSignal): Promise<boolean> {\n    const shouldWait = !this._agentCardDiscoveryFailed;\n    const load = this.loadAgentCard();\n    if (signal.aborted) return false;\n    if (!shouldWait) return true;\n\n    let onAbort!: () => void;\n    const abort = new Promise<void>((resolve) => {\n      onAbort = resolve;\n      signal.addEventListener(\"abort\", onAbort, { once: true });\n    });\n\n    try {\n      await Promise.race([load, abort]);\n    } finally {\n      signal.removeEventListener(\"abort\", onAbort);\n    }\n    return !signal.aborted;\n  }\n\n  __internal_load(): Promise<void> {\n    this._loadRequested = true;\n    const agentCardPromise = this.loadAgentCard();\n\n    if (this._loadPromise) return this._loadPromise;\n    if (!this.history) return agentCardPromise;\n\n    this._isLoading = true;\n\n    const generation = this._historyLoadGeneration;\n    const historyPromise = this.history.load();\n\n    this._loadPromise = Promise.all([historyPromise, agentCardPromise])\n      .then(([repo]) => {\n        if (generation !== this._historyLoadGeneration) return;\n        if (repo) {\n          this.session.applyExternalMessageRepository(repo);\n          this.finalizeExternalApply();\n        }\n      })\n      .catch((error) => {\n        if (generation !== this._historyLoadGeneration) return;\n        invokeRuntimeCallback(\n          \"onError\",\n          this.onError,\n          error instanceof Error ? error : new Error(String(error)),\n        );\n      })\n      .finally(() => {\n        if (generation !== this._historyLoadGeneration) return;\n        this._isLoading = false;\n        this.notifyUpdate();\n      });\n\n    this.notifyUpdate();\n    return this._loadPromise;\n  }\n\n  async append(message: AppendMessage): Promise<void> {\n    const startRun = message.startRun ?? message.role === \"user\";\n\n    const threadMessage = fromThreadMessageLike(\n      message as any,\n      generateId(),\n      FALLBACK_USER_STATUS,\n    );\n    const parentId =\n      message.parentId === null\n        ? null\n        : message.parentId && this.session.hasMessage(message.parentId)\n          ? message.parentId\n          : this.session.headId;\n    this.session.addOrUpdateMessage(parentId, threadMessage);\n    this.session.switchToBranch(threadMessage.id);\n    this.notifyUpdate();\n    this.recordHistoryEntry(parentId, threadMessage);\n\n    if (!startRun) return;\n    await this.startRun(threadMessage);\n  }\n\n  async edit(message: AppendMessage): Promise<void> {\n    await this.append(message);\n  }\n\n  async reload(\n    parentId: string | null,\n    _config: { runConfig?: Record<string, unknown> } = {},\n  ): Promise<void> {\n    const messages =\n      parentId === null\n        ? []\n        : (this.session.tryGetMessages(parentId) ?? this.getMessages());\n    for (let i = messages.length - 1; i >= 0; i--) {\n      if (messages[i]!.role === \"user\") {\n        await this.startRun(messages[i]!);\n        return;\n      }\n    }\n  }\n\n  async cancel(): Promise<void> {\n    if (!this.abortController) return;\n\n    // Read the server target before aborting: the abort listener runs the\n    // onCancel callback synchronously, which may clear the thread and with it\n    // the task this cancellation is for, or start a new run.\n    const task = this.currentTask;\n    const generation = this.runGeneration;\n\n    // Abort locally first so the stream stops immediately\n    this.abortController.abort();\n\n    // Then try to cancel the task on the server\n    if (task?.id) {\n      try {\n        const updated = await this.client.cancelTask(task.id);\n        // Only apply the response while nothing newer exists. A newer snapshot\n        // or a cleared thread replaces the task object; a follow-up run that\n        // has not emitted yet keeps it, so the run generation is what rules\n        // that case out.\n        if (this.currentTask === task && this.runGeneration === generation) {\n          this.currentTask = updated;\n        }\n      } catch {\n        // Server cancel failed; local abort already handled\n      }\n    }\n  }\n\n  private appendLinearChain(messages: readonly ThreadMessage[]): string | null {\n    let parentId: string | null = null;\n    let lastId: string | null = null;\n    const seen = new Set<string>();\n\n    for (const message of messages) {\n      if (seen.has(message.id)) continue;\n      seen.add(message.id);\n      this.session.addOrUpdateMessage(parentId, message);\n      parentId = message.id;\n      lastId = message.id;\n    }\n\n    return lastId;\n  }\n\n  private finalizeExternalApply(): void {\n    this.assistantHistoryParents.clear();\n    this.recordedHistoryIds.clear();\n    for (const { message } of this.getMessageRepository().messages) {\n      this.recordedHistoryIds.add(message.id);\n    }\n    this.currentTask = undefined;\n    this.currentArtifacts = [];\n    this.notifyUpdate();\n  }\n\n  applyExternalMessages(messages: readonly ThreadMessage[]): void {\n    if (messages.length === 0) {\n      this.session.clear();\n    } else {\n      let expectedParentId: string | null = null;\n      let lastAppliedId: string | null = null;\n      let hardReplace = false;\n      const seen = new Set<string>();\n\n      for (const message of messages) {\n        if (seen.has(message.id)) continue;\n        seen.add(message.id);\n        const existing = this.session.tryGetMessage(message.id);\n        if (existing && existing.parentId !== expectedParentId) {\n          hardReplace = true;\n          break;\n        }\n        this.session.addOrUpdateMessage(expectedParentId, message);\n        expectedParentId = message.id;\n        lastAppliedId = message.id;\n      }\n\n      if (hardReplace) {\n        this.session.clear();\n        lastAppliedId = this.appendLinearChain(messages);\n      }\n\n      this.session.resetHead(lastAppliedId);\n    }\n\n    this.finalizeExternalApply();\n  }\n\n  // --- Run logic ---\n\n  private async startRun(userThreadMessage: ThreadMessage): Promise<void> {\n    this.runGeneration++;\n\n    // Cancel any in-progress run before starting a new one\n    if (this.abortController) {\n      this.abortController.abort();\n      this.abortController = null;\n    }\n\n    const a2aMessage = threadMessageToA2AMessage(userThreadMessage, {\n      contextId: this.contextId,\n      taskId:\n        this.currentTask?.id &&\n        !isTerminalTaskState(this.currentTask.status.state)\n          ? this.currentTask.id\n          : undefined,\n    });\n\n    // Clear task if previous task reached terminal state\n    if (\n      this.currentTask &&\n      isTerminalTaskState(this.currentTask.status.state)\n    ) {\n      this.currentTask = undefined;\n    }\n\n    this.currentArtifacts = [];\n\n    const assistantParentId = userThreadMessage.id;\n    const assistantId = this.insertAssistantPlaceholder(assistantParentId);\n    this.markPendingAssistantHistory(assistantId, assistantParentId);\n\n    const abortController = new AbortController();\n    this.abortController = abortController;\n\n    abortController.signal.addEventListener(\n      \"abort\",\n      () => {\n        this.updateAssistantStatus(assistantId, {\n          type: \"incomplete\",\n          reason: \"cancelled\",\n        });\n        this.finishRun(abortController);\n        invokeRuntimeCallback(\"onCancel\", this.onCancel);\n      },\n      { once: true },\n    );\n\n    this.setRunning(true);\n\n    try {\n      if (!(await this.waitForAgentCard(abortController.signal))) return;\n\n      const supportsStreaming =\n        this.agentCardValue?.capabilities?.streaming !== false;\n      if (supportsStreaming) {\n        await this.runStreaming(a2aMessage, assistantId, abortController);\n      } else {\n        await this.runSync(a2aMessage, assistantId, abortController);\n      }\n    } catch (error) {\n      if (!abortController.signal.aborted) {\n        const err = error instanceof Error ? error : new Error(String(error));\n        this.updateAssistantStatus(assistantId, {\n          type: \"incomplete\",\n          reason: \"error\",\n        });\n        invokeRuntimeCallback(\"onError\", this.onError, err);\n        this.pendingError = this.pendingError ?? err;\n      }\n    } finally {\n      this.finishRun(abortController);\n    }\n\n    if (this.pendingError) {\n      const err = this.pendingError;\n      this.pendingError = null;\n      throw err;\n    }\n  }\n\n  private async runStreaming(\n    a2aMessage: A2AMessage,\n    assistantId: string,\n    abortController: AbortController,\n  ): Promise<void> {\n    const stream = this.client.streamMessage(\n      a2aMessage,\n      this.configuration,\n      undefined, // metadata\n      abortController.signal,\n    );\n\n    let receivedEvent = false;\n    let skipReason: string | undefined;\n    const diagnosedStream = (async function* () {\n      const reason = yield* stream;\n      if (typeof reason === \"string\") skipReason = reason;\n    })();\n\n    for await (const event of diagnosedStream) {\n      if (abortController.signal.aborted) break;\n      receivedEvent = true;\n      this.handleStreamEvent(assistantId, event);\n    }\n\n    if (!abortController.signal.aborted) {\n      if (!receivedEvent) {\n        throw new Error(\n          skipReason\n            ? `A2A message stream ended without any events. First skipped frame: ${skipReason}`\n            : \"A2A message stream ended without any events.\",\n        );\n      }\n\n      const lastStatus = this.getAssistantStatus(assistantId);\n      if (lastStatus?.type === \"running\") {\n        this.updateAssistantStatus(assistantId, {\n          type: \"complete\",\n          reason: \"stop\",\n        });\n      }\n    }\n  }\n\n  private async runSync(\n    a2aMessage: A2AMessage,\n    assistantId: string,\n    abortController: AbortController,\n  ): Promise<void> {\n    const result = await this.client.sendMessage(\n      a2aMessage,\n      this.configuration,\n      undefined, // metadata\n      abortController.signal,\n    );\n\n    if (abortController.signal.aborted) return;\n\n    // Result is either A2ATask or A2AMessage\n    if (\"id\" in result && \"status\" in result) {\n      // It's a Task\n      this.handleTaskSnapshot(assistantId, result as A2ATask);\n    } else if (\"messageId\" in result && \"parts\" in result) {\n      // It's a Message\n      this.handleMessage(assistantId, result as A2AMessage);\n      this.updateAssistantStatus(assistantId, {\n        type: \"complete\",\n        reason: \"stop\",\n      });\n    }\n  }\n\n  private handleStreamEvent(assistantId: string, event: A2AStreamEvent) {\n    switch (event.type) {\n      case \"statusUpdate\":\n        this.handleStatusUpdate(assistantId, event.event);\n        break;\n      case \"artifactUpdate\":\n        this.handleArtifactUpdate(event.event);\n        break;\n      case \"message\":\n        this.handleMessage(assistantId, event.message);\n        break;\n      case \"task\":\n        this.handleTaskSnapshot(assistantId, event.task);\n        break;\n    }\n  }\n\n  private handleStatusUpdate(\n    assistantId: string,\n    event: A2ATaskStatusUpdateEvent,\n  ) {\n    if (!this.currentTask) {\n      this.currentTask = {\n        id: event.taskId,\n        contextId: event.contextId,\n        status: event.status,\n      };\n    } else {\n      this.currentTask = { ...this.currentTask, status: event.status };\n    }\n\n    if (event.contextId) {\n      this.contextId = event.contextId;\n    }\n\n    if (event.status.message) {\n      const content = a2aMessageToContent(event.status.message);\n      this.updateAssistantContent(assistantId, content);\n    }\n\n    const status = taskStateToMessageStatus(event.status.state);\n    this.updateAssistantStatus(assistantId, status);\n\n    this.notifyUpdate();\n  }\n\n  private handleArtifactUpdate(event: A2ATaskArtifactUpdateEvent) {\n    const { append, lastChunk } = event;\n    const artifact = normalizeArtifact(event.artifact);\n    const existingIdx = this.currentArtifacts.findIndex(\n      (a) => a.artifactId === artifact.artifactId,\n    );\n\n    let updated: A2AArtifact;\n    if (existingIdx >= 0 && append) {\n      const existing = this.currentArtifacts[existingIdx]!;\n      updated = {\n        ...existing,\n        parts: [...existing.parts, ...artifact.parts],\n      };\n      this.currentArtifacts = [\n        ...this.currentArtifacts.slice(0, existingIdx),\n        updated,\n        ...this.currentArtifacts.slice(existingIdx + 1),\n      ];\n    } else if (existingIdx >= 0) {\n      updated = artifact;\n      this.currentArtifacts = [\n        ...this.currentArtifacts.slice(0, existingIdx),\n        updated,\n        ...this.currentArtifacts.slice(existingIdx + 1),\n      ];\n    } else {\n      updated = artifact;\n      this.currentArtifacts = [...this.currentArtifacts, updated];\n    }\n\n    if (lastChunk) {\n      invokeRuntimeCallback(\n        \"onArtifactComplete\",\n        this.onArtifactComplete,\n        updated,\n      );\n    }\n\n    this.notifyUpdate();\n  }\n\n  private handleMessage(assistantId: string, message: A2AMessage) {\n    if (message.role !== \"agent\") return;\n\n    const content = a2aMessageToContent(message);\n    this.updateAssistantContent(assistantId, content);\n    this.notifyUpdate();\n  }\n\n  private handleTaskSnapshot(assistantId: string, task: A2ATask) {\n    const artifacts =\n      task.artifacts === undefined\n        ? undefined\n        : Array.isArray(task.artifacts)\n          ? task.artifacts.map(normalizeArtifact)\n          : [];\n    const history =\n      task.history === undefined\n        ? undefined\n        : Array.isArray(task.history)\n          ? task.history\n          : [];\n    this.currentTask = {\n      ...task,\n      ...(artifacts === undefined ? {} : { artifacts }),\n      ...(history === undefined ? {} : { history }),\n    };\n\n    if (task.contextId) {\n      this.contextId = task.contextId;\n    }\n    if (artifacts) {\n      this.currentArtifacts = artifacts;\n    }\n\n    if (task.status.message) {\n      const content = a2aMessageToContent(task.status.message);\n      this.updateAssistantContent(assistantId, content);\n    }\n\n    const status = taskStateToMessageStatus(task.status.state);\n    this.updateAssistantStatus(assistantId, status);\n\n    this.notifyUpdate();\n  }\n\n  // --- Message helpers ---\n\n  private insertAssistantPlaceholder(parentId: string): string {\n    const id = generateId();\n    const assistant: ThreadAssistantMessage = {\n      id,\n      role: \"assistant\",\n      createdAt: new Date(),\n      status: { type: \"running\" },\n      content: [],\n      metadata: {\n        unstable_state: null,\n        unstable_annotations: [],\n        unstable_data: [],\n        steps: [],\n        custom: {},\n      },\n    };\n    this.session.addOrUpdateMessage(parentId, assistant);\n    this.session.switchToBranch(id);\n    this.notifyUpdate();\n    return id;\n  }\n\n  private updateAssistantContent(\n    messageId: string,\n    content: ThreadAssistantMessage[\"content\"],\n  ) {\n    this.session.updateMessage(messageId, (message) => {\n      if (message.role !== \"assistant\") return message;\n      return { ...message, content };\n    });\n  }\n\n  private updateAssistantStatus(messageId: string, status: MessageStatus) {\n    const touched = this.session.updateMessage(messageId, (message) => {\n      if (message.role !== \"assistant\") return message;\n      return { ...message, status };\n    });\n    if (touched) {\n      this.notifyUpdate();\n      if (status.type === \"complete\" || status.type === \"incomplete\") {\n        this.persistAssistantHistory(messageId);\n      }\n    }\n  }\n\n  private getAssistantStatus(messageId: string): MessageStatus | undefined {\n    const msg = this.session.tryGetMessage(messageId)?.message;\n    if (msg?.role !== \"assistant\") return undefined;\n    return msg.status;\n  }\n\n  // --- Lifecycle helpers ---\n\n  private setRunning(running: boolean) {\n    this.isRunningFlag = running;\n    this.notifyUpdate();\n  }\n\n  private finishRun(controller: AbortController | null) {\n    if (this.abortController !== controller) return;\n    this.abortController = null;\n    this.setRunning(false);\n  }\n\n  // --- History persistence ---\n\n  private recordHistoryEntry(parentId: string | null, message: ThreadMessage) {\n    this.appendHistoryItem(parentId, message);\n  }\n\n  private markPendingAssistantHistory(\n    messageId: string,\n    parentId: string | null,\n  ) {\n    if (!this.history) return;\n    this.assistantHistoryParents.set(messageId, parentId);\n  }\n\n  private persistAssistantHistory(messageId: string) {\n    if (!this.history) return;\n    const parentId = this.assistantHistoryParents.get(messageId);\n    if (parentId === undefined) return;\n    const message = this.session.tryGetMessage(messageId)?.message;\n    if (!message || message.role !== \"assistant\") return;\n    if (\n      message.status?.type !== \"complete\" &&\n      message.status?.type !== \"incomplete\"\n    )\n      return;\n    this.assistantHistoryParents.delete(messageId);\n    this.appendHistoryItem(parentId, message);\n  }\n\n  private appendHistoryItem(parentId: string | null, message: ThreadMessage) {\n    if (!this.history || this.recordedHistoryIds.has(message.id)) return;\n    this.recordedHistoryIds.add(message.id);\n    void this.history.append({ parentId, message }).catch(() => {\n      this.recordedHistoryIds.delete(message.id);\n    });\n  }\n}\n"],"mappings":";;;;;AAmCA,MAAM,oCAAoC;AAC1C,MAAM,gCAAgC;AAatC,MAAM,uBAAuB;CAC3B,MAAM;CACN,QAAQ;AACV;AAIA,MAAM,yBACJ,MACA,UACA,GAAG,SACM;CACT,mBAAwB,aAAa,MAAM,UAAU,GAAG,IAAI;AAC9D;AAEA,SAAS,kBAAkB,UAAoC;CAC7D,OAAO;EACL,GAAG;EACH,OAAO,MAAM,QAAQ,SAAS,KAAK,IAAI,SAAS,QAAQ,CAAC;CAC3D;AACF;AAEA,IAAa,uBAAb,MAAkC;CAChC;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CAEA;CACA,UAA2B,+BAA+B;CAC1D,gBAAwB;CACxB,kBAAkD;CAClD,gBAAwB;CACxB,eAAqC;CAGrC;CACA,mBAA0C,CAAC;CAC3C;CAGA,0CAA2C,IAAI,IAA2B;CAC1E,qCAAsC,IAAI,IAAY;CACtD,aAAqB;CACrB;CACA,yBAAiC;CACjC,iBAAyB;CACzB;CACA,uBAA+B;CAC/B,uBAA+B;CAC/B,4BAAoC;CAEpC;CAEA,YAAY,SAAsC;EAChD,KAAK,SAAS,QAAQ;EACtB,KAAK,YAAY,QAAQ;EACzB,KAAK,uBAAuB,QAAQ;EACpC,KAAK,gBAAgB,QAAQ;EAC7B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,qBAAqB,QAAQ;EAClC,KAAK,UAAU,QAAQ;EACvB,KAAK,eAAe,QAAQ;CAC9B;CAEA,cAAc,SAA4D;EACxE,KAAK,SAAS,QAAQ;EAKtB,IAAI,QAAQ,cAAc,KAAK,sBAAsB;GACnD,KAAK,YAAY,QAAQ;GACzB,KAAK,uBAAuB,QAAQ;EACtC;EACA,KAAK,gBAAgB,QAAQ;EAC7B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,qBAAqB,QAAQ;EAClC,MAAM,kBAAkB,KAAK;EAC7B,KAAK,UAAU,QAAQ;EAEvB,IACE,KAAK,kBACL,CAAC,KAAK,gBACN,CAAC,mBACD,QAAQ,WACR,KAAK,QAAQ,YAAY,CAAC,CAAC,WAAW,GAEtC,KAAU,gBAAgB;CAE9B;;;CAIA,eAAqB;EACnB,KAAK;EACL,KAAK,aAAa;EAIlB,MAAM,aAAa,KAAK;EACxB,KAAK,YAAY,KAAK;EACtB,IAAI,YAAY;GACd,WAAW,MAAM;GACjB,IAAI,KAAK,oBAAoB,YAC3B,KAAK,kBAAkB;EAE3B;CACF;CAEA,cAAc,SAA2B;EACvC,KAAK,UAAU;CACjB;CAEA,gBAAgB;EACd,KAAK,UAAU,KAAA;EAEf,IAAI,KAAK,iBAAiB;GACxB,KAAK,gBAAgB,MAAM;GAC3B,KAAK,kBAAkB;EACzB;CACF;CAEA,aAA2C;EACzC,OAAO,KAAK;CACd;CAEA,cAAwC;EACtC,OAAO,KAAK,QAAQ,YAAY;CAClC;CAEA,uBAAkD;EAChD,OAAO,KAAK,QAAQ,OAAO;CAC7B;CAEA,UAA+B;EAC7B,OAAO,KAAK;CACd;CAEA,eAAuC;EACrC,OAAO,KAAK;CACd;CAEA,eAAyC;EACvC,OAAO,KAAK;CACd;CAEA,YAAqB;EACnB,OAAO,KAAK;CACd;CAEA,IAAI,YAAqB;EACvB,OAAO,KAAK;CACd;CAEA,gBAAuC;EACrC,IAAI,KAAK,IAAI,IAAI,KAAK,sBAAsB,OAAO,QAAQ,QAAQ;EAEnE,KAAK,sBAAsB,KAAK,OAAO,aAAa,CAAC,CAAC,MACnD,cAAc;GACb,KAAK,iBAAiB;GACtB,KAAK,uBAAuB;GAC5B,KAAK,uBAAuB;GAC5B,KAAK,4BAA4B;GACjC,KAAK,aAAa;EACpB,SACM;GACJ,KAAK,4BAA4B;GACjC,KAAK,uBAAuB,KAAK,IAAI,IAAI,KAAK;GAC9C,KAAK,uBAAuB,KAAK,IAC/B,KAAK,uBAAuB,GAC5B,6BACF;GACA,KAAK,oBAAoB,KAAA;EAC3B,CACF;EACA,OAAO,KAAK;CACd;CAEA,MAAc,iBAAiB,QAAuC;EACpE,MAAM,aAAa,CAAC,KAAK;EACzB,MAAM,OAAO,KAAK,cAAc;EAChC,IAAI,OAAO,SAAS,OAAO;EAC3B,IAAI,CAAC,YAAY,OAAO;EAExB,IAAI;EACJ,MAAM,QAAQ,IAAI,SAAe,YAAY;GAC3C,UAAU;GACV,OAAO,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;EAC1D,CAAC;EAED,IAAI;GACF,MAAM,QAAQ,KAAK,CAAC,MAAM,KAAK,CAAC;EAClC,UAAU;GACR,OAAO,oBAAoB,SAAS,OAAO;EAC7C;EACA,OAAO,CAAC,OAAO;CACjB;CAEA,kBAAiC;EAC/B,KAAK,iBAAiB;EACtB,MAAM,mBAAmB,KAAK,cAAc;EAE5C,IAAI,KAAK,cAAc,OAAO,KAAK;EACnC,IAAI,CAAC,KAAK,SAAS,OAAO;EAE1B,KAAK,aAAa;EAElB,MAAM,aAAa,KAAK;EACxB,MAAM,iBAAiB,KAAK,QAAQ,KAAK;EAEzC,KAAK,eAAe,QAAQ,IAAI,CAAC,gBAAgB,gBAAgB,CAAC,CAAC,CAChE,MAAM,CAAC,UAAU;GAChB,IAAI,eAAe,KAAK,wBAAwB;GAChD,IAAI,MAAM;IACR,KAAK,QAAQ,+BAA+B,IAAI;IAChD,KAAK,sBAAsB;GAC7B;EACF,CAAC,CAAC,CACD,OAAO,UAAU;GAChB,IAAI,eAAe,KAAK,wBAAwB;GAChD,sBACE,WACA,KAAK,SACL,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAC1D;EACF,CAAC,CAAC,CACD,cAAc;GACb,IAAI,eAAe,KAAK,wBAAwB;GAChD,KAAK,aAAa;GAClB,KAAK,aAAa;EACpB,CAAC;EAEH,KAAK,aAAa;EAClB,OAAO,KAAK;CACd;CAEA,MAAM,OAAO,SAAuC;EAClD,MAAM,WAAW,QAAQ,YAAY,QAAQ,SAAS;EAEtD,MAAM,gBAAgB,sBACpB,SACA,WAAW,GACX,oBACF;EACA,MAAM,WACJ,QAAQ,aAAa,OACjB,OACA,QAAQ,YAAY,KAAK,QAAQ,WAAW,QAAQ,QAAQ,IAC1D,QAAQ,WACR,KAAK,QAAQ;EACrB,KAAK,QAAQ,mBAAmB,UAAU,aAAa;EACvD,KAAK,QAAQ,eAAe,cAAc,EAAE;EAC5C,KAAK,aAAa;EAClB,KAAK,mBAAmB,UAAU,aAAa;EAE/C,IAAI,CAAC,UAAU;EACf,MAAM,KAAK,SAAS,aAAa;CACnC;CAEA,MAAM,KAAK,SAAuC;EAChD,MAAM,KAAK,OAAO,OAAO;CAC3B;CAEA,MAAM,OACJ,UACA,UAAmD,CAAC,GACrC;EACf,MAAM,WACJ,aAAa,OACT,CAAC,IACA,KAAK,QAAQ,eAAe,QAAQ,KAAK,KAAK,YAAY;EACjE,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KACxC,IAAI,SAAS,EAAE,CAAE,SAAS,QAAQ;GAChC,MAAM,KAAK,SAAS,SAAS,EAAG;GAChC;EACF;CAEJ;CAEA,MAAM,SAAwB;EAC5B,IAAI,CAAC,KAAK,iBAAiB;EAK3B,MAAM,OAAO,KAAK;EAClB,MAAM,aAAa,KAAK;EAGxB,KAAK,gBAAgB,MAAM;EAG3B,IAAI,MAAM,IACR,IAAI;GACF,MAAM,UAAU,MAAM,KAAK,OAAO,WAAW,KAAK,EAAE;GAKpD,IAAI,KAAK,gBAAgB,QAAQ,KAAK,kBAAkB,YACtD,KAAK,cAAc;EAEvB,QAAQ,CAER;CAEJ;CAEA,kBAA0B,UAAmD;EAC3E,IAAI,WAA0B;EAC9B,IAAI,SAAwB;EAC5B,MAAM,uBAAO,IAAI,IAAY;EAE7B,KAAK,MAAM,WAAW,UAAU;GAC9B,IAAI,KAAK,IAAI,QAAQ,EAAE,GAAG;GAC1B,KAAK,IAAI,QAAQ,EAAE;GACnB,KAAK,QAAQ,mBAAmB,UAAU,OAAO;GACjD,WAAW,QAAQ;GACnB,SAAS,QAAQ;EACnB;EAEA,OAAO;CACT;CAEA,wBAAsC;EACpC,KAAK,wBAAwB,MAAM;EACnC,KAAK,mBAAmB,MAAM;EAC9B,KAAK,MAAM,EAAE,aAAa,KAAK,qBAAqB,CAAC,CAAC,UACpD,KAAK,mBAAmB,IAAI,QAAQ,EAAE;EAExC,KAAK,cAAc,KAAA;EACnB,KAAK,mBAAmB,CAAC;EACzB,KAAK,aAAa;CACpB;CAEA,sBAAsB,UAA0C;EAC9D,IAAI,SAAS,WAAW,GACtB,KAAK,QAAQ,MAAM;OACd;GACL,IAAI,mBAAkC;GACtC,IAAI,gBAA+B;GACnC,IAAI,cAAc;GAClB,MAAM,uBAAO,IAAI,IAAY;GAE7B,KAAK,MAAM,WAAW,UAAU;IAC9B,IAAI,KAAK,IAAI,QAAQ,EAAE,GAAG;IAC1B,KAAK,IAAI,QAAQ,EAAE;IACnB,MAAM,WAAW,KAAK,QAAQ,cAAc,QAAQ,EAAE;IACtD,IAAI,YAAY,SAAS,aAAa,kBAAkB;KACtD,cAAc;KACd;IACF;IACA,KAAK,QAAQ,mBAAmB,kBAAkB,OAAO;IACzD,mBAAmB,QAAQ;IAC3B,gBAAgB,QAAQ;GAC1B;GAEA,IAAI,aAAa;IACf,KAAK,QAAQ,MAAM;IACnB,gBAAgB,KAAK,kBAAkB,QAAQ;GACjD;GAEA,KAAK,QAAQ,UAAU,aAAa;EACtC;EAEA,KAAK,sBAAsB;CAC7B;CAIA,MAAc,SAAS,mBAAiD;EACtE,KAAK;EAGL,IAAI,KAAK,iBAAiB;GACxB,KAAK,gBAAgB,MAAM;GAC3B,KAAK,kBAAkB;EACzB;EAEA,MAAM,aAAa,0BAA0B,mBAAmB;GAC9D,WAAW,KAAK;GAChB,QACE,KAAK,aAAa,MAClB,CAAC,oBAAoB,KAAK,YAAY,OAAO,KAAK,IAC9C,KAAK,YAAY,KACjB,KAAA;EACR,CAAC;EAGD,IACE,KAAK,eACL,oBAAoB,KAAK,YAAY,OAAO,KAAK,GAEjD,KAAK,cAAc,KAAA;EAGrB,KAAK,mBAAmB,CAAC;EAEzB,MAAM,oBAAoB,kBAAkB;EAC5C,MAAM,cAAc,KAAK,2BAA2B,iBAAiB;EACrE,KAAK,4BAA4B,aAAa,iBAAiB;EAE/D,MAAM,kBAAkB,IAAI,gBAAgB;EAC5C,KAAK,kBAAkB;EAEvB,gBAAgB,OAAO,iBACrB,eACM;GACJ,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;GACD,KAAK,UAAU,eAAe;GAC9B,sBAAsB,YAAY,KAAK,QAAQ;EACjD,GACA,EAAE,MAAM,KAAK,CACf;EAEA,KAAK,WAAW,IAAI;EAEpB,IAAI;GACF,IAAI,CAAE,MAAM,KAAK,iBAAiB,gBAAgB,MAAM,GAAI;GAI5D,IADE,KAAK,gBAAgB,cAAc,cAAc,OAEjD,MAAM,KAAK,aAAa,YAAY,aAAa,eAAe;QAEhE,MAAM,KAAK,QAAQ,YAAY,aAAa,eAAe;EAE/D,SAAS,OAAO;GACd,IAAI,CAAC,gBAAgB,OAAO,SAAS;IACnC,MAAM,MAAM,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;IACpE,KAAK,sBAAsB,aAAa;KACtC,MAAM;KACN,QAAQ;IACV,CAAC;IACD,sBAAsB,WAAW,KAAK,SAAS,GAAG;IAClD,KAAK,eAAe,KAAK,gBAAgB;GAC3C;EACF,UAAU;GACR,KAAK,UAAU,eAAe;EAChC;EAEA,IAAI,KAAK,cAAc;GACrB,MAAM,MAAM,KAAK;GACjB,KAAK,eAAe;GACpB,MAAM;EACR;CACF;CAEA,MAAc,aACZ,YACA,aACA,iBACe;EACf,MAAM,SAAS,KAAK,OAAO,cACzB,YACA,KAAK,eACL,KAAA,GACA,gBAAgB,MAClB;EAEA,IAAI,gBAAgB;EACpB,IAAI;EACJ,MAAM,mBAAmB,mBAAmB;GAC1C,MAAM,SAAS,OAAO;GACtB,IAAI,OAAO,WAAW,UAAU,aAAa;EAC/C,EAAA,CAAG;EAEH,WAAW,MAAM,SAAS,iBAAiB;GACzC,IAAI,gBAAgB,OAAO,SAAS;GACpC,gBAAgB;GAChB,KAAK,kBAAkB,aAAa,KAAK;EAC3C;EAEA,IAAI,CAAC,gBAAgB,OAAO,SAAS;GACnC,IAAI,CAAC,eACH,MAAM,IAAI,MACR,aACI,qEAAqE,eACrE,8CACN;GAIF,IADmB,KAAK,mBAAmB,WAC9B,CAAC,EAAE,SAAS,WACvB,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;EAEL;CACF;CAEA,MAAc,QACZ,YACA,aACA,iBACe;EACf,MAAM,SAAS,MAAM,KAAK,OAAO,YAC/B,YACA,KAAK,eACL,KAAA,GACA,gBAAgB,MAClB;EAEA,IAAI,gBAAgB,OAAO,SAAS;EAGpC,IAAI,QAAQ,UAAU,YAAY,QAEhC,KAAK,mBAAmB,aAAa,MAAiB;OACjD,IAAI,eAAe,UAAU,WAAW,QAAQ;GAErD,KAAK,cAAc,aAAa,MAAoB;GACpD,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;EACH;CACF;CAEA,kBAA0B,aAAqB,OAAuB;EACpE,QAAQ,MAAM,MAAd;GACE,KAAK;IACH,KAAK,mBAAmB,aAAa,MAAM,KAAK;IAChD;GACF,KAAK;IACH,KAAK,qBAAqB,MAAM,KAAK;IACrC;GACF,KAAK;IACH,KAAK,cAAc,aAAa,MAAM,OAAO;IAC7C;GACF,KAAK,QACH,KAAK,mBAAmB,aAAa,MAAM,IAAI;EAEnD;CACF;CAEA,mBACE,aACA,OACA;EACA,IAAI,CAAC,KAAK,aACR,KAAK,cAAc;GACjB,IAAI,MAAM;GACV,WAAW,MAAM;GACjB,QAAQ,MAAM;EAChB;OAEA,KAAK,cAAc;GAAE,GAAG,KAAK;GAAa,QAAQ,MAAM;EAAO;EAGjE,IAAI,MAAM,WACR,KAAK,YAAY,MAAM;EAGzB,IAAI,MAAM,OAAO,SAAS;GACxB,MAAM,UAAU,oBAAoB,MAAM,OAAO,OAAO;GACxD,KAAK,uBAAuB,aAAa,OAAO;EAClD;EAEA,MAAM,SAAS,yBAAyB,MAAM,OAAO,KAAK;EAC1D,KAAK,sBAAsB,aAAa,MAAM;EAE9C,KAAK,aAAa;CACpB;CAEA,qBAA6B,OAAmC;EAC9D,MAAM,EAAE,QAAQ,cAAc;EAC9B,MAAM,WAAW,kBAAkB,MAAM,QAAQ;EACjD,MAAM,cAAc,KAAK,iBAAiB,WACvC,MAAM,EAAE,eAAe,SAAS,UACnC;EAEA,IAAI;EACJ,IAAI,eAAe,KAAK,QAAQ;GAC9B,MAAM,WAAW,KAAK,iBAAiB;GACvC,UAAU;IACR,GAAG;IACH,OAAO,CAAC,GAAG,SAAS,OAAO,GAAG,SAAS,KAAK;GAC9C;GACA,KAAK,mBAAmB;IACtB,GAAG,KAAK,iBAAiB,MAAM,GAAG,WAAW;IAC7C;IACA,GAAG,KAAK,iBAAiB,MAAM,cAAc,CAAC;GAChD;EACF,OAAO,IAAI,eAAe,GAAG;GAC3B,UAAU;GACV,KAAK,mBAAmB;IACtB,GAAG,KAAK,iBAAiB,MAAM,GAAG,WAAW;IAC7C;IACA,GAAG,KAAK,iBAAiB,MAAM,cAAc,CAAC;GAChD;EACF,OAAO;GACL,UAAU;GACV,KAAK,mBAAmB,CAAC,GAAG,KAAK,kBAAkB,OAAO;EAC5D;EAEA,IAAI,WACF,sBACE,sBACA,KAAK,oBACL,OACF;EAGF,KAAK,aAAa;CACpB;CAEA,cAAsB,aAAqB,SAAqB;EAC9D,IAAI,QAAQ,SAAS,SAAS;EAE9B,MAAM,UAAU,oBAAoB,OAAO;EAC3C,KAAK,uBAAuB,aAAa,OAAO;EAChD,KAAK,aAAa;CACpB;CAEA,mBAA2B,aAAqB,MAAe;EAC7D,MAAM,YACJ,KAAK,cAAc,KAAA,IACf,KAAA,IACA,MAAM,QAAQ,KAAK,SAAS,IAC1B,KAAK,UAAU,IAAI,iBAAiB,IACpC,CAAC;EACT,MAAM,UACJ,KAAK,YAAY,KAAA,IACb,KAAA,IACA,MAAM,QAAQ,KAAK,OAAO,IACxB,KAAK,UACL,CAAC;EACT,KAAK,cAAc;GACjB,GAAG;GACH,GAAI,cAAc,KAAA,IAAY,CAAC,IAAI,EAAE,UAAU;GAC/C,GAAI,YAAY,KAAA,IAAY,CAAC,IAAI,EAAE,QAAQ;EAC7C;EAEA,IAAI,KAAK,WACP,KAAK,YAAY,KAAK;EAExB,IAAI,WACF,KAAK,mBAAmB;EAG1B,IAAI,KAAK,OAAO,SAAS;GACvB,MAAM,UAAU,oBAAoB,KAAK,OAAO,OAAO;GACvD,KAAK,uBAAuB,aAAa,OAAO;EAClD;EAEA,MAAM,SAAS,yBAAyB,KAAK,OAAO,KAAK;EACzD,KAAK,sBAAsB,aAAa,MAAM;EAE9C,KAAK,aAAa;CACpB;CAIA,2BAAmC,UAA0B;EAC3D,MAAM,KAAK,WAAW;EACtB,MAAM,YAAoC;GACxC;GACA,MAAM;GACN,2BAAW,IAAI,KAAK;GACpB,QAAQ,EAAE,MAAM,UAAU;GAC1B,SAAS,CAAC;GACV,UAAU;IACR,gBAAgB;IAChB,sBAAsB,CAAC;IACvB,eAAe,CAAC;IAChB,OAAO,CAAC;IACR,QAAQ,CAAC;GACX;EACF;EACA,KAAK,QAAQ,mBAAmB,UAAU,SAAS;EACnD,KAAK,QAAQ,eAAe,EAAE;EAC9B,KAAK,aAAa;EAClB,OAAO;CACT;CAEA,uBACE,WACA,SACA;EACA,KAAK,QAAQ,cAAc,YAAY,YAAY;GACjD,IAAI,QAAQ,SAAS,aAAa,OAAO;GACzC,OAAO;IAAE,GAAG;IAAS;GAAQ;EAC/B,CAAC;CACH;CAEA,sBAA8B,WAAmB,QAAuB;EAKtE,IAJgB,KAAK,QAAQ,cAAc,YAAY,YAAY;GACjE,IAAI,QAAQ,SAAS,aAAa,OAAO;GACzC,OAAO;IAAE,GAAG;IAAS;GAAO;EAC9B,CACU,GAAG;GACX,KAAK,aAAa;GAClB,IAAI,OAAO,SAAS,cAAc,OAAO,SAAS,cAChD,KAAK,wBAAwB,SAAS;EAE1C;CACF;CAEA,mBAA2B,WAA8C;EACvE,MAAM,MAAM,KAAK,QAAQ,cAAc,SAAS,CAAC,EAAE;EACnD,IAAI,KAAK,SAAS,aAAa,OAAO,KAAA;EACtC,OAAO,IAAI;CACb;CAIA,WAAmB,SAAkB;EACnC,KAAK,gBAAgB;EACrB,KAAK,aAAa;CACpB;CAEA,UAAkB,YAAoC;EACpD,IAAI,KAAK,oBAAoB,YAAY;EACzC,KAAK,kBAAkB;EACvB,KAAK,WAAW,KAAK;CACvB;CAIA,mBAA2B,UAAyB,SAAwB;EAC1E,KAAK,kBAAkB,UAAU,OAAO;CAC1C;CAEA,4BACE,WACA,UACA;EACA,IAAI,CAAC,KAAK,SAAS;EACnB,KAAK,wBAAwB,IAAI,WAAW,QAAQ;CACtD;CAEA,wBAAgC,WAAmB;EACjD,IAAI,CAAC,KAAK,SAAS;EACnB,MAAM,WAAW,KAAK,wBAAwB,IAAI,SAAS;EAC3D,IAAI,aAAa,KAAA,GAAW;EAC5B,MAAM,UAAU,KAAK,QAAQ,cAAc,SAAS,CAAC,EAAE;EACvD,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;EAC9C,IACE,QAAQ,QAAQ,SAAS,cACzB,QAAQ,QAAQ,SAAS,cAEzB;EACF,KAAK,wBAAwB,OAAO,SAAS;EAC7C,KAAK,kBAAkB,UAAU,OAAO;CAC1C;CAEA,kBAA0B,UAAyB,SAAwB;EACzE,IAAI,CAAC,KAAK,WAAW,KAAK,mBAAmB,IAAI,QAAQ,EAAE,GAAG;EAC9D,KAAK,mBAAmB,IAAI,QAAQ,EAAE;EACtC,KAAU,QAAQ,OAAO;GAAE;GAAU;EAAQ,CAAC,CAAC,CAAC,YAAY;GAC1D,KAAK,mBAAmB,OAAO,QAAQ,EAAE;EAC3C,CAAC;CACH;AACF"}

package/dist/a2aExtras.d.ts

@@ -1,6 +1,5 @@
 import { A2AExtras } from "./types.js";
 //#region src/a2aExtras.d.ts
-declare const a2aExtras: import("@assistant-ui/core/react").RuntimeExtras<A2AExtras>;
+export declare const a2aExtras: import("@assistant-ui/core/react").RuntimeExtras<A2AExtras>;
 //#endregion
-export { a2aExtras };
 //# sourceMappingURL=a2aExtras.d.ts.map

package/dist/a2aExtras.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"a2aExtras.d.ts","names":[],"sources":["../src/a2aExtras.ts"],"mappings":";;cAGa,8CAAS,cAAA"}
+{"version":3,"file":"a2aExtras.d.ts","names":[],"sources":["../src/a2aExtras.ts"],"mappings":";;qBAGa,8CAAS,cAAA"}

package/dist/conversions.d.ts

@@ -1,12 +1,12 @@
 import { A2AMessage, A2APart, A2ATaskState } from "./types.js";
 import { MessageStatus, ThreadAssistantMessage, ThreadMessage } from "@assistant-ui/core";
 //#region src/conversions.d.ts
-declare function a2aPartToContent(part: A2APart): ThreadAssistantMessage["content"][number];
-declare function a2aPartsToContent(parts: A2APart[]): ThreadAssistantMessage["content"];
-declare function isTerminalTaskState(state: A2ATaskState): boolean;
-declare function isInterruptedTaskState(state: A2ATaskState): boolean;
-declare function taskStateToMessageStatus(state: A2ATaskState): MessageStatus;
-declare function contentPartsToA2AParts(content: ReadonlyArray<{
+export declare function a2aPartToContent(part: A2APart): ThreadAssistantMessage["content"][number];
+export declare function a2aPartsToContent(parts: A2APart[]): ThreadAssistantMessage["content"];
+export declare function isTerminalTaskState(state: A2ATaskState): boolean;
+export declare function isInterruptedTaskState(state: A2ATaskState): boolean;
+export declare function taskStateToMessageStatus(state: A2ATaskState): MessageStatus;
+export declare function contentPartsToA2AParts(content: ReadonlyArray<{
   type: string;
   text?: string | undefined;
   image?: string | undefined;
@@ -19,11 +19,10 @@ declare function contentPartsToA2AParts(content: ReadonlyArray<{
     format: string;
   } | undefined;
 }>, fallbackMimeType?: string): A2APart[];
-declare function a2aMessageToContent(message: A2AMessage): ThreadAssistantMessage["content"];
-declare function threadMessageToA2AMessage(message: ThreadMessage, options?: {
+export declare function a2aMessageToContent(message: A2AMessage): ThreadAssistantMessage["content"];
+export declare function threadMessageToA2AMessage(message: ThreadMessage, options?: {
   contextId?: string | undefined;
   taskId?: string | undefined;
 }): A2AMessage;
 //#endregion
-export { a2aMessageToContent, a2aPartToContent, a2aPartsToContent, contentPartsToA2AParts, isInterruptedTaskState, isTerminalTaskState, taskStateToMessageStatus, threadMessageToA2AMessage };
 //# sourceMappingURL=conversions.d.ts.map

package/dist/conversions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"conversions.d.ts","names":[],"sources":["../src/conversions.ts"],"mappings":";;;iBAiBgB,iBACd,MAAM,UACL;iBAyCa,kBACd,OAAO,YACN;iBAgBa,oBAAoB,OAAO;iBAI3B,uBAAuB,OAAO;iBAI9B,yBAAyB,OAAO,eAAe;iBAoB/C,uBACd,SAAS;EACP;EACA;EACA;EACA;EACA;EACA;EACA;EACA;IAAU;IAAc;;IAE1B,4BACC;iBA4Ea,oBACd,SAAS,aACR;iBAIa,0BACd,SAAS,eACT;EACE;EACA;IAED"}
+{"version":3,"file":"conversions.d.ts","names":[],"sources":["../src/conversions.ts"],"mappings":";;;wBAiBgB,iBACd,MAAM,UACL;wBAyCa,kBACd,OAAO,YACN;wBAgBa,oBAAoB,OAAO;wBAI3B,uBAAuB,OAAO;wBAI9B,yBAAyB,OAAO,eAAe;wBAoB/C,uBACd,SAAS;EACP;EACA;EACA;EACA;EACA;EACA;EACA;EACA;IAAU;IAAc;;IAE1B,4BACC;wBA4Ea,oBACd,SAAS,aACR;wBAIa,0BACd,SAAS,eACT;EACE;EACA;IAED"}

package/dist/hooks.d.ts

@@ -1,11 +1,10 @@
 import { A2AAgentCard, A2AArtifact, A2ATask } from "./types.js";
 //#region src/hooks.d.ts
 /** The current A2A task, if a run is in progress. */
-declare const useA2ATask: () => A2ATask | undefined;
+export declare const useA2ATask: () => A2ATask | undefined;
 /** The artifacts accumulated for the current A2A run. */
-declare const useA2AArtifacts: () => readonly A2AArtifact[];
+export declare const useA2AArtifacts: () => readonly A2AArtifact[];
 /** The discovered A2A agent card. */
-declare const useA2AAgentCard: () => A2AAgentCard | undefined;
+export declare const useA2AAgentCard: () => A2AAgentCard | undefined;
 //#endregion
-export { useA2AAgentCard, useA2AArtifacts, useA2ATask };
 //# sourceMappingURL=hooks.d.ts.map

package/dist/hooks.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"hooks.d.ts","names":[],"sources":["../src/hooks.ts"],"mappings":";;;cAKa,kBAAA;;cAGA,gCAAA;;cAGA,uBAAA"}
+{"version":3,"file":"hooks.d.ts","names":[],"sources":["../src/hooks.ts"],"mappings":";;;qBAKa,kBAAA;;qBAGA,gCAAA;;qBAGA,uBAAA"}

package/dist/types.d.ts

@@ -1,9 +1,9 @@
 import { A2AClient, A2AClientOptions } from "./A2AClient.js";
 import { AttachmentAdapter, DictationAdapter, ExternalStoreSharedOptions, FeedbackAdapter, RealtimeVoiceAdapter, SpeechSynthesisAdapter, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
 //#region src/types.d.ts
-declare const A2A_PROTOCOL_VERSION = "1.0";
-type A2ARole = "unspecified" | "user" | "agent";
-type A2APart = {
+export declare const A2A_PROTOCOL_VERSION = "1.0";
+export type A2ARole = "unspecified" | "user" | "agent";
+export type A2APart = {
   text?: string | undefined;
   raw?: string | undefined;
   url?: string | undefined;
@@ -12,7 +12,7 @@ type A2APart = {
   filename?: string | undefined;
   mediaType?: string | undefined;
 };
-type A2AMessage = {
+export type A2AMessage = {
   messageId: string;
   contextId?: string | undefined;
   taskId?: string | undefined;
@@ -22,13 +22,13 @@ type A2AMessage = {
   extensions?: string[] | undefined;
   referenceTaskIds?: string[] | undefined;
 };
-type A2ATaskState = "unspecified" | "submitted" | "working" | "completed" | "failed" | "canceled" | "input_required" | "rejected" | "auth_required";
-type A2ATaskStatus = {
+export type A2ATaskState = "unspecified" | "submitted" | "working" | "completed" | "failed" | "canceled" | "input_required" | "rejected" | "auth_required";
+export type A2ATaskStatus = {
   state: A2ATaskState;
   message?: A2AMessage | undefined;
   timestamp?: string | undefined;
 };
-type A2AArtifact = {
+export type A2AArtifact = {
   artifactId: string;
   name?: string | undefined;
   description?: string | undefined;
@@ -36,7 +36,7 @@ type A2AArtifact = {
   metadata?: Record<string, unknown> | undefined;
   extensions?: string[] | undefined;
 };
-type A2ATask = {
+export type A2ATask = {
   id: string;
   contextId?: string | undefined;
   status: A2ATaskStatus;
@@ -44,13 +44,13 @@ type A2ATask = {
   history?: A2AMessage[] | undefined;
   metadata?: Record<string, unknown> | undefined;
 };
-type A2ATaskStatusUpdateEvent = {
+export type A2ATaskStatusUpdateEvent = {
   taskId: string;
   contextId: string;
   status: A2ATaskStatus;
   metadata?: Record<string, unknown> | undefined;
 };
-type A2ATaskArtifactUpdateEvent = {
+export type A2ATaskArtifactUpdateEvent = {
   taskId: string;
   contextId: string;
   artifact: A2AArtifact;
@@ -58,7 +58,7 @@ type A2ATaskArtifactUpdateEvent = {
   lastChunk?: boolean | undefined;
   metadata?: Record<string, unknown> | undefined;
 };
-type A2AStreamEvent = {
+export type A2AStreamEvent = {
   type: "task";
   task: A2ATask;
 } | {
@@ -71,11 +71,11 @@ type A2AStreamEvent = {
   type: "artifactUpdate";
   event: A2ATaskArtifactUpdateEvent;
 };
-type A2AAuthenticationInfo = {
+export type A2AAuthenticationInfo = {
   scheme: string;
   credentials?: string | undefined;
 };
-type A2ATaskPushNotificationConfig = {
+export type A2ATaskPushNotificationConfig = {
   tenant?: string | undefined;
   id?: string | undefined;
   taskId?: string | undefined;
@@ -83,17 +83,17 @@ type A2ATaskPushNotificationConfig = {
   token?: string | undefined;
   authentication?: A2AAuthenticationInfo | undefined;
 };
-type A2AListTaskPushNotificationConfigsResponse = {
+export type A2AListTaskPushNotificationConfigsResponse = {
   configs: A2ATaskPushNotificationConfig[];
   nextPageToken?: string | undefined;
 };
-type A2ASendMessageConfiguration = {
+export type A2ASendMessageConfiguration = {
   acceptedOutputModes?: string[] | undefined;
   taskPushNotificationConfig?: A2ATaskPushNotificationConfig | undefined;
   historyLength?: number | undefined;
   returnImmediately?: boolean | undefined;
 };
-type A2AListTasksRequest = {
+export type A2AListTasksRequest = {
   contextId?: string | undefined;
   status?: A2ATaskState | undefined;
   pageSize?: number | undefined;
@@ -102,59 +102,59 @@ type A2AListTasksRequest = {
   statusTimestampAfter?: string | undefined;
   includeArtifacts?: boolean | undefined;
 };
-type A2AListTasksResponse = {
+export type A2AListTasksResponse = {
   tasks: A2ATask[];
   nextPageToken: string;
   pageSize: number;
   totalSize: number;
 };
-type A2AErrorInfo = {
+export type A2AErrorInfo = {
   code: number;
   status: string;
   message: string;
   details?: unknown[] | undefined;
 };
-type A2AApiKeySecurityScheme = {
+export type A2AApiKeySecurityScheme = {
   description?: string | undefined;
   location: string;
   name: string;
 };
-type A2AHttpAuthSecurityScheme = {
+export type A2AHttpAuthSecurityScheme = {
   description?: string | undefined;
   scheme: string;
   bearerFormat?: string | undefined;
 };
-type A2AAuthorizationCodeOAuthFlow = {
+export type A2AAuthorizationCodeOAuthFlow = {
   authorizationUrl: string;
   tokenUrl: string;
   refreshUrl?: string | undefined;
   scopes: Record<string, string>;
   pkceRequired?: boolean | undefined;
 };
-type A2AClientCredentialsOAuthFlow = {
+export type A2AClientCredentialsOAuthFlow = {
   tokenUrl: string;
   refreshUrl?: string | undefined;
   scopes: Record<string, string>;
 };
-type A2ADeviceCodeOAuthFlow = {
+export type A2ADeviceCodeOAuthFlow = {
   deviceAuthorizationUrl: string;
   tokenUrl: string;
   refreshUrl?: string | undefined;
   scopes: Record<string, string>;
 };
 /** @deprecated Use Authorization Code + PKCE instead. */
-type A2AImplicitOAuthFlow = {
+export type A2AImplicitOAuthFlow = {
   authorizationUrl?: string | undefined;
   refreshUrl?: string | undefined;
   scopes?: Record<string, string> | undefined;
 };
 /** @deprecated Use Authorization Code + PKCE or Device Code. */
-type A2APasswordOAuthFlow = {
+export type A2APasswordOAuthFlow = {
   tokenUrl?: string | undefined;
   refreshUrl?: string | undefined;
   scopes?: Record<string, string> | undefined;
 };
-type A2AOAuthFlows = {
+export type A2AOAuthFlows = {
   authorizationCode?: A2AAuthorizationCodeOAuthFlow | undefined;
   clientCredentials?: A2AClientCredentialsOAuthFlow | undefined;
   /** @deprecated */
@@ -163,36 +163,36 @@ type A2AOAuthFlows = {
   password?: A2APasswordOAuthFlow | undefined;
   deviceCode?: A2ADeviceCodeOAuthFlow | undefined;
 };
-type A2AOAuth2SecurityScheme = {
+export type A2AOAuth2SecurityScheme = {
   description?: string | undefined;
   flows: A2AOAuthFlows;
   oauth2MetadataUrl?: string | undefined;
 };
-type A2AOpenIdConnectSecurityScheme = {
+export type A2AOpenIdConnectSecurityScheme = {
   description?: string | undefined;
   openIdConnectUrl: string;
 };
-type A2AMutualTlsSecurityScheme = {
+export type A2AMutualTlsSecurityScheme = {
   description?: string | undefined;
 };
-type A2ASecurityScheme = {
+export type A2ASecurityScheme = {
   apiKeySecurityScheme?: A2AApiKeySecurityScheme | undefined;
   httpAuthSecurityScheme?: A2AHttpAuthSecurityScheme | undefined;
   oauth2SecurityScheme?: A2AOAuth2SecurityScheme | undefined;
   openIdConnectSecurityScheme?: A2AOpenIdConnectSecurityScheme | undefined;
   mtlsSecurityScheme?: A2AMutualTlsSecurityScheme | undefined;
 };
-type A2ASecurityRequirement = {
+export type A2ASecurityRequirement = {
   schemes: Record<string, {
     list: string[];
   }>;
 };
-type A2AAgentCardSignature = {
+export type A2AAgentCardSignature = {
   protected: string;
   signature: string;
   header?: Record<string, unknown> | undefined;
 };
-type A2AAgentSkill = {
+export type A2AAgentSkill = {
   id: string;
   name: string;
   description: string;
@@ -202,7 +202,7 @@ type A2AAgentSkill = {
   outputModes?: string[] | undefined;
   securityRequirements?: A2ASecurityRequirement[] | undefined;
 };
-type A2AAgentCapabilities = {
+export type A2AAgentCapabilities = {
   streaming?: boolean | undefined;
   pushNotifications?: boolean | undefined;
   extensions?: Array<{
@@ -213,13 +213,13 @@ type A2AAgentCapabilities = {
   }> | undefined;
   extendedAgentCard?: boolean | undefined;
 };
-type A2AAgentInterface = {
+export type A2AAgentInterface = {
   url: string;
   protocolBinding: string;
   protocolVersion: string;
   tenant?: string | undefined;
 };
-type A2AAgentCard = {
+export type A2AAgentCard = {
   name: string;
   description: string;
   supportedInterfaces: A2AAgentInterface[];
@@ -239,19 +239,19 @@ type A2AAgentCard = {
   iconUrl?: string | undefined;
 };
 /** Private state `useA2ARuntime` exposes through `thread.extras`. */
-type A2AExtras = {
+export type A2AExtras = {
   task: A2ATask | undefined;
   artifacts: readonly A2AArtifact[];
   agentCard: A2AAgentCard | undefined;
 };
-type UseA2AThreadListAdapter = {
+export type UseA2AThreadListAdapter = {
   threadId?: string;
   onSwitchToNewThread?: () => Promise<void> | void;
   onSwitchToThread?: (threadId: string) => Promise<{
     messages: readonly ThreadMessage[];
   }>;
 };
-type UseA2ARuntimeOptions = ExternalStoreSharedOptions & {
+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. */
@@ -287,5 +287,4 @@ type UseA2ARuntimeOptions = ExternalStoreSharedOptions & {
   };
 };
 //#endregion
-export { A2AAgentCapabilities, A2AAgentCard, A2AAgentCardSignature, A2AAgentInterface, A2AAgentSkill, A2AApiKeySecurityScheme, A2AArtifact, A2AAuthenticationInfo, A2AAuthorizationCodeOAuthFlow, A2AClientCredentialsOAuthFlow, A2ADeviceCodeOAuthFlow, A2AErrorInfo, A2AExtras, A2AHttpAuthSecurityScheme, A2AImplicitOAuthFlow, A2AListTaskPushNotificationConfigsResponse, A2AListTasksRequest, A2AListTasksResponse, A2AMessage, A2AMutualTlsSecurityScheme, A2AOAuth2SecurityScheme, A2AOAuthFlows, A2AOpenIdConnectSecurityScheme, A2APart, A2APasswordOAuthFlow, A2ARole, A2ASecurityRequirement, A2ASecurityScheme, A2ASendMessageConfiguration, A2AStreamEvent, A2ATask, A2ATaskArtifactUpdateEvent, A2ATaskPushNotificationConfig, A2ATaskState, A2ATaskStatus, A2ATaskStatusUpdateEvent, A2A_PROTOCOL_VERSION, UseA2ARuntimeOptions, UseA2AThreadListAdapter };
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;cAgBa;KAED;KAEA;EACV;EACA;EACA;EACA;EAEA,WAAW;EACX;EACA;;KAGU;EACV;EACA;EACA;EACA,MAAM;EACN,OAAO;EACP,WAAW;EACX;EACA;;KAGU;KAWA;EACV,OAAO;EACP,UAAU;EACV;;KAGU;EACV;EACA;EACA;EACA,OAAO;EACP,WAAW;EACX;;KAGU;EACV;EACA;EACA,QAAQ;EACR,YAAY;EACZ,UAAU;EACV,WAAW;;KAID;EACV;EACA;EACA,QAAQ;EACR,WAAW;;KAGD;EACV;EACA;EACA,UAAU;EACV;EACA;EACA,WAAW;;KAGD;EACN;EAAc,MAAM;;EACpB;EAAiB,SAAS;;EAC1B;EAAsB,OAAO;;EAC7B;EAAwB,OAAO;;KAEzB;EACV;EACA;;KAGU;EACV;EACA;EACA;EACA;EACA;EACA,iBAAiB;;KAGP;EACV,SAAS;EACT;;KAGU;EACV;EACA,6BAA6B;EAC7B;EACA;;KAGU;EACV;EACA,SAAS;EACT;EACA;EACA;EACA;EACA;;KAGU;EACV,OAAO;EACP;EACA;EACA;;KAIU;EACV;EACA;EACA;EACA;;KAGU;EACV;EACA;EACA;;KAGU;EACV;EACA;EACA;;KAGU;EACV;EACA;EACA;EACA,QAAQ;EACR;;KAGU;EACV;EACA;EACA,QAAQ;;KAGE;EACV;EACA;EACA;EACA,QAAQ;;;KAIE;EACV;EACA;EACA,SAAS;;;KAIC;EACV;EACA;EACA,SAAS;;KAGC;EACV,oBAAoB;EACpB,oBAAoB;;EAEpB,WAAW;;EAEX,WAAW;EACX,aAAa;;KAGH;EACV;EACA,OAAO;EACP;;KAGU;EACV;EACA;;KAGU;EACV;;KAGU;EACV,uBAAuB;EACvB,yBAAyB;EACzB,uBAAuB;EACvB,8BAA8B;EAC9B,qBAAqB;;KAGX;EACV,SAAS;IAAiB;;;KAIhB;EACV;EACA;EACA,SAAS;;KAGC;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA,uBAAuB;;KAGb;EACV;EACA;EACA,aACI;IACE;IACA;IACA;IACA,SAAS;;EAGf;;KAGU;EACV;EACA;EACA;EACA;;KAGU;EACV;EACA;EACA,qBAAqB;EACrB;IAEM;IACA;;EAGN;EACA;EACA,cAAc;EACd,kBAAkB,eAAe;EACjC,uBAAuB;EACvB;EACA;EACA,QAAQ;EACR,aAAa;EACb;;;KAIU;EACV,MAAM;EACN,oBAAoB;EACpB,WAAW;;KAGD;EACV;EACA,4BAA4B;EAC5B,oBAAoB,qBAAqB;IACvC,mBAAmB;;;KAIX,uBAAuB;;EAEjC,SAAS;;EAET;;EAEA;;EAEA;;EAEA,UAAU;;EAEV;;EAEA,eAAe;;EAGf;;EAEA,gBAAgB;;EAGhB,WAAW,OAAO;;EAElB;;EAEA,sBAAsB,UAAU;EAEhC;IACE,cAAc;IACd,SAAS;IACT,YAAY;IACZ,QAAQ;IACR,WAAW;IACX,UAAU;IACV,aAAa"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;qBAgBa;YAED;YAEA;EACV;EACA;EACA;EACA;EAEA,WAAW;EACX;EACA;;YAGU;EACV;EACA;EACA;EACA,MAAM;EACN,OAAO;EACP,WAAW;EACX;EACA;;YAGU;YAWA;EACV,OAAO;EACP,UAAU;EACV;;YAGU;EACV;EACA;EACA;EACA,OAAO;EACP,WAAW;EACX;;YAGU;EACV;EACA;EACA,QAAQ;EACR,YAAY;EACZ,UAAU;EACV,WAAW;;YAID;EACV;EACA;EACA,QAAQ;EACR,WAAW;;YAGD;EACV;EACA;EACA,UAAU;EACV;EACA;EACA,WAAW;;YAGD;EACN;EAAc,MAAM;;EACpB;EAAiB,SAAS;;EAC1B;EAAsB,OAAO;;EAC7B;EAAwB,OAAO;;YAEzB;EACV;EACA;;YAGU;EACV;EACA;EACA;EACA;EACA;EACA,iBAAiB;;YAGP;EACV,SAAS;EACT;;YAGU;EACV;EACA,6BAA6B;EAC7B;EACA;;YAGU;EACV;EACA,SAAS;EACT;EACA;EACA;EACA;EACA;;YAGU;EACV,OAAO;EACP;EACA;EACA;;YAIU;EACV;EACA;EACA;EACA;;YAGU;EACV;EACA;EACA;;YAGU;EACV;EACA;EACA;;YAGU;EACV;EACA;EACA;EACA,QAAQ;EACR;;YAGU;EACV;EACA;EACA,QAAQ;;YAGE;EACV;EACA;EACA;EACA,QAAQ;;;YAIE;EACV;EACA;EACA,SAAS;;;YAIC;EACV;EACA;EACA,SAAS;;YAGC;EACV,oBAAoB;EACpB,oBAAoB;;EAEpB,WAAW;;EAEX,WAAW;EACX,aAAa;;YAGH;EACV;EACA,OAAO;EACP;;YAGU;EACV;EACA;;YAGU;EACV;;YAGU;EACV,uBAAuB;EACvB,yBAAyB;EACzB,uBAAuB;EACvB,8BAA8B;EAC9B,qBAAqB;;YAGX;EACV,SAAS;IAAiB;;;YAIhB;EACV;EACA;EACA,SAAS;;YAGC;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA,uBAAuB;;YAGb;EACV;EACA;EACA,aACI;IACE;IACA;IACA;IACA,SAAS;;EAGf;;YAGU;EACV;EACA;EACA;EACA;;YAGU;EACV;EACA;EACA,qBAAqB;EACrB;IAEM;IACA;;EAGN;EACA;EACA,cAAc;EACd,kBAAkB,eAAe;EACjC,uBAAuB;EACvB;EACA;EACA,QAAQ;EACR,aAAa;EACb;;;YAIU;EACV,MAAM;EACN,oBAAoB;EACpB,WAAW;;YAGD;EACV;EACA,4BAA4B;EAC5B,oBAAoB,qBAAqB;IACvC,mBAAmB;;;YAIX,uBAAuB;;EAEjC,SAAS;;EAET;;EAEA;;EAEA;;EAEA,UAAU;;EAEV;;EAEA,eAAe;;EAGf;;EAEA,gBAAgB;;EAGhB,WAAW,OAAO;;EAElB;;EAEA,sBAAsB,UAAU;EAEhC;IACE,cAAc;IACd,SAAS;IACT,YAAY;IACZ,QAAQ;IACR,WAAW;IACX,UAAU;IACV,aAAa"}

package/dist/useA2ARuntime.d.ts

@@ -1,7 +1,6 @@
 import { UseA2ARuntimeOptions } from "./types.js";
 import { AssistantRuntime } from "@assistant-ui/core";
 //#region src/useA2ARuntime.d.ts
-declare function useA2ARuntime(options: UseA2ARuntimeOptions): AssistantRuntime;
+export declare function useA2ARuntime(options: UseA2ARuntimeOptions): AssistantRuntime;
 //#endregion
-export { useA2ARuntime };
 //# sourceMappingURL=useA2ARuntime.d.ts.map

package/dist/useA2ARuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useA2ARuntime.d.ts","names":[],"sources":["../src/useA2ARuntime.ts"],"mappings":";;;iBA2CgB,cAAc,SAAS,uBAAuB"}
+{"version":3,"file":"useA2ARuntime.d.ts","names":[],"sources":["../src/useA2ARuntime.ts"],"mappings":";;;wBA2CgB,cAAc,SAAS,uBAAuB"}

package/package.json

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

package/src/A2AThreadRuntimeCore.test.ts

@@ -7,7 +7,11 @@ import type {
   A2AStreamEvent,
   A2ATask,
 } from "./types";
-import type { AppendMessage, ThreadMessage } from "@assistant-ui/core";
+import type {
+  AppendMessage,
+  ExportedMessageRepository,
+  ThreadMessage,
+} from "@assistant-ui/core";
 
 // --- Mock client factory ---
 
@@ -269,6 +273,118 @@ describe("A2AThreadRuntimeCore", () => {
   });
 
   describe("history loading", () => {
+    it("keeps initial history loading across same-thread message resyncs", async () => {
+      let resolve!: (repo: ExportedMessageRepository) => void;
+      const pending = new Promise<ExportedMessageRepository>((res) => {
+        resolve = res;
+      });
+      const core = createCore(
+        {},
+        { history: { load: () => pending, append: async () => {} } },
+      );
+      const loading = core.__internal_load();
+      core.applyExternalMessages([]);
+      const wasLoading = core.isLoading;
+      const restored = createHistoryMessage(
+        "restored",
+        "user",
+        "Saved history",
+      );
+      resolve({
+        headId: restored.id,
+        messages: [{ parentId: null, message: restored }],
+      });
+      await loading;
+      expect(wasLoading).toBe(true);
+      expect(core.getMessages()).toEqual([restored]);
+      expect(core.isLoading).toBe(false);
+    });
+
+    it.each(["resolve", "reject"])(
+      "ignores an initial history load that later %s after a thread switch",
+      async (outcome) => {
+        let resolve!: (repo: ExportedMessageRepository) => void;
+        let reject!: (error: Error) => void;
+        const pending = new Promise<ExportedMessageRepository>((res, rej) => {
+          resolve = res;
+          reject = rej;
+        });
+        const onError = vi.fn();
+        const history = {
+          load: vi.fn(() => pending),
+          append: vi.fn().mockResolvedValue(undefined),
+        };
+        const core = createCore({}, { history, onError });
+        const loading = core.__internal_load();
+        const replacement = createHistoryMessage(
+          "selected",
+          "user",
+          "Selected thread",
+        );
+        core.applyExternalMessages([replacement]);
+        core.resetContext();
+
+        if (outcome === "resolve") {
+          const previous = createHistoryMessage(
+            "previous",
+            "user",
+            "Previous thread",
+          );
+          resolve({
+            headId: previous.id,
+            messages: [{ parentId: null, message: previous }],
+          });
+        } else {
+          reject(new Error("Old history failed"));
+        }
+        await loading;
+
+        expect(core.getMessages()).toEqual([replacement]);
+        expect(core.isLoading).toBe(false);
+        expect(onError).not.toHaveBeenCalled();
+        await core.__internal_load();
+        expect(history.load).toHaveBeenCalledTimes(1);
+      },
+    );
+
+    it("ends initial loading immediately when switching threads", async () => {
+      let resolve!: (repo: ExportedMessageRepository) => void;
+      const pending = new Promise<ExportedMessageRepository>((res) => {
+        resolve = res;
+      });
+      const core = createCore(
+        {},
+        { history: { load: () => pending, append: async () => {} } },
+      );
+      const loading = core.__internal_load();
+      expect(core.isLoading).toBe(true);
+      core.applyExternalMessages([]);
+      core.resetContext();
+      const updatesAfterReplacement = notifyUpdate.mock.calls.length;
+      expect(core.isLoading).toBe(false);
+      resolve({ messages: [] });
+      await loading;
+      expect(notifyUpdate).toHaveBeenCalledTimes(updatesAfterReplacement);
+    });
+
+    it("does not import old history when agent-card discovery settles after a thread switch", async () => {
+      let resolveCard!: (card: A2AAgentCard) => void;
+      const card = new Promise<A2AAgentCard>((resolve) => {
+        resolveCard = resolve;
+      });
+      const { history } = createBranchedHistory();
+      const core = createCore({ getAgentCard: () => card }, { history });
+      const loading = core.__internal_load();
+      await Promise.resolve();
+      core.applyExternalMessages([]);
+      core.resetContext();
+      resolveCard({ name: "Agent" } as A2AAgentCard);
+      await loading;
+      expect(core.getMessages()).toEqual([]);
+      expect(core.getAgentCard()?.name).toBe("Agent");
+      expect(core.isLoading).toBe(false);
+    });
+
     it("preserves sibling branches and selects the persisted head", async () => {
       const { user, firstAssistant, secondAssistant, history } =
         createBranchedHistory();
@@ -1402,6 +1518,136 @@ describe("A2AThreadRuntimeCore", () => {
       expect(cancelTask).not.toHaveBeenCalled();
     });
 
+    it("ignores a cancellation response that resolves after a newer task snapshot", async () => {
+      let resolveCancel!: (task: A2ATask) => void;
+      const cancelTask = vi.fn().mockReturnValue(
+        new Promise<A2ATask>((resolve) => {
+          resolveCancel = resolve;
+        }),
+      );
+      const streamMessage = vi.fn().mockImplementation(async function* (
+        _msg: any,
+        _cfg: any,
+        _meta: any,
+        signal: AbortSignal,
+      ) {
+        if (streamMessage.mock.calls.length === 1) {
+          yield statusUpdateEvent("working");
+          await new Promise<void>((resolve) => {
+            if (signal.aborted) resolve();
+            else
+              signal.addEventListener("abort", () => resolve(), {
+                once: true,
+              });
+          });
+          return;
+        }
+        yield statusUpdateEvent("working", "Second answer");
+      });
+      const core = createCore({ cancelTask, streamMessage });
+
+      const firstRun = core.append(createUserAppendMessage("First"));
+      await vi.waitFor(() => expect(core.getTask()?.id).toBe("t1"));
+
+      const cancelPromise = core.cancel();
+      await firstRun;
+
+      await core.append(createUserAppendMessage("Second"));
+      expect(core.getTask()?.status.state).toBe("working");
+
+      resolveCancel({ id: "t1", status: { state: "canceled" } });
+      await cancelPromise;
+
+      expect(cancelTask).toHaveBeenCalledExactlyOnceWith("t1");
+      expect(core.getTask()).toMatchObject({
+        id: "t1",
+        status: { state: "working" },
+      });
+    });
+
+    it("ignores a cancellation response that resolves after a newer run starts", async () => {
+      let resolveCancel!: (task: A2ATask) => void;
+      const cancelTask = vi.fn().mockReturnValue(
+        new Promise<A2ATask>((resolve) => {
+          resolveCancel = resolve;
+        }),
+      );
+      // The follow-up run never emits, so it leaves the non-terminal task
+      // object from the first run in place: only the run itself is newer.
+      const streamMessage = vi.fn().mockImplementation(async function* (
+        _msg: any,
+        _cfg: any,
+        _meta: any,
+        signal: AbortSignal,
+      ) {
+        if (streamMessage.mock.calls.length === 1) {
+          yield statusUpdateEvent("working");
+        }
+        await new Promise<void>((resolve) => {
+          if (signal.aborted) resolve();
+          else
+            signal.addEventListener("abort", () => resolve(), { once: true });
+        });
+      });
+      const core = createCore({ cancelTask, streamMessage });
+
+      const firstRun = core.append(createUserAppendMessage("First"));
+      await vi.waitFor(() => expect(core.getTask()?.id).toBe("t1"));
+
+      const cancelPromise = core.cancel();
+      await firstRun;
+
+      void core.append(createUserAppendMessage("Second"));
+      await vi.waitFor(() => expect(streamMessage).toHaveBeenCalledTimes(2));
+      expect(core.isRunning()).toBe(true);
+
+      resolveCancel({ id: "t1", status: { state: "canceled" } });
+      await cancelPromise;
+
+      expect(core.getTask()?.status.state).toBe("working");
+    });
+
+    it("still cancels the server task when onCancel clears the thread", async () => {
+      let resolveCancel!: (task: A2ATask) => void;
+      const cancelTask = vi.fn().mockReturnValue(
+        new Promise<A2ATask>((resolve) => {
+          resolveCancel = resolve;
+        }),
+      );
+      const streamMessage = vi.fn().mockImplementation(async function* (
+        _msg: any,
+        _cfg: any,
+        _meta: any,
+        signal: AbortSignal,
+      ) {
+        yield statusUpdateEvent("working");
+        await new Promise<void>((resolve) => {
+          if (signal.aborted) resolve();
+          else
+            signal.addEventListener("abort", () => resolve(), { once: true });
+        });
+      });
+      let core!: A2AThreadRuntimeCore;
+      core = createCore(
+        { cancelTask, streamMessage },
+        { onCancel: () => core.applyExternalMessages([]) },
+      );
+
+      const runPromise = core.append(createUserAppendMessage("Go"));
+      await vi.waitFor(() => expect(core.getTask()?.id).toBe("t1"));
+
+      const cancelPromise = core.cancel();
+      await runPromise;
+
+      expect(cancelTask).toHaveBeenCalledExactlyOnceWith("t1");
+
+      resolveCancel({ id: "t1", status: { state: "canceled" } });
+      await cancelPromise;
+
+      expect(core.getTask()).toBeUndefined();
+      expect(core.getMessages()).toEqual([]);
+    });
+
     it.each(["throws", "rejects"] as const)(
       "isolates onCancel callbacks that %s",
       async (failureMode) => {

package/src/A2AThreadRuntimeCore.ts

@@ -83,6 +83,7 @@ export class A2AThreadRuntimeCore {
   private readonly session = createMessageRepositorySession();
   private isRunningFlag = false;
   private abortController: AbortController | null = null;
+  private runGeneration = 0;
   private pendingError: Error | null = null;
 
   // A2A-specific state
@@ -95,6 +96,7 @@ export class A2AThreadRuntimeCore {
   private readonly recordedHistoryIds = new Set<string>();
   private _isLoading = false;
   private _loadPromise: Promise<void> | undefined;
+  private _historyLoadGeneration = 0;
   private _loadRequested = false;
   private _agentCardPromise: Promise<void> | undefined;
   private _agentCardRetryAfter = 0;
@@ -146,6 +148,8 @@ export class A2AThreadRuntimeCore {
   /** Thread-boundary reset: applyExternalMessages alone also serves branch
    * switches, deletes, and cancel resyncs, which must keep the live context. */
   resetContext(): void {
+    this._historyLoadGeneration++;
+    this._isLoading = false;
     // Restore the seed before aborting: an onCancel callback that starts a
     // new run must not pick up the old thread's context, and its controller
     // must not be discarded.
@@ -257,16 +261,19 @@ export class A2AThreadRuntimeCore {
 
     this._isLoading = true;
 
+    const generation = this._historyLoadGeneration;
     const historyPromise = this.history.load();
 
     this._loadPromise = Promise.all([historyPromise, agentCardPromise])
       .then(([repo]) => {
+        if (generation !== this._historyLoadGeneration) return;
         if (repo) {
           this.session.applyExternalMessageRepository(repo);
           this.finalizeExternalApply();
         }
       })
       .catch((error) => {
+        if (generation !== this._historyLoadGeneration) return;
         invokeRuntimeCallback(
           "onError",
           this.onError,
@@ -274,6 +281,7 @@ export class A2AThreadRuntimeCore {
         );
       })
       .finally(() => {
+        if (generation !== this._historyLoadGeneration) return;
         this._isLoading = false;
         this.notifyUpdate();
       });
@@ -328,14 +336,26 @@ export class A2AThreadRuntimeCore {
   async cancel(): Promise<void> {
     if (!this.abortController) return;
 
+    // Read the server target before aborting: the abort listener runs the
+    // onCancel callback synchronously, which may clear the thread and with it
+    // the task this cancellation is for, or start a new run.
+    const task = this.currentTask;
+    const generation = this.runGeneration;
+
     // Abort locally first so the stream stops immediately
     this.abortController.abort();
 
     // Then try to cancel the task on the server
-    if (this.currentTask?.id) {
+    if (task?.id) {
       try {
-        const updated = await this.client.cancelTask(this.currentTask.id);
-        this.currentTask = updated;
+        const updated = await this.client.cancelTask(task.id);
+        // Only apply the response while nothing newer exists. A newer snapshot
+        // or a cleared thread replaces the task object; a follow-up run that
+        // has not emitted yet keeps it, so the run generation is what rules
+        // that case out.
+        if (this.currentTask === task && this.runGeneration === generation) {
+          this.currentTask = updated;
+        }
       } catch {
         // Server cancel failed; local abort already handled
       }
@@ -405,6 +425,8 @@ export class A2AThreadRuntimeCore {
   // --- Run logic ---
 
   private async startRun(userThreadMessage: ThreadMessage): Promise<void> {
+    this.runGeneration++;
+
     // Cancel any in-progress run before starting a new one
     if (this.abortController) {
       this.abortController.abort();

package/src/useA2ARuntime.test.tsx

@@ -8,7 +8,10 @@ import {
   type PropsWithChildren,
 } from "react";
 import { afterEach, describe, expect, it, vi } from "vitest";
-import type { ThreadMessage } from "@assistant-ui/core";
+import type {
+  ExportedMessageRepository,
+  ThreadMessage,
+} from "@assistant-ui/core";
 import type { A2AClient } from "./A2AClient";
 import type { A2AStreamEvent } from "./types";
 import { useA2ARuntime } from "./useA2ARuntime";
@@ -357,6 +360,93 @@ describe("useA2ARuntime", () => {
     ]);
   });
 
+  it("keeps pending history when a run is cancelled in the same thread", async () => {
+    const { client, streamMessage } = createMockClient(true);
+    let resolve!: (repo: ExportedMessageRepository) => void;
+    const pending = new Promise<ExportedMessageRepository>((res) => {
+      resolve = res;
+    });
+    const { result } = renderHook(() =>
+      useA2ARuntime({
+        client,
+        adapters: { history: { load: () => pending, append: async () => {} } },
+      }),
+    );
+    act(() => {
+      result.current.thread.append("Hello");
+    });
+    await waitFor(() => expect(streamMessage).toHaveBeenCalledOnce());
+    await act(async () => {
+      result.current.thread.cancelRun();
+      await new Promise((done) => setTimeout(done, 0));
+    });
+    const wasLoading = result.current.thread.getState().isLoading;
+    const restored = createThreadMessage("restored");
+    await act(async () => {
+      resolve({
+        headId: restored.id,
+        messages: [{ parentId: null, message: restored }],
+      });
+      await pending;
+    });
+    expect(wasLoading).toBe(true);
+    expect(result.current.thread.getState().messages).toEqual([restored]);
+    expect(result.current.thread.getState().isLoading).toBe(false);
+  });
+
+  it.each(["existing", "new"])(
+    "keeps the selected %s thread when initial history finishes later",
+    async (target) => {
+      const { client } = createMockClient();
+      let resolveHistory!: (repo: ExportedMessageRepository) => void;
+      const pending = new Promise<ExportedMessageRepository>((resolve) => {
+        resolveHistory = resolve;
+      });
+      const history = { load: () => pending, append: async () => {} };
+      const { result } = renderHook(() => {
+        const [threadId, setThreadId] = useState("thread-a");
+        return useA2ARuntime({
+          client,
+          adapters: {
+            history,
+            threadList: {
+              threadId,
+              onSwitchToThread: async (id) => {
+                setThreadId(id);
+                return { messages: [createThreadMessage("message-b")] };
+              },
+              onSwitchToNewThread: async () => {
+                setThreadId("thread-new");
+              },
+            },
+          },
+        });
+      });
+
+      await act(async () => {
+        if (target === "existing")
+          await result.current.threads.switchToThread("thread-b");
+        else await result.current.threads.switchToNewThread();
+      });
+      const selectedId = result.current.threads.getState().mainThreadId;
+      const selectedMessages = result.current.thread.getState().messages;
+      await act(async () => {
+        resolveHistory({
+          headId: "message-a",
+          messages: [
+            { parentId: null, message: createThreadMessage("message-a") },
+          ],
+        });
+        await pending;
+      });
+      expect(result.current.threads.getState().mainThreadId).toBe(selectedId);
+      expect(result.current.thread.getState().messages).toEqual(
+        selectedMessages,
+      );
+      expect(result.current.thread.getState().isLoading).toBe(false);
+    },
+  );
+
   it("ignores a thread load superseded by a new thread", async () => {
     const { client } = createMockClient();
     let resolveLoad!: (value: { messages: ThreadMessage[] }) => void;