@assistant-ui/react-ag-ui 0.0.44 → 0.0.46

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,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,QAgCA,cAAA;EAAA,QA2BA,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"}
+{"version":3,"file":"run-aggregator.d.ts","names":[],"sources":["../../../src/runtime/adapter/run-aggregator.ts"],"mappings":";;;;cAaa;KAED;EACV,aAAa;;KAGV,QAAQ,QAAQ;cAiBR;cAEA,gBACX,mBACC,SAAS;KAGA;EACV;EACA,QAAQ;EACR,MAAM;EACN,qBAAqB;;;;;;;;cASV;mBACM;mBACA;mBACA;mBACA;UAET;UACA;mBACS;UAIT;mBACS;UACT;UACA;mBACS;UACT;mBACS;UAKT;UACA;UAEA;UACA;UACA;EAEI,YAAA,SAAS;EAOrB,YAAY;EAIZ,OAAO,OAAO;UAiNN;UAMA;UAIA;UAKA;UAOA;UAiBA;UAWA;UAMA;UAQA;UAiCA;UA2BA;UAgBA;UAkDA;UA4EA;UASA;UAgCA;UAcA;UAYA;;iBAOM,aAAa"}

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

@@ -3,6 +3,7 @@ import { isMcpAppUri } from "@assistant-ui/core";
 //#region src/runtime/adapter/run-aggregator.ts
 const AG_UI_METADATA_NAMESPACE = "agui";
 const MCP_APPS_ACTIVITY_TYPE = "mcp-apps";
+const isPlainObject = (value) => !!value && typeof value === "object" && !Array.isArray(value);
 /**
 * Collects AG-UI events into assistant-ui run snapshots that can be yielded from a ChatModelAdapter.
 *
@@ -157,16 +158,30 @@ var RunAggregator = class {
 				this.emit();
 				break;
 			case "TOOL_CALL_RESULT":
-				this.finishToolCall(event.toolCallId, event.content ?? "", event.role === "tool" ? false : void 0, event.messageId);
+				this.finishToolCall(event.toolCallId, event.content ?? "", typeof event.mcpResult?.isError === "boolean" ? event.mcpResult.isError : event.role === "tool" ? false : void 0, event.messageId, event.mcpResult);
 				this.emit();
 				break;
 			case "ACTIVITY_SNAPSHOT": {
-				if (event.activityType !== MCP_APPS_ACTIVITY_TYPE) break;
-				const id = this.lastResolvedToolCallId;
-				const entry = id ? this.toolCalls.get(id) : void 0;
+				if (event.activityType !== "mcp-apps") break;
+				const toolCallId = event.content.toolCallId;
+				const entry = typeof toolCallId === "string" ? this.toolCalls.get(toolCallId) : this.lastResolvedToolCallId ? this.toolCalls.get(this.lastResolvedToolCallId) : void 0;
 				const resourceUri = event.content.resourceUri;
 				if (entry && typeof resourceUri === "string" && isMcpAppUri(resourceUri)) {
 					entry.mcpAppResourceUri = resourceUri;
+					const id = event.content.serverId;
+					const hash = event.content.serverHash;
+					if (typeof id === "string" && id.length > 0) entry.mcpAppServerId = id;
+					else if (typeof hash === "string" && hash.length > 0) entry.mcpAppServerId = hash;
+					const result = event.content.result;
+					if (isPlainObject(result)) {
+						if (entry.result !== void 0 && entry.modelContent === void 0) entry.modelContent = [{
+							type: "text",
+							text: typeof entry.result === "string" ? entry.result : JSON.stringify(entry.result)
+						}];
+						entry.result = result;
+						entry.snapshotResultApplied = true;
+						if (typeof result.isError === "boolean") entry.isError = result.isError;
+					}
 					this.emit();
 				}
 				break;
@@ -238,7 +253,8 @@ var RunAggregator = class {
 			argsText: "",
 			parsedArgs: void 0,
 			result: void 0,
-			isError: void 0
+			isError: void 0,
+			snapshotResultApplied: false
 		};
 		if (parentMessageId) state.parentMessageId = parentMessageId;
 		this.toolCalls.set(id, state);
@@ -278,7 +294,7 @@ var RunAggregator = class {
 			entry.parsedArgs = void 0;
 		}
 	}
-	finishToolCall(id, content, isError, toolMessageId) {
+	finishToolCall(id, content, isError, toolMessageId, mcpResult) {
 		if (!id) return;
 		let entry = this.toolCalls.get(id);
 		if (!entry) {
@@ -288,7 +304,8 @@ var RunAggregator = class {
 				argsText: "",
 				parsedArgs: void 0,
 				result: void 0,
-				isError: void 0
+				isError: void 0,
+				snapshotResultApplied: false
 			};
 			this.toolCalls.set(id, entry);
 		}
@@ -296,8 +313,24 @@ var RunAggregator = class {
 			kind: "tool-call",
 			toolCallId: id
 		});
-		entry.result = tryParseJSON(content);
-		entry.isError = isError;
+		if (mcpResult !== void 0 && !entry.snapshotResultApplied) {
+			entry.result = mcpResult;
+			entry.modelContent = [{
+				type: "text",
+				text: content
+			}];
+			entry.snapshotResultApplied = true;
+			entry.isError = isError;
+		} else if (entry.snapshotResultApplied) {
+			if (entry.modelContent === void 0) entry.modelContent = [{
+				type: "text",
+				text: content
+			}];
+			if (entry.isError === void 0) entry.isError = isError;
+		} else {
+			entry.result = tryParseJSON(content);
+			entry.isError = isError;
+		}
 		if (toolMessageId) entry.toolMessageId = toolMessageId;
 		this.lastResolvedToolCallId = id;
 	}
@@ -331,8 +364,12 @@ var RunAggregator = class {
 				args: entry.parsedArgs ?? {},
 				argsText: entry.argsText,
 				...entry.result !== void 0 ? { result: entry.result } : {},
+				...entry.modelContent !== void 0 ? { modelContent: entry.modelContent } : {},
 				...entry.isError !== void 0 ? { isError: entry.isError } : {},
-				...entry.mcpAppResourceUri ? { mcp: { app: { resourceUri: entry.mcpAppResourceUri } } } : {},
+				...entry.mcpAppResourceUri ? { mcp: { app: {
+					resourceUri: entry.mcpAppResourceUri,
+					...entry.mcpAppServerId ? { serverId: entry.mcpAppServerId } : {}
+				} } } : {},
 				...entry.parentMessageId ? { parentId: entry.parentMessageId } : {},
 				...entry.toolMessageId ? { unstable_toolMessageId: entry.toolMessageId } : {}
 			};
@@ -405,6 +442,6 @@ function tryParseJSON(value) {
 	}
 }
 //#endregion
-export { AG_UI_METADATA_NAMESPACE, RunAggregator, tryParseJSON };
+export { AG_UI_METADATA_NAMESPACE, MCP_APPS_ACTIVITY_TYPE, RunAggregator, isPlainObject, 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  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.insertToolPart(id, parentMessageId);\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  // The message and tool-call channels are unordered on the wire, so anchor a\n  // tool call under its parentMessageId text part instead of appending it.\n  private insertToolPart(id: string, parentMessageId?: string) {\n    const entry = { kind: \"tool-call\", toolCallId: id } as const;\n    if (parentMessageId) {\n      const parentIndex = this.partOrder.findIndex(\n        (part) => part.kind === \"text\" && part.key === parentMessageId,\n      );\n      if (parentIndex !== -1) {\n        let insertAt = parentIndex + 1;\n        while (insertAt < this.partOrder.length) {\n          const part = this.partOrder[insertAt];\n          if (\n            part?.kind === \"tool-call\" &&\n            this.toolCalls.get(part.toolCallId)?.parentMessageId ===\n              parentMessageId\n          ) {\n            insertAt += 1;\n            continue;\n          }\n          break;\n        }\n        this.partOrder.splice(insertAt, 0, entry);\n        return;\n      }\n    }\n    this.partOrder.push(entry);\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,eAAe,IAAI,eAAe;EAEzC,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;CAIA,eAAuB,IAAY,iBAA0B;EAC3D,MAAM,QAAQ;GAAE,MAAM;GAAa,YAAY;EAAG;EAClD,IAAI,iBAAiB;GACnB,MAAM,cAAc,KAAK,UAAU,WAChC,SAAS,KAAK,SAAS,UAAU,KAAK,QAAQ,eACjD;GACA,IAAI,gBAAgB,IAAI;IACtB,IAAI,WAAW,cAAc;IAC7B,OAAO,WAAW,KAAK,UAAU,QAAQ;KACvC,MAAM,OAAO,KAAK,UAAU;KAC5B,IACE,MAAM,SAAS,eACf,KAAK,UAAU,IAAI,KAAK,UAAU,CAAC,EAAE,oBACnC,iBACF;MACA,YAAY;MACZ;KACF;KACA;IACF;IACA,KAAK,UAAU,OAAO,UAAU,GAAG,KAAK;IACxC;GACF;EACF;EACA,KAAK,UAAU,KAAK,KAAK;CAC3B;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"}
+{"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  type ToolModelContentPart,\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  mcpAppServerId?: string;\n  modelContent?: ToolModelContentPart[];\n  snapshotResultApplied: boolean;\n};\n\nexport const MCP_APPS_ACTIVITY_TYPE = \"mcp-apps\";\n\nexport const isPlainObject = (\n  value: unknown,\n): value is Record<string, unknown> =>\n  !!value && typeof value === \"object\" && !Array.isArray(value);\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          typeof event.mcpResult?.isError === \"boolean\"\n            ? event.mcpResult.isError\n            : event.role === \"tool\"\n              ? false\n              : undefined,\n          event.messageId,\n          event.mcpResult,\n        );\n        this.emit();\n        break;\n      }\n      case \"ACTIVITY_SNAPSHOT\": {\n        if (event.activityType !== MCP_APPS_ACTIVITY_TYPE) break;\n        const toolCallId = event.content.toolCallId;\n        const entry =\n          typeof toolCallId === \"string\"\n            ? this.toolCalls.get(toolCallId)\n            : this.lastResolvedToolCallId\n              ? this.toolCalls.get(this.lastResolvedToolCallId)\n              : undefined;\n        const resourceUri = event.content.resourceUri;\n        if (\n          entry &&\n          typeof resourceUri === \"string\" &&\n          isMcpAppUri(resourceUri)\n        ) {\n          entry.mcpAppResourceUri = resourceUri;\n          const id = event.content.serverId;\n          const hash = event.content.serverHash;\n          if (typeof id === \"string\" && id.length > 0) {\n            entry.mcpAppServerId = id;\n          } else if (typeof hash === \"string\" && hash.length > 0) {\n            entry.mcpAppServerId = hash;\n          }\n          const result = event.content.result;\n          if (isPlainObject(result)) {\n            if (\n              entry.result !== undefined &&\n              entry.modelContent === undefined\n            ) {\n              entry.modelContent = [\n                {\n                  type: \"text\",\n                  text:\n                    typeof entry.result === \"string\"\n                      ? entry.result\n                      : JSON.stringify(entry.result),\n                },\n              ];\n            }\n            entry.result = result;\n            entry.snapshotResultApplied = true;\n            if (typeof result.isError === \"boolean\") {\n              entry.isError = result.isError;\n            }\n          }\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.insertToolPart(id, parentMessageId);\n    }\n    const state: ToolCallState = {\n      toolCallId: id,\n      toolCallName: name ?? \"tool\",\n      argsText: \"\",\n      parsedArgs: undefined,\n      result: undefined,\n      isError: undefined,\n      snapshotResultApplied: false,\n    };\n    if (parentMessageId) {\n      state.parentMessageId = parentMessageId;\n    }\n    this.toolCalls.set(id, state);\n  }\n\n  // The message and tool-call channels are unordered on the wire, so anchor a\n  // tool call under its parentMessageId text part instead of appending it.\n  private insertToolPart(id: string, parentMessageId?: string) {\n    const entry = { kind: \"tool-call\", toolCallId: id } as const;\n    if (parentMessageId) {\n      const parentIndex = this.partOrder.findIndex(\n        (part) => part.kind === \"text\" && part.key === parentMessageId,\n      );\n      if (parentIndex !== -1) {\n        let insertAt = parentIndex + 1;\n        while (insertAt < this.partOrder.length) {\n          const part = this.partOrder[insertAt];\n          if (\n            part?.kind === \"tool-call\" &&\n            this.toolCalls.get(part.toolCallId)?.parentMessageId ===\n              parentMessageId\n          ) {\n            insertAt += 1;\n            continue;\n          }\n          break;\n        }\n        this.partOrder.splice(insertAt, 0, entry);\n        return;\n      }\n    }\n    this.partOrder.push(entry);\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    mcpResult?: Record<string, unknown>,\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        snapshotResultApplied: false,\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    if (mcpResult !== undefined && !entry.snapshotResultApplied) {\n      entry.result = mcpResult;\n      entry.modelContent = [{ type: \"text\", text: content }];\n      entry.snapshotResultApplied = true;\n      entry.isError = isError;\n    } else if (entry.snapshotResultApplied) {\n      if (entry.modelContent === undefined) {\n        entry.modelContent = [{ type: \"text\", text: content }];\n      }\n      if (entry.isError === undefined) {\n        entry.isError = isError;\n      }\n    } else {\n      entry.result = tryParseJSON(content);\n      entry.isError = isError;\n    }\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.modelContent !== undefined\n          ? { modelContent: entry.modelContent }\n          : {}),\n        ...(entry.isError !== undefined ? { isError: entry.isError } : {}),\n        ...(entry.mcpAppResourceUri\n          ? {\n              mcp: {\n                app: {\n                  resourceUri: entry.mcpAppResourceUri,\n                  ...(entry.mcpAppServerId\n                    ? { serverId: entry.mcpAppServerId }\n                    : {}),\n                },\n              },\n            }\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":";;;AAaA,MAAa,2BAA2B;AAuBxC,MAAa,yBAAyB;AAEtC,MAAa,iBACX,UAEA,CAAC,CAAC,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;;;;;;;AAe9D,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,OAAO,MAAM,WAAW,YAAY,YAChC,MAAM,UAAU,UAChB,MAAM,SAAS,SACb,QACA,KAAA,GACN,MAAM,WACN,MAAM,SACR;IACA,KAAK,KAAK;IACV;GAEF,KAAK,qBAAqB;IACxB,IAAI,MAAM,iBAAA,YAAyC;IACnD,MAAM,aAAa,MAAM,QAAQ;IACjC,MAAM,QACJ,OAAO,eAAe,WAClB,KAAK,UAAU,IAAI,UAAU,IAC7B,KAAK,yBACH,KAAK,UAAU,IAAI,KAAK,sBAAsB,IAC9C,KAAA;IACR,MAAM,cAAc,MAAM,QAAQ;IAClC,IACE,SACA,OAAO,gBAAgB,YACvB,YAAY,WAAW,GACvB;KACA,MAAM,oBAAoB;KAC1B,MAAM,KAAK,MAAM,QAAQ;KACzB,MAAM,OAAO,MAAM,QAAQ;KAC3B,IAAI,OAAO,OAAO,YAAY,GAAG,SAAS,GACxC,MAAM,iBAAiB;UAClB,IAAI,OAAO,SAAS,YAAY,KAAK,SAAS,GACnD,MAAM,iBAAiB;KAEzB,MAAM,SAAS,MAAM,QAAQ;KAC7B,IAAI,cAAc,MAAM,GAAG;MACzB,IACE,MAAM,WAAW,KAAA,KACjB,MAAM,iBAAiB,KAAA,GAEvB,MAAM,eAAe,CACnB;OACE,MAAM;OACN,MACE,OAAO,MAAM,WAAW,WACpB,MAAM,SACN,KAAK,UAAU,MAAM,MAAM;MACnC,CACF;MAEF,MAAM,SAAS;MACf,MAAM,wBAAwB;MAC9B,IAAI,OAAO,OAAO,YAAY,WAC5B,MAAM,UAAU,OAAO;KAE3B;KACA,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,eAAe,IAAI,eAAe;EAEzC,MAAM,QAAuB;GAC3B,YAAY;GACZ,cAAc,QAAQ;GACtB,UAAU;GACV,YAAY,KAAA;GACZ,QAAQ,KAAA;GACR,SAAS,KAAA;GACT,uBAAuB;EACzB;EACA,IAAI,iBACF,MAAM,kBAAkB;EAE1B,KAAK,UAAU,IAAI,IAAI,KAAK;CAC9B;CAIA,eAAuB,IAAY,iBAA0B;EAC3D,MAAM,QAAQ;GAAE,MAAM;GAAa,YAAY;EAAG;EAClD,IAAI,iBAAiB;GACnB,MAAM,cAAc,KAAK,UAAU,WAChC,SAAS,KAAK,SAAS,UAAU,KAAK,QAAQ,eACjD;GACA,IAAI,gBAAgB,IAAI;IACtB,IAAI,WAAW,cAAc;IAC7B,OAAO,WAAW,KAAK,UAAU,QAAQ;KACvC,MAAM,OAAO,KAAK,UAAU;KAC5B,IACE,MAAM,SAAS,eACf,KAAK,UAAU,IAAI,KAAK,UAAU,CAAC,EAAE,oBACnC,iBACF;MACA,YAAY;MACZ;KACF;KACA;IACF;IACA,KAAK,UAAU,OAAO,UAAU,GAAG,KAAK;IACxC;GACF;EACF;EACA,KAAK,UAAU,KAAK,KAAK;CAC3B;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,WACA;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;IACT,uBAAuB;GACzB;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,IAAI,cAAc,KAAA,KAAa,CAAC,MAAM,uBAAuB;GAC3D,MAAM,SAAS;GACf,MAAM,eAAe,CAAC;IAAE,MAAM;IAAQ,MAAM;GAAQ,CAAC;GACrD,MAAM,wBAAwB;GAC9B,MAAM,UAAU;EAClB,OAAO,IAAI,MAAM,uBAAuB;GACtC,IAAI,MAAM,iBAAiB,KAAA,GACzB,MAAM,eAAe,CAAC;IAAE,MAAM;IAAQ,MAAM;GAAQ,CAAC;GAEvD,IAAI,MAAM,YAAY,KAAA,GACpB,MAAM,UAAU;EAEpB,OAAO;GACL,MAAM,SAAS,aAAa,OAAO;GACnC,MAAM,UAAU;EAClB;EACA,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,iBAAiB,KAAA,IACvB,EAAE,cAAc,MAAM,aAAa,IACnC,CAAC;IACL,GAAI,MAAM,YAAY,KAAA,IAAY,EAAE,SAAS,MAAM,QAAQ,IAAI,CAAC;IAChE,GAAI,MAAM,oBACN,EACE,KAAK,EACH,KAAK;KACH,aAAa,MAAM;KACnB,GAAI,MAAM,iBACN,EAAE,UAAU,MAAM,eAAe,IACjC,CAAC;IACP,EACF,EACF,IACA,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

@@ -1,6 +1,5 @@
 import { Logger } from "../logger.js";
 import { AgUiEvent } from "../types.js";
-
 //#region src/runtime/adapter/subscriber.d.ts
 type Dispatch = (event: AgUiEvent) => void;
 type Subscriber = {

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,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"}
+{"version":3,"file":"subscriber.d.ts","names":[],"sources":["../../../src/runtime/adapter/subscriber.ts"],"mappings":";;;KAMK,YAAY,OAAO;KAEnB;EACH,WAAW;IAAW;;EACtB,2BAA2B;IAAW;;EACtC,6BAA6B;IAAW;;EACxC,yBAAyB;IAAW;;EACpC,2BAA2B;IAAW;;EACtC,wBAAwB;IAAW;;EACnC,sBAAsB;IAAW;;EACjC,mCAAmC;IAAW;;EAC9C,qCAAqC;IAAW;;EAChD,iCAAiC;IAAW;;EAC5C,yBAAyB;IAAW;;EACpC,uBAAuB;IAAW;;EAClC,gCAAgC;IAAW;;EAC3C,kCAAkC;IAAW;;EAC7C,8BAA8B;IAAW;;EACzC,wBAAwB;IAAW;;EACnC,uBAAuB;IAAW;;EAClC,sBAAsB;IAAW;;EACjC,wBAAwB;IAAW;;EACnC,yBAAyB;IAAW;;EACpC,2BAA2B;IAAW;;EACtC,wBAAwB;IAAW;;EACnC,qBAAqB;IAAW;;EAChC,2BAA2B;IAAW;;EACtC,iBAAiB;IAAW;;EAC5B,cAAc;IAAW;;EACzB,sBAAsB;IAAW;;EACjC;EACA,eAAe;IAAW,OAAO;;;KAiC9B;EACH,UAAU;EACV;EACA,SAAS;EACT,eAAe,OAAO;;cAGX,uBACX,SAAS,sBACR;KAiGS,iBAAiB,kBAAkB"}

package/dist/runtime/event-parser.d.ts

@@ -1,6 +1,5 @@
 import { Logger } from "./logger.js";
 import { AgUiEvent } from "./types.js";
-
 //#region src/runtime/event-parser.d.ts
 type ParseAgUiEventOptions = {
   logger?: Logger;

package/dist/runtime/event-parser.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"event-parser.d.ts","names":[],"sources":["../../src/runtime/event-parser.ts"],"mappings":";;;;KAGY,qBAAA;EACV,MAAA,GAAS,MAAM;AAAA;AAAA,cAkEJ,cAAA,GACX,KAAA,WACA,OAAA,GAAU,qBAAA,KACT,SAAS"}
+{"version":3,"file":"event-parser.d.ts","names":[],"sources":["../../src/runtime/event-parser.ts"],"mappings":";;;KAIY;EACV,SAAS;;cAkEE,iBACX,gBACA,UAAU,0BACT"}

package/dist/runtime/event-parser.js

@@ -1,3 +1,4 @@
+import { parseMcpToolCallResult } from "./mcp-tool-result.js";
 //#region src/runtime/event-parser.ts
 const isString = (value) => typeof value === "string";
 const isNonEmptyString = (value) => isString(value) && value.length > 0;
@@ -66,7 +67,10 @@ const parseAgUiEvent = (event, options) => {
 				runId
 			}, { outcome: parseRunFinishedOutcome(payload.outcome, options?.logger) });
 		}
-		case "RUN_CANCELLED": return withOptional({ type: "RUN_CANCELLED" }, { runId: getString("runId") });
+		case "RUN_CANCELLED": {
+			const runId = getString("runId");
+			return withOptional({ type: "RUN_CANCELLED" }, { runId });
+		}
 		case "RUN_ERROR": return withOptional({ type: "RUN_ERROR" }, {
 			message: getString("message"),
 			code: getString("code")
@@ -81,10 +85,13 @@ const parseAgUiEvent = (event, options) => {
 			}, { messageId: getString("messageId") });
 		}
 		case "TEXT_MESSAGE_END": return withOptional({ type: "TEXT_MESSAGE_END" }, { messageId: getString("messageId") });
-		case "TEXT_MESSAGE_CHUNK": return withOptional({
-			type: "TEXT_MESSAGE_CHUNK",
-			delta: getString("delta") ?? ""
-		}, { messageId: getString("messageId") });
+		case "TEXT_MESSAGE_CHUNK": {
+			const delta = getString("delta") ?? "";
+			return withOptional({
+				type: "TEXT_MESSAGE_CHUNK",
+				delta
+			}, { messageId: getString("messageId") });
+		}
 		case "THINKING_START": return withOptional({ type: "THINKING_START" }, { title: getString("title") });
 		case "THINKING_TEXT_MESSAGE_START": return { type: "THINKING_TEXT_MESSAGE_START" };
 		case "THINKING_TEXT_MESSAGE_CONTENT": return {
@@ -95,10 +102,13 @@ const parseAgUiEvent = (event, options) => {
 		case "THINKING_END": return { type: "THINKING_END" };
 		case "REASONING_START": return withOptional({ type: "REASONING_START" }, { messageId: getString("messageId") });
 		case "REASONING_MESSAGE_START": return withOptional({ type: "REASONING_MESSAGE_START" }, { messageId: getString("messageId") });
-		case "REASONING_MESSAGE_CONTENT": return withOptional({
-			type: "REASONING_MESSAGE_CONTENT",
-			delta: getString("delta") ?? ""
-		}, { messageId: getString("messageId") });
+		case "REASONING_MESSAGE_CONTENT": {
+			const delta = getString("delta") ?? "";
+			return withOptional({
+				type: "REASONING_MESSAGE_CONTENT",
+				delta
+			}, { messageId: getString("messageId") });
+		}
 		case "REASONING_MESSAGE_END": return withOptional({ type: "REASONING_MESSAGE_END" }, { messageId: getString("messageId") });
 		case "REASONING_END": return withOptional({ type: "REASONING_END" }, { messageId: getString("messageId") });
 		case "TOOL_CALL_START": {
@@ -137,13 +147,15 @@ const parseAgUiEvent = (event, options) => {
 		case "TOOL_CALL_RESULT": {
 			const toolCallId = getString("toolCallId");
 			if (!toolCallId) return null;
+			const content = getString("content") ?? "";
 			return withOptional({
 				type: "TOOL_CALL_RESULT",
 				toolCallId,
-				content: getString("content") ?? ""
+				content
 			}, {
 				messageId: getString("messageId"),
-				role: payload.role === "tool" ? "tool" : void 0
+				role: payload.role === "tool" ? "tool" : void 0,
+				mcpResult: parseMcpToolCallResult(payload, content)
 			});
 		}
 		case "STATE_SNAPSHOT": return {

package/dist/runtime/event-parser.js.map

@@ -1 +1 @@
-{"version":3,"file":"event-parser.js","names":[],"sources":["../../src/runtime/event-parser.ts"],"sourcesContent":["import type { AgUiEvent, AgUiInterrupt, AgUiRunFinishedOutcome } from \"./types\";\nimport type { Logger } from \"./logger\";\n\nexport type ParseAgUiEventOptions = {\n  logger?: Logger;\n};\n\nconst isString = (value: unknown): value is string => typeof value === \"string\";\nconst isNonEmptyString = (value: unknown): value is string =>\n  isString(value) && value.length > 0;\n\nconst withOptional = <T extends object>(\n  base: T,\n  optionals: Record<string, unknown>,\n) => {\n  const definedEntries = Object.entries(optionals).filter(\n    ([, value]) => value !== undefined,\n  );\n  return definedEntries.length === 0\n    ? base\n    : ({ ...base, ...Object.fromEntries(definedEntries) } as T);\n};\n\nconst isPlainObject = (value: unknown): value is Record<string, unknown> =>\n  !!value && typeof value === \"object\" && !Array.isArray(value);\n\nconst parseInterrupt = (raw: unknown): AgUiInterrupt | null => {\n  if (!isPlainObject(raw)) return null;\n  const id = raw.id;\n  const reason = raw.reason;\n  if (typeof id !== \"string\" || typeof reason !== \"string\") return null;\n  const interrupt: AgUiInterrupt = { id, reason };\n  if (typeof raw.message === \"string\") interrupt.message = raw.message;\n  if (typeof raw.toolCallId === \"string\") interrupt.toolCallId = raw.toolCallId;\n  if (typeof raw.expiresAt === \"string\") interrupt.expiresAt = raw.expiresAt;\n  if (isPlainObject(raw.responseSchema))\n    interrupt.responseSchema = raw.responseSchema;\n  if (isPlainObject(raw.metadata)) interrupt.metadata = raw.metadata;\n  return interrupt;\n};\n\nconst parseRunFinishedOutcome = (\n  raw: unknown,\n  logger: Logger | undefined,\n): AgUiRunFinishedOutcome | undefined => {\n  if (!isPlainObject(raw)) return undefined;\n  if (raw.type === \"success\") return { type: \"success\" };\n  if (raw.type === \"interrupt\") {\n    if (!Array.isArray(raw.interrupts)) {\n      logger?.debug?.(\n        \"[agui] RUN_FINISHED interrupt outcome missing interrupts array\",\n        raw,\n      );\n      return undefined;\n    }\n    const parsed = raw.interrupts\n      .map((entry) => parseInterrupt(entry))\n      .filter((entry): entry is AgUiInterrupt => entry !== null);\n    if (parsed.length === 0) {\n      logger?.debug?.(\n        \"[agui] RUN_FINISHED interrupt outcome has no valid interrupts\",\n        raw.interrupts,\n      );\n      return undefined;\n    }\n    return { type: \"interrupt\", interrupts: parsed };\n  }\n  return undefined;\n};\n\nexport const parseAgUiEvent = (\n  event: unknown,\n  options?: ParseAgUiEventOptions,\n): AgUiEvent | null => {\n  if (!event || typeof event !== \"object\") return null;\n  const payload = event as Record<string, unknown>;\n  const typeValue = payload.type;\n  if (!isString(typeValue)) return null;\n\n  const getString = (key: string) =>\n    isString(payload[key]) ? (payload[key] as string) : undefined;\n\n  switch (typeValue) {\n    case \"RUN_STARTED\": {\n      const runId = getString(\"runId\");\n      return runId ? { type: \"RUN_STARTED\", runId } : null;\n    }\n    case \"RUN_FINISHED\": {\n      const runId = getString(\"runId\");\n      if (!runId) return null;\n      return withOptional(\n        { type: \"RUN_FINISHED\" as const, runId },\n        {\n          outcome: parseRunFinishedOutcome(payload.outcome, options?.logger),\n        },\n      );\n    }\n    case \"RUN_CANCELLED\": {\n      const runId = getString(\"runId\");\n      return withOptional({ type: \"RUN_CANCELLED\" as const }, { runId });\n    }\n    case \"RUN_ERROR\": {\n      return withOptional(\n        { type: \"RUN_ERROR\" as const },\n        {\n          message: getString(\"message\"),\n          code: getString(\"code\"),\n        },\n      );\n    }\n    case \"TEXT_MESSAGE_START\":\n      return withOptional(\n        { type: \"TEXT_MESSAGE_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TEXT_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\");\n      if (!isNonEmptyString(delta)) return null;\n      return withOptional(\n        { type: \"TEXT_MESSAGE_CONTENT\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"TEXT_MESSAGE_END\":\n      return withOptional(\n        { type: \"TEXT_MESSAGE_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TEXT_MESSAGE_CHUNK\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return withOptional(\n        { type: \"TEXT_MESSAGE_CHUNK\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"THINKING_START\":\n      return withOptional(\n        { type: \"THINKING_START\" as const },\n        { title: getString(\"title\") },\n      );\n    case \"THINKING_TEXT_MESSAGE_START\":\n      return { type: \"THINKING_TEXT_MESSAGE_START\" };\n    case \"THINKING_TEXT_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return { type: \"THINKING_TEXT_MESSAGE_CONTENT\", delta };\n    }\n    case \"THINKING_TEXT_MESSAGE_END\":\n      return { type: \"THINKING_TEXT_MESSAGE_END\" };\n    case \"THINKING_END\":\n      return { type: \"THINKING_END\" };\n    case \"REASONING_START\":\n      return withOptional(\n        { type: \"REASONING_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_MESSAGE_START\":\n      return withOptional(\n        { type: \"REASONING_MESSAGE_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return withOptional(\n        { type: \"REASONING_MESSAGE_CONTENT\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"REASONING_MESSAGE_END\":\n      return withOptional(\n        { type: \"REASONING_MESSAGE_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_END\":\n      return withOptional(\n        { type: \"REASONING_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TOOL_CALL_START\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      return withOptional(\n        { type: \"TOOL_CALL_START\" as const, toolCallId },\n        {\n          toolCallName: getString(\"toolCallName\"),\n          parentMessageId: getString(\"parentMessageId\"),\n        },\n      );\n    }\n    case \"TOOL_CALL_ARGS\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      const delta = getString(\"delta\") ?? \"\";\n      return { type: \"TOOL_CALL_ARGS\", toolCallId, delta };\n    }\n    case \"TOOL_CALL_END\": {\n      const toolCallId = getString(\"toolCallId\");\n      return toolCallId ? { type: \"TOOL_CALL_END\", toolCallId } : null;\n    }\n    case \"TOOL_CALL_CHUNK\":\n      return withOptional(\n        { type: \"TOOL_CALL_CHUNK\" as const },\n        {\n          toolCallId: getString(\"toolCallId\"),\n          toolCallName: getString(\"toolCallName\"),\n          parentMessageId: getString(\"parentMessageId\"),\n          delta: getString(\"delta\"),\n        },\n      );\n    case \"TOOL_CALL_RESULT\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      return withOptional(\n        {\n          type: \"TOOL_CALL_RESULT\" as const,\n          toolCallId,\n          content: getString(\"content\") ?? \"\",\n        },\n        {\n          messageId: getString(\"messageId\"),\n          role: payload.role === \"tool\" ? \"tool\" : undefined,\n        },\n      );\n    }\n    case \"STATE_SNAPSHOT\":\n      return { type: \"STATE_SNAPSHOT\", snapshot: payload.snapshot };\n    case \"STATE_DELTA\":\n      return {\n        type: \"STATE_DELTA\",\n        delta: Array.isArray(payload.delta) ? (payload.delta as any[]) : [],\n      };\n    case \"MESSAGES_SNAPSHOT\":\n      return {\n        type: \"MESSAGES_SNAPSHOT\",\n        messages: Array.isArray(payload.messages)\n          ? (payload.messages as any[])\n          : [],\n      };\n    case \"ACTIVITY_SNAPSHOT\": {\n      const activityType = getString(\"activityType\");\n      if (!activityType || !isPlainObject(payload.content)) return null;\n      return {\n        type: \"ACTIVITY_SNAPSHOT\" as const,\n        activityType,\n        content: payload.content,\n      };\n    }\n    case \"RAW\":\n      return withOptional(\n        { type: \"RAW\" as const, event: payload.event },\n        { source: getString(\"source\") },\n      );\n    case \"CUSTOM\": {\n      const name = getString(\"name\");\n      if (!name) return null;\n      return { type: \"CUSTOM\", name, value: payload.value };\n    }\n    default:\n      return withOptional(\n        { type: \"RAW\" as const, event: payload },\n        {\n          source: isString(payload.type) ? (payload.type as string) : undefined,\n        },\n      );\n  }\n};\n"],"mappings":";AAOA,MAAM,YAAY,UAAoC,OAAO,UAAU;AACvE,MAAM,oBAAoB,UACxB,SAAS,KAAK,KAAK,MAAM,SAAS;AAEpC,MAAM,gBACJ,MACA,cACG;CACH,MAAM,iBAAiB,OAAO,QAAQ,SAAS,CAAC,CAAC,QAC9C,GAAG,WAAW,UAAU,KAAA,CAC3B;CACA,OAAO,eAAe,WAAW,IAC7B,OACC;EAAE,GAAG;EAAM,GAAG,OAAO,YAAY,cAAc;CAAE;AACxD;AAEA,MAAM,iBAAiB,UACrB,CAAC,CAAC,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;AAE9D,MAAM,kBAAkB,QAAuC;CAC7D,IAAI,CAAC,cAAc,GAAG,GAAG,OAAO;CAChC,MAAM,KAAK,IAAI;CACf,MAAM,SAAS,IAAI;CACnB,IAAI,OAAO,OAAO,YAAY,OAAO,WAAW,UAAU,OAAO;CACjE,MAAM,YAA2B;EAAE;EAAI;CAAO;CAC9C,IAAI,OAAO,IAAI,YAAY,UAAU,UAAU,UAAU,IAAI;CAC7D,IAAI,OAAO,IAAI,eAAe,UAAU,UAAU,aAAa,IAAI;CACnE,IAAI,OAAO,IAAI,cAAc,UAAU,UAAU,YAAY,IAAI;CACjE,IAAI,cAAc,IAAI,cAAc,GAClC,UAAU,iBAAiB,IAAI;CACjC,IAAI,cAAc,IAAI,QAAQ,GAAG,UAAU,WAAW,IAAI;CAC1D,OAAO;AACT;AAEA,MAAM,2BACJ,KACA,WACuC;CACvC,IAAI,CAAC,cAAc,GAAG,GAAG,OAAO,KAAA;CAChC,IAAI,IAAI,SAAS,WAAW,OAAO,EAAE,MAAM,UAAU;CACrD,IAAI,IAAI,SAAS,aAAa;EAC5B,IAAI,CAAC,MAAM,QAAQ,IAAI,UAAU,GAAG;GAClC,QAAQ,QACN,kEACA,GACF;GACA;EACF;EACA,MAAM,SAAS,IAAI,WAChB,KAAK,UAAU,eAAe,KAAK,CAAC,CAAC,CACrC,QAAQ,UAAkC,UAAU,IAAI;EAC3D,IAAI,OAAO,WAAW,GAAG;GACvB,QAAQ,QACN,iEACA,IAAI,UACN;GACA;EACF;EACA,OAAO;GAAE,MAAM;GAAa,YAAY;EAAO;CACjD;AAEF;AAEA,MAAa,kBACX,OACA,YACqB;CACrB,IAAI,CAAC,SAAS,OAAO,UAAU,UAAU,OAAO;CAChD,MAAM,UAAU;CAChB,MAAM,YAAY,QAAQ;CAC1B,IAAI,CAAC,SAAS,SAAS,GAAG,OAAO;CAEjC,MAAM,aAAa,QACjB,SAAS,QAAQ,IAAI,IAAK,QAAQ,OAAkB,KAAA;CAEtD,QAAQ,WAAR;EACE,KAAK,eAAe;GAClB,MAAM,QAAQ,UAAU,OAAO;GAC/B,OAAO,QAAQ;IAAE,MAAM;IAAe;GAAM,IAAI;EAClD;EACA,KAAK,gBAAgB;GACnB,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,CAAC,OAAO,OAAO;GACnB,OAAO,aACL;IAAE,MAAM;IAAyB;GAAM,GACvC,EACE,SAAS,wBAAwB,QAAQ,SAAS,SAAS,MAAM,EACnE,CACF;EACF;EACA,KAAK,iBAEH,OAAO,aAAa,EAAE,MAAM,gBAAyB,GAAG,EAAE,OAD5C,UAAU,OACsC,EAAE,CAAC;EAEnE,KAAK,aACH,OAAO,aACL,EAAE,MAAM,YAAqB,GAC7B;GACE,SAAS,UAAU,SAAS;GAC5B,MAAM,UAAU,MAAM;EACxB,CACF;EAEF,KAAK,sBACH,OAAO,aACL,EAAE,MAAM,qBAA8B,GACtC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,wBAAwB;GAC3B,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,CAAC,iBAAiB,KAAK,GAAG,OAAO;GACrC,OAAO,aACL;IAAE,MAAM;IAAiC;GAAM,GAC/C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF;EACA,KAAK,oBACH,OAAO,aACL,EAAE,MAAM,mBAA4B,GACpC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,sBAEH,OAAO,aACL;GAAE,MAAM;GAA+B,OAF3B,UAAU,OAAO,KAAK;EAEW,GAC7C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EAEF,KAAK,kBACH,OAAO,aACL,EAAE,MAAM,iBAA0B,GAClC,EAAE,OAAO,UAAU,OAAO,EAAE,CAC9B;EACF,KAAK,+BACH,OAAO,EAAE,MAAM,8BAA8B;EAC/C,KAAK,iCAEH,OAAO;GAAE,MAAM;GAAiC,OADlC,UAAU,OAAO,KAAK;EACkB;EAExD,KAAK,6BACH,OAAO,EAAE,MAAM,4BAA4B;EAC7C,KAAK,gBACH,OAAO,EAAE,MAAM,eAAe;EAChC,KAAK,mBACH,OAAO,aACL,EAAE,MAAM,kBAA2B,GACnC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,2BACH,OAAO,aACL,EAAE,MAAM,0BAAmC,GAC3C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,6BAEH,OAAO,aACL;GAAE,MAAM;GAAsC,OAFlC,UAAU,OAAO,KAAK;EAEkB,GACpD,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EAEF,KAAK,yBACH,OAAO,aACL,EAAE,MAAM,wBAAiC,GACzC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,iBACH,OAAO,aACL,EAAE,MAAM,gBAAyB,GACjC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,mBAAmB;GACtB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,aACL;IAAE,MAAM;IAA4B;GAAW,GAC/C;IACE,cAAc,UAAU,cAAc;IACtC,iBAAiB,UAAU,iBAAiB;GAC9C,CACF;EACF;EACA,KAAK,kBAAkB;GACrB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GAExB,OAAO;IAAE,MAAM;IAAkB;IAAY,OAD/B,UAAU,OAAO,KAAK;GACe;EACrD;EACA,KAAK,iBAAiB;GACpB,MAAM,aAAa,UAAU,YAAY;GACzC,OAAO,aAAa;IAAE,MAAM;IAAiB;GAAW,IAAI;EAC9D;EACA,KAAK,mBACH,OAAO,aACL,EAAE,MAAM,kBAA2B,GACnC;GACE,YAAY,UAAU,YAAY;GAClC,cAAc,UAAU,cAAc;GACtC,iBAAiB,UAAU,iBAAiB;GAC5C,OAAO,UAAU,OAAO;EAC1B,CACF;EACF,KAAK,oBAAoB;GACvB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,aACL;IACE,MAAM;IACN;IACA,SAAS,UAAU,SAAS,KAAK;GACnC,GACA;IACE,WAAW,UAAU,WAAW;IAChC,MAAM,QAAQ,SAAS,SAAS,SAAS,KAAA;GAC3C,CACF;EACF;EACA,KAAK,kBACH,OAAO;GAAE,MAAM;GAAkB,UAAU,QAAQ;EAAS;EAC9D,KAAK,eACH,OAAO;GACL,MAAM;GACN,OAAO,MAAM,QAAQ,QAAQ,KAAK,IAAK,QAAQ,QAAkB,CAAC;EACpE;EACF,KAAK,qBACH,OAAO;GACL,MAAM;GACN,UAAU,MAAM,QAAQ,QAAQ,QAAQ,IACnC,QAAQ,WACT,CAAC;EACP;EACF,KAAK,qBAAqB;GACxB,MAAM,eAAe,UAAU,cAAc;GAC7C,IAAI,CAAC,gBAAgB,CAAC,cAAc,QAAQ,OAAO,GAAG,OAAO;GAC7D,OAAO;IACL,MAAM;IACN;IACA,SAAS,QAAQ;GACnB;EACF;EACA,KAAK,OACH,OAAO,aACL;GAAE,MAAM;GAAgB,OAAO,QAAQ;EAAM,GAC7C,EAAE,QAAQ,UAAU,QAAQ,EAAE,CAChC;EACF,KAAK,UAAU;GACb,MAAM,OAAO,UAAU,MAAM;GAC7B,IAAI,CAAC,MAAM,OAAO;GAClB,OAAO;IAAE,MAAM;IAAU;IAAM,OAAO,QAAQ;GAAM;EACtD;EACA,SACE,OAAO,aACL;GAAE,MAAM;GAAgB,OAAO;EAAQ,GACvC,EACE,QAAQ,SAAS,QAAQ,IAAI,IAAK,QAAQ,OAAkB,KAAA,EAC9D,CACF;CACJ;AACF"}
+{"version":3,"file":"event-parser.js","names":[],"sources":["../../src/runtime/event-parser.ts"],"sourcesContent":["import type { AgUiEvent, AgUiInterrupt, AgUiRunFinishedOutcome } from \"./types\";\nimport type { Logger } from \"./logger\";\nimport { parseMcpToolCallResult } from \"./mcp-tool-result\";\n\nexport type ParseAgUiEventOptions = {\n  logger?: Logger;\n};\n\nconst isString = (value: unknown): value is string => typeof value === \"string\";\nconst isNonEmptyString = (value: unknown): value is string =>\n  isString(value) && value.length > 0;\n\nconst withOptional = <T extends object>(\n  base: T,\n  optionals: Record<string, unknown>,\n) => {\n  const definedEntries = Object.entries(optionals).filter(\n    ([, value]) => value !== undefined,\n  );\n  return definedEntries.length === 0\n    ? base\n    : ({ ...base, ...Object.fromEntries(definedEntries) } as T);\n};\n\nconst isPlainObject = (value: unknown): value is Record<string, unknown> =>\n  !!value && typeof value === \"object\" && !Array.isArray(value);\n\nconst parseInterrupt = (raw: unknown): AgUiInterrupt | null => {\n  if (!isPlainObject(raw)) return null;\n  const id = raw.id;\n  const reason = raw.reason;\n  if (typeof id !== \"string\" || typeof reason !== \"string\") return null;\n  const interrupt: AgUiInterrupt = { id, reason };\n  if (typeof raw.message === \"string\") interrupt.message = raw.message;\n  if (typeof raw.toolCallId === \"string\") interrupt.toolCallId = raw.toolCallId;\n  if (typeof raw.expiresAt === \"string\") interrupt.expiresAt = raw.expiresAt;\n  if (isPlainObject(raw.responseSchema))\n    interrupt.responseSchema = raw.responseSchema;\n  if (isPlainObject(raw.metadata)) interrupt.metadata = raw.metadata;\n  return interrupt;\n};\n\nconst parseRunFinishedOutcome = (\n  raw: unknown,\n  logger: Logger | undefined,\n): AgUiRunFinishedOutcome | undefined => {\n  if (!isPlainObject(raw)) return undefined;\n  if (raw.type === \"success\") return { type: \"success\" };\n  if (raw.type === \"interrupt\") {\n    if (!Array.isArray(raw.interrupts)) {\n      logger?.debug?.(\n        \"[agui] RUN_FINISHED interrupt outcome missing interrupts array\",\n        raw,\n      );\n      return undefined;\n    }\n    const parsed = raw.interrupts\n      .map((entry) => parseInterrupt(entry))\n      .filter((entry): entry is AgUiInterrupt => entry !== null);\n    if (parsed.length === 0) {\n      logger?.debug?.(\n        \"[agui] RUN_FINISHED interrupt outcome has no valid interrupts\",\n        raw.interrupts,\n      );\n      return undefined;\n    }\n    return { type: \"interrupt\", interrupts: parsed };\n  }\n  return undefined;\n};\n\nexport const parseAgUiEvent = (\n  event: unknown,\n  options?: ParseAgUiEventOptions,\n): AgUiEvent | null => {\n  if (!event || typeof event !== \"object\") return null;\n  const payload = event as Record<string, unknown>;\n  const typeValue = payload.type;\n  if (!isString(typeValue)) return null;\n\n  const getString = (key: string) =>\n    isString(payload[key]) ? (payload[key] as string) : undefined;\n\n  switch (typeValue) {\n    case \"RUN_STARTED\": {\n      const runId = getString(\"runId\");\n      return runId ? { type: \"RUN_STARTED\", runId } : null;\n    }\n    case \"RUN_FINISHED\": {\n      const runId = getString(\"runId\");\n      if (!runId) return null;\n      return withOptional(\n        { type: \"RUN_FINISHED\" as const, runId },\n        {\n          outcome: parseRunFinishedOutcome(payload.outcome, options?.logger),\n        },\n      );\n    }\n    case \"RUN_CANCELLED\": {\n      const runId = getString(\"runId\");\n      return withOptional({ type: \"RUN_CANCELLED\" as const }, { runId });\n    }\n    case \"RUN_ERROR\": {\n      return withOptional(\n        { type: \"RUN_ERROR\" as const },\n        {\n          message: getString(\"message\"),\n          code: getString(\"code\"),\n        },\n      );\n    }\n    case \"TEXT_MESSAGE_START\":\n      return withOptional(\n        { type: \"TEXT_MESSAGE_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TEXT_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\");\n      if (!isNonEmptyString(delta)) return null;\n      return withOptional(\n        { type: \"TEXT_MESSAGE_CONTENT\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"TEXT_MESSAGE_END\":\n      return withOptional(\n        { type: \"TEXT_MESSAGE_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TEXT_MESSAGE_CHUNK\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return withOptional(\n        { type: \"TEXT_MESSAGE_CHUNK\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"THINKING_START\":\n      return withOptional(\n        { type: \"THINKING_START\" as const },\n        { title: getString(\"title\") },\n      );\n    case \"THINKING_TEXT_MESSAGE_START\":\n      return { type: \"THINKING_TEXT_MESSAGE_START\" };\n    case \"THINKING_TEXT_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return { type: \"THINKING_TEXT_MESSAGE_CONTENT\", delta };\n    }\n    case \"THINKING_TEXT_MESSAGE_END\":\n      return { type: \"THINKING_TEXT_MESSAGE_END\" };\n    case \"THINKING_END\":\n      return { type: \"THINKING_END\" };\n    case \"REASONING_START\":\n      return withOptional(\n        { type: \"REASONING_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_MESSAGE_START\":\n      return withOptional(\n        { type: \"REASONING_MESSAGE_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return withOptional(\n        { type: \"REASONING_MESSAGE_CONTENT\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"REASONING_MESSAGE_END\":\n      return withOptional(\n        { type: \"REASONING_MESSAGE_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_END\":\n      return withOptional(\n        { type: \"REASONING_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TOOL_CALL_START\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      return withOptional(\n        { type: \"TOOL_CALL_START\" as const, toolCallId },\n        {\n          toolCallName: getString(\"toolCallName\"),\n          parentMessageId: getString(\"parentMessageId\"),\n        },\n      );\n    }\n    case \"TOOL_CALL_ARGS\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      const delta = getString(\"delta\") ?? \"\";\n      return { type: \"TOOL_CALL_ARGS\", toolCallId, delta };\n    }\n    case \"TOOL_CALL_END\": {\n      const toolCallId = getString(\"toolCallId\");\n      return toolCallId ? { type: \"TOOL_CALL_END\", toolCallId } : null;\n    }\n    case \"TOOL_CALL_CHUNK\":\n      return withOptional(\n        { type: \"TOOL_CALL_CHUNK\" as const },\n        {\n          toolCallId: getString(\"toolCallId\"),\n          toolCallName: getString(\"toolCallName\"),\n          parentMessageId: getString(\"parentMessageId\"),\n          delta: getString(\"delta\"),\n        },\n      );\n    case \"TOOL_CALL_RESULT\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      const content = getString(\"content\") ?? \"\";\n      return withOptional(\n        {\n          type: \"TOOL_CALL_RESULT\" as const,\n          toolCallId,\n          content,\n        },\n        {\n          messageId: getString(\"messageId\"),\n          role: payload.role === \"tool\" ? \"tool\" : undefined,\n          mcpResult: parseMcpToolCallResult(payload, content),\n        },\n      );\n    }\n    case \"STATE_SNAPSHOT\":\n      return { type: \"STATE_SNAPSHOT\", snapshot: payload.snapshot };\n    case \"STATE_DELTA\":\n      return {\n        type: \"STATE_DELTA\",\n        delta: Array.isArray(payload.delta) ? (payload.delta as any[]) : [],\n      };\n    case \"MESSAGES_SNAPSHOT\":\n      return {\n        type: \"MESSAGES_SNAPSHOT\",\n        messages: Array.isArray(payload.messages)\n          ? (payload.messages as any[])\n          : [],\n      };\n    case \"ACTIVITY_SNAPSHOT\": {\n      const activityType = getString(\"activityType\");\n      if (!activityType || !isPlainObject(payload.content)) return null;\n      return {\n        type: \"ACTIVITY_SNAPSHOT\" as const,\n        activityType,\n        content: payload.content,\n      };\n    }\n    case \"RAW\":\n      return withOptional(\n        { type: \"RAW\" as const, event: payload.event },\n        { source: getString(\"source\") },\n      );\n    case \"CUSTOM\": {\n      const name = getString(\"name\");\n      if (!name) return null;\n      return { type: \"CUSTOM\", name, value: payload.value };\n    }\n    default:\n      return withOptional(\n        { type: \"RAW\" as const, event: payload },\n        {\n          source: isString(payload.type) ? (payload.type as string) : undefined,\n        },\n      );\n  }\n};\n"],"mappings":";;AAQA,MAAM,YAAY,UAAoC,OAAO,UAAU;AACvE,MAAM,oBAAoB,UACxB,SAAS,KAAK,KAAK,MAAM,SAAS;AAEpC,MAAM,gBACJ,MACA,cACG;CACH,MAAM,iBAAiB,OAAO,QAAQ,SAAS,CAAC,CAAC,QAC9C,GAAG,WAAW,UAAU,KAAA,CAC3B;CACA,OAAO,eAAe,WAAW,IAC7B,OACC;EAAE,GAAG;EAAM,GAAG,OAAO,YAAY,cAAc;CAAE;AACxD;AAEA,MAAM,iBAAiB,UACrB,CAAC,CAAC,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;AAE9D,MAAM,kBAAkB,QAAuC;CAC7D,IAAI,CAAC,cAAc,GAAG,GAAG,OAAO;CAChC,MAAM,KAAK,IAAI;CACf,MAAM,SAAS,IAAI;CACnB,IAAI,OAAO,OAAO,YAAY,OAAO,WAAW,UAAU,OAAO;CACjE,MAAM,YAA2B;EAAE;EAAI;CAAO;CAC9C,IAAI,OAAO,IAAI,YAAY,UAAU,UAAU,UAAU,IAAI;CAC7D,IAAI,OAAO,IAAI,eAAe,UAAU,UAAU,aAAa,IAAI;CACnE,IAAI,OAAO,IAAI,cAAc,UAAU,UAAU,YAAY,IAAI;CACjE,IAAI,cAAc,IAAI,cAAc,GAClC,UAAU,iBAAiB,IAAI;CACjC,IAAI,cAAc,IAAI,QAAQ,GAAG,UAAU,WAAW,IAAI;CAC1D,OAAO;AACT;AAEA,MAAM,2BACJ,KACA,WACuC;CACvC,IAAI,CAAC,cAAc,GAAG,GAAG,OAAO,KAAA;CAChC,IAAI,IAAI,SAAS,WAAW,OAAO,EAAE,MAAM,UAAU;CACrD,IAAI,IAAI,SAAS,aAAa;EAC5B,IAAI,CAAC,MAAM,QAAQ,IAAI,UAAU,GAAG;GAClC,QAAQ,QACN,kEACA,GACF;GACA;EACF;EACA,MAAM,SAAS,IAAI,WAChB,KAAK,UAAU,eAAe,KAAK,CAAC,CAAC,CACrC,QAAQ,UAAkC,UAAU,IAAI;EAC3D,IAAI,OAAO,WAAW,GAAG;GACvB,QAAQ,QACN,iEACA,IAAI,UACN;GACA;EACF;EACA,OAAO;GAAE,MAAM;GAAa,YAAY;EAAO;CACjD;AAEF;AAEA,MAAa,kBACX,OACA,YACqB;CACrB,IAAI,CAAC,SAAS,OAAO,UAAU,UAAU,OAAO;CAChD,MAAM,UAAU;CAChB,MAAM,YAAY,QAAQ;CAC1B,IAAI,CAAC,SAAS,SAAS,GAAG,OAAO;CAEjC,MAAM,aAAa,QACjB,SAAS,QAAQ,IAAI,IAAK,QAAQ,OAAkB,KAAA;CAEtD,QAAQ,WAAR;EACE,KAAK,eAAe;GAClB,MAAM,QAAQ,UAAU,OAAO;GAC/B,OAAO,QAAQ;IAAE,MAAM;IAAe;GAAM,IAAI;EAClD;EACA,KAAK,gBAAgB;GACnB,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,CAAC,OAAO,OAAO;GACnB,OAAO,aACL;IAAE,MAAM;IAAyB;GAAM,GACvC,EACE,SAAS,wBAAwB,QAAQ,SAAS,SAAS,MAAM,EACnE,CACF;EACF;EACA,KAAK,iBAAiB;GACpB,MAAM,QAAQ,UAAU,OAAO;GAC/B,OAAO,aAAa,EAAE,MAAM,gBAAyB,GAAG,EAAE,MAAM,CAAC;EACnE;EACA,KAAK,aACH,OAAO,aACL,EAAE,MAAM,YAAqB,GAC7B;GACE,SAAS,UAAU,SAAS;GAC5B,MAAM,UAAU,MAAM;EACxB,CACF;EAEF,KAAK,sBACH,OAAO,aACL,EAAE,MAAM,qBAA8B,GACtC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,wBAAwB;GAC3B,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,CAAC,iBAAiB,KAAK,GAAG,OAAO;GACrC,OAAO,aACL;IAAE,MAAM;IAAiC;GAAM,GAC/C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF;EACA,KAAK,oBACH,OAAO,aACL,EAAE,MAAM,mBAA4B,GACpC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,sBAAsB;GACzB,MAAM,QAAQ,UAAU,OAAO,KAAK;GACpC,OAAO,aACL;IAAE,MAAM;IAA+B;GAAM,GAC7C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF;EACA,KAAK,kBACH,OAAO,aACL,EAAE,MAAM,iBAA0B,GAClC,EAAE,OAAO,UAAU,OAAO,EAAE,CAC9B;EACF,KAAK,+BACH,OAAO,EAAE,MAAM,8BAA8B;EAC/C,KAAK,iCAEH,OAAO;GAAE,MAAM;GAAiC,OADlC,UAAU,OAAO,KAAK;EACkB;EAExD,KAAK,6BACH,OAAO,EAAE,MAAM,4BAA4B;EAC7C,KAAK,gBACH,OAAO,EAAE,MAAM,eAAe;EAChC,KAAK,mBACH,OAAO,aACL,EAAE,MAAM,kBAA2B,GACnC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,2BACH,OAAO,aACL,EAAE,MAAM,0BAAmC,GAC3C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,6BAA6B;GAChC,MAAM,QAAQ,UAAU,OAAO,KAAK;GACpC,OAAO,aACL;IAAE,MAAM;IAAsC;GAAM,GACpD,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF;EACA,KAAK,yBACH,OAAO,aACL,EAAE,MAAM,wBAAiC,GACzC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,iBACH,OAAO,aACL,EAAE,MAAM,gBAAyB,GACjC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,mBAAmB;GACtB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,aACL;IAAE,MAAM;IAA4B;GAAW,GAC/C;IACE,cAAc,UAAU,cAAc;IACtC,iBAAiB,UAAU,iBAAiB;GAC9C,CACF;EACF;EACA,KAAK,kBAAkB;GACrB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GAExB,OAAO;IAAE,MAAM;IAAkB;IAAY,OAD/B,UAAU,OAAO,KAAK;GACe;EACrD;EACA,KAAK,iBAAiB;GACpB,MAAM,aAAa,UAAU,YAAY;GACzC,OAAO,aAAa;IAAE,MAAM;IAAiB;GAAW,IAAI;EAC9D;EACA,KAAK,mBACH,OAAO,aACL,EAAE,MAAM,kBAA2B,GACnC;GACE,YAAY,UAAU,YAAY;GAClC,cAAc,UAAU,cAAc;GACtC,iBAAiB,UAAU,iBAAiB;GAC5C,OAAO,UAAU,OAAO;EAC1B,CACF;EACF,KAAK,oBAAoB;GACvB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GACxB,MAAM,UAAU,UAAU,SAAS,KAAK;GACxC,OAAO,aACL;IACE,MAAM;IACN;IACA;GACF,GACA;IACE,WAAW,UAAU,WAAW;IAChC,MAAM,QAAQ,SAAS,SAAS,SAAS,KAAA;IACzC,WAAW,uBAAuB,SAAS,OAAO;GACpD,CACF;EACF;EACA,KAAK,kBACH,OAAO;GAAE,MAAM;GAAkB,UAAU,QAAQ;EAAS;EAC9D,KAAK,eACH,OAAO;GACL,MAAM;GACN,OAAO,MAAM,QAAQ,QAAQ,KAAK,IAAK,QAAQ,QAAkB,CAAC;EACpE;EACF,KAAK,qBACH,OAAO;GACL,MAAM;GACN,UAAU,MAAM,QAAQ,QAAQ,QAAQ,IACnC,QAAQ,WACT,CAAC;EACP;EACF,KAAK,qBAAqB;GACxB,MAAM,eAAe,UAAU,cAAc;GAC7C,IAAI,CAAC,gBAAgB,CAAC,cAAc,QAAQ,OAAO,GAAG,OAAO;GAC7D,OAAO;IACL,MAAM;IACN;IACA,SAAS,QAAQ;GACnB;EACF;EACA,KAAK,OACH,OAAO,aACL;GAAE,MAAM;GAAgB,OAAO,QAAQ;EAAM,GAC7C,EAAE,QAAQ,UAAU,QAAQ,EAAE,CAChC;EACF,KAAK,UAAU;GACb,MAAM,OAAO,UAAU,MAAM;GAC7B,IAAI,CAAC,MAAM,OAAO;GAClB,OAAO;IAAE,MAAM;IAAU;IAAM,OAAO,QAAQ;GAAM;EACtD;EACA,SACE,OAAO,aACL;GAAE,MAAM;GAAgB,OAAO;EAAQ,GACvC,EACE,QAAQ,SAAS,QAAQ,IAAI,IAAK,QAAQ,OAAkB,KAAA,EAC9D,CACF;CACJ;AACF"}

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

@@ -1 +1 @@
-{"version":3,"file":"logger.d.ts","names":[],"sources":["../../src/runtime/logger.ts"],"mappings":";KAAY,MAAA;EACV,KAAA,MAAW,CAAA;EACX,KAAA,MAAW,CAAA;AAAA;AAAA,cAGA,UAAA,GAAc,CAAA,GAAI,OAAA,CAAQ,MAAA,MAAU,MAAA"}
+{"version":3,"file":"logger.d.ts","names":[],"sources":["../../src/runtime/logger.ts"],"mappings":";KAAY;EACV,WAAW;EACX,WAAW;;cAGA,aAAc,IAAI,QAAQ,YAAU"}

package/dist/runtime/mcp-tool-result.d.ts

@@ -0,0 +1,11 @@
+//#region src/runtime/mcp-tool-result.d.ts
+type McpToolCallResult = Record<string, unknown> & {
+  content: unknown[];
+  structuredContent?: Record<string, unknown>;
+  _meta?: Record<string, unknown>;
+  isError?: boolean;
+};
+declare const parseMcpToolCallResult: (source: Record<string, unknown>, modelContent: string) => McpToolCallResult | undefined;
+//#endregion
+export { McpToolCallResult, parseMcpToolCallResult };
+//# sourceMappingURL=mcp-tool-result.d.ts.map

package/dist/runtime/mcp-tool-result.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-tool-result.d.ts","names":[],"sources":["../../src/runtime/mcp-tool-result.ts"],"mappings":";KAEY,oBAAoB;EAC9B;EACA,oBAAoB;EACpB,QAAQ;EACR;;cAGW,yBACX,QAAQ,yBACR,yBACC"}

package/dist/runtime/mcp-tool-result.js

@@ -0,0 +1,20 @@
+import { isRecord } from "@assistant-ui/core/internal";
+//#region src/runtime/mcp-tool-result.ts
+const parseMcpToolCallResult = (source, modelContent) => {
+	const structuredContent = isRecord(source.structuredContent) ? source.structuredContent : void 0;
+	const meta = isRecord(source._meta) ? source._meta : void 0;
+	if (!structuredContent && !meta) return void 0;
+	return {
+		content: modelContent ? [{
+			type: "text",
+			text: modelContent
+		}] : [],
+		...structuredContent ? { structuredContent } : {},
+		...meta ? { _meta: meta } : {},
+		...typeof source.isError === "boolean" ? { isError: source.isError } : {}
+	};
+};
+//#endregion
+export { parseMcpToolCallResult };
+
+//# sourceMappingURL=mcp-tool-result.js.map

package/dist/runtime/mcp-tool-result.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-tool-result.js","names":[],"sources":["../../src/runtime/mcp-tool-result.ts"],"sourcesContent":["import { isRecord } from \"@assistant-ui/core/internal\";\n\nexport type McpToolCallResult = Record<string, unknown> & {\n  content: unknown[];\n  structuredContent?: Record<string, unknown>;\n  _meta?: Record<string, unknown>;\n  isError?: boolean;\n};\n\nexport const parseMcpToolCallResult = (\n  source: Record<string, unknown>,\n  modelContent: string,\n): McpToolCallResult | undefined => {\n  const structuredContent = isRecord(source.structuredContent)\n    ? source.structuredContent\n    : undefined;\n  const meta = isRecord(source._meta) ? source._meta : undefined;\n  if (!structuredContent && !meta) return undefined;\n\n  return {\n    content: modelContent ? [{ type: \"text\", text: modelContent }] : [],\n    ...(structuredContent ? { structuredContent } : {}),\n    ...(meta ? { _meta: meta } : {}),\n    ...(typeof source.isError === \"boolean\" ? { isError: source.isError } : {}),\n  };\n};\n"],"mappings":";;AASA,MAAa,0BACX,QACA,iBACkC;CAClC,MAAM,oBAAoB,SAAS,OAAO,iBAAiB,IACvD,OAAO,oBACP,KAAA;CACJ,MAAM,OAAO,SAAS,OAAO,KAAK,IAAI,OAAO,QAAQ,KAAA;CACrD,IAAI,CAAC,qBAAqB,CAAC,MAAM,OAAO,KAAA;CAExC,OAAO;EACL,SAAS,eAAe,CAAC;GAAE,MAAM;GAAQ,MAAM;EAAa,CAAC,IAAI,CAAC;EAClE,GAAI,oBAAoB,EAAE,kBAAkB,IAAI,CAAC;EACjD,GAAI,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;EAC9B,GAAI,OAAO,OAAO,YAAY,YAAY,EAAE,SAAS,OAAO,QAAQ,IAAI,CAAC;CAC3E;AACF"}

package/dist/runtime/types.d.ts

@@ -1,8 +1,8 @@
 import { Logger } from "./logger.js";
+import { McpToolCallResult } from "./mcp-tool-result.js";
 import { AttachmentAdapter, CreateAppendMessage, 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.
@@ -39,6 +39,17 @@ type UseAgUiRuntimeOptions = ExternalStoreSharedOptions & {
   agent: AbstractAgent;
   logger?: Partial<Logger>;
   showThinking?: boolean;
+  /**
+   * When the user sends, edits, or reloads a message while client-side tool
+   * calls are still pending, automatically cancel the unresolved tool calls
+   * with an error result so the agent's tool-call accounting stays
+   * consistent. Pending AG-UI interrupts are exempt: they still reject the
+   * run and must be answered with `useAgUiSubmitInterruptResponses` or
+   * discarded with `useAgUiSteerAway`. When disabled, `useAgUiSteerAway`
+   * remains the explicit way to cancel pending tool calls.
+   * Defaults to `true`.
+   */
+  autoCancelPendingToolCalls?: boolean | undefined;
   onError?: (e: Error) => void;
   onCancel?: () => void;
   adapters?: UseAgUiRuntimeAdapters;
@@ -62,6 +73,8 @@ type AgUiRuntimeExtras = {
   interrupts: readonly AgUiInterrupt[];
   submitInterruptResponses: (responses: readonly AgUiResumeEntry[]) => Promise<void>;
   steerAway: (message: CreateAppendMessage, responses?: readonly AgUiResumeEntry[]) => Promise<void>;
+  state: ReadonlyJSONValue | undefined;
+  setState: (next: ReadonlyJSONValue | ((prev: ReadonlyJSONValue | undefined) => ReadonlyJSONValue)) => void;
 };
 type AgUiRunFinishedOutcome = {
   type: "success";
@@ -148,6 +161,7 @@ type AgUiEvent = {
   toolCallId: string;
   content: string;
   role?: "tool";
+  mcpResult?: McpToolCallResult;
 } | {
   type: "ACTIVITY_SNAPSHOT";
   activityType: string;

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

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../../src/runtime/types.ts"],"mappings":";;;;;;;;AAcgE;;;;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,iBAAA;EACV,UAAA,WAAqB,aAAA;EACrB,wBAAA,GACE,SAAA,WAAoB,eAAA,OACjB,OAAA;EACL,SAAA,GACE,OAAA,EAAS,mBAAA,EACT,SAAA,YAAqB,eAAA,OAClB,OAAA;AAAA;AAAA,KAGK,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":";;;;;;;;;;;;KAuBK;EACH,mBAAmB;EACnB,QAAQ;;;;;;EAMR;;KAGU,2BAA2B,KACrC;EAGA,qBAEM,qBACG,QAAQ,wBAAwB;;KAI/B;EACV,cAAc;EACd,SAAS;EACT,YAAY;EACZ,QAAQ;EACR,WAAW;EACX,UAAU;;;;EAIV,aAAa;;KAGH,wBAAwB;EAClC,OAAO;EACP,SAAS,QAAQ;EACjB;;;;;;;;;;;EAWA;EACA,WAAW,GAAG;EACd;EACA,WAAW;;KAGD;KAMA;EACV;EACA,QAAQ;EACR;EACA;EACA,iBAAiB;EACjB;EACA,WAAW;;KAGD;EACV;EACA;EACA;;KAGU;EACV,qBAAqB;EACrB,2BACE,oBAAoB,sBACjB;EACL,YACE,SAAS,qBACT,qBAAqB,sBAClB;EACL,OAAO;EACP,WACE,MACI,sBACE,MAAM,kCAAkC;;KAItC;EACN;;EACA;EAAmB,YAAY;;KAEzB;EACN;EAAqB;;EAErB;EACA;EACA,UAAU;;EAEV;EAAuB;;EACvB;EAAmB;EAAkB;;EACrC;EAA4B;;EAC5B;EAA8B;EAAoB;;EAClD;EAA0B;;EAC1B;EAA4B;;EAC5B;EAAwB;;EACxB;;EACA;EAAuC;;EACvC;;EACA;;EACA;EAAyB;;EACzB;EAAiC;;EACjC;EAAmC;EAAoB;;EACvD;EAA+B;;EAC/B;EAAuB;;EAEvB;EACA;EACA;EACA;;EAEA;EAAwB;EAAoB;;EAC5C;EAAuB;;EAEvB;EACA;EACA;EACA;EACA;;EAGA;EACA;EACA;EACA;EACA;EACA,YAAY;;EAGZ;EACA;EACA,SAAS;;EAET;EAAa;EAAY;;EACzB;EAAgB;EAAc;;EAC9B;EAAwB;;EACxB;EAAqB;;EACrB;EAA2B"}

package/dist/useAgUiRuntime.d.ts

@@ -1,6 +1,5 @@
 import { AgUiInterrupt, AgUiResumeEntry, UseAgUiRuntimeOptions } from "./runtime/types.js";
 import { AssistantRuntime } from "@assistant-ui/core";
-
 //#region src/useAgUiRuntime.d.ts
 type AgUiAssistantRuntime = AssistantRuntime & {
   /**

package/dist/useAgUiRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAgUiRuntime.d.ts","names":[],"sources":["../src/useAgUiRuntime.ts"],"mappings":";;;;KA2BY,oBAAA,GAAuB,gBAAA;;AAAnC;;;EAKE,6BAAA,iBAA8C,aAAA;EAAA;;;;;EAM9C,iCAAA,GACE,SAAA,WAAoB,eAAA,OACjB,OAAA;AAAA;AAAA,iBAGS,cAAA,CACd,OAAA,EAAS,qBAAA,GACR,oBAAoB"}
+{"version":3,"file":"useAgUiRuntime.d.ts","names":[],"sources":["../src/useAgUiRuntime.ts"],"mappings":";;;KA2BY,uBAAuB;;;;;EAKjC,8CAA8C;;;;;;EAM9C,oCACE,oBAAoB,sBACjB;;iBAGS,eACd,SAAS,wBACR"}

package/dist/useAgUiRuntime.js

@@ -18,6 +18,7 @@ function useAgUiRuntime(options) {
 		agent: options.agent,
 		logger,
 		showThinking: options.showThinking ?? true,
+		autoCancelPendingToolCalls: options.autoCancelPendingToolCalls,
 		...options.onError && { onError: options.onError },
 		...options.onCancel && { onCancel: options.onCancel },
 		...historyAdapter && { history: historyAdapter },
@@ -28,6 +29,7 @@ function useAgUiRuntime(options) {
 		agent: options.agent,
 		logger,
 		showThinking: options.showThinking ?? true,
+		autoCancelPendingToolCalls: options.autoCancelPendingToolCalls,
 		...options.onError && { onError: options.onError },
 		...options.onCancel && { onCancel: options.onCancel },
 		...historyAdapter && { history: historyAdapter }
@@ -42,12 +44,14 @@ function useAgUiRuntime(options) {
 			onSwitchToNewThread: onSwitchToNewThread ? async () => {
 				await onSwitchToNewThread();
 				core.applyExternalMessages([]);
+				core.resetState();
 			} : void 0,
 			onSwitchToThread: onSwitchToThread ? async (threadId) => {
 				core.applyExternalMessages([]);
 				const result = await onSwitchToThread(threadId);
 				core.applyExternalMessages(result.messages);
 				if (result.state !== void 0) core.loadExternalState(result.state);
+				else core.resetState();
 				if (result.unstable_resume) core.resumeInFlightRun(result.messages);
 			} : void 0
 		};
@@ -70,13 +74,15 @@ function useAgUiRuntime(options) {
 		return {
 			...shared,
 			isLoading: core.isLoading,
-			messages: core.getMessages(),
+			messageRepository: core.getMessageRepository(),
 			state: core.getState(),
 			isRunning: core.isRunning() || hasExecutingTools,
 			extras: agUiExtras.provide({
 				interrupts: core.getPendingInterrupts()?.interrupts ?? EMPTY_INTERRUPTS,
 				submitInterruptResponses: (responses) => core.submitInterruptResponses(responses),
-				steerAway: (message, responses) => core.steerAway(message, responses)
+				steerAway: (message, responses) => core.steerAway(message, responses),
+				state: core.getState(),
+				setState: (next) => core.setState(next)
 			}),
 			unstable_enableToolInvocations: true,
 			setToolStatuses,