@assistant-ui/react-ag-ui 0.1.0 → 0.1.1

package/dist/runtime/adapter/replay-content.js

@@ -0,0 +1,185 @@
+import { preserveToolApprovalDecision } from "./tool-approval.js";
+import { iterateToolCallParts, mapToolCallPartsDeep } from "@assistant-ui/core/internal";
+// src/runtime/adapter/replay-content.ts
+function indexReplayToolOwners(owners) {
+	const ownerByToolCallId = /* @__PURE__ */ new Map();
+	for (const owner of owners) for (const part of iterateToolCallParts(owner.content)) if (!ownerByToolCallId.has(part.toolCallId)) ownerByToolCallId.set(part.toolCallId, owner.id);
+	return ownerByToolCallId;
+}
+function groupReplayToolContent(owners, content, fallbackMessageId) {
+	const ownerByToolCallId = indexReplayToolOwners(owners);
+	const grouped = /* @__PURE__ */ new Map([[fallbackMessageId, []]]);
+	for (const part of content) {
+		const ownerId = part.type === "tool-call" ? ownerByToolCallId.get(part.toolCallId) ?? fallbackMessageId : fallbackMessageId;
+		const parts = grouped.get(ownerId) ?? [];
+		parts.push(part);
+		grouped.set(ownerId, parts);
+	}
+	return grouped;
+}
+function preserveToolResult(previous, next) {
+	if (previous.result === void 0 || next.result !== void 0) return next;
+	return {
+		...next,
+		result: previous.result,
+		...previous.artifact !== void 0 ? { artifact: previous.artifact } : {},
+		...previous.isError !== void 0 ? { isError: previous.isError } : {},
+		...previous.modelContent !== void 0 ? { modelContent: previous.modelContent } : {}
+	};
+}
+function mergeOrdered(previous, next, merge) {
+	const old = new Map(previous.map(([value, key]) => [key, value]));
+	const indexes = new Map(next.map(([, key], index) => [key, index]));
+	const insertions = Array.from({ length: next.length + 1 }, () => []);
+	let predecessor = -1;
+	for (const [value, key] of previous) {
+		const index = indexes.get(key);
+		if (index === void 0) insertions[predecessor + 1].push(value);
+		else predecessor = index;
+	}
+	const result = [];
+	for (const [index, [value, key]] of next.entries()) {
+		const prior = old.get(key);
+		result.push(...insertions[index], prior === void 0 ? value : merge(prior, value));
+	}
+	result.push(...insertions[next.length]);
+	return result;
+}
+function mergeMessages(previous, next) {
+	return mergeOrdered(previous.map((message) => [message, message.id]), next.map((message) => [message, message.id]), (old, fresh) => old.role === "assistant" && fresh.role === "assistant" ? {
+		...fresh,
+		content: mergeReplayContent(old.content, fresh.content)
+	} : fresh);
+}
+function mergeTool(previous, next) {
+	const result = preserveToolResult(previous, next);
+	const approval = preserveToolApprovalDecision(previous, next);
+	return {
+		...result,
+		...approval ? { approval } : {},
+		...result.unstable_interactions === void 0 && previous.unstable_interactions !== void 0 ? { unstable_interactions: previous.unstable_interactions } : {},
+		...previous.messages !== void 0 || next.messages !== void 0 ? { messages: mergeMessages(previous.messages ?? [], next.messages ?? []) } : {}
+	};
+}
+function partKey(part) {
+	if (part.type === "tool-call") return `tool:${part.toolCallId}`;
+	if (part.type === "text" || part.type === "reasoning") return part.type;
+	if ("id" in part && typeof part.id === "string" && part.id) return `${part.type}:${part.id}`;
+	if (part.type === "data") return `data:${part.name}`;
+	return part.type;
+}
+function keyed(content) {
+	const ordinals = /* @__PURE__ */ new Map();
+	return content.map((part) => {
+		const base = partKey(part);
+		const ordinal = ordinals.get(base) ?? 0;
+		ordinals.set(base, ordinal + 1);
+		return [part, `${base}:${ordinal}`];
+	});
+}
+function identity(part, key, scope) {
+	return JSON.stringify(part.type === "tool-call" ? ["tool", part.toolCallId] : [...scope, key]);
+}
+function visitMessages(messages, scope, visit) {
+	for (const message of messages) {
+		const path = [
+			...scope,
+			"message",
+			message.id
+		];
+		visit(JSON.stringify(path));
+		if (message.role === "assistant") visitContent(message.content, path, visit);
+	}
+}
+function visitContent(content, scope, visit) {
+	for (const [part, key] of keyed(content)) {
+		visit(identity(part, key, scope));
+		if (part.type === "tool-call" && part.messages) visitMessages(part.messages, ["tool", part.toolCallId], visit);
+	}
+}
+function pruneMessages(messages, scope, present, owned) {
+	let hasCurrent = false;
+	return {
+		messages: messages.flatMap((message) => {
+			const path = [
+				...scope,
+				"message",
+				message.id
+			];
+			const key = JSON.stringify(path);
+			const nested = message.role === "assistant" ? pruneContent(message.content, path, present, owned) : void 0;
+			const current = present.has(key) || (nested?.hasCurrent ?? false);
+			if (owned.has(key) && !current) return [];
+			hasCurrent ||= current;
+			return [nested && message.role === "assistant" ? {
+				...message,
+				content: nested.content
+			} : message];
+		}),
+		hasCurrent
+	};
+}
+function pruneContent(content, scope, present, owned) {
+	let hasCurrent = false;
+	return {
+		content: keyed(content).flatMap(([part, localKey]) => {
+			const key = identity(part, localKey, scope);
+			const nested = part.type === "tool-call" && part.messages ? pruneMessages(part.messages, ["tool", part.toolCallId], present, owned) : void 0;
+			const current = present.has(key) || (nested?.hasCurrent ?? false);
+			if (owned.has(key) && !current) return [];
+			hasCurrent ||= current;
+			return [nested ? {
+				...part,
+				messages: nested.messages
+			} : part];
+		}),
+		hasCurrent
+	};
+}
+function createReplayContentMerger() {
+	const owned = /* @__PURE__ */ new Set();
+	return (previous, next) => {
+		const present = /* @__PURE__ */ new Set();
+		visitContent(next, [], (key) => present.add(key));
+		for (const key of present) owned.add(key);
+		return pruneContent(mergeReplayContent(previous, next), [], present, owned).content;
+	};
+}
+function nestedToolIds(content) {
+	const nestedIds = /* @__PURE__ */ new Set();
+	for (const part of content) {
+		if (part.type !== "tool-call") continue;
+		for (const message of part.messages ?? []) {
+			if (message.role !== "assistant") continue;
+			for (const child of iterateToolCallParts(message.content)) nestedIds.add(child.toolCallId);
+		}
+	}
+	return nestedIds;
+}
+function mergeReplayContent(previous, next) {
+	const nestedIds = nestedToolIds(previous);
+	const incomingNestedIds = nestedToolIds(next);
+	const movingRoots = new Map(previous.flatMap((part) => part.type === "tool-call" && incomingNestedIds.has(part.toolCallId) ? [[part.toolCallId, part]] : []));
+	const flattened = /* @__PURE__ */ new Map();
+	for (const part of next) if (part.type === "tool-call" && nestedIds.has(part.toolCallId)) flattened.set(part.toolCallId, part);
+	const prior = (flattened.size ? mapToolCallPartsDeep(previous, (part) => {
+		const fresh = flattened.get(part.toolCallId);
+		return fresh ? mergeTool(part, fresh) : part;
+	}).content : previous).filter((part) => part.type !== "tool-call" || !movingRoots.has(part.toolCallId));
+	let incoming = next.filter((part) => part.type !== "tool-call" || !flattened.has(part.toolCallId));
+	if (movingRoots.size) incoming = [...mapToolCallPartsDeep(incoming, (part) => {
+		const prior = movingRoots.get(part.toolCallId);
+		return prior ? mergeTool(prior, part) : part;
+	}).content];
+	return mergeOrdered(keyed(prior), keyed(incoming), (old, fresh) => {
+		if (old.type === "tool-call" && fresh.type === "tool-call") return mergeTool(old, fresh);
+		if ((old.type === "text" || old.type === "reasoning") && (fresh.type === "text" || fresh.type === "reasoning") && old.text.startsWith(fresh.text) && old.text.length > fresh.text.length) return {
+			...fresh,
+			text: old.text
+		};
+		return fresh;
+	});
+}
+export { createReplayContentMerger, groupReplayToolContent, indexReplayToolOwners, preserveToolResult };
+
+//# sourceMappingURL=replay-content.js.map

package/dist/runtime/adapter/replay-content.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"replay-content.js","names":[],"sources":["../../../src/runtime/adapter/replay-content.ts"],"sourcesContent":["import type {\n  ThreadAssistantMessage,\n  ThreadMessage,\n  ToolCallMessagePart,\n} from \"@assistant-ui/core\";\nimport {\n  iterateToolCallParts,\n  mapToolCallPartsDeep,\n} from \"@assistant-ui/core/internal\";\n\nimport { preserveToolApprovalDecision } from \"./tool-approval\";\n\ntype Content = ThreadAssistantMessage[\"content\"];\ntype Part = Content[number];\n\nexport function indexReplayToolOwners(\n  owners: readonly ThreadAssistantMessage[],\n): ReadonlyMap<string, string> {\n  const ownerByToolCallId = new Map<string, string>();\n  for (const owner of owners) {\n    for (const part of iterateToolCallParts(owner.content)) {\n      if (!ownerByToolCallId.has(part.toolCallId))\n        ownerByToolCallId.set(part.toolCallId, owner.id);\n    }\n  }\n  return ownerByToolCallId;\n}\n\nexport function groupReplayToolContent(\n  owners: readonly ThreadAssistantMessage[],\n  content: Content,\n  fallbackMessageId: string,\n): Map<string, Part[]> {\n  const ownerByToolCallId = indexReplayToolOwners(owners);\n  const grouped = new Map<string, Part[]>([[fallbackMessageId, []]]);\n  for (const part of content) {\n    const ownerId =\n      part.type === \"tool-call\"\n        ? (ownerByToolCallId.get(part.toolCallId) ?? fallbackMessageId)\n        : fallbackMessageId;\n    const parts = grouped.get(ownerId) ?? [];\n    parts.push(part);\n    grouped.set(ownerId, parts);\n  }\n  return grouped;\n}\n\nexport function preserveToolResult(\n  previous: ToolCallMessagePart,\n  next: ToolCallMessagePart,\n): ToolCallMessagePart {\n  if (previous.result === undefined || next.result !== undefined) return next;\n  return {\n    ...next,\n    result: previous.result,\n    ...(previous.artifact !== undefined ? { artifact: previous.artifact } : {}),\n    ...(previous.isError !== undefined ? { isError: previous.isError } : {}),\n    ...(previous.modelContent !== undefined\n      ? { modelContent: previous.modelContent }\n      : {}),\n  };\n}\n\nfunction mergeOrdered<T>(\n  previous: readonly (readonly [T, string])[],\n  next: readonly (readonly [T, string])[],\n  merge: (old: T, fresh: T) => T,\n): T[] {\n  const old = new Map(previous.map(([value, key]) => [key, value] as const));\n  const indexes = new Map(next.map(([, key], index) => [key, index] as const));\n  const insertions: T[][] = Array.from({ length: next.length + 1 }, () => []);\n  let predecessor = -1;\n  for (const [value, key] of previous) {\n    const index = indexes.get(key);\n    if (index === undefined) insertions[predecessor + 1]!.push(value);\n    else predecessor = index;\n  }\n  const result: T[] = [];\n  for (const [index, [value, key]] of next.entries()) {\n    const prior = old.get(key);\n    result.push(\n      ...insertions[index]!,\n      prior === undefined ? value : merge(prior, value),\n    );\n  }\n  result.push(...insertions[next.length]!);\n  return result;\n}\n\nfunction mergeMessages(\n  previous: readonly ThreadMessage[],\n  next: readonly ThreadMessage[],\n): readonly ThreadMessage[] {\n  return mergeOrdered(\n    previous.map((message) => [message, message.id] as const),\n    next.map((message) => [message, message.id] as const),\n    (old, fresh) =>\n      old.role === \"assistant\" && fresh.role === \"assistant\"\n        ? { ...fresh, content: mergeReplayContent(old.content, fresh.content) }\n        : fresh,\n  );\n}\n\nfunction mergeTool(\n  previous: ToolCallMessagePart,\n  next: ToolCallMessagePart,\n): ToolCallMessagePart {\n  const result = preserveToolResult(previous, next);\n  const approval = preserveToolApprovalDecision(previous, next);\n  return {\n    ...result,\n    ...(approval ? { approval } : {}),\n    ...(result.unstable_interactions === undefined &&\n    previous.unstable_interactions !== undefined\n      ? { unstable_interactions: previous.unstable_interactions }\n      : {}),\n    ...(previous.messages !== undefined || next.messages !== undefined\n      ? {\n          messages: mergeMessages(previous.messages ?? [], next.messages ?? []),\n        }\n      : {}),\n  };\n}\n\nfunction partKey(part: Part): string {\n  if (part.type === \"tool-call\") return `tool:${part.toolCallId}`;\n  if (part.type === \"text\" || part.type === \"reasoning\") return part.type;\n  if (\"id\" in part && typeof part.id === \"string\" && part.id)\n    return `${part.type}:${part.id}`;\n  if (part.type === \"data\") return `data:${part.name}`;\n  return part.type;\n}\n\nfunction keyed(content: Content): [Part, string][] {\n  const ordinals = new Map<string, number>();\n  return content.map((part) => {\n    const base = partKey(part);\n    const ordinal = ordinals.get(base) ?? 0;\n    ordinals.set(base, ordinal + 1);\n    return [part, `${base}:${ordinal}`] as [Part, string];\n  });\n}\n\nfunction identity(part: Part, key: string, scope: readonly string[]): string {\n  return JSON.stringify(\n    part.type === \"tool-call\" ? [\"tool\", part.toolCallId] : [...scope, key],\n  );\n}\n\nfunction visitMessages(\n  messages: readonly ThreadMessage[],\n  scope: readonly string[],\n  visit: (key: string) => void,\n): void {\n  for (const message of messages) {\n    const path = [...scope, \"message\", message.id];\n    visit(JSON.stringify(path));\n    if (message.role === \"assistant\")\n      visitContent(message.content, path, visit);\n  }\n}\n\nfunction visitContent(\n  content: Content,\n  scope: readonly string[],\n  visit: (key: string) => void,\n): void {\n  for (const [part, key] of keyed(content)) {\n    visit(identity(part, key, scope));\n    if (part.type === \"tool-call\" && part.messages)\n      visitMessages(part.messages, [\"tool\", part.toolCallId], visit);\n  }\n}\n\nfunction pruneMessages(\n  messages: readonly ThreadMessage[],\n  scope: readonly string[],\n  present: ReadonlySet<string>,\n  owned: ReadonlySet<string>,\n): { messages: readonly ThreadMessage[]; hasCurrent: boolean } {\n  let hasCurrent = false;\n  const kept = messages.flatMap((message) => {\n    const path = [...scope, \"message\", message.id];\n    const key = JSON.stringify(path);\n    const nested =\n      message.role === \"assistant\"\n        ? pruneContent(message.content, path, present, owned)\n        : undefined;\n    const current = present.has(key) || (nested?.hasCurrent ?? false);\n    if (owned.has(key) && !current) return [];\n    hasCurrent ||= current;\n    return [\n      nested && message.role === \"assistant\"\n        ? { ...message, content: nested.content }\n        : message,\n    ];\n  });\n  return { messages: kept, hasCurrent };\n}\n\nfunction pruneContent(\n  content: Content,\n  scope: readonly string[],\n  present: ReadonlySet<string>,\n  owned: ReadonlySet<string>,\n): { content: Content; hasCurrent: boolean } {\n  let hasCurrent = false;\n  const kept = keyed(content).flatMap(([part, localKey]) => {\n    const key = identity(part, localKey, scope);\n    const nested =\n      part.type === \"tool-call\" && part.messages\n        ? pruneMessages(\n            part.messages,\n            [\"tool\", part.toolCallId],\n            present,\n            owned,\n          )\n        : undefined;\n    const current = present.has(key) || (nested?.hasCurrent ?? false);\n    if (owned.has(key) && !current) return [];\n    hasCurrent ||= current;\n    return [nested ? { ...part, messages: nested.messages } : part];\n  });\n  return { content: kept, hasCurrent };\n}\n\nexport function createReplayContentMerger(): (\n  previous: Content,\n  next: Content,\n) => Content {\n  // Remember stream-owned identities, never a second copy of tool payloads.\n  const owned = new Set<string>();\n  return (previous, next) => {\n    const present = new Set<string>();\n    visitContent(next, [], (key) => present.add(key));\n    for (const key of present) owned.add(key);\n    return pruneContent(mergeReplayContent(previous, next), [], present, owned)\n      .content;\n  };\n}\n\nfunction nestedToolIds(content: Content): Set<string> {\n  const nestedIds = new Set<string>();\n  for (const part of content) {\n    if (part.type !== \"tool-call\") continue;\n    for (const message of part.messages ?? []) {\n      if (message.role !== \"assistant\") continue;\n      for (const child of iterateToolCallParts(message.content)) {\n        nestedIds.add(child.toolCallId);\n      }\n    }\n  }\n  return nestedIds;\n}\n\nfunction mergeReplayContent(previous: Content, next: Content): Content {\n  const nestedIds = nestedToolIds(previous);\n  const incomingNestedIds = nestedToolIds(next);\n  const movingRoots = new Map(\n    previous.flatMap((part) =>\n      part.type === \"tool-call\" && incomingNestedIds.has(part.toolCallId)\n        ? [[part.toolCallId, part] as const]\n        : [],\n    ),\n  );\n  const flattened = new Map<string, ToolCallMessagePart>();\n  for (const part of next) {\n    if (part.type === \"tool-call\" && nestedIds.has(part.toolCallId)) {\n      flattened.set(part.toolCallId, part);\n    }\n  }\n  const prior = (\n    flattened.size\n      ? mapToolCallPartsDeep(previous, (part) => {\n          const fresh = flattened.get(part.toolCallId);\n          return fresh ? mergeTool(part, fresh) : part;\n        }).content\n      : previous\n  ).filter(\n    (part) => part.type !== \"tool-call\" || !movingRoots.has(part.toolCallId),\n  );\n  let incoming = next.filter(\n    (part) => part.type !== \"tool-call\" || !flattened.has(part.toolCallId),\n  );\n  if (movingRoots.size) {\n    incoming = [\n      ...mapToolCallPartsDeep(incoming, (part) => {\n        const prior = movingRoots.get(part.toolCallId);\n        return prior ? mergeTool(prior, part) : part;\n      }).content,\n    ];\n  }\n  return mergeOrdered(keyed(prior), keyed(incoming), (old, fresh) => {\n    if (old.type === \"tool-call\" && fresh.type === \"tool-call\")\n      return mergeTool(old, fresh);\n    if (\n      (old.type === \"text\" || old.type === \"reasoning\") &&\n      (fresh.type === \"text\" || fresh.type === \"reasoning\") &&\n      old.text.startsWith(fresh.text) &&\n      old.text.length > fresh.text.length\n    ) {\n      return { ...fresh, text: old.text };\n    }\n    return fresh;\n  });\n}\n"],"mappings":";;;AAeA,SAAgB,sBACd,QAC6B;CAC7B,MAAM,oCAAoB,IAAI,IAAoB;CAClD,KAAK,MAAM,SAAS,QAClB,KAAK,MAAM,QAAQ,qBAAqB,MAAM,OAAO,GACnD,IAAI,CAAC,kBAAkB,IAAI,KAAK,UAAU,GACxC,kBAAkB,IAAI,KAAK,YAAY,MAAM,EAAE;CAGrD,OAAO;AACT;AAEA,SAAgB,uBACd,QACA,SACA,mBACqB;CACrB,MAAM,oBAAoB,sBAAsB,MAAM;CACtD,MAAM,0BAAU,IAAI,IAAoB,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;CACjE,KAAK,MAAM,QAAQ,SAAS;EAC1B,MAAM,UACJ,KAAK,SAAS,cACT,kBAAkB,IAAI,KAAK,UAAU,KAAK,oBAC3C;EACN,MAAM,QAAQ,QAAQ,IAAI,OAAO,KAAK,CAAC;EACvC,MAAM,KAAK,IAAI;EACf,QAAQ,IAAI,SAAS,KAAK;CAC5B;CACA,OAAO;AACT;AAEA,SAAgB,mBACd,UACA,MACqB;CACrB,IAAI,SAAS,WAAW,KAAA,KAAa,KAAK,WAAW,KAAA,GAAW,OAAO;CACvE,OAAO;EACL,GAAG;EACH,QAAQ,SAAS;EACjB,GAAI,SAAS,aAAa,KAAA,IAAY,EAAE,UAAU,SAAS,SAAS,IAAI,CAAC;EACzE,GAAI,SAAS,YAAY,KAAA,IAAY,EAAE,SAAS,SAAS,QAAQ,IAAI,CAAC;EACtE,GAAI,SAAS,iBAAiB,KAAA,IAC1B,EAAE,cAAc,SAAS,aAAa,IACtC,CAAC;CACP;AACF;AAEA,SAAS,aACP,UACA,MACA,OACK;CACL,MAAM,MAAM,IAAI,IAAI,SAAS,KAAK,CAAC,OAAO,SAAS,CAAC,KAAK,KAAK,CAAU,CAAC;CACzE,MAAM,UAAU,IAAI,IAAI,KAAK,KAAK,GAAG,MAAM,UAAU,CAAC,KAAK,KAAK,CAAU,CAAC;CAC3E,MAAM,aAAoB,MAAM,KAAK,EAAE,QAAQ,KAAK,SAAS,EAAE,SAAS,CAAC,CAAC;CAC1E,IAAI,cAAc;CAClB,KAAK,MAAM,CAAC,OAAO,QAAQ,UAAU;EACnC,MAAM,QAAQ,QAAQ,IAAI,GAAG;EAC7B,IAAI,UAAU,KAAA,GAAW,WAAW,cAAc,EAAE,CAAE,KAAK,KAAK;OAC3D,cAAc;CACrB;CACA,MAAM,SAAc,CAAC;CACrB,KAAK,MAAM,CAAC,OAAO,CAAC,OAAO,SAAS,KAAK,QAAQ,GAAG;EAClD,MAAM,QAAQ,IAAI,IAAI,GAAG;EACzB,OAAO,KACL,GAAG,WAAW,QACd,UAAU,KAAA,IAAY,QAAQ,MAAM,OAAO,KAAK,CAClD;CACF;CACA,OAAO,KAAK,GAAG,WAAW,KAAK,OAAQ;CACvC,OAAO;AACT;AAEA,SAAS,cACP,UACA,MAC0B;CAC1B,OAAO,aACL,SAAS,KAAK,YAAY,CAAC,SAAS,QAAQ,EAAE,CAAU,GACxD,KAAK,KAAK,YAAY,CAAC,SAAS,QAAQ,EAAE,CAAU,IACnD,KAAK,UACJ,IAAI,SAAS,eAAe,MAAM,SAAS,cACvC;EAAE,GAAG;EAAO,SAAS,mBAAmB,IAAI,SAAS,MAAM,OAAO;CAAE,IACpE,KACR;AACF;AAEA,SAAS,UACP,UACA,MACqB;CACrB,MAAM,SAAS,mBAAmB,UAAU,IAAI;CAChD,MAAM,WAAW,6BAA6B,UAAU,IAAI;CAC5D,OAAO;EACL,GAAG;EACH,GAAI,WAAW,EAAE,SAAS,IAAI,CAAC;EAC/B,GAAI,OAAO,0BAA0B,KAAA,KACrC,SAAS,0BAA0B,KAAA,IAC/B,EAAE,uBAAuB,SAAS,sBAAsB,IACxD,CAAC;EACL,GAAI,SAAS,aAAa,KAAA,KAAa,KAAK,aAAa,KAAA,IACrD,EACE,UAAU,cAAc,SAAS,YAAY,CAAC,GAAG,KAAK,YAAY,CAAC,CAAC,EACtE,IACA,CAAC;CACP;AACF;AAEA,SAAS,QAAQ,MAAoB;CACnC,IAAI,KAAK,SAAS,aAAa,OAAO,QAAQ,KAAK;CACnD,IAAI,KAAK,SAAS,UAAU,KAAK,SAAS,aAAa,OAAO,KAAK;CACnE,IAAI,QAAQ,QAAQ,OAAO,KAAK,OAAO,YAAY,KAAK,IACtD,OAAO,GAAG,KAAK,KAAK,GAAG,KAAK;CAC9B,IAAI,KAAK,SAAS,QAAQ,OAAO,QAAQ,KAAK;CAC9C,OAAO,KAAK;AACd;AAEA,SAAS,MAAM,SAAoC;CACjD,MAAM,2BAAW,IAAI,IAAoB;CACzC,OAAO,QAAQ,KAAK,SAAS;EAC3B,MAAM,OAAO,QAAQ,IAAI;EACzB,MAAM,UAAU,SAAS,IAAI,IAAI,KAAK;EACtC,SAAS,IAAI,MAAM,UAAU,CAAC;EAC9B,OAAO,CAAC,MAAM,GAAG,KAAK,GAAG,SAAS;CACpC,CAAC;AACH;AAEA,SAAS,SAAS,MAAY,KAAa,OAAkC;CAC3E,OAAO,KAAK,UACV,KAAK,SAAS,cAAc,CAAC,QAAQ,KAAK,UAAU,IAAI,CAAC,GAAG,OAAO,GAAG,CACxE;AACF;AAEA,SAAS,cACP,UACA,OACA,OACM;CACN,KAAK,MAAM,WAAW,UAAU;EAC9B,MAAM,OAAO;GAAC,GAAG;GAAO;GAAW,QAAQ;EAAE;EAC7C,MAAM,KAAK,UAAU,IAAI,CAAC;EAC1B,IAAI,QAAQ,SAAS,aACnB,aAAa,QAAQ,SAAS,MAAM,KAAK;CAC7C;AACF;AAEA,SAAS,aACP,SACA,OACA,OACM;CACN,KAAK,MAAM,CAAC,MAAM,QAAQ,MAAM,OAAO,GAAG;EACxC,MAAM,SAAS,MAAM,KAAK,KAAK,CAAC;EAChC,IAAI,KAAK,SAAS,eAAe,KAAK,UACpC,cAAc,KAAK,UAAU,CAAC,QAAQ,KAAK,UAAU,GAAG,KAAK;CACjE;AACF;AAEA,SAAS,cACP,UACA,OACA,SACA,OAC6D;CAC7D,IAAI,aAAa;CAiBjB,OAAO;EAAE,UAhBI,SAAS,SAAS,YAAY;GACzC,MAAM,OAAO;IAAC,GAAG;IAAO;IAAW,QAAQ;GAAE;GAC7C,MAAM,MAAM,KAAK,UAAU,IAAI;GAC/B,MAAM,SACJ,QAAQ,SAAS,cACb,aAAa,QAAQ,SAAS,MAAM,SAAS,KAAK,IAClD,KAAA;GACN,MAAM,UAAU,QAAQ,IAAI,GAAG,MAAM,QAAQ,cAAc;GAC3D,IAAI,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,OAAO,CAAC;GACxC,eAAe;GACf,OAAO,CACL,UAAU,QAAQ,SAAS,cACvB;IAAE,GAAG;IAAS,SAAS,OAAO;GAAQ,IACtC,OACN;EACF,CACsB;EAAG;CAAW;AACtC;AAEA,SAAS,aACP,SACA,OACA,SACA,OAC2C;CAC3C,IAAI,aAAa;CAiBjB,OAAO;EAAE,SAhBI,MAAM,OAAO,CAAC,CAAC,SAAS,CAAC,MAAM,cAAc;GACxD,MAAM,MAAM,SAAS,MAAM,UAAU,KAAK;GAC1C,MAAM,SACJ,KAAK,SAAS,eAAe,KAAK,WAC9B,cACE,KAAK,UACL,CAAC,QAAQ,KAAK,UAAU,GACxB,SACA,KACF,IACA,KAAA;GACN,MAAM,UAAU,QAAQ,IAAI,GAAG,MAAM,QAAQ,cAAc;GAC3D,IAAI,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,OAAO,CAAC;GACxC,eAAe;GACf,OAAO,CAAC,SAAS;IAAE,GAAG;IAAM,UAAU,OAAO;GAAS,IAAI,IAAI;EAChE,CACqB;EAAG;CAAW;AACrC;AAEA,SAAgB,4BAGH;CAEX,MAAM,wBAAQ,IAAI,IAAY;CAC9B,QAAQ,UAAU,SAAS;EACzB,MAAM,0BAAU,IAAI,IAAY;EAChC,aAAa,MAAM,CAAC,IAAI,QAAQ,QAAQ,IAAI,GAAG,CAAC;EAChD,KAAK,MAAM,OAAO,SAAS,MAAM,IAAI,GAAG;EACxC,OAAO,aAAa,mBAAmB,UAAU,IAAI,GAAG,CAAC,GAAG,SAAS,KAAK,CAAC,CACxE;CACL;AACF;AAEA,SAAS,cAAc,SAA+B;CACpD,MAAM,4BAAY,IAAI,IAAY;CAClC,KAAK,MAAM,QAAQ,SAAS;EAC1B,IAAI,KAAK,SAAS,aAAa;EAC/B,KAAK,MAAM,WAAW,KAAK,YAAY,CAAC,GAAG;GACzC,IAAI,QAAQ,SAAS,aAAa;GAClC,KAAK,MAAM,SAAS,qBAAqB,QAAQ,OAAO,GACtD,UAAU,IAAI,MAAM,UAAU;EAElC;CACF;CACA,OAAO;AACT;AAEA,SAAS,mBAAmB,UAAmB,MAAwB;CACrE,MAAM,YAAY,cAAc,QAAQ;CACxC,MAAM,oBAAoB,cAAc,IAAI;CAC5C,MAAM,cAAc,IAAI,IACtB,SAAS,SAAS,SAChB,KAAK,SAAS,eAAe,kBAAkB,IAAI,KAAK,UAAU,IAC9D,CAAC,CAAC,KAAK,YAAY,IAAI,CAAU,IACjC,CAAC,CACP,CACF;CACA,MAAM,4BAAY,IAAI,IAAiC;CACvD,KAAK,MAAM,QAAQ,MACjB,IAAI,KAAK,SAAS,eAAe,UAAU,IAAI,KAAK,UAAU,GAC5D,UAAU,IAAI,KAAK,YAAY,IAAI;CAGvC,MAAM,SACJ,UAAU,OACN,qBAAqB,WAAW,SAAS;EACvC,MAAM,QAAQ,UAAU,IAAI,KAAK,UAAU;EAC3C,OAAO,QAAQ,UAAU,MAAM,KAAK,IAAI;CAC1C,CAAC,CAAC,CAAC,UACH,SAAA,CACJ,QACC,SAAS,KAAK,SAAS,eAAe,CAAC,YAAY,IAAI,KAAK,UAAU,CACzE;CACA,IAAI,WAAW,KAAK,QACjB,SAAS,KAAK,SAAS,eAAe,CAAC,UAAU,IAAI,KAAK,UAAU,CACvE;CACA,IAAI,YAAY,MACd,WAAW,CACT,GAAG,qBAAqB,WAAW,SAAS;EAC1C,MAAM,QAAQ,YAAY,IAAI,KAAK,UAAU;EAC7C,OAAO,QAAQ,UAAU,OAAO,IAAI,IAAI;CAC1C,CAAC,CAAC,CAAC,OACL;CAEF,OAAO,aAAa,MAAM,KAAK,GAAG,MAAM,QAAQ,IAAI,KAAK,UAAU;EACjE,IAAI,IAAI,SAAS,eAAe,MAAM,SAAS,aAC7C,OAAO,UAAU,KAAK,KAAK;EAC7B,KACG,IAAI,SAAS,UAAU,IAAI,SAAS,iBACpC,MAAM,SAAS,UAAU,MAAM,SAAS,gBACzC,IAAI,KAAK,WAAW,MAAM,IAAI,KAC9B,IAAI,KAAK,SAAS,MAAM,KAAK,QAE7B,OAAO;GAAE,GAAG;GAAO,MAAM,IAAI;EAAK;EAEpC,OAAO;CACT,CAAC;AACH"}

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

@@ -5,7 +5,7 @@ import { projectAgUiToolApprovals } from "./tool-approval.js";
 import { isMcpAppUri } from "@assistant-ui/core";
 import jsonpatch from "fast-json-patch";
 import { applyA2uiOperations, surfaceToPresentToolCall } from "@assistant-ui/generative-ui/a2ui";
-//#region src/runtime/adapter/run-aggregator.ts
+// src/runtime/adapter/run-aggregator.ts
 const ROOT_SCOPE = "";
 const MAX_SUBAGENT_DEPTH = 16;
 const createJSONContainerScanner = () => ({
@@ -1032,7 +1032,6 @@ function tryParseJSON(value) {
 		return value;
 	}
 }
-//#endregion
 export { RunAggregator, isPlainObject, tryParseJSON };
 
 //# sourceMappingURL=run-aggregator.js.map

package/dist/runtime/adapter/subscriber.js

@@ -1,6 +1,6 @@
 "use client";
 import { parseAgUiEvent } from "../event-parser.js";
-//#region src/runtime/adapter/subscriber.ts
+// src/runtime/adapter/subscriber.ts
 const isAbortError = (error) => {
 	if (error.name === "AbortError") return true;
 	const message = typeof error.message === "string" ? error.message : "";
@@ -84,7 +84,6 @@ const createAgUiSubscriber = (options) => {
 		}
 	};
 };
-//#endregion
 export { createAgUiSubscriber };
 
 //# sourceMappingURL=subscriber.js.map

package/dist/runtime/adapter/tool-approval.d.ts

@@ -19,6 +19,8 @@ type ToolApproval = NonNullable<ToolCallMessagePart["approval"]>;
  * message renders nowhere leaves the batch bespoke.
  */
 export declare const projectAgUiToolApprovals: (interrupts: readonly AgUiInterrupt[] | undefined, boundToolCallIds?: ReadonlySet<string>) => ReadonlyMap<string, ToolApproval>;
+/** A recorded decision authorizes one exact call and gate, not a reused id. */
+export declare const preserveToolApprovalDecision: (previous: ToolCallMessagePart | undefined, next: ToolCallMessagePart) => ToolApproval | undefined;
 /**
  * Records a decision on the gated tool call. The decision is held on the part
  * until every gate in the batch is answered, because AG-UI resumes a run with

package/dist/runtime/adapter/tool-approval.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tool-approval.d.ts","sourceRoot":"","sources":["../../../src/runtime/adapter/tool-approval.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,4BAA4B,EAC5B,0BAA0B,EAC1B,mBAAmB,EACpB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,aAAa,EAAE,eAAe,EAAE,oBAAiB;AAM/D,KAAK,YAAY,GAAG,WAAW,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC;AA0KjE;;;;;;;;;;;;;;;;GAgBG;AACH,eAAO,MAAM,wBAAwB,GACnC,YAAY,SAAS,aAAa,EAAE,GAAG,SAAS,EAChD,mBAAmB,WAAW,CAAC,MAAM,CAAC,KACrC,WAAW,CAAC,MAAM,EAAE,YAAY,CAsBlC,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,wBAAwB,GACnC,SAAS,SAAS,0BAA0B,EAAE,EAC9C,kCAAkC,4BAA4B,KAC7D,SAAS,0BAA0B,EAcrC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,uBAAuB,GAClC,SAAS,SAAS,0BAA0B,EAAE,EAC9C,YAAY,SAAS,aAAa,EAAE,KACnC,eAAe,EAAE,GAAG,IA8BtB,CAAC;AA0CF;;;;;;;;;;;;;;GAcG;AACH,eAAO,MAAM,wBAAwB,GACnC,SAAS,SAAS,0BAA0B,EAAE,EAC9C,QAAQ,SAAS,eAAe,EAAE,KACjC,SAAS,0BAA0B,EAarC,CAAC"}
+{"version":3,"file":"tool-approval.d.ts","sourceRoot":"","sources":["../../../src/runtime/adapter/tool-approval.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,4BAA4B,EAC5B,0BAA0B,EAC1B,mBAAmB,EACpB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,aAAa,EAAE,eAAe,EAAE,oBAAiB;AAM/D,KAAK,YAAY,GAAG,WAAW,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC;AA0KjE;;;;;;;;;;;;;;;;GAgBG;AACH,eAAO,MAAM,wBAAwB,GACnC,YAAY,SAAS,aAAa,EAAE,GAAG,SAAS,EAChD,mBAAmB,WAAW,CAAC,MAAM,CAAC,KACrC,WAAW,CAAC,MAAM,EAAE,YAAY,CAsBlC,CAAC;AAEF,+EAA+E;AAC/E,eAAO,MAAM,4BAA4B,GACvC,UAAU,mBAAmB,GAAG,SAAS,EACzC,MAAM,mBAAmB,KACxB,YAAY,GAAG,SA4BjB,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,wBAAwB,GACnC,SAAS,SAAS,0BAA0B,EAAE,EAC9C,kCAAkC,4BAA4B,KAC7D,SAAS,0BAA0B,EAcrC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,uBAAuB,GAClC,SAAS,SAAS,0BAA0B,EAAE,EAC9C,YAAY,SAAS,aAAa,EAAE,KACnC,eAAe,EAAE,GAAG,IAkCtB,CAAC;AA0CF;;;;;;;;;;;;;;GAcG;AACH,eAAO,MAAM,wBAAwB,GACnC,SAAS,SAAS,0BAA0B,EAAE,EAC9C,QAAQ,SAAS,eAAe,EAAE,KACjC,SAAS,0BAA0B,EAarC,CAAC"}

package/dist/runtime/adapter/tool-approval.js

@@ -1,7 +1,7 @@
 import { readRawResponseSchema, withoutRawResponseSchema } from "../interrupt-internals.js";
 import { iterateToolCallParts, mapToolCallPartsDeep } from "@assistant-ui/core/internal";
 import { buildResumeArray } from "@ag-ui/client";
-//#region src/runtime/adapter/tool-approval.ts
+// src/runtime/adapter/tool-approval.ts
 const EMPTY_APPROVALS = /* @__PURE__ */ new Map();
 const isPlainObject = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
 const isString = (value) => typeof value === "string";
@@ -136,6 +136,23 @@ const projectAgUiToolApprovals = (interrupts, boundToolCallIds) => {
 		...interrupt.message !== void 0 && { prompt: interrupt.message }
 	}]));
 };
+/** A recorded decision authorizes one exact call and gate, not a reused id. */
+const preserveToolApprovalDecision = (previous, next) => {
+	const prior = previous?.approval;
+	let approval = next.approval;
+	if (!approval && prior) {
+		const { approved: _approved, reason: _reason, resolution: _resolution, ...pending } = prior;
+		approval = pending;
+	}
+	if (!approval) return void 0;
+	if (!previous || prior?.id !== approval.id || previous.toolCallId !== next.toolCallId || previous.toolName !== next.toolName || (previous.argsText || "{}") !== (next.argsText || "{}")) return approval;
+	return {
+		...approval,
+		...prior.approved !== void 0 ? { approved: prior.approved } : {},
+		...prior.reason !== void 0 ? { reason: prior.reason } : {},
+		...prior.resolution !== void 0 ? { resolution: prior.resolution } : {}
+	};
+};
 /**
 * Records a decision on the gated tool call. The decision is held on the part
 * until every gate in the batch is answered, because AG-UI resumes a run with
@@ -167,12 +184,13 @@ const withToolApprovalDecision = (content, { approvalId, approved, reason }) =>
 * drop the denial reason on the wire.
 */
 const buildToolApprovalResume = (content, interrupts) => {
-	const open = new Set(interrupts.map((interrupt) => interrupt.id));
+	const calls = Array.from(iterateToolCallParts(content));
+	const gates = projectAgUiToolApprovals(interrupts, new Set(calls.map((part) => part.toolCallId)));
 	const responses = /* @__PURE__ */ new Map();
-	for (const part of iterateToolCallParts(content)) {
+	for (const part of calls) {
 		const approval = part.approval;
 		if (!approval || approval.approved === void 0) continue;
-		if (!open.has(approval.id)) continue;
+		if (gates.get(part.toolCallId)?.id !== approval.id) continue;
 		responses.set(approval.id, {
 			status: "resolved",
 			payload: {
@@ -237,7 +255,6 @@ const withSettledToolApprovals = (content, resume) => {
 	});
 	return changed ? next : content;
 };
-//#endregion
-export { buildToolApprovalResume, projectAgUiToolApprovals, withSettledToolApprovals, withToolApprovalDecision };
+export { buildToolApprovalResume, preserveToolApprovalDecision, projectAgUiToolApprovals, withSettledToolApprovals, withToolApprovalDecision };
 
 //# sourceMappingURL=tool-approval.js.map

package/dist/runtime/adapter/tool-approval.js.map

@@ -1 +1 @@
-{"version":3,"file":"tool-approval.js","names":[],"sources":["../../../src/runtime/adapter/tool-approval.ts"],"sourcesContent":["import {\n  iterateToolCallParts,\n  mapToolCallPartsDeep,\n} from \"@assistant-ui/core/internal\";\nimport { buildResumeArray } from \"@ag-ui/client\";\nimport type {\n  RespondToToolApprovalOptions,\n  ThreadAssistantMessagePart,\n  ToolCallMessagePart,\n} from \"@assistant-ui/core\";\nimport type { AgUiInterrupt, AgUiResumeEntry } from \"../types\";\nimport {\n  readRawResponseSchema,\n  withoutRawResponseSchema,\n} from \"../interrupt-internals\";\n\ntype ToolApproval = NonNullable<ToolCallMessagePart[\"approval\"]>;\n\nconst EMPTY_APPROVALS: ReadonlyMap<string, ToolApproval> = new Map();\n\nconst isPlainObject = (value: unknown): value is Record<string, unknown> =>\n  typeof value === \"object\" && value !== null && !Array.isArray(value);\n\nconst isString = (value: unknown) => typeof value === \"string\";\n\nconst JSON_SCHEMA_TYPES = new Set([\n  \"null\",\n  \"boolean\",\n  \"object\",\n  \"array\",\n  \"number\",\n  \"integer\",\n  \"string\",\n]);\n\nconst isUniqueStrings = (value: unknown) =>\n  Array.isArray(value) &&\n  value.every(isString) &&\n  new Set(value).size === value.length;\n\nconst isSchemaType = (type: unknown) =>\n  (isString(type) && JSON_SCHEMA_TYPES.has(type as string)) ||\n  (Array.isArray(type) &&\n    type.length > 0 &&\n    isUniqueStrings(type) &&\n    type.every((entry) => JSON_SCHEMA_TYPES.has(entry as string)));\n\nconst acceptsType = (type: unknown, primitive: string) =>\n  type === primitive ||\n  (Array.isArray(type) && isSchemaType(type) && type.includes(primitive));\n\nconst acceptsObject = (type: unknown) => acceptsType(type, \"object\");\n\nconst SUB_SCHEMA_ANNOTATIONS = new Set([\n  \"$schema\",\n  \"$id\",\n  \"title\",\n  \"description\",\n]);\n\n/**\n * A field schema accepts a value of `primitive` only where nothing in it can\n * narrow past the type: an annotation carries no constraint, and an absent\n * `type` constrains nothing. Any other keyword — `const`, `enum`, `anyOf`,\n * `format` — can reject the value this seam sends, so it is not interpreted.\n */\nconst acceptsOnly = (schema: unknown, primitive: string) =>\n  isPlainObject(schema) &&\n  Object.entries(schema).every(([keyword, value]) =>\n    keyword === \"type\"\n      ? acceptsType(value, primitive)\n      : SUB_SCHEMA_ANNOTATIONS.has(keyword) && isString(value),\n  );\n\n/**\n * Keywords of a sub-schema whose value shape a validator reads, each paired\n * with that shape: the same shapes the top-level schema is held to, plus\n * `properties` and `items`, which carry schemas, so a malformed one nested\n * under them is as uncompilable as a malformed one at the top. A keyword this\n * seam cannot judge is left alone: unlike the top-level schema, a field the\n * seam never sends needs only to be compilable, not permissive.\n */\nconst SUB_SCHEMA_SHAPES = new Map<string, (value: unknown) => boolean>([\n  [\"type\", isSchemaType],\n  [\"required\", isUniqueStrings],\n  ...[...SUB_SCHEMA_ANNOTATIONS].map(\n    (annotation) =>\n      [annotation, isString] as [string, (value: unknown) => boolean],\n  ),\n  [\n    \"properties\",\n    (value) =>\n      isPlainObject(value) &&\n      Object.values(value).every((schema) => isWellFormedSubSchema(schema)),\n  ],\n  [\"items\", (value) => isWellFormedSubSchema(value)],\n]);\n\n/**\n * A schema the server cannot compile fails the run whatever payload follows, so\n * a field this seam never sends is still checked for the shape a validator\n * reads it as. A boolean is a schema in its own right.\n */\nconst isWellFormedSubSchema = (schema: unknown): boolean =>\n  typeof schema === \"boolean\" ||\n  (isPlainObject(schema) &&\n    Object.entries(schema).every(([keyword, value]) => {\n      const holds = SUB_SCHEMA_SHAPES.get(keyword);\n      return holds === undefined || holds(value);\n    }));\n\n/**\n * AG-UI's own tool-approval schema declares the fields it expects under\n * `properties`, so a declaration is read where it cannot reject what the seam\n * sends: `approved` must accept a boolean and `reason` a string. A field the\n * seam never sends is unconstrained by `required` (which admits only\n * `approved`), so only its well-formedness as a sub-schema is checked — the\n * `editedArgs` of the spec's approve-with-edits shape is such a field.\n */\nconst acceptsSeamProperties = (properties: unknown) =>\n  isPlainObject(properties) &&\n  Object.entries(properties).every(([field, schema]) => {\n    if (field === \"approved\") return acceptsOnly(schema, \"boolean\");\n    if (field === \"reason\") return acceptsOnly(schema, \"string\");\n    return isWellFormedSubSchema(schema);\n  });\n\nconst acceptsSeamPayload = (required: unknown) =>\n  Array.isArray(required) &&\n  required.every((field) => field === \"approved\") &&\n  new Set(required).size === required.length;\n\n/**\n * Keywords that cannot reject a payload this seam emits, each paired with the\n * check that its value is the well-formed shape the keyword is read as. A\n * keyword whose value a validator would reject as an invalid schema is not\n * decided here, because a server that compiles the schema fails the run. An\n * unlisted keyword — `additionalProperties`, `allOf`, `$ref` — makes the gate\n * bespoke rather than being interpreted. The map is keyed by\n * `Map` rather than by object property, because a schema key of `constructor`\n * or `toString` would otherwise read an `Object.prototype` member as its\n * validator and `__proto__` would read the prototype itself.\n */\nconst SEAM_SAFE_KEYWORDS = new Map<string, (value: unknown) => boolean>([\n  [\"$schema\", isString],\n  [\"$id\", isString],\n  [\"title\", isString],\n  [\"description\", isString],\n  [\"type\", acceptsObject],\n  [\"required\", acceptsSeamPayload],\n  [\"properties\", acceptsSeamProperties],\n]);\n\n/**\n * The seam sends `{ approved }` and may add `reason`, so it may answer a\n * `responseSchema` only where acceptance of those exact payloads is established\n * by construction. AG-UI answers a resume payload its schema rejects with\n * `RunError`, and the host can always build the exact payload through the\n * bespoke interrupt hooks, so an undecidable schema is left to them. Only an\n * absent schema and a plain object of well-formed allowed keywords establish\n * that: a boolean schema (`false` rejects every payload), a malformed one, and\n * a keyword whose value is not the shape it interprets are all left bespoke.\n */\nconst isSeamAnswerable = (schema: unknown) => {\n  if (schema === undefined) return true;\n  if (!isPlainObject(schema)) return false;\n  return Object.entries(schema).every(([keyword, value]) => {\n    const accepts = SEAM_SAFE_KEYWORDS.get(keyword);\n    return accepts !== undefined && accepts(value);\n  });\n};\n\nconst isGate = (\n  interrupt: AgUiInterrupt,\n): interrupt is AgUiInterrupt & { toolCallId: string } =>\n  interrupt.reason === \"tool_call\" &&\n  !!interrupt.id &&\n  !!interrupt.toolCallId &&\n  // A restored interrupt is untyped, so `responseSchema` may hold any persisted\n  // shape; a non-object schema received live arrives on the internal carrier.\n  readRawResponseSchema(interrupt) === undefined &&\n  isSeamAnswerable(interrupt.responseSchema);\n\nconst isPending = (approval: ToolCallMessagePart[\"approval\"]) =>\n  !!approval && approval.approved === undefined && !approval.resolution;\n\n/**\n * AG-UI binds a decision to a tool call with `reason: \"tool_call\"`;\n * `confirmation` is a free-standing yes/no and `input_required` wants structured\n * input, so both stay on the bespoke interrupt hooks, as does a `tool_call`\n * whose `responseSchema` asks for more than this seam can send. A batch mixing a\n * gate with any other interrupt projects nothing: AG-UI resumes with one\n * response per open interrupt, so a half-owned batch cannot be completed from\n * either side alone.\n *\n * The same rule covers binding. A gate naming a tool call the message does not\n * render can never be decided through this seam, so the batch it belongs to\n * could never be completed either: the decisions taken on its visible siblings\n * would sit on the message while the run waited for a response that no click\n * can produce. `boundToolCallIds` spans every call the run produced — root or\n * nested on ToolCallMessagePart.messages — so only a gate naming a call the\n * message renders nowhere leaves the batch bespoke.\n */\nexport const projectAgUiToolApprovals = (\n  interrupts: readonly AgUiInterrupt[] | undefined,\n  boundToolCallIds?: ReadonlySet<string>,\n): ReadonlyMap<string, ToolApproval> => {\n  if (!interrupts?.length || !interrupts.every(isGate)) return EMPTY_APPROVALS;\n  if (\n    boundToolCallIds &&\n    !interrupts.every((interrupt) => boundToolCallIds.has(interrupt.toolCallId))\n  ) {\n    return EMPTY_APPROVALS;\n  }\n  // Two gates naming one tool call would collapse into a single approval, so\n  // the batch could never be completed: the surviving gate takes one decision\n  // while the resume array still waits on the id that no part carries.\n  const toolCallIds = interrupts.map((interrupt) => interrupt.toolCallId);\n  if (new Set(toolCallIds).size !== toolCallIds.length) return EMPTY_APPROVALS;\n  return new Map(\n    interrupts.map((interrupt) => [\n      interrupt.toolCallId,\n      {\n        id: interrupt.id,\n        ...(interrupt.message !== undefined && { prompt: interrupt.message }),\n      },\n    ]),\n  );\n};\n\n/**\n * Records a decision on the gated tool call. The decision is held on the part\n * until every gate in the batch is answered, because AG-UI resumes a run with\n * one response per open interrupt rather than one response per gate. A chosen\n * `optionId` is not recorded: the projection exposes an approval without\n * options, and core rejects an `optionId` absent from them, so no option can\n * reach this seam.\n */\nexport const withToolApprovalDecision = (\n  content: readonly ThreadAssistantMessagePart[],\n  { approvalId, approved, reason }: RespondToToolApprovalOptions,\n): readonly ThreadAssistantMessagePart[] => {\n  const { content: next, changed } = mapToolCallPartsDeep(content, (part) => {\n    const approval = part.approval;\n    if (approval?.id !== approvalId || !isPending(approval)) return part;\n    return {\n      ...part,\n      approval: {\n        ...approval,\n        approved,\n        ...(reason !== undefined && { reason }),\n      },\n    };\n  });\n  return changed ? next : content;\n};\n\n/**\n * Maps the recorded decisions onto AG-UI resume entries, or `null` while the\n * batch is incomplete. A denial is a resolved response carrying\n * `approved: false`, not a cancellation: `cancelled` means the user abandoned\n * the interrupt without meaningful input and carries no payload, which would\n * drop the denial reason on the wire.\n */\nexport const buildToolApprovalResume = (\n  content: readonly ThreadAssistantMessagePart[],\n  interrupts: readonly AgUiInterrupt[],\n): AgUiResumeEntry[] | null => {\n  const open = new Set(interrupts.map((interrupt) => interrupt.id));\n  // Decisions accumulate in a `Map` rather than in an object literal, because an\n  // interrupt id of `__proto__` would otherwise write to the prototype chain\n  // instead of recording a response, and `constructor` or `toString` would\n  // report themselves as answered while undecided.\n  const responses = new Map<string, { status: \"resolved\"; payload: unknown }>();\n  for (const part of iterateToolCallParts(content)) {\n    const approval = part.approval;\n    if (!approval || approval.approved === undefined) continue;\n    if (!open.has(approval.id)) continue;\n    responses.set(approval.id, {\n      status: \"resolved\",\n      payload: {\n        approved: approval.approved,\n        ...(approval.reason !== undefined && { reason: approval.reason }),\n      },\n    });\n  }\n  if (interrupts.some((interrupt) => !responses.has(interrupt.id))) return null;\n  // The schema is dropped rather than cast: AG-UI's own interrupt type cannot\n  // express a boolean schema, and `buildResumeArray` reads only the ids.\n  return buildResumeArray(\n    interrupts.map((interrupt) => {\n      const { responseSchema: _schema, ...rest } =\n        withoutRawResponseSchema(interrupt);\n      return rest;\n    }),\n    Object.fromEntries(responses),\n  );\n};\n\nconst settle = (\n  approval: ToolApproval,\n  entry: AgUiResumeEntry,\n): ToolApproval | undefined => {\n  // A decision this gate recorded locally but never sent is dropped rather than\n  // displayed beside the settlement it contradicts: a submitted entry carries\n  // the whole outcome, so no field of an earlier one may survive it.\n  const {\n    approved: _approved,\n    reason: _reason,\n    resolution: _resolution,\n    ...rest\n  } = approval;\n  if (entry.status === \"cancelled\") {\n    return { ...rest, resolution: \"cancelled\" };\n  }\n  const payload = entry.payload;\n  if (\n    typeof payload !== \"object\" ||\n    payload === null ||\n    typeof (payload as { approved?: unknown }).approved !== \"boolean\"\n  ) {\n    // The entry settles the gate with no decision. Nothing is fabricated, and a\n    // decision recorded locally is dropped: the payload that went out carried\n    // none, so displaying one would show what was never sent. The gate is still\n    // marked terminal, because the interrupt it belonged to is closed and a\n    // click on it afterwards would find nothing pending.\n    return { ...rest, resolution: \"cancelled\" };\n  }\n  const { approved, reason } = payload as {\n    approved: boolean;\n    reason?: unknown;\n  };\n  return {\n    ...rest,\n    approved,\n    ...(typeof reason === \"string\" && { reason }),\n  };\n};\n\n/**\n * Settles every gate the submitted resume array names, so the displayed outcome\n * cannot contradict the wire. A decision recorded locally is not exempt: a gate\n * approved while the batch waited on its sibling is still cancelled when\n * `steerAway` submits a cancellation for it. A gate is shown as cancelled only\n * where its own entry is cancelled; a resolved entry supplies the decision it\n * carries, and one that carries none settles the gate with no decision rather\n * than fabricating one or keeping an unsent local decision beside it.\n *\n * A settlement always overrides an earlier one, in either direction. A resolved\n * settlement leaves no state a locally recorded decision does not also leave —\n * core's `resolution` covers only `cancelled` and `expired` — so honoring the\n * first settlement instead would mean treating an unsent local decision as\n * terminal, which is exactly what the cancellation case must override.\n */\nexport const withSettledToolApprovals = (\n  content: readonly ThreadAssistantMessagePart[],\n  resume: readonly AgUiResumeEntry[],\n): readonly ThreadAssistantMessagePart[] => {\n  if (resume.length === 0) return content;\n  const byId = new Map(resume.map((entry) => [entry.interruptId, entry]));\n  const { content: next, changed } = mapToolCallPartsDeep(content, (part) => {\n    const approval = part.approval;\n    if (!approval) return part;\n    const entry = byId.get(approval.id);\n    if (!entry) return part;\n    const settled = settle(approval, entry);\n    if (!settled) return part;\n    return { ...part, approval: settled };\n  });\n  return changed ? next : content;\n};\n"],"mappings":";;;;AAkBA,MAAM,kCAAqD,IAAI,IAAI;AAEnE,MAAM,iBAAiB,UACrB,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAErE,MAAM,YAAY,UAAmB,OAAO,UAAU;AAEtD,MAAM,oCAAoB,IAAI,IAAI;CAChC;CACA;CACA;CACA;CACA;CACA;CACA;AACF,CAAC;AAED,MAAM,mBAAmB,UACvB,MAAM,QAAQ,KAAK,KACnB,MAAM,MAAM,QAAQ,KACpB,IAAI,IAAI,KAAK,CAAC,CAAC,SAAS,MAAM;AAEhC,MAAM,gBAAgB,SACnB,SAAS,IAAI,KAAK,kBAAkB,IAAI,IAAc,KACtD,MAAM,QAAQ,IAAI,KACjB,KAAK,SAAS,KACd,gBAAgB,IAAI,KACpB,KAAK,OAAO,UAAU,kBAAkB,IAAI,KAAe,CAAC;AAEhE,MAAM,eAAe,MAAe,cAClC,SAAS,aACR,MAAM,QAAQ,IAAI,KAAK,aAAa,IAAI,KAAK,KAAK,SAAS,SAAS;AAEvE,MAAM,iBAAiB,SAAkB,YAAY,MAAM,QAAQ;AAEnE,MAAM,yCAAyB,IAAI,IAAI;CACrC;CACA;CACA;CACA;AACF,CAAC;;;;;;;AAQD,MAAM,eAAe,QAAiB,cACpC,cAAc,MAAM,KACpB,OAAO,QAAQ,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,WACtC,YAAY,SACR,YAAY,OAAO,SAAS,IAC5B,uBAAuB,IAAI,OAAO,KAAK,SAAS,KAAK,CAC3D;;;;;;;;;AAUF,MAAM,oBAAoB,IAAI,IAAyC;CACrE,CAAC,QAAQ,YAAY;CACrB,CAAC,YAAY,eAAe;CAC5B,GAAG,CAAC,GAAG,sBAAsB,CAAC,CAAC,KAC5B,eACC,CAAC,YAAY,QAAQ,CACzB;CACA,CACE,eACC,UACC,cAAc,KAAK,KACnB,OAAO,OAAO,KAAK,CAAC,CAAC,OAAO,WAAW,sBAAsB,MAAM,CAAC,CACxE;CACA,CAAC,UAAU,UAAU,sBAAsB,KAAK,CAAC;AACnD,CAAC;;;;;;AAOD,MAAM,yBAAyB,WAC7B,OAAO,WAAW,aACjB,cAAc,MAAM,KACnB,OAAO,QAAQ,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,WAAW;CACjD,MAAM,QAAQ,kBAAkB,IAAI,OAAO;CAC3C,OAAO,UAAU,KAAA,KAAa,MAAM,KAAK;AAC3C,CAAC;;;;;;;;;AAUL,MAAM,yBAAyB,eAC7B,cAAc,UAAU,KACxB,OAAO,QAAQ,UAAU,CAAC,CAAC,OAAO,CAAC,OAAO,YAAY;CACpD,IAAI,UAAU,YAAY,OAAO,YAAY,QAAQ,SAAS;CAC9D,IAAI,UAAU,UAAU,OAAO,YAAY,QAAQ,QAAQ;CAC3D,OAAO,sBAAsB,MAAM;AACrC,CAAC;AAEH,MAAM,sBAAsB,aAC1B,MAAM,QAAQ,QAAQ,KACtB,SAAS,OAAO,UAAU,UAAU,UAAU,KAC9C,IAAI,IAAI,QAAQ,CAAC,CAAC,SAAS,SAAS;;;;;;;;;;;;AAatC,MAAM,qCAAqB,IAAI,IAAyC;CACtE,CAAC,WAAW,QAAQ;CACpB,CAAC,OAAO,QAAQ;CAChB,CAAC,SAAS,QAAQ;CAClB,CAAC,eAAe,QAAQ;CACxB,CAAC,QAAQ,aAAa;CACtB,CAAC,YAAY,kBAAkB;CAC/B,CAAC,cAAc,qBAAqB;AACtC,CAAC;;;;;;;;;;;AAYD,MAAM,oBAAoB,WAAoB;CAC5C,IAAI,WAAW,KAAA,GAAW,OAAO;CACjC,IAAI,CAAC,cAAc,MAAM,GAAG,OAAO;CACnC,OAAO,OAAO,QAAQ,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,WAAW;EACxD,MAAM,UAAU,mBAAmB,IAAI,OAAO;EAC9C,OAAO,YAAY,KAAA,KAAa,QAAQ,KAAK;CAC/C,CAAC;AACH;AAEA,MAAM,UACJ,cAEA,UAAU,WAAW,eACrB,CAAC,CAAC,UAAU,MACZ,CAAC,CAAC,UAAU,cAGZ,sBAAsB,SAAS,MAAM,KAAA,KACrC,iBAAiB,UAAU,cAAc;AAE3C,MAAM,aAAa,aACjB,CAAC,CAAC,YAAY,SAAS,aAAa,KAAA,KAAa,CAAC,SAAS;;;;;;;;;;;;;;;;;;AAmB7D,MAAa,4BACX,YACA,qBACsC;CACtC,IAAI,CAAC,YAAY,UAAU,CAAC,WAAW,MAAM,MAAM,GAAG,OAAO;CAC7D,IACE,oBACA,CAAC,WAAW,OAAO,cAAc,iBAAiB,IAAI,UAAU,UAAU,CAAC,GAE3E,OAAO;CAKT,MAAM,cAAc,WAAW,KAAK,cAAc,UAAU,UAAU;CACtE,IAAI,IAAI,IAAI,WAAW,CAAC,CAAC,SAAS,YAAY,QAAQ,OAAO;CAC7D,OAAO,IAAI,IACT,WAAW,KAAK,cAAc,CAC5B,UAAU,YACV;EACE,IAAI,UAAU;EACd,GAAI,UAAU,YAAY,KAAA,KAAa,EAAE,QAAQ,UAAU,QAAQ;CACrE,CACF,CAAC,CACH;AACF;;;;;;;;;AAUA,MAAa,4BACX,SACA,EAAE,YAAY,UAAU,aACkB;CAC1C,MAAM,EAAE,SAAS,MAAM,YAAY,qBAAqB,UAAU,SAAS;EACzE,MAAM,WAAW,KAAK;EACtB,IAAI,UAAU,OAAO,cAAc,CAAC,UAAU,QAAQ,GAAG,OAAO;EAChE,OAAO;GACL,GAAG;GACH,UAAU;IACR,GAAG;IACH;IACA,GAAI,WAAW,KAAA,KAAa,EAAE,OAAO;GACvC;EACF;CACF,CAAC;CACD,OAAO,UAAU,OAAO;AAC1B;;;;;;;;AASA,MAAa,2BACX,SACA,eAC6B;CAC7B,MAAM,OAAO,IAAI,IAAI,WAAW,KAAK,cAAc,UAAU,EAAE,CAAC;CAKhE,MAAM,4BAAY,IAAI,IAAsD;CAC5E,KAAK,MAAM,QAAQ,qBAAqB,OAAO,GAAG;EAChD,MAAM,WAAW,KAAK;EACtB,IAAI,CAAC,YAAY,SAAS,aAAa,KAAA,GAAW;EAClD,IAAI,CAAC,KAAK,IAAI,SAAS,EAAE,GAAG;EAC5B,UAAU,IAAI,SAAS,IAAI;GACzB,QAAQ;GACR,SAAS;IACP,UAAU,SAAS;IACnB,GAAI,SAAS,WAAW,KAAA,KAAa,EAAE,QAAQ,SAAS,OAAO;GACjE;EACF,CAAC;CACH;CACA,IAAI,WAAW,MAAM,cAAc,CAAC,UAAU,IAAI,UAAU,EAAE,CAAC,GAAG,OAAO;CAGzE,OAAO,iBACL,WAAW,KAAK,cAAc;EAC5B,MAAM,EAAE,gBAAgB,SAAS,GAAG,SAClC,yBAAyB,SAAS;EACpC,OAAO;CACT,CAAC,GACD,OAAO,YAAY,SAAS,CAC9B;AACF;AAEA,MAAM,UACJ,UACA,UAC6B;CAI7B,MAAM,EACJ,UAAU,WACV,QAAQ,SACR,YAAY,aACZ,GAAG,SACD;CACJ,IAAI,MAAM,WAAW,aACnB,OAAO;EAAE,GAAG;EAAM,YAAY;CAAY;CAE5C,MAAM,UAAU,MAAM;CACtB,IACE,OAAO,YAAY,YACnB,YAAY,QACZ,OAAQ,QAAmC,aAAa,WAOxD,OAAO;EAAE,GAAG;EAAM,YAAY;CAAY;CAE5C,MAAM,EAAE,UAAU,WAAW;CAI7B,OAAO;EACL,GAAG;EACH;EACA,GAAI,OAAO,WAAW,YAAY,EAAE,OAAO;CAC7C;AACF;;;;;;;;;;;;;;;;AAiBA,MAAa,4BACX,SACA,WAC0C;CAC1C,IAAI,OAAO,WAAW,GAAG,OAAO;CAChC,MAAM,OAAO,IAAI,IAAI,OAAO,KAAK,UAAU,CAAC,MAAM,aAAa,KAAK,CAAC,CAAC;CACtE,MAAM,EAAE,SAAS,MAAM,YAAY,qBAAqB,UAAU,SAAS;EACzE,MAAM,WAAW,KAAK;EACtB,IAAI,CAAC,UAAU,OAAO;EACtB,MAAM,QAAQ,KAAK,IAAI,SAAS,EAAE;EAClC,IAAI,CAAC,OAAO,OAAO;EACnB,MAAM,UAAU,OAAO,UAAU,KAAK;EACtC,IAAI,CAAC,SAAS,OAAO;EACrB,OAAO;GAAE,GAAG;GAAM,UAAU;EAAQ;CACtC,CAAC;CACD,OAAO,UAAU,OAAO;AAC1B"}
+{"version":3,"file":"tool-approval.js","names":[],"sources":["../../../src/runtime/adapter/tool-approval.ts"],"sourcesContent":["import {\n  iterateToolCallParts,\n  mapToolCallPartsDeep,\n} from \"@assistant-ui/core/internal\";\nimport { buildResumeArray } from \"@ag-ui/client\";\nimport type {\n  RespondToToolApprovalOptions,\n  ThreadAssistantMessagePart,\n  ToolCallMessagePart,\n} from \"@assistant-ui/core\";\nimport type { AgUiInterrupt, AgUiResumeEntry } from \"../types\";\nimport {\n  readRawResponseSchema,\n  withoutRawResponseSchema,\n} from \"../interrupt-internals\";\n\ntype ToolApproval = NonNullable<ToolCallMessagePart[\"approval\"]>;\n\nconst EMPTY_APPROVALS: ReadonlyMap<string, ToolApproval> = new Map();\n\nconst isPlainObject = (value: unknown): value is Record<string, unknown> =>\n  typeof value === \"object\" && value !== null && !Array.isArray(value);\n\nconst isString = (value: unknown) => typeof value === \"string\";\n\nconst JSON_SCHEMA_TYPES = new Set([\n  \"null\",\n  \"boolean\",\n  \"object\",\n  \"array\",\n  \"number\",\n  \"integer\",\n  \"string\",\n]);\n\nconst isUniqueStrings = (value: unknown) =>\n  Array.isArray(value) &&\n  value.every(isString) &&\n  new Set(value).size === value.length;\n\nconst isSchemaType = (type: unknown) =>\n  (isString(type) && JSON_SCHEMA_TYPES.has(type as string)) ||\n  (Array.isArray(type) &&\n    type.length > 0 &&\n    isUniqueStrings(type) &&\n    type.every((entry) => JSON_SCHEMA_TYPES.has(entry as string)));\n\nconst acceptsType = (type: unknown, primitive: string) =>\n  type === primitive ||\n  (Array.isArray(type) && isSchemaType(type) && type.includes(primitive));\n\nconst acceptsObject = (type: unknown) => acceptsType(type, \"object\");\n\nconst SUB_SCHEMA_ANNOTATIONS = new Set([\n  \"$schema\",\n  \"$id\",\n  \"title\",\n  \"description\",\n]);\n\n/**\n * A field schema accepts a value of `primitive` only where nothing in it can\n * narrow past the type: an annotation carries no constraint, and an absent\n * `type` constrains nothing. Any other keyword — `const`, `enum`, `anyOf`,\n * `format` — can reject the value this seam sends, so it is not interpreted.\n */\nconst acceptsOnly = (schema: unknown, primitive: string) =>\n  isPlainObject(schema) &&\n  Object.entries(schema).every(([keyword, value]) =>\n    keyword === \"type\"\n      ? acceptsType(value, primitive)\n      : SUB_SCHEMA_ANNOTATIONS.has(keyword) && isString(value),\n  );\n\n/**\n * Keywords of a sub-schema whose value shape a validator reads, each paired\n * with that shape: the same shapes the top-level schema is held to, plus\n * `properties` and `items`, which carry schemas, so a malformed one nested\n * under them is as uncompilable as a malformed one at the top. A keyword this\n * seam cannot judge is left alone: unlike the top-level schema, a field the\n * seam never sends needs only to be compilable, not permissive.\n */\nconst SUB_SCHEMA_SHAPES = new Map<string, (value: unknown) => boolean>([\n  [\"type\", isSchemaType],\n  [\"required\", isUniqueStrings],\n  ...[...SUB_SCHEMA_ANNOTATIONS].map(\n    (annotation) =>\n      [annotation, isString] as [string, (value: unknown) => boolean],\n  ),\n  [\n    \"properties\",\n    (value) =>\n      isPlainObject(value) &&\n      Object.values(value).every((schema) => isWellFormedSubSchema(schema)),\n  ],\n  [\"items\", (value) => isWellFormedSubSchema(value)],\n]);\n\n/**\n * A schema the server cannot compile fails the run whatever payload follows, so\n * a field this seam never sends is still checked for the shape a validator\n * reads it as. A boolean is a schema in its own right.\n */\nconst isWellFormedSubSchema = (schema: unknown): boolean =>\n  typeof schema === \"boolean\" ||\n  (isPlainObject(schema) &&\n    Object.entries(schema).every(([keyword, value]) => {\n      const holds = SUB_SCHEMA_SHAPES.get(keyword);\n      return holds === undefined || holds(value);\n    }));\n\n/**\n * AG-UI's own tool-approval schema declares the fields it expects under\n * `properties`, so a declaration is read where it cannot reject what the seam\n * sends: `approved` must accept a boolean and `reason` a string. A field the\n * seam never sends is unconstrained by `required` (which admits only\n * `approved`), so only its well-formedness as a sub-schema is checked — the\n * `editedArgs` of the spec's approve-with-edits shape is such a field.\n */\nconst acceptsSeamProperties = (properties: unknown) =>\n  isPlainObject(properties) &&\n  Object.entries(properties).every(([field, schema]) => {\n    if (field === \"approved\") return acceptsOnly(schema, \"boolean\");\n    if (field === \"reason\") return acceptsOnly(schema, \"string\");\n    return isWellFormedSubSchema(schema);\n  });\n\nconst acceptsSeamPayload = (required: unknown) =>\n  Array.isArray(required) &&\n  required.every((field) => field === \"approved\") &&\n  new Set(required).size === required.length;\n\n/**\n * Keywords that cannot reject a payload this seam emits, each paired with the\n * check that its value is the well-formed shape the keyword is read as. A\n * keyword whose value a validator would reject as an invalid schema is not\n * decided here, because a server that compiles the schema fails the run. An\n * unlisted keyword — `additionalProperties`, `allOf`, `$ref` — makes the gate\n * bespoke rather than being interpreted. The map is keyed by\n * `Map` rather than by object property, because a schema key of `constructor`\n * or `toString` would otherwise read an `Object.prototype` member as its\n * validator and `__proto__` would read the prototype itself.\n */\nconst SEAM_SAFE_KEYWORDS = new Map<string, (value: unknown) => boolean>([\n  [\"$schema\", isString],\n  [\"$id\", isString],\n  [\"title\", isString],\n  [\"description\", isString],\n  [\"type\", acceptsObject],\n  [\"required\", acceptsSeamPayload],\n  [\"properties\", acceptsSeamProperties],\n]);\n\n/**\n * The seam sends `{ approved }` and may add `reason`, so it may answer a\n * `responseSchema` only where acceptance of those exact payloads is established\n * by construction. AG-UI answers a resume payload its schema rejects with\n * `RunError`, and the host can always build the exact payload through the\n * bespoke interrupt hooks, so an undecidable schema is left to them. Only an\n * absent schema and a plain object of well-formed allowed keywords establish\n * that: a boolean schema (`false` rejects every payload), a malformed one, and\n * a keyword whose value is not the shape it interprets are all left bespoke.\n */\nconst isSeamAnswerable = (schema: unknown) => {\n  if (schema === undefined) return true;\n  if (!isPlainObject(schema)) return false;\n  return Object.entries(schema).every(([keyword, value]) => {\n    const accepts = SEAM_SAFE_KEYWORDS.get(keyword);\n    return accepts !== undefined && accepts(value);\n  });\n};\n\nconst isGate = (\n  interrupt: AgUiInterrupt,\n): interrupt is AgUiInterrupt & { toolCallId: string } =>\n  interrupt.reason === \"tool_call\" &&\n  !!interrupt.id &&\n  !!interrupt.toolCallId &&\n  // A restored interrupt is untyped, so `responseSchema` may hold any persisted\n  // shape; a non-object schema received live arrives on the internal carrier.\n  readRawResponseSchema(interrupt) === undefined &&\n  isSeamAnswerable(interrupt.responseSchema);\n\nconst isPending = (approval: ToolCallMessagePart[\"approval\"]) =>\n  !!approval && approval.approved === undefined && !approval.resolution;\n\n/**\n * AG-UI binds a decision to a tool call with `reason: \"tool_call\"`;\n * `confirmation` is a free-standing yes/no and `input_required` wants structured\n * input, so both stay on the bespoke interrupt hooks, as does a `tool_call`\n * whose `responseSchema` asks for more than this seam can send. A batch mixing a\n * gate with any other interrupt projects nothing: AG-UI resumes with one\n * response per open interrupt, so a half-owned batch cannot be completed from\n * either side alone.\n *\n * The same rule covers binding. A gate naming a tool call the message does not\n * render can never be decided through this seam, so the batch it belongs to\n * could never be completed either: the decisions taken on its visible siblings\n * would sit on the message while the run waited for a response that no click\n * can produce. `boundToolCallIds` spans every call the run produced — root or\n * nested on ToolCallMessagePart.messages — so only a gate naming a call the\n * message renders nowhere leaves the batch bespoke.\n */\nexport const projectAgUiToolApprovals = (\n  interrupts: readonly AgUiInterrupt[] | undefined,\n  boundToolCallIds?: ReadonlySet<string>,\n): ReadonlyMap<string, ToolApproval> => {\n  if (!interrupts?.length || !interrupts.every(isGate)) return EMPTY_APPROVALS;\n  if (\n    boundToolCallIds &&\n    !interrupts.every((interrupt) => boundToolCallIds.has(interrupt.toolCallId))\n  ) {\n    return EMPTY_APPROVALS;\n  }\n  // Two gates naming one tool call would collapse into a single approval, so\n  // the batch could never be completed: the surviving gate takes one decision\n  // while the resume array still waits on the id that no part carries.\n  const toolCallIds = interrupts.map((interrupt) => interrupt.toolCallId);\n  if (new Set(toolCallIds).size !== toolCallIds.length) return EMPTY_APPROVALS;\n  return new Map(\n    interrupts.map((interrupt) => [\n      interrupt.toolCallId,\n      {\n        id: interrupt.id,\n        ...(interrupt.message !== undefined && { prompt: interrupt.message }),\n      },\n    ]),\n  );\n};\n\n/** A recorded decision authorizes one exact call and gate, not a reused id. */\nexport const preserveToolApprovalDecision = (\n  previous: ToolCallMessagePart | undefined,\n  next: ToolCallMessagePart,\n): ToolApproval | undefined => {\n  const prior = previous?.approval;\n  let approval = next.approval;\n  if (!approval && prior) {\n    const {\n      approved: _approved,\n      reason: _reason,\n      resolution: _resolution,\n      ...pending\n    } = prior;\n    approval = pending;\n  }\n  if (!approval) return undefined;\n  // AG-UI may omit argument events for a no-argument call; both forms mean {}.\n  if (\n    !previous ||\n    prior?.id !== approval.id ||\n    previous.toolCallId !== next.toolCallId ||\n    previous.toolName !== next.toolName ||\n    (previous.argsText || \"{}\") !== (next.argsText || \"{}\")\n  )\n    return approval;\n  return {\n    ...approval,\n    ...(prior.approved !== undefined ? { approved: prior.approved } : {}),\n    ...(prior.reason !== undefined ? { reason: prior.reason } : {}),\n    ...(prior.resolution !== undefined ? { resolution: prior.resolution } : {}),\n  };\n};\n\n/**\n * Records a decision on the gated tool call. The decision is held on the part\n * until every gate in the batch is answered, because AG-UI resumes a run with\n * one response per open interrupt rather than one response per gate. A chosen\n * `optionId` is not recorded: the projection exposes an approval without\n * options, and core rejects an `optionId` absent from them, so no option can\n * reach this seam.\n */\nexport const withToolApprovalDecision = (\n  content: readonly ThreadAssistantMessagePart[],\n  { approvalId, approved, reason }: RespondToToolApprovalOptions,\n): readonly ThreadAssistantMessagePart[] => {\n  const { content: next, changed } = mapToolCallPartsDeep(content, (part) => {\n    const approval = part.approval;\n    if (approval?.id !== approvalId || !isPending(approval)) return part;\n    return {\n      ...part,\n      approval: {\n        ...approval,\n        approved,\n        ...(reason !== undefined && { reason }),\n      },\n    };\n  });\n  return changed ? next : content;\n};\n\n/**\n * Maps the recorded decisions onto AG-UI resume entries, or `null` while the\n * batch is incomplete. A denial is a resolved response carrying\n * `approved: false`, not a cancellation: `cancelled` means the user abandoned\n * the interrupt without meaningful input and carries no payload, which would\n * drop the denial reason on the wire.\n */\nexport const buildToolApprovalResume = (\n  content: readonly ThreadAssistantMessagePart[],\n  interrupts: readonly AgUiInterrupt[],\n): AgUiResumeEntry[] | null => {\n  const calls = Array.from(iterateToolCallParts(content));\n  const gates = projectAgUiToolApprovals(\n    interrupts,\n    new Set(calls.map((part) => part.toolCallId)),\n  );\n  // Decisions accumulate in a `Map` rather than in an object literal, because an\n  // interrupt id of `__proto__` would otherwise write to the prototype chain\n  // instead of recording a response, and `constructor` or `toString` would\n  // report themselves as answered while undecided.\n  const responses = new Map<string, { status: \"resolved\"; payload: unknown }>();\n  for (const part of calls) {\n    const approval = part.approval;\n    if (!approval || approval.approved === undefined) continue;\n    if (gates.get(part.toolCallId)?.id !== approval.id) continue;\n    responses.set(approval.id, {\n      status: \"resolved\",\n      payload: {\n        approved: approval.approved,\n        ...(approval.reason !== undefined && { reason: approval.reason }),\n      },\n    });\n  }\n  if (interrupts.some((interrupt) => !responses.has(interrupt.id))) return null;\n  // The schema is dropped rather than cast: AG-UI's own interrupt type cannot\n  // express a boolean schema, and `buildResumeArray` reads only the ids.\n  return buildResumeArray(\n    interrupts.map((interrupt) => {\n      const { responseSchema: _schema, ...rest } =\n        withoutRawResponseSchema(interrupt);\n      return rest;\n    }),\n    Object.fromEntries(responses),\n  );\n};\n\nconst settle = (\n  approval: ToolApproval,\n  entry: AgUiResumeEntry,\n): ToolApproval | undefined => {\n  // A decision this gate recorded locally but never sent is dropped rather than\n  // displayed beside the settlement it contradicts: a submitted entry carries\n  // the whole outcome, so no field of an earlier one may survive it.\n  const {\n    approved: _approved,\n    reason: _reason,\n    resolution: _resolution,\n    ...rest\n  } = approval;\n  if (entry.status === \"cancelled\") {\n    return { ...rest, resolution: \"cancelled\" };\n  }\n  const payload = entry.payload;\n  if (\n    typeof payload !== \"object\" ||\n    payload === null ||\n    typeof (payload as { approved?: unknown }).approved !== \"boolean\"\n  ) {\n    // The entry settles the gate with no decision. Nothing is fabricated, and a\n    // decision recorded locally is dropped: the payload that went out carried\n    // none, so displaying one would show what was never sent. The gate is still\n    // marked terminal, because the interrupt it belonged to is closed and a\n    // click on it afterwards would find nothing pending.\n    return { ...rest, resolution: \"cancelled\" };\n  }\n  const { approved, reason } = payload as {\n    approved: boolean;\n    reason?: unknown;\n  };\n  return {\n    ...rest,\n    approved,\n    ...(typeof reason === \"string\" && { reason }),\n  };\n};\n\n/**\n * Settles every gate the submitted resume array names, so the displayed outcome\n * cannot contradict the wire. A decision recorded locally is not exempt: a gate\n * approved while the batch waited on its sibling is still cancelled when\n * `steerAway` submits a cancellation for it. A gate is shown as cancelled only\n * where its own entry is cancelled; a resolved entry supplies the decision it\n * carries, and one that carries none settles the gate with no decision rather\n * than fabricating one or keeping an unsent local decision beside it.\n *\n * A settlement always overrides an earlier one, in either direction. A resolved\n * settlement leaves no state a locally recorded decision does not also leave —\n * core's `resolution` covers only `cancelled` and `expired` — so honoring the\n * first settlement instead would mean treating an unsent local decision as\n * terminal, which is exactly what the cancellation case must override.\n */\nexport const withSettledToolApprovals = (\n  content: readonly ThreadAssistantMessagePart[],\n  resume: readonly AgUiResumeEntry[],\n): readonly ThreadAssistantMessagePart[] => {\n  if (resume.length === 0) return content;\n  const byId = new Map(resume.map((entry) => [entry.interruptId, entry]));\n  const { content: next, changed } = mapToolCallPartsDeep(content, (part) => {\n    const approval = part.approval;\n    if (!approval) return part;\n    const entry = byId.get(approval.id);\n    if (!entry) return part;\n    const settled = settle(approval, entry);\n    if (!settled) return part;\n    return { ...part, approval: settled };\n  });\n  return changed ? next : content;\n};\n"],"mappings":";;;;AAkBA,MAAM,kCAAqD,IAAI,IAAI;AAEnE,MAAM,iBAAiB,UACrB,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAErE,MAAM,YAAY,UAAmB,OAAO,UAAU;AAEtD,MAAM,oCAAoB,IAAI,IAAI;CAChC;CACA;CACA;CACA;CACA;CACA;CACA;AACF,CAAC;AAED,MAAM,mBAAmB,UACvB,MAAM,QAAQ,KAAK,KACnB,MAAM,MAAM,QAAQ,KACpB,IAAI,IAAI,KAAK,CAAC,CAAC,SAAS,MAAM;AAEhC,MAAM,gBAAgB,SACnB,SAAS,IAAI,KAAK,kBAAkB,IAAI,IAAc,KACtD,MAAM,QAAQ,IAAI,KACjB,KAAK,SAAS,KACd,gBAAgB,IAAI,KACpB,KAAK,OAAO,UAAU,kBAAkB,IAAI,KAAe,CAAC;AAEhE,MAAM,eAAe,MAAe,cAClC,SAAS,aACR,MAAM,QAAQ,IAAI,KAAK,aAAa,IAAI,KAAK,KAAK,SAAS,SAAS;AAEvE,MAAM,iBAAiB,SAAkB,YAAY,MAAM,QAAQ;AAEnE,MAAM,yCAAyB,IAAI,IAAI;CACrC;CACA;CACA;CACA;AACF,CAAC;;;;;;;AAQD,MAAM,eAAe,QAAiB,cACpC,cAAc,MAAM,KACpB,OAAO,QAAQ,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,WACtC,YAAY,SACR,YAAY,OAAO,SAAS,IAC5B,uBAAuB,IAAI,OAAO,KAAK,SAAS,KAAK,CAC3D;;;;;;;;;AAUF,MAAM,oBAAoB,IAAI,IAAyC;CACrE,CAAC,QAAQ,YAAY;CACrB,CAAC,YAAY,eAAe;CAC5B,GAAG,CAAC,GAAG,sBAAsB,CAAC,CAAC,KAC5B,eACC,CAAC,YAAY,QAAQ,CACzB;CACA,CACE,eACC,UACC,cAAc,KAAK,KACnB,OAAO,OAAO,KAAK,CAAC,CAAC,OAAO,WAAW,sBAAsB,MAAM,CAAC,CACxE;CACA,CAAC,UAAU,UAAU,sBAAsB,KAAK,CAAC;AACnD,CAAC;;;;;;AAOD,MAAM,yBAAyB,WAC7B,OAAO,WAAW,aACjB,cAAc,MAAM,KACnB,OAAO,QAAQ,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,WAAW;CACjD,MAAM,QAAQ,kBAAkB,IAAI,OAAO;CAC3C,OAAO,UAAU,KAAA,KAAa,MAAM,KAAK;AAC3C,CAAC;;;;;;;;;AAUL,MAAM,yBAAyB,eAC7B,cAAc,UAAU,KACxB,OAAO,QAAQ,UAAU,CAAC,CAAC,OAAO,CAAC,OAAO,YAAY;CACpD,IAAI,UAAU,YAAY,OAAO,YAAY,QAAQ,SAAS;CAC9D,IAAI,UAAU,UAAU,OAAO,YAAY,QAAQ,QAAQ;CAC3D,OAAO,sBAAsB,MAAM;AACrC,CAAC;AAEH,MAAM,sBAAsB,aAC1B,MAAM,QAAQ,QAAQ,KACtB,SAAS,OAAO,UAAU,UAAU,UAAU,KAC9C,IAAI,IAAI,QAAQ,CAAC,CAAC,SAAS,SAAS;;;;;;;;;;;;AAatC,MAAM,qCAAqB,IAAI,IAAyC;CACtE,CAAC,WAAW,QAAQ;CACpB,CAAC,OAAO,QAAQ;CAChB,CAAC,SAAS,QAAQ;CAClB,CAAC,eAAe,QAAQ;CACxB,CAAC,QAAQ,aAAa;CACtB,CAAC,YAAY,kBAAkB;CAC/B,CAAC,cAAc,qBAAqB;AACtC,CAAC;;;;;;;;;;;AAYD,MAAM,oBAAoB,WAAoB;CAC5C,IAAI,WAAW,KAAA,GAAW,OAAO;CACjC,IAAI,CAAC,cAAc,MAAM,GAAG,OAAO;CACnC,OAAO,OAAO,QAAQ,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,WAAW;EACxD,MAAM,UAAU,mBAAmB,IAAI,OAAO;EAC9C,OAAO,YAAY,KAAA,KAAa,QAAQ,KAAK;CAC/C,CAAC;AACH;AAEA,MAAM,UACJ,cAEA,UAAU,WAAW,eACrB,CAAC,CAAC,UAAU,MACZ,CAAC,CAAC,UAAU,cAGZ,sBAAsB,SAAS,MAAM,KAAA,KACrC,iBAAiB,UAAU,cAAc;AAE3C,MAAM,aAAa,aACjB,CAAC,CAAC,YAAY,SAAS,aAAa,KAAA,KAAa,CAAC,SAAS;;;;;;;;;;;;;;;;;;AAmB7D,MAAa,4BACX,YACA,qBACsC;CACtC,IAAI,CAAC,YAAY,UAAU,CAAC,WAAW,MAAM,MAAM,GAAG,OAAO;CAC7D,IACE,oBACA,CAAC,WAAW,OAAO,cAAc,iBAAiB,IAAI,UAAU,UAAU,CAAC,GAE3E,OAAO;CAKT,MAAM,cAAc,WAAW,KAAK,cAAc,UAAU,UAAU;CACtE,IAAI,IAAI,IAAI,WAAW,CAAC,CAAC,SAAS,YAAY,QAAQ,OAAO;CAC7D,OAAO,IAAI,IACT,WAAW,KAAK,cAAc,CAC5B,UAAU,YACV;EACE,IAAI,UAAU;EACd,GAAI,UAAU,YAAY,KAAA,KAAa,EAAE,QAAQ,UAAU,QAAQ;CACrE,CACF,CAAC,CACH;AACF;;AAGA,MAAa,gCACX,UACA,SAC6B;CAC7B,MAAM,QAAQ,UAAU;CACxB,IAAI,WAAW,KAAK;CACpB,IAAI,CAAC,YAAY,OAAO;EACtB,MAAM,EACJ,UAAU,WACV,QAAQ,SACR,YAAY,aACZ,GAAG,YACD;EACJ,WAAW;CACb;CACA,IAAI,CAAC,UAAU,OAAO,KAAA;CAEtB,IACE,CAAC,YACD,OAAO,OAAO,SAAS,MACvB,SAAS,eAAe,KAAK,cAC7B,SAAS,aAAa,KAAK,aAC1B,SAAS,YAAY,WAAW,KAAK,YAAY,OAElD,OAAO;CACT,OAAO;EACL,GAAG;EACH,GAAI,MAAM,aAAa,KAAA,IAAY,EAAE,UAAU,MAAM,SAAS,IAAI,CAAC;EACnE,GAAI,MAAM,WAAW,KAAA,IAAY,EAAE,QAAQ,MAAM,OAAO,IAAI,CAAC;EAC7D,GAAI,MAAM,eAAe,KAAA,IAAY,EAAE,YAAY,MAAM,WAAW,IAAI,CAAC;CAC3E;AACF;;;;;;;;;AAUA,MAAa,4BACX,SACA,EAAE,YAAY,UAAU,aACkB;CAC1C,MAAM,EAAE,SAAS,MAAM,YAAY,qBAAqB,UAAU,SAAS;EACzE,MAAM,WAAW,KAAK;EACtB,IAAI,UAAU,OAAO,cAAc,CAAC,UAAU,QAAQ,GAAG,OAAO;EAChE,OAAO;GACL,GAAG;GACH,UAAU;IACR,GAAG;IACH;IACA,GAAI,WAAW,KAAA,KAAa,EAAE,OAAO;GACvC;EACF;CACF,CAAC;CACD,OAAO,UAAU,OAAO;AAC1B;;;;;;;;AASA,MAAa,2BACX,SACA,eAC6B;CAC7B,MAAM,QAAQ,MAAM,KAAK,qBAAqB,OAAO,CAAC;CACtD,MAAM,QAAQ,yBACZ,YACA,IAAI,IAAI,MAAM,KAAK,SAAS,KAAK,UAAU,CAAC,CAC9C;CAKA,MAAM,4BAAY,IAAI,IAAsD;CAC5E,KAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,WAAW,KAAK;EACtB,IAAI,CAAC,YAAY,SAAS,aAAa,KAAA,GAAW;EAClD,IAAI,MAAM,IAAI,KAAK,UAAU,CAAC,EAAE,OAAO,SAAS,IAAI;EACpD,UAAU,IAAI,SAAS,IAAI;GACzB,QAAQ;GACR,SAAS;IACP,UAAU,SAAS;IACnB,GAAI,SAAS,WAAW,KAAA,KAAa,EAAE,QAAQ,SAAS,OAAO;GACjE;EACF,CAAC;CACH;CACA,IAAI,WAAW,MAAM,cAAc,CAAC,UAAU,IAAI,UAAU,EAAE,CAAC,GAAG,OAAO;CAGzE,OAAO,iBACL,WAAW,KAAK,cAAc;EAC5B,MAAM,EAAE,gBAAgB,SAAS,GAAG,SAClC,yBAAyB,SAAS;EACpC,OAAO;CACT,CAAC,GACD,OAAO,YAAY,SAAS,CAC9B;AACF;AAEA,MAAM,UACJ,UACA,UAC6B;CAI7B,MAAM,EACJ,UAAU,WACV,QAAQ,SACR,YAAY,aACZ,GAAG,SACD;CACJ,IAAI,MAAM,WAAW,aACnB,OAAO;EAAE,GAAG;EAAM,YAAY;CAAY;CAE5C,MAAM,UAAU,MAAM;CACtB,IACE,OAAO,YAAY,YACnB,YAAY,QACZ,OAAQ,QAAmC,aAAa,WAOxD,OAAO;EAAE,GAAG;EAAM,YAAY;CAAY;CAE5C,MAAM,EAAE,UAAU,WAAW;CAI7B,OAAO;EACL,GAAG;EACH;EACA,GAAI,OAAO,WAAW,YAAY,EAAE,OAAO;CAC7C;AACF;;;;;;;;;;;;;;;;AAiBA,MAAa,4BACX,SACA,WAC0C;CAC1C,IAAI,OAAO,WAAW,GAAG,OAAO;CAChC,MAAM,OAAO,IAAI,IAAI,OAAO,KAAK,UAAU,CAAC,MAAM,aAAa,KAAK,CAAC,CAAC;CACtE,MAAM,EAAE,SAAS,MAAM,YAAY,qBAAqB,UAAU,SAAS;EACzE,MAAM,WAAW,KAAK;EACtB,IAAI,CAAC,UAAU,OAAO;EACtB,MAAM,QAAQ,KAAK,IAAI,SAAS,EAAE;EAClC,IAAI,CAAC,OAAO,OAAO;EACnB,MAAM,UAAU,OAAO,UAAU,KAAK;EACtC,IAAI,CAAC,SAAS,OAAO;EACrB,OAAO;GAAE,GAAG;GAAM,UAAU;EAAQ;CACtC,CAAC;CACD,OAAO,UAAU,OAAO;AAC1B"}

package/dist/runtime/agUiThreadState.d.ts

@@ -9,7 +9,7 @@ export type AgUiThreadState = {
     lastRunConfig: RunConfig | undefined;
     pendingResume: {
         owner: AbortController;
-        messageId: string;
+        messageIds: readonly string[];
     } | null;
     pendingA2uiResumeOwner: AbortController | null;
     pendingA2uiAction: Record<string, unknown> | undefined;

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

@@ -1 +1 @@
-{"version":3,"file":"agUiThreadState.d.ts","sourceRoot":"","sources":["../../src/runtime/agUiThreadState.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAEhE,MAAM,MAAM,eAAe,GAAG;IAC5B,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,aAAa,EAAE,OAAO,CAAC;IACvB,yBAAyB,EAAE,MAAM,CAAC;IAClC,aAAa,EAAE,iBAAiB,GAAG,SAAS,CAAC;IAC7C,aAAa,EAAE,SAAS,GAAG,SAAS,CAAC;IACrC,aAAa,EAAE;QAAE,KAAK,EAAE,eAAe,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IACpE,sBAAsB,EAAE,eAAe,GAAG,IAAI,CAAC;IAC/C,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;CACxD,CAAC;AAEF,MAAM,MAAM,gBAAgB,GACxB;IAAE,IAAI,EAAE,YAAY,CAAA;CAAE,GACtB;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,GACvB;IAAE,IAAI,EAAE,eAAe,CAAA;CAAE,GACzB;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,GACvB;IAAE,IAAI,EAAE,cAAc,CAAA;CAAE,GACxB;IAAE,IAAI,EAAE,yBAAyB,CAAA;CAAE,GACnC;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,QAAQ,EAAE,iBAAiB,CAAA;CAAE,GACtD;IAAE,IAAI,EAAE,YAAY,CAAA;CAAE,GACtB;IAAE,IAAI,EAAE,mBAAmB,CAAC;IAAC,SAAS,EAAE,SAAS,CAAA;CAAE,GACnD;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,KAAK,EAAE,eAAe,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GACrE;IAAE,IAAI,EAAE,eAAe,CAAA;CAAE,GACzB;IAAE,IAAI,EAAE,oBAAoB,CAAC;IAAC,KAAK,EAAE,eAAe,CAAA;CAAE,GACtD;IAAE,IAAI,EAAE,mBAAmB,CAAA;CAAE,GAC7B;IAAE,IAAI,EAAE,oBAAoB,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,GAC/D;IAAE,IAAI,EAAE,mBAAmB,CAAA;CAAE,GAC7B;IAAE,IAAI,EAAE,0BAA0B,CAAA;CAAE,GACpC;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,CAAC;AAE5B,eAAO,MAAM,qBAAqB,QAAO,eAUvC,CAAC;AAEH,eAAO,MAAM,qBAAqB,GAChC,OAAO,eAAe,EACtB,QAAQ,gBAAgB,KACvB,eAgFF,CAAC"}
+{"version":3,"file":"agUiThreadState.d.ts","sourceRoot":"","sources":["../../src/runtime/agUiThreadState.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAEhE,MAAM,MAAM,eAAe,GAAG;IAC5B,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,aAAa,EAAE,OAAO,CAAC;IACvB,yBAAyB,EAAE,MAAM,CAAC;IAClC,aAAa,EAAE,iBAAiB,GAAG,SAAS,CAAC;IAC7C,aAAa,EAAE,SAAS,GAAG,SAAS,CAAC;IACrC,aAAa,EAAE;QACb,KAAK,EAAE,eAAe,CAAC;QACvB,UAAU,EAAE,SAAS,MAAM,EAAE,CAAC;KAC/B,GAAG,IAAI,CAAC;IACT,sBAAsB,EAAE,eAAe,GAAG,IAAI,CAAC;IAC/C,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;CACxD,CAAC;AAEF,MAAM,MAAM,gBAAgB,GACxB;IAAE,IAAI,EAAE,YAAY,CAAA;CAAE,GACtB;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,GACvB;IAAE,IAAI,EAAE,eAAe,CAAA;CAAE,GACzB;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,GACvB;IAAE,IAAI,EAAE,cAAc,CAAA;CAAE,GACxB;IAAE,IAAI,EAAE,yBAAyB,CAAA;CAAE,GACnC;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,QAAQ,EAAE,iBAAiB,CAAA;CAAE,GACtD;IAAE,IAAI,EAAE,YAAY,CAAA;CAAE,GACtB;IAAE,IAAI,EAAE,mBAAmB,CAAC;IAAC,SAAS,EAAE,SAAS,CAAA;CAAE,GACnD;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,KAAK,EAAE,eAAe,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GACrE;IAAE,IAAI,EAAE,eAAe,CAAA;CAAE,GACzB;IAAE,IAAI,EAAE,oBAAoB,CAAC;IAAC,KAAK,EAAE,eAAe,CAAA;CAAE,GACtD;IAAE,IAAI,EAAE,mBAAmB,CAAA;CAAE,GAC7B;IAAE,IAAI,EAAE,oBAAoB,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,GAC/D;IAAE,IAAI,EAAE,mBAAmB,CAAA;CAAE,GAC7B;IAAE,IAAI,EAAE,0BAA0B,CAAA;CAAE,GACpC;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,CAAC;AAE5B,eAAO,MAAM,qBAAqB,QAAO,eAUvC,CAAC;AAEH,eAAO,MAAM,qBAAqB,GAChC,OAAO,eAAe,EACtB,QAAQ,gBAAgB,KACvB,eAsFF,CAAC"}

package/dist/runtime/agUiThreadState.js

@@ -1,4 +1,4 @@
-//#region src/runtime/agUiThreadState.ts
+// src/runtime/agUiThreadState.ts
 const createAgUiThreadState = () => ({
 	isRunning: false,
 	isLoading: false,
@@ -48,11 +48,11 @@ const reduceAgUiThreadState = (state, action) => {
 			...state,
 			lastRunConfig: action.runConfig
 		};
-		case "resumeDeferred": return state.pendingResume?.owner === action.owner && state.pendingResume.messageId === action.messageId ? state : {
+		case "resumeDeferred": return state.pendingResume?.owner === action.owner && state.pendingResume.messageIds.includes(action.messageId) ? state : {
 			...state,
 			pendingResume: {
 				owner: action.owner,
-				messageId: action.messageId
+				messageIds: state.pendingResume?.owner === action.owner ? [...state.pendingResume.messageIds, action.messageId] : [action.messageId]
 			}
 		};
 		case "resumeCleared": return state.pendingResume === null ? state : {
@@ -89,7 +89,6 @@ const reduceAgUiThreadState = (state, action) => {
 		};
 	}
 };
-//#endregion
 export { createAgUiThreadState, reduceAgUiThreadState };
 
 //# sourceMappingURL=agUiThreadState.js.map

package/dist/runtime/agUiThreadState.js.map

@@ -1 +1 @@
-{"version":3,"file":"agUiThreadState.js","names":[],"sources":["../../src/runtime/agUiThreadState.ts"],"sourcesContent":["import type { RunConfig } from \"@assistant-ui/core\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\n\nexport type AgUiThreadState = {\n  isRunning: boolean;\n  isLoading: boolean;\n  loadRequested: boolean;\n  historyMutationGeneration: number;\n  stateSnapshot: ReadonlyJSONValue | undefined;\n  lastRunConfig: RunConfig | undefined;\n  pendingResume: { owner: AbortController; messageId: string } | null;\n  pendingA2uiResumeOwner: AbortController | null;\n  pendingA2uiAction: Record<string, unknown> | undefined;\n};\n\nexport type AgUiThreadAction =\n  | { type: \"runStarted\" }\n  | { type: \"runFinished\" }\n  | { type: \"loadRequested\" }\n  | { type: \"loadStarted\" }\n  | { type: \"loadFinished\" }\n  | { type: \"historyMutationAdvanced\" }\n  | { type: \"stateSnapshot\"; snapshot: ReadonlyJSONValue }\n  | { type: \"resetState\" }\n  | { type: \"runConfigRecorded\"; runConfig: RunConfig }\n  | { type: \"resumeDeferred\"; owner: AbortController; messageId: string }\n  | { type: \"resumeCleared\" }\n  | { type: \"a2uiResumeDeferred\"; owner: AbortController }\n  | { type: \"a2uiResumeCleared\" }\n  | { type: \"a2uiActionDeferred\"; action: Record<string, unknown> }\n  | { type: \"a2uiActionCleared\" }\n  | { type: \"a2uiContinuationsCleared\" }\n  | { type: \"resetThread\" };\n\nexport const createAgUiThreadState = (): AgUiThreadState => ({\n  isRunning: false,\n  isLoading: false,\n  loadRequested: false,\n  historyMutationGeneration: 0,\n  stateSnapshot: undefined,\n  lastRunConfig: undefined,\n  pendingResume: null,\n  pendingA2uiResumeOwner: null,\n  pendingA2uiAction: undefined,\n});\n\nexport const reduceAgUiThreadState = (\n  state: AgUiThreadState,\n  action: AgUiThreadAction,\n): AgUiThreadState => {\n  switch (action.type) {\n    case \"runStarted\":\n      return state.isRunning ? state : { ...state, isRunning: true };\n    case \"runFinished\":\n      return !state.isRunning ? state : { ...state, isRunning: false };\n    case \"loadRequested\":\n      return state.loadRequested ? state : { ...state, loadRequested: true };\n    case \"loadStarted\":\n      return state.isLoading ? state : { ...state, isLoading: true };\n    case \"loadFinished\":\n      return !state.isLoading ? state : { ...state, isLoading: false };\n    case \"historyMutationAdvanced\":\n      return {\n        ...state,\n        historyMutationGeneration: state.historyMutationGeneration + 1,\n      };\n    case \"stateSnapshot\":\n      return state.stateSnapshot === action.snapshot\n        ? state\n        : { ...state, stateSnapshot: action.snapshot };\n    case \"resetState\":\n      return state.stateSnapshot === undefined\n        ? state\n        : { ...state, stateSnapshot: undefined };\n    case \"runConfigRecorded\":\n      return state.lastRunConfig === action.runConfig\n        ? state\n        : { ...state, lastRunConfig: action.runConfig };\n    case \"resumeDeferred\":\n      return state.pendingResume?.owner === action.owner &&\n        state.pendingResume.messageId === action.messageId\n        ? state\n        : {\n            ...state,\n            pendingResume: { owner: action.owner, messageId: action.messageId },\n          };\n    case \"resumeCleared\":\n      return state.pendingResume === null\n        ? state\n        : { ...state, pendingResume: null };\n    case \"a2uiResumeDeferred\":\n      return state.pendingA2uiResumeOwner === action.owner\n        ? state\n        : { ...state, pendingA2uiResumeOwner: action.owner };\n    case \"a2uiResumeCleared\":\n      return state.pendingA2uiResumeOwner === null\n        ? state\n        : { ...state, pendingA2uiResumeOwner: null };\n    case \"a2uiActionDeferred\":\n      return state.pendingA2uiAction === action.action\n        ? state\n        : { ...state, pendingA2uiAction: action.action };\n    case \"a2uiActionCleared\":\n      return state.pendingA2uiAction === undefined\n        ? state\n        : { ...state, pendingA2uiAction: undefined };\n    case \"a2uiContinuationsCleared\":\n      return state.pendingA2uiResumeOwner === null &&\n        state.pendingA2uiAction === undefined\n        ? state\n        : {\n            ...state,\n            pendingA2uiResumeOwner: null,\n            pendingA2uiAction: undefined,\n          };\n    case \"resetThread\":\n      return state.stateSnapshot === undefined &&\n        state.pendingResume === null &&\n        state.pendingA2uiResumeOwner === null &&\n        state.pendingA2uiAction === undefined\n        ? state\n        : {\n            ...state,\n            stateSnapshot: undefined,\n            pendingResume: null,\n            pendingA2uiResumeOwner: null,\n            pendingA2uiAction: undefined,\n          };\n  }\n};\n"],"mappings":";AAkCA,MAAa,+BAAgD;CAC3D,WAAW;CACX,WAAW;CACX,eAAe;CACf,2BAA2B;CAC3B,eAAe,KAAA;CACf,eAAe,KAAA;CACf,eAAe;CACf,wBAAwB;CACxB,mBAAmB,KAAA;AACrB;AAEA,MAAa,yBACX,OACA,WACoB;CACpB,QAAQ,OAAO,MAAf;EACE,KAAK,cACH,OAAO,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAK;EAC/D,KAAK,eACH,OAAO,CAAC,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAM;EACjE,KAAK,iBACH,OAAO,MAAM,gBAAgB,QAAQ;GAAE,GAAG;GAAO,eAAe;EAAK;EACvE,KAAK,eACH,OAAO,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAK;EAC/D,KAAK,gBACH,OAAO,CAAC,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAM;EACjE,KAAK,2BACH,OAAO;GACL,GAAG;GACH,2BAA2B,MAAM,4BAA4B;EAC/D;EACF,KAAK,iBACH,OAAO,MAAM,kBAAkB,OAAO,WAClC,QACA;GAAE,GAAG;GAAO,eAAe,OAAO;EAAS;EACjD,KAAK,cACH,OAAO,MAAM,kBAAkB,KAAA,IAC3B,QACA;GAAE,GAAG;GAAO,eAAe,KAAA;EAAU;EAC3C,KAAK,qBACH,OAAO,MAAM,kBAAkB,OAAO,YAClC,QACA;GAAE,GAAG;GAAO,eAAe,OAAO;EAAU;EAClD,KAAK,kBACH,OAAO,MAAM,eAAe,UAAU,OAAO,SAC3C,MAAM,cAAc,cAAc,OAAO,YACvC,QACA;GACE,GAAG;GACH,eAAe;IAAE,OAAO,OAAO;IAAO,WAAW,OAAO;GAAU;EACpE;EACN,KAAK,iBACH,OAAO,MAAM,kBAAkB,OAC3B,QACA;GAAE,GAAG;GAAO,eAAe;EAAK;EACtC,KAAK,sBACH,OAAO,MAAM,2BAA2B,OAAO,QAC3C,QACA;GAAE,GAAG;GAAO,wBAAwB,OAAO;EAAM;EACvD,KAAK,qBACH,OAAO,MAAM,2BAA2B,OACpC,QACA;GAAE,GAAG;GAAO,wBAAwB;EAAK;EAC/C,KAAK,sBACH,OAAO,MAAM,sBAAsB,OAAO,SACtC,QACA;GAAE,GAAG;GAAO,mBAAmB,OAAO;EAAO;EACnD,KAAK,qBACH,OAAO,MAAM,sBAAsB,KAAA,IAC/B,QACA;GAAE,GAAG;GAAO,mBAAmB,KAAA;EAAU;EAC/C,KAAK,4BACH,OAAO,MAAM,2BAA2B,QACtC,MAAM,sBAAsB,KAAA,IAC1B,QACA;GACE,GAAG;GACH,wBAAwB;GACxB,mBAAmB,KAAA;EACrB;EACN,KAAK,eACH,OAAO,MAAM,kBAAkB,KAAA,KAC7B,MAAM,kBAAkB,QACxB,MAAM,2BAA2B,QACjC,MAAM,sBAAsB,KAAA,IAC1B,QACA;GACE,GAAG;GACH,eAAe,KAAA;GACf,eAAe;GACf,wBAAwB;GACxB,mBAAmB,KAAA;EACrB;CACR;AACF"}
+{"version":3,"file":"agUiThreadState.js","names":[],"sources":["../../src/runtime/agUiThreadState.ts"],"sourcesContent":["import type { RunConfig } from \"@assistant-ui/core\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\n\nexport type AgUiThreadState = {\n  isRunning: boolean;\n  isLoading: boolean;\n  loadRequested: boolean;\n  historyMutationGeneration: number;\n  stateSnapshot: ReadonlyJSONValue | undefined;\n  lastRunConfig: RunConfig | undefined;\n  pendingResume: {\n    owner: AbortController;\n    messageIds: readonly string[];\n  } | null;\n  pendingA2uiResumeOwner: AbortController | null;\n  pendingA2uiAction: Record<string, unknown> | undefined;\n};\n\nexport type AgUiThreadAction =\n  | { type: \"runStarted\" }\n  | { type: \"runFinished\" }\n  | { type: \"loadRequested\" }\n  | { type: \"loadStarted\" }\n  | { type: \"loadFinished\" }\n  | { type: \"historyMutationAdvanced\" }\n  | { type: \"stateSnapshot\"; snapshot: ReadonlyJSONValue }\n  | { type: \"resetState\" }\n  | { type: \"runConfigRecorded\"; runConfig: RunConfig }\n  | { type: \"resumeDeferred\"; owner: AbortController; messageId: string }\n  | { type: \"resumeCleared\" }\n  | { type: \"a2uiResumeDeferred\"; owner: AbortController }\n  | { type: \"a2uiResumeCleared\" }\n  | { type: \"a2uiActionDeferred\"; action: Record<string, unknown> }\n  | { type: \"a2uiActionCleared\" }\n  | { type: \"a2uiContinuationsCleared\" }\n  | { type: \"resetThread\" };\n\nexport const createAgUiThreadState = (): AgUiThreadState => ({\n  isRunning: false,\n  isLoading: false,\n  loadRequested: false,\n  historyMutationGeneration: 0,\n  stateSnapshot: undefined,\n  lastRunConfig: undefined,\n  pendingResume: null,\n  pendingA2uiResumeOwner: null,\n  pendingA2uiAction: undefined,\n});\n\nexport const reduceAgUiThreadState = (\n  state: AgUiThreadState,\n  action: AgUiThreadAction,\n): AgUiThreadState => {\n  switch (action.type) {\n    case \"runStarted\":\n      return state.isRunning ? state : { ...state, isRunning: true };\n    case \"runFinished\":\n      return !state.isRunning ? state : { ...state, isRunning: false };\n    case \"loadRequested\":\n      return state.loadRequested ? state : { ...state, loadRequested: true };\n    case \"loadStarted\":\n      return state.isLoading ? state : { ...state, isLoading: true };\n    case \"loadFinished\":\n      return !state.isLoading ? state : { ...state, isLoading: false };\n    case \"historyMutationAdvanced\":\n      return {\n        ...state,\n        historyMutationGeneration: state.historyMutationGeneration + 1,\n      };\n    case \"stateSnapshot\":\n      return state.stateSnapshot === action.snapshot\n        ? state\n        : { ...state, stateSnapshot: action.snapshot };\n    case \"resetState\":\n      return state.stateSnapshot === undefined\n        ? state\n        : { ...state, stateSnapshot: undefined };\n    case \"runConfigRecorded\":\n      return state.lastRunConfig === action.runConfig\n        ? state\n        : { ...state, lastRunConfig: action.runConfig };\n    case \"resumeDeferred\":\n      return state.pendingResume?.owner === action.owner &&\n        state.pendingResume.messageIds.includes(action.messageId)\n        ? state\n        : {\n            ...state,\n            pendingResume: {\n              owner: action.owner,\n              messageIds:\n                state.pendingResume?.owner === action.owner\n                  ? [...state.pendingResume.messageIds, action.messageId]\n                  : [action.messageId],\n            },\n          };\n    case \"resumeCleared\":\n      return state.pendingResume === null\n        ? state\n        : { ...state, pendingResume: null };\n    case \"a2uiResumeDeferred\":\n      return state.pendingA2uiResumeOwner === action.owner\n        ? state\n        : { ...state, pendingA2uiResumeOwner: action.owner };\n    case \"a2uiResumeCleared\":\n      return state.pendingA2uiResumeOwner === null\n        ? state\n        : { ...state, pendingA2uiResumeOwner: null };\n    case \"a2uiActionDeferred\":\n      return state.pendingA2uiAction === action.action\n        ? state\n        : { ...state, pendingA2uiAction: action.action };\n    case \"a2uiActionCleared\":\n      return state.pendingA2uiAction === undefined\n        ? state\n        : { ...state, pendingA2uiAction: undefined };\n    case \"a2uiContinuationsCleared\":\n      return state.pendingA2uiResumeOwner === null &&\n        state.pendingA2uiAction === undefined\n        ? state\n        : {\n            ...state,\n            pendingA2uiResumeOwner: null,\n            pendingA2uiAction: undefined,\n          };\n    case \"resetThread\":\n      return state.stateSnapshot === undefined &&\n        state.pendingResume === null &&\n        state.pendingA2uiResumeOwner === null &&\n        state.pendingA2uiAction === undefined\n        ? state\n        : {\n            ...state,\n            stateSnapshot: undefined,\n            pendingResume: null,\n            pendingA2uiResumeOwner: null,\n            pendingA2uiAction: undefined,\n          };\n  }\n};\n"],"mappings":";AAqCA,MAAa,+BAAgD;CAC3D,WAAW;CACX,WAAW;CACX,eAAe;CACf,2BAA2B;CAC3B,eAAe,KAAA;CACf,eAAe,KAAA;CACf,eAAe;CACf,wBAAwB;CACxB,mBAAmB,KAAA;AACrB;AAEA,MAAa,yBACX,OACA,WACoB;CACpB,QAAQ,OAAO,MAAf;EACE,KAAK,cACH,OAAO,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAK;EAC/D,KAAK,eACH,OAAO,CAAC,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAM;EACjE,KAAK,iBACH,OAAO,MAAM,gBAAgB,QAAQ;GAAE,GAAG;GAAO,eAAe;EAAK;EACvE,KAAK,eACH,OAAO,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAK;EAC/D,KAAK,gBACH,OAAO,CAAC,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAM;EACjE,KAAK,2BACH,OAAO;GACL,GAAG;GACH,2BAA2B,MAAM,4BAA4B;EAC/D;EACF,KAAK,iBACH,OAAO,MAAM,kBAAkB,OAAO,WAClC,QACA;GAAE,GAAG;GAAO,eAAe,OAAO;EAAS;EACjD,KAAK,cACH,OAAO,MAAM,kBAAkB,KAAA,IAC3B,QACA;GAAE,GAAG;GAAO,eAAe,KAAA;EAAU;EAC3C,KAAK,qBACH,OAAO,MAAM,kBAAkB,OAAO,YAClC,QACA;GAAE,GAAG;GAAO,eAAe,OAAO;EAAU;EAClD,KAAK,kBACH,OAAO,MAAM,eAAe,UAAU,OAAO,SAC3C,MAAM,cAAc,WAAW,SAAS,OAAO,SAAS,IACtD,QACA;GACE,GAAG;GACH,eAAe;IACb,OAAO,OAAO;IACd,YACE,MAAM,eAAe,UAAU,OAAO,QAClC,CAAC,GAAG,MAAM,cAAc,YAAY,OAAO,SAAS,IACpD,CAAC,OAAO,SAAS;GACzB;EACF;EACN,KAAK,iBACH,OAAO,MAAM,kBAAkB,OAC3B,QACA;GAAE,GAAG;GAAO,eAAe;EAAK;EACtC,KAAK,sBACH,OAAO,MAAM,2BAA2B,OAAO,QAC3C,QACA;GAAE,GAAG;GAAO,wBAAwB,OAAO;EAAM;EACvD,KAAK,qBACH,OAAO,MAAM,2BAA2B,OACpC,QACA;GAAE,GAAG;GAAO,wBAAwB;EAAK;EAC/C,KAAK,sBACH,OAAO,MAAM,sBAAsB,OAAO,SACtC,QACA;GAAE,GAAG;GAAO,mBAAmB,OAAO;EAAO;EACnD,KAAK,qBACH,OAAO,MAAM,sBAAsB,KAAA,IAC/B,QACA;GAAE,GAAG;GAAO,mBAAmB,KAAA;EAAU;EAC/C,KAAK,4BACH,OAAO,MAAM,2BAA2B,QACtC,MAAM,sBAAsB,KAAA,IAC1B,QACA;GACE,GAAG;GACH,wBAAwB;GACxB,mBAAmB,KAAA;EACrB;EACN,KAAK,eACH,OAAO,MAAM,kBAAkB,KAAA,KAC7B,MAAM,kBAAkB,QACxB,MAAM,2BAA2B,QACjC,MAAM,sBAAsB,KAAA,IAC1B,QACA;GACE,GAAG;GACH,eAAe,KAAA;GACf,eAAe;GACf,wBAAwB;GACxB,mBAAmB,KAAA;EACrB;CACR;AACF"}

package/dist/runtime/event-parser.js

@@ -1,6 +1,6 @@
 import { parseMcpToolCallResult } from "./mcp-tool-result.js";
 import { withRawResponseSchema } from "./interrupt-internals.js";
-//#region src/runtime/event-parser.ts
+// src/runtime/event-parser.ts
 const isString = (value) => typeof value === "string";
 const isNonEmptyString = (value) => isString(value) && value.length > 0;
 const withOptional = (base, optionals) => {
@@ -362,7 +362,6 @@ const parseAgUiEvent = (event, options) => {
 		}, { source: isString(payload.type) ? payload.type : void 0 });
 	}
 };
-//#endregion
 export { parseAgUiEvent };
 
 //# sourceMappingURL=event-parser.js.map

package/dist/runtime/interrupt-internals.js

@@ -1,4 +1,4 @@
-//#region src/runtime/interrupt-internals.ts
+// src/runtime/interrupt-internals.ts
 /**
 * Where a received `responseSchema` travels when it is not an object. JSON
 * Schema allows a boolean, and `false` rejects every payload, so a schema
@@ -19,7 +19,6 @@ const withoutRawResponseSchema = (interrupt) => {
 	const { [RAW_RESPONSE_SCHEMA]: _raw, ...rest } = interrupt;
 	return rest;
 };
-//#endregion
 export { readRawResponseSchema, withRawResponseSchema, withoutRawResponseSchema };
 
 //# sourceMappingURL=interrupt-internals.js.map

package/dist/runtime/logger.js

@@ -1,9 +1,8 @@
-//#region src/runtime/logger.ts
+// src/runtime/logger.ts
 const makeLogger = (l) => ({
 	debug: (...a) => l?.debug?.(...a),
 	error: (...a) => l?.error?.(...a)
 });
-//#endregion
 export { makeLogger };
 
 //# sourceMappingURL=logger.js.map

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

@@ -1,7 +1,7 @@
 import { isMcpAppUri } from "@assistant-ui/core";
 import { isRecord } from "@assistant-ui/core/internal";
 import { contentToText } from "@ag-ui/client";
-//#region src/runtime/mcp-tool-result.ts
+// src/runtime/mcp-tool-result.ts
 const toolContentText = (content) => contentToText(Array.isArray(content) ? content.filter(isRecord) : content);
 const parseMcpToolCallResult = (source, content) => {
 	const structuredContent = isRecord(source.structuredContent) ? source.structuredContent : void 0;
@@ -26,7 +26,6 @@ const readMcpAppResourceUri = (meta) => {
 	const flat = meta["ui/resourceUri"];
 	return typeof flat === "string" && isMcpAppUri(flat) ? flat : void 0;
 };
-//#endregion
 export { parseMcpToolCallResult, readMcpAppResourceUri, toolContentText };
 
 //# sourceMappingURL=mcp-tool-result.js.map

package/dist/runtime/types.js

@@ -1,8 +1,7 @@
-//#region src/runtime/types.ts
+// src/runtime/types.ts
 const AG_UI_METADATA_NAMESPACE = "agui";
 const MCP_APPS_ACTIVITY_TYPE = "mcp-apps";
 const A2UI_SURFACE_ACTIVITY_TYPE = "a2ui-surface";
-//#endregion
 export { A2UI_SURFACE_ACTIVITY_TYPE, AG_UI_METADATA_NAMESPACE, MCP_APPS_ACTIVITY_TYPE };
 
 //# sourceMappingURL=types.js.map

package/dist/useAgUiRuntime.js

@@ -6,7 +6,7 @@ import { useExternalStoreRuntime, useExternalStoreSharedOptions, useRuntimeAdapt
 import { useEffect, useMemo, useRef, useState, useSyncExternalStore } from "react";
 import { useIsomorphicLayoutEffect, useReplaySafeEffect } from "@assistant-ui/store/internal";
 import { createMessageQueue } from "@assistant-ui/core";
-//#region src/useAgUiRuntime.ts
+// src/useAgUiRuntime.ts
 const EMPTY_INTERRUPTS = [];
 const EMPTY_QUEUE_ITEMS = Object.freeze([]);
 const subscribeNoop = () => () => {};
@@ -262,7 +262,6 @@ function useAgUiRuntime(options) {
 	}, [core]);
 	return runtime;
 }
-//#endregion
 export { useAgUiRuntime };
 
 //# sourceMappingURL=useAgUiRuntime.js.map