@assistant-ui/react-ag-ui 0.0.34 → 0.0.35

package/dist/runtime/AgUiThreadRuntimeCore.d.ts

@@ -1,16 +1,17 @@
 import { Logger } from "./logger.js";
 import { AgUiInterrupt, AgUiResumeEntry } from "./types.js";
-import { AddToolResultOptions, AppendMessage, AssistantRuntime, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
+import { AddToolResultOptions, AppendMessage, AssistantRuntime, ChatModelRunOptions, ChatModelRunResult, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
 import { HttpAgent } from "@ag-ui/client";
 import { ReadonlyJSONValue } from "assistant-stream/utils";
 
 //#region src/runtime/AgUiThreadRuntimeCore.d.ts
 type RunConfig = NonNullable<AppendMessage["runConfig"]>;
+type ResumeStream = (options: ChatModelRunOptions) => AsyncGenerator<ChatModelRunResult, void, unknown>;
 type ResumeRunConfig = {
   parentId: string | null;
   sourceId: string | null;
   runConfig: RunConfig;
-  stream?: unknown;
+  stream?: ResumeStream;
 };
 type CoreOptions = {
   agent: HttpAgent;
@@ -40,6 +41,7 @@ declare class AgUiThreadRuntimeCore {
   private readonly recordedHistoryIds;
   private _isLoading;
   private _loadPromise;
+  private pendingResumeMessageId;
   constructor(options: CoreOptions);
   updateOptions(options: Omit<CoreOptions, "notifyUpdate">): void;
   attachRuntime(runtime: AssistantRuntime): void;
@@ -65,9 +67,12 @@ declare class AgUiThreadRuntimeCore {
   private clearPendingInterrupts;
   findMessageIdForToolCall(toolCallId: string): string | undefined;
   addToolResult(options: AddToolResultOptions): void;
+  private maybeResumeAfterToolResults;
+  private startResumeRun;
   applyExternalMessages(messages: readonly ThreadMessage[]): void;
   loadExternalState(state: ReadonlyJSONValue): void;
   private startRun;
+  private consumeResumeStream;
   private buildRunInput;
   private installResumeShim;
   private setRunning;
@@ -75,6 +80,7 @@ declare class AgUiThreadRuntimeCore {
   private insertAssistantPlaceholder;
   private reassignAssistantId;
   private updateAssistantMessage;
+  private preserveToolResults;
   private mergeAssistantMetadata;
   private handleEvent;
   private importMessagesSnapshot;

package/dist/runtime/AgUiThreadRuntimeCore.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AgUiThreadRuntimeCore.d.ts","names":[],"sources":["../../src/runtime/AgUiThreadRuntimeCore.ts"],"mappings":";;;;;;;KAmCK,SAAA,GAAY,WAAW,CAAC,aAAA;AAAA,KACxB,eAAA;EACH,QAAA;EACA,QAAA;EACA,SAAA,EAAW,SAAS;EACpB,MAAA;AAAA;AAAA,KAGG,WAAA;EACH,KAAA,EAAO,SAAA;EACP,MAAA,EAAQ,MAAA;EACR,YAAA;EACA,OAAA,IAAW,KAAA,EAAO,KAAA;EAClB,QAAA;EACA,OAAA,GAAU,oBAAA;EACV,YAAA;AAAA;AAAA,cAKW,qBAAA;EAAA,QACH,KAAA;EAAA,QACA,MAAA;EAAA,QACA,YAAA;EAAA,QACA,OAAA;EAAA,QACA,QAAA;EAAA,iBACS,YAAA;EAAA,QAET,OAAA;EAAA,QACA,QAAA;EAAA,QACA,aAAA;EAAA,QACA,eAAA;EAAA,QACA,aAAA;EAAA,QACA,YAAA;EAAA,QACA,OAAA;EAAA,QACA,aAAA;EAAA,iBACS,uBAAA;EAAA,iBACA,kBAAA;EAAA,QACT,UAAA;EAAA,QACA,YAAA;cAEI,OAAA,EAAS,WAAA;EAWrB,aAAA,CAAc,OAAA,EAAS,IAAA,CAAK,WAAA;EAU5B,aAAA,CAAc,OAAA,EAAS,gBAAA;EAIvB,aAAA;EAIA,WAAA,aAAwB,aAAA;EAIxB,QAAA,IAAY,iBAAA;EAIZ,SAAA;EAAA,IAII,SAAA;EAIJ,eAAA,IAAmB,OAAA;EAkCb,MAAA,CAAO,OAAA,EAAS,aAAA,GAAgB,OAAA;EAmBhC,IAAA,CAAK,OAAA,EAAS,aAAA,GAAgB,OAAA;EAI9B,MAAA,CACJ,QAAA,iBACA,MAAA;IAAU,SAAA,GAAY,SAAA;EAAA,IACrB,OAAA;EAOG,MAAA,IAAU,OAAA;EAKV,MAAA,CAAO,MAAA,EAAQ,eAAA,GAAkB,OAAA;EAAA,QAa/B,yBAAA;EAOR,oBAAA;IACE,SAAA;IACA,UAAA,WAAqB,aAAA;EAAA;EAqBjB,wBAAA,CACJ,SAAA,WAAoB,eAAA,KACnB,OAAA;EAAA,QAyEK,sBAAA;EAgCR,wBAAA,CAAyB,UAAA;EAiBzB,aAAA,CAAc,OAAA,EAAS,oBAAA;EAgEvB,qBAAA,CAAsB,QAAA,WAAmB,aAAA;EAUzC,iBAAA,CAAkB,KAAA,EAAO,iBAAA;EAAA,QAKX,QAAA;EAAA,QAsGN,aAAA;EAAA,QA2BA,iBAAA;EAAA,QAqBA,UAAA;EAAA,QAKA,SAAA;EAAA,QAOA,0BAAA;EAAA,QAqBA,mBAAA;EAAA,QAiCA,sBAAA;EAAA,QAyCA,sBAAA;EAAA,QA4BA,WAAA;EAAA,QAoBA,sBAAA;EAAA,QA0BA,eAAA;EAAA,QAQA,SAAA;EAAA,QAYA,gBAAA;EAAA,QAIA,mBAAA;EAAA,QAKA,kBAAA;EAAA,QAIA,2BAAA;EAAA,QAQA,uBAAA;EAAA,QAWA,iBAAA;AAAA"}
+{"version":3,"file":"AgUiThreadRuntimeCore.d.ts","names":[],"sources":["../../src/runtime/AgUiThreadRuntimeCore.ts"],"mappings":";;;;;;;KAsCK,SAAA,GAAY,WAAW,CAAC,aAAA;AAAA,KACxB,YAAA,IACH,OAAA,EAAS,mBAAA,KACN,cAAA,CAAe,kBAAA;AAAA,KACf,eAAA;EACH,QAAA;EACA,QAAA;EACA,SAAA,EAAW,SAAA;EACX,MAAA,GAAS,YAAY;AAAA;AAAA,KAGlB,WAAA;EACH,KAAA,EAAO,SAAA;EACP,MAAA,EAAQ,MAAA;EACR,YAAA;EACA,OAAA,IAAW,KAAA,EAAO,KAAA;EAClB,QAAA;EACA,OAAA,GAAU,oBAAA;EACV,YAAA;AAAA;AAAA,cAKW,qBAAA;EAAA,QACH,KAAA;EAAA,QACA,MAAA;EAAA,QACA,YAAA;EAAA,QACA,OAAA;EAAA,QACA,QAAA;EAAA,iBACS,YAAA;EAAA,QAET,OAAA;EAAA,QACA,QAAA;EAAA,QACA,aAAA;EAAA,QACA,eAAA;EAAA,QACA,aAAA;EAAA,QACA,YAAA;EAAA,QACA,OAAA;EAAA,QACA,aAAA;EAAA,iBACS,uBAAA;EAAA,iBACA,kBAAA;EAAA,QACT,UAAA;EAAA,QACA,YAAA;EAAA,QACA,sBAAA;cAEI,OAAA,EAAS,WAAA;EAWrB,aAAA,CAAc,OAAA,EAAS,IAAA,CAAK,WAAA;EAU5B,aAAA,CAAc,OAAA,EAAS,gBAAA;EAIvB,aAAA;EAIA,WAAA,aAAwB,aAAA;EAIxB,QAAA,IAAY,iBAAA;EAIZ,SAAA;EAAA,IAII,SAAA;EAIJ,eAAA,IAAmB,OAAA;EAwCb,MAAA,CAAO,OAAA,EAAS,aAAA,GAAgB,OAAA;EAmBhC,IAAA,CAAK,OAAA,EAAS,aAAA,GAAgB,OAAA;EAI9B,MAAA,CACJ,QAAA,iBACA,MAAA;IAAU,SAAA,GAAY,SAAA;EAAA,IACrB,OAAA;EAOG,MAAA,IAAU,OAAA;EAKV,MAAA,CAAO,MAAA,EAAQ,eAAA,GAAkB,OAAA;EAAA,QAU/B,yBAAA;EAOR,oBAAA;IACE,SAAA;IACA,UAAA,WAAqB,aAAA;EAAA;EAqBjB,wBAAA,CACJ,SAAA,WAAoB,eAAA,KACnB,OAAA;EAAA,QAyEK,sBAAA;EAgCR,wBAAA,CAAyB,UAAA;EAiBzB,aAAA,CAAc,OAAA,EAAS,oBAAA;EAAA,QAiCf,2BAAA;EAAA,QAqCA,cAAA;EAMR,qBAAA,CAAsB,QAAA,WAAmB,aAAA;EAUzC,iBAAA,CAAkB,KAAA,EAAO,iBAAA;EAAA,QAKX,QAAA;EAAA,QAqIA,mBAAA;EAAA,QAyDN,aAAA;EAAA,QA2BA,iBAAA;EAAA,QAqBA,UAAA;EAAA,QAKA,SAAA;EAAA,QAOA,0BAAA;EAAA,QAqBA,mBAAA;EAAA,QAiCA,sBAAA;EAAA,QAoDA,mBAAA;EAAA,QAiCA,sBAAA;EAAA,QA4BA,WAAA;EAAA,QAiCA,sBAAA;EAAA,QA0BA,eAAA;EAAA,QAQA,SAAA;EAAA,QAYA,gBAAA;EAAA,QAIA,mBAAA;EAAA,QAKA,kBAAA;EAAA,QAIA,2BAAA;EAAA,QAQA,uBAAA;EAAA,QAWA,iBAAA;AAAA"}

package/dist/runtime/AgUiThreadRuntimeCore.js

@@ -3,6 +3,7 @@ import { AG_UI_METADATA_NAMESPACE, RunAggregator } from "./adapter/run-aggregato
 import { fromAgUiMessages, toAgUiMessages, toAgUiTools } from "./adapter/conversions.js";
 import { createAgUiSubscriber } from "./adapter/subscriber.js";
 import { fromThreadMessageLike, generateId } from "@assistant-ui/core/internal";
+import jsonpatch from "fast-json-patch";
 //#region src/runtime/AgUiThreadRuntimeCore.ts
 const optimisticPrefix = "__optimistic__";
 const generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;
@@ -31,6 +32,7 @@ var AgUiThreadRuntimeCore = class {
 	recordedHistoryIds = /* @__PURE__ */ new Set();
 	_isLoading = false;
 	_loadPromise;
+	pendingResumeMessageId = null;
 	constructor(options) {
 		this.agent = options.agent;
 		this.logger = options.logger;
@@ -78,7 +80,8 @@ var AgUiThreadRuntimeCore = class {
 			this.applyExternalMessages(messages);
 			if (repo.unstable_resume) {
 				const parentId = repo.headId ?? messages.at(-1)?.id ?? null;
-				await this.startRun(parentId, this.lastRunConfig);
+				const resumeStream = this.history?.resume?.bind(this.history);
+				await this.startRun(parentId, this.lastRunConfig, void 0, resumeStream);
 			}
 		}).catch((error) => {
 			this.logger.error?.("[agui] failed to load history", error);
@@ -117,8 +120,7 @@ var AgUiThreadRuntimeCore = class {
 	}
 	async resume(config) {
 		this.assertNoPendingInterrupts();
-		if (config.stream) this.logger.debug?.("[agui] resume stream is not supported, falling back to regular run");
-		await this.startRun(config.parentId, config.runConfig ?? this.lastRunConfig);
+		await this.startRun(config.parentId, config.runConfig ?? this.lastRunConfig, void 0, config.stream);
 	}
 	assertNoPendingInterrupts() {
 		if (!this.getPendingInterrupts()) return;
@@ -202,7 +204,6 @@ var AgUiThreadRuntimeCore = class {
 	}
 	addToolResult(options) {
 		let updated = false;
-		let shouldResume = false;
 		this.messages = this.messages.map((message) => {
 			if (message.id !== options.messageId || message.role !== "assistant") return message;
 			const assistant = message;
@@ -219,31 +220,40 @@ var AgUiThreadRuntimeCore = class {
 			});
 			if (!matchedToolCall) return message;
 			updated = true;
-			if (assistant.status?.type === "requires-action" && assistant.status.reason === "tool-calls" && content.every((part) => part.type !== "tool-call" || "result" in part && part.result !== void 0)) {
-				shouldResume = true;
-				return {
-					...assistant,
-					content,
-					status: {
-						type: "complete",
-						reason: "unknown"
-					}
-				};
-			}
 			return {
 				...assistant,
 				content
 			};
 		});
-		if (updated) {
-			this.notifyUpdate();
-			if (shouldResume) {
-				this.persistAssistantHistory(options.messageId);
-				if (!this.isRunningFlag) this.startRun(options.messageId, this.lastRunConfig).catch((error) => {
-					this.onError?.(error instanceof Error ? error : new Error(String(error)));
-				});
+		if (!updated) return;
+		this.notifyUpdate();
+		this.maybeResumeAfterToolResults(options.messageId);
+	}
+	maybeResumeAfterToolResults(messageId) {
+		const message = this.messages.find((m) => m.id === messageId);
+		if (!message || message.role !== "assistant") return;
+		const assistant = message;
+		if (assistant.status?.type !== "requires-action" || assistant.status.reason !== "tool-calls") return;
+		if (!assistant.content.every((part) => part.type !== "tool-call" || "result" in part && part.result !== void 0)) return;
+		this.messages = this.messages.map((m) => m.id === messageId && m.role === "assistant" ? {
+			...m,
+			status: {
+				type: "complete",
+				reason: "unknown"
 			}
+		} : m);
+		this.notifyUpdate();
+		this.persistAssistantHistory(messageId);
+		if (this.isRunningFlag) {
+			this.pendingResumeMessageId = messageId;
+			return;
 		}
+		this.startResumeRun(messageId);
+	}
+	startResumeRun(messageId) {
+		this.startRun(messageId, this.lastRunConfig).catch((error) => {
+			this.onError?.(error instanceof Error ? error : new Error(String(error)));
+		});
 	}
 	applyExternalMessages(messages) {
 		this.assistantHistoryParents.clear();
@@ -256,14 +266,12 @@ var AgUiThreadRuntimeCore = class {
 		this.stateSnapshot = state;
 		this.notifyUpdate();
 	}
-	async startRun(parentId, runConfig, resume) {
+	async startRun(parentId, runConfig, resume, resumeStream) {
 		const normalizedRunConfig = runConfig ?? {};
 		this.lastRunConfig = normalizedRunConfig;
 		this.resetHead(parentId);
 		const historicalMessages = [...this.messages];
-		const runId = generateId();
 		this.pendingError = null;
-		const input = this.buildRunInput(runId, normalizedRunConfig, historicalMessages, resume);
 		const assistantParentId = parentId ?? this.messages.at(-1)?.id ?? null;
 		let assistantMessageId;
 		const ensureAssistant = () => {
@@ -273,13 +281,14 @@ var AgUiThreadRuntimeCore = class {
 			this.markPendingAssistantHistory(created, assistantParentId ?? null);
 			return created;
 		};
+		const applyUpdate = (update) => {
+			const resolved = this.updateAssistantMessage(ensureAssistant(), update);
+			if (resolved !== assistantMessageId) assistantMessageId = resolved;
+		};
 		const aggregator = new RunAggregator({
 			showThinking: this.showThinking,
 			logger: this.logger,
-			emit: (update) => {
-				const resolved = this.updateAssistantMessage(ensureAssistant(), update);
-				if (resolved !== assistantMessageId) assistantMessageId = resolved;
-			},
+			emit: applyUpdate,
 			onServerMessageId: (serverId) => {
 				const placeholder = ensureAssistant();
 				if (placeholder === serverId) return;
@@ -291,32 +300,52 @@ var AgUiThreadRuntimeCore = class {
 		const abortController = new AbortController();
 		const abortSignal = abortController.signal;
 		this.abortController = abortController;
+		let cancelRun = () => dispatch({ type: "RUN_CANCELLED" });
 		abortSignal.addEventListener("abort", () => {
-			dispatch({ type: "RUN_CANCELLED" });
+			cancelRun();
 			this.finishRun(abortController);
 			this.onCancel?.();
 		}, { once: true });
-		const subscriber = createAgUiSubscriber({
-			dispatch,
-			runId,
-			logger: this.logger,
-			onRunFailed: (error) => {
-				this.pendingError = error;
-				this.onError?.(error);
-			}
-		});
-		aggregator.handle({
-			type: "RUN_STARTED",
-			runId
-		});
 		this.setRunning(true);
 		try {
-			try {
-				this.agent.messages = input.messages;
-				this.agent.threadId = input.threadId;
-				this.agent.state = input.state ?? null;
-			} catch {}
-			await this.agent.runAgent(input, subscriber, { signal: abortSignal });
+			if (resumeStream) {
+				cancelRun = () => applyUpdate({ status: {
+					type: "incomplete",
+					reason: "cancelled"
+				} });
+				await this.consumeResumeStream(resumeStream, {
+					runConfig: normalizedRunConfig,
+					threadId: this.agent.threadId || "main",
+					parentId: assistantParentId,
+					historicalMessages,
+					abortSignal,
+					ensureAssistant,
+					applyUpdate,
+					getAssistantMessageId: () => assistantMessageId
+				});
+			} else {
+				const runId = generateId();
+				aggregator.handle({
+					type: "RUN_STARTED",
+					runId
+				});
+				const input = this.buildRunInput(runId, normalizedRunConfig, historicalMessages, resume);
+				const subscriber = createAgUiSubscriber({
+					dispatch,
+					runId,
+					logger: this.logger,
+					onRunFailed: (error) => {
+						this.pendingError = error;
+						this.onError?.(error);
+					}
+				});
+				try {
+					this.agent.messages = input.messages;
+					this.agent.threadId = input.threadId;
+					this.agent.state = input.state ?? null;
+				} catch {}
+				await this.agent.runAgent(input, subscriber, { signal: abortSignal });
+			}
 		} catch (error) {
 			if (!abortSignal.aborted) {
 				const err = error instanceof Error ? error : new Error(String(error));
@@ -333,8 +362,55 @@ var AgUiThreadRuntimeCore = class {
 		if (this.pendingError) {
 			const err = this.pendingError;
 			this.pendingError = null;
+			this.pendingResumeMessageId = null;
 			throw err;
 		}
+		if (this.pendingResumeMessageId !== null) {
+			const resumeMessageId = this.pendingResumeMessageId;
+			this.pendingResumeMessageId = null;
+			if (!abortSignal.aborted) this.startResumeRun(resumeMessageId);
+		}
+	}
+	async consumeResumeStream(stream, ctx) {
+		const assistantId = ctx.ensureAssistant();
+		const currentId = () => ctx.getAssistantMessageId() ?? assistantId;
+		const options = {
+			messages: ctx.historicalMessages,
+			runConfig: ctx.runConfig,
+			abortSignal: ctx.abortSignal,
+			context: this.runtime?.thread.getModelContext() ?? {},
+			unstable_assistantMessageId: assistantId,
+			unstable_threadId: ctx.threadId,
+			unstable_parentId: ctx.parentId,
+			unstable_getMessage: () => {
+				const message = this.messages.find((m) => m.id === currentId());
+				if (!message) throw new Error("[agui] resume stream requested the assistant message before it existed");
+				return message;
+			}
+		};
+		try {
+			for await (const result of stream(options)) {
+				if (ctx.abortSignal.aborted) return;
+				ctx.applyUpdate(result);
+			}
+		} catch (error) {
+			if (ctx.abortSignal.aborted) return;
+			const err = error instanceof Error ? error : new Error(String(error));
+			ctx.applyUpdate({ status: {
+				type: "incomplete",
+				reason: "error",
+				error: err.message
+			} });
+			this.onError?.(err);
+			this.pendingError = this.pendingError ?? err;
+			return;
+		}
+		if (ctx.abortSignal.aborted) return;
+		const current = this.messages.find((m) => m.id === currentId());
+		if (!current || current.status?.type === "running") ctx.applyUpdate({ status: {
+			type: "complete",
+			reason: "unknown"
+		} });
 	}
 	buildRunInput(runId, runConfig, historyMessages, resume) {
 		const threadId = this.agent.threadId || "main";
@@ -433,9 +509,10 @@ var AgUiThreadRuntimeCore = class {
 			const assistant = message;
 			const metadata = update.metadata ? this.mergeAssistantMetadata(assistant.metadata, update.metadata) : assistant.metadata;
 			latestStatus = update.status ?? assistant.status;
+			const content = update.content !== void 0 ? this.preserveToolResults(assistant.content, update.content) : assistant.content;
 			return {
 				...assistant,
-				content: update.content ?? assistant.content,
+				content,
 				status: latestStatus,
 				metadata
 			};
@@ -448,8 +525,29 @@ var AgUiThreadRuntimeCore = class {
 			resolvedMessageId = stableId;
 		} else this.notifyUpdate();
 		if (this.isPersistableStatus(latestStatus)) this.persistAssistantHistory(resolvedMessageId);
+		this.maybeResumeAfterToolResults(resolvedMessageId);
 		return resolvedMessageId;
 	}
+	preserveToolResults(previous, next) {
+		const resolved = /* @__PURE__ */ new Map();
+		for (const part of previous) if (part.type === "tool-call" && "result" in part && part.result !== void 0) resolved.set(part.toolCallId, part);
+		if (resolved.size === 0) return next;
+		let changed = false;
+		const merged = next.map((part) => {
+			if (part.type !== "tool-call") return part;
+			if ("result" in part && part.result !== void 0) return part;
+			const prior = resolved.get(part.toolCallId);
+			if (!prior) return part;
+			changed = true;
+			return {
+				...part,
+				result: prior.result,
+				...prior.artifact !== void 0 ? { artifact: prior.artifact } : {},
+				...prior.isError !== void 0 ? { isError: prior.isError } : {}
+			};
+		});
+		return changed ? merged : next;
+	}
 	mergeAssistantMetadata(current, incoming) {
 		const annotations = incoming.unstable_annotations ? [...current.unstable_annotations, ...incoming.unstable_annotations] : current.unstable_annotations;
 		const data = incoming.unstable_data ? [...current.unstable_data, ...incoming.unstable_data] : current.unstable_data;
@@ -473,7 +571,15 @@ var AgUiThreadRuntimeCore = class {
 				this.notifyUpdate();
 				return;
 			case "STATE_DELTA":
-				this.logger.debug?.("[agui] state delta event ignored", event.delta);
+				if (event.delta.length === 0) return;
+				try {
+					const state = this.stateSnapshot ?? {};
+					const result = jsonpatch.applyPatch(state, event.delta, true, false);
+					this.stateSnapshot = result.newDocument;
+					this.notifyUpdate();
+				} catch (error) {
+					this.logger.error?.("[agui] failed to apply state delta", error);
+				}
 				return;
 			case "MESSAGES_SNAPSHOT":
 				this.importMessagesSnapshot(event.messages);

package/dist/runtime/AgUiThreadRuntimeCore.js.map

@@ -1 +1 @@
-{"version":3,"file":"AgUiThreadRuntimeCore.js","names":[],"sources":["../../src/runtime/AgUiThreadRuntimeCore.ts"],"sourcesContent":["\"use client\";\n\nimport { generateId, fromThreadMessageLike } from \"@assistant-ui/core/internal\";\nimport type {\n  AddToolResultOptions,\n  AppendMessage,\n  AssistantRuntime,\n  ChatModelRunResult,\n  MessageStatus,\n  ThreadAssistantMessage,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport type { HttpAgent } from \"@ag-ui/client\";\nimport type { Logger } from \"./logger\";\nimport type { AgUiEvent, AgUiInterrupt, AgUiResumeEntry } from \"./types\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport {\n  AG_UI_METADATA_NAMESPACE,\n  type AgUiCustomMetadata,\n  RunAggregator,\n} from \"./adapter/run-aggregator\";\nimport {\n  fromAgUiMessages,\n  toAgUiMessages,\n  toAgUiTools,\n} from \"./adapter/conversions\";\nimport { createAgUiSubscriber } from \"./adapter/subscriber\";\n\nconst optimisticPrefix = \"__optimistic__\";\nconst generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;\nconst isOptimisticId = (id: string) => id.startsWith(optimisticPrefix);\n\nconst symbolResumeShim = Symbol(\"agui-resume-shim\");\n\ntype RunConfig = NonNullable<AppendMessage[\"runConfig\"]>;\ntype ResumeRunConfig = {\n  parentId: string | null;\n  sourceId: string | null;\n  runConfig: RunConfig;\n  stream?: unknown;\n};\n\ntype CoreOptions = {\n  agent: HttpAgent;\n  logger: Logger;\n  showThinking: boolean;\n  onError?: (error: Error) => void;\n  onCancel?: () => void;\n  history?: ThreadHistoryAdapter;\n  notifyUpdate: () => void;\n};\n\nconst FALLBACK_USER_STATUS = { type: \"complete\", reason: \"unknown\" } as const;\n\nexport class AgUiThreadRuntimeCore {\n  private agent: HttpAgent;\n  private logger: Logger;\n  private showThinking: boolean;\n  private onError: ((error: Error) => void) | undefined;\n  private onCancel: (() => void) | undefined;\n  private readonly notifyUpdate: () => void;\n\n  private runtime: AssistantRuntime | undefined;\n  private messages: ThreadMessage[] = [];\n  private isRunningFlag = false;\n  private abortController: AbortController | null = null;\n  private stateSnapshot: ReadonlyJSONValue | undefined;\n  private pendingError: Error | null = null;\n  private history: ThreadHistoryAdapter | undefined;\n  private lastRunConfig: RunConfig | undefined;\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\n  constructor(options: CoreOptions) {\n    this.agent = options.agent;\n    this.logger = options.logger;\n    this.showThinking = options.showThinking;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.history = options.history;\n    this.notifyUpdate = options.notifyUpdate;\n    this.installResumeShim();\n  }\n\n  updateOptions(options: Omit<CoreOptions, \"notifyUpdate\">) {\n    this.agent = options.agent;\n    this.logger = options.logger;\n    this.showThinking = options.showThinking;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.history = options.history;\n    this.installResumeShim();\n  }\n\n  attachRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  detachRuntime() {\n    this.runtime = undefined;\n  }\n\n  getMessages(): readonly ThreadMessage[] {\n    return this.messages;\n  }\n\n  getState(): ReadonlyJSONValue | undefined {\n    return this.stateSnapshot;\n  }\n\n  isRunning(): boolean {\n    return this.isRunningFlag;\n  }\n\n  get isLoading(): boolean {\n    return this._isLoading;\n  }\n\n  __internal_load(): Promise<void> {\n    if (this._loadPromise) return this._loadPromise;\n\n    const promise = this.history?.load() ?? Promise.resolve(null);\n\n    this._isLoading = true;\n\n    this._loadPromise = promise\n      .then(async (repo) => {\n        if (!repo) return;\n\n        const messages = repo.messages.map((item) => item.message);\n        this.applyExternalMessages(messages);\n\n        if (repo.unstable_resume) {\n          const parentId = repo.headId ?? messages.at(-1)?.id ?? null;\n          await this.startRun(parentId, this.lastRunConfig);\n        }\n      })\n      .catch((error) => {\n        this.logger.error?.(\"[agui] failed to load history\", error);\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    if (startRun) this.assertNoPendingInterrupts();\n    if (message.sourceId) {\n      this.messages = this.messages.filter(\n        (entry) => entry.id !== message.sourceId,\n      );\n    }\n    this.resetHead(message.parentId);\n\n    const threadMessage = this.toThreadMessage(message);\n    this.messages = [...this.messages, threadMessage];\n    this.notifyUpdate();\n    this.recordHistoryEntry(message.parentId ?? null, threadMessage);\n\n    if (!startRun) return;\n    await this.startRun(threadMessage.id, message.runConfig);\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?: RunConfig } = {},\n  ): Promise<void> {\n    this.assertNoPendingInterrupts();\n    this.resetHead(parentId);\n    this.notifyUpdate();\n    await this.startRun(parentId, config.runConfig);\n  }\n\n  async cancel(): Promise<void> {\n    if (!this.abortController) return;\n    this.abortController.abort();\n  }\n\n  async resume(config: ResumeRunConfig): Promise<void> {\n    this.assertNoPendingInterrupts();\n    if (config.stream) {\n      this.logger.debug?.(\n        \"[agui] resume stream is not supported, falling back to regular run\",\n      );\n    }\n    await this.startRun(\n      config.parentId,\n      config.runConfig ?? this.lastRunConfig,\n    );\n  }\n\n  private assertNoPendingInterrupts(): void {\n    if (!this.getPendingInterrupts()) return;\n    throw new Error(\n      \"[agui] cannot start a new run while interrupts are pending; resolve them with submitInterruptResponses()\",\n    );\n  }\n\n  getPendingInterrupts(): {\n    messageId: string;\n    interrupts: readonly AgUiInterrupt[];\n  } | null {\n    const assistant = this.messages.findLast((m) => m.role === \"assistant\") as\n      | ThreadAssistantMessage\n      | undefined;\n    if (\n      !assistant ||\n      assistant.status?.type !== \"requires-action\" ||\n      assistant.status.reason !== \"interrupt\"\n    ) {\n      return null;\n    }\n    const stored = (\n      assistant.metadata.custom[AG_UI_METADATA_NAMESPACE] as\n        | AgUiCustomMetadata\n        | undefined\n    )?.interrupts;\n    if (!stored?.length) return null;\n    return { messageId: assistant.id, interrupts: stored };\n  }\n\n  async submitInterruptResponses(\n    responses: readonly AgUiResumeEntry[],\n  ): Promise<void> {\n    const pending = this.getPendingInterrupts();\n    if (!pending) {\n      throw new Error(\n        \"[agui] submitInterruptResponses: no pending interrupts on this thread\",\n      );\n    }\n\n    const responsesById = new Map<string, AgUiResumeEntry>();\n    for (const entry of responses) {\n      if (!entry || typeof entry.interruptId !== \"string\") {\n        throw new Error(\n          \"[agui] submitInterruptResponses: every entry must have an interruptId\",\n        );\n      }\n      if (entry.status !== \"resolved\" && entry.status !== \"cancelled\") {\n        throw new Error(\n          `[agui] submitInterruptResponses: invalid status \"${entry.status}\" for interrupt ${entry.interruptId}`,\n        );\n      }\n      if (responsesById.has(entry.interruptId)) {\n        throw new Error(\n          `[agui] submitInterruptResponses: duplicate response for interrupt ${entry.interruptId}`,\n        );\n      }\n      responsesById.set(entry.interruptId, entry);\n    }\n\n    const openIds = pending.interrupts.map((i) => i.id);\n    const missing = openIds.filter((id) => !responsesById.has(id));\n    if (missing.length > 0) {\n      throw new Error(\n        `[agui] submitInterruptResponses: missing responses for open interrupts: ${missing.join(\", \")}`,\n      );\n    }\n    const known = new Set(openIds);\n    const unknownIds = [...responsesById.keys()].filter((id) => !known.has(id));\n    if (unknownIds.length > 0) {\n      throw new Error(\n        `[agui] submitInterruptResponses: unknown interrupt ids: ${unknownIds.join(\", \")}`,\n      );\n    }\n\n    const now = Date.now();\n    for (const interrupt of pending.interrupts) {\n      if (!interrupt.expiresAt) continue;\n      const expiry = new Date(interrupt.expiresAt).getTime();\n      if (Number.isNaN(expiry)) {\n        throw new Error(\n          `[agui] submitInterruptResponses: interrupt ${interrupt.id} has malformed expiresAt \"${interrupt.expiresAt}\"`,\n        );\n      }\n      if (expiry <= now) {\n        throw new Error(\n          `[agui] submitInterruptResponses: interrupt ${interrupt.id} expired at ${interrupt.expiresAt}`,\n        );\n      }\n    }\n\n    const resume: AgUiResumeEntry[] = openIds.map(\n      (id) => responsesById.get(id)!,\n    );\n\n    if (this.isRunningFlag) {\n      throw new Error(\n        \"[agui] submitInterruptResponses: a run is already in progress\",\n      );\n    }\n\n    this.clearPendingInterrupts(pending.messageId);\n    await this.startRun(pending.messageId, this.lastRunConfig, resume);\n  }\n\n  private clearPendingInterrupts(messageId: string): void {\n    let touched = false;\n    this.messages = this.messages.map((message) => {\n      if (message.id !== messageId || message.role !== \"assistant\")\n        return message;\n      const assistant = message as ThreadAssistantMessage;\n      if (\n        assistant.status?.type !== \"requires-action\" ||\n        assistant.status.reason !== \"interrupt\"\n      ) {\n        return assistant;\n      }\n      touched = true;\n      const aguiMeta = assistant.metadata.custom[AG_UI_METADATA_NAMESPACE] as\n        | AgUiCustomMetadata\n        | undefined;\n      const { interrupts: _drop, ...restAgui } = aguiMeta ?? {};\n      const newCustom = { ...assistant.metadata.custom };\n      if (Object.keys(restAgui).length > 0) {\n        newCustom[AG_UI_METADATA_NAMESPACE] = restAgui;\n      } else {\n        delete newCustom[AG_UI_METADATA_NAMESPACE];\n      }\n      return {\n        ...assistant,\n        status: { type: \"complete\" as const, reason: \"unknown\" as const },\n        metadata: { ...assistant.metadata, custom: newCustom },\n      };\n    });\n    if (touched) this.notifyUpdate();\n  }\n\n  findMessageIdForToolCall(toolCallId: string): string | undefined {\n    let fallbackMessageId: string | undefined;\n    for (let index = this.messages.length - 1; index >= 0; index--) {\n      const message = this.messages[index];\n      if (!message || message.role !== \"assistant\") continue;\n      for (const part of message.content) {\n        if (part.type !== \"tool-call\" || part.toolCallId !== toolCallId)\n          continue;\n        if (!(\"result\" in part) || part.result === undefined) {\n          return message.id;\n        }\n        fallbackMessageId ??= message.id;\n      }\n    }\n    return fallbackMessageId;\n  }\n\n  addToolResult(options: AddToolResultOptions): void {\n    let updated = false;\n    let shouldResume = false;\n    this.messages = this.messages.map((message) => {\n      if (message.id !== options.messageId || message.role !== \"assistant\")\n        return message;\n      const assistant = message as ThreadAssistantMessage;\n      let matchedToolCall = false;\n      const content = assistant.content.map((part) => {\n        if (part.type !== \"tool-call\" || part.toolCallId !== options.toolCallId)\n          return part;\n        matchedToolCall = true;\n        return {\n          ...part,\n          result: options.result,\n          artifact: options.artifact,\n          isError: options.isError,\n        };\n      });\n      if (!matchedToolCall) return message;\n      updated = true;\n\n      if (\n        assistant.status?.type === \"requires-action\" &&\n        assistant.status.reason === \"tool-calls\" &&\n        content.every(\n          (part) =>\n            part.type !== \"tool-call\" ||\n            (\"result\" in part && part.result !== undefined),\n        )\n      ) {\n        shouldResume = true;\n        return {\n          ...assistant,\n          content,\n          status: { type: \"complete\" as const, reason: \"unknown\" as const },\n        };\n      }\n\n      return {\n        ...assistant,\n        content,\n      };\n    });\n\n    if (updated) {\n      this.notifyUpdate();\n\n      if (shouldResume) {\n        this.persistAssistantHistory(options.messageId);\n\n        if (!this.isRunningFlag) {\n          void this.startRun(options.messageId, this.lastRunConfig).catch(\n            (error) => {\n              this.onError?.(\n                error instanceof Error ? error : new Error(String(error)),\n              );\n            },\n          );\n        }\n      }\n    }\n  }\n\n  applyExternalMessages(messages: readonly ThreadMessage[]): void {\n    this.assistantHistoryParents.clear();\n    this.messages = [...messages];\n    this.recordedHistoryIds.clear();\n    for (const message of this.messages) {\n      this.recordedHistoryIds.add(message.id);\n    }\n    this.notifyUpdate();\n  }\n\n  loadExternalState(state: ReadonlyJSONValue): void {\n    this.stateSnapshot = state;\n    this.notifyUpdate();\n  }\n\n  private async startRun(\n    parentId: string | null,\n    runConfig?: RunConfig,\n    resume?: AgUiResumeEntry[],\n  ): Promise<void> {\n    const normalizedRunConfig = runConfig ?? {};\n    this.lastRunConfig = normalizedRunConfig;\n    this.resetHead(parentId);\n    const historicalMessages = [...this.messages];\n\n    const runId = generateId();\n    this.pendingError = null;\n    const input = this.buildRunInput(\n      runId,\n      normalizedRunConfig,\n      historicalMessages,\n      resume,\n    );\n    const assistantParentId = parentId ?? this.messages.at(-1)?.id ?? null;\n    let assistantMessageId: string | undefined;\n    const ensureAssistant = () => {\n      if (assistantMessageId) return assistantMessageId;\n      const created = this.insertAssistantPlaceholder();\n      assistantMessageId = created;\n      this.markPendingAssistantHistory(created, assistantParentId ?? null);\n      return created;\n    };\n\n    const aggregator = new RunAggregator({\n      showThinking: this.showThinking,\n      logger: this.logger,\n      emit: (update) => {\n        const resolved = this.updateAssistantMessage(ensureAssistant(), update);\n        if (resolved !== assistantMessageId) {\n          assistantMessageId = resolved;\n        }\n      },\n      onServerMessageId: (serverId) => {\n        const placeholder = ensureAssistant();\n        if (placeholder === serverId) return;\n        this.reassignAssistantId(placeholder, serverId);\n        assistantMessageId = serverId;\n      },\n    });\n    const dispatch = (event: AgUiEvent) => this.handleEvent(aggregator, event);\n\n    const abortController = new AbortController();\n    const abortSignal = abortController.signal;\n    this.abortController = abortController;\n\n    abortSignal.addEventListener(\n      \"abort\",\n      () => {\n        dispatch({ type: \"RUN_CANCELLED\" });\n        this.finishRun(abortController);\n        this.onCancel?.();\n      },\n      { once: true },\n    );\n\n    const subscriber = createAgUiSubscriber({\n      dispatch,\n      runId,\n      logger: this.logger,\n      onRunFailed: (error) => {\n        this.pendingError = error;\n        this.onError?.(error);\n      },\n    });\n\n    aggregator.handle({ type: \"RUN_STARTED\", runId });\n    this.setRunning(true);\n\n    try {\n      try {\n        (this.agent as any).messages = input.messages;\n        (this.agent as any).threadId = input.threadId;\n        (this.agent as any).state = input.state ?? null;\n      } catch {\n        // ignore\n      }\n      await (this.agent as any).runAgent(input, subscriber, {\n        signal: abortSignal,\n      });\n    } catch (error) {\n      if (!abortSignal.aborted) {\n        const err = error instanceof Error ? error : new Error(String(error));\n        dispatch({ type: \"RUN_ERROR\", message: err.message });\n        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 buildRunInput(\n    runId: string,\n    runConfig: RunConfig | undefined,\n    historyMessages: readonly ThreadMessage[] | undefined,\n    resume?: AgUiResumeEntry[],\n  ) {\n    const threadId = this.agent.threadId || \"main\";\n    const messages = toAgUiMessages(historyMessages ?? this.messages);\n    const context = this.runtime?.thread.getModelContext();\n    return {\n      threadId,\n      runId,\n      state: this.stateSnapshot ?? null,\n      messages,\n      tools: toAgUiTools(context?.tools),\n      context: context?.system\n        ? [{ description: \"system\", value: context.system }]\n        : [],\n      forwardedProps: {\n        ...(context?.callSettings ?? {}),\n        ...(context?.config ?? {}),\n        ...(runConfig?.custom ? { runConfig: runConfig.custom } : {}),\n      },\n      ...(resume !== undefined ? { resume } : {}),\n    };\n  }\n\n  private installResumeShim(): void {\n    const agent = this.agent as any;\n    if (agent[symbolResumeShim]) return;\n    agent[symbolResumeShim] = true;\n    const onInstance = Object.hasOwn(agent, \"prepareRunAgentInput\");\n    const original = onInstance\n      ? agent.prepareRunAgentInput\n      : Object.getPrototypeOf(agent)?.prepareRunAgentInput;\n    if (typeof original !== \"function\") return;\n    agent.prepareRunAgentInput = function (\n      this: unknown,\n      params: { resume?: unknown } | undefined,\n    ) {\n      const input = original.call(this, params);\n      if (params?.resume !== undefined && input && typeof input === \"object\") {\n        return { ...(input as object), resume: params.resume };\n      }\n      return input;\n    };\n  }\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) {\n      this.abortController = null;\n    }\n    this.setRunning(false);\n  }\n\n  private insertAssistantPlaceholder(): string {\n    const id = generateOptimisticId();\n    const assistant: ThreadAssistantMessage = {\n      id,\n      role: \"assistant\",\n      createdAt: new Date(),\n      status: { type: \"running\" },\n      content: [],\n      metadata: {\n        unstable_state: this.stateSnapshot ?? null,\n        unstable_annotations: [],\n        unstable_data: [],\n        steps: [],\n        custom: {},\n      },\n    };\n    this.messages = [...this.messages, assistant];\n    this.notifyUpdate();\n    return id;\n  }\n\n  private reassignAssistantId(oldId: string, newId: string): void {\n    if (oldId === newId) return;\n\n    const collidesWithExisting = this.messages.some((m) => m.id === newId);\n\n    if (collidesWithExisting) {\n      this.logger.debug?.(\n        \"[agui] reassignAssistantId: server id already present in messages, dropping placeholder\",\n        { oldId, newId },\n      );\n      this.messages = this.messages.filter((m) => m.id !== oldId);\n    } else {\n      this.messages = this.messages.map((m) =>\n        m.id === oldId ? { ...m, id: newId } : m,\n      );\n    }\n\n    const pendingParent = this.assistantHistoryParents.get(oldId);\n    if (pendingParent !== undefined) {\n      this.assistantHistoryParents.delete(oldId);\n      if (!this.assistantHistoryParents.has(newId)) {\n        this.assistantHistoryParents.set(newId, pendingParent);\n      }\n    }\n\n    if (this.recordedHistoryIds.has(oldId)) {\n      this.recordedHistoryIds.delete(oldId);\n      this.recordedHistoryIds.add(newId);\n    }\n\n    this.notifyUpdate();\n  }\n\n  private updateAssistantMessage(\n    messageId: string,\n    update: ChatModelRunResult,\n  ): string {\n    let touched = false;\n    let latestStatus: MessageStatus | undefined;\n    this.messages = this.messages.map((message) => {\n      if (message.id !== messageId || message.role !== \"assistant\")\n        return message;\n      touched = true;\n      const assistant = message as ThreadAssistantMessage;\n      const metadata = update.metadata\n        ? this.mergeAssistantMetadata(assistant.metadata, update.metadata)\n        : assistant.metadata;\n      latestStatus = update.status ?? assistant.status;\n      return {\n        ...assistant,\n        content: (update.content ??\n          assistant.content) as ThreadAssistantMessage[\"content\"],\n        status: latestStatus,\n        metadata,\n      };\n    });\n    if (!touched) return messageId;\n\n    let resolvedMessageId = messageId;\n    const isSettled =\n      latestStatus !== undefined && latestStatus.type !== \"running\";\n    if (isSettled && isOptimisticId(messageId)) {\n      const stableId = generateId();\n      this.reassignAssistantId(messageId, stableId);\n      resolvedMessageId = stableId;\n    } else {\n      this.notifyUpdate();\n    }\n    if (this.isPersistableStatus(latestStatus)) {\n      this.persistAssistantHistory(resolvedMessageId);\n    }\n    return resolvedMessageId;\n  }\n\n  private mergeAssistantMetadata(\n    current: ThreadAssistantMessage[\"metadata\"],\n    incoming: NonNullable<ChatModelRunResult[\"metadata\"]>,\n  ): ThreadAssistantMessage[\"metadata\"] {\n    const annotations = incoming.unstable_annotations\n      ? [...current.unstable_annotations, ...incoming.unstable_annotations]\n      : current.unstable_annotations;\n    const data = incoming.unstable_data\n      ? [...current.unstable_data, ...incoming.unstable_data]\n      : current.unstable_data;\n    const steps = incoming.steps\n      ? [...current.steps, ...incoming.steps]\n      : current.steps;\n    return {\n      unstable_state:\n        incoming.unstable_state !== undefined\n          ? incoming.unstable_state\n          : current.unstable_state,\n      unstable_annotations: annotations,\n      unstable_data: data,\n      steps,\n      ...(incoming.timing ? { timing: incoming.timing } : {}),\n      custom: incoming.custom\n        ? { ...current.custom, ...incoming.custom }\n        : current.custom,\n    };\n  }\n\n  private handleEvent(aggregator: RunAggregator, event: AgUiEvent) {\n    switch (event.type) {\n      case \"STATE_SNAPSHOT\": {\n        this.stateSnapshot = event.snapshot as ReadonlyJSONValue;\n        this.notifyUpdate();\n        return;\n      }\n      case \"STATE_DELTA\": {\n        this.logger.debug?.(\"[agui] state delta event ignored\", event.delta);\n        return;\n      }\n      case \"MESSAGES_SNAPSHOT\": {\n        this.importMessagesSnapshot(event.messages);\n        return;\n      }\n      default:\n        aggregator.handle(event);\n    }\n  }\n\n  private importMessagesSnapshot(rawMessages: readonly unknown[]) {\n    try {\n      const normalized = fromAgUiMessages(rawMessages);\n      const converted: ThreadMessage[] = [];\n      for (const message of normalized) {\n        try {\n          converted.push(\n            fromThreadMessageLike(\n              message as any,\n              generateId(),\n              FALLBACK_USER_STATUS,\n            ),\n          );\n        } catch (error) {\n          this.logger.error?.(\n            \"[agui] failed to import message from snapshot\",\n            error,\n          );\n        }\n      }\n      this.applyExternalMessages(converted);\n    } catch (error) {\n      this.logger.error?.(\"[agui] failed to import messages snapshot\", error);\n    }\n  }\n\n  private toThreadMessage(message: AppendMessage): ThreadMessage {\n    return fromThreadMessageLike(\n      message as any,\n      generateId(),\n      FALLBACK_USER_STATUS,\n    );\n  }\n\n  private resetHead(parentId: string | null | undefined) {\n    if (!parentId) {\n      if (this.messages.length) {\n        this.messages = [];\n      }\n      return;\n    }\n    const idx = this.messages.findIndex((message) => message.id === parentId);\n    if (idx === -1) return;\n    this.messages = this.messages.slice(0, idx + 1);\n  }\n\n  private isTerminalStatus(status?: MessageStatus): boolean {\n    return status?.type === \"complete\" || status?.type === \"incomplete\";\n  }\n\n  private isPersistableStatus(status?: MessageStatus): boolean {\n    if (this.isTerminalStatus(status)) return true;\n    return status?.type === \"requires-action\" && status.reason === \"interrupt\";\n  }\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.messages.find((m) => m.id === messageId);\n    if (!message || message.role !== \"assistant\") return;\n    if (!this.isPersistableStatus(message.status)) 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((error) => {\n      this.recordedHistoryIds.delete(message.id);\n      this.logger.error?.(\"[agui] failed to append history entry\", error);\n    });\n  }\n}\n"],"mappings":";;;;;;AA6BA,MAAM,mBAAmB;AACzB,MAAM,6BAA6B,GAAG,mBAAmB,WAAW;AACpE,MAAM,kBAAkB,OAAe,GAAG,WAAW,gBAAgB;AAErE,MAAM,mBAAmB,OAAO,kBAAkB;AAoBlD,MAAM,uBAAuB;CAAE,MAAM;CAAY,QAAQ;AAAU;AAEnE,IAAa,wBAAb,MAAmC;CACjC;CACA;CACA;CACA;CACA;CACA;CAEA;CACA,WAAoC,CAAC;CACrC,gBAAwB;CACxB,kBAAkD;CAClD;CACA,eAAqC;CACrC;CACA;CACA,0CAA2C,IAAI,IAA2B;CAC1E,qCAAsC,IAAI,IAAY;CACtD,aAAqB;CACrB;CAEA,YAAY,SAAsB;EAChC,KAAK,QAAQ,QAAQ;EACrB,KAAK,SAAS,QAAQ;EACtB,KAAK,eAAe,QAAQ;EAC5B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,UAAU,QAAQ;EACvB,KAAK,eAAe,QAAQ;EAC5B,KAAK,kBAAkB;CACzB;CAEA,cAAc,SAA4C;EACxD,KAAK,QAAQ,QAAQ;EACrB,KAAK,SAAS,QAAQ;EACtB,KAAK,eAAe,QAAQ;EAC5B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,UAAU,QAAQ;EACvB,KAAK,kBAAkB;CACzB;CAEA,cAAc,SAA2B;EACvC,KAAK,UAAU;CACjB;CAEA,gBAAgB;EACd,KAAK,UAAU,KAAA;CACjB;CAEA,cAAwC;EACtC,OAAO,KAAK;CACd;CAEA,WAA0C;EACxC,OAAO,KAAK;CACd;CAEA,YAAqB;EACnB,OAAO,KAAK;CACd;CAEA,IAAI,YAAqB;EACvB,OAAO,KAAK;CACd;CAEA,kBAAiC;EAC/B,IAAI,KAAK,cAAc,OAAO,KAAK;EAEnC,MAAM,UAAU,KAAK,SAAS,KAAK,KAAK,QAAQ,QAAQ,IAAI;EAE5D,KAAK,aAAa;EAElB,KAAK,eAAe,QACjB,KAAK,OAAO,SAAS;GACpB,IAAI,CAAC,MAAM;GAEX,MAAM,WAAW,KAAK,SAAS,KAAK,SAAS,KAAK,OAAO;GACzD,KAAK,sBAAsB,QAAQ;GAEnC,IAAI,KAAK,iBAAiB;IACxB,MAAM,WAAW,KAAK,UAAU,SAAS,GAAG,EAAE,GAAG,MAAM;IACvD,MAAM,KAAK,SAAS,UAAU,KAAK,aAAa;GAClD;EACF,CAAC,EACA,OAAO,UAAU;GAChB,KAAK,OAAO,QAAQ,iCAAiC,KAAK;GAC1D,KAAK,UACH,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAC1D;EACF,CAAC,EACA,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;EACtD,IAAI,UAAU,KAAK,0BAA0B;EAC7C,IAAI,QAAQ,UACV,KAAK,WAAW,KAAK,SAAS,QAC3B,UAAU,MAAM,OAAO,QAAQ,QAClC;EAEF,KAAK,UAAU,QAAQ,QAAQ;EAE/B,MAAM,gBAAgB,KAAK,gBAAgB,OAAO;EAClD,KAAK,WAAW,CAAC,GAAG,KAAK,UAAU,aAAa;EAChD,KAAK,aAAa;EAClB,KAAK,mBAAmB,QAAQ,YAAY,MAAM,aAAa;EAE/D,IAAI,CAAC,UAAU;EACf,MAAM,KAAK,SAAS,cAAc,IAAI,QAAQ,SAAS;CACzD;CAEA,MAAM,KAAK,SAAuC;EAChD,MAAM,KAAK,OAAO,OAAO;CAC3B;CAEA,MAAM,OACJ,UACA,SAAoC,CAAC,GACtB;EACf,KAAK,0BAA0B;EAC/B,KAAK,UAAU,QAAQ;EACvB,KAAK,aAAa;EAClB,MAAM,KAAK,SAAS,UAAU,OAAO,SAAS;CAChD;CAEA,MAAM,SAAwB;EAC5B,IAAI,CAAC,KAAK,iBAAiB;EAC3B,KAAK,gBAAgB,MAAM;CAC7B;CAEA,MAAM,OAAO,QAAwC;EACnD,KAAK,0BAA0B;EAC/B,IAAI,OAAO,QACT,KAAK,OAAO,QACV,oEACF;EAEF,MAAM,KAAK,SACT,OAAO,UACP,OAAO,aAAa,KAAK,aAC3B;CACF;CAEA,4BAA0C;EACxC,IAAI,CAAC,KAAK,qBAAqB,GAAG;EAClC,MAAM,IAAI,MACR,0GACF;CACF;CAEA,uBAGS;EACP,MAAM,YAAY,KAAK,SAAS,UAAU,MAAM,EAAE,SAAS,WAAW;EAGtE,IACE,CAAC,aACD,UAAU,QAAQ,SAAS,qBAC3B,UAAU,OAAO,WAAW,aAE5B,OAAO;EAET,MAAM,SACJ,UAAU,SAAS,OAAO,2BAGzB;EACH,IAAI,CAAC,QAAQ,QAAQ,OAAO;EAC5B,OAAO;GAAE,WAAW,UAAU;GAAI,YAAY;EAAO;CACvD;CAEA,MAAM,yBACJ,WACe;EACf,MAAM,UAAU,KAAK,qBAAqB;EAC1C,IAAI,CAAC,SACH,MAAM,IAAI,MACR,uEACF;EAGF,MAAM,gCAAgB,IAAI,IAA6B;EACvD,KAAK,MAAM,SAAS,WAAW;GAC7B,IAAI,CAAC,SAAS,OAAO,MAAM,gBAAgB,UACzC,MAAM,IAAI,MACR,uEACF;GAEF,IAAI,MAAM,WAAW,cAAc,MAAM,WAAW,aAClD,MAAM,IAAI,MACR,oDAAoD,MAAM,OAAO,kBAAkB,MAAM,aAC3F;GAEF,IAAI,cAAc,IAAI,MAAM,WAAW,GACrC,MAAM,IAAI,MACR,qEAAqE,MAAM,aAC7E;GAEF,cAAc,IAAI,MAAM,aAAa,KAAK;EAC5C;EAEA,MAAM,UAAU,QAAQ,WAAW,KAAK,MAAM,EAAE,EAAE;EAClD,MAAM,UAAU,QAAQ,QAAQ,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC;EAC7D,IAAI,QAAQ,SAAS,GACnB,MAAM,IAAI,MACR,2EAA2E,QAAQ,KAAK,IAAI,GAC9F;EAEF,MAAM,QAAQ,IAAI,IAAI,OAAO;EAC7B,MAAM,aAAa,CAAC,GAAG,cAAc,KAAK,CAAC,EAAE,QAAQ,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;EAC1E,IAAI,WAAW,SAAS,GACtB,MAAM,IAAI,MACR,2DAA2D,WAAW,KAAK,IAAI,GACjF;EAGF,MAAM,MAAM,KAAK,IAAI;EACrB,KAAK,MAAM,aAAa,QAAQ,YAAY;GAC1C,IAAI,CAAC,UAAU,WAAW;GAC1B,MAAM,SAAS,IAAI,KAAK,UAAU,SAAS,EAAE,QAAQ;GACrD,IAAI,OAAO,MAAM,MAAM,GACrB,MAAM,IAAI,MACR,8CAA8C,UAAU,GAAG,4BAA4B,UAAU,UAAU,EAC7G;GAEF,IAAI,UAAU,KACZ,MAAM,IAAI,MACR,8CAA8C,UAAU,GAAG,cAAc,UAAU,WACrF;EAEJ;EAEA,MAAM,SAA4B,QAAQ,KACvC,OAAO,cAAc,IAAI,EAAE,CAC9B;EAEA,IAAI,KAAK,eACP,MAAM,IAAI,MACR,+DACF;EAGF,KAAK,uBAAuB,QAAQ,SAAS;EAC7C,MAAM,KAAK,SAAS,QAAQ,WAAW,KAAK,eAAe,MAAM;CACnE;CAEA,uBAA+B,WAAyB;EACtD,IAAI,UAAU;EACd,KAAK,WAAW,KAAK,SAAS,KAAK,YAAY;GAC7C,IAAI,QAAQ,OAAO,aAAa,QAAQ,SAAS,aAC/C,OAAO;GACT,MAAM,YAAY;GAClB,IACE,UAAU,QAAQ,SAAS,qBAC3B,UAAU,OAAO,WAAW,aAE5B,OAAO;GAET,UAAU;GAIV,MAAM,EAAE,YAAY,OAAO,GAAG,aAHb,UAAU,SAAS,OAAA,WAGmB,CAAC;GACxD,MAAM,YAAY,EAAE,GAAG,UAAU,SAAS,OAAO;GACjD,IAAI,OAAO,KAAK,QAAQ,EAAE,SAAS,GACjC,UAAU,4BAA4B;QAEtC,OAAO,UAAU;GAEnB,OAAO;IACL,GAAG;IACH,QAAQ;KAAE,MAAM;KAAqB,QAAQ;IAAmB;IAChE,UAAU;KAAE,GAAG,UAAU;KAAU,QAAQ;IAAU;GACvD;EACF,CAAC;EACD,IAAI,SAAS,KAAK,aAAa;CACjC;CAEA,yBAAyB,YAAwC;EAC/D,IAAI;EACJ,KAAK,IAAI,QAAQ,KAAK,SAAS,SAAS,GAAG,SAAS,GAAG,SAAS;GAC9D,MAAM,UAAU,KAAK,SAAS;GAC9B,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;GAC9C,KAAK,MAAM,QAAQ,QAAQ,SAAS;IAClC,IAAI,KAAK,SAAS,eAAe,KAAK,eAAe,YACnD;IACF,IAAI,EAAE,YAAY,SAAS,KAAK,WAAW,KAAA,GACzC,OAAO,QAAQ;IAEjB,sBAAsB,QAAQ;GAChC;EACF;EACA,OAAO;CACT;CAEA,cAAc,SAAqC;EACjD,IAAI,UAAU;EACd,IAAI,eAAe;EACnB,KAAK,WAAW,KAAK,SAAS,KAAK,YAAY;GAC7C,IAAI,QAAQ,OAAO,QAAQ,aAAa,QAAQ,SAAS,aACvD,OAAO;GACT,MAAM,YAAY;GAClB,IAAI,kBAAkB;GACtB,MAAM,UAAU,UAAU,QAAQ,KAAK,SAAS;IAC9C,IAAI,KAAK,SAAS,eAAe,KAAK,eAAe,QAAQ,YAC3D,OAAO;IACT,kBAAkB;IAClB,OAAO;KACL,GAAG;KACH,QAAQ,QAAQ;KAChB,UAAU,QAAQ;KAClB,SAAS,QAAQ;IACnB;GACF,CAAC;GACD,IAAI,CAAC,iBAAiB,OAAO;GAC7B,UAAU;GAEV,IACE,UAAU,QAAQ,SAAS,qBAC3B,UAAU,OAAO,WAAW,gBAC5B,QAAQ,OACL,SACC,KAAK,SAAS,eACb,YAAY,QAAQ,KAAK,WAAW,KAAA,CACzC,GACA;IACA,eAAe;IACf,OAAO;KACL,GAAG;KACH;KACA,QAAQ;MAAE,MAAM;MAAqB,QAAQ;KAAmB;IAClE;GACF;GAEA,OAAO;IACL,GAAG;IACH;GACF;EACF,CAAC;EAED,IAAI,SAAS;GACX,KAAK,aAAa;GAElB,IAAI,cAAc;IAChB,KAAK,wBAAwB,QAAQ,SAAS;IAE9C,IAAI,CAAC,KAAK,eACR,KAAU,SAAS,QAAQ,WAAW,KAAK,aAAa,EAAE,OACvD,UAAU;KACT,KAAK,UACH,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAC1D;IACF,CACF;GAEJ;EACF;CACF;CAEA,sBAAsB,UAA0C;EAC9D,KAAK,wBAAwB,MAAM;EACnC,KAAK,WAAW,CAAC,GAAG,QAAQ;EAC5B,KAAK,mBAAmB,MAAM;EAC9B,KAAK,MAAM,WAAW,KAAK,UACzB,KAAK,mBAAmB,IAAI,QAAQ,EAAE;EAExC,KAAK,aAAa;CACpB;CAEA,kBAAkB,OAAgC;EAChD,KAAK,gBAAgB;EACrB,KAAK,aAAa;CACpB;CAEA,MAAc,SACZ,UACA,WACA,QACe;EACf,MAAM,sBAAsB,aAAa,CAAC;EAC1C,KAAK,gBAAgB;EACrB,KAAK,UAAU,QAAQ;EACvB,MAAM,qBAAqB,CAAC,GAAG,KAAK,QAAQ;EAE5C,MAAM,QAAQ,WAAW;EACzB,KAAK,eAAe;EACpB,MAAM,QAAQ,KAAK,cACjB,OACA,qBACA,oBACA,MACF;EACA,MAAM,oBAAoB,YAAY,KAAK,SAAS,GAAG,EAAE,GAAG,MAAM;EAClE,IAAI;EACJ,MAAM,wBAAwB;GAC5B,IAAI,oBAAoB,OAAO;GAC/B,MAAM,UAAU,KAAK,2BAA2B;GAChD,qBAAqB;GACrB,KAAK,4BAA4B,SAAS,qBAAqB,IAAI;GACnE,OAAO;EACT;EAEA,MAAM,aAAa,IAAI,cAAc;GACnC,cAAc,KAAK;GACnB,QAAQ,KAAK;GACb,OAAO,WAAW;IAChB,MAAM,WAAW,KAAK,uBAAuB,gBAAgB,GAAG,MAAM;IACtE,IAAI,aAAa,oBACf,qBAAqB;GAEzB;GACA,oBAAoB,aAAa;IAC/B,MAAM,cAAc,gBAAgB;IACpC,IAAI,gBAAgB,UAAU;IAC9B,KAAK,oBAAoB,aAAa,QAAQ;IAC9C,qBAAqB;GACvB;EACF,CAAC;EACD,MAAM,YAAY,UAAqB,KAAK,YAAY,YAAY,KAAK;EAEzE,MAAM,kBAAkB,IAAI,gBAAgB;EAC5C,MAAM,cAAc,gBAAgB;EACpC,KAAK,kBAAkB;EAEvB,YAAY,iBACV,eACM;GACJ,SAAS,EAAE,MAAM,gBAAgB,CAAC;GAClC,KAAK,UAAU,eAAe;GAC9B,KAAK,WAAW;EAClB,GACA,EAAE,MAAM,KAAK,CACf;EAEA,MAAM,aAAa,qBAAqB;GACtC;GACA;GACA,QAAQ,KAAK;GACb,cAAc,UAAU;IACtB,KAAK,eAAe;IACpB,KAAK,UAAU,KAAK;GACtB;EACF,CAAC;EAED,WAAW,OAAO;GAAE,MAAM;GAAe;EAAM,CAAC;EAChD,KAAK,WAAW,IAAI;EAEpB,IAAI;GACF,IAAI;IACF,KAAM,MAAc,WAAW,MAAM;IACrC,KAAM,MAAc,WAAW,MAAM;IACrC,KAAM,MAAc,QAAQ,MAAM,SAAS;GAC7C,QAAQ,CAER;GACA,MAAO,KAAK,MAAc,SAAS,OAAO,YAAY,EACpD,QAAQ,YACV,CAAC;EACH,SAAS,OAAO;GACd,IAAI,CAAC,YAAY,SAAS;IACxB,MAAM,MAAM,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;IACpE,SAAS;KAAE,MAAM;KAAa,SAAS,IAAI;IAAQ,CAAC;IACpD,KAAK,UAAU,GAAG;IAClB,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,cACE,OACA,WACA,iBACA,QACA;EACA,MAAM,WAAW,KAAK,MAAM,YAAY;EACxC,MAAM,WAAW,eAAe,mBAAmB,KAAK,QAAQ;EAChE,MAAM,UAAU,KAAK,SAAS,OAAO,gBAAgB;EACrD,OAAO;GACL;GACA;GACA,OAAO,KAAK,iBAAiB;GAC7B;GACA,OAAO,YAAY,SAAS,KAAK;GACjC,SAAS,SAAS,SACd,CAAC;IAAE,aAAa;IAAU,OAAO,QAAQ;GAAO,CAAC,IACjD,CAAC;GACL,gBAAgB;IACd,GAAI,SAAS,gBAAgB,CAAC;IAC9B,GAAI,SAAS,UAAU,CAAC;IACxB,GAAI,WAAW,SAAS,EAAE,WAAW,UAAU,OAAO,IAAI,CAAC;GAC7D;GACA,GAAI,WAAW,KAAA,IAAY,EAAE,OAAO,IAAI,CAAC;EAC3C;CACF;CAEA,oBAAkC;EAChC,MAAM,QAAQ,KAAK;EACnB,IAAI,MAAM,mBAAmB;EAC7B,MAAM,oBAAoB;EAE1B,MAAM,WADa,OAAO,OAAO,OAAO,sBACd,IACtB,MAAM,uBACN,OAAO,eAAe,KAAK,GAAG;EAClC,IAAI,OAAO,aAAa,YAAY;EACpC,MAAM,uBAAuB,SAE3B,QACA;GACA,MAAM,QAAQ,SAAS,KAAK,MAAM,MAAM;GACxC,IAAI,QAAQ,WAAW,KAAA,KAAa,SAAS,OAAO,UAAU,UAC5D,OAAO;IAAE,GAAI;IAAkB,QAAQ,OAAO;GAAO;GAEvD,OAAO;EACT;CACF;CAEA,WAAmB,SAAkB;EACnC,KAAK,gBAAgB;EACrB,KAAK,aAAa;CACpB;CAEA,UAAkB,YAAoC;EACpD,IAAI,KAAK,oBAAoB,YAC3B,KAAK,kBAAkB;EAEzB,KAAK,WAAW,KAAK;CACvB;CAEA,6BAA6C;EAC3C,MAAM,KAAK,qBAAqB;EAChC,MAAM,YAAoC;GACxC;GACA,MAAM;GACN,2BAAW,IAAI,KAAK;GACpB,QAAQ,EAAE,MAAM,UAAU;GAC1B,SAAS,CAAC;GACV,UAAU;IACR,gBAAgB,KAAK,iBAAiB;IACtC,sBAAsB,CAAC;IACvB,eAAe,CAAC;IAChB,OAAO,CAAC;IACR,QAAQ,CAAC;GACX;EACF;EACA,KAAK,WAAW,CAAC,GAAG,KAAK,UAAU,SAAS;EAC5C,KAAK,aAAa;EAClB,OAAO;CACT;CAEA,oBAA4B,OAAe,OAAqB;EAC9D,IAAI,UAAU,OAAO;EAIrB,IAF6B,KAAK,SAAS,MAAM,MAAM,EAAE,OAAO,KAEzC,GAAG;GACxB,KAAK,OAAO,QACV,2FACA;IAAE;IAAO;GAAM,CACjB;GACA,KAAK,WAAW,KAAK,SAAS,QAAQ,MAAM,EAAE,OAAO,KAAK;EAC5D,OACE,KAAK,WAAW,KAAK,SAAS,KAAK,MACjC,EAAE,OAAO,QAAQ;GAAE,GAAG;GAAG,IAAI;EAAM,IAAI,CACzC;EAGF,MAAM,gBAAgB,KAAK,wBAAwB,IAAI,KAAK;EAC5D,IAAI,kBAAkB,KAAA,GAAW;GAC/B,KAAK,wBAAwB,OAAO,KAAK;GACzC,IAAI,CAAC,KAAK,wBAAwB,IAAI,KAAK,GACzC,KAAK,wBAAwB,IAAI,OAAO,aAAa;EAEzD;EAEA,IAAI,KAAK,mBAAmB,IAAI,KAAK,GAAG;GACtC,KAAK,mBAAmB,OAAO,KAAK;GACpC,KAAK,mBAAmB,IAAI,KAAK;EACnC;EAEA,KAAK,aAAa;CACpB;CAEA,uBACE,WACA,QACQ;EACR,IAAI,UAAU;EACd,IAAI;EACJ,KAAK,WAAW,KAAK,SAAS,KAAK,YAAY;GAC7C,IAAI,QAAQ,OAAO,aAAa,QAAQ,SAAS,aAC/C,OAAO;GACT,UAAU;GACV,MAAM,YAAY;GAClB,MAAM,WAAW,OAAO,WACpB,KAAK,uBAAuB,UAAU,UAAU,OAAO,QAAQ,IAC/D,UAAU;GACd,eAAe,OAAO,UAAU,UAAU;GAC1C,OAAO;IACL,GAAG;IACH,SAAU,OAAO,WACf,UAAU;IACZ,QAAQ;IACR;GACF;EACF,CAAC;EACD,IAAI,CAAC,SAAS,OAAO;EAErB,IAAI,oBAAoB;EAGxB,IADE,iBAAiB,KAAA,KAAa,aAAa,SAAS,aACrC,eAAe,SAAS,GAAG;GAC1C,MAAM,WAAW,WAAW;GAC5B,KAAK,oBAAoB,WAAW,QAAQ;GAC5C,oBAAoB;EACtB,OACE,KAAK,aAAa;EAEpB,IAAI,KAAK,oBAAoB,YAAY,GACvC,KAAK,wBAAwB,iBAAiB;EAEhD,OAAO;CACT;CAEA,uBACE,SACA,UACoC;EACpC,MAAM,cAAc,SAAS,uBACzB,CAAC,GAAG,QAAQ,sBAAsB,GAAG,SAAS,oBAAoB,IAClE,QAAQ;EACZ,MAAM,OAAO,SAAS,gBAClB,CAAC,GAAG,QAAQ,eAAe,GAAG,SAAS,aAAa,IACpD,QAAQ;EACZ,MAAM,QAAQ,SAAS,QACnB,CAAC,GAAG,QAAQ,OAAO,GAAG,SAAS,KAAK,IACpC,QAAQ;EACZ,OAAO;GACL,gBACE,SAAS,mBAAmB,KAAA,IACxB,SAAS,iBACT,QAAQ;GACd,sBAAsB;GACtB,eAAe;GACf;GACA,GAAI,SAAS,SAAS,EAAE,QAAQ,SAAS,OAAO,IAAI,CAAC;GACrD,QAAQ,SAAS,SACb;IAAE,GAAG,QAAQ;IAAQ,GAAG,SAAS;GAAO,IACxC,QAAQ;EACd;CACF;CAEA,YAAoB,YAA2B,OAAkB;EAC/D,QAAQ,MAAM,MAAd;GACE,KAAK;IACH,KAAK,gBAAgB,MAAM;IAC3B,KAAK,aAAa;IAClB;GAEF,KAAK;IACH,KAAK,OAAO,QAAQ,oCAAoC,MAAM,KAAK;IACnE;GAEF,KAAK;IACH,KAAK,uBAAuB,MAAM,QAAQ;IAC1C;GAEF,SACE,WAAW,OAAO,KAAK;EAC3B;CACF;CAEA,uBAA+B,aAAiC;EAC9D,IAAI;GACF,MAAM,aAAa,iBAAiB,WAAW;GAC/C,MAAM,YAA6B,CAAC;GACpC,KAAK,MAAM,WAAW,YACpB,IAAI;IACF,UAAU,KACR,sBACE,SACA,WAAW,GACX,oBACF,CACF;GACF,SAAS,OAAO;IACd,KAAK,OAAO,QACV,iDACA,KACF;GACF;GAEF,KAAK,sBAAsB,SAAS;EACtC,SAAS,OAAO;GACd,KAAK,OAAO,QAAQ,6CAA6C,KAAK;EACxE;CACF;CAEA,gBAAwB,SAAuC;EAC7D,OAAO,sBACL,SACA,WAAW,GACX,oBACF;CACF;CAEA,UAAkB,UAAqC;EACrD,IAAI,CAAC,UAAU;GACb,IAAI,KAAK,SAAS,QAChB,KAAK,WAAW,CAAC;GAEnB;EACF;EACA,MAAM,MAAM,KAAK,SAAS,WAAW,YAAY,QAAQ,OAAO,QAAQ;EACxE,IAAI,QAAQ,IAAI;EAChB,KAAK,WAAW,KAAK,SAAS,MAAM,GAAG,MAAM,CAAC;CAChD;CAEA,iBAAyB,QAAiC;EACxD,OAAO,QAAQ,SAAS,cAAc,QAAQ,SAAS;CACzD;CAEA,oBAA4B,QAAiC;EAC3D,IAAI,KAAK,iBAAiB,MAAM,GAAG,OAAO;EAC1C,OAAO,QAAQ,SAAS,qBAAqB,OAAO,WAAW;CACjE;CAEA,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,SAAS,MAAM,MAAM,EAAE,OAAO,SAAS;EAC5D,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;EAC9C,IAAI,CAAC,KAAK,oBAAoB,QAAQ,MAAM,GAAG;EAC/C,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,EAAE,OAAO,UAAU;GAC/D,KAAK,mBAAmB,OAAO,QAAQ,EAAE;GACzC,KAAK,OAAO,QAAQ,yCAAyC,KAAK;EACpE,CAAC;CACH;AACF"}
+{"version":3,"file":"AgUiThreadRuntimeCore.js","names":[],"sources":["../../src/runtime/AgUiThreadRuntimeCore.ts"],"sourcesContent":["\"use client\";\n\nimport { generateId, fromThreadMessageLike } from \"@assistant-ui/core/internal\";\nimport type {\n  AddToolResultOptions,\n  AppendMessage,\n  AssistantRuntime,\n  ChatModelRunOptions,\n  ChatModelRunResult,\n  MessageStatus,\n  ThreadAssistantMessage,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n  ToolCallMessagePart,\n} from \"@assistant-ui/core\";\nimport type { HttpAgent } from \"@ag-ui/client\";\nimport jsonpatch, { type Operation } from \"fast-json-patch\";\nimport type { Logger } from \"./logger\";\nimport type { AgUiEvent, AgUiInterrupt, AgUiResumeEntry } from \"./types\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport {\n  AG_UI_METADATA_NAMESPACE,\n  type AgUiCustomMetadata,\n  RunAggregator,\n} from \"./adapter/run-aggregator\";\nimport {\n  fromAgUiMessages,\n  toAgUiMessages,\n  toAgUiTools,\n} from \"./adapter/conversions\";\nimport { createAgUiSubscriber } from \"./adapter/subscriber\";\n\nconst optimisticPrefix = \"__optimistic__\";\nconst generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;\nconst isOptimisticId = (id: string) => id.startsWith(optimisticPrefix);\n\nconst symbolResumeShim = Symbol(\"agui-resume-shim\");\n\ntype RunConfig = NonNullable<AppendMessage[\"runConfig\"]>;\ntype ResumeStream = (\n  options: ChatModelRunOptions,\n) => AsyncGenerator<ChatModelRunResult, void, unknown>;\ntype ResumeRunConfig = {\n  parentId: string | null;\n  sourceId: string | null;\n  runConfig: RunConfig;\n  stream?: ResumeStream;\n};\n\ntype CoreOptions = {\n  agent: HttpAgent;\n  logger: Logger;\n  showThinking: boolean;\n  onError?: (error: Error) => void;\n  onCancel?: () => void;\n  history?: ThreadHistoryAdapter;\n  notifyUpdate: () => void;\n};\n\nconst FALLBACK_USER_STATUS = { type: \"complete\", reason: \"unknown\" } as const;\n\nexport class AgUiThreadRuntimeCore {\n  private agent: HttpAgent;\n  private logger: Logger;\n  private showThinking: boolean;\n  private onError: ((error: Error) => void) | undefined;\n  private onCancel: (() => void) | undefined;\n  private readonly notifyUpdate: () => void;\n\n  private runtime: AssistantRuntime | undefined;\n  private messages: ThreadMessage[] = [];\n  private isRunningFlag = false;\n  private abortController: AbortController | null = null;\n  private stateSnapshot: ReadonlyJSONValue | undefined;\n  private pendingError: Error | null = null;\n  private history: ThreadHistoryAdapter | undefined;\n  private lastRunConfig: RunConfig | undefined;\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 pendingResumeMessageId: string | null = null;\n\n  constructor(options: CoreOptions) {\n    this.agent = options.agent;\n    this.logger = options.logger;\n    this.showThinking = options.showThinking;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.history = options.history;\n    this.notifyUpdate = options.notifyUpdate;\n    this.installResumeShim();\n  }\n\n  updateOptions(options: Omit<CoreOptions, \"notifyUpdate\">) {\n    this.agent = options.agent;\n    this.logger = options.logger;\n    this.showThinking = options.showThinking;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.history = options.history;\n    this.installResumeShim();\n  }\n\n  attachRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  detachRuntime() {\n    this.runtime = undefined;\n  }\n\n  getMessages(): readonly ThreadMessage[] {\n    return this.messages;\n  }\n\n  getState(): ReadonlyJSONValue | undefined {\n    return this.stateSnapshot;\n  }\n\n  isRunning(): boolean {\n    return this.isRunningFlag;\n  }\n\n  get isLoading(): boolean {\n    return this._isLoading;\n  }\n\n  __internal_load(): Promise<void> {\n    if (this._loadPromise) return this._loadPromise;\n\n    const promise = this.history?.load() ?? Promise.resolve(null);\n\n    this._isLoading = true;\n\n    this._loadPromise = promise\n      .then(async (repo) => {\n        if (!repo) return;\n\n        const messages = repo.messages.map((item) => item.message);\n        this.applyExternalMessages(messages);\n\n        if (repo.unstable_resume) {\n          const parentId = repo.headId ?? messages.at(-1)?.id ?? null;\n          const resumeStream = this.history?.resume?.bind(this.history);\n          await this.startRun(\n            parentId,\n            this.lastRunConfig,\n            undefined,\n            resumeStream,\n          );\n        }\n      })\n      .catch((error) => {\n        this.logger.error?.(\"[agui] failed to load history\", error);\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    if (startRun) this.assertNoPendingInterrupts();\n    if (message.sourceId) {\n      this.messages = this.messages.filter(\n        (entry) => entry.id !== message.sourceId,\n      );\n    }\n    this.resetHead(message.parentId);\n\n    const threadMessage = this.toThreadMessage(message);\n    this.messages = [...this.messages, threadMessage];\n    this.notifyUpdate();\n    this.recordHistoryEntry(message.parentId ?? null, threadMessage);\n\n    if (!startRun) return;\n    await this.startRun(threadMessage.id, message.runConfig);\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?: RunConfig } = {},\n  ): Promise<void> {\n    this.assertNoPendingInterrupts();\n    this.resetHead(parentId);\n    this.notifyUpdate();\n    await this.startRun(parentId, config.runConfig);\n  }\n\n  async cancel(): Promise<void> {\n    if (!this.abortController) return;\n    this.abortController.abort();\n  }\n\n  async resume(config: ResumeRunConfig): Promise<void> {\n    this.assertNoPendingInterrupts();\n    await this.startRun(\n      config.parentId,\n      config.runConfig ?? this.lastRunConfig,\n      undefined,\n      config.stream,\n    );\n  }\n\n  private assertNoPendingInterrupts(): void {\n    if (!this.getPendingInterrupts()) return;\n    throw new Error(\n      \"[agui] cannot start a new run while interrupts are pending; resolve them with submitInterruptResponses()\",\n    );\n  }\n\n  getPendingInterrupts(): {\n    messageId: string;\n    interrupts: readonly AgUiInterrupt[];\n  } | null {\n    const assistant = this.messages.findLast((m) => m.role === \"assistant\") as\n      | ThreadAssistantMessage\n      | undefined;\n    if (\n      !assistant ||\n      assistant.status?.type !== \"requires-action\" ||\n      assistant.status.reason !== \"interrupt\"\n    ) {\n      return null;\n    }\n    const stored = (\n      assistant.metadata.custom[AG_UI_METADATA_NAMESPACE] as\n        | AgUiCustomMetadata\n        | undefined\n    )?.interrupts;\n    if (!stored?.length) return null;\n    return { messageId: assistant.id, interrupts: stored };\n  }\n\n  async submitInterruptResponses(\n    responses: readonly AgUiResumeEntry[],\n  ): Promise<void> {\n    const pending = this.getPendingInterrupts();\n    if (!pending) {\n      throw new Error(\n        \"[agui] submitInterruptResponses: no pending interrupts on this thread\",\n      );\n    }\n\n    const responsesById = new Map<string, AgUiResumeEntry>();\n    for (const entry of responses) {\n      if (!entry || typeof entry.interruptId !== \"string\") {\n        throw new Error(\n          \"[agui] submitInterruptResponses: every entry must have an interruptId\",\n        );\n      }\n      if (entry.status !== \"resolved\" && entry.status !== \"cancelled\") {\n        throw new Error(\n          `[agui] submitInterruptResponses: invalid status \"${entry.status}\" for interrupt ${entry.interruptId}`,\n        );\n      }\n      if (responsesById.has(entry.interruptId)) {\n        throw new Error(\n          `[agui] submitInterruptResponses: duplicate response for interrupt ${entry.interruptId}`,\n        );\n      }\n      responsesById.set(entry.interruptId, entry);\n    }\n\n    const openIds = pending.interrupts.map((i) => i.id);\n    const missing = openIds.filter((id) => !responsesById.has(id));\n    if (missing.length > 0) {\n      throw new Error(\n        `[agui] submitInterruptResponses: missing responses for open interrupts: ${missing.join(\", \")}`,\n      );\n    }\n    const known = new Set(openIds);\n    const unknownIds = [...responsesById.keys()].filter((id) => !known.has(id));\n    if (unknownIds.length > 0) {\n      throw new Error(\n        `[agui] submitInterruptResponses: unknown interrupt ids: ${unknownIds.join(\", \")}`,\n      );\n    }\n\n    const now = Date.now();\n    for (const interrupt of pending.interrupts) {\n      if (!interrupt.expiresAt) continue;\n      const expiry = new Date(interrupt.expiresAt).getTime();\n      if (Number.isNaN(expiry)) {\n        throw new Error(\n          `[agui] submitInterruptResponses: interrupt ${interrupt.id} has malformed expiresAt \"${interrupt.expiresAt}\"`,\n        );\n      }\n      if (expiry <= now) {\n        throw new Error(\n          `[agui] submitInterruptResponses: interrupt ${interrupt.id} expired at ${interrupt.expiresAt}`,\n        );\n      }\n    }\n\n    const resume: AgUiResumeEntry[] = openIds.map(\n      (id) => responsesById.get(id)!,\n    );\n\n    if (this.isRunningFlag) {\n      throw new Error(\n        \"[agui] submitInterruptResponses: a run is already in progress\",\n      );\n    }\n\n    this.clearPendingInterrupts(pending.messageId);\n    await this.startRun(pending.messageId, this.lastRunConfig, resume);\n  }\n\n  private clearPendingInterrupts(messageId: string): void {\n    let touched = false;\n    this.messages = this.messages.map((message) => {\n      if (message.id !== messageId || message.role !== \"assistant\")\n        return message;\n      const assistant = message as ThreadAssistantMessage;\n      if (\n        assistant.status?.type !== \"requires-action\" ||\n        assistant.status.reason !== \"interrupt\"\n      ) {\n        return assistant;\n      }\n      touched = true;\n      const aguiMeta = assistant.metadata.custom[AG_UI_METADATA_NAMESPACE] as\n        | AgUiCustomMetadata\n        | undefined;\n      const { interrupts: _drop, ...restAgui } = aguiMeta ?? {};\n      const newCustom = { ...assistant.metadata.custom };\n      if (Object.keys(restAgui).length > 0) {\n        newCustom[AG_UI_METADATA_NAMESPACE] = restAgui;\n      } else {\n        delete newCustom[AG_UI_METADATA_NAMESPACE];\n      }\n      return {\n        ...assistant,\n        status: { type: \"complete\" as const, reason: \"unknown\" as const },\n        metadata: { ...assistant.metadata, custom: newCustom },\n      };\n    });\n    if (touched) this.notifyUpdate();\n  }\n\n  findMessageIdForToolCall(toolCallId: string): string | undefined {\n    let fallbackMessageId: string | undefined;\n    for (let index = this.messages.length - 1; index >= 0; index--) {\n      const message = this.messages[index];\n      if (!message || message.role !== \"assistant\") continue;\n      for (const part of message.content) {\n        if (part.type !== \"tool-call\" || part.toolCallId !== toolCallId)\n          continue;\n        if (!(\"result\" in part) || part.result === undefined) {\n          return message.id;\n        }\n        fallbackMessageId ??= message.id;\n      }\n    }\n    return fallbackMessageId;\n  }\n\n  addToolResult(options: AddToolResultOptions): void {\n    let updated = false;\n    this.messages = this.messages.map((message) => {\n      if (message.id !== options.messageId || message.role !== \"assistant\")\n        return message;\n      const assistant = message as ThreadAssistantMessage;\n      let matchedToolCall = false;\n      const content = assistant.content.map((part) => {\n        if (part.type !== \"tool-call\" || part.toolCallId !== options.toolCallId)\n          return part;\n        matchedToolCall = true;\n        return {\n          ...part,\n          result: options.result,\n          artifact: options.artifact,\n          isError: options.isError,\n        };\n      });\n      if (!matchedToolCall) return message;\n      updated = true;\n      return { ...assistant, content };\n    });\n\n    if (!updated) return;\n    this.notifyUpdate();\n    this.maybeResumeAfterToolResults(options.messageId);\n  }\n\n  // Re-evaluate whether every tool call on the assistant message now has a\n  // result and, if so, drive the follow-up run. Centralizing this lets the\n  // continuation fire whether the frontend result lands before RUN_FINISHED\n  // (status flips to requires-action only later, while a run is still draining)\n  // or after it. Idempotent: once the message is marked complete it no-ops.\n  private maybeResumeAfterToolResults(messageId: string): void {\n    const message = this.messages.find((m) => m.id === messageId);\n    if (!message || message.role !== \"assistant\") return;\n    const assistant = message as ThreadAssistantMessage;\n    if (\n      assistant.status?.type !== \"requires-action\" ||\n      assistant.status.reason !== \"tool-calls\"\n    ) {\n      return;\n    }\n    const allResolved = assistant.content.every(\n      (part) =>\n        part.type !== \"tool-call\" ||\n        (\"result\" in part && part.result !== undefined),\n    );\n    if (!allResolved) return;\n\n    this.messages = this.messages.map((m) =>\n      m.id === messageId && m.role === \"assistant\"\n        ? {\n            ...(m as ThreadAssistantMessage),\n            status: { type: \"complete\" as const, reason: \"unknown\" as const },\n          }\n        : m,\n    );\n    this.notifyUpdate();\n    this.persistAssistantHistory(messageId);\n\n    if (this.isRunningFlag) {\n      // A run is still draining (RUN_FINISHED arrived but the stream has not\n      // closed). Defer until startRun's tail so we never start two runs.\n      this.pendingResumeMessageId = messageId;\n      return;\n    }\n    this.startResumeRun(messageId);\n  }\n\n  private startResumeRun(messageId: string): void {\n    void this.startRun(messageId, this.lastRunConfig).catch((error) => {\n      this.onError?.(error instanceof Error ? error : new Error(String(error)));\n    });\n  }\n\n  applyExternalMessages(messages: readonly ThreadMessage[]): void {\n    this.assistantHistoryParents.clear();\n    this.messages = [...messages];\n    this.recordedHistoryIds.clear();\n    for (const message of this.messages) {\n      this.recordedHistoryIds.add(message.id);\n    }\n    this.notifyUpdate();\n  }\n\n  loadExternalState(state: ReadonlyJSONValue): void {\n    this.stateSnapshot = state;\n    this.notifyUpdate();\n  }\n\n  private async startRun(\n    parentId: string | null,\n    runConfig?: RunConfig,\n    resume?: AgUiResumeEntry[],\n    resumeStream?: ResumeStream,\n  ): Promise<void> {\n    const normalizedRunConfig = runConfig ?? {};\n    this.lastRunConfig = normalizedRunConfig;\n    this.resetHead(parentId);\n    const historicalMessages = [...this.messages];\n\n    this.pendingError = null;\n    const assistantParentId = parentId ?? this.messages.at(-1)?.id ?? null;\n    let assistantMessageId: string | undefined;\n    const ensureAssistant = () => {\n      if (assistantMessageId) return assistantMessageId;\n      const created = this.insertAssistantPlaceholder();\n      assistantMessageId = created;\n      this.markPendingAssistantHistory(created, assistantParentId ?? null);\n      return created;\n    };\n\n    const applyUpdate = (update: ChatModelRunResult) => {\n      const resolved = this.updateAssistantMessage(ensureAssistant(), update);\n      if (resolved !== assistantMessageId) {\n        assistantMessageId = resolved;\n      }\n    };\n\n    const aggregator = new RunAggregator({\n      showThinking: this.showThinking,\n      logger: this.logger,\n      emit: applyUpdate,\n      onServerMessageId: (serverId) => {\n        const placeholder = ensureAssistant();\n        if (placeholder === serverId) return;\n        this.reassignAssistantId(placeholder, serverId);\n        assistantMessageId = serverId;\n      },\n    });\n    const dispatch = (event: AgUiEvent) => this.handleEvent(aggregator, event);\n\n    const abortController = new AbortController();\n    const abortSignal = abortController.signal;\n    this.abortController = abortController;\n\n    let cancelRun = () => dispatch({ type: \"RUN_CANCELLED\" });\n    abortSignal.addEventListener(\n      \"abort\",\n      () => {\n        cancelRun();\n        this.finishRun(abortController);\n        this.onCancel?.();\n      },\n      { once: true },\n    );\n\n    this.setRunning(true);\n\n    try {\n      if (resumeStream) {\n        // Cancel flips only the status; an aggregator RUN_CANCELLED would emit an empty snapshot and wipe the replayed content.\n        cancelRun = () =>\n          applyUpdate({ status: { type: \"incomplete\", reason: \"cancelled\" } });\n        await this.consumeResumeStream(resumeStream, {\n          runConfig: normalizedRunConfig,\n          threadId: this.agent.threadId || \"main\",\n          parentId: assistantParentId,\n          historicalMessages,\n          abortSignal,\n          ensureAssistant,\n          applyUpdate,\n          getAssistantMessageId: () => assistantMessageId,\n        });\n      } else {\n        const runId = generateId();\n        aggregator.handle({ type: \"RUN_STARTED\", runId });\n        const input = this.buildRunInput(\n          runId,\n          normalizedRunConfig,\n          historicalMessages,\n          resume,\n        );\n        const subscriber = createAgUiSubscriber({\n          dispatch,\n          runId,\n          logger: this.logger,\n          onRunFailed: (error) => {\n            this.pendingError = error;\n            this.onError?.(error);\n          },\n        });\n        try {\n          (this.agent as any).messages = input.messages;\n          (this.agent as any).threadId = input.threadId;\n          (this.agent as any).state = input.state ?? null;\n        } catch {\n          // ignore\n        }\n        await (this.agent as any).runAgent(input, subscriber, {\n          signal: abortSignal,\n        });\n      }\n    } catch (error) {\n      if (!abortSignal.aborted) {\n        const err = error instanceof Error ? error : new Error(String(error));\n        dispatch({ type: \"RUN_ERROR\", message: err.message });\n        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      this.pendingResumeMessageId = null;\n      throw err;\n    }\n\n    // A tool result that landed before the run settled deferred its\n    // continuation here so a second run never overlaps the first.\n    if (this.pendingResumeMessageId !== null) {\n      const resumeMessageId = this.pendingResumeMessageId;\n      this.pendingResumeMessageId = null;\n      if (!abortSignal.aborted) {\n        this.startResumeRun(resumeMessageId);\n      }\n    }\n  }\n\n  // Replays a persisted run's snapshots into the existing assistant message, bypassing agent.runAgent so it is not re-invoked.\n  private async consumeResumeStream(\n    stream: ResumeStream,\n    ctx: {\n      runConfig: RunConfig;\n      threadId: string;\n      parentId: string | null;\n      historicalMessages: readonly ThreadMessage[];\n      abortSignal: AbortSignal;\n      ensureAssistant: () => string;\n      applyUpdate: (update: ChatModelRunResult) => void;\n      getAssistantMessageId: () => string | undefined;\n    },\n  ): Promise<void> {\n    const assistantId = ctx.ensureAssistant();\n    const currentId = () => ctx.getAssistantMessageId() ?? assistantId;\n    const options: ChatModelRunOptions = {\n      messages: ctx.historicalMessages,\n      runConfig: ctx.runConfig,\n      abortSignal: ctx.abortSignal,\n      context: this.runtime?.thread.getModelContext() ?? {},\n      unstable_assistantMessageId: assistantId,\n      unstable_threadId: ctx.threadId,\n      unstable_parentId: ctx.parentId,\n      unstable_getMessage: () => {\n        const message = this.messages.find((m) => m.id === currentId());\n        if (!message) {\n          throw new Error(\n            \"[agui] resume stream requested the assistant message before it existed\",\n          );\n        }\n        return message;\n      },\n    };\n\n    try {\n      for await (const result of stream(options)) {\n        if (ctx.abortSignal.aborted) return;\n        ctx.applyUpdate(result);\n      }\n    } catch (error) {\n      if (ctx.abortSignal.aborted) return;\n      const err = error instanceof Error ? error : new Error(String(error));\n      ctx.applyUpdate({\n        status: { type: \"incomplete\", reason: \"error\", error: err.message },\n      });\n      this.onError?.(err);\n      this.pendingError = this.pendingError ?? err;\n      return;\n    }\n\n    if (ctx.abortSignal.aborted) return;\n    const current = this.messages.find((m) => m.id === currentId());\n    if (!current || current.status?.type === \"running\") {\n      ctx.applyUpdate({ status: { type: \"complete\", reason: \"unknown\" } });\n    }\n  }\n\n  private buildRunInput(\n    runId: string,\n    runConfig: RunConfig | undefined,\n    historyMessages: readonly ThreadMessage[] | undefined,\n    resume?: AgUiResumeEntry[],\n  ) {\n    const threadId = this.agent.threadId || \"main\";\n    const messages = toAgUiMessages(historyMessages ?? this.messages);\n    const context = this.runtime?.thread.getModelContext();\n    return {\n      threadId,\n      runId,\n      state: this.stateSnapshot ?? null,\n      messages,\n      tools: toAgUiTools(context?.tools),\n      context: context?.system\n        ? [{ description: \"system\", value: context.system }]\n        : [],\n      forwardedProps: {\n        ...(context?.callSettings ?? {}),\n        ...(context?.config ?? {}),\n        ...(runConfig?.custom ? { runConfig: runConfig.custom } : {}),\n      },\n      ...(resume !== undefined ? { resume } : {}),\n    };\n  }\n\n  private installResumeShim(): void {\n    const agent = this.agent as any;\n    if (agent[symbolResumeShim]) return;\n    agent[symbolResumeShim] = true;\n    const onInstance = Object.hasOwn(agent, \"prepareRunAgentInput\");\n    const original = onInstance\n      ? agent.prepareRunAgentInput\n      : Object.getPrototypeOf(agent)?.prepareRunAgentInput;\n    if (typeof original !== \"function\") return;\n    agent.prepareRunAgentInput = function (\n      this: unknown,\n      params: { resume?: unknown } | undefined,\n    ) {\n      const input = original.call(this, params);\n      if (params?.resume !== undefined && input && typeof input === \"object\") {\n        return { ...(input as object), resume: params.resume };\n      }\n      return input;\n    };\n  }\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) {\n      this.abortController = null;\n    }\n    this.setRunning(false);\n  }\n\n  private insertAssistantPlaceholder(): string {\n    const id = generateOptimisticId();\n    const assistant: ThreadAssistantMessage = {\n      id,\n      role: \"assistant\",\n      createdAt: new Date(),\n      status: { type: \"running\" },\n      content: [],\n      metadata: {\n        unstable_state: this.stateSnapshot ?? null,\n        unstable_annotations: [],\n        unstable_data: [],\n        steps: [],\n        custom: {},\n      },\n    };\n    this.messages = [...this.messages, assistant];\n    this.notifyUpdate();\n    return id;\n  }\n\n  private reassignAssistantId(oldId: string, newId: string): void {\n    if (oldId === newId) return;\n\n    const collidesWithExisting = this.messages.some((m) => m.id === newId);\n\n    if (collidesWithExisting) {\n      this.logger.debug?.(\n        \"[agui] reassignAssistantId: server id already present in messages, dropping placeholder\",\n        { oldId, newId },\n      );\n      this.messages = this.messages.filter((m) => m.id !== oldId);\n    } else {\n      this.messages = this.messages.map((m) =>\n        m.id === oldId ? { ...m, id: newId } : m,\n      );\n    }\n\n    const pendingParent = this.assistantHistoryParents.get(oldId);\n    if (pendingParent !== undefined) {\n      this.assistantHistoryParents.delete(oldId);\n      if (!this.assistantHistoryParents.has(newId)) {\n        this.assistantHistoryParents.set(newId, pendingParent);\n      }\n    }\n\n    if (this.recordedHistoryIds.has(oldId)) {\n      this.recordedHistoryIds.delete(oldId);\n      this.recordedHistoryIds.add(newId);\n    }\n\n    this.notifyUpdate();\n  }\n\n  private updateAssistantMessage(\n    messageId: string,\n    update: ChatModelRunResult,\n  ): string {\n    let touched = false;\n    let latestStatus: MessageStatus | undefined;\n    this.messages = this.messages.map((message) => {\n      if (message.id !== messageId || message.role !== \"assistant\")\n        return message;\n      touched = true;\n      const assistant = message as ThreadAssistantMessage;\n      const metadata = update.metadata\n        ? this.mergeAssistantMetadata(assistant.metadata, update.metadata)\n        : assistant.metadata;\n      latestStatus = update.status ?? assistant.status;\n      const content =\n        update.content !== undefined\n          ? this.preserveToolResults(\n              assistant.content,\n              update.content as ThreadAssistantMessage[\"content\"],\n            )\n          : assistant.content;\n      return {\n        ...assistant,\n        content,\n        status: latestStatus,\n        metadata,\n      };\n    });\n    if (!touched) return messageId;\n\n    let resolvedMessageId = messageId;\n    const isSettled =\n      latestStatus !== undefined && latestStatus.type !== \"running\";\n    if (isSettled && isOptimisticId(messageId)) {\n      const stableId = generateId();\n      this.reassignAssistantId(messageId, stableId);\n      resolvedMessageId = stableId;\n    } else {\n      this.notifyUpdate();\n    }\n    if (this.isPersistableStatus(latestStatus)) {\n      this.persistAssistantHistory(resolvedMessageId);\n    }\n    this.maybeResumeAfterToolResults(resolvedMessageId);\n    return resolvedMessageId;\n  }\n\n  // The RunAggregator rebuilds the assistant content from stream events only,\n  // so a fresh snapshot omits results injected via addToolResult (frontend tool\n  // execution). Carry those results forward so the aggregator never clobbers\n  // them. Results are only ever added in this flow, so preserving is safe.\n  private preserveToolResults(\n    previous: ThreadAssistantMessage[\"content\"],\n    next: ThreadAssistantMessage[\"content\"],\n  ): ThreadAssistantMessage[\"content\"] {\n    const resolved = new Map<string, ToolCallMessagePart>();\n    for (const part of previous) {\n      if (\n        part.type === \"tool-call\" &&\n        \"result\" in part &&\n        part.result !== undefined\n      ) {\n        resolved.set(part.toolCallId, part);\n      }\n    }\n    if (resolved.size === 0) return next;\n\n    let changed = false;\n    const merged = next.map((part) => {\n      if (part.type !== \"tool-call\") return part;\n      if (\"result\" in part && part.result !== undefined) return part;\n      const prior = resolved.get(part.toolCallId);\n      if (!prior) return part;\n      changed = true;\n      return {\n        ...part,\n        result: prior.result,\n        ...(prior.artifact !== undefined ? { artifact: prior.artifact } : {}),\n        ...(prior.isError !== undefined ? { isError: prior.isError } : {}),\n      };\n    });\n    return changed ? (merged as ThreadAssistantMessage[\"content\"]) : next;\n  }\n\n  private mergeAssistantMetadata(\n    current: ThreadAssistantMessage[\"metadata\"],\n    incoming: NonNullable<ChatModelRunResult[\"metadata\"]>,\n  ): ThreadAssistantMessage[\"metadata\"] {\n    const annotations = incoming.unstable_annotations\n      ? [...current.unstable_annotations, ...incoming.unstable_annotations]\n      : current.unstable_annotations;\n    const data = incoming.unstable_data\n      ? [...current.unstable_data, ...incoming.unstable_data]\n      : current.unstable_data;\n    const steps = incoming.steps\n      ? [...current.steps, ...incoming.steps]\n      : current.steps;\n    return {\n      unstable_state:\n        incoming.unstable_state !== undefined\n          ? incoming.unstable_state\n          : current.unstable_state,\n      unstable_annotations: annotations,\n      unstable_data: data,\n      steps,\n      ...(incoming.timing ? { timing: incoming.timing } : {}),\n      custom: incoming.custom\n        ? { ...current.custom, ...incoming.custom }\n        : current.custom,\n    };\n  }\n\n  private handleEvent(aggregator: RunAggregator, event: AgUiEvent) {\n    switch (event.type) {\n      case \"STATE_SNAPSHOT\": {\n        this.stateSnapshot = event.snapshot as ReadonlyJSONValue;\n        this.notifyUpdate();\n        return;\n      }\n      case \"STATE_DELTA\": {\n        if (event.delta.length === 0) return;\n        try {\n          const state = this.stateSnapshot ?? {};\n          const result = jsonpatch.applyPatch(\n            state,\n            event.delta as Operation[],\n            /* validateOperation */ true,\n            /* mutateDocument */ false,\n          );\n          this.stateSnapshot = result.newDocument as ReadonlyJSONValue;\n          this.notifyUpdate();\n        } catch (error) {\n          this.logger.error?.(\"[agui] failed to apply state delta\", error);\n        }\n        return;\n      }\n      case \"MESSAGES_SNAPSHOT\": {\n        this.importMessagesSnapshot(event.messages);\n        return;\n      }\n      default:\n        aggregator.handle(event);\n    }\n  }\n\n  private importMessagesSnapshot(rawMessages: readonly unknown[]) {\n    try {\n      const normalized = fromAgUiMessages(rawMessages);\n      const converted: ThreadMessage[] = [];\n      for (const message of normalized) {\n        try {\n          converted.push(\n            fromThreadMessageLike(\n              message as any,\n              generateId(),\n              FALLBACK_USER_STATUS,\n            ),\n          );\n        } catch (error) {\n          this.logger.error?.(\n            \"[agui] failed to import message from snapshot\",\n            error,\n          );\n        }\n      }\n      this.applyExternalMessages(converted);\n    } catch (error) {\n      this.logger.error?.(\"[agui] failed to import messages snapshot\", error);\n    }\n  }\n\n  private toThreadMessage(message: AppendMessage): ThreadMessage {\n    return fromThreadMessageLike(\n      message as any,\n      generateId(),\n      FALLBACK_USER_STATUS,\n    );\n  }\n\n  private resetHead(parentId: string | null | undefined) {\n    if (!parentId) {\n      if (this.messages.length) {\n        this.messages = [];\n      }\n      return;\n    }\n    const idx = this.messages.findIndex((message) => message.id === parentId);\n    if (idx === -1) return;\n    this.messages = this.messages.slice(0, idx + 1);\n  }\n\n  private isTerminalStatus(status?: MessageStatus): boolean {\n    return status?.type === \"complete\" || status?.type === \"incomplete\";\n  }\n\n  private isPersistableStatus(status?: MessageStatus): boolean {\n    if (this.isTerminalStatus(status)) return true;\n    return status?.type === \"requires-action\" && status.reason === \"interrupt\";\n  }\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.messages.find((m) => m.id === messageId);\n    if (!message || message.role !== \"assistant\") return;\n    if (!this.isPersistableStatus(message.status)) 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((error) => {\n      this.recordedHistoryIds.delete(message.id);\n      this.logger.error?.(\"[agui] failed to append history entry\", error);\n    });\n  }\n}\n"],"mappings":";;;;;;;AAgCA,MAAM,mBAAmB;AACzB,MAAM,6BAA6B,GAAG,mBAAmB,WAAW;AACpE,MAAM,kBAAkB,OAAe,GAAG,WAAW,gBAAgB;AAErE,MAAM,mBAAmB,OAAO,kBAAkB;AAuBlD,MAAM,uBAAuB;CAAE,MAAM;CAAY,QAAQ;AAAU;AAEnE,IAAa,wBAAb,MAAmC;CACjC;CACA;CACA;CACA;CACA;CACA;CAEA;CACA,WAAoC,CAAC;CACrC,gBAAwB;CACxB,kBAAkD;CAClD;CACA,eAAqC;CACrC;CACA;CACA,0CAA2C,IAAI,IAA2B;CAC1E,qCAAsC,IAAI,IAAY;CACtD,aAAqB;CACrB;CACA,yBAAgD;CAEhD,YAAY,SAAsB;EAChC,KAAK,QAAQ,QAAQ;EACrB,KAAK,SAAS,QAAQ;EACtB,KAAK,eAAe,QAAQ;EAC5B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,UAAU,QAAQ;EACvB,KAAK,eAAe,QAAQ;EAC5B,KAAK,kBAAkB;CACzB;CAEA,cAAc,SAA4C;EACxD,KAAK,QAAQ,QAAQ;EACrB,KAAK,SAAS,QAAQ;EACtB,KAAK,eAAe,QAAQ;EAC5B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,UAAU,QAAQ;EACvB,KAAK,kBAAkB;CACzB;CAEA,cAAc,SAA2B;EACvC,KAAK,UAAU;CACjB;CAEA,gBAAgB;EACd,KAAK,UAAU,KAAA;CACjB;CAEA,cAAwC;EACtC,OAAO,KAAK;CACd;CAEA,WAA0C;EACxC,OAAO,KAAK;CACd;CAEA,YAAqB;EACnB,OAAO,KAAK;CACd;CAEA,IAAI,YAAqB;EACvB,OAAO,KAAK;CACd;CAEA,kBAAiC;EAC/B,IAAI,KAAK,cAAc,OAAO,KAAK;EAEnC,MAAM,UAAU,KAAK,SAAS,KAAK,KAAK,QAAQ,QAAQ,IAAI;EAE5D,KAAK,aAAa;EAElB,KAAK,eAAe,QACjB,KAAK,OAAO,SAAS;GACpB,IAAI,CAAC,MAAM;GAEX,MAAM,WAAW,KAAK,SAAS,KAAK,SAAS,KAAK,OAAO;GACzD,KAAK,sBAAsB,QAAQ;GAEnC,IAAI,KAAK,iBAAiB;IACxB,MAAM,WAAW,KAAK,UAAU,SAAS,GAAG,EAAE,GAAG,MAAM;IACvD,MAAM,eAAe,KAAK,SAAS,QAAQ,KAAK,KAAK,OAAO;IAC5D,MAAM,KAAK,SACT,UACA,KAAK,eACL,KAAA,GACA,YACF;GACF;EACF,CAAC,EACA,OAAO,UAAU;GAChB,KAAK,OAAO,QAAQ,iCAAiC,KAAK;GAC1D,KAAK,UACH,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAC1D;EACF,CAAC,EACA,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;EACtD,IAAI,UAAU,KAAK,0BAA0B;EAC7C,IAAI,QAAQ,UACV,KAAK,WAAW,KAAK,SAAS,QAC3B,UAAU,MAAM,OAAO,QAAQ,QAClC;EAEF,KAAK,UAAU,QAAQ,QAAQ;EAE/B,MAAM,gBAAgB,KAAK,gBAAgB,OAAO;EAClD,KAAK,WAAW,CAAC,GAAG,KAAK,UAAU,aAAa;EAChD,KAAK,aAAa;EAClB,KAAK,mBAAmB,QAAQ,YAAY,MAAM,aAAa;EAE/D,IAAI,CAAC,UAAU;EACf,MAAM,KAAK,SAAS,cAAc,IAAI,QAAQ,SAAS;CACzD;CAEA,MAAM,KAAK,SAAuC;EAChD,MAAM,KAAK,OAAO,OAAO;CAC3B;CAEA,MAAM,OACJ,UACA,SAAoC,CAAC,GACtB;EACf,KAAK,0BAA0B;EAC/B,KAAK,UAAU,QAAQ;EACvB,KAAK,aAAa;EAClB,MAAM,KAAK,SAAS,UAAU,OAAO,SAAS;CAChD;CAEA,MAAM,SAAwB;EAC5B,IAAI,CAAC,KAAK,iBAAiB;EAC3B,KAAK,gBAAgB,MAAM;CAC7B;CAEA,MAAM,OAAO,QAAwC;EACnD,KAAK,0BAA0B;EAC/B,MAAM,KAAK,SACT,OAAO,UACP,OAAO,aAAa,KAAK,eACzB,KAAA,GACA,OAAO,MACT;CACF;CAEA,4BAA0C;EACxC,IAAI,CAAC,KAAK,qBAAqB,GAAG;EAClC,MAAM,IAAI,MACR,0GACF;CACF;CAEA,uBAGS;EACP,MAAM,YAAY,KAAK,SAAS,UAAU,MAAM,EAAE,SAAS,WAAW;EAGtE,IACE,CAAC,aACD,UAAU,QAAQ,SAAS,qBAC3B,UAAU,OAAO,WAAW,aAE5B,OAAO;EAET,MAAM,SACJ,UAAU,SAAS,OAAO,2BAGzB;EACH,IAAI,CAAC,QAAQ,QAAQ,OAAO;EAC5B,OAAO;GAAE,WAAW,UAAU;GAAI,YAAY;EAAO;CACvD;CAEA,MAAM,yBACJ,WACe;EACf,MAAM,UAAU,KAAK,qBAAqB;EAC1C,IAAI,CAAC,SACH,MAAM,IAAI,MACR,uEACF;EAGF,MAAM,gCAAgB,IAAI,IAA6B;EACvD,KAAK,MAAM,SAAS,WAAW;GAC7B,IAAI,CAAC,SAAS,OAAO,MAAM,gBAAgB,UACzC,MAAM,IAAI,MACR,uEACF;GAEF,IAAI,MAAM,WAAW,cAAc,MAAM,WAAW,aAClD,MAAM,IAAI,MACR,oDAAoD,MAAM,OAAO,kBAAkB,MAAM,aAC3F;GAEF,IAAI,cAAc,IAAI,MAAM,WAAW,GACrC,MAAM,IAAI,MACR,qEAAqE,MAAM,aAC7E;GAEF,cAAc,IAAI,MAAM,aAAa,KAAK;EAC5C;EAEA,MAAM,UAAU,QAAQ,WAAW,KAAK,MAAM,EAAE,EAAE;EAClD,MAAM,UAAU,QAAQ,QAAQ,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC;EAC7D,IAAI,QAAQ,SAAS,GACnB,MAAM,IAAI,MACR,2EAA2E,QAAQ,KAAK,IAAI,GAC9F;EAEF,MAAM,QAAQ,IAAI,IAAI,OAAO;EAC7B,MAAM,aAAa,CAAC,GAAG,cAAc,KAAK,CAAC,EAAE,QAAQ,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;EAC1E,IAAI,WAAW,SAAS,GACtB,MAAM,IAAI,MACR,2DAA2D,WAAW,KAAK,IAAI,GACjF;EAGF,MAAM,MAAM,KAAK,IAAI;EACrB,KAAK,MAAM,aAAa,QAAQ,YAAY;GAC1C,IAAI,CAAC,UAAU,WAAW;GAC1B,MAAM,SAAS,IAAI,KAAK,UAAU,SAAS,EAAE,QAAQ;GACrD,IAAI,OAAO,MAAM,MAAM,GACrB,MAAM,IAAI,MACR,8CAA8C,UAAU,GAAG,4BAA4B,UAAU,UAAU,EAC7G;GAEF,IAAI,UAAU,KACZ,MAAM,IAAI,MACR,8CAA8C,UAAU,GAAG,cAAc,UAAU,WACrF;EAEJ;EAEA,MAAM,SAA4B,QAAQ,KACvC,OAAO,cAAc,IAAI,EAAE,CAC9B;EAEA,IAAI,KAAK,eACP,MAAM,IAAI,MACR,+DACF;EAGF,KAAK,uBAAuB,QAAQ,SAAS;EAC7C,MAAM,KAAK,SAAS,QAAQ,WAAW,KAAK,eAAe,MAAM;CACnE;CAEA,uBAA+B,WAAyB;EACtD,IAAI,UAAU;EACd,KAAK,WAAW,KAAK,SAAS,KAAK,YAAY;GAC7C,IAAI,QAAQ,OAAO,aAAa,QAAQ,SAAS,aAC/C,OAAO;GACT,MAAM,YAAY;GAClB,IACE,UAAU,QAAQ,SAAS,qBAC3B,UAAU,OAAO,WAAW,aAE5B,OAAO;GAET,UAAU;GAIV,MAAM,EAAE,YAAY,OAAO,GAAG,aAHb,UAAU,SAAS,OAAA,WAGmB,CAAC;GACxD,MAAM,YAAY,EAAE,GAAG,UAAU,SAAS,OAAO;GACjD,IAAI,OAAO,KAAK,QAAQ,EAAE,SAAS,GACjC,UAAU,4BAA4B;QAEtC,OAAO,UAAU;GAEnB,OAAO;IACL,GAAG;IACH,QAAQ;KAAE,MAAM;KAAqB,QAAQ;IAAmB;IAChE,UAAU;KAAE,GAAG,UAAU;KAAU,QAAQ;IAAU;GACvD;EACF,CAAC;EACD,IAAI,SAAS,KAAK,aAAa;CACjC;CAEA,yBAAyB,YAAwC;EAC/D,IAAI;EACJ,KAAK,IAAI,QAAQ,KAAK,SAAS,SAAS,GAAG,SAAS,GAAG,SAAS;GAC9D,MAAM,UAAU,KAAK,SAAS;GAC9B,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;GAC9C,KAAK,MAAM,QAAQ,QAAQ,SAAS;IAClC,IAAI,KAAK,SAAS,eAAe,KAAK,eAAe,YACnD;IACF,IAAI,EAAE,YAAY,SAAS,KAAK,WAAW,KAAA,GACzC,OAAO,QAAQ;IAEjB,sBAAsB,QAAQ;GAChC;EACF;EACA,OAAO;CACT;CAEA,cAAc,SAAqC;EACjD,IAAI,UAAU;EACd,KAAK,WAAW,KAAK,SAAS,KAAK,YAAY;GAC7C,IAAI,QAAQ,OAAO,QAAQ,aAAa,QAAQ,SAAS,aACvD,OAAO;GACT,MAAM,YAAY;GAClB,IAAI,kBAAkB;GACtB,MAAM,UAAU,UAAU,QAAQ,KAAK,SAAS;IAC9C,IAAI,KAAK,SAAS,eAAe,KAAK,eAAe,QAAQ,YAC3D,OAAO;IACT,kBAAkB;IAClB,OAAO;KACL,GAAG;KACH,QAAQ,QAAQ;KAChB,UAAU,QAAQ;KAClB,SAAS,QAAQ;IACnB;GACF,CAAC;GACD,IAAI,CAAC,iBAAiB,OAAO;GAC7B,UAAU;GACV,OAAO;IAAE,GAAG;IAAW;GAAQ;EACjC,CAAC;EAED,IAAI,CAAC,SAAS;EACd,KAAK,aAAa;EAClB,KAAK,4BAA4B,QAAQ,SAAS;CACpD;CAOA,4BAAoC,WAAyB;EAC3D,MAAM,UAAU,KAAK,SAAS,MAAM,MAAM,EAAE,OAAO,SAAS;EAC5D,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;EAC9C,MAAM,YAAY;EAClB,IACE,UAAU,QAAQ,SAAS,qBAC3B,UAAU,OAAO,WAAW,cAE5B;EAOF,IAAI,CALgB,UAAU,QAAQ,OACnC,SACC,KAAK,SAAS,eACb,YAAY,QAAQ,KAAK,WAAW,KAAA,CAE1B,GAAG;EAElB,KAAK,WAAW,KAAK,SAAS,KAAK,MACjC,EAAE,OAAO,aAAa,EAAE,SAAS,cAC7B;GACE,GAAI;GACJ,QAAQ;IAAE,MAAM;IAAqB,QAAQ;GAAmB;EAClE,IACA,CACN;EACA,KAAK,aAAa;EAClB,KAAK,wBAAwB,SAAS;EAEtC,IAAI,KAAK,eAAe;GAGtB,KAAK,yBAAyB;GAC9B;EACF;EACA,KAAK,eAAe,SAAS;CAC/B;CAEA,eAAuB,WAAyB;EAC9C,KAAU,SAAS,WAAW,KAAK,aAAa,EAAE,OAAO,UAAU;GACjE,KAAK,UAAU,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAAC;EAC1E,CAAC;CACH;CAEA,sBAAsB,UAA0C;EAC9D,KAAK,wBAAwB,MAAM;EACnC,KAAK,WAAW,CAAC,GAAG,QAAQ;EAC5B,KAAK,mBAAmB,MAAM;EAC9B,KAAK,MAAM,WAAW,KAAK,UACzB,KAAK,mBAAmB,IAAI,QAAQ,EAAE;EAExC,KAAK,aAAa;CACpB;CAEA,kBAAkB,OAAgC;EAChD,KAAK,gBAAgB;EACrB,KAAK,aAAa;CACpB;CAEA,MAAc,SACZ,UACA,WACA,QACA,cACe;EACf,MAAM,sBAAsB,aAAa,CAAC;EAC1C,KAAK,gBAAgB;EACrB,KAAK,UAAU,QAAQ;EACvB,MAAM,qBAAqB,CAAC,GAAG,KAAK,QAAQ;EAE5C,KAAK,eAAe;EACpB,MAAM,oBAAoB,YAAY,KAAK,SAAS,GAAG,EAAE,GAAG,MAAM;EAClE,IAAI;EACJ,MAAM,wBAAwB;GAC5B,IAAI,oBAAoB,OAAO;GAC/B,MAAM,UAAU,KAAK,2BAA2B;GAChD,qBAAqB;GACrB,KAAK,4BAA4B,SAAS,qBAAqB,IAAI;GACnE,OAAO;EACT;EAEA,MAAM,eAAe,WAA+B;GAClD,MAAM,WAAW,KAAK,uBAAuB,gBAAgB,GAAG,MAAM;GACtE,IAAI,aAAa,oBACf,qBAAqB;EAEzB;EAEA,MAAM,aAAa,IAAI,cAAc;GACnC,cAAc,KAAK;GACnB,QAAQ,KAAK;GACb,MAAM;GACN,oBAAoB,aAAa;IAC/B,MAAM,cAAc,gBAAgB;IACpC,IAAI,gBAAgB,UAAU;IAC9B,KAAK,oBAAoB,aAAa,QAAQ;IAC9C,qBAAqB;GACvB;EACF,CAAC;EACD,MAAM,YAAY,UAAqB,KAAK,YAAY,YAAY,KAAK;EAEzE,MAAM,kBAAkB,IAAI,gBAAgB;EAC5C,MAAM,cAAc,gBAAgB;EACpC,KAAK,kBAAkB;EAEvB,IAAI,kBAAkB,SAAS,EAAE,MAAM,gBAAgB,CAAC;EACxD,YAAY,iBACV,eACM;GACJ,UAAU;GACV,KAAK,UAAU,eAAe;GAC9B,KAAK,WAAW;EAClB,GACA,EAAE,MAAM,KAAK,CACf;EAEA,KAAK,WAAW,IAAI;EAEpB,IAAI;GACF,IAAI,cAAc;IAEhB,kBACE,YAAY,EAAE,QAAQ;KAAE,MAAM;KAAc,QAAQ;IAAY,EAAE,CAAC;IACrE,MAAM,KAAK,oBAAoB,cAAc;KAC3C,WAAW;KACX,UAAU,KAAK,MAAM,YAAY;KACjC,UAAU;KACV;KACA;KACA;KACA;KACA,6BAA6B;IAC/B,CAAC;GACH,OAAO;IACL,MAAM,QAAQ,WAAW;IACzB,WAAW,OAAO;KAAE,MAAM;KAAe;IAAM,CAAC;IAChD,MAAM,QAAQ,KAAK,cACjB,OACA,qBACA,oBACA,MACF;IACA,MAAM,aAAa,qBAAqB;KACtC;KACA;KACA,QAAQ,KAAK;KACb,cAAc,UAAU;MACtB,KAAK,eAAe;MACpB,KAAK,UAAU,KAAK;KACtB;IACF,CAAC;IACD,IAAI;KACF,KAAM,MAAc,WAAW,MAAM;KACrC,KAAM,MAAc,WAAW,MAAM;KACrC,KAAM,MAAc,QAAQ,MAAM,SAAS;IAC7C,QAAQ,CAER;IACA,MAAO,KAAK,MAAc,SAAS,OAAO,YAAY,EACpD,QAAQ,YACV,CAAC;GACH;EACF,SAAS,OAAO;GACd,IAAI,CAAC,YAAY,SAAS;IACxB,MAAM,MAAM,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;IACpE,SAAS;KAAE,MAAM;KAAa,SAAS,IAAI;IAAQ,CAAC;IACpD,KAAK,UAAU,GAAG;IAClB,KAAK,eAAe,KAAK,gBAAgB;GAC3C;EACF,UAAU;GACR,KAAK,UAAU,eAAe;EAChC;EAEA,IAAI,KAAK,cAAc;GACrB,MAAM,MAAM,KAAK;GACjB,KAAK,eAAe;GACpB,KAAK,yBAAyB;GAC9B,MAAM;EACR;EAIA,IAAI,KAAK,2BAA2B,MAAM;GACxC,MAAM,kBAAkB,KAAK;GAC7B,KAAK,yBAAyB;GAC9B,IAAI,CAAC,YAAY,SACf,KAAK,eAAe,eAAe;EAEvC;CACF;CAGA,MAAc,oBACZ,QACA,KAUe;EACf,MAAM,cAAc,IAAI,gBAAgB;EACxC,MAAM,kBAAkB,IAAI,sBAAsB,KAAK;EACvD,MAAM,UAA+B;GACnC,UAAU,IAAI;GACd,WAAW,IAAI;GACf,aAAa,IAAI;GACjB,SAAS,KAAK,SAAS,OAAO,gBAAgB,KAAK,CAAC;GACpD,6BAA6B;GAC7B,mBAAmB,IAAI;GACvB,mBAAmB,IAAI;GACvB,2BAA2B;IACzB,MAAM,UAAU,KAAK,SAAS,MAAM,MAAM,EAAE,OAAO,UAAU,CAAC;IAC9D,IAAI,CAAC,SACH,MAAM,IAAI,MACR,wEACF;IAEF,OAAO;GACT;EACF;EAEA,IAAI;GACF,WAAW,MAAM,UAAU,OAAO,OAAO,GAAG;IAC1C,IAAI,IAAI,YAAY,SAAS;IAC7B,IAAI,YAAY,MAAM;GACxB;EACF,SAAS,OAAO;GACd,IAAI,IAAI,YAAY,SAAS;GAC7B,MAAM,MAAM,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;GACpE,IAAI,YAAY,EACd,QAAQ;IAAE,MAAM;IAAc,QAAQ;IAAS,OAAO,IAAI;GAAQ,EACpE,CAAC;GACD,KAAK,UAAU,GAAG;GAClB,KAAK,eAAe,KAAK,gBAAgB;GACzC;EACF;EAEA,IAAI,IAAI,YAAY,SAAS;EAC7B,MAAM,UAAU,KAAK,SAAS,MAAM,MAAM,EAAE,OAAO,UAAU,CAAC;EAC9D,IAAI,CAAC,WAAW,QAAQ,QAAQ,SAAS,WACvC,IAAI,YAAY,EAAE,QAAQ;GAAE,MAAM;GAAY,QAAQ;EAAU,EAAE,CAAC;CAEvE;CAEA,cACE,OACA,WACA,iBACA,QACA;EACA,MAAM,WAAW,KAAK,MAAM,YAAY;EACxC,MAAM,WAAW,eAAe,mBAAmB,KAAK,QAAQ;EAChE,MAAM,UAAU,KAAK,SAAS,OAAO,gBAAgB;EACrD,OAAO;GACL;GACA;GACA,OAAO,KAAK,iBAAiB;GAC7B;GACA,OAAO,YAAY,SAAS,KAAK;GACjC,SAAS,SAAS,SACd,CAAC;IAAE,aAAa;IAAU,OAAO,QAAQ;GAAO,CAAC,IACjD,CAAC;GACL,gBAAgB;IACd,GAAI,SAAS,gBAAgB,CAAC;IAC9B,GAAI,SAAS,UAAU,CAAC;IACxB,GAAI,WAAW,SAAS,EAAE,WAAW,UAAU,OAAO,IAAI,CAAC;GAC7D;GACA,GAAI,WAAW,KAAA,IAAY,EAAE,OAAO,IAAI,CAAC;EAC3C;CACF;CAEA,oBAAkC;EAChC,MAAM,QAAQ,KAAK;EACnB,IAAI,MAAM,mBAAmB;EAC7B,MAAM,oBAAoB;EAE1B,MAAM,WADa,OAAO,OAAO,OAAO,sBACd,IACtB,MAAM,uBACN,OAAO,eAAe,KAAK,GAAG;EAClC,IAAI,OAAO,aAAa,YAAY;EACpC,MAAM,uBAAuB,SAE3B,QACA;GACA,MAAM,QAAQ,SAAS,KAAK,MAAM,MAAM;GACxC,IAAI,QAAQ,WAAW,KAAA,KAAa,SAAS,OAAO,UAAU,UAC5D,OAAO;IAAE,GAAI;IAAkB,QAAQ,OAAO;GAAO;GAEvD,OAAO;EACT;CACF;CAEA,WAAmB,SAAkB;EACnC,KAAK,gBAAgB;EACrB,KAAK,aAAa;CACpB;CAEA,UAAkB,YAAoC;EACpD,IAAI,KAAK,oBAAoB,YAC3B,KAAK,kBAAkB;EAEzB,KAAK,WAAW,KAAK;CACvB;CAEA,6BAA6C;EAC3C,MAAM,KAAK,qBAAqB;EAChC,MAAM,YAAoC;GACxC;GACA,MAAM;GACN,2BAAW,IAAI,KAAK;GACpB,QAAQ,EAAE,MAAM,UAAU;GAC1B,SAAS,CAAC;GACV,UAAU;IACR,gBAAgB,KAAK,iBAAiB;IACtC,sBAAsB,CAAC;IACvB,eAAe,CAAC;IAChB,OAAO,CAAC;IACR,QAAQ,CAAC;GACX;EACF;EACA,KAAK,WAAW,CAAC,GAAG,KAAK,UAAU,SAAS;EAC5C,KAAK,aAAa;EAClB,OAAO;CACT;CAEA,oBAA4B,OAAe,OAAqB;EAC9D,IAAI,UAAU,OAAO;EAIrB,IAF6B,KAAK,SAAS,MAAM,MAAM,EAAE,OAAO,KAEzC,GAAG;GACxB,KAAK,OAAO,QACV,2FACA;IAAE;IAAO;GAAM,CACjB;GACA,KAAK,WAAW,KAAK,SAAS,QAAQ,MAAM,EAAE,OAAO,KAAK;EAC5D,OACE,KAAK,WAAW,KAAK,SAAS,KAAK,MACjC,EAAE,OAAO,QAAQ;GAAE,GAAG;GAAG,IAAI;EAAM,IAAI,CACzC;EAGF,MAAM,gBAAgB,KAAK,wBAAwB,IAAI,KAAK;EAC5D,IAAI,kBAAkB,KAAA,GAAW;GAC/B,KAAK,wBAAwB,OAAO,KAAK;GACzC,IAAI,CAAC,KAAK,wBAAwB,IAAI,KAAK,GACzC,KAAK,wBAAwB,IAAI,OAAO,aAAa;EAEzD;EAEA,IAAI,KAAK,mBAAmB,IAAI,KAAK,GAAG;GACtC,KAAK,mBAAmB,OAAO,KAAK;GACpC,KAAK,mBAAmB,IAAI,KAAK;EACnC;EAEA,KAAK,aAAa;CACpB;CAEA,uBACE,WACA,QACQ;EACR,IAAI,UAAU;EACd,IAAI;EACJ,KAAK,WAAW,KAAK,SAAS,KAAK,YAAY;GAC7C,IAAI,QAAQ,OAAO,aAAa,QAAQ,SAAS,aAC/C,OAAO;GACT,UAAU;GACV,MAAM,YAAY;GAClB,MAAM,WAAW,OAAO,WACpB,KAAK,uBAAuB,UAAU,UAAU,OAAO,QAAQ,IAC/D,UAAU;GACd,eAAe,OAAO,UAAU,UAAU;GAC1C,MAAM,UACJ,OAAO,YAAY,KAAA,IACf,KAAK,oBACH,UAAU,SACV,OAAO,OACT,IACA,UAAU;GAChB,OAAO;IACL,GAAG;IACH;IACA,QAAQ;IACR;GACF;EACF,CAAC;EACD,IAAI,CAAC,SAAS,OAAO;EAErB,IAAI,oBAAoB;EAGxB,IADE,iBAAiB,KAAA,KAAa,aAAa,SAAS,aACrC,eAAe,SAAS,GAAG;GAC1C,MAAM,WAAW,WAAW;GAC5B,KAAK,oBAAoB,WAAW,QAAQ;GAC5C,oBAAoB;EACtB,OACE,KAAK,aAAa;EAEpB,IAAI,KAAK,oBAAoB,YAAY,GACvC,KAAK,wBAAwB,iBAAiB;EAEhD,KAAK,4BAA4B,iBAAiB;EAClD,OAAO;CACT;CAMA,oBACE,UACA,MACmC;EACnC,MAAM,2BAAW,IAAI,IAAiC;EACtD,KAAK,MAAM,QAAQ,UACjB,IACE,KAAK,SAAS,eACd,YAAY,QACZ,KAAK,WAAW,KAAA,GAEhB,SAAS,IAAI,KAAK,YAAY,IAAI;EAGtC,IAAI,SAAS,SAAS,GAAG,OAAO;EAEhC,IAAI,UAAU;EACd,MAAM,SAAS,KAAK,KAAK,SAAS;GAChC,IAAI,KAAK,SAAS,aAAa,OAAO;GACtC,IAAI,YAAY,QAAQ,KAAK,WAAW,KAAA,GAAW,OAAO;GAC1D,MAAM,QAAQ,SAAS,IAAI,KAAK,UAAU;GAC1C,IAAI,CAAC,OAAO,OAAO;GACnB,UAAU;GACV,OAAO;IACL,GAAG;IACH,QAAQ,MAAM;IACd,GAAI,MAAM,aAAa,KAAA,IAAY,EAAE,UAAU,MAAM,SAAS,IAAI,CAAC;IACnE,GAAI,MAAM,YAAY,KAAA,IAAY,EAAE,SAAS,MAAM,QAAQ,IAAI,CAAC;GAClE;EACF,CAAC;EACD,OAAO,UAAW,SAA+C;CACnE;CAEA,uBACE,SACA,UACoC;EACpC,MAAM,cAAc,SAAS,uBACzB,CAAC,GAAG,QAAQ,sBAAsB,GAAG,SAAS,oBAAoB,IAClE,QAAQ;EACZ,MAAM,OAAO,SAAS,gBAClB,CAAC,GAAG,QAAQ,eAAe,GAAG,SAAS,aAAa,IACpD,QAAQ;EACZ,MAAM,QAAQ,SAAS,QACnB,CAAC,GAAG,QAAQ,OAAO,GAAG,SAAS,KAAK,IACpC,QAAQ;EACZ,OAAO;GACL,gBACE,SAAS,mBAAmB,KAAA,IACxB,SAAS,iBACT,QAAQ;GACd,sBAAsB;GACtB,eAAe;GACf;GACA,GAAI,SAAS,SAAS,EAAE,QAAQ,SAAS,OAAO,IAAI,CAAC;GACrD,QAAQ,SAAS,SACb;IAAE,GAAG,QAAQ;IAAQ,GAAG,SAAS;GAAO,IACxC,QAAQ;EACd;CACF;CAEA,YAAoB,YAA2B,OAAkB;EAC/D,QAAQ,MAAM,MAAd;GACE,KAAK;IACH,KAAK,gBAAgB,MAAM;IAC3B,KAAK,aAAa;IAClB;GAEF,KAAK;IACH,IAAI,MAAM,MAAM,WAAW,GAAG;IAC9B,IAAI;KACF,MAAM,QAAQ,KAAK,iBAAiB,CAAC;KACrC,MAAM,SAAS,UAAU,WACvB,OACA,MAAM,OACkB,MACH,KACvB;KACA,KAAK,gBAAgB,OAAO;KAC5B,KAAK,aAAa;IACpB,SAAS,OAAO;KACd,KAAK,OAAO,QAAQ,sCAAsC,KAAK;IACjE;IACA;GAEF,KAAK;IACH,KAAK,uBAAuB,MAAM,QAAQ;IAC1C;GAEF,SACE,WAAW,OAAO,KAAK;EAC3B;CACF;CAEA,uBAA+B,aAAiC;EAC9D,IAAI;GACF,MAAM,aAAa,iBAAiB,WAAW;GAC/C,MAAM,YAA6B,CAAC;GACpC,KAAK,MAAM,WAAW,YACpB,IAAI;IACF,UAAU,KACR,sBACE,SACA,WAAW,GACX,oBACF,CACF;GACF,SAAS,OAAO;IACd,KAAK,OAAO,QACV,iDACA,KACF;GACF;GAEF,KAAK,sBAAsB,SAAS;EACtC,SAAS,OAAO;GACd,KAAK,OAAO,QAAQ,6CAA6C,KAAK;EACxE;CACF;CAEA,gBAAwB,SAAuC;EAC7D,OAAO,sBACL,SACA,WAAW,GACX,oBACF;CACF;CAEA,UAAkB,UAAqC;EACrD,IAAI,CAAC,UAAU;GACb,IAAI,KAAK,SAAS,QAChB,KAAK,WAAW,CAAC;GAEnB;EACF;EACA,MAAM,MAAM,KAAK,SAAS,WAAW,YAAY,QAAQ,OAAO,QAAQ;EACxE,IAAI,QAAQ,IAAI;EAChB,KAAK,WAAW,KAAK,SAAS,MAAM,GAAG,MAAM,CAAC;CAChD;CAEA,iBAAyB,QAAiC;EACxD,OAAO,QAAQ,SAAS,cAAc,QAAQ,SAAS;CACzD;CAEA,oBAA4B,QAAiC;EAC3D,IAAI,KAAK,iBAAiB,MAAM,GAAG,OAAO;EAC1C,OAAO,QAAQ,SAAS,qBAAqB,OAAO,WAAW;CACjE;CAEA,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,SAAS,MAAM,MAAM,EAAE,OAAO,SAAS;EAC5D,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;EAC9C,IAAI,CAAC,KAAK,oBAAoB,QAAQ,MAAM,GAAG;EAC/C,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,EAAE,OAAO,UAAU;GAC/D,KAAK,mBAAmB,OAAO,QAAQ,EAAE;GACzC,KAAK,OAAO,QAAQ,yCAAyC,KAAK;EACpE,CAAC;CACH;AACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ag-ui",
-  "version": "0.0.34",
+  "version": "0.0.35",
   "description": "AG-UI protocol adapter for assistant-ui",
   "keywords": [
     "ag-ui",
@@ -30,8 +30,9 @@
   "sideEffects": false,
   "dependencies": {
     "@ag-ui/client": "^0.0.54",
-    "@assistant-ui/core": "^0.2.8",
-    "assistant-stream": "^0.3.18"
+    "@assistant-ui/core": "^0.2.10",
+    "assistant-stream": "^0.3.20",
+    "fast-json-patch": "^3.1.1"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -47,7 +48,7 @@
     "react": "^19.2.6",
     "vitest": "^4.1.7",
     "zod": "^4.4.3",
-    "@assistant-ui/x-buildutils": "0.0.10"
+    "@assistant-ui/x-buildutils": "0.0.11"
   },
   "publishConfig": {
     "access": "public",

package/src/runtime/AgUiThreadRuntimeCore.ts

@@ -5,13 +5,16 @@ import type {
   AddToolResultOptions,
   AppendMessage,
   AssistantRuntime,
+  ChatModelRunOptions,
   ChatModelRunResult,
   MessageStatus,
   ThreadAssistantMessage,
   ThreadHistoryAdapter,
   ThreadMessage,
+  ToolCallMessagePart,
 } from "@assistant-ui/core";
 import type { HttpAgent } from "@ag-ui/client";
+import jsonpatch, { type Operation } from "fast-json-patch";
 import type { Logger } from "./logger";
 import type { AgUiEvent, AgUiInterrupt, AgUiResumeEntry } from "./types";
 import type { ReadonlyJSONValue } from "assistant-stream/utils";
@@ -34,11 +37,14 @@ const isOptimisticId = (id: string) => id.startsWith(optimisticPrefix);
 const symbolResumeShim = Symbol("agui-resume-shim");
 
 type RunConfig = NonNullable<AppendMessage["runConfig"]>;
+type ResumeStream = (
+  options: ChatModelRunOptions,
+) => AsyncGenerator<ChatModelRunResult, void, unknown>;
 type ResumeRunConfig = {
   parentId: string | null;
   sourceId: string | null;
   runConfig: RunConfig;
-  stream?: unknown;
+  stream?: ResumeStream;
 };
 
 type CoreOptions = {
@@ -73,6 +79,7 @@ export class AgUiThreadRuntimeCore {
   private readonly recordedHistoryIds = new Set<string>();
   private _isLoading = false;
   private _loadPromise: Promise<void> | undefined;
+  private pendingResumeMessageId: string | null = null;
 
   constructor(options: CoreOptions) {
     this.agent = options.agent;
@@ -135,7 +142,13 @@ export class AgUiThreadRuntimeCore {
 
         if (repo.unstable_resume) {
           const parentId = repo.headId ?? messages.at(-1)?.id ?? null;
-          await this.startRun(parentId, this.lastRunConfig);
+          const resumeStream = this.history?.resume?.bind(this.history);
+          await this.startRun(
+            parentId,
+            this.lastRunConfig,
+            undefined,
+            resumeStream,
+          );
         }
       })
       .catch((error) => {
@@ -193,14 +206,11 @@ export class AgUiThreadRuntimeCore {
 
   async resume(config: ResumeRunConfig): Promise<void> {
     this.assertNoPendingInterrupts();
-    if (config.stream) {
-      this.logger.debug?.(
-        "[agui] resume stream is not supported, falling back to regular run",
-      );
-    }
     await this.startRun(
       config.parentId,
       config.runConfig ?? this.lastRunConfig,
+      undefined,
+      config.stream,
     );
   }
 
@@ -360,7 +370,6 @@ export class AgUiThreadRuntimeCore {
 
   addToolResult(options: AddToolResultOptions): void {
     let updated = false;
-    let shouldResume = false;
     this.messages = this.messages.map((message) => {
       if (message.id !== options.messageId || message.role !== "assistant")
         return message;
@@ -379,47 +388,60 @@ export class AgUiThreadRuntimeCore {
       });
       if (!matchedToolCall) return message;
       updated = true;
-
-      if (
-        assistant.status?.type === "requires-action" &&
-        assistant.status.reason === "tool-calls" &&
-        content.every(
-          (part) =>
-            part.type !== "tool-call" ||
-            ("result" in part && part.result !== undefined),
-        )
-      ) {
-        shouldResume = true;
-        return {
-          ...assistant,
-          content,
-          status: { type: "complete" as const, reason: "unknown" as const },
-        };
-      }
-
-      return {
-        ...assistant,
-        content,
-      };
+      return { ...assistant, content };
     });
 
-    if (updated) {
-      this.notifyUpdate();
+    if (!updated) return;
+    this.notifyUpdate();
+    this.maybeResumeAfterToolResults(options.messageId);
+  }
 
-      if (shouldResume) {
-        this.persistAssistantHistory(options.messageId);
+  // Re-evaluate whether every tool call on the assistant message now has a
+  // result and, if so, drive the follow-up run. Centralizing this lets the
+  // continuation fire whether the frontend result lands before RUN_FINISHED
+  // (status flips to requires-action only later, while a run is still draining)
+  // or after it. Idempotent: once the message is marked complete it no-ops.
+  private maybeResumeAfterToolResults(messageId: string): void {
+    const message = this.messages.find((m) => m.id === messageId);
+    if (!message || message.role !== "assistant") return;
+    const assistant = message as ThreadAssistantMessage;
+    if (
+      assistant.status?.type !== "requires-action" ||
+      assistant.status.reason !== "tool-calls"
+    ) {
+      return;
+    }
+    const allResolved = assistant.content.every(
+      (part) =>
+        part.type !== "tool-call" ||
+        ("result" in part && part.result !== undefined),
+    );
+    if (!allResolved) return;
+
+    this.messages = this.messages.map((m) =>
+      m.id === messageId && m.role === "assistant"
+        ? {
+            ...(m as ThreadAssistantMessage),
+            status: { type: "complete" as const, reason: "unknown" as const },
+          }
+        : m,
+    );
+    this.notifyUpdate();
+    this.persistAssistantHistory(messageId);
 
-        if (!this.isRunningFlag) {
-          void this.startRun(options.messageId, this.lastRunConfig).catch(
-            (error) => {
-              this.onError?.(
-                error instanceof Error ? error : new Error(String(error)),
-              );
-            },
-          );
-        }
-      }
+    if (this.isRunningFlag) {
+      // A run is still draining (RUN_FINISHED arrived but the stream has not
+      // closed). Defer until startRun's tail so we never start two runs.
+      this.pendingResumeMessageId = messageId;
+      return;
     }
+    this.startResumeRun(messageId);
+  }
+
+  private startResumeRun(messageId: string): void {
+    void this.startRun(messageId, this.lastRunConfig).catch((error) => {
+      this.onError?.(error instanceof Error ? error : new Error(String(error)));
+    });
   }
 
   applyExternalMessages(messages: readonly ThreadMessage[]): void {
@@ -441,20 +463,14 @@ export class AgUiThreadRuntimeCore {
     parentId: string | null,
     runConfig?: RunConfig,
     resume?: AgUiResumeEntry[],
+    resumeStream?: ResumeStream,
   ): Promise<void> {
     const normalizedRunConfig = runConfig ?? {};
     this.lastRunConfig = normalizedRunConfig;
     this.resetHead(parentId);
     const historicalMessages = [...this.messages];
 
-    const runId = generateId();
     this.pendingError = null;
-    const input = this.buildRunInput(
-      runId,
-      normalizedRunConfig,
-      historicalMessages,
-      resume,
-    );
     const assistantParentId = parentId ?? this.messages.at(-1)?.id ?? null;
     let assistantMessageId: string | undefined;
     const ensureAssistant = () => {
@@ -465,15 +481,17 @@ export class AgUiThreadRuntimeCore {
       return created;
     };
 
+    const applyUpdate = (update: ChatModelRunResult) => {
+      const resolved = this.updateAssistantMessage(ensureAssistant(), update);
+      if (resolved !== assistantMessageId) {
+        assistantMessageId = resolved;
+      }
+    };
+
     const aggregator = new RunAggregator({
       showThinking: this.showThinking,
       logger: this.logger,
-      emit: (update) => {
-        const resolved = this.updateAssistantMessage(ensureAssistant(), update);
-        if (resolved !== assistantMessageId) {
-          assistantMessageId = resolved;
-        }
-      },
+      emit: applyUpdate,
       onServerMessageId: (serverId) => {
         const placeholder = ensureAssistant();
         if (placeholder === serverId) return;
@@ -487,40 +505,63 @@ export class AgUiThreadRuntimeCore {
     const abortSignal = abortController.signal;
     this.abortController = abortController;
 
+    let cancelRun = () => dispatch({ type: "RUN_CANCELLED" });
     abortSignal.addEventListener(
       "abort",
       () => {
-        dispatch({ type: "RUN_CANCELLED" });
+        cancelRun();
         this.finishRun(abortController);
         this.onCancel?.();
       },
       { once: true },
     );
 
-    const subscriber = createAgUiSubscriber({
-      dispatch,
-      runId,
-      logger: this.logger,
-      onRunFailed: (error) => {
-        this.pendingError = error;
-        this.onError?.(error);
-      },
-    });
-
-    aggregator.handle({ type: "RUN_STARTED", runId });
     this.setRunning(true);
 
     try {
-      try {
-        (this.agent as any).messages = input.messages;
-        (this.agent as any).threadId = input.threadId;
-        (this.agent as any).state = input.state ?? null;
-      } catch {
-        // ignore
+      if (resumeStream) {
+        // Cancel flips only the status; an aggregator RUN_CANCELLED would emit an empty snapshot and wipe the replayed content.
+        cancelRun = () =>
+          applyUpdate({ status: { type: "incomplete", reason: "cancelled" } });
+        await this.consumeResumeStream(resumeStream, {
+          runConfig: normalizedRunConfig,
+          threadId: this.agent.threadId || "main",
+          parentId: assistantParentId,
+          historicalMessages,
+          abortSignal,
+          ensureAssistant,
+          applyUpdate,
+          getAssistantMessageId: () => assistantMessageId,
+        });
+      } else {
+        const runId = generateId();
+        aggregator.handle({ type: "RUN_STARTED", runId });
+        const input = this.buildRunInput(
+          runId,
+          normalizedRunConfig,
+          historicalMessages,
+          resume,
+        );
+        const subscriber = createAgUiSubscriber({
+          dispatch,
+          runId,
+          logger: this.logger,
+          onRunFailed: (error) => {
+            this.pendingError = error;
+            this.onError?.(error);
+          },
+        });
+        try {
+          (this.agent as any).messages = input.messages;
+          (this.agent as any).threadId = input.threadId;
+          (this.agent as any).state = input.state ?? null;
+        } catch {
+          // ignore
+        }
+        await (this.agent as any).runAgent(input, subscriber, {
+          signal: abortSignal,
+        });
       }
-      await (this.agent as any).runAgent(input, subscriber, {
-        signal: abortSignal,
-      });
     } catch (error) {
       if (!abortSignal.aborted) {
         const err = error instanceof Error ? error : new Error(String(error));
@@ -535,8 +576,77 @@ export class AgUiThreadRuntimeCore {
     if (this.pendingError) {
       const err = this.pendingError;
       this.pendingError = null;
+      this.pendingResumeMessageId = null;
       throw err;
     }
+
+    // A tool result that landed before the run settled deferred its
+    // continuation here so a second run never overlaps the first.
+    if (this.pendingResumeMessageId !== null) {
+      const resumeMessageId = this.pendingResumeMessageId;
+      this.pendingResumeMessageId = null;
+      if (!abortSignal.aborted) {
+        this.startResumeRun(resumeMessageId);
+      }
+    }
+  }
+
+  // Replays a persisted run's snapshots into the existing assistant message, bypassing agent.runAgent so it is not re-invoked.
+  private async consumeResumeStream(
+    stream: ResumeStream,
+    ctx: {
+      runConfig: RunConfig;
+      threadId: string;
+      parentId: string | null;
+      historicalMessages: readonly ThreadMessage[];
+      abortSignal: AbortSignal;
+      ensureAssistant: () => string;
+      applyUpdate: (update: ChatModelRunResult) => void;
+      getAssistantMessageId: () => string | undefined;
+    },
+  ): Promise<void> {
+    const assistantId = ctx.ensureAssistant();
+    const currentId = () => ctx.getAssistantMessageId() ?? assistantId;
+    const options: ChatModelRunOptions = {
+      messages: ctx.historicalMessages,
+      runConfig: ctx.runConfig,
+      abortSignal: ctx.abortSignal,
+      context: this.runtime?.thread.getModelContext() ?? {},
+      unstable_assistantMessageId: assistantId,
+      unstable_threadId: ctx.threadId,
+      unstable_parentId: ctx.parentId,
+      unstable_getMessage: () => {
+        const message = this.messages.find((m) => m.id === currentId());
+        if (!message) {
+          throw new Error(
+            "[agui] resume stream requested the assistant message before it existed",
+          );
+        }
+        return message;
+      },
+    };
+
+    try {
+      for await (const result of stream(options)) {
+        if (ctx.abortSignal.aborted) return;
+        ctx.applyUpdate(result);
+      }
+    } catch (error) {
+      if (ctx.abortSignal.aborted) return;
+      const err = error instanceof Error ? error : new Error(String(error));
+      ctx.applyUpdate({
+        status: { type: "incomplete", reason: "error", error: err.message },
+      });
+      this.onError?.(err);
+      this.pendingError = this.pendingError ?? err;
+      return;
+    }
+
+    if (ctx.abortSignal.aborted) return;
+    const current = this.messages.find((m) => m.id === currentId());
+    if (!current || current.status?.type === "running") {
+      ctx.applyUpdate({ status: { type: "complete", reason: "unknown" } });
+    }
   }
 
   private buildRunInput(
@@ -668,10 +778,16 @@ export class AgUiThreadRuntimeCore {
         ? this.mergeAssistantMetadata(assistant.metadata, update.metadata)
         : assistant.metadata;
       latestStatus = update.status ?? assistant.status;
+      const content =
+        update.content !== undefined
+          ? this.preserveToolResults(
+              assistant.content,
+              update.content as ThreadAssistantMessage["content"],
+            )
+          : assistant.content;
       return {
         ...assistant,
-        content: (update.content ??
-          assistant.content) as ThreadAssistantMessage["content"],
+        content,
         status: latestStatus,
         metadata,
       };
@@ -691,9 +807,47 @@ export class AgUiThreadRuntimeCore {
     if (this.isPersistableStatus(latestStatus)) {
       this.persistAssistantHistory(resolvedMessageId);
     }
+    this.maybeResumeAfterToolResults(resolvedMessageId);
     return resolvedMessageId;
   }
 
+  // The RunAggregator rebuilds the assistant content from stream events only,
+  // so a fresh snapshot omits results injected via addToolResult (frontend tool
+  // execution). Carry those results forward so the aggregator never clobbers
+  // them. Results are only ever added in this flow, so preserving is safe.
+  private preserveToolResults(
+    previous: ThreadAssistantMessage["content"],
+    next: ThreadAssistantMessage["content"],
+  ): ThreadAssistantMessage["content"] {
+    const resolved = new Map<string, ToolCallMessagePart>();
+    for (const part of previous) {
+      if (
+        part.type === "tool-call" &&
+        "result" in part &&
+        part.result !== undefined
+      ) {
+        resolved.set(part.toolCallId, part);
+      }
+    }
+    if (resolved.size === 0) return next;
+
+    let changed = false;
+    const merged = next.map((part) => {
+      if (part.type !== "tool-call") return part;
+      if ("result" in part && part.result !== undefined) return part;
+      const prior = resolved.get(part.toolCallId);
+      if (!prior) return part;
+      changed = true;
+      return {
+        ...part,
+        result: prior.result,
+        ...(prior.artifact !== undefined ? { artifact: prior.artifact } : {}),
+        ...(prior.isError !== undefined ? { isError: prior.isError } : {}),
+      };
+    });
+    return changed ? (merged as ThreadAssistantMessage["content"]) : next;
+  }
+
   private mergeAssistantMetadata(
     current: ThreadAssistantMessage["metadata"],
     incoming: NonNullable<ChatModelRunResult["metadata"]>,
@@ -730,7 +884,20 @@ export class AgUiThreadRuntimeCore {
         return;
       }
       case "STATE_DELTA": {
-        this.logger.debug?.("[agui] state delta event ignored", event.delta);
+        if (event.delta.length === 0) return;
+        try {
+          const state = this.stateSnapshot ?? {};
+          const result = jsonpatch.applyPatch(
+            state,
+            event.delta as Operation[],
+            /* validateOperation */ true,
+            /* mutateDocument */ false,
+          );
+          this.stateSnapshot = result.newDocument as ReadonlyJSONValue;
+          this.notifyUpdate();
+        } catch (error) {
+          this.logger.error?.("[agui] failed to apply state delta", error);
+        }
         return;
       }
       case "MESSAGES_SNAPSHOT": {