@assistant-ui/react-ag-ui 0.0.38 → 0.0.39

package/dist/runtime/AgUiThreadRuntimeCore.d.ts

@@ -1,7 +1,7 @@
 import { Logger } from "./logger.js";
 import { AgUiInterrupt, AgUiResumeEntry } from "./types.js";
 import { AddToolResultOptions, AppendMessage, AssistantRuntime, ChatModelRunOptions, ChatModelRunResult, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
-import { HttpAgent } from "@ag-ui/client";
+import { AbstractAgent } from "@ag-ui/client";
 import { ReadonlyJSONValue } from "assistant-stream/utils";
 
 //#region src/runtime/AgUiThreadRuntimeCore.d.ts
@@ -14,7 +14,7 @@ type ResumeRunConfig = {
   stream?: ResumeStream;
 };
 type CoreOptions = {
-  agent: HttpAgent;
+  agent: AbstractAgent;
   logger: Logger;
   showThinking: boolean;
   onError?: (error: Error) => void;
@@ -58,6 +58,7 @@ declare class AgUiThreadRuntimeCore {
   }): Promise<void>;
   cancel(): Promise<void>;
   resume(config: ResumeRunConfig): Promise<void>;
+  resumeInFlightRun(messages: readonly ThreadMessage[]): Promise<void>;
   private assertNoPendingInterrupts;
   getPendingInterrupts(): {
     messageId: string;
@@ -68,6 +69,7 @@ declare class AgUiThreadRuntimeCore {
   findMessageIdForToolCall(toolCallId: string): string | undefined;
   addToolResult(options: AddToolResultOptions): void;
   private maybeResumeAfterToolResults;
+  private maybeCompleteAfterToolResults;
   private startResumeRun;
   applyExternalMessages(messages: readonly ThreadMessage[]): void;
   loadExternalState(state: ReadonlyJSONValue): void;
@@ -83,6 +85,7 @@ declare class AgUiThreadRuntimeCore {
   private preserveToolResults;
   private mergeAssistantMetadata;
   private handleEvent;
+  private applyCrossRunToolResult;
   private importMessagesSnapshot;
   private toThreadMessage;
   private resetHead;

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

@@ -1 +1 @@
-{"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,QAsBA,mBAAA;EAAA,QAmCA,sBAAA;EAAA,QAoDA,mBAAA;EAAA,QAiCA,sBAAA;EAAA,QA6BA,WAAA;EAAA,QAiCA,sBAAA;EAAA,QAwBA,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":";;;;;;;KAuCK,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,aAAA;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;EAUjC,iBAAA,CAAkB,QAAA,WAAmB,aAAA,KAAkB,OAAA;EAAA,QAwBrD,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,QA+Bf,2BAAA;EAAA,QAYA,6BAAA;EAAA,QA8BA,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,QAsBA,mBAAA;EAAA,QAmCA,sBAAA;EAAA,QAoDA,mBAAA;EAAA,QAiCA,sBAAA;EAAA,QA6BA,WAAA;EAAA,QA4CA,uBAAA;EAAA,QAmCA,sBAAA;EAAA,QAwBA,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

@@ -1,5 +1,5 @@
 "use client";
-import { AG_UI_METADATA_NAMESPACE, RunAggregator } from "./adapter/run-aggregator.js";
+import { AG_UI_METADATA_NAMESPACE, RunAggregator, tryParseJSON } from "./adapter/run-aggregator.js";
 import { fromAgUiMessages, toAgUiMessages, toAgUiTools } from "./adapter/conversions.js";
 import { createAgUiSubscriber } from "./adapter/subscriber.js";
 import { fromThreadMessageLike, generateId } from "@assistant-ui/core/internal";
@@ -122,6 +122,19 @@ var AgUiThreadRuntimeCore = class {
 		this.assertNoPendingInterrupts();
 		await this.startRun(config.parentId, config.runConfig ?? this.lastRunConfig, void 0, config.stream);
 	}
+	async resumeInFlightRun(messages) {
+		const resumeStream = this.history?.resume?.bind(this.history);
+		if (!resumeStream) {
+			const error = /* @__PURE__ */ new Error("[agui] unstable_resume requires a ThreadHistoryAdapter with a resume() method; skipping resume after thread switch");
+			this.logger.error?.(error.message);
+			this.onError?.(error);
+			return;
+		}
+		const parentId = messages.at(-1)?.id ?? null;
+		try {
+			await this.startRun(parentId, this.lastRunConfig, void 0, resumeStream);
+		} catch {}
+	}
 	assertNoPendingInterrupts() {
 		if (!this.getPendingInterrupts()) return;
 		throw new Error("[agui] cannot start a new run while interrupts are pending; resolve them with submitInterruptResponses()");
@@ -230,11 +243,19 @@ var AgUiThreadRuntimeCore = class {
 		this.maybeResumeAfterToolResults(options.messageId);
 	}
 	maybeResumeAfterToolResults(messageId) {
+		if (!this.maybeCompleteAfterToolResults(messageId)) return;
+		if (this.isRunningFlag) {
+			this.pendingResumeMessageId = messageId;
+			return;
+		}
+		this.startResumeRun(messageId);
+	}
+	maybeCompleteAfterToolResults(messageId) {
 		const message = this.messages.find((m) => m.id === messageId);
-		if (!message || message.role !== "assistant") return;
+		if (!message || message.role !== "assistant") return false;
 		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;
+		if (assistant.status?.type !== "requires-action" || assistant.status.reason !== "tool-calls") return false;
+		if (!assistant.content.every((part) => part.type !== "tool-call" || "result" in part && part.result !== void 0)) return false;
 		this.messages = this.messages.map((m) => m.id === messageId && m.role === "assistant" ? {
 			...m,
 			status: {
@@ -244,11 +265,7 @@ var AgUiThreadRuntimeCore = class {
 		} : m);
 		this.notifyUpdate();
 		this.persistAssistantHistory(messageId);
-		if (this.isRunningFlag) {
-			this.pendingResumeMessageId = messageId;
-			return;
-		}
-		this.startResumeRun(messageId);
+		return true;
 	}
 	startResumeRun(messageId) {
 		this.startRun(messageId, this.lastRunConfig).catch((error) => {
@@ -591,9 +608,46 @@ var AgUiThreadRuntimeCore = class {
 			case "MESSAGES_SNAPSHOT":
 				this.importMessagesSnapshot(event.messages);
 				return;
+			case "TOOL_CALL_RESULT":
+				if (!aggregator.hasToolCall(event.toolCallId)) {
+					const messageId = this.findMessageIdForToolCall(event.toolCallId);
+					if (messageId !== void 0) {
+						this.applyCrossRunToolResult(messageId, event);
+						return;
+					}
+				}
+				aggregator.handle(event);
+				return;
 			default: aggregator.handle(event);
 		}
 	}
+	applyCrossRunToolResult(messageId, event) {
+		let updated = false;
+		this.messages = this.messages.map((message) => {
+			if (message.id !== messageId || message.role !== "assistant") return message;
+			const assistant = message;
+			let matchedToolCall = false;
+			const content = assistant.content.map((part) => {
+				if (part.type !== "tool-call" || part.toolCallId !== event.toolCallId) return part;
+				matchedToolCall = true;
+				return {
+					...part,
+					result: tryParseJSON(event.content ?? ""),
+					...event.role === "tool" ? { isError: false } : {},
+					...event.messageId ? { unstable_toolMessageId: event.messageId } : {}
+				};
+			});
+			if (!matchedToolCall) return message;
+			updated = true;
+			return {
+				...assistant,
+				content
+			};
+		});
+		if (!updated) return;
+		this.notifyUpdate();
+		this.maybeCompleteAfterToolResults(messageId);
+	}
 	importMessagesSnapshot(rawMessages) {
 		try {
 			const normalized = fromAgUiMessages(rawMessages, { showThinking: this.showThinking });

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  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        isOptimistic: true,\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        if (m.id !== oldId) return m;\n        const { isOptimistic: _, ...metadata } = m.metadata;\n        return { ...m, id: newId, metadata } as ThreadMessage;\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      ...(current.isOptimistic ? { isOptimistic: true } : {}),\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        showThinking: this.showThinking,\n      });\n      const converted: ThreadMessage[] = [];\n      for (const message of normalized) {\n        try {\n          converted.push(\n            fromThreadMessageLike(message, generateId(), FALLBACK_USER_STATUS),\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,CAAC,EAAE,MAAM;IACvD,MAAM,eAAe,KAAK,SAAS,QAAQ,KAAK,KAAK,OAAO;IAC5D,MAAM,KAAK,SACT,UACA,KAAK,eACL,KAAA,GACA,YACF;GACF;EACF,CAAC,CAAC,CACD,OAAO,UAAU;GAChB,KAAK,OAAO,QAAQ,iCAAiC,KAAK;GAC1D,KAAK,UACH,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAC1D;EACF,CAAC,CAAC,CACD,cAAc;GACb,KAAK,aAAa;GAClB,KAAK,aAAa;EACpB,CAAC;EAEH,KAAK,aAAa;EAClB,OAAO,KAAK;CACd;CAEA,MAAM,OAAO,SAAuC;EAClD,MAAM,WAAW,QAAQ,YAAY,QAAQ,SAAS;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,yBAAyB,EAGlD;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,CAAC,CAAC,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,CAAC,CAAC,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,CAAC,CAAC,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,CAAC,CAAC,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,CAAC,EAAE,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,CAAC,EAAE;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,cAAc;IACd,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,MAAM;GACvC,IAAI,EAAE,OAAO,OAAO,OAAO;GAC3B,MAAM,EAAE,cAAc,GAAG,GAAG,aAAa,EAAE;GAC3C,OAAO;IAAE,GAAG;IAAG,IAAI;IAAO;GAAS;EACrC,CAAC;EAGH,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,QAAQ,eAAe,EAAE,cAAc,KAAK,IAAI,CAAC;GACrD,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,aAAa,EAC/C,cAAc,KAAK,aACrB,CAAC;GACD,MAAM,YAA6B,CAAC;GACpC,KAAK,MAAM,WAAW,YACpB,IAAI;IACF,UAAU,KACR,sBAAsB,SAAS,WAAW,GAAG,oBAAoB,CACnE;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,CAAC,CAAC,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 { AbstractAgent } 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  tryParseJSON,\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: AbstractAgent;\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: AbstractAgent;\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  async resumeInFlightRun(messages: readonly ThreadMessage[]): Promise<void> {\n    // Without a resume stream startRun would re-run the agent from scratch.\n    const resumeStream = this.history?.resume?.bind(this.history);\n    if (!resumeStream) {\n      const error = new Error(\n        \"[agui] unstable_resume requires a ThreadHistoryAdapter with a resume() method; skipping resume after thread switch\",\n      );\n      this.logger.error?.(error.message);\n      this.onError?.(error);\n      return;\n    }\n    const parentId = messages.at(-1)?.id ?? null;\n    try {\n      await this.startRun(\n        parentId,\n        this.lastRunConfig,\n        undefined,\n        resumeStream,\n      );\n    } catch {\n      // startRun already reported via onError; don't reject the switch.\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  // The continuation fires whether the frontend result lands before\n  // RUN_FINISHED (the status flips to requires-action only later, while the\n  // run is still draining) or after it.\n  private maybeResumeAfterToolResults(messageId: string): void {\n    if (!this.maybeCompleteAfterToolResults(messageId)) return;\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 maybeCompleteAfterToolResults(messageId: string): boolean {\n    const message = this.messages.find((m) => m.id === messageId);\n    if (!message || message.role !== \"assistant\") return false;\n    const assistant = message as ThreadAssistantMessage;\n    if (\n      assistant.status?.type !== \"requires-action\" ||\n      assistant.status.reason !== \"tool-calls\"\n    ) {\n      return false;\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 false;\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    return true;\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        isOptimistic: true,\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        if (m.id !== oldId) return m;\n        const { isOptimistic: _, ...metadata } = m.metadata;\n        return { ...m, id: newId, metadata } as ThreadMessage;\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      ...(current.isOptimistic ? { isOptimistic: true } : {}),\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      case \"TOOL_CALL_RESULT\": {\n        if (!aggregator.hasToolCall(event.toolCallId)) {\n          const messageId = this.findMessageIdForToolCall(event.toolCallId);\n          if (messageId !== undefined) {\n            this.applyCrossRunToolResult(messageId, event);\n            return;\n          }\n        }\n        aggregator.handle(event);\n        return;\n      }\n      default:\n        aggregator.handle(event);\n    }\n  }\n\n  private applyCrossRunToolResult(\n    messageId: string,\n    event: Extract<AgUiEvent, { type: \"TOOL_CALL_RESULT\" }>,\n  ): void {\n    let updated = 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      let matchedToolCall = false;\n      const content = assistant.content.map((part) => {\n        if (part.type !== \"tool-call\" || part.toolCallId !== event.toolCallId)\n          return part;\n        matchedToolCall = true;\n        return {\n          ...part,\n          result: tryParseJSON(event.content ?? \"\") as ReadonlyJSONValue,\n          ...(event.role === \"tool\" ? { isError: false } : {}),\n          ...(event.messageId\n            ? { unstable_toolMessageId: event.messageId }\n            : {}),\n        };\n      });\n      if (!matchedToolCall) return message;\n      updated = true;\n      return { ...assistant, content };\n    });\n\n    if (!updated) return;\n    this.notifyUpdate();\n    // Not maybeResumeAfterToolResults: the delivering run is already in\n    // flight, and a resume from the owner would reset the head past it.\n    this.maybeCompleteAfterToolResults(messageId);\n  }\n\n  private importMessagesSnapshot(rawMessages: readonly unknown[]) {\n    try {\n      const normalized = fromAgUiMessages(rawMessages, {\n        showThinking: this.showThinking,\n      });\n      const converted: ThreadMessage[] = [];\n      for (const message of normalized) {\n        try {\n          converted.push(\n            fromThreadMessageLike(message, generateId(), FALLBACK_USER_STATUS),\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":";;;;;;;AAiCA,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,CAAC,EAAE,MAAM;IACvD,MAAM,eAAe,KAAK,SAAS,QAAQ,KAAK,KAAK,OAAO;IAC5D,MAAM,KAAK,SACT,UACA,KAAK,eACL,KAAA,GACA,YACF;GACF;EACF,CAAC,CAAC,CACD,OAAO,UAAU;GAChB,KAAK,OAAO,QAAQ,iCAAiC,KAAK;GAC1D,KAAK,UACH,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAC1D;EACF,CAAC,CAAC,CACD,cAAc;GACb,KAAK,aAAa;GAClB,KAAK,aAAa;EACpB,CAAC;EAEH,KAAK,aAAa;EAClB,OAAO,KAAK;CACd;CAEA,MAAM,OAAO,SAAuC;EAClD,MAAM,WAAW,QAAQ,YAAY,QAAQ,SAAS;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,MAAM,kBAAkB,UAAmD;EAEzE,MAAM,eAAe,KAAK,SAAS,QAAQ,KAAK,KAAK,OAAO;EAC5D,IAAI,CAAC,cAAc;GACjB,MAAM,wBAAQ,IAAI,MAChB,oHACF;GACA,KAAK,OAAO,QAAQ,MAAM,OAAO;GACjC,KAAK,UAAU,KAAK;GACpB;EACF;EACA,MAAM,WAAW,SAAS,GAAG,EAAE,CAAC,EAAE,MAAM;EACxC,IAAI;GACF,MAAM,KAAK,SACT,UACA,KAAK,eACL,KAAA,GACA,YACF;EACF,QAAQ,CAER;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,yBAAyB,EAGlD;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,CAAC,CAAC,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,CAAC,CAAC,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,CAAC,CAAC,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;CAKA,4BAAoC,WAAyB;EAC3D,IAAI,CAAC,KAAK,8BAA8B,SAAS,GAAG;EAEpD,IAAI,KAAK,eAAe;GAGtB,KAAK,yBAAyB;GAC9B;EACF;EACA,KAAK,eAAe,SAAS;CAC/B;CAEA,8BAAsC,WAA4B;EAChE,MAAM,UAAU,KAAK,SAAS,MAAM,MAAM,EAAE,OAAO,SAAS;EAC5D,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa,OAAO;EACrD,MAAM,YAAY;EAClB,IACE,UAAU,QAAQ,SAAS,qBAC3B,UAAU,OAAO,WAAW,cAE5B,OAAO;EAOT,IAAI,CALgB,UAAU,QAAQ,OACnC,SACC,KAAK,SAAS,eACb,YAAY,QAAQ,KAAK,WAAW,KAAA,CAE1B,GAAG,OAAO;EAEzB,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;EACtC,OAAO;CACT;CAEA,eAAuB,WAAyB;EAC9C,KAAU,SAAS,WAAW,KAAK,aAAa,CAAC,CAAC,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,CAAC,EAAE,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,CAAC,EAAE;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,cAAc;IACd,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,MAAM;GACvC,IAAI,EAAE,OAAO,OAAO,OAAO;GAC3B,MAAM,EAAE,cAAc,GAAG,GAAG,aAAa,EAAE;GAC3C,OAAO;IAAE,GAAG;IAAG,IAAI;IAAO;GAAS;EACrC,CAAC;EAGH,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,QAAQ,eAAe,EAAE,cAAc,KAAK,IAAI,CAAC;GACrD,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,KAAK;IACH,IAAI,CAAC,WAAW,YAAY,MAAM,UAAU,GAAG;KAC7C,MAAM,YAAY,KAAK,yBAAyB,MAAM,UAAU;KAChE,IAAI,cAAc,KAAA,GAAW;MAC3B,KAAK,wBAAwB,WAAW,KAAK;MAC7C;KACF;IACF;IACA,WAAW,OAAO,KAAK;IACvB;GAEF,SACE,WAAW,OAAO,KAAK;EAC3B;CACF;CAEA,wBACE,WACA,OACM;EACN,IAAI,UAAU;EACd,KAAK,WAAW,KAAK,SAAS,KAAK,YAAY;GAC7C,IAAI,QAAQ,OAAO,aAAa,QAAQ,SAAS,aAC/C,OAAO;GACT,MAAM,YAAY;GAClB,IAAI,kBAAkB;GACtB,MAAM,UAAU,UAAU,QAAQ,KAAK,SAAS;IAC9C,IAAI,KAAK,SAAS,eAAe,KAAK,eAAe,MAAM,YACzD,OAAO;IACT,kBAAkB;IAClB,OAAO;KACL,GAAG;KACH,QAAQ,aAAa,MAAM,WAAW,EAAE;KACxC,GAAI,MAAM,SAAS,SAAS,EAAE,SAAS,MAAM,IAAI,CAAC;KAClD,GAAI,MAAM,YACN,EAAE,wBAAwB,MAAM,UAAU,IAC1C,CAAC;IACP;GACF,CAAC;GACD,IAAI,CAAC,iBAAiB,OAAO;GAC7B,UAAU;GACV,OAAO;IAAE,GAAG;IAAW;GAAQ;EACjC,CAAC;EAED,IAAI,CAAC,SAAS;EACd,KAAK,aAAa;EAGlB,KAAK,8BAA8B,SAAS;CAC9C;CAEA,uBAA+B,aAAiC;EAC9D,IAAI;GACF,MAAM,aAAa,iBAAiB,aAAa,EAC/C,cAAc,KAAK,aACrB,CAAC;GACD,MAAM,YAA6B,CAAC;GACpC,KAAK,MAAM,WAAW,YACpB,IAAI;IACF,UAAU,KACR,sBAAsB,SAAS,WAAW,GAAG,oBAAoB,CACnE;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,CAAC,CAAC,OAAO,UAAU;GAC/D,KAAK,mBAAmB,OAAO,QAAQ,EAAE;GACzC,KAAK,OAAO,QAAQ,yCAAyC,KAAK;EACpE,CAAC;CACH;AACF"}

package/dist/runtime/adapter/run-aggregator.d.ts

@@ -41,6 +41,7 @@ declare class RunAggregator {
   private firstTokenTime;
   private totalChunks;
   constructor(options: RunAggregatorOptions);
+  hasToolCall(toolCallId: string): boolean;
   handle(event: AgUiEvent): void;
   private reportServerMessageId;
   private clearTextParts;
@@ -53,7 +54,6 @@ declare class RunAggregator {
   private startToolCall;
   private appendToolArgs;
   private finishToolCall;
-  private tryParseJSON;
   private emit;
   private recordFirstToken;
   private getTiming;
@@ -61,6 +61,7 @@ declare class RunAggregator {
   private handleReasoningContent;
   private handleReasoningEnd;
 }
+declare function tryParseJSON(value: string): unknown;
 //#endregion
-export { AG_UI_METADATA_NAMESPACE, AgUiCustomMetadata, RunAggregator, RunAggregatorOptions };
+export { AG_UI_METADATA_NAMESPACE, AgUiCustomMetadata, RunAggregator, RunAggregatorOptions, tryParseJSON };
 //# sourceMappingURL=run-aggregator.d.ts.map

package/dist/runtime/adapter/run-aggregator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"run-aggregator.d.ts","names":[],"sources":["../../../src/runtime/adapter/run-aggregator.ts"],"mappings":";;;;;cAYa,wBAAA;AAAA,KAED,kBAAA;EACV,UAAA,GAAa,aAAa;AAAA;AAAA,KAGvB,IAAA,IAAQ,MAA0B,EAAlB,kBAAkB;AAAA,KAgB3B,oBAAA;EACV,YAAA;EACA,MAAA,EAAQ,MAAA;EACR,IAAA,EAAM,IAAI;EACV,iBAAA,IAAqB,SAAA;AAAA;AAvBK;AAC1B;;;;AAEqC;AAHX,cAgCf,aAAA;EAAA,iBACM,UAAA;EAAA,iBACA,YAAA;EAAA,iBACA,MAAA;EAAA,iBACA,iBAAA;EAAA,QAET,MAAA;EAAA,QACA,UAAA;EAAA,iBACS,SAAA;EAAA,QAIT,mBAAA;EAAA,iBACS,cAAA;EAAA,QACT,kBAAA;EAAA,QACA,oBAAA;EAAA,iBACS,SAAA;EAAA,QACT,sBAAA;EAAA,iBACS,SAAA;EAAA,QAKT,eAAA;EAAA,QACA,uBAAA;EAAA,QAEA,eAAA;EAAA,QACA,cAAA;EAAA,QACA,WAAA;cAEI,OAAA,EAAS,oBAAA;EAOrB,MAAA,CAAO,KAAA,EAAO,SAAS;EAAA,QA0Kf,qBAAA;EAAA,QAMA,cAAA;EAAA,QAIA,eAAA;EAAA,QAKA,gBAAA;EAAA,QAOA,oBAAA;EAAA,QAiBA,cAAA;EAAA,QAWA,mBAAA;EAAA,QAMA,UAAA;EAAA,QAQA,aAAA;EAAA,QA8BA,cAAA;EAAA,QAgBA,cAAA;EAAA,QAkCA,YAAA;EAAA,QASA,IAAA;EAAA,QAgEA,gBAAA;EAAA,QASA,SAAA;EAAA,QAgCA,oBAAA;EAAA,QAcA,sBAAA;EAAA,QAYA,kBAAA;AAAA"}
+{"version":3,"file":"run-aggregator.d.ts","names":[],"sources":["../../../src/runtime/adapter/run-aggregator.ts"],"mappings":";;;;;cAYa,wBAAA;AAAA,KAED,kBAAA;EACV,UAAA,GAAa,aAAa;AAAA;AAAA,KAGvB,IAAA,IAAQ,MAA0B,EAAlB,kBAAkB;AAAA,KAgB3B,oBAAA;EACV,YAAA;EACA,MAAA,EAAQ,MAAA;EACR,IAAA,EAAM,IAAI;EACV,iBAAA,IAAqB,SAAA;AAAA;AAvBK;AAC1B;;;;AAEqC;AAHX,cAgCf,aAAA;EAAA,iBACM,UAAA;EAAA,iBACA,YAAA;EAAA,iBACA,MAAA;EAAA,iBACA,iBAAA;EAAA,QAET,MAAA;EAAA,QACA,UAAA;EAAA,iBACS,SAAA;EAAA,QAIT,mBAAA;EAAA,iBACS,cAAA;EAAA,QACT,kBAAA;EAAA,QACA,oBAAA;EAAA,iBACS,SAAA;EAAA,QACT,sBAAA;EAAA,iBACS,SAAA;EAAA,QAKT,eAAA;EAAA,QACA,uBAAA;EAAA,QAEA,eAAA;EAAA,QACA,cAAA;EAAA,QACA,WAAA;cAEI,OAAA,EAAS,oBAAA;EAOrB,WAAA,CAAY,UAAA;EAIZ,MAAA,CAAO,KAAA,EAAO,SAAS;EAAA,QA0Kf,qBAAA;EAAA,QAMA,cAAA;EAAA,QAIA,eAAA;EAAA,QAKA,gBAAA;EAAA,QAOA,oBAAA;EAAA,QAiBA,cAAA;EAAA,QAWA,mBAAA;EAAA,QAMA,UAAA;EAAA,QAQA,aAAA;EAAA,QA8BA,cAAA;EAAA,QAgBA,cAAA;EAAA,QAkCA,IAAA;EAAA,QAgEA,gBAAA;EAAA,QASA,SAAA;EAAA,QAgCA,oBAAA;EAAA,QAcA,sBAAA;EAAA,QAYA,kBAAA;AAAA;AAAA,iBAOM,YAAA,CAAa,KAAa"}

package/dist/runtime/adapter/run-aggregator.js

@@ -35,6 +35,9 @@ var RunAggregator = class {
 		this.logger = options.logger;
 		this.onServerMessageId = options.onServerMessageId;
 	}
+	hasToolCall(toolCallId) {
+		return this.toolCalls.has(toolCallId);
+	}
 	handle(event) {
 		switch (event.type) {
 			case "RUN_STARTED":
@@ -273,19 +276,11 @@ var RunAggregator = class {
 			kind: "tool-call",
 			toolCallId: id
 		});
-		entry.result = this.tryParseJSON(content);
+		entry.result = tryParseJSON(content);
 		entry.isError = isError;
 		if (toolMessageId) entry.toolMessageId = toolMessageId;
 		this.lastResolvedToolCallId = id;
 	}
-	tryParseJSON(value) {
-		if (!value) return value;
-		try {
-			return JSON.parse(value);
-		} catch {
-			return value;
-		}
-	}
 	emit() {
 		const snapshot = [];
 		for (const part of this.partOrder) {
@@ -381,7 +376,15 @@ var RunAggregator = class {
 		this.emit();
 	}
 };
+function tryParseJSON(value) {
+	if (!value) return value;
+	try {
+		return JSON.parse(value);
+	} catch {
+		return value;
+	}
+}
 //#endregion
-export { AG_UI_METADATA_NAMESPACE, RunAggregator };
+export { AG_UI_METADATA_NAMESPACE, RunAggregator, tryParseJSON };
 
 //# sourceMappingURL=run-aggregator.js.map

package/dist/runtime/adapter/run-aggregator.js.map

@@ -1 +1 @@
-{"version":3,"file":"run-aggregator.js","names":[],"sources":["../../../src/runtime/adapter/run-aggregator.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  isMcpAppUri,\n  type ChatModelRunResult,\n  type MessageTiming,\n  type ThreadAssistantMessagePart,\n  type ToolCallMessagePart,\n} from \"@assistant-ui/core\";\nimport type { AgUiEvent, AgUiInterrupt } from \"../types\";\nimport type { Logger } from \"../logger\";\n\nexport const AG_UI_METADATA_NAMESPACE = \"agui\";\n\nexport type AgUiCustomMetadata = {\n  interrupts?: AgUiInterrupt[];\n};\n\ntype Emit = (update: ChatModelRunResult) => void;\n\ntype ToolCallState = {\n  toolCallId: string;\n  toolCallName: string;\n  argsText: string;\n  parsedArgs: Record<string, unknown> | undefined;\n  result: unknown;\n  isError: boolean | undefined;\n  parentMessageId?: string;\n  toolMessageId?: string;\n  mcpAppResourceUri?: string;\n};\n\nconst MCP_APPS_ACTIVITY_TYPE = \"mcp-apps\";\n\nexport type RunAggregatorOptions = {\n  showThinking: boolean;\n  logger: Logger;\n  emit: Emit;\n  onServerMessageId?: (messageId: string) => void;\n};\n\n/**\n * Collects AG-UI events into assistant-ui run snapshots that can be yielded from a ChatModelAdapter.\n *\n * The aggregator keeps a single assistant message worth of parts. Each incoming event updates the parts and\n * emits a fresh snapshot through the provided `emit` callback.\n */\nexport class RunAggregator {\n  private readonly emitUpdate: Emit;\n  private readonly showThinking: boolean;\n  private readonly logger: Logger;\n  private readonly onServerMessageId: ((messageId: string) => void) | undefined;\n\n  private status: ChatModelRunResult[\"status\"] | undefined;\n  private interrupts: AgUiInterrupt[] | undefined;\n  private readonly textParts = new Map<\n    string,\n    { buffer: string; touched: boolean }\n  >();\n  private activeTextMessageId: string | undefined;\n  private readonly reasoningParts = new Map<string, string>(); // key → buffer\n  private activeReasoningKey: string | undefined;\n  private reasoningPartCounter = 0;\n  private readonly toolCalls = new Map<string, ToolCallState>();\n  private lastResolvedToolCallId: string | undefined;\n  private readonly partOrder: (\n    | { kind: \"text\"; key: string }\n    | { kind: \"reasoning\"; key: string }\n    | { kind: \"tool-call\"; toolCallId: string }\n  )[] = [];\n  private textPartCounter = 0;\n  private serverMessageIdReported = false;\n\n  private streamStartTime: number | undefined;\n  private firstTokenTime: number | undefined;\n  private totalChunks = 0;\n\n  constructor(options: RunAggregatorOptions) {\n    this.emitUpdate = options.emit;\n    this.showThinking = options.showThinking;\n    this.logger = options.logger;\n    this.onServerMessageId = options.onServerMessageId;\n  }\n\n  handle(event: AgUiEvent): void {\n    switch (event.type) {\n      case \"RUN_STARTED\": {\n        this.clearTextParts();\n        this.reasoningParts.clear();\n        this.activeReasoningKey = undefined;\n        this.reasoningPartCounter = 0;\n        this.toolCalls.clear();\n        this.lastResolvedToolCallId = undefined;\n        this.partOrder.length = 0;\n        this.textPartCounter = 0;\n        this.activeTextMessageId = undefined;\n        this.interrupts = undefined;\n        this.serverMessageIdReported = false;\n        this.streamStartTime = Date.now();\n        this.firstTokenTime = undefined;\n        this.totalChunks = 0;\n        this.status = { type: \"running\" };\n        this.emit();\n        break;\n      }\n      case \"RUN_FINISHED\": {\n        if (event.outcome?.type === \"interrupt\") {\n          this.interrupts = event.outcome.interrupts;\n          this.status = { type: \"requires-action\", reason: \"interrupt\" };\n          this.emit();\n          break;\n        }\n\n        this.interrupts = undefined;\n        const hasUnresolvedToolCalls = Array.from(this.toolCalls.values()).some(\n          (tc) => tc.result === undefined,\n        );\n\n        this.status =\n          event.outcome?.type === \"success\" || !hasUnresolvedToolCalls\n            ? { type: \"complete\", reason: \"unknown\" }\n            : { type: \"requires-action\", reason: \"tool-calls\" };\n        this.emit();\n        break;\n      }\n      case \"RUN_ERROR\": {\n        this.status = {\n          type: \"incomplete\",\n          reason: \"error\",\n          ...(event.message !== undefined ? { error: event.message } : {}),\n        };\n        this.emit();\n        break;\n      }\n      case \"RUN_CANCELLED\": {\n        this.status = { type: \"incomplete\", reason: \"cancelled\" };\n        this.emit();\n        break;\n      }\n\n      case \"TEXT_MESSAGE_START\": {\n        this.reportServerMessageId(event.messageId);\n        const id = this.startTextMessage(event.messageId);\n        if (id) {\n          this.markTextPartTouched(id);\n        }\n        this.emit();\n        break;\n      }\n      case \"TEXT_MESSAGE_CONTENT\":\n      case \"TEXT_MESSAGE_CHUNK\": {\n        const incomingId = \"messageId\" in event ? event.messageId : undefined;\n        this.reportServerMessageId(incomingId);\n        if (!event.delta) break;\n        this.recordFirstToken();\n        const id = this.resolveTextMessageId(incomingId);\n        this.appendText(id, event.delta);\n        this.totalChunks++;\n        this.emit();\n        break;\n      }\n      case \"TEXT_MESSAGE_END\": {\n        this.reportServerMessageId(event.messageId);\n        if (event.messageId && this.activeTextMessageId === event.messageId) {\n          this.activeTextMessageId = undefined;\n        }\n        this.emit();\n        break;\n      }\n\n      case \"THINKING_START\":\n      case \"THINKING_TEXT_MESSAGE_START\":\n      case \"REASONING_START\":\n      case \"REASONING_MESSAGE_START\":\n        this.handleReasoningStart(\n          \"messageId\" in event ? event.messageId : undefined,\n        );\n        break;\n      case \"THINKING_TEXT_MESSAGE_CONTENT\":\n        this.handleReasoningContent(event.delta);\n        this.totalChunks++;\n        this.recordFirstToken();\n        break;\n      case \"REASONING_MESSAGE_CONTENT\":\n        this.handleReasoningContent(\n          event.delta,\n          \"messageId\" in event ? event.messageId : undefined,\n        );\n        this.totalChunks++;\n        this.recordFirstToken();\n        break;\n      case \"THINKING_TEXT_MESSAGE_END\":\n      case \"THINKING_END\":\n      case \"REASONING_MESSAGE_END\":\n      case \"REASONING_END\":\n        this.handleReasoningEnd();\n        break;\n\n      case \"TOOL_CALL_START\": {\n        this.reportServerMessageId(event.parentMessageId);\n        this.startToolCall(\n          event.toolCallId,\n          event.toolCallName,\n          event.parentMessageId,\n        );\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_ARGS\":\n      case \"TOOL_CALL_CHUNK\": {\n        if (event.type === \"TOOL_CALL_CHUNK\") {\n          this.reportServerMessageId(event.parentMessageId);\n        }\n        if (!event.delta) break;\n        this.appendToolArgs(event.toolCallId, event.delta);\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_END\": {\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_RESULT\": {\n        this.finishToolCall(\n          event.toolCallId,\n          event.content ?? \"\",\n          event.role === \"tool\" ? false : undefined,\n          event.messageId,\n        );\n        this.emit();\n        break;\n      }\n      case \"ACTIVITY_SNAPSHOT\": {\n        if (event.activityType !== MCP_APPS_ACTIVITY_TYPE) break;\n        const id = this.lastResolvedToolCallId;\n        const entry = id ? this.toolCalls.get(id) : undefined;\n        const resourceUri = event.content.resourceUri;\n        if (\n          entry &&\n          typeof resourceUri === \"string\" &&\n          isMcpAppUri(resourceUri)\n        ) {\n          entry.mcpAppResourceUri = resourceUri;\n          this.emit();\n        }\n        break;\n      }\n\n      default: {\n        this.logger.debug?.(\"[agui] aggregator ignored event\", event);\n      }\n    }\n  }\n\n  private reportServerMessageId(messageId: string | undefined): void {\n    if (this.serverMessageIdReported || !messageId) return;\n    this.serverMessageIdReported = true;\n    this.onServerMessageId?.(messageId);\n  }\n\n  private clearTextParts(): void {\n    this.textParts.clear();\n  }\n\n  private generateTextKey(): string {\n    this.textPartCounter += 1;\n    return `text-${this.textPartCounter}`;\n  }\n\n  private startTextMessage(messageId?: string): string {\n    const id = messageId ?? this.generateTextKey();\n    this.ensureTextPart(id);\n    this.activeTextMessageId = id;\n    return id;\n  }\n\n  private resolveTextMessageId(messageId?: string): string {\n    if (messageId) {\n      this.ensureTextPart(messageId);\n      this.activeTextMessageId = messageId;\n      return messageId;\n    }\n\n    if (this.activeTextMessageId) {\n      return this.activeTextMessageId;\n    }\n\n    const generated = this.generateTextKey();\n    this.ensureTextPart(generated);\n    this.activeTextMessageId = generated;\n    return generated;\n  }\n\n  private ensureTextPart(id: string): void {\n    if (!this.textParts.has(id)) {\n      this.textParts.set(id, { buffer: \"\", touched: false });\n      if (\n        !this.partOrder.some((part) => part.kind === \"text\" && part.key === id)\n      ) {\n        this.partOrder.push({ kind: \"text\", key: id });\n      }\n    }\n  }\n\n  private markTextPartTouched(id: string): void {\n    const entry = this.textParts.get(id);\n    if (!entry) return;\n    entry.touched = true;\n  }\n\n  private appendText(id: string, delta: string): void {\n    this.ensureTextPart(id);\n    const entry = this.textParts.get(id);\n    if (!entry) return;\n    entry.buffer += delta;\n    entry.touched = true;\n  }\n\n  private startToolCall(\n    id: string | undefined,\n    name?: string,\n    parentMessageId?: string,\n  ) {\n    if (!id) return;\n    // A new tool call acts as a boundary: any anonymous text that arrives\n    // after it should be a new part, not appended to the pre-tool text.\n    this.activeTextMessageId = undefined;\n    if (\n      !this.partOrder.some(\n        (part) => part.kind === \"tool-call\" && part.toolCallId === id,\n      )\n    ) {\n      this.partOrder.push({ kind: \"tool-call\", toolCallId: id });\n    }\n    const state: ToolCallState = {\n      toolCallId: id,\n      toolCallName: name ?? \"tool\",\n      argsText: \"\",\n      parsedArgs: undefined,\n      result: undefined,\n      isError: undefined,\n    };\n    if (parentMessageId) {\n      state.parentMessageId = parentMessageId;\n    }\n    this.toolCalls.set(id, state);\n  }\n\n  private appendToolArgs(id: string | undefined, delta: string) {\n    const entry = id ? this.toolCalls.get(id) : undefined;\n    if (!entry) return;\n    entry.argsText += delta;\n    try {\n      const parsed = JSON.parse(entry.argsText);\n      if (parsed && typeof parsed === \"object\") {\n        entry.parsedArgs = parsed as Record<string, unknown>;\n      } else {\n        entry.parsedArgs = undefined;\n      }\n    } catch {\n      entry.parsedArgs = undefined;\n    }\n  }\n\n  private finishToolCall(\n    id: string,\n    content: string,\n    isError?: boolean,\n    toolMessageId?: string,\n  ) {\n    if (!id) return;\n    let entry = this.toolCalls.get(id);\n    if (!entry) {\n      entry = {\n        toolCallId: id,\n        toolCallName: \"tool\",\n        argsText: \"\",\n        parsedArgs: undefined,\n        result: undefined,\n        isError: undefined,\n      };\n      this.toolCalls.set(id, entry);\n    }\n    if (\n      !this.partOrder.some(\n        (part) => part.kind === \"tool-call\" && part.toolCallId === id,\n      )\n    ) {\n      this.partOrder.push({ kind: \"tool-call\", toolCallId: id });\n    }\n    entry.result = this.tryParseJSON(content);\n    entry.isError = isError;\n    if (toolMessageId) {\n      entry.toolMessageId = toolMessageId;\n    }\n    this.lastResolvedToolCallId = id;\n  }\n\n  private tryParseJSON(value: string): unknown {\n    if (!value) return value;\n    try {\n      return JSON.parse(value);\n    } catch {\n      return value;\n    }\n  }\n\n  private emit(): void {\n    const snapshot: ThreadAssistantMessagePart[] = [];\n\n    for (const part of this.partOrder) {\n      if (part.kind === \"reasoning\") {\n        if (this.showThinking) {\n          const buffer = this.reasoningParts.get(part.key) ?? \"\";\n          if (buffer.length > 0 || this.activeReasoningKey === part.key) {\n            snapshot.push({ type: \"reasoning\", text: buffer } as const);\n          }\n        }\n        continue;\n      }\n\n      if (part.kind === \"text\") {\n        const entry = this.textParts.get(part.key);\n        if (entry?.touched) {\n          snapshot.push({ type: \"text\", text: entry.buffer } as const);\n        }\n        continue;\n      }\n\n      const entry = this.toolCalls.get(part.toolCallId);\n      if (!entry) continue;\n      const toolPart: ToolCallMessagePart = {\n        type: \"tool-call\",\n        toolCallId: entry.toolCallId,\n        toolName: entry.toolCallName,\n        args: (entry.parsedArgs ?? {}) as any,\n        argsText: entry.argsText,\n        ...(entry.result !== undefined ? { result: entry.result } : {}),\n        ...(entry.isError !== undefined ? { isError: entry.isError } : {}),\n        ...(entry.mcpAppResourceUri\n          ? { mcp: { app: { resourceUri: entry.mcpAppResourceUri } } }\n          : {}),\n        ...(entry.parentMessageId ? { parentId: entry.parentMessageId } : {}),\n        ...(entry.toolMessageId\n          ? { unstable_toolMessageId: entry.toolMessageId }\n          : {}),\n      } as ToolCallMessagePart & { unstable_toolMessageId?: string };\n      snapshot.push(toolPart);\n    }\n\n    const timing = this.getTiming();\n    const metadata = {\n      ...(timing ? { timing } : {}),\n      ...(this.interrupts\n        ? {\n            custom: {\n              [AG_UI_METADATA_NAMESPACE]: {\n                interrupts: this.interrupts,\n              } satisfies AgUiCustomMetadata,\n            },\n          }\n        : {}),\n    };\n    const result: ChatModelRunResult = {\n      content: snapshot,\n      ...(this.status ? { status: this.status } : undefined),\n      ...(Object.keys(metadata).length > 0 ? { metadata } : undefined),\n    };\n    this.emitUpdate(result);\n  }\n\n  private recordFirstToken(): void {\n    if (\n      this.firstTokenTime === undefined &&\n      this.streamStartTime !== undefined\n    ) {\n      this.firstTokenTime = Date.now() - this.streamStartTime;\n    }\n  }\n\n  private getTiming(): MessageTiming | undefined {\n    if (this.streamStartTime === undefined) return undefined;\n\n    const now = Date.now();\n    const totalStreamTime = now - this.streamStartTime;\n    const tokenCount =\n      this.totalChunks > 0\n        ? Math.ceil(\n            Array.from(this.textParts.values()).reduce(\n              (sum, p) => sum + p.buffer.length,\n              0,\n            ) / 4,\n          )\n        : undefined;\n    const tokensPerSecond =\n      tokenCount && totalStreamTime > 0\n        ? (tokenCount / totalStreamTime) * 1000\n        : undefined;\n\n    return {\n      streamStartTime: this.streamStartTime,\n      ...(this.firstTokenTime !== undefined\n        ? { firstTokenTime: this.firstTokenTime }\n        : {}),\n      totalStreamTime,\n      ...(tokenCount !== undefined ? { tokenCount } : {}),\n      ...(tokensPerSecond !== undefined ? { tokensPerSecond } : {}),\n      totalChunks: this.totalChunks,\n      toolCallCount: this.toolCalls.size,\n    };\n  }\n\n  private handleReasoningStart(messageId?: string): void {\n    if (!this.showThinking) return;\n    // A reasoning block acts as a boundary: anonymous text arriving after it\n    // should be a new part, not appended to any pre-reasoning text.\n    this.activeTextMessageId = undefined;\n    const key = messageId ?? `__auto-reasoning-${++this.reasoningPartCounter}`;\n    if (!this.reasoningParts.has(key)) {\n      this.reasoningParts.set(key, \"\");\n      this.partOrder.push({ kind: \"reasoning\", key });\n    }\n    this.activeReasoningKey = key;\n    this.emit();\n  }\n\n  private handleReasoningContent(delta: string, messageId?: string): void {\n    if (!this.showThinking || !delta) return;\n    if (!this.activeReasoningKey) {\n      // Content arrived without a preceding START — create the slot lazily.\n      this.handleReasoningStart(messageId);\n    }\n    const key = this.activeReasoningKey;\n    if (!key) return;\n    this.reasoningParts.set(key, (this.reasoningParts.get(key) ?? \"\") + delta);\n    this.emit();\n  }\n\n  private handleReasoningEnd(): void {\n    if (!this.showThinking) return;\n    this.activeReasoningKey = undefined;\n    this.emit();\n  }\n}\n"],"mappings":";;;AAYA,MAAa,2BAA2B;AAoBxC,MAAM,yBAAyB;;;;;;;AAe/B,IAAa,gBAAb,MAA2B;CACzB;CACA;CACA;CACA;CAEA;CACA;CACA,4BAA6B,IAAI,IAG/B;CACF;CACA,iCAAkC,IAAI,IAAoB;CAC1D;CACA,uBAA+B;CAC/B,4BAA6B,IAAI,IAA2B;CAC5D;CACA,YAIM,CAAC;CACP,kBAA0B;CAC1B,0BAAkC;CAElC;CACA;CACA,cAAsB;CAEtB,YAAY,SAA+B;EACzC,KAAK,aAAa,QAAQ;EAC1B,KAAK,eAAe,QAAQ;EAC5B,KAAK,SAAS,QAAQ;EACtB,KAAK,oBAAoB,QAAQ;CACnC;CAEA,OAAO,OAAwB;EAC7B,QAAQ,MAAM,MAAd;GACE,KAAK;IACH,KAAK,eAAe;IACpB,KAAK,eAAe,MAAM;IAC1B,KAAK,qBAAqB,KAAA;IAC1B,KAAK,uBAAuB;IAC5B,KAAK,UAAU,MAAM;IACrB,KAAK,yBAAyB,KAAA;IAC9B,KAAK,UAAU,SAAS;IACxB,KAAK,kBAAkB;IACvB,KAAK,sBAAsB,KAAA;IAC3B,KAAK,aAAa,KAAA;IAClB,KAAK,0BAA0B;IAC/B,KAAK,kBAAkB,KAAK,IAAI;IAChC,KAAK,iBAAiB,KAAA;IACtB,KAAK,cAAc;IACnB,KAAK,SAAS,EAAE,MAAM,UAAU;IAChC,KAAK,KAAK;IACV;GAEF,KAAK,gBAAgB;IACnB,IAAI,MAAM,SAAS,SAAS,aAAa;KACvC,KAAK,aAAa,MAAM,QAAQ;KAChC,KAAK,SAAS;MAAE,MAAM;MAAmB,QAAQ;KAAY;KAC7D,KAAK,KAAK;KACV;IACF;IAEA,KAAK,aAAa,KAAA;IAClB,MAAM,yBAAyB,MAAM,KAAK,KAAK,UAAU,OAAO,CAAC,CAAC,CAAC,MAChE,OAAO,GAAG,WAAW,KAAA,CACxB;IAEA,KAAK,SACH,MAAM,SAAS,SAAS,aAAa,CAAC,yBAClC;KAAE,MAAM;KAAY,QAAQ;IAAU,IACtC;KAAE,MAAM;KAAmB,QAAQ;IAAa;IACtD,KAAK,KAAK;IACV;GACF;GACA,KAAK;IACH,KAAK,SAAS;KACZ,MAAM;KACN,QAAQ;KACR,GAAI,MAAM,YAAY,KAAA,IAAY,EAAE,OAAO,MAAM,QAAQ,IAAI,CAAC;IAChE;IACA,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,SAAS;KAAE,MAAM;KAAc,QAAQ;IAAY;IACxD,KAAK,KAAK;IACV;GAGF,KAAK,sBAAsB;IACzB,KAAK,sBAAsB,MAAM,SAAS;IAC1C,MAAM,KAAK,KAAK,iBAAiB,MAAM,SAAS;IAChD,IAAI,IACF,KAAK,oBAAoB,EAAE;IAE7B,KAAK,KAAK;IACV;GACF;GACA,KAAK;GACL,KAAK,sBAAsB;IACzB,MAAM,aAAa,eAAe,QAAQ,MAAM,YAAY,KAAA;IAC5D,KAAK,sBAAsB,UAAU;IACrC,IAAI,CAAC,MAAM,OAAO;IAClB,KAAK,iBAAiB;IACtB,MAAM,KAAK,KAAK,qBAAqB,UAAU;IAC/C,KAAK,WAAW,IAAI,MAAM,KAAK;IAC/B,KAAK;IACL,KAAK,KAAK;IACV;GACF;GACA,KAAK;IACH,KAAK,sBAAsB,MAAM,SAAS;IAC1C,IAAI,MAAM,aAAa,KAAK,wBAAwB,MAAM,WACxD,KAAK,sBAAsB,KAAA;IAE7B,KAAK,KAAK;IACV;GAGF,KAAK;GACL,KAAK;GACL,KAAK;GACL,KAAK;IACH,KAAK,qBACH,eAAe,QAAQ,MAAM,YAAY,KAAA,CAC3C;IACA;GACF,KAAK;IACH,KAAK,uBAAuB,MAAM,KAAK;IACvC,KAAK;IACL,KAAK,iBAAiB;IACtB;GACF,KAAK;IACH,KAAK,uBACH,MAAM,OACN,eAAe,QAAQ,MAAM,YAAY,KAAA,CAC3C;IACA,KAAK;IACL,KAAK,iBAAiB;IACtB;GACF,KAAK;GACL,KAAK;GACL,KAAK;GACL,KAAK;IACH,KAAK,mBAAmB;IACxB;GAEF,KAAK;IACH,KAAK,sBAAsB,MAAM,eAAe;IAChD,KAAK,cACH,MAAM,YACN,MAAM,cACN,MAAM,eACR;IACA,KAAK,KAAK;IACV;GAEF,KAAK;GACL,KAAK;IACH,IAAI,MAAM,SAAS,mBACjB,KAAK,sBAAsB,MAAM,eAAe;IAElD,IAAI,CAAC,MAAM,OAAO;IAClB,KAAK,eAAe,MAAM,YAAY,MAAM,KAAK;IACjD,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,eACH,MAAM,YACN,MAAM,WAAW,IACjB,MAAM,SAAS,SAAS,QAAQ,KAAA,GAChC,MAAM,SACR;IACA,KAAK,KAAK;IACV;GAEF,KAAK,qBAAqB;IACxB,IAAI,MAAM,iBAAiB,wBAAwB;IACnD,MAAM,KAAK,KAAK;IAChB,MAAM,QAAQ,KAAK,KAAK,UAAU,IAAI,EAAE,IAAI,KAAA;IAC5C,MAAM,cAAc,MAAM,QAAQ;IAClC,IACE,SACA,OAAO,gBAAgB,YACvB,YAAY,WAAW,GACvB;KACA,MAAM,oBAAoB;KAC1B,KAAK,KAAK;IACZ;IACA;GACF;GAEA,SACE,KAAK,OAAO,QAAQ,mCAAmC,KAAK;EAEhE;CACF;CAEA,sBAA8B,WAAqC;EACjE,IAAI,KAAK,2BAA2B,CAAC,WAAW;EAChD,KAAK,0BAA0B;EAC/B,KAAK,oBAAoB,SAAS;CACpC;CAEA,iBAA+B;EAC7B,KAAK,UAAU,MAAM;CACvB;CAEA,kBAAkC;EAChC,KAAK,mBAAmB;EACxB,OAAO,QAAQ,KAAK;CACtB;CAEA,iBAAyB,WAA4B;EACnD,MAAM,KAAK,aAAa,KAAK,gBAAgB;EAC7C,KAAK,eAAe,EAAE;EACtB,KAAK,sBAAsB;EAC3B,OAAO;CACT;CAEA,qBAA6B,WAA4B;EACvD,IAAI,WAAW;GACb,KAAK,eAAe,SAAS;GAC7B,KAAK,sBAAsB;GAC3B,OAAO;EACT;EAEA,IAAI,KAAK,qBACP,OAAO,KAAK;EAGd,MAAM,YAAY,KAAK,gBAAgB;EACvC,KAAK,eAAe,SAAS;EAC7B,KAAK,sBAAsB;EAC3B,OAAO;CACT;CAEA,eAAuB,IAAkB;EACvC,IAAI,CAAC,KAAK,UAAU,IAAI,EAAE,GAAG;GAC3B,KAAK,UAAU,IAAI,IAAI;IAAE,QAAQ;IAAI,SAAS;GAAM,CAAC;GACrD,IACE,CAAC,KAAK,UAAU,MAAM,SAAS,KAAK,SAAS,UAAU,KAAK,QAAQ,EAAE,GAEtE,KAAK,UAAU,KAAK;IAAE,MAAM;IAAQ,KAAK;GAAG,CAAC;EAEjD;CACF;CAEA,oBAA4B,IAAkB;EAC5C,MAAM,QAAQ,KAAK,UAAU,IAAI,EAAE;EACnC,IAAI,CAAC,OAAO;EACZ,MAAM,UAAU;CAClB;CAEA,WAAmB,IAAY,OAAqB;EAClD,KAAK,eAAe,EAAE;EACtB,MAAM,QAAQ,KAAK,UAAU,IAAI,EAAE;EACnC,IAAI,CAAC,OAAO;EACZ,MAAM,UAAU;EAChB,MAAM,UAAU;CAClB;CAEA,cACE,IACA,MACA,iBACA;EACA,IAAI,CAAC,IAAI;EAGT,KAAK,sBAAsB,KAAA;EAC3B,IACE,CAAC,KAAK,UAAU,MACb,SAAS,KAAK,SAAS,eAAe,KAAK,eAAe,EAC7D,GAEA,KAAK,UAAU,KAAK;GAAE,MAAM;GAAa,YAAY;EAAG,CAAC;EAE3D,MAAM,QAAuB;GAC3B,YAAY;GACZ,cAAc,QAAQ;GACtB,UAAU;GACV,YAAY,KAAA;GACZ,QAAQ,KAAA;GACR,SAAS,KAAA;EACX;EACA,IAAI,iBACF,MAAM,kBAAkB;EAE1B,KAAK,UAAU,IAAI,IAAI,KAAK;CAC9B;CAEA,eAAuB,IAAwB,OAAe;EAC5D,MAAM,QAAQ,KAAK,KAAK,UAAU,IAAI,EAAE,IAAI,KAAA;EAC5C,IAAI,CAAC,OAAO;EACZ,MAAM,YAAY;EAClB,IAAI;GACF,MAAM,SAAS,KAAK,MAAM,MAAM,QAAQ;GACxC,IAAI,UAAU,OAAO,WAAW,UAC9B,MAAM,aAAa;QAEnB,MAAM,aAAa,KAAA;EAEvB,QAAQ;GACN,MAAM,aAAa,KAAA;EACrB;CACF;CAEA,eACE,IACA,SACA,SACA,eACA;EACA,IAAI,CAAC,IAAI;EACT,IAAI,QAAQ,KAAK,UAAU,IAAI,EAAE;EACjC,IAAI,CAAC,OAAO;GACV,QAAQ;IACN,YAAY;IACZ,cAAc;IACd,UAAU;IACV,YAAY,KAAA;IACZ,QAAQ,KAAA;IACR,SAAS,KAAA;GACX;GACA,KAAK,UAAU,IAAI,IAAI,KAAK;EAC9B;EACA,IACE,CAAC,KAAK,UAAU,MACb,SAAS,KAAK,SAAS,eAAe,KAAK,eAAe,EAC7D,GAEA,KAAK,UAAU,KAAK;GAAE,MAAM;GAAa,YAAY;EAAG,CAAC;EAE3D,MAAM,SAAS,KAAK,aAAa,OAAO;EACxC,MAAM,UAAU;EAChB,IAAI,eACF,MAAM,gBAAgB;EAExB,KAAK,yBAAyB;CAChC;CAEA,aAAqB,OAAwB;EAC3C,IAAI,CAAC,OAAO,OAAO;EACnB,IAAI;GACF,OAAO,KAAK,MAAM,KAAK;EACzB,QAAQ;GACN,OAAO;EACT;CACF;CAEA,OAAqB;EACnB,MAAM,WAAyC,CAAC;EAEhD,KAAK,MAAM,QAAQ,KAAK,WAAW;GACjC,IAAI,KAAK,SAAS,aAAa;IAC7B,IAAI,KAAK,cAAc;KACrB,MAAM,SAAS,KAAK,eAAe,IAAI,KAAK,GAAG,KAAK;KACpD,IAAI,OAAO,SAAS,KAAK,KAAK,uBAAuB,KAAK,KACxD,SAAS,KAAK;MAAE,MAAM;MAAa,MAAM;KAAO,CAAU;IAE9D;IACA;GACF;GAEA,IAAI,KAAK,SAAS,QAAQ;IACxB,MAAM,QAAQ,KAAK,UAAU,IAAI,KAAK,GAAG;IACzC,IAAI,OAAO,SACT,SAAS,KAAK;KAAE,MAAM;KAAQ,MAAM,MAAM;IAAO,CAAU;IAE7D;GACF;GAEA,MAAM,QAAQ,KAAK,UAAU,IAAI,KAAK,UAAU;GAChD,IAAI,CAAC,OAAO;GACZ,MAAM,WAAgC;IACpC,MAAM;IACN,YAAY,MAAM;IAClB,UAAU,MAAM;IAChB,MAAO,MAAM,cAAc,CAAC;IAC5B,UAAU,MAAM;IAChB,GAAI,MAAM,WAAW,KAAA,IAAY,EAAE,QAAQ,MAAM,OAAO,IAAI,CAAC;IAC7D,GAAI,MAAM,YAAY,KAAA,IAAY,EAAE,SAAS,MAAM,QAAQ,IAAI,CAAC;IAChE,GAAI,MAAM,oBACN,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,MAAM,kBAAkB,EAAE,EAAE,IACzD,CAAC;IACL,GAAI,MAAM,kBAAkB,EAAE,UAAU,MAAM,gBAAgB,IAAI,CAAC;IACnE,GAAI,MAAM,gBACN,EAAE,wBAAwB,MAAM,cAAc,IAC9C,CAAC;GACP;GACA,SAAS,KAAK,QAAQ;EACxB;EAEA,MAAM,SAAS,KAAK,UAAU;EAC9B,MAAM,WAAW;GACf,GAAI,SAAS,EAAE,OAAO,IAAI,CAAC;GAC3B,GAAI,KAAK,aACL,EACE,QAAQ,GACL,2BAA2B,EAC1B,YAAY,KAAK,WACnB,EACF,EACF,IACA,CAAC;EACP;EACA,MAAM,SAA6B;GACjC,SAAS;GACT,GAAI,KAAK,SAAS,EAAE,QAAQ,KAAK,OAAO,IAAI,KAAA;GAC5C,GAAI,OAAO,KAAK,QAAQ,CAAC,CAAC,SAAS,IAAI,EAAE,SAAS,IAAI,KAAA;EACxD;EACA,KAAK,WAAW,MAAM;CACxB;CAEA,mBAAiC;EAC/B,IACE,KAAK,mBAAmB,KAAA,KACxB,KAAK,oBAAoB,KAAA,GAEzB,KAAK,iBAAiB,KAAK,IAAI,IAAI,KAAK;CAE5C;CAEA,YAA+C;EAC7C,IAAI,KAAK,oBAAoB,KAAA,GAAW,OAAO,KAAA;EAG/C,MAAM,kBADM,KAAK,IACS,IAAI,KAAK;EACnC,MAAM,aACJ,KAAK,cAAc,IACf,KAAK,KACH,MAAM,KAAK,KAAK,UAAU,OAAO,CAAC,CAAC,CAAC,QACjC,KAAK,MAAM,MAAM,EAAE,OAAO,QAC3B,CACF,IAAI,CACN,IACA,KAAA;EACN,MAAM,kBACJ,cAAc,kBAAkB,IAC3B,aAAa,kBAAmB,MACjC,KAAA;EAEN,OAAO;GACL,iBAAiB,KAAK;GACtB,GAAI,KAAK,mBAAmB,KAAA,IACxB,EAAE,gBAAgB,KAAK,eAAe,IACtC,CAAC;GACL;GACA,GAAI,eAAe,KAAA,IAAY,EAAE,WAAW,IAAI,CAAC;GACjD,GAAI,oBAAoB,KAAA,IAAY,EAAE,gBAAgB,IAAI,CAAC;GAC3D,aAAa,KAAK;GAClB,eAAe,KAAK,UAAU;EAChC;CACF;CAEA,qBAA6B,WAA0B;EACrD,IAAI,CAAC,KAAK,cAAc;EAGxB,KAAK,sBAAsB,KAAA;EAC3B,MAAM,MAAM,aAAa,oBAAoB,EAAE,KAAK;EACpD,IAAI,CAAC,KAAK,eAAe,IAAI,GAAG,GAAG;GACjC,KAAK,eAAe,IAAI,KAAK,EAAE;GAC/B,KAAK,UAAU,KAAK;IAAE,MAAM;IAAa;GAAI,CAAC;EAChD;EACA,KAAK,qBAAqB;EAC1B,KAAK,KAAK;CACZ;CAEA,uBAA+B,OAAe,WAA0B;EACtE,IAAI,CAAC,KAAK,gBAAgB,CAAC,OAAO;EAClC,IAAI,CAAC,KAAK,oBAER,KAAK,qBAAqB,SAAS;EAErC,MAAM,MAAM,KAAK;EACjB,IAAI,CAAC,KAAK;EACV,KAAK,eAAe,IAAI,MAAM,KAAK,eAAe,IAAI,GAAG,KAAK,MAAM,KAAK;EACzE,KAAK,KAAK;CACZ;CAEA,qBAAmC;EACjC,IAAI,CAAC,KAAK,cAAc;EACxB,KAAK,qBAAqB,KAAA;EAC1B,KAAK,KAAK;CACZ;AACF"}
+{"version":3,"file":"run-aggregator.js","names":[],"sources":["../../../src/runtime/adapter/run-aggregator.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  isMcpAppUri,\n  type ChatModelRunResult,\n  type MessageTiming,\n  type ThreadAssistantMessagePart,\n  type ToolCallMessagePart,\n} from \"@assistant-ui/core\";\nimport type { AgUiEvent, AgUiInterrupt } from \"../types\";\nimport type { Logger } from \"../logger\";\n\nexport const AG_UI_METADATA_NAMESPACE = \"agui\";\n\nexport type AgUiCustomMetadata = {\n  interrupts?: AgUiInterrupt[];\n};\n\ntype Emit = (update: ChatModelRunResult) => void;\n\ntype ToolCallState = {\n  toolCallId: string;\n  toolCallName: string;\n  argsText: string;\n  parsedArgs: Record<string, unknown> | undefined;\n  result: unknown;\n  isError: boolean | undefined;\n  parentMessageId?: string;\n  toolMessageId?: string;\n  mcpAppResourceUri?: string;\n};\n\nconst MCP_APPS_ACTIVITY_TYPE = \"mcp-apps\";\n\nexport type RunAggregatorOptions = {\n  showThinking: boolean;\n  logger: Logger;\n  emit: Emit;\n  onServerMessageId?: (messageId: string) => void;\n};\n\n/**\n * Collects AG-UI events into assistant-ui run snapshots that can be yielded from a ChatModelAdapter.\n *\n * The aggregator keeps a single assistant message worth of parts. Each incoming event updates the parts and\n * emits a fresh snapshot through the provided `emit` callback.\n */\nexport class RunAggregator {\n  private readonly emitUpdate: Emit;\n  private readonly showThinking: boolean;\n  private readonly logger: Logger;\n  private readonly onServerMessageId: ((messageId: string) => void) | undefined;\n\n  private status: ChatModelRunResult[\"status\"] | undefined;\n  private interrupts: AgUiInterrupt[] | undefined;\n  private readonly textParts = new Map<\n    string,\n    { buffer: string; touched: boolean }\n  >();\n  private activeTextMessageId: string | undefined;\n  private readonly reasoningParts = new Map<string, string>(); // key → buffer\n  private activeReasoningKey: string | undefined;\n  private reasoningPartCounter = 0;\n  private readonly toolCalls = new Map<string, ToolCallState>();\n  private lastResolvedToolCallId: string | undefined;\n  private readonly partOrder: (\n    | { kind: \"text\"; key: string }\n    | { kind: \"reasoning\"; key: string }\n    | { kind: \"tool-call\"; toolCallId: string }\n  )[] = [];\n  private textPartCounter = 0;\n  private serverMessageIdReported = false;\n\n  private streamStartTime: number | undefined;\n  private firstTokenTime: number | undefined;\n  private totalChunks = 0;\n\n  constructor(options: RunAggregatorOptions) {\n    this.emitUpdate = options.emit;\n    this.showThinking = options.showThinking;\n    this.logger = options.logger;\n    this.onServerMessageId = options.onServerMessageId;\n  }\n\n  hasToolCall(toolCallId: string): boolean {\n    return this.toolCalls.has(toolCallId);\n  }\n\n  handle(event: AgUiEvent): void {\n    switch (event.type) {\n      case \"RUN_STARTED\": {\n        this.clearTextParts();\n        this.reasoningParts.clear();\n        this.activeReasoningKey = undefined;\n        this.reasoningPartCounter = 0;\n        this.toolCalls.clear();\n        this.lastResolvedToolCallId = undefined;\n        this.partOrder.length = 0;\n        this.textPartCounter = 0;\n        this.activeTextMessageId = undefined;\n        this.interrupts = undefined;\n        this.serverMessageIdReported = false;\n        this.streamStartTime = Date.now();\n        this.firstTokenTime = undefined;\n        this.totalChunks = 0;\n        this.status = { type: \"running\" };\n        this.emit();\n        break;\n      }\n      case \"RUN_FINISHED\": {\n        if (event.outcome?.type === \"interrupt\") {\n          this.interrupts = event.outcome.interrupts;\n          this.status = { type: \"requires-action\", reason: \"interrupt\" };\n          this.emit();\n          break;\n        }\n\n        this.interrupts = undefined;\n        const hasUnresolvedToolCalls = Array.from(this.toolCalls.values()).some(\n          (tc) => tc.result === undefined,\n        );\n\n        this.status =\n          event.outcome?.type === \"success\" || !hasUnresolvedToolCalls\n            ? { type: \"complete\", reason: \"unknown\" }\n            : { type: \"requires-action\", reason: \"tool-calls\" };\n        this.emit();\n        break;\n      }\n      case \"RUN_ERROR\": {\n        this.status = {\n          type: \"incomplete\",\n          reason: \"error\",\n          ...(event.message !== undefined ? { error: event.message } : {}),\n        };\n        this.emit();\n        break;\n      }\n      case \"RUN_CANCELLED\": {\n        this.status = { type: \"incomplete\", reason: \"cancelled\" };\n        this.emit();\n        break;\n      }\n\n      case \"TEXT_MESSAGE_START\": {\n        this.reportServerMessageId(event.messageId);\n        const id = this.startTextMessage(event.messageId);\n        if (id) {\n          this.markTextPartTouched(id);\n        }\n        this.emit();\n        break;\n      }\n      case \"TEXT_MESSAGE_CONTENT\":\n      case \"TEXT_MESSAGE_CHUNK\": {\n        const incomingId = \"messageId\" in event ? event.messageId : undefined;\n        this.reportServerMessageId(incomingId);\n        if (!event.delta) break;\n        this.recordFirstToken();\n        const id = this.resolveTextMessageId(incomingId);\n        this.appendText(id, event.delta);\n        this.totalChunks++;\n        this.emit();\n        break;\n      }\n      case \"TEXT_MESSAGE_END\": {\n        this.reportServerMessageId(event.messageId);\n        if (event.messageId && this.activeTextMessageId === event.messageId) {\n          this.activeTextMessageId = undefined;\n        }\n        this.emit();\n        break;\n      }\n\n      case \"THINKING_START\":\n      case \"THINKING_TEXT_MESSAGE_START\":\n      case \"REASONING_START\":\n      case \"REASONING_MESSAGE_START\":\n        this.handleReasoningStart(\n          \"messageId\" in event ? event.messageId : undefined,\n        );\n        break;\n      case \"THINKING_TEXT_MESSAGE_CONTENT\":\n        this.handleReasoningContent(event.delta);\n        this.totalChunks++;\n        this.recordFirstToken();\n        break;\n      case \"REASONING_MESSAGE_CONTENT\":\n        this.handleReasoningContent(\n          event.delta,\n          \"messageId\" in event ? event.messageId : undefined,\n        );\n        this.totalChunks++;\n        this.recordFirstToken();\n        break;\n      case \"THINKING_TEXT_MESSAGE_END\":\n      case \"THINKING_END\":\n      case \"REASONING_MESSAGE_END\":\n      case \"REASONING_END\":\n        this.handleReasoningEnd();\n        break;\n\n      case \"TOOL_CALL_START\": {\n        this.reportServerMessageId(event.parentMessageId);\n        this.startToolCall(\n          event.toolCallId,\n          event.toolCallName,\n          event.parentMessageId,\n        );\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_ARGS\":\n      case \"TOOL_CALL_CHUNK\": {\n        if (event.type === \"TOOL_CALL_CHUNK\") {\n          this.reportServerMessageId(event.parentMessageId);\n        }\n        if (!event.delta) break;\n        this.appendToolArgs(event.toolCallId, event.delta);\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_END\": {\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_RESULT\": {\n        this.finishToolCall(\n          event.toolCallId,\n          event.content ?? \"\",\n          event.role === \"tool\" ? false : undefined,\n          event.messageId,\n        );\n        this.emit();\n        break;\n      }\n      case \"ACTIVITY_SNAPSHOT\": {\n        if (event.activityType !== MCP_APPS_ACTIVITY_TYPE) break;\n        const id = this.lastResolvedToolCallId;\n        const entry = id ? this.toolCalls.get(id) : undefined;\n        const resourceUri = event.content.resourceUri;\n        if (\n          entry &&\n          typeof resourceUri === \"string\" &&\n          isMcpAppUri(resourceUri)\n        ) {\n          entry.mcpAppResourceUri = resourceUri;\n          this.emit();\n        }\n        break;\n      }\n\n      default: {\n        this.logger.debug?.(\"[agui] aggregator ignored event\", event);\n      }\n    }\n  }\n\n  private reportServerMessageId(messageId: string | undefined): void {\n    if (this.serverMessageIdReported || !messageId) return;\n    this.serverMessageIdReported = true;\n    this.onServerMessageId?.(messageId);\n  }\n\n  private clearTextParts(): void {\n    this.textParts.clear();\n  }\n\n  private generateTextKey(): string {\n    this.textPartCounter += 1;\n    return `text-${this.textPartCounter}`;\n  }\n\n  private startTextMessage(messageId?: string): string {\n    const id = messageId ?? this.generateTextKey();\n    this.ensureTextPart(id);\n    this.activeTextMessageId = id;\n    return id;\n  }\n\n  private resolveTextMessageId(messageId?: string): string {\n    if (messageId) {\n      this.ensureTextPart(messageId);\n      this.activeTextMessageId = messageId;\n      return messageId;\n    }\n\n    if (this.activeTextMessageId) {\n      return this.activeTextMessageId;\n    }\n\n    const generated = this.generateTextKey();\n    this.ensureTextPart(generated);\n    this.activeTextMessageId = generated;\n    return generated;\n  }\n\n  private ensureTextPart(id: string): void {\n    if (!this.textParts.has(id)) {\n      this.textParts.set(id, { buffer: \"\", touched: false });\n      if (\n        !this.partOrder.some((part) => part.kind === \"text\" && part.key === id)\n      ) {\n        this.partOrder.push({ kind: \"text\", key: id });\n      }\n    }\n  }\n\n  private markTextPartTouched(id: string): void {\n    const entry = this.textParts.get(id);\n    if (!entry) return;\n    entry.touched = true;\n  }\n\n  private appendText(id: string, delta: string): void {\n    this.ensureTextPart(id);\n    const entry = this.textParts.get(id);\n    if (!entry) return;\n    entry.buffer += delta;\n    entry.touched = true;\n  }\n\n  private startToolCall(\n    id: string | undefined,\n    name?: string,\n    parentMessageId?: string,\n  ) {\n    if (!id) return;\n    // A new tool call acts as a boundary: any anonymous text that arrives\n    // after it should be a new part, not appended to the pre-tool text.\n    this.activeTextMessageId = undefined;\n    if (\n      !this.partOrder.some(\n        (part) => part.kind === \"tool-call\" && part.toolCallId === id,\n      )\n    ) {\n      this.partOrder.push({ kind: \"tool-call\", toolCallId: id });\n    }\n    const state: ToolCallState = {\n      toolCallId: id,\n      toolCallName: name ?? \"tool\",\n      argsText: \"\",\n      parsedArgs: undefined,\n      result: undefined,\n      isError: undefined,\n    };\n    if (parentMessageId) {\n      state.parentMessageId = parentMessageId;\n    }\n    this.toolCalls.set(id, state);\n  }\n\n  private appendToolArgs(id: string | undefined, delta: string) {\n    const entry = id ? this.toolCalls.get(id) : undefined;\n    if (!entry) return;\n    entry.argsText += delta;\n    try {\n      const parsed = JSON.parse(entry.argsText);\n      if (parsed && typeof parsed === \"object\") {\n        entry.parsedArgs = parsed as Record<string, unknown>;\n      } else {\n        entry.parsedArgs = undefined;\n      }\n    } catch {\n      entry.parsedArgs = undefined;\n    }\n  }\n\n  private finishToolCall(\n    id: string,\n    content: string,\n    isError?: boolean,\n    toolMessageId?: string,\n  ) {\n    if (!id) return;\n    let entry = this.toolCalls.get(id);\n    if (!entry) {\n      entry = {\n        toolCallId: id,\n        toolCallName: \"tool\",\n        argsText: \"\",\n        parsedArgs: undefined,\n        result: undefined,\n        isError: undefined,\n      };\n      this.toolCalls.set(id, entry);\n    }\n    if (\n      !this.partOrder.some(\n        (part) => part.kind === \"tool-call\" && part.toolCallId === id,\n      )\n    ) {\n      this.partOrder.push({ kind: \"tool-call\", toolCallId: id });\n    }\n    entry.result = tryParseJSON(content);\n    entry.isError = isError;\n    if (toolMessageId) {\n      entry.toolMessageId = toolMessageId;\n    }\n    this.lastResolvedToolCallId = id;\n  }\n\n  private emit(): void {\n    const snapshot: ThreadAssistantMessagePart[] = [];\n\n    for (const part of this.partOrder) {\n      if (part.kind === \"reasoning\") {\n        if (this.showThinking) {\n          const buffer = this.reasoningParts.get(part.key) ?? \"\";\n          if (buffer.length > 0 || this.activeReasoningKey === part.key) {\n            snapshot.push({ type: \"reasoning\", text: buffer } as const);\n          }\n        }\n        continue;\n      }\n\n      if (part.kind === \"text\") {\n        const entry = this.textParts.get(part.key);\n        if (entry?.touched) {\n          snapshot.push({ type: \"text\", text: entry.buffer } as const);\n        }\n        continue;\n      }\n\n      const entry = this.toolCalls.get(part.toolCallId);\n      if (!entry) continue;\n      const toolPart: ToolCallMessagePart = {\n        type: \"tool-call\",\n        toolCallId: entry.toolCallId,\n        toolName: entry.toolCallName,\n        args: (entry.parsedArgs ?? {}) as any,\n        argsText: entry.argsText,\n        ...(entry.result !== undefined ? { result: entry.result } : {}),\n        ...(entry.isError !== undefined ? { isError: entry.isError } : {}),\n        ...(entry.mcpAppResourceUri\n          ? { mcp: { app: { resourceUri: entry.mcpAppResourceUri } } }\n          : {}),\n        ...(entry.parentMessageId ? { parentId: entry.parentMessageId } : {}),\n        ...(entry.toolMessageId\n          ? { unstable_toolMessageId: entry.toolMessageId }\n          : {}),\n      } as ToolCallMessagePart & { unstable_toolMessageId?: string };\n      snapshot.push(toolPart);\n    }\n\n    const timing = this.getTiming();\n    const metadata = {\n      ...(timing ? { timing } : {}),\n      ...(this.interrupts\n        ? {\n            custom: {\n              [AG_UI_METADATA_NAMESPACE]: {\n                interrupts: this.interrupts,\n              } satisfies AgUiCustomMetadata,\n            },\n          }\n        : {}),\n    };\n    const result: ChatModelRunResult = {\n      content: snapshot,\n      ...(this.status ? { status: this.status } : undefined),\n      ...(Object.keys(metadata).length > 0 ? { metadata } : undefined),\n    };\n    this.emitUpdate(result);\n  }\n\n  private recordFirstToken(): void {\n    if (\n      this.firstTokenTime === undefined &&\n      this.streamStartTime !== undefined\n    ) {\n      this.firstTokenTime = Date.now() - this.streamStartTime;\n    }\n  }\n\n  private getTiming(): MessageTiming | undefined {\n    if (this.streamStartTime === undefined) return undefined;\n\n    const now = Date.now();\n    const totalStreamTime = now - this.streamStartTime;\n    const tokenCount =\n      this.totalChunks > 0\n        ? Math.ceil(\n            Array.from(this.textParts.values()).reduce(\n              (sum, p) => sum + p.buffer.length,\n              0,\n            ) / 4,\n          )\n        : undefined;\n    const tokensPerSecond =\n      tokenCount && totalStreamTime > 0\n        ? (tokenCount / totalStreamTime) * 1000\n        : undefined;\n\n    return {\n      streamStartTime: this.streamStartTime,\n      ...(this.firstTokenTime !== undefined\n        ? { firstTokenTime: this.firstTokenTime }\n        : {}),\n      totalStreamTime,\n      ...(tokenCount !== undefined ? { tokenCount } : {}),\n      ...(tokensPerSecond !== undefined ? { tokensPerSecond } : {}),\n      totalChunks: this.totalChunks,\n      toolCallCount: this.toolCalls.size,\n    };\n  }\n\n  private handleReasoningStart(messageId?: string): void {\n    if (!this.showThinking) return;\n    // A reasoning block acts as a boundary: anonymous text arriving after it\n    // should be a new part, not appended to any pre-reasoning text.\n    this.activeTextMessageId = undefined;\n    const key = messageId ?? `__auto-reasoning-${++this.reasoningPartCounter}`;\n    if (!this.reasoningParts.has(key)) {\n      this.reasoningParts.set(key, \"\");\n      this.partOrder.push({ kind: \"reasoning\", key });\n    }\n    this.activeReasoningKey = key;\n    this.emit();\n  }\n\n  private handleReasoningContent(delta: string, messageId?: string): void {\n    if (!this.showThinking || !delta) return;\n    if (!this.activeReasoningKey) {\n      // Content arrived without a preceding START — create the slot lazily.\n      this.handleReasoningStart(messageId);\n    }\n    const key = this.activeReasoningKey;\n    if (!key) return;\n    this.reasoningParts.set(key, (this.reasoningParts.get(key) ?? \"\") + delta);\n    this.emit();\n  }\n\n  private handleReasoningEnd(): void {\n    if (!this.showThinking) return;\n    this.activeReasoningKey = undefined;\n    this.emit();\n  }\n}\n\nexport function tryParseJSON(value: string): unknown {\n  if (!value) return value;\n  try {\n    return JSON.parse(value);\n  } catch {\n    return value;\n  }\n}\n"],"mappings":";;;AAYA,MAAa,2BAA2B;AAoBxC,MAAM,yBAAyB;;;;;;;AAe/B,IAAa,gBAAb,MAA2B;CACzB;CACA;CACA;CACA;CAEA;CACA;CACA,4BAA6B,IAAI,IAG/B;CACF;CACA,iCAAkC,IAAI,IAAoB;CAC1D;CACA,uBAA+B;CAC/B,4BAA6B,IAAI,IAA2B;CAC5D;CACA,YAIM,CAAC;CACP,kBAA0B;CAC1B,0BAAkC;CAElC;CACA;CACA,cAAsB;CAEtB,YAAY,SAA+B;EACzC,KAAK,aAAa,QAAQ;EAC1B,KAAK,eAAe,QAAQ;EAC5B,KAAK,SAAS,QAAQ;EACtB,KAAK,oBAAoB,QAAQ;CACnC;CAEA,YAAY,YAA6B;EACvC,OAAO,KAAK,UAAU,IAAI,UAAU;CACtC;CAEA,OAAO,OAAwB;EAC7B,QAAQ,MAAM,MAAd;GACE,KAAK;IACH,KAAK,eAAe;IACpB,KAAK,eAAe,MAAM;IAC1B,KAAK,qBAAqB,KAAA;IAC1B,KAAK,uBAAuB;IAC5B,KAAK,UAAU,MAAM;IACrB,KAAK,yBAAyB,KAAA;IAC9B,KAAK,UAAU,SAAS;IACxB,KAAK,kBAAkB;IACvB,KAAK,sBAAsB,KAAA;IAC3B,KAAK,aAAa,KAAA;IAClB,KAAK,0BAA0B;IAC/B,KAAK,kBAAkB,KAAK,IAAI;IAChC,KAAK,iBAAiB,KAAA;IACtB,KAAK,cAAc;IACnB,KAAK,SAAS,EAAE,MAAM,UAAU;IAChC,KAAK,KAAK;IACV;GAEF,KAAK,gBAAgB;IACnB,IAAI,MAAM,SAAS,SAAS,aAAa;KACvC,KAAK,aAAa,MAAM,QAAQ;KAChC,KAAK,SAAS;MAAE,MAAM;MAAmB,QAAQ;KAAY;KAC7D,KAAK,KAAK;KACV;IACF;IAEA,KAAK,aAAa,KAAA;IAClB,MAAM,yBAAyB,MAAM,KAAK,KAAK,UAAU,OAAO,CAAC,CAAC,CAAC,MAChE,OAAO,GAAG,WAAW,KAAA,CACxB;IAEA,KAAK,SACH,MAAM,SAAS,SAAS,aAAa,CAAC,yBAClC;KAAE,MAAM;KAAY,QAAQ;IAAU,IACtC;KAAE,MAAM;KAAmB,QAAQ;IAAa;IACtD,KAAK,KAAK;IACV;GACF;GACA,KAAK;IACH,KAAK,SAAS;KACZ,MAAM;KACN,QAAQ;KACR,GAAI,MAAM,YAAY,KAAA,IAAY,EAAE,OAAO,MAAM,QAAQ,IAAI,CAAC;IAChE;IACA,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,SAAS;KAAE,MAAM;KAAc,QAAQ;IAAY;IACxD,KAAK,KAAK;IACV;GAGF,KAAK,sBAAsB;IACzB,KAAK,sBAAsB,MAAM,SAAS;IAC1C,MAAM,KAAK,KAAK,iBAAiB,MAAM,SAAS;IAChD,IAAI,IACF,KAAK,oBAAoB,EAAE;IAE7B,KAAK,KAAK;IACV;GACF;GACA,KAAK;GACL,KAAK,sBAAsB;IACzB,MAAM,aAAa,eAAe,QAAQ,MAAM,YAAY,KAAA;IAC5D,KAAK,sBAAsB,UAAU;IACrC,IAAI,CAAC,MAAM,OAAO;IAClB,KAAK,iBAAiB;IACtB,MAAM,KAAK,KAAK,qBAAqB,UAAU;IAC/C,KAAK,WAAW,IAAI,MAAM,KAAK;IAC/B,KAAK;IACL,KAAK,KAAK;IACV;GACF;GACA,KAAK;IACH,KAAK,sBAAsB,MAAM,SAAS;IAC1C,IAAI,MAAM,aAAa,KAAK,wBAAwB,MAAM,WACxD,KAAK,sBAAsB,KAAA;IAE7B,KAAK,KAAK;IACV;GAGF,KAAK;GACL,KAAK;GACL,KAAK;GACL,KAAK;IACH,KAAK,qBACH,eAAe,QAAQ,MAAM,YAAY,KAAA,CAC3C;IACA;GACF,KAAK;IACH,KAAK,uBAAuB,MAAM,KAAK;IACvC,KAAK;IACL,KAAK,iBAAiB;IACtB;GACF,KAAK;IACH,KAAK,uBACH,MAAM,OACN,eAAe,QAAQ,MAAM,YAAY,KAAA,CAC3C;IACA,KAAK;IACL,KAAK,iBAAiB;IACtB;GACF,KAAK;GACL,KAAK;GACL,KAAK;GACL,KAAK;IACH,KAAK,mBAAmB;IACxB;GAEF,KAAK;IACH,KAAK,sBAAsB,MAAM,eAAe;IAChD,KAAK,cACH,MAAM,YACN,MAAM,cACN,MAAM,eACR;IACA,KAAK,KAAK;IACV;GAEF,KAAK;GACL,KAAK;IACH,IAAI,MAAM,SAAS,mBACjB,KAAK,sBAAsB,MAAM,eAAe;IAElD,IAAI,CAAC,MAAM,OAAO;IAClB,KAAK,eAAe,MAAM,YAAY,MAAM,KAAK;IACjD,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,eACH,MAAM,YACN,MAAM,WAAW,IACjB,MAAM,SAAS,SAAS,QAAQ,KAAA,GAChC,MAAM,SACR;IACA,KAAK,KAAK;IACV;GAEF,KAAK,qBAAqB;IACxB,IAAI,MAAM,iBAAiB,wBAAwB;IACnD,MAAM,KAAK,KAAK;IAChB,MAAM,QAAQ,KAAK,KAAK,UAAU,IAAI,EAAE,IAAI,KAAA;IAC5C,MAAM,cAAc,MAAM,QAAQ;IAClC,IACE,SACA,OAAO,gBAAgB,YACvB,YAAY,WAAW,GACvB;KACA,MAAM,oBAAoB;KAC1B,KAAK,KAAK;IACZ;IACA;GACF;GAEA,SACE,KAAK,OAAO,QAAQ,mCAAmC,KAAK;EAEhE;CACF;CAEA,sBAA8B,WAAqC;EACjE,IAAI,KAAK,2BAA2B,CAAC,WAAW;EAChD,KAAK,0BAA0B;EAC/B,KAAK,oBAAoB,SAAS;CACpC;CAEA,iBAA+B;EAC7B,KAAK,UAAU,MAAM;CACvB;CAEA,kBAAkC;EAChC,KAAK,mBAAmB;EACxB,OAAO,QAAQ,KAAK;CACtB;CAEA,iBAAyB,WAA4B;EACnD,MAAM,KAAK,aAAa,KAAK,gBAAgB;EAC7C,KAAK,eAAe,EAAE;EACtB,KAAK,sBAAsB;EAC3B,OAAO;CACT;CAEA,qBAA6B,WAA4B;EACvD,IAAI,WAAW;GACb,KAAK,eAAe,SAAS;GAC7B,KAAK,sBAAsB;GAC3B,OAAO;EACT;EAEA,IAAI,KAAK,qBACP,OAAO,KAAK;EAGd,MAAM,YAAY,KAAK,gBAAgB;EACvC,KAAK,eAAe,SAAS;EAC7B,KAAK,sBAAsB;EAC3B,OAAO;CACT;CAEA,eAAuB,IAAkB;EACvC,IAAI,CAAC,KAAK,UAAU,IAAI,EAAE,GAAG;GAC3B,KAAK,UAAU,IAAI,IAAI;IAAE,QAAQ;IAAI,SAAS;GAAM,CAAC;GACrD,IACE,CAAC,KAAK,UAAU,MAAM,SAAS,KAAK,SAAS,UAAU,KAAK,QAAQ,EAAE,GAEtE,KAAK,UAAU,KAAK;IAAE,MAAM;IAAQ,KAAK;GAAG,CAAC;EAEjD;CACF;CAEA,oBAA4B,IAAkB;EAC5C,MAAM,QAAQ,KAAK,UAAU,IAAI,EAAE;EACnC,IAAI,CAAC,OAAO;EACZ,MAAM,UAAU;CAClB;CAEA,WAAmB,IAAY,OAAqB;EAClD,KAAK,eAAe,EAAE;EACtB,MAAM,QAAQ,KAAK,UAAU,IAAI,EAAE;EACnC,IAAI,CAAC,OAAO;EACZ,MAAM,UAAU;EAChB,MAAM,UAAU;CAClB;CAEA,cACE,IACA,MACA,iBACA;EACA,IAAI,CAAC,IAAI;EAGT,KAAK,sBAAsB,KAAA;EAC3B,IACE,CAAC,KAAK,UAAU,MACb,SAAS,KAAK,SAAS,eAAe,KAAK,eAAe,EAC7D,GAEA,KAAK,UAAU,KAAK;GAAE,MAAM;GAAa,YAAY;EAAG,CAAC;EAE3D,MAAM,QAAuB;GAC3B,YAAY;GACZ,cAAc,QAAQ;GACtB,UAAU;GACV,YAAY,KAAA;GACZ,QAAQ,KAAA;GACR,SAAS,KAAA;EACX;EACA,IAAI,iBACF,MAAM,kBAAkB;EAE1B,KAAK,UAAU,IAAI,IAAI,KAAK;CAC9B;CAEA,eAAuB,IAAwB,OAAe;EAC5D,MAAM,QAAQ,KAAK,KAAK,UAAU,IAAI,EAAE,IAAI,KAAA;EAC5C,IAAI,CAAC,OAAO;EACZ,MAAM,YAAY;EAClB,IAAI;GACF,MAAM,SAAS,KAAK,MAAM,MAAM,QAAQ;GACxC,IAAI,UAAU,OAAO,WAAW,UAC9B,MAAM,aAAa;QAEnB,MAAM,aAAa,KAAA;EAEvB,QAAQ;GACN,MAAM,aAAa,KAAA;EACrB;CACF;CAEA,eACE,IACA,SACA,SACA,eACA;EACA,IAAI,CAAC,IAAI;EACT,IAAI,QAAQ,KAAK,UAAU,IAAI,EAAE;EACjC,IAAI,CAAC,OAAO;GACV,QAAQ;IACN,YAAY;IACZ,cAAc;IACd,UAAU;IACV,YAAY,KAAA;IACZ,QAAQ,KAAA;IACR,SAAS,KAAA;GACX;GACA,KAAK,UAAU,IAAI,IAAI,KAAK;EAC9B;EACA,IACE,CAAC,KAAK,UAAU,MACb,SAAS,KAAK,SAAS,eAAe,KAAK,eAAe,EAC7D,GAEA,KAAK,UAAU,KAAK;GAAE,MAAM;GAAa,YAAY;EAAG,CAAC;EAE3D,MAAM,SAAS,aAAa,OAAO;EACnC,MAAM,UAAU;EAChB,IAAI,eACF,MAAM,gBAAgB;EAExB,KAAK,yBAAyB;CAChC;CAEA,OAAqB;EACnB,MAAM,WAAyC,CAAC;EAEhD,KAAK,MAAM,QAAQ,KAAK,WAAW;GACjC,IAAI,KAAK,SAAS,aAAa;IAC7B,IAAI,KAAK,cAAc;KACrB,MAAM,SAAS,KAAK,eAAe,IAAI,KAAK,GAAG,KAAK;KACpD,IAAI,OAAO,SAAS,KAAK,KAAK,uBAAuB,KAAK,KACxD,SAAS,KAAK;MAAE,MAAM;MAAa,MAAM;KAAO,CAAU;IAE9D;IACA;GACF;GAEA,IAAI,KAAK,SAAS,QAAQ;IACxB,MAAM,QAAQ,KAAK,UAAU,IAAI,KAAK,GAAG;IACzC,IAAI,OAAO,SACT,SAAS,KAAK;KAAE,MAAM;KAAQ,MAAM,MAAM;IAAO,CAAU;IAE7D;GACF;GAEA,MAAM,QAAQ,KAAK,UAAU,IAAI,KAAK,UAAU;GAChD,IAAI,CAAC,OAAO;GACZ,MAAM,WAAgC;IACpC,MAAM;IACN,YAAY,MAAM;IAClB,UAAU,MAAM;IAChB,MAAO,MAAM,cAAc,CAAC;IAC5B,UAAU,MAAM;IAChB,GAAI,MAAM,WAAW,KAAA,IAAY,EAAE,QAAQ,MAAM,OAAO,IAAI,CAAC;IAC7D,GAAI,MAAM,YAAY,KAAA,IAAY,EAAE,SAAS,MAAM,QAAQ,IAAI,CAAC;IAChE,GAAI,MAAM,oBACN,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,MAAM,kBAAkB,EAAE,EAAE,IACzD,CAAC;IACL,GAAI,MAAM,kBAAkB,EAAE,UAAU,MAAM,gBAAgB,IAAI,CAAC;IACnE,GAAI,MAAM,gBACN,EAAE,wBAAwB,MAAM,cAAc,IAC9C,CAAC;GACP;GACA,SAAS,KAAK,QAAQ;EACxB;EAEA,MAAM,SAAS,KAAK,UAAU;EAC9B,MAAM,WAAW;GACf,GAAI,SAAS,EAAE,OAAO,IAAI,CAAC;GAC3B,GAAI,KAAK,aACL,EACE,QAAQ,GACL,2BAA2B,EAC1B,YAAY,KAAK,WACnB,EACF,EACF,IACA,CAAC;EACP;EACA,MAAM,SAA6B;GACjC,SAAS;GACT,GAAI,KAAK,SAAS,EAAE,QAAQ,KAAK,OAAO,IAAI,KAAA;GAC5C,GAAI,OAAO,KAAK,QAAQ,CAAC,CAAC,SAAS,IAAI,EAAE,SAAS,IAAI,KAAA;EACxD;EACA,KAAK,WAAW,MAAM;CACxB;CAEA,mBAAiC;EAC/B,IACE,KAAK,mBAAmB,KAAA,KACxB,KAAK,oBAAoB,KAAA,GAEzB,KAAK,iBAAiB,KAAK,IAAI,IAAI,KAAK;CAE5C;CAEA,YAA+C;EAC7C,IAAI,KAAK,oBAAoB,KAAA,GAAW,OAAO,KAAA;EAG/C,MAAM,kBADM,KAAK,IACS,IAAI,KAAK;EACnC,MAAM,aACJ,KAAK,cAAc,IACf,KAAK,KACH,MAAM,KAAK,KAAK,UAAU,OAAO,CAAC,CAAC,CAAC,QACjC,KAAK,MAAM,MAAM,EAAE,OAAO,QAC3B,CACF,IAAI,CACN,IACA,KAAA;EACN,MAAM,kBACJ,cAAc,kBAAkB,IAC3B,aAAa,kBAAmB,MACjC,KAAA;EAEN,OAAO;GACL,iBAAiB,KAAK;GACtB,GAAI,KAAK,mBAAmB,KAAA,IACxB,EAAE,gBAAgB,KAAK,eAAe,IACtC,CAAC;GACL;GACA,GAAI,eAAe,KAAA,IAAY,EAAE,WAAW,IAAI,CAAC;GACjD,GAAI,oBAAoB,KAAA,IAAY,EAAE,gBAAgB,IAAI,CAAC;GAC3D,aAAa,KAAK;GAClB,eAAe,KAAK,UAAU;EAChC;CACF;CAEA,qBAA6B,WAA0B;EACrD,IAAI,CAAC,KAAK,cAAc;EAGxB,KAAK,sBAAsB,KAAA;EAC3B,MAAM,MAAM,aAAa,oBAAoB,EAAE,KAAK;EACpD,IAAI,CAAC,KAAK,eAAe,IAAI,GAAG,GAAG;GACjC,KAAK,eAAe,IAAI,KAAK,EAAE;GAC/B,KAAK,UAAU,KAAK;IAAE,MAAM;IAAa;GAAI,CAAC;EAChD;EACA,KAAK,qBAAqB;EAC1B,KAAK,KAAK;CACZ;CAEA,uBAA+B,OAAe,WAA0B;EACtE,IAAI,CAAC,KAAK,gBAAgB,CAAC,OAAO;EAClC,IAAI,CAAC,KAAK,oBAER,KAAK,qBAAqB,SAAS;EAErC,MAAM,MAAM,KAAK;EACjB,IAAI,CAAC,KAAK;EACV,KAAK,eAAe,IAAI,MAAM,KAAK,eAAe,IAAI,GAAG,KAAK,MAAM,KAAK;EACzE,KAAK,KAAK;CACZ;CAEA,qBAAmC;EACjC,IAAI,CAAC,KAAK,cAAc;EACxB,KAAK,qBAAqB,KAAA;EAC1B,KAAK,KAAK;CACZ;AACF;AAEA,SAAgB,aAAa,OAAwB;CACnD,IAAI,CAAC,OAAO,OAAO;CACnB,IAAI;EACF,OAAO,KAAK,MAAM,KAAK;CACzB,QAAQ;EACN,OAAO;CACT;AACF"}

package/dist/runtime/adapter/subscriber.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"subscriber.d.ts","names":[],"sources":["../../../src/runtime/adapter/subscriber.ts"],"mappings":";;;;KAMK,QAAA,IAAY,KAAgB,EAAT,SAAS;AAAA,KAE5B,UAAA;EACH,OAAA,IAAW,OAAA;IAAW,KAAA;EAAA;EACtB,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,yBAAA,IAA6B,OAAA;IAAW,KAAA;EAAA;EACxC,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,+BAAA,IAAmC,OAAA;IAAW,KAAA;EAAA;EAC9C,iCAAA,IAAqC,OAAA;IAAW,KAAA;EAAA;EAChD,6BAAA,IAAiC,OAAA;IAAW,KAAA;EAAA;EAC5C,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,mBAAA,IAAuB,OAAA;IAAW,KAAA;EAAA;EAClC,4BAAA,IAAgC,OAAA;IAAW,KAAA;EAAA;EAC3C,8BAAA,IAAkC,OAAA;IAAW,KAAA;EAAA;EAC7C,0BAAA,IAA8B,OAAA;IAAW,KAAA;EAAA;EACzC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,mBAAA,IAAuB,OAAA;IAAW,KAAA;EAAA;EAClC,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,iBAAA,IAAqB,OAAA;IAAW,KAAA;EAAA;EAChC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,aAAA,IAAiB,OAAA;IAAW,KAAA;EAAA;EAC5B,UAAA,IAAc,OAAA;IAAW,KAAA;EAAA;EACzB,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,cAAA;EACA,WAAA,IAAe,OAAA;IAAW,KAAA,EAAO,KAAK;EAAA;AAAA;AAAA,KAuBnC,iBAAA;EACH,QAAA,EAAU,QAAA;EACV,KAAA;EACA,MAAA,GAAS,MAAA;EACT,WAAA,IAAe,KAAA,EAAO,KAAA;AAAA;AAAA,cAGX,oBAAA,GACX,OAAA,EAAS,iBAAA,KACR,UA0FF;AAAA,KAEW,cAAA,GAAiB,UAAU,QAAQ,oBAAA"}
+{"version":3,"file":"subscriber.d.ts","names":[],"sources":["../../../src/runtime/adapter/subscriber.ts"],"mappings":";;;;KAMK,QAAA,IAAY,KAAgB,EAAT,SAAS;AAAA,KAE5B,UAAA;EACH,OAAA,IAAW,OAAA;IAAW,KAAA;EAAA;EACtB,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,yBAAA,IAA6B,OAAA;IAAW,KAAA;EAAA;EACxC,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,+BAAA,IAAmC,OAAA;IAAW,KAAA;EAAA;EAC9C,iCAAA,IAAqC,OAAA;IAAW,KAAA;EAAA;EAChD,6BAAA,IAAiC,OAAA;IAAW,KAAA;EAAA;EAC5C,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,mBAAA,IAAuB,OAAA;IAAW,KAAA;EAAA;EAClC,4BAAA,IAAgC,OAAA;IAAW,KAAA;EAAA;EAC3C,8BAAA,IAAkC,OAAA;IAAW,KAAA;EAAA;EAC7C,0BAAA,IAA8B,OAAA;IAAW,KAAA;EAAA;EACzC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,mBAAA,IAAuB,OAAA;IAAW,KAAA;EAAA;EAClC,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,iBAAA,IAAqB,OAAA;IAAW,KAAA;EAAA;EAChC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,aAAA,IAAiB,OAAA;IAAW,KAAA;EAAA;EAC5B,UAAA,IAAc,OAAA;IAAW,KAAA;EAAA;EACzB,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,cAAA;EACA,WAAA,IAAe,OAAA;IAAW,KAAA,EAAO,KAAK;EAAA;AAAA;AAAA,KAiCnC,iBAAA;EACH,QAAA,EAAU,QAAA;EACV,KAAA;EACA,MAAA,GAAS,MAAA;EACT,WAAA,IAAe,KAAA,EAAO,KAAA;AAAA;AAAA,cAGX,oBAAA,GACX,OAAA,EAAS,iBAAA,KACR,UA+FF;AAAA,KAEW,cAAA,GAAiB,UAAU,QAAQ,oBAAA"}

package/dist/runtime/adapter/subscriber.js

@@ -1,6 +1,11 @@
 "use client";
 import { parseAgUiEvent } from "../event-parser.js";
 //#region src/runtime/adapter/subscriber.ts
+const isAbortError = (error) => {
+	if (error.name === "AbortError") return true;
+	const message = typeof error.message === "string" ? error.message : "";
+	return message === "Fetch is aborted" || message === "signal is aborted without reason" || message === "component unmounted";
+};
 const ensureEvent = (raw, type, logger) => {
 	const parseOptions = logger ? { logger } : void 0;
 	let parsed;
@@ -67,7 +72,12 @@ const createAgUiSubscriber = (options) => {
 			});
 		},
 		onRunFailed: ({ error }) => {
+			runFinishedDispatched = true;
 			onRunFailed?.(error);
+			if (isAbortError(error)) {
+				dispatch({ type: "RUN_CANCELLED" });
+				return;
+			}
 			const message = typeof error.message === "string" ? error.message : "Run failed";
 			const code = typeof error?.code === "string" ? error.code : void 0;
 			dispatch({

package/dist/runtime/adapter/subscriber.js.map

@@ -1 +1 @@
-{"version":3,"file":"subscriber.js","names":[],"sources":["../../../src/runtime/adapter/subscriber.ts"],"sourcesContent":["\"use client\";\n\nimport type { AgUiEvent } from \"../types\";\nimport { parseAgUiEvent } from \"../event-parser\";\nimport type { Logger } from \"../logger\";\n\ntype Dispatch = (event: AgUiEvent) => void;\n\ntype Subscriber = {\n  onEvent?: (payload: { event: unknown }) => void;\n  onTextMessageStartEvent?: (payload: { event: unknown }) => void;\n  onTextMessageContentEvent?: (payload: { event: unknown }) => void;\n  onTextMessageEndEvent?: (payload: { event: unknown }) => void;\n  onTextMessageChunkEvent?: (payload: { event: unknown }) => void;\n  onThinkingStartEvent?: (payload: { event: unknown }) => void;\n  onThinkingEndEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageStartEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageContentEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageEndEvent?: (payload: { event: unknown }) => void;\n  onReasoningStartEvent?: (payload: { event: unknown }) => void;\n  onReasoningEndEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageStartEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageContentEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageEndEvent?: (payload: { event: unknown }) => void;\n  onToolCallStartEvent?: (payload: { event: unknown }) => void;\n  onToolCallArgsEvent?: (payload: { event: unknown }) => void;\n  onToolCallEndEvent?: (payload: { event: unknown }) => void;\n  onToolCallChunkEvent?: (payload: { event: unknown }) => void;\n  onToolCallResultEvent?: (payload: { event: unknown }) => void;\n  onActivitySnapshotEvent?: (payload: { event: unknown }) => void;\n  onStateSnapshotEvent?: (payload: { event: unknown }) => void;\n  onStateDeltaEvent?: (payload: { event: unknown }) => void;\n  onMessagesSnapshotEvent?: (payload: { event: unknown }) => void;\n  onCustomEvent?: (payload: { event: unknown }) => void;\n  onRawEvent?: (payload: { event: unknown }) => void;\n  onRunFinishedEvent?: (payload: { event: unknown }) => void;\n  onRunFinalized?: () => void;\n  onRunFailed?: (payload: { error: Error }) => void;\n};\n\nconst ensureEvent = (\n  raw: unknown,\n  type: AgUiEvent[\"type\"],\n  logger?: Logger,\n): AgUiEvent | null => {\n  const parseOptions = logger ? { logger } : undefined;\n  let parsed: AgUiEvent | null;\n  if (raw && typeof raw === \"object\") {\n    const payload = raw as Record<string, unknown>;\n    if (typeof payload.type === \"string\") {\n      parsed = parseAgUiEvent(payload, parseOptions);\n    } else {\n      parsed = parseAgUiEvent({ type, ...payload }, parseOptions);\n    }\n  } else {\n    parsed = parseAgUiEvent({ type }, parseOptions);\n  }\n  return parsed && parsed.type === type ? parsed : null;\n};\n\ntype SubscriberOptions = {\n  dispatch: Dispatch;\n  runId: string;\n  logger?: Logger;\n  onRunFailed?: (error: Error) => void;\n};\n\nexport const createAgUiSubscriber = (\n  options: SubscriberOptions,\n): Subscriber => {\n  const { dispatch, runId, logger, onRunFailed } = options;\n  let runFinishedDispatched = false;\n  const dispatchIfValid = (raw: unknown, type: AgUiEvent[\"type\"]) => {\n    const event = ensureEvent(raw, type, logger);\n    if (!event) return;\n    dispatch(event);\n  };\n  return {\n    onEvent: ({ event }) => {\n      const typeCandidate =\n        event && typeof event === \"object\"\n          ? (event as Record<string, unknown>).type\n          : undefined;\n      if (typeof typeCandidate === \"string\") {\n        // Typed handlers will receive this via the discriminated callbacks; avoid duplicates.\n        return;\n      }\n      const parsed = parseAgUiEvent(event);\n      if (parsed) dispatch(parsed);\n    },\n    onTextMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_START\"),\n    onTextMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_CONTENT\"),\n    onTextMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_END\"),\n    onTextMessageChunkEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_CHUNK\"),\n    onThinkingStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_START\"),\n    onThinkingEndEvent: ({ event }) => dispatchIfValid(event, \"THINKING_END\"),\n    onThinkingTextMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_START\"),\n    onThinkingTextMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_CONTENT\"),\n    onThinkingTextMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_END\"),\n    onReasoningStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_START\"),\n    onReasoningEndEvent: ({ event }) => dispatchIfValid(event, \"REASONING_END\"),\n    onReasoningMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_START\"),\n    onReasoningMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_CONTENT\"),\n    onReasoningMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_END\"),\n    onToolCallStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_START\"),\n    onToolCallArgsEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_ARGS\"),\n    onToolCallEndEvent: ({ event }) => dispatchIfValid(event, \"TOOL_CALL_END\"),\n    onToolCallChunkEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_CHUNK\"),\n    onToolCallResultEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_RESULT\"),\n    onActivitySnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"ACTIVITY_SNAPSHOT\"),\n    onStateSnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"STATE_SNAPSHOT\"),\n    onStateDeltaEvent: ({ event }) => dispatchIfValid(event, \"STATE_DELTA\"),\n    onMessagesSnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"MESSAGES_SNAPSHOT\"),\n    onCustomEvent: ({ event }) => dispatchIfValid(event, \"CUSTOM\"),\n    onRawEvent: ({ event }) => dispatchIfValid(event, \"RAW\"),\n    onRunFinishedEvent: ({ event }) => {\n      const parsed = ensureEvent(event, \"RUN_FINISHED\", logger);\n      if (!parsed) return;\n      runFinishedDispatched = true;\n      dispatch(parsed);\n    },\n    onRunFinalized: () => {\n      if (runFinishedDispatched) return;\n      dispatch({ type: \"RUN_FINISHED\", runId });\n    },\n    onRunFailed: ({ error }) => {\n      onRunFailed?.(error);\n      const message =\n        typeof error.message === \"string\" ? error.message : \"Run failed\";\n      const code =\n        typeof (error as any)?.code === \"string\"\n          ? (error as any).code\n          : undefined;\n      dispatch({\n        type: \"RUN_ERROR\" as const,\n        ...(message !== undefined ? { message } : {}),\n        ...(code !== undefined ? { code } : {}),\n      } satisfies AgUiEvent);\n    },\n  };\n};\n\nexport type AgUiSubscriber = ReturnType<typeof createAgUiSubscriber>;\n"],"mappings":";;;AAwCA,MAAM,eACJ,KACA,MACA,WACqB;CACrB,MAAM,eAAe,SAAS,EAAE,OAAO,IAAI,KAAA;CAC3C,IAAI;CACJ,IAAI,OAAO,OAAO,QAAQ,UAAU;EAClC,MAAM,UAAU;EAChB,IAAI,OAAO,QAAQ,SAAS,UAC1B,SAAS,eAAe,SAAS,YAAY;OAE7C,SAAS,eAAe;GAAE;GAAM,GAAG;EAAQ,GAAG,YAAY;CAE9D,OACE,SAAS,eAAe,EAAE,KAAK,GAAG,YAAY;CAEhD,OAAO,UAAU,OAAO,SAAS,OAAO,SAAS;AACnD;AASA,MAAa,wBACX,YACe;CACf,MAAM,EAAE,UAAU,OAAO,QAAQ,gBAAgB;CACjD,IAAI,wBAAwB;CAC5B,MAAM,mBAAmB,KAAc,SAA4B;EACjE,MAAM,QAAQ,YAAY,KAAK,MAAM,MAAM;EAC3C,IAAI,CAAC,OAAO;EACZ,SAAS,KAAK;CAChB;CACA,OAAO;EACL,UAAU,EAAE,YAAY;GAKtB,IAAI,QAHF,SAAS,OAAO,UAAU,WACrB,MAAkC,OACnC,KAAA,OACuB,UAE3B;GAEF,MAAM,SAAS,eAAe,KAAK;GACnC,IAAI,QAAQ,SAAS,MAAM;EAC7B;EACA,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,oBAAoB;EAC7C,4BAA4B,EAAE,YAC5B,gBAAgB,OAAO,sBAAsB;EAC/C,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,kBAAkB;EAC3C,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,oBAAoB;EAC7C,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,gBAAgB;EACzC,qBAAqB,EAAE,YAAY,gBAAgB,OAAO,cAAc;EACxE,kCAAkC,EAAE,YAClC,gBAAgB,OAAO,6BAA6B;EACtD,oCAAoC,EAAE,YACpC,gBAAgB,OAAO,+BAA+B;EACxD,gCAAgC,EAAE,YAChC,gBAAgB,OAAO,2BAA2B;EACpD,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,iBAAiB;EAC1C,sBAAsB,EAAE,YAAY,gBAAgB,OAAO,eAAe;EAC1E,+BAA+B,EAAE,YAC/B,gBAAgB,OAAO,yBAAyB;EAClD,iCAAiC,EAAE,YACjC,gBAAgB,OAAO,2BAA2B;EACpD,6BAA6B,EAAE,YAC7B,gBAAgB,OAAO,uBAAuB;EAChD,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,iBAAiB;EAC1C,sBAAsB,EAAE,YACtB,gBAAgB,OAAO,gBAAgB;EACzC,qBAAqB,EAAE,YAAY,gBAAgB,OAAO,eAAe;EACzE,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,iBAAiB;EAC1C,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,kBAAkB;EAC3C,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,mBAAmB;EAC5C,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,gBAAgB;EACzC,oBAAoB,EAAE,YAAY,gBAAgB,OAAO,aAAa;EACtE,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,mBAAmB;EAC5C,gBAAgB,EAAE,YAAY,gBAAgB,OAAO,QAAQ;EAC7D,aAAa,EAAE,YAAY,gBAAgB,OAAO,KAAK;EACvD,qBAAqB,EAAE,YAAY;GACjC,MAAM,SAAS,YAAY,OAAO,gBAAgB,MAAM;GACxD,IAAI,CAAC,QAAQ;GACb,wBAAwB;GACxB,SAAS,MAAM;EACjB;EACA,sBAAsB;GACpB,IAAI,uBAAuB;GAC3B,SAAS;IAAE,MAAM;IAAgB;GAAM,CAAC;EAC1C;EACA,cAAc,EAAE,YAAY;GAC1B,cAAc,KAAK;GACnB,MAAM,UACJ,OAAO,MAAM,YAAY,WAAW,MAAM,UAAU;GACtD,MAAM,OACJ,OAAQ,OAAe,SAAS,WAC3B,MAAc,OACf,KAAA;GACN,SAAS;IACP,MAAM;IACN,GAAI,YAAY,KAAA,IAAY,EAAE,QAAQ,IAAI,CAAC;IAC3C,GAAI,SAAS,KAAA,IAAY,EAAE,KAAK,IAAI,CAAC;GACvC,CAAqB;EACvB;CACF;AACF"}
+{"version":3,"file":"subscriber.js","names":[],"sources":["../../../src/runtime/adapter/subscriber.ts"],"sourcesContent":["\"use client\";\n\nimport type { AgUiEvent } from \"../types\";\nimport { parseAgUiEvent } from \"../event-parser\";\nimport type { Logger } from \"../logger\";\n\ntype Dispatch = (event: AgUiEvent) => void;\n\ntype Subscriber = {\n  onEvent?: (payload: { event: unknown }) => void;\n  onTextMessageStartEvent?: (payload: { event: unknown }) => void;\n  onTextMessageContentEvent?: (payload: { event: unknown }) => void;\n  onTextMessageEndEvent?: (payload: { event: unknown }) => void;\n  onTextMessageChunkEvent?: (payload: { event: unknown }) => void;\n  onThinkingStartEvent?: (payload: { event: unknown }) => void;\n  onThinkingEndEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageStartEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageContentEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageEndEvent?: (payload: { event: unknown }) => void;\n  onReasoningStartEvent?: (payload: { event: unknown }) => void;\n  onReasoningEndEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageStartEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageContentEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageEndEvent?: (payload: { event: unknown }) => void;\n  onToolCallStartEvent?: (payload: { event: unknown }) => void;\n  onToolCallArgsEvent?: (payload: { event: unknown }) => void;\n  onToolCallEndEvent?: (payload: { event: unknown }) => void;\n  onToolCallChunkEvent?: (payload: { event: unknown }) => void;\n  onToolCallResultEvent?: (payload: { event: unknown }) => void;\n  onActivitySnapshotEvent?: (payload: { event: unknown }) => void;\n  onStateSnapshotEvent?: (payload: { event: unknown }) => void;\n  onStateDeltaEvent?: (payload: { event: unknown }) => void;\n  onMessagesSnapshotEvent?: (payload: { event: unknown }) => void;\n  onCustomEvent?: (payload: { event: unknown }) => void;\n  onRawEvent?: (payload: { event: unknown }) => void;\n  onRunFinishedEvent?: (payload: { event: unknown }) => void;\n  onRunFinalized?: () => void;\n  onRunFailed?: (payload: { error: Error }) => void;\n};\n\nconst isAbortError = (error: Error): boolean => {\n  if (error.name === \"AbortError\") return true;\n  const message = typeof error.message === \"string\" ? error.message : \"\";\n  return (\n    message === \"Fetch is aborted\" ||\n    message === \"signal is aborted without reason\" ||\n    message === \"component unmounted\"\n  );\n};\n\nconst ensureEvent = (\n  raw: unknown,\n  type: AgUiEvent[\"type\"],\n  logger?: Logger,\n): AgUiEvent | null => {\n  const parseOptions = logger ? { logger } : undefined;\n  let parsed: AgUiEvent | null;\n  if (raw && typeof raw === \"object\") {\n    const payload = raw as Record<string, unknown>;\n    if (typeof payload.type === \"string\") {\n      parsed = parseAgUiEvent(payload, parseOptions);\n    } else {\n      parsed = parseAgUiEvent({ type, ...payload }, parseOptions);\n    }\n  } else {\n    parsed = parseAgUiEvent({ type }, parseOptions);\n  }\n  return parsed && parsed.type === type ? parsed : null;\n};\n\ntype SubscriberOptions = {\n  dispatch: Dispatch;\n  runId: string;\n  logger?: Logger;\n  onRunFailed?: (error: Error) => void;\n};\n\nexport const createAgUiSubscriber = (\n  options: SubscriberOptions,\n): Subscriber => {\n  const { dispatch, runId, logger, onRunFailed } = options;\n  let runFinishedDispatched = false;\n  const dispatchIfValid = (raw: unknown, type: AgUiEvent[\"type\"]) => {\n    const event = ensureEvent(raw, type, logger);\n    if (!event) return;\n    dispatch(event);\n  };\n  return {\n    onEvent: ({ event }) => {\n      const typeCandidate =\n        event && typeof event === \"object\"\n          ? (event as Record<string, unknown>).type\n          : undefined;\n      if (typeof typeCandidate === \"string\") {\n        // Typed handlers will receive this via the discriminated callbacks; avoid duplicates.\n        return;\n      }\n      const parsed = parseAgUiEvent(event);\n      if (parsed) dispatch(parsed);\n    },\n    onTextMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_START\"),\n    onTextMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_CONTENT\"),\n    onTextMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_END\"),\n    onTextMessageChunkEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_CHUNK\"),\n    onThinkingStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_START\"),\n    onThinkingEndEvent: ({ event }) => dispatchIfValid(event, \"THINKING_END\"),\n    onThinkingTextMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_START\"),\n    onThinkingTextMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_CONTENT\"),\n    onThinkingTextMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_END\"),\n    onReasoningStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_START\"),\n    onReasoningEndEvent: ({ event }) => dispatchIfValid(event, \"REASONING_END\"),\n    onReasoningMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_START\"),\n    onReasoningMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_CONTENT\"),\n    onReasoningMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_END\"),\n    onToolCallStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_START\"),\n    onToolCallArgsEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_ARGS\"),\n    onToolCallEndEvent: ({ event }) => dispatchIfValid(event, \"TOOL_CALL_END\"),\n    onToolCallChunkEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_CHUNK\"),\n    onToolCallResultEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_RESULT\"),\n    onActivitySnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"ACTIVITY_SNAPSHOT\"),\n    onStateSnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"STATE_SNAPSHOT\"),\n    onStateDeltaEvent: ({ event }) => dispatchIfValid(event, \"STATE_DELTA\"),\n    onMessagesSnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"MESSAGES_SNAPSHOT\"),\n    onCustomEvent: ({ event }) => dispatchIfValid(event, \"CUSTOM\"),\n    onRawEvent: ({ event }) => dispatchIfValid(event, \"RAW\"),\n    onRunFinishedEvent: ({ event }) => {\n      const parsed = ensureEvent(event, \"RUN_FINISHED\", logger);\n      if (!parsed) return;\n      runFinishedDispatched = true;\n      dispatch(parsed);\n    },\n    onRunFinalized: () => {\n      if (runFinishedDispatched) return;\n      dispatch({ type: \"RUN_FINISHED\", runId });\n    },\n    onRunFailed: ({ error }) => {\n      runFinishedDispatched = true;\n      onRunFailed?.(error);\n      if (isAbortError(error)) {\n        dispatch({ type: \"RUN_CANCELLED\" } satisfies AgUiEvent);\n        return;\n      }\n      const message =\n        typeof error.message === \"string\" ? error.message : \"Run failed\";\n      const code =\n        typeof (error as any)?.code === \"string\"\n          ? (error as any).code\n          : undefined;\n      dispatch({\n        type: \"RUN_ERROR\" as const,\n        ...(message !== undefined ? { message } : {}),\n        ...(code !== undefined ? { code } : {}),\n      } satisfies AgUiEvent);\n    },\n  };\n};\n\nexport type AgUiSubscriber = ReturnType<typeof createAgUiSubscriber>;\n"],"mappings":";;;AAwCA,MAAM,gBAAgB,UAA0B;CAC9C,IAAI,MAAM,SAAS,cAAc,OAAO;CACxC,MAAM,UAAU,OAAO,MAAM,YAAY,WAAW,MAAM,UAAU;CACpE,OACE,YAAY,sBACZ,YAAY,sCACZ,YAAY;AAEhB;AAEA,MAAM,eACJ,KACA,MACA,WACqB;CACrB,MAAM,eAAe,SAAS,EAAE,OAAO,IAAI,KAAA;CAC3C,IAAI;CACJ,IAAI,OAAO,OAAO,QAAQ,UAAU;EAClC,MAAM,UAAU;EAChB,IAAI,OAAO,QAAQ,SAAS,UAC1B,SAAS,eAAe,SAAS,YAAY;OAE7C,SAAS,eAAe;GAAE;GAAM,GAAG;EAAQ,GAAG,YAAY;CAE9D,OACE,SAAS,eAAe,EAAE,KAAK,GAAG,YAAY;CAEhD,OAAO,UAAU,OAAO,SAAS,OAAO,SAAS;AACnD;AASA,MAAa,wBACX,YACe;CACf,MAAM,EAAE,UAAU,OAAO,QAAQ,gBAAgB;CACjD,IAAI,wBAAwB;CAC5B,MAAM,mBAAmB,KAAc,SAA4B;EACjE,MAAM,QAAQ,YAAY,KAAK,MAAM,MAAM;EAC3C,IAAI,CAAC,OAAO;EACZ,SAAS,KAAK;CAChB;CACA,OAAO;EACL,UAAU,EAAE,YAAY;GAKtB,IAAI,QAHF,SAAS,OAAO,UAAU,WACrB,MAAkC,OACnC,KAAA,OACuB,UAE3B;GAEF,MAAM,SAAS,eAAe,KAAK;GACnC,IAAI,QAAQ,SAAS,MAAM;EAC7B;EACA,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,oBAAoB;EAC7C,4BAA4B,EAAE,YAC5B,gBAAgB,OAAO,sBAAsB;EAC/C,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,kBAAkB;EAC3C,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,oBAAoB;EAC7C,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,gBAAgB;EACzC,qBAAqB,EAAE,YAAY,gBAAgB,OAAO,cAAc;EACxE,kCAAkC,EAAE,YAClC,gBAAgB,OAAO,6BAA6B;EACtD,oCAAoC,EAAE,YACpC,gBAAgB,OAAO,+BAA+B;EACxD,gCAAgC,EAAE,YAChC,gBAAgB,OAAO,2BAA2B;EACpD,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,iBAAiB;EAC1C,sBAAsB,EAAE,YAAY,gBAAgB,OAAO,eAAe;EAC1E,+BAA+B,EAAE,YAC/B,gBAAgB,OAAO,yBAAyB;EAClD,iCAAiC,EAAE,YACjC,gBAAgB,OAAO,2BAA2B;EACpD,6BAA6B,EAAE,YAC7B,gBAAgB,OAAO,uBAAuB;EAChD,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,iBAAiB;EAC1C,sBAAsB,EAAE,YACtB,gBAAgB,OAAO,gBAAgB;EACzC,qBAAqB,EAAE,YAAY,gBAAgB,OAAO,eAAe;EACzE,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,iBAAiB;EAC1C,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,kBAAkB;EAC3C,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,mBAAmB;EAC5C,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,gBAAgB;EACzC,oBAAoB,EAAE,YAAY,gBAAgB,OAAO,aAAa;EACtE,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,mBAAmB;EAC5C,gBAAgB,EAAE,YAAY,gBAAgB,OAAO,QAAQ;EAC7D,aAAa,EAAE,YAAY,gBAAgB,OAAO,KAAK;EACvD,qBAAqB,EAAE,YAAY;GACjC,MAAM,SAAS,YAAY,OAAO,gBAAgB,MAAM;GACxD,IAAI,CAAC,QAAQ;GACb,wBAAwB;GACxB,SAAS,MAAM;EACjB;EACA,sBAAsB;GACpB,IAAI,uBAAuB;GAC3B,SAAS;IAAE,MAAM;IAAgB;GAAM,CAAC;EAC1C;EACA,cAAc,EAAE,YAAY;GAC1B,wBAAwB;GACxB,cAAc,KAAK;GACnB,IAAI,aAAa,KAAK,GAAG;IACvB,SAAS,EAAE,MAAM,gBAAgB,CAAqB;IACtD;GACF;GACA,MAAM,UACJ,OAAO,MAAM,YAAY,WAAW,MAAM,UAAU;GACtD,MAAM,OACJ,OAAQ,OAAe,SAAS,WAC3B,MAAc,OACf,KAAA;GACN,SAAS;IACP,MAAM;IACN,GAAI,YAAY,KAAA,IAAY,EAAE,QAAQ,IAAI,CAAC;IAC3C,GAAI,SAAS,KAAA,IAAY,EAAE,KAAK,IAAI,CAAC;GACvC,CAAqB;EACvB;CACF;AACF"}

package/dist/runtime/types.d.ts

@@ -1,22 +1,27 @@
 import { Logger } from "./logger.js";
-import { AttachmentAdapter, DictationAdapter, ExternalStoreSharedOptions, FeedbackAdapter, RealtimeVoiceAdapter, SpeechSynthesisAdapter, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
-import { HttpAgent } from "@ag-ui/client";
+import { AttachmentAdapter, DictationAdapter, ExternalStoreSharedOptions, ExternalStoreThreadListAdapter, FeedbackAdapter, RealtimeVoiceAdapter, SpeechSynthesisAdapter, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
+import { AbstractAgent } from "@ag-ui/client";
 import { ReadonlyJSONValue } from "assistant-stream/utils";
 
 //#region src/runtime/types.d.ts
 /**
  * @experimental This API is still under active development and might change without notice.
+ *
+ * Same as ExternalStoreThreadListAdapter, except `onSwitchToThread` returns
+ * the messages (and optional state) to hydrate the thread with.
  */
-type UseAgUiThreadListAdapter = {
-  threadId?: string | undefined;
-  onSwitchToNewThread?: (() => Promise<void> | void) | undefined;
-  onSwitchToThread?: ((threadId: string) => Promise<{
-    messages: readonly ThreadMessage[];
-    state?: ReadonlyJSONValue;
-  }> | {
-    messages: readonly ThreadMessage[];
-    state?: ReadonlyJSONValue;
-  }) | undefined;
+type SwitchToThreadResult = {
+  messages: readonly ThreadMessage[];
+  state?: ReadonlyJSONValue;
+  /**
+   * Set when the thread has a run in flight. The runtime resumes the run
+   * after hydrating, the same way `ThreadHistoryAdapter.load()` does when it
+   * returns `unstable_resume: true`.
+   */
+  unstable_resume?: boolean;
+};
+type UseAgUiThreadListAdapter = Omit<ExternalStoreThreadListAdapter, "onSwitchToThread"> & {
+  onSwitchToThread?: ((threadId: string) => Promise<SwitchToThreadResult> | SwitchToThreadResult) | undefined;
 };
 type UseAgUiRuntimeAdapters = {
   attachments?: AttachmentAdapter;
@@ -31,7 +36,7 @@ type UseAgUiRuntimeAdapters = {
   threadList?: UseAgUiThreadListAdapter;
 };
 type UseAgUiRuntimeOptions = ExternalStoreSharedOptions & {
-  agent: HttpAgent;
+  agent: AbstractAgent;
   logger?: Partial<Logger>;
   showThinking?: boolean;
   onError?: (e: Error) => void;

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

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../../src/runtime/types.ts"],"mappings":";;;;;;;;AAiBA;KAAY,wBAAA;EACV,QAAA;EACA,mBAAA,UAA6B,OAAA;EAC7B,gBAAA,KACM,QAAA,aACE,OAAA;IACE,QAAA,WAAmB,aAAA;IACnB,KAAA,GAAQ,iBAAA;EAAA;IAER,QAAA,WAAmB,aAAA;IAAiB,KAAA,GAAQ,iBAAA;EAAA;AAAA;AAAA,KAI5C,sBAAA;EACV,WAAA,GAAc,iBAAA;EACd,MAAA,GAAS,sBAAA;EACT,SAAA,GAAY,gBAAA;EACZ,KAAA,GAAQ,oBAAA;EACR,QAAA,GAAW,eAAA;EACX,OAAA,GAAU,oBAAA;EAZA;;;EAgBV,UAAA,GAAa,wBAAA;AAAA;AAAA,KAGH,qBAAA,GAAwB,0BAAA;EAClC,KAAA,EAAO,SAAA;EACP,MAAA,GAAS,OAAA,CAAQ,MAAA;EACjB,YAAA;EACA,OAAA,IAAW,CAAA,EAAG,KAAA;EACd,QAAA;EACA,QAAA,GAAW,sBAAA;AAAA;AAAA,KAGD,mBAAA;AAAA,KAMA,aAAA;EACV,EAAA;EACA,MAAA,EAAQ,mBAAA;EACR,OAAA;EACA,UAAA;EACA,cAAA,GAAiB,MAAA;EACjB,SAAA;EACA,QAAA,GAAW,MAAA;AAAA;AAAA,KAGD,eAAA;EACV,WAAA;EACA,MAAA;EACA,OAAA;AAAA;AAAA,KAGU,sBAAA;EACN,IAAA;AAAA;EACA,IAAA;EAAmB,UAAA,EAAY,aAAa;AAAA;AAAA,KAEtC,SAAA;EACN,IAAA;EAAqB,KAAA;AAAA;EAErB,IAAA;EACA,KAAA;EACA,OAAA,GAAU,sBAAA;AAAA;EAEV,IAAA;EAAuB,KAAA;AAAA;EACvB,IAAA;EAAmB,OAAA;EAAkB,IAAA;AAAA;EACrC,IAAA;EAA4B,SAAA;AAAA;EAC5B,IAAA;EAA8B,SAAA;EAAoB,KAAA;AAAA;EAClD,IAAA;EAA0B,SAAA;AAAA;EAC1B,IAAA;EAA4B,KAAA;AAAA;EAC5B,IAAA;EAAwB,KAAA;AAAA;EACxB,IAAA;AAAA;EACA,IAAA;EAAuC,KAAA;AAAA;EACvC,IAAA;AAAA;EACA,IAAA;AAAA;EACA,IAAA;EAAyB,SAAA;AAAA;EACzB,IAAA;EAAiC,SAAA;AAAA;EACjC,IAAA;EAAmC,SAAA;EAAoB,KAAA;AAAA;EACvD,IAAA;EAA+B,SAAA;AAAA;EAC/B,IAAA;EAAuB,SAAA;AAAA;EAEvB,IAAA;EACA,UAAA;EACA,YAAA;EACA,eAAA;AAAA;EAEA,IAAA;EAAwB,UAAA;EAAoB,KAAA;AAAA;EAC5C,IAAA;EAAuB,UAAA;AAAA;EAEvB,IAAA;EACA,UAAA;EACA,YAAA;EACA,eAAA;EACA,KAAA;AAAA;EAGA,IAAA;EACA,SAAA;EACA,UAAA;EACA,OAAA;EACA,IAAA;AAAA;EAGA,IAAA;EACA,YAAA;EACA,OAAA,EAAS,MAAM;AAAA;EAEf,IAAA;EAAa,KAAA;EAAY,MAAA;AAAA;EACzB,IAAA;EAAgB,IAAA;EAAc,KAAA;AAAA;EAC9B,IAAA;EAAwB,QAAA;AAAA;EACxB,IAAA;EAAqB,KAAA;AAAA;EACrB,IAAA;EAA2B,QAAA;AAAA"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../src/runtime/types.ts"],"mappings":";;;;;;;;AAagE;;;;KAQ3D,oBAAA;EACH,QAAA,WAAmB,aAAA;EACnB,KAAA,GAAQ,iBAAiB;EAAjB;;;AAMO;AAGjB;EAHE,eAAA;AAAA;AAAA,KAGU,wBAAA,GAA2B,IAAA,CACrC,8BAAA;EAGA,gBAAA,KAEM,QAAA,aACG,OAAA,CAAQ,oBAAA,IAAwB,oBAAA;AAAA;AAAA,KAI/B,sBAAA;EACV,WAAA,GAAc,iBAAA;EACd,MAAA,GAAS,sBAAA;EACT,SAAA,GAAY,gBAAA;EACZ,KAAA,GAAQ,oBAAA;EACR,QAAA,GAAW,eAAA;EACX,OAAA,GAAU,oBAAA;EAXJ;;;EAeN,UAAA,GAAa,wBAAA;AAAA;AAAA,KAGH,qBAAA,GAAwB,0BAAA;EAClC,KAAA,EAAO,aAAA;EACP,MAAA,GAAS,OAAA,CAAQ,MAAA;EACjB,YAAA;EACA,OAAA,IAAW,CAAA,EAAG,KAAA;EACd,QAAA;EACA,QAAA,GAAW,sBAAA;AAAA;AAAA,KAGD,mBAAA;AAAA,KAMA,aAAA;EACV,EAAA;EACA,MAAA,EAAQ,mBAAA;EACR,OAAA;EACA,UAAA;EACA,cAAA,GAAiB,MAAA;EACjB,SAAA;EACA,QAAA,GAAW,MAAA;AAAA;AAAA,KAGD,eAAA;EACV,WAAA;EACA,MAAA;EACA,OAAA;AAAA;AAAA,KAGU,sBAAA;EACN,IAAA;AAAA;EACA,IAAA;EAAmB,UAAA,EAAY,aAAa;AAAA;AAAA,KAEtC,SAAA;EACN,IAAA;EAAqB,KAAA;AAAA;EAErB,IAAA;EACA,KAAA;EACA,OAAA,GAAU,sBAAA;AAAA;EAEV,IAAA;EAAuB,KAAA;AAAA;EACvB,IAAA;EAAmB,OAAA;EAAkB,IAAA;AAAA;EACrC,IAAA;EAA4B,SAAA;AAAA;EAC5B,IAAA;EAA8B,SAAA;EAAoB,KAAA;AAAA;EAClD,IAAA;EAA0B,SAAA;AAAA;EAC1B,IAAA;EAA4B,KAAA;AAAA;EAC5B,IAAA;EAAwB,KAAA;AAAA;EACxB,IAAA;AAAA;EACA,IAAA;EAAuC,KAAA;AAAA;EACvC,IAAA;AAAA;EACA,IAAA;AAAA;EACA,IAAA;EAAyB,SAAA;AAAA;EACzB,IAAA;EAAiC,SAAA;AAAA;EACjC,IAAA;EAAmC,SAAA;EAAoB,KAAA;AAAA;EACvD,IAAA;EAA+B,SAAA;AAAA;EAC/B,IAAA;EAAuB,SAAA;AAAA;EAEvB,IAAA;EACA,UAAA;EACA,YAAA;EACA,eAAA;AAAA;EAEA,IAAA;EAAwB,UAAA;EAAoB,KAAA;AAAA;EAC5C,IAAA;EAAuB,UAAA;AAAA;EAEvB,IAAA;EACA,UAAA;EACA,YAAA;EACA,eAAA;EACA,KAAA;AAAA;EAGA,IAAA;EACA,SAAA;EACA,UAAA;EACA,OAAA;EACA,IAAA;AAAA;EAGA,IAAA;EACA,YAAA;EACA,OAAA,EAAS,MAAM;AAAA;EAEf,IAAA;EAAa,KAAA;EAAY,MAAA;AAAA;EACzB,IAAA;EAAgB,IAAA;EAAc,KAAA;AAAA;EAC9B,IAAA;EAAwB,QAAA;AAAA;EACxB,IAAA;EAAqB,KAAA;AAAA;EACrB,IAAA;EAA2B,QAAA;AAAA"}

package/dist/useAgUiRuntime.js

@@ -34,17 +34,19 @@ function useAgUiRuntime(options) {
 	const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
 	const threadList = useMemo(() => {
 		if (!threadListAdapter) return void 0;
-		const { onSwitchToNewThread, onSwitchToThread } = threadListAdapter;
+		const { onSwitchToNewThread, onSwitchToThread, ...rest } = threadListAdapter;
 		return {
-			threadId: threadListAdapter.threadId,
+			...rest,
 			onSwitchToNewThread: onSwitchToNewThread ? async () => {
 				await onSwitchToNewThread();
 				core.applyExternalMessages([]);
 			} : void 0,
 			onSwitchToThread: onSwitchToThread ? async (threadId) => {
+				core.applyExternalMessages([]);
 				const result = await onSwitchToThread(threadId);
 				core.applyExternalMessages(result.messages);
 				if (result.state) core.loadExternalState(result.state);
+				if (result.unstable_resume) core.resumeInFlightRun(result.messages);
 			} : void 0
 		};
 	}, [threadListAdapter, core]);

package/dist/useAgUiRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAgUiRuntime.js","names":[],"sources":["../src/useAgUiRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useCallback, useEffect, useMemo, useRef, useState } from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport type {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport { makeLogger } from \"./runtime/logger\";\nimport type {\n  AgUiInterrupt,\n  AgUiResumeEntry,\n  UseAgUiRuntimeOptions,\n} from \"./runtime/types\";\nimport { AgUiThreadRuntimeCore } from \"./runtime/AgUiThreadRuntimeCore\";\n\nexport type AgUiAssistantRuntime = AssistantRuntime & {\n  unstable_getPendingInterrupts: () => readonly AgUiInterrupt[];\n  unstable_submitInterruptResponses: (\n    responses: readonly AgUiResumeEntry[],\n  ) => Promise<void>;\n};\n\nexport function useAgUiRuntime(\n  options: UseAgUiRuntimeOptions,\n): AgUiAssistantRuntime {\n  const logger = useMemo(() => makeLogger(options.logger), [options.logger]);\n  const [_version, setVersion] = useState(0);\n  const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);\n  const coreRef = useRef<AgUiThreadRuntimeCore | null>(null);\n  const runtimeAdapters = useRuntimeAdapters();\n\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  if (!coreRef.current) {\n    coreRef.current = new AgUiThreadRuntimeCore({\n      agent: options.agent,\n      logger,\n      showThinking: options.showThinking ?? true,\n      ...(options.onError && { onError: options.onError }),\n      ...(options.onCancel && { onCancel: options.onCancel }),\n      ...(historyAdapter && { history: historyAdapter }),\n      notifyUpdate,\n    });\n  }\n\n  const core = coreRef.current;\n  core.updateOptions({\n    agent: options.agent,\n    logger,\n    showThinking: options.showThinking ?? true,\n    ...(options.onError && { onError: options.onError }),\n    ...(options.onCancel && { onCancel: options.onCancel }),\n    ...(historyAdapter && { history: historyAdapter }),\n  });\n\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n\n  const threadList = useMemo(() => {\n    if (!threadListAdapter) return undefined;\n\n    const { onSwitchToNewThread, onSwitchToThread } = threadListAdapter;\n\n    return {\n      threadId: threadListAdapter.threadId,\n      onSwitchToNewThread: onSwitchToNewThread\n        ? async () => {\n            await onSwitchToNewThread();\n            core.applyExternalMessages([]);\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const result = await onSwitchToThread(threadId);\n            core.applyExternalMessages(result.messages);\n            if (result.state) {\n              core.loadExternalState(result.state);\n            }\n          }\n        : undefined,\n    };\n  }, [threadListAdapter, core]);\n\n  const adapters = options.adapters;\n  const adapterAdapters = useMemo(\n    () => ({\n      attachments: adapters?.attachments ?? runtimeAdapters?.attachments,\n      speech: adapters?.speech,\n      dictation: adapters?.dictation,\n      voice: adapters?.voice,\n      feedback: adapters?.feedback,\n      threadList,\n    }),\n    [adapters, runtimeAdapters, threadList],\n  );\n\n  const shared = useExternalStoreSharedOptions(options);\n  const store = useMemo(\n    () => {\n      void _version; // rerender on version change\n\n      return {\n        ...shared,\n        isLoading: core.isLoading,\n        messages: core.getMessages(),\n        state: core.getState(),\n        isRunning: core.isRunning() || hasExecutingTools,\n        unstable_enableToolInvocations: true,\n        setToolStatuses,\n        onNew: (message: AppendMessage) => core.append(message),\n        onEdit: (message: AppendMessage) => core.edit(message),\n        onReload: (parentId: string | null, config: { runConfig?: any }) =>\n          core.reload(parentId, config),\n        onCancel: async () => {\n          core.cancel();\n        },\n        onAddToolResult: (options) => core.addToolResult(options),\n        onResume: (config) => core.resume(config),\n        setMessages: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onImport: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onLoadExternalState: (state: ReadonlyJSONValue) =>\n          core.loadExternalState(state),\n        adapters: adapterAdapters,\n      } satisfies ExternalStoreAdapter<ThreadMessage>;\n    },\n    // _version is intentionally included to trigger re-computation when core state changes via notifyUpdate\n    // toolInvocations intentionally excluded: abort/resume use refs internally and work with stale captures\n    [adapterAdapters, core, _version, hasExecutingTools, shared],\n  );\n\n  const baseRuntime = useExternalStoreRuntime(store);\n\n  const runtime = useMemo<AgUiAssistantRuntime>(() => {\n    const wrapper = Object.create(baseRuntime) as AgUiAssistantRuntime;\n    wrapper.unstable_getPendingInterrupts = () =>\n      core.getPendingInterrupts()?.interrupts ?? [];\n    wrapper.unstable_submitInterruptResponses = (responses) =>\n      core.submitInterruptResponses(responses);\n    return wrapper;\n  }, [baseRuntime, core]);\n\n  useEffect(() => {\n    core.attachRuntime(runtime);\n    return () => {\n      core.detachRuntime();\n    };\n  }, [core, runtime]);\n\n  useEffect(() => {\n    core.__internal_load();\n  }, [core]);\n\n  return runtime;\n}\n"],"mappings":";;;;;;AA+BA,SAAgB,eACd,SACsB;CACtB,MAAM,SAAS,cAAc,WAAW,QAAQ,MAAM,GAAG,CAAC,QAAQ,MAAM,CAAC;CACzE,MAAM,CAAC,UAAU,cAAc,SAAS,CAAC;CACzC,MAAM,eAAe,kBAAkB,YAAY,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;CACnE,MAAM,UAAU,OAAqC,IAAI;CACzD,MAAM,kBAAkB,mBAAmB;CAE3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,IAAI,CAAC,QAAQ,SACX,QAAQ,UAAU,IAAI,sBAAsB;EAC1C,OAAO,QAAQ;EACf;EACA,cAAc,QAAQ,gBAAgB;EACtC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;EAChD;CACF,CAAC;CAGH,MAAM,OAAO,QAAQ;CACrB,KAAK,cAAc;EACjB,OAAO,QAAQ;EACf;EACA,cAAc,QAAQ,gBAAgB;EACtC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;CAClD,CAAC;CAED,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CAEJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,CAAC,CAAC,MACnD,MAAM,GAAG,SAAS,WACrB;CAEA,MAAM,aAAa,cAAc;EAC/B,IAAI,CAAC,mBAAmB,OAAO,KAAA;EAE/B,MAAM,EAAE,qBAAqB,qBAAqB;EAElD,OAAO;GACL,UAAU,kBAAkB;GAC5B,qBAAqB,sBACjB,YAAY;IACV,MAAM,oBAAoB;IAC1B,KAAK,sBAAsB,CAAC,CAAC;GAC/B,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,SAAS,MAAM,iBAAiB,QAAQ;IAC9C,KAAK,sBAAsB,OAAO,QAAQ;IAC1C,IAAI,OAAO,OACT,KAAK,kBAAkB,OAAO,KAAK;GAEvC,IACA,KAAA;EACN;CACF,GAAG,CAAC,mBAAmB,IAAI,CAAC;CAE5B,MAAM,WAAW,QAAQ;CACzB,MAAM,kBAAkB,eACf;EACL,aAAa,UAAU,eAAe,iBAAiB;EACvD,QAAQ,UAAU;EAClB,WAAW,UAAU;EACrB,OAAO,UAAU;EACjB,UAAU,UAAU;EACpB;CACF,IACA;EAAC;EAAU;EAAiB;CAAU,CACxC;CAEA,MAAM,SAAS,8BAA8B,OAAO;CAoCpD,MAAM,cAAc,wBAnCN,cACN;EAGJ,OAAO;GACL,GAAG;GACH,WAAW,KAAK;GAChB,UAAU,KAAK,YAAY;GAC3B,OAAO,KAAK,SAAS;GACrB,WAAW,KAAK,UAAU,KAAK;GAC/B,gCAAgC;GAChC;GACA,QAAQ,YAA2B,KAAK,OAAO,OAAO;GACtD,SAAS,YAA2B,KAAK,KAAK,OAAO;GACrD,WAAW,UAAyB,WAClC,KAAK,OAAO,UAAU,MAAM;GAC9B,UAAU,YAAY;IACpB,KAAK,OAAO;GACd;GACA,kBAAkB,YAAY,KAAK,cAAc,OAAO;GACxD,WAAW,WAAW,KAAK,OAAO,MAAM;GACxC,cAAc,aACZ,KAAK,sBAAsB,QAAQ;GACrC,WAAW,aACT,KAAK,sBAAsB,QAAQ;GACrC,sBAAsB,UACpB,KAAK,kBAAkB,KAAK;GAC9B,UAAU;EACZ;CACF,GAGA;EAAC;EAAiB;EAAM;EAAU;EAAmB;CAAM,CAGb,CAAC;CAEjD,MAAM,UAAU,cAAoC;EAClD,MAAM,UAAU,OAAO,OAAO,WAAW;EACzC,QAAQ,sCACN,KAAK,qBAAqB,CAAC,EAAE,cAAc,CAAC;EAC9C,QAAQ,qCAAqC,cAC3C,KAAK,yBAAyB,SAAS;EACzC,OAAO;CACT,GAAG,CAAC,aAAa,IAAI,CAAC;CAEtB,gBAAgB;EACd,KAAK,cAAc,OAAO;EAC1B,aAAa;GACX,KAAK,cAAc;EACrB;CACF,GAAG,CAAC,MAAM,OAAO,CAAC;CAElB,gBAAgB;EACd,KAAK,gBAAgB;CACvB,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;AACT"}
+{"version":3,"file":"useAgUiRuntime.js","names":[],"sources":["../src/useAgUiRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useCallback, useEffect, useMemo, useRef, useState } from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport type {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport { makeLogger } from \"./runtime/logger\";\nimport type {\n  AgUiInterrupt,\n  AgUiResumeEntry,\n  UseAgUiRuntimeOptions,\n} from \"./runtime/types\";\nimport { AgUiThreadRuntimeCore } from \"./runtime/AgUiThreadRuntimeCore\";\n\nexport type AgUiAssistantRuntime = AssistantRuntime & {\n  unstable_getPendingInterrupts: () => readonly AgUiInterrupt[];\n  unstable_submitInterruptResponses: (\n    responses: readonly AgUiResumeEntry[],\n  ) => Promise<void>;\n};\n\nexport function useAgUiRuntime(\n  options: UseAgUiRuntimeOptions,\n): AgUiAssistantRuntime {\n  const logger = useMemo(() => makeLogger(options.logger), [options.logger]);\n  const [_version, setVersion] = useState(0);\n  const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);\n  const coreRef = useRef<AgUiThreadRuntimeCore | null>(null);\n  const runtimeAdapters = useRuntimeAdapters();\n\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  if (!coreRef.current) {\n    coreRef.current = new AgUiThreadRuntimeCore({\n      agent: options.agent,\n      logger,\n      showThinking: options.showThinking ?? true,\n      ...(options.onError && { onError: options.onError }),\n      ...(options.onCancel && { onCancel: options.onCancel }),\n      ...(historyAdapter && { history: historyAdapter }),\n      notifyUpdate,\n    });\n  }\n\n  const core = coreRef.current;\n  core.updateOptions({\n    agent: options.agent,\n    logger,\n    showThinking: options.showThinking ?? true,\n    ...(options.onError && { onError: options.onError }),\n    ...(options.onCancel && { onCancel: options.onCancel }),\n    ...(historyAdapter && { history: historyAdapter }),\n  });\n\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n\n  const threadList = useMemo(() => {\n    if (!threadListAdapter) return undefined;\n\n    const { onSwitchToNewThread, onSwitchToThread, ...rest } =\n      threadListAdapter;\n\n    return {\n      ...rest,\n      onSwitchToNewThread: onSwitchToNewThread\n        ? async () => {\n            await onSwitchToNewThread();\n            core.applyExternalMessages([]);\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            // Clear before the thread id flips, or the old messages leak\n            // into the new thread as a sibling branch.\n            core.applyExternalMessages([]);\n            const result = await onSwitchToThread(threadId);\n            core.applyExternalMessages(result.messages);\n            if (result.state) {\n              core.loadExternalState(result.state);\n            }\n            if (result.unstable_resume) {\n              void core.resumeInFlightRun(result.messages);\n            }\n          }\n        : undefined,\n    };\n  }, [threadListAdapter, core]);\n\n  const adapters = options.adapters;\n  const adapterAdapters = useMemo(\n    () => ({\n      attachments: adapters?.attachments ?? runtimeAdapters?.attachments,\n      speech: adapters?.speech,\n      dictation: adapters?.dictation,\n      voice: adapters?.voice,\n      feedback: adapters?.feedback,\n      threadList,\n    }),\n    [adapters, runtimeAdapters, threadList],\n  );\n\n  const shared = useExternalStoreSharedOptions(options);\n  const store = useMemo(\n    () => {\n      void _version; // rerender on version change\n\n      return {\n        ...shared,\n        isLoading: core.isLoading,\n        messages: core.getMessages(),\n        state: core.getState(),\n        isRunning: core.isRunning() || hasExecutingTools,\n        unstable_enableToolInvocations: true,\n        setToolStatuses,\n        onNew: (message: AppendMessage) => core.append(message),\n        onEdit: (message: AppendMessage) => core.edit(message),\n        onReload: (parentId: string | null, config: { runConfig?: any }) =>\n          core.reload(parentId, config),\n        onCancel: async () => {\n          core.cancel();\n        },\n        onAddToolResult: (options) => core.addToolResult(options),\n        onResume: (config) => core.resume(config),\n        setMessages: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onImport: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onLoadExternalState: (state: ReadonlyJSONValue) =>\n          core.loadExternalState(state),\n        adapters: adapterAdapters,\n      } satisfies ExternalStoreAdapter<ThreadMessage>;\n    },\n    // _version is intentionally included to trigger re-computation when core state changes via notifyUpdate\n    // toolInvocations intentionally excluded: abort/resume use refs internally and work with stale captures\n    [adapterAdapters, core, _version, hasExecutingTools, shared],\n  );\n\n  const baseRuntime = useExternalStoreRuntime(store);\n\n  const runtime = useMemo<AgUiAssistantRuntime>(() => {\n    const wrapper = Object.create(baseRuntime) as AgUiAssistantRuntime;\n    wrapper.unstable_getPendingInterrupts = () =>\n      core.getPendingInterrupts()?.interrupts ?? [];\n    wrapper.unstable_submitInterruptResponses = (responses) =>\n      core.submitInterruptResponses(responses);\n    return wrapper;\n  }, [baseRuntime, core]);\n\n  useEffect(() => {\n    core.attachRuntime(runtime);\n    return () => {\n      core.detachRuntime();\n    };\n  }, [core, runtime]);\n\n  useEffect(() => {\n    core.__internal_load();\n  }, [core]);\n\n  return runtime;\n}\n"],"mappings":";;;;;;AA+BA,SAAgB,eACd,SACsB;CACtB,MAAM,SAAS,cAAc,WAAW,QAAQ,MAAM,GAAG,CAAC,QAAQ,MAAM,CAAC;CACzE,MAAM,CAAC,UAAU,cAAc,SAAS,CAAC;CACzC,MAAM,eAAe,kBAAkB,YAAY,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;CACnE,MAAM,UAAU,OAAqC,IAAI;CACzD,MAAM,kBAAkB,mBAAmB;CAE3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,IAAI,CAAC,QAAQ,SACX,QAAQ,UAAU,IAAI,sBAAsB;EAC1C,OAAO,QAAQ;EACf;EACA,cAAc,QAAQ,gBAAgB;EACtC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;EAChD;CACF,CAAC;CAGH,MAAM,OAAO,QAAQ;CACrB,KAAK,cAAc;EACjB,OAAO,QAAQ;EACf;EACA,cAAc,QAAQ,gBAAgB;EACtC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;CAClD,CAAC;CAED,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CAEJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,CAAC,CAAC,MACnD,MAAM,GAAG,SAAS,WACrB;CAEA,MAAM,aAAa,cAAc;EAC/B,IAAI,CAAC,mBAAmB,OAAO,KAAA;EAE/B,MAAM,EAAE,qBAAqB,kBAAkB,GAAG,SAChD;EAEF,OAAO;GACL,GAAG;GACH,qBAAqB,sBACjB,YAAY;IACV,MAAM,oBAAoB;IAC1B,KAAK,sBAAsB,CAAC,CAAC;GAC/B,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAG1B,KAAK,sBAAsB,CAAC,CAAC;IAC7B,MAAM,SAAS,MAAM,iBAAiB,QAAQ;IAC9C,KAAK,sBAAsB,OAAO,QAAQ;IAC1C,IAAI,OAAO,OACT,KAAK,kBAAkB,OAAO,KAAK;IAErC,IAAI,OAAO,iBACT,KAAU,kBAAkB,OAAO,QAAQ;GAE/C,IACA,KAAA;EACN;CACF,GAAG,CAAC,mBAAmB,IAAI,CAAC;CAE5B,MAAM,WAAW,QAAQ;CACzB,MAAM,kBAAkB,eACf;EACL,aAAa,UAAU,eAAe,iBAAiB;EACvD,QAAQ,UAAU;EAClB,WAAW,UAAU;EACrB,OAAO,UAAU;EACjB,UAAU,UAAU;EACpB;CACF,IACA;EAAC;EAAU;EAAiB;CAAU,CACxC;CAEA,MAAM,SAAS,8BAA8B,OAAO;CAoCpD,MAAM,cAAc,wBAnCN,cACN;EAGJ,OAAO;GACL,GAAG;GACH,WAAW,KAAK;GAChB,UAAU,KAAK,YAAY;GAC3B,OAAO,KAAK,SAAS;GACrB,WAAW,KAAK,UAAU,KAAK;GAC/B,gCAAgC;GAChC;GACA,QAAQ,YAA2B,KAAK,OAAO,OAAO;GACtD,SAAS,YAA2B,KAAK,KAAK,OAAO;GACrD,WAAW,UAAyB,WAClC,KAAK,OAAO,UAAU,MAAM;GAC9B,UAAU,YAAY;IACpB,KAAK,OAAO;GACd;GACA,kBAAkB,YAAY,KAAK,cAAc,OAAO;GACxD,WAAW,WAAW,KAAK,OAAO,MAAM;GACxC,cAAc,aACZ,KAAK,sBAAsB,QAAQ;GACrC,WAAW,aACT,KAAK,sBAAsB,QAAQ;GACrC,sBAAsB,UACpB,KAAK,kBAAkB,KAAK;GAC9B,UAAU;EACZ;CACF,GAGA;EAAC;EAAiB;EAAM;EAAU;EAAmB;CAAM,CAGb,CAAC;CAEjD,MAAM,UAAU,cAAoC;EAClD,MAAM,UAAU,OAAO,OAAO,WAAW;EACzC,QAAQ,sCACN,KAAK,qBAAqB,CAAC,EAAE,cAAc,CAAC;EAC9C,QAAQ,qCAAqC,cAC3C,KAAK,yBAAyB,SAAS;EACzC,OAAO;CACT,GAAG,CAAC,aAAa,IAAI,CAAC;CAEtB,gBAAgB;EACd,KAAK,cAAc,OAAO;EAC1B,aAAa;GACX,KAAK,cAAc;EACrB;CACF,GAAG,CAAC,MAAM,OAAO,CAAC;CAElB,gBAAgB;EACd,KAAK,gBAAgB;CACvB,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;AACT"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ag-ui",
-  "version": "0.0.38",
+  "version": "0.0.39",
   "description": "AG-UI protocol adapter for assistant-ui",
   "keywords": [
     "ag-ui",
@@ -30,8 +30,8 @@
   "sideEffects": false,
   "dependencies": {
     "@ag-ui/client": "^0.0.55",
-    "@assistant-ui/core": "^0.2.14",
-    "assistant-stream": "^0.3.21",
+    "@assistant-ui/core": "^0.2.15",
+    "assistant-stream": "^0.3.22",
     "fast-json-patch": "^3.1.1"
   },
   "peerDependencies": {
@@ -48,7 +48,7 @@
     "react": "^19.2.7",
     "vitest": "^4.1.8",
     "zod": "^4.4.3",
-    "@assistant-ui/x-buildutils": "0.0.12"
+    "@assistant-ui/x-buildutils": "0.0.13"
   },
   "publishConfig": {
     "access": "public",

package/src/runtime/AgUiThreadRuntimeCore.ts

@@ -13,7 +13,7 @@ import type {
   ThreadMessage,
   ToolCallMessagePart,
 } from "@assistant-ui/core";
-import type { HttpAgent } from "@ag-ui/client";
+import type { AbstractAgent } from "@ag-ui/client";
 import jsonpatch, { type Operation } from "fast-json-patch";
 import type { Logger } from "./logger";
 import type { AgUiEvent, AgUiInterrupt, AgUiResumeEntry } from "./types";
@@ -22,6 +22,7 @@ import {
   AG_UI_METADATA_NAMESPACE,
   type AgUiCustomMetadata,
   RunAggregator,
+  tryParseJSON,
 } from "./adapter/run-aggregator";
 import {
   fromAgUiMessages,
@@ -48,7 +49,7 @@ type ResumeRunConfig = {
 };
 
 type CoreOptions = {
-  agent: HttpAgent;
+  agent: AbstractAgent;
   logger: Logger;
   showThinking: boolean;
   onError?: (error: Error) => void;
@@ -60,7 +61,7 @@ type CoreOptions = {
 const FALLBACK_USER_STATUS = { type: "complete", reason: "unknown" } as const;
 
 export class AgUiThreadRuntimeCore {
-  private agent: HttpAgent;
+  private agent: AbstractAgent;
   private logger: Logger;
   private showThinking: boolean;
   private onError: ((error: Error) => void) | undefined;
@@ -214,6 +215,30 @@ export class AgUiThreadRuntimeCore {
     );
   }
 
+  async resumeInFlightRun(messages: readonly ThreadMessage[]): Promise<void> {
+    // Without a resume stream startRun would re-run the agent from scratch.
+    const resumeStream = this.history?.resume?.bind(this.history);
+    if (!resumeStream) {
+      const error = new Error(
+        "[agui] unstable_resume requires a ThreadHistoryAdapter with a resume() method; skipping resume after thread switch",
+      );
+      this.logger.error?.(error.message);
+      this.onError?.(error);
+      return;
+    }
+    const parentId = messages.at(-1)?.id ?? null;
+    try {
+      await this.startRun(
+        parentId,
+        this.lastRunConfig,
+        undefined,
+        resumeStream,
+      );
+    } catch {
+      // startRun already reported via onError; don't reject the switch.
+    }
+  }
+
   private assertNoPendingInterrupts(): void {
     if (!this.getPendingInterrupts()) return;
     throw new Error(
@@ -396,27 +421,37 @@ export class AgUiThreadRuntimeCore {
     this.maybeResumeAfterToolResults(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.
+  // The continuation fires whether the frontend result lands before
+  // RUN_FINISHED (the status flips to requires-action only later, while the
+  // run is still draining) or after it.
   private maybeResumeAfterToolResults(messageId: string): void {
+    if (!this.maybeCompleteAfterToolResults(messageId)) return;
+
+    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 maybeCompleteAfterToolResults(messageId: string): boolean {
     const message = this.messages.find((m) => m.id === messageId);
-    if (!message || message.role !== "assistant") return;
+    if (!message || message.role !== "assistant") return false;
     const assistant = message as ThreadAssistantMessage;
     if (
       assistant.status?.type !== "requires-action" ||
       assistant.status.reason !== "tool-calls"
     ) {
-      return;
+      return false;
     }
     const allResolved = assistant.content.every(
       (part) =>
         part.type !== "tool-call" ||
         ("result" in part && part.result !== undefined),
     );
-    if (!allResolved) return;
+    if (!allResolved) return false;
 
     this.messages = this.messages.map((m) =>
       m.id === messageId && m.role === "assistant"
@@ -428,14 +463,7 @@ export class AgUiThreadRuntimeCore {
     );
     this.notifyUpdate();
     this.persistAssistantHistory(messageId);
-
-    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);
+    return true;
   }
 
   private startResumeRun(messageId: string): void {
@@ -908,11 +936,57 @@ export class AgUiThreadRuntimeCore {
         this.importMessagesSnapshot(event.messages);
         return;
       }
+      case "TOOL_CALL_RESULT": {
+        if (!aggregator.hasToolCall(event.toolCallId)) {
+          const messageId = this.findMessageIdForToolCall(event.toolCallId);
+          if (messageId !== undefined) {
+            this.applyCrossRunToolResult(messageId, event);
+            return;
+          }
+        }
+        aggregator.handle(event);
+        return;
+      }
       default:
         aggregator.handle(event);
     }
   }
 
+  private applyCrossRunToolResult(
+    messageId: string,
+    event: Extract<AgUiEvent, { type: "TOOL_CALL_RESULT" }>,
+  ): void {
+    let updated = false;
+    this.messages = this.messages.map((message) => {
+      if (message.id !== messageId || message.role !== "assistant")
+        return message;
+      const assistant = message as ThreadAssistantMessage;
+      let matchedToolCall = false;
+      const content = assistant.content.map((part) => {
+        if (part.type !== "tool-call" || part.toolCallId !== event.toolCallId)
+          return part;
+        matchedToolCall = true;
+        return {
+          ...part,
+          result: tryParseJSON(event.content ?? "") as ReadonlyJSONValue,
+          ...(event.role === "tool" ? { isError: false } : {}),
+          ...(event.messageId
+            ? { unstable_toolMessageId: event.messageId }
+            : {}),
+        };
+      });
+      if (!matchedToolCall) return message;
+      updated = true;
+      return { ...assistant, content };
+    });
+
+    if (!updated) return;
+    this.notifyUpdate();
+    // Not maybeResumeAfterToolResults: the delivering run is already in
+    // flight, and a resume from the owner would reset the head past it.
+    this.maybeCompleteAfterToolResults(messageId);
+  }
+
   private importMessagesSnapshot(rawMessages: readonly unknown[]) {
     try {
       const normalized = fromAgUiMessages(rawMessages, {

package/src/runtime/adapter/run-aggregator.ts

@@ -82,6 +82,10 @@ export class RunAggregator {
     this.onServerMessageId = options.onServerMessageId;
   }
 
+  hasToolCall(toolCallId: string): boolean {
+    return this.toolCalls.has(toolCallId);
+  }
+
   handle(event: AgUiEvent): void {
     switch (event.type) {
       case "RUN_STARTED": {
@@ -388,7 +392,7 @@ export class RunAggregator {
     ) {
       this.partOrder.push({ kind: "tool-call", toolCallId: id });
     }
-    entry.result = this.tryParseJSON(content);
+    entry.result = tryParseJSON(content);
     entry.isError = isError;
     if (toolMessageId) {
       entry.toolMessageId = toolMessageId;
@@ -396,15 +400,6 @@ export class RunAggregator {
     this.lastResolvedToolCallId = id;
   }
 
-  private tryParseJSON(value: string): unknown {
-    if (!value) return value;
-    try {
-      return JSON.parse(value);
-    } catch {
-      return value;
-    }
-  }
-
   private emit(): void {
     const snapshot: ThreadAssistantMessagePart[] = [];
 
@@ -542,3 +537,12 @@ export class RunAggregator {
     this.emit();
   }
 }
+
+export function tryParseJSON(value: string): unknown {
+  if (!value) return value;
+  try {
+    return JSON.parse(value);
+  } catch {
+    return value;
+  }
+}

package/src/runtime/adapter/subscriber.ts

@@ -38,6 +38,16 @@ type Subscriber = {
   onRunFailed?: (payload: { error: Error }) => void;
 };
 
+const isAbortError = (error: Error): boolean => {
+  if (error.name === "AbortError") return true;
+  const message = typeof error.message === "string" ? error.message : "";
+  return (
+    message === "Fetch is aborted" ||
+    message === "signal is aborted without reason" ||
+    message === "component unmounted"
+  );
+};
+
 const ensureEvent = (
   raw: unknown,
   type: AgUiEvent["type"],
@@ -143,7 +153,12 @@ export const createAgUiSubscriber = (
       dispatch({ type: "RUN_FINISHED", runId });
     },
     onRunFailed: ({ error }) => {
+      runFinishedDispatched = true;
       onRunFailed?.(error);
+      if (isAbortError(error)) {
+        dispatch({ type: "RUN_CANCELLED" } satisfies AgUiEvent);
+        return;
+      }
       const message =
         typeof error.message === "string" ? error.message : "Run failed";
       const code =

package/src/runtime/types.ts

@@ -2,29 +2,42 @@ import type {
   AttachmentAdapter,
   DictationAdapter,
   ExternalStoreSharedOptions,
+  ExternalStoreThreadListAdapter,
   FeedbackAdapter,
   RealtimeVoiceAdapter,
   SpeechSynthesisAdapter,
   ThreadHistoryAdapter,
   ThreadMessage,
 } from "@assistant-ui/core";
-import type { HttpAgent } from "@ag-ui/client";
+import type { AbstractAgent } from "@ag-ui/client";
 import type { Logger } from "./logger";
 import type { ReadonlyJSONValue } from "assistant-stream/utils";
 
 /**
  * @experimental This API is still under active development and might change without notice.
+ *
+ * Same as ExternalStoreThreadListAdapter, except `onSwitchToThread` returns
+ * the messages (and optional state) to hydrate the thread with.
  */
-export type UseAgUiThreadListAdapter = {
-  threadId?: string | undefined;
-  onSwitchToNewThread?: (() => Promise<void> | void) | undefined;
+type SwitchToThreadResult = {
+  messages: readonly ThreadMessage[];
+  state?: ReadonlyJSONValue;
+  /**
+   * Set when the thread has a run in flight. The runtime resumes the run
+   * after hydrating, the same way `ThreadHistoryAdapter.load()` does when it
+   * returns `unstable_resume: true`.
+   */
+  unstable_resume?: boolean;
+};
+
+export type UseAgUiThreadListAdapter = Omit<
+  ExternalStoreThreadListAdapter,
+  "onSwitchToThread"
+> & {
   onSwitchToThread?:
-    | ((threadId: string) =>
-        | Promise<{
-            messages: readonly ThreadMessage[];
-            state?: ReadonlyJSONValue;
-          }>
-        | { messages: readonly ThreadMessage[]; state?: ReadonlyJSONValue })
+    | ((
+        threadId: string,
+      ) => Promise<SwitchToThreadResult> | SwitchToThreadResult)
     | undefined;
 };
 
@@ -42,7 +55,7 @@ export type UseAgUiRuntimeAdapters = {
 };
 
 export type UseAgUiRuntimeOptions = ExternalStoreSharedOptions & {
-  agent: HttpAgent;
+  agent: AbstractAgent;
   logger?: Partial<Logger>;
   showThinking?: boolean;
   onError?: (e: Error) => void;

package/src/useAgUiRuntime.ts

@@ -74,10 +74,11 @@ export function useAgUiRuntime(
   const threadList = useMemo(() => {
     if (!threadListAdapter) return undefined;
 
-    const { onSwitchToNewThread, onSwitchToThread } = threadListAdapter;
+    const { onSwitchToNewThread, onSwitchToThread, ...rest } =
+      threadListAdapter;
 
     return {
-      threadId: threadListAdapter.threadId,
+      ...rest,
       onSwitchToNewThread: onSwitchToNewThread
         ? async () => {
             await onSwitchToNewThread();
@@ -86,11 +87,17 @@ export function useAgUiRuntime(
         : undefined,
       onSwitchToThread: onSwitchToThread
         ? async (threadId: string) => {
+            // Clear before the thread id flips, or the old messages leak
+            // into the new thread as a sibling branch.
+            core.applyExternalMessages([]);
             const result = await onSwitchToThread(threadId);
             core.applyExternalMessages(result.messages);
             if (result.state) {
               core.loadExternalState(result.state);
             }
+            if (result.unstable_resume) {
+              void core.resumeInFlightRun(result.messages);
+            }
           }
         : undefined,
     };