@assistant-ui/react-a2a 0.2.36 → 0.2.37

package/dist/A2AThreadRuntimeCore.d.ts

@@ -1,91 +1,103 @@
-import { A2AAgentCard, A2AArtifact, A2ASendMessageConfiguration, A2ATask } from "./types.js";
-import { A2AClient } from "./A2AClient.js";
-import { AppendMessage, AssistantRuntime, ExportedMessageRepository, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
-//#region src/A2AThreadRuntimeCore.d.ts
+import type { AppendMessage, AssistantRuntime, ExportedMessageRepository, ThreadHistoryAdapter, ThreadMessage, Unstable_RecordToolInteractionOptions } from "@assistant-ui/core";
+import type { A2AClient } from "./A2AClient.js";
+import type { A2AArtifact, A2AAgentCard, A2ASendMessageConfiguration, A2ATask } from "./types.js";
 export type A2AThreadRuntimeCoreOptions = {
-  client: A2AClient;
-  contextId?: string | undefined;
-  configuration?: A2ASendMessageConfiguration | undefined;
-  onError?: ((error: Error) => void) | undefined;
-  onCancel?: (() => void) | undefined;
-  onArtifactComplete?: ((artifact: A2AArtifact) => void) | undefined;
-  history?: ThreadHistoryAdapter | undefined;
-  notifyUpdate: () => void;
+    client: A2AClient;
+    contextId?: string | undefined;
+    configuration?: A2ASendMessageConfiguration | undefined;
+    onError?: ((error: Error) => void) | undefined;
+    onCancel?: (() => void) | undefined;
+    onArtifactComplete?: ((artifact: A2AArtifact) => void) | undefined;
+    history?: ThreadHistoryAdapter | undefined;
+    notifyUpdate: () => void;
 };
 export declare class A2AThreadRuntimeCore {
-  private client;
-  private contextId;
-  private configuration;
-  private onError;
-  private onCancel;
-  private onArtifactComplete;
-  private history;
-  private readonly notifyUpdate;
-  private runtime;
-  private readonly session;
-  private isRunningFlag;
-  private abortController;
-  private runGeneration;
-  private pendingError;
-  private currentTask;
-  private currentArtifacts;
-  private agentCardValue;
-  private readonly assistantHistoryParents;
-  private readonly recordedHistoryIds;
-  private _isLoading;
-  private _loadPromise;
-  private _historyLoadGeneration;
-  private _loadRequested;
-  private _agentCardPromise;
-  private _agentCardRetryAfter;
-  private _agentCardRetryDelay;
-  private _agentCardDiscoveryFailed;
-  private lastOptionsContextId;
-  constructor(options: A2AThreadRuntimeCoreOptions);
-  updateOptions(options: Omit<A2AThreadRuntimeCoreOptions, "notifyUpdate">): void;
-  /** Thread-boundary reset: applyExternalMessages alone also serves branch
-   * switches, deletes, and cancel resyncs, which must keep the live context. */
-  resetContext(): void;
-  attachRuntime(runtime: AssistantRuntime): void;
-  detachRuntime(): void;
-  getRuntime(): AssistantRuntime | undefined;
-  getMessages(): readonly ThreadMessage[];
-  getMessageRepository(): ExportedMessageRepository;
-  getTask(): A2ATask | undefined;
-  getArtifacts(): readonly A2AArtifact[];
-  getAgentCard(): A2AAgentCard | undefined;
-  isRunning(): boolean;
-  get isLoading(): boolean;
-  private loadAgentCard;
-  private waitForAgentCard;
-  __internal_load(): Promise<void>;
-  append(message: AppendMessage): Promise<void>;
-  edit(message: AppendMessage): Promise<void>;
-  reload(parentId: string | null, _config?: {
-    runConfig?: Record<string, unknown>;
-  }): Promise<void>;
-  cancel(): Promise<void>;
-  private appendLinearChain;
-  private finalizeExternalApply;
-  applyExternalMessages(messages: readonly ThreadMessage[]): void;
-  private startRun;
-  private runStreaming;
-  private runSync;
-  private handleStreamEvent;
-  private handleStatusUpdate;
-  private handleArtifactUpdate;
-  private handleMessage;
-  private handleTaskSnapshot;
-  private insertAssistantPlaceholder;
-  private updateAssistantContent;
-  private updateAssistantStatus;
-  private getAssistantStatus;
-  private setRunning;
-  private finishRun;
-  private recordHistoryEntry;
-  private markPendingAssistantHistory;
-  private persistAssistantHistory;
-  private appendHistoryItem;
+    private client;
+    private contextId;
+    private configuration;
+    private onError;
+    private onCancel;
+    private onArtifactComplete;
+    private history;
+    private readonly notifyUpdate;
+    private runtime;
+    private readonly session;
+    private isRunningFlag;
+    private abortController;
+    private runGeneration;
+    private pendingError;
+    private currentTask;
+    private currentArtifacts;
+    private a2uiState;
+    private readonly a2uiMessageIds;
+    private agentCardValue;
+    private readonly assistantHistoryParents;
+    private readonly recordedHistoryIds;
+    private readonly historyWrites;
+    private _isLoading;
+    private _loadPromise;
+    private _historyLoadGeneration;
+    private _loadRequested;
+    private _agentCardPromise;
+    private _agentCardRetryAfter;
+    private _agentCardRetryDelay;
+    private _agentCardDiscoveryFailed;
+    private lastOptionsContextId;
+    constructor(options: A2AThreadRuntimeCoreOptions);
+    updateOptions(options: Omit<A2AThreadRuntimeCoreOptions, "notifyUpdate">): void;
+    /** Thread-boundary reset: applyExternalMessages alone also serves branch
+     * switches, deletes, and cancel resyncs, which must keep the live context. */
+    resetContext(): void;
+    attachRuntime(runtime: AssistantRuntime): void;
+    detachRuntime(): void;
+    getRuntime(): AssistantRuntime | undefined;
+    getMessages(): readonly ThreadMessage[];
+    getMessageRepository(): ExportedMessageRepository;
+    recordToolInteraction({ messageId, toolCallId, interaction, }: Unstable_RecordToolInteractionOptions): Promise<void>;
+    getTask(): A2ATask | undefined;
+    getArtifacts(): readonly A2AArtifact[];
+    getAgentCard(): A2AAgentCard | undefined;
+    isRunning(): boolean;
+    get isLoading(): boolean;
+    private loadAgentCard;
+    private waitForAgentCard;
+    __internal_load(): Promise<void>;
+    append(message: AppendMessage): Promise<void>;
+    appendVoiceTranscript(message: ThreadMessage): void;
+    edit(message: AppendMessage): Promise<void>;
+    reload(parentId: string | null, _config?: {
+        runConfig?: Record<string, unknown>;
+    }): Promise<void>;
+    cancel(): Promise<void>;
+    private appendLinearChain;
+    private finalizeExternalApply;
+    applyExternalMessages(messages: readonly ThreadMessage[]): void;
+    private startRun;
+    private runStreaming;
+    private runSync;
+    private handleStreamEvent;
+    private handleStatusUpdate;
+    private handleArtifactUpdate;
+    private handleMessage;
+    private handleTaskSnapshot;
+    private insertAssistantPlaceholder;
+    private updateAssistantContent;
+    private applyA2uiMessage;
+    private applyA2uiParts;
+    private a2uiSurfaceParts;
+    private withA2uiSurfaces;
+    private withPreservedToolInteractions;
+    private rebuildAssistantA2uiSurfaces;
+    private updateAssistantArtifacts;
+    private updateAssistantStatus;
+    private getAssistantStatus;
+    private isPersistableAssistantStatus;
+    private setRunning;
+    private finishRun;
+    private recordHistoryEntry;
+    private markPendingAssistantHistory;
+    private persistAssistantHistory;
+    private appendHistoryItem;
+    private chainHistoryWrite;
 }
-//#endregion
 //# 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":";;;;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"}
+{"version":3,"file":"A2AThreadRuntimeCore.d.ts","sourceRoot":"","sources":["../src/A2AThreadRuntimeCore.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,aAAa,EACb,gBAAgB,EAChB,yBAAyB,EAIzB,oBAAoB,EACpB,aAAa,EAEb,qCAAqC,EACtC,MAAM,oBAAoB,CAAC;AAa5B,OAAO,KAAK,EAAE,SAAS,EAAE,uBAAoB;AAC7C,OAAO,KAAK,EACV,WAAW,EACX,YAAY,EAEZ,2BAA2B,EAE3B,OAAO,EAGR,mBAAgB;AAqBjB,MAAM,MAAM,2BAA2B,GAAG;IACxC,MAAM,EAAE,SAAS,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC/B,aAAa,CAAC,EAAE,2BAA2B,GAAG,SAAS,CAAC;IACxD,OAAO,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC;IAC/C,QAAQ,CAAC,EAAE,CAAC,MAAM,IAAI,CAAC,GAAG,SAAS,CAAC;IACpC,kBAAkB,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,WAAW,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC;IACnE,OAAO,CAAC,EAAE,oBAAoB,GAAG,SAAS,CAAC;IAC3C,YAAY,EAAE,MAAM,IAAI,CAAC;CAC1B,CAAC;AAwBF,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,SAAS,CAAqB;IACtC,OAAO,CAAC,aAAa,CAA0C;IAC/D,OAAO,CAAC,OAAO,CAAuC;IACtD,OAAO,CAAC,QAAQ,CAA2B;IAC3C,OAAO,CAAC,kBAAkB,CAAgD;IAC1E,OAAO,CAAC,OAAO,CAAmC;IAClD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAa;IAE1C,OAAO,CAAC,OAAO,CAA+B;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAoC;IAC5D,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,eAAe,CAAgC;IACvD,OAAO,CAAC,aAAa,CAAK;IAC1B,OAAO,CAAC,YAAY,CAAsB;IAG1C,OAAO,CAAC,WAAW,CAAsB;IACzC,OAAO,CAAC,gBAAgB,CAAqB;IAC7C,OAAO,CAAC,SAAS,CAAwB;IACzC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAqB;IACpD,OAAO,CAAC,cAAc,CAA2B;IAGjD,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAoC;IAC5E,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAqB;IACxD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAoC;IAClE,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,YAAY,CAA4B;IAChD,OAAO,CAAC,sBAAsB,CAAK;IACnC,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,iBAAiB,CAA4B;IACrD,OAAO,CAAC,oBAAoB,CAAK;IACjC,OAAO,CAAC,oBAAoB,CAAqC;IACjE,OAAO,CAAC,yBAAyB,CAAS;IAE1C,OAAO,CAAC,oBAAoB,CAAqB;gBAErC,OAAO,EAAE,2BAA2B;IAYhD,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,2BAA2B,EAAE,cAAc,CAAC;IA4BxE;kFAC8E;IAC9E,YAAY,IAAI,IAAI;IAgBpB,aAAa,CAAC,OAAO,EAAE,gBAAgB;IAIvC,aAAa;IASb,UAAU,IAAI,gBAAgB,GAAG,SAAS;IAI1C,WAAW,IAAI,SAAS,aAAa,EAAE;IAIvC,oBAAoB,IAAI,yBAAyB;IAIpC,qBAAqB,CAAC,EACjC,SAAS,EACT,UAAU,EACV,WAAW,GACZ,EAAE,qCAAqC,GAAG,OAAO,CAAC,IAAI,CAAC;IA4CxD,OAAO,IAAI,OAAO,GAAG,SAAS;IAI9B,YAAY,IAAI,SAAS,WAAW,EAAE;IAItC,YAAY,IAAI,YAAY,GAAG,SAAS;IAIxC,SAAS,IAAI,OAAO;IAIpB,IAAI,SAAS,IAAI,OAAO,CAEvB;IAED,OAAO,CAAC,aAAa;YAwBP,gBAAgB;IAoB9B,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAsC1B,MAAM,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IAuBnD,qBAAqB,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI;IAQ7C,IAAI,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IAI3C,MAAM,CACV,QAAQ,EAAE,MAAM,GAAG,IAAI,EACvB,OAAO,GAAE;QAAE,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KAAO,GACpD,OAAO,CAAC,IAAI,CAAC;IAaV,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC;IA6B7B,OAAO,CAAC,iBAAiB;IAgBzB,OAAO,CAAC,qBAAqB;IAa7B,qBAAqB,CAAC,QAAQ,EAAE,SAAS,aAAa,EAAE,GAAG,IAAI;YAmCjD,QAAQ;YAmFR,YAAY;YA4CZ,OAAO;IA4BrB,OAAO,CAAC,iBAAiB;IAiBzB,OAAO,CAAC,kBAAkB;IA8B1B,OAAO,CAAC,oBAAoB;IAiD5B,OAAO,CAAC,aAAa;IASrB,OAAO,CAAC,kBAAkB;IAqE1B,OAAO,CAAC,0BAA0B;IAsBlC,OAAO,CAAC,sBAAsB;IAa9B,OAAO,CAAC,gBAAgB;IAQxB,OAAO,CAAC,cAAc;IAMtB,OAAO,CAAC,gBAAgB;IAkBxB,OAAO,CAAC,gBAAgB;IAexB,OAAO,CAAC,6BAA6B;IA+BrC,OAAO,CAAC,4BAA4B;IAWpC,OAAO,CAAC,wBAAwB;IAsBhC,OAAO,CAAC,qBAAqB;IAa7B,OAAO,CAAC,kBAAkB;IAM1B,OAAO,CAAC,4BAA4B;IAUpC,OAAO,CAAC,UAAU;IAKlB,OAAO,CAAC,SAAS;IAQjB,OAAO,CAAC,kBAAkB;IAM1B,OAAO,CAAC,2BAA2B;IAQnC,OAAO,CAAC,uBAAuB;IA2D/B,OAAO,CAAC,iBAAiB;IAgBzB,OAAO,CAAC,iBAAiB;CA8B1B"}

package/dist/A2AThreadRuntimeCore.js

@@ -1,10 +1,16 @@
 "use client";
-import { a2aMessageToContent, isTerminalTaskState, taskStateToMessageStatus, threadMessageToA2AMessage } from "./conversions.js";
-import { createMessageRepositorySession, invokeUserCallback } from "@assistant-ui/core/internal";
+import { a2aMessageToContent, a2uiPartsToOperations, isTerminalTaskState, taskStateToMessageStatus, threadMessageToA2AMessage } from "./conversions.js";
+import { appendToolInteraction, createMessageRepositorySession, invokeUserCallback } from "@assistant-ui/core/internal";
 import { fromThreadMessageLike, generateId } from "@assistant-ui/core";
+import { applyA2uiOperations, convertSurfaceToUISpec, surfaceToOperations } from "@assistant-ui/react-generative-ui/a2ui";
 //#region src/A2AThreadRuntimeCore.ts
 const INITIAL_AGENT_CARD_RETRY_DELAY_MS = 5e3;
 const MAX_AGENT_CARD_RETRY_DELAY_MS = 3e5;
+const isRecord$1 = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+const storedArtifact = (artifact) => ({
+	...artifact,
+	parts: artifact.parts.map(({ raw: _raw, ...part }) => part)
+});
 const FALLBACK_USER_STATUS = {
 	type: "complete",
 	reason: "unknown"
@@ -35,9 +41,12 @@ var A2AThreadRuntimeCore = class {
 	pendingError = null;
 	currentTask;
 	currentArtifacts = [];
+	a2uiState = /* @__PURE__ */ new Map();
+	a2uiMessageIds = /* @__PURE__ */ new Set();
 	agentCardValue;
 	assistantHistoryParents = /* @__PURE__ */ new Map();
 	recordedHistoryIds = /* @__PURE__ */ new Set();
+	historyWrites = /* @__PURE__ */ new Map();
 	_isLoading = false;
 	_loadPromise;
 	_historyLoadGeneration = 0;
@@ -103,6 +112,26 @@ var A2AThreadRuntimeCore = class {
 	getMessageRepository() {
 		return this.session.export();
 	}
+	async recordToolInteraction({ messageId, toolCallId, interaction }) {
+		const message = this.session.tryGetMessage(messageId)?.message;
+		if (!message) throw new Error("Tried to record a tool interaction on a non-existing message");
+		if (message.role !== "assistant") throw new Error("Tried to record a tool interaction on a non-assistant message");
+		const target = message.content.find((part) => part.type === "tool-call" && part.toolCallId === toolCallId);
+		if (!target || target.type !== "tool-call") throw new Error("Tried to record a tool interaction on a non-existing tool call");
+		if (this.session.updateMessage(messageId, (current) => {
+			if (current.role !== "assistant") return current;
+			return {
+				...current,
+				content: current.content.map((part) => part.type === "tool-call" && part.toolCallId === toolCallId ? {
+					...part,
+					unstable_interactions: appendToolInteraction(part.unstable_interactions, interaction)
+				} : part)
+			};
+		})) {
+			this.notifyUpdate();
+			this.persistAssistantHistory(messageId);
+		}
+	}
 	getTask() {
 		return this.currentTask;
 	}
@@ -159,7 +188,7 @@ var A2AThreadRuntimeCore = class {
 		this._isLoading = true;
 		const generation = this._historyLoadGeneration;
 		const historyPromise = this.history.load();
-		this._loadPromise = Promise.all([historyPromise, agentCardPromise]).then(([repo]) => {
+		this._loadPromise = historyPromise.then((repo) => {
 			if (generation !== this._historyLoadGeneration) return;
 			if (repo) {
 				this.session.applyExternalMessageRepository(repo);
@@ -187,6 +216,13 @@ var A2AThreadRuntimeCore = class {
 		if (!startRun) return;
 		await this.startRun(threadMessage);
 	}
+	appendVoiceTranscript(message) {
+		const parentId = this.session.headId;
+		this.session.addOrUpdateMessage(parentId, message);
+		this.session.switchToBranch(message.id);
+		this.notifyUpdate();
+		this.recordHistoryEntry(parentId, message);
+	}
 	async edit(message) {
 		await this.append(message);
 	}
@@ -226,6 +262,8 @@ var A2AThreadRuntimeCore = class {
 		for (const { message } of this.getMessageRepository().messages) this.recordedHistoryIds.add(message.id);
 		this.currentTask = void 0;
 		this.currentArtifacts = [];
+		this.a2uiState = /* @__PURE__ */ new Map();
+		this.a2uiMessageIds.clear();
 		this.notifyUpdate();
 	}
 	applyExternalMessages(messages) {
@@ -267,6 +305,8 @@ var A2AThreadRuntimeCore = class {
 		});
 		if (this.currentTask && isTerminalTaskState(this.currentTask.status.state)) this.currentTask = void 0;
 		this.currentArtifacts = [];
+		this.a2uiState = /* @__PURE__ */ new Map();
+		this.a2uiMessageIds.clear();
 		const assistantParentId = userThreadMessage.id;
 		const assistantId = this.insertAssistantPlaceholder(assistantParentId);
 		this.markPendingAssistantHistory(assistantId, assistantParentId);
@@ -343,7 +383,7 @@ var A2AThreadRuntimeCore = class {
 				this.handleStatusUpdate(assistantId, event.event);
 				break;
 			case "artifactUpdate":
-				this.handleArtifactUpdate(event.event);
+				this.handleArtifactUpdate(assistantId, event.event);
 				break;
 			case "message":
 				this.handleMessage(assistantId, event.message);
@@ -363,6 +403,7 @@ var A2AThreadRuntimeCore = class {
 		};
 		if (event.contextId) this.contextId = event.contextId;
 		if (event.status.message) {
+			this.applyA2uiMessage(event.status.message);
 			const content = a2aMessageToContent(event.status.message);
 			this.updateAssistantContent(assistantId, content);
 		}
@@ -370,7 +411,7 @@ var A2AThreadRuntimeCore = class {
 		this.updateAssistantStatus(assistantId, status);
 		this.notifyUpdate();
 	}
-	handleArtifactUpdate(event) {
+	handleArtifactUpdate(assistantId, event) {
 		const { append, lastChunk } = event;
 		const artifact = normalizeArtifact(event.artifact);
 		const existingIdx = this.currentArtifacts.findIndex((a) => a.artifactId === artifact.artifactId);
@@ -397,11 +438,15 @@ var A2AThreadRuntimeCore = class {
 			updated = artifact;
 			this.currentArtifacts = [...this.currentArtifacts, updated];
 		}
+		this.applyA2uiParts(artifact.parts);
+		this.updateAssistantArtifacts(assistantId);
+		this.rebuildAssistantA2uiSurfaces(assistantId);
 		if (lastChunk) invokeRuntimeCallback("onArtifactComplete", this.onArtifactComplete, updated);
 		this.notifyUpdate();
 	}
 	handleMessage(assistantId, message) {
 		if (message.role !== "agent") return;
+		this.applyA2uiMessage(message);
 		const content = a2aMessageToContent(message);
 		this.updateAssistantContent(assistantId, content);
 		this.notifyUpdate();
@@ -415,11 +460,21 @@ var A2AThreadRuntimeCore = class {
 			...history === void 0 ? {} : { history }
 		};
 		if (task.contextId) this.contextId = task.contextId;
-		if (artifacts) this.currentArtifacts = artifacts;
+		const isCompleteSnapshot = history !== void 0 && artifacts !== void 0;
+		const artifactsToApply = isCompleteSnapshot ? artifacts : artifacts?.filter((artifact) => !this.currentArtifacts.some(({ artifactId }) => artifactId === artifact.artifactId));
+		if (isCompleteSnapshot) {
+			this.currentArtifacts = artifacts;
+			this.a2uiState = /* @__PURE__ */ new Map();
+			this.a2uiMessageIds.clear();
+		} else if (artifactsToApply) this.currentArtifacts = [...this.currentArtifacts, ...artifactsToApply];
+		for (const message of history ?? []) if (message.role === "agent") this.applyA2uiMessage(message, true);
+		for (const artifact of artifactsToApply ?? []) this.applyA2uiParts(artifact.parts);
+		if (task.status.message) this.applyA2uiMessage(task.status.message, true);
+		if (isCompleteSnapshot || artifactsToApply?.length) this.updateAssistantArtifacts(assistantId);
 		if (task.status.message) {
 			const content = a2aMessageToContent(task.status.message);
 			this.updateAssistantContent(assistantId, content);
-		}
+		} else this.rebuildAssistantA2uiSurfaces(assistantId);
 		const status = taskStateToMessageStatus(task.status.state);
 		this.updateAssistantStatus(assistantId, status);
 		this.notifyUpdate();
@@ -450,10 +505,86 @@ var A2AThreadRuntimeCore = class {
 			if (message.role !== "assistant") return message;
 			return {
 				...message,
-				content
+				content: this.withA2uiSurfaces(content, message.content)
 			};
 		});
 	}
+	applyA2uiMessage(message, replay = false) {
+		if (message.messageId) {
+			if (replay && this.a2uiMessageIds.has(message.messageId)) return;
+			this.a2uiMessageIds.add(message.messageId);
+		}
+		this.applyA2uiParts(message.parts);
+	}
+	applyA2uiParts(parts) {
+		const operations = a2uiPartsToOperations(parts);
+		if (operations.length === 0) return;
+		this.a2uiState = applyA2uiOperations(this.a2uiState, operations).state;
+	}
+	a2uiSurfaceParts() {
+		const parts = [];
+		for (const [surfaceId, surface] of this.a2uiState) {
+			const { spec } = convertSurfaceToUISpec(surface);
+			if (!spec) continue;
+			parts.push({
+				type: "tool-call",
+				toolCallId: `a2ui:${surfaceId}`,
+				toolName: "present",
+				args: spec,
+				argsText: JSON.stringify(spec),
+				result: {},
+				artifact: { a2ui: surfaceToOperations(surface) }
+			});
+		}
+		return parts;
+	}
+	withA2uiSurfaces(content, previousContent = content) {
+		const preserved = content.filter((part) => !(part.type === "tool-call" && part.toolCallId.startsWith("a2ui:")));
+		const interactionSource = [...previousContent, ...content];
+		return this.withPreservedToolInteractions(interactionSource, [...preserved, ...this.a2uiSurfaceParts()]);
+	}
+	withPreservedToolInteractions(previousContent, content) {
+		const interactions = /* @__PURE__ */ new Map();
+		for (const part of previousContent) if (part.type === "tool-call" && part.unstable_interactions !== void 0) interactions.set(part.toolCallId, part.unstable_interactions);
+		if (interactions.size === 0) return content;
+		return content.map((part) => {
+			if (part.type !== "tool-call" || part.unstable_interactions !== void 0) return part;
+			const unstable_interactions = interactions.get(part.toolCallId);
+			return unstable_interactions === void 0 ? part : {
+				...part,
+				unstable_interactions
+			};
+		});
+	}
+	rebuildAssistantA2uiSurfaces(messageId) {
+		if (this.session.updateMessage(messageId, (message) => {
+			if (message.role !== "assistant") return message;
+			return {
+				...message,
+				content: this.withA2uiSurfaces(message.content)
+			};
+		})) this.notifyUpdate();
+	}
+	updateAssistantArtifacts(messageId) {
+		if (this.session.updateMessage(messageId, (message) => {
+			if (message.role !== "assistant") return message;
+			const custom = message.metadata.custom;
+			const a2a = isRecord$1(custom.a2a) ? custom.a2a : {};
+			return {
+				...message,
+				metadata: {
+					...message.metadata,
+					custom: {
+						...custom,
+						a2a: {
+							...a2a,
+							artifacts: this.currentArtifacts.map(storedArtifact)
+						}
+					}
+				}
+			};
+		})) this.notifyUpdate();
+	}
 	updateAssistantStatus(messageId, status) {
 		if (this.session.updateMessage(messageId, (message) => {
 			if (message.role !== "assistant") return message;
@@ -463,7 +594,7 @@ var A2AThreadRuntimeCore = class {
 			};
 		})) {
 			this.notifyUpdate();
-			if (status.type === "complete" || status.type === "incomplete") this.persistAssistantHistory(messageId);
+			if (this.isPersistableAssistantStatus(status)) this.persistAssistantHistory(messageId);
 		}
 	}
 	getAssistantStatus(messageId) {
@@ -471,6 +602,9 @@ var A2AThreadRuntimeCore = class {
 		if (msg?.role !== "assistant") return void 0;
 		return msg.status;
 	}
+	isPersistableAssistantStatus(status) {
+		return status.type === "complete" || status.type === "incomplete" || status.type === "requires-action" && status.reason === "interrupt";
+	}
 	setRunning(running) {
 		this.isRunningFlag = running;
 		this.notifyUpdate();
@@ -481,7 +615,9 @@ var A2AThreadRuntimeCore = class {
 		this.setRunning(false);
 	}
 	recordHistoryEntry(parentId, message) {
-		this.appendHistoryItem(parentId, message);
+		this.appendHistoryItem(parentId, message)?.catch((error) => {
+			console.error("[react-a2a] failed to append history entry", error);
+		});
 	}
 	markPendingAssistantHistory(messageId, parentId) {
 		if (!this.history) return;
@@ -489,23 +625,74 @@ var A2AThreadRuntimeCore = class {
 	}
 	persistAssistantHistory(messageId) {
 		if (!this.history) return;
+		const messageData = this.session.tryGetMessage(messageId);
 		const parentId = this.assistantHistoryParents.get(messageId);
-		if (parentId === void 0) return;
-		const message = this.session.tryGetMessage(messageId)?.message;
+		if (parentId === void 0 && !this.recordedHistoryIds.has(messageId)) return;
+		const resolvedParentId = parentId === void 0 ? messageData?.parentId : parentId;
+		if (resolvedParentId === void 0) return;
+		const message = messageData?.message;
 		if (!message || message.role !== "assistant") return;
-		if (message.status?.type !== "complete" && message.status?.type !== "incomplete") return;
-		this.assistantHistoryParents.delete(messageId);
-		this.appendHistoryItem(parentId, message);
+		if (!this.isPersistableAssistantStatus(message.status)) return;
+		const isPausing = message.status.type === "requires-action";
+		if (isPausing && !this.history.update) return;
+		if (this.recordedHistoryIds.has(messageId)) {
+			if (this.history.update) {
+				const update = this.history.update.bind(this.history);
+				const write = this.chainHistoryWrite(messageId, () => update({
+					parentId: resolvedParentId,
+					message
+				}));
+				if (!isPausing) this.assistantHistoryParents.delete(messageId);
+				write.then(() => {
+					this.recordedHistoryIds.add(messageId);
+				}, (error) => {
+					const pending = this.historyWrites.get(messageId);
+					if (pending === void 0 || pending === write) this.assistantHistoryParents.set(messageId, resolvedParentId);
+					console.error("[react-a2a] failed to update history entry", error);
+				});
+				return;
+			}
+			if (!isPausing) this.assistantHistoryParents.delete(messageId);
+			return;
+		}
+		const write = this.appendHistoryItem(resolvedParentId, message);
+		if (!write) return;
+		if (!isPausing) this.assistantHistoryParents.delete(messageId);
+		write.catch((error) => {
+			const pending = this.historyWrites.get(messageId);
+			if (pending === void 0 || pending === write) this.assistantHistoryParents.set(messageId, resolvedParentId);
+			console.error("[react-a2a] failed to append history entry", error);
+		});
 	}
 	appendHistoryItem(parentId, message) {
 		if (!this.history || this.recordedHistoryIds.has(message.id)) return;
 		this.recordedHistoryIds.add(message.id);
-		this.history.append({
+		const append = this.history.append.bind(this.history);
+		const write = this.chainHistoryWrite(message.id, () => append({
 			parentId,
 			message
-		}).catch(() => {
+		}));
+		write.catch(() => {
 			this.recordedHistoryIds.delete(message.id);
 		});
+		return write;
+	}
+	chainHistoryWrite(id, write) {
+		const pending = this.historyWrites.get(id);
+		let next;
+		if (pending) next = pending.then(write, write);
+		else try {
+			next = Promise.resolve(write());
+		} catch (error) {
+			next = Promise.reject(error);
+		}
+		this.historyWrites.set(id, next);
+		next.then(() => {
+			if (this.historyWrites.get(id) === next) this.historyWrites.delete(id);
+		}, () => {
+			if (this.historyWrites.get(id) === next) this.historyWrites.delete(id);
+		});
+		return next;
 	}
 };
 //#endregion