@assistant-ui/core 0.2.12 → 0.2.15

package/dist/react/primitives/message/MessageParts.js

@@ -4,8 +4,9 @@ import { TextMessagePartProvider } from "../../providers/TextMessagePartProvider
 import { ChainOfThoughtByIndicesProvider } from "../../providers/ChainOfThoughtByIndicesProvider.js";
 import { getMessageQuote } from "../../utils/getMessageQuote.js";
 import { GenerativeUIRender } from "../generativeUI/GenerativeUI.js";
-import { memo, useMemo } from "@assistant-ui/tap/react-shim";
+import { memo } from "@assistant-ui/tap/react-shim";
 import { RenderChildrenWithAccessor, useAui, useAuiState } from "@assistant-ui/store";
+import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { Fragment as Fragment$1, jsx, jsxs } from "react/jsx-runtime";
 import { useShallow } from "zustand/shallow";
 //#region src/react/primitives/message/MessageParts.tsx
@@ -42,8 +43,10 @@ const createGroupState = (groupType) => {
 * Groups consecutive tool-call and reasoning message parts into ranges.
 * Always groups tool calls and reasoning parts, even if there's only one.
 * When useChainOfThought is true, groups tool-call and reasoning parts together.
+* `partIds[i]` optionally carries a stable identity for part `i`; group
+* ranges derive an `idKey` from their first part's id (first claim wins).
 */
-const groupMessageParts = (messageTypes, useChainOfThought) => {
+const groupMessageParts = (messageTypes, useChainOfThought, partIds) => {
 	const ranges = [];
 	if (useChainOfThought) {
 		const chainOfThoughtGroup = createGroupState("chainOfThoughtGroup");
@@ -82,29 +85,131 @@ const groupMessageParts = (messageTypes, useChainOfThought) => {
 		toolGroup.finalize(messageTypes.length - 1, ranges);
 		reasoningGroup.finalize(messageTypes.length - 1, ranges);
 	}
+	if (partIds) {
+		const claimed = /* @__PURE__ */ new Set();
+		for (const range of ranges) {
+			if (range.type === "single") continue;
+			const id = partIds[range.startIndex];
+			if (id !== void 0 && !claimed.has(id)) {
+				claimed.add(id);
+				range.idKey = `id:${id}`;
+			}
+		}
+	}
 	return ranges;
 };
 const useMessagePartsGroups = (useChainOfThought) => {
-	const messageTypes = useAuiState(useShallow((s) => s.message.parts.map((c) => c.type)));
-	return useMemo(() => {
-		if (messageTypes.length === 0) return [];
-		return groupMessageParts(messageTypes, useChainOfThought);
-	}, [messageTypes, useChainOfThought]);
+	const $ = c(10);
+	const messageTypes = useAuiState(useShallow(_temp2));
+	const partIds = useAuiState(useShallow(_temp4));
+	let t0;
+	bb0: {
+		if (messageTypes.length === 0) {
+			let t1;
+			if ($[0] === Symbol.for("react.memo_cache_sentinel")) {
+				t1 = [];
+				$[0] = t1;
+			} else t1 = $[0];
+			let t2;
+			if ($[1] !== partIds) {
+				t2 = {
+					ranges: t1,
+					partIds
+				};
+				$[1] = partIds;
+				$[2] = t2;
+			} else t2 = $[2];
+			t0 = t2;
+			break bb0;
+		}
+		let t1;
+		if ($[3] !== messageTypes || $[4] !== partIds || $[5] !== useChainOfThought) {
+			t1 = groupMessageParts(messageTypes, useChainOfThought, partIds);
+			$[3] = messageTypes;
+			$[4] = partIds;
+			$[5] = useChainOfThought;
+			$[6] = t1;
+		} else t1 = $[6];
+		let t2;
+		if ($[7] !== partIds || $[8] !== t1) {
+			t2 = {
+				ranges: t1,
+				partIds
+			};
+			$[7] = partIds;
+			$[8] = t1;
+			$[9] = t2;
+		} else t2 = $[9];
+		t0 = t2;
+	}
+	return t0;
 };
-const ToolUIDisplay = ({ Fallback, ...props }) => {
-	const Render = useAuiState((s) => s.tools.toolUIs[props.toolName]?.[0]?.render ?? Fallback);
+const ToolUIDisplay = (t0) => {
+	const $ = c(9);
+	let Fallback;
+	let props;
+	if ($[0] !== t0) {
+		({Fallback, ...props} = t0);
+		$[0] = t0;
+		$[1] = Fallback;
+		$[2] = props;
+	} else {
+		Fallback = $[1];
+		props = $[2];
+	}
+	let t1;
+	if ($[3] !== Fallback || $[4] !== props.toolName) {
+		t1 = (s) => s.tools.toolUIs[props.toolName]?.[0]?.render ?? Fallback;
+		$[3] = Fallback;
+		$[4] = props.toolName;
+		$[5] = t1;
+	} else t1 = $[5];
+	const Render = useAuiState(t1);
 	if (!Render) return null;
-	return /* @__PURE__ */ jsx(Render, { ...props });
+	let t2;
+	if ($[6] !== Render || $[7] !== props) {
+		t2 = /* @__PURE__ */ jsx(Render, { ...props });
+		$[6] = Render;
+		$[7] = props;
+		$[8] = t2;
+	} else t2 = $[8];
+	return t2;
 };
 const getDataRenderer = (dataRenderers, name, inlineFallback) => {
 	const named = dataRenderers.renderers[name]?.[0];
 	if (named) return named;
 	return dataRenderers.fallbacks[0] ?? inlineFallback;
 };
-const DataUIDisplay = ({ Fallback, ...props }) => {
-	const Render = useAuiState((s) => getDataRenderer(s.dataRenderers, props.name, Fallback));
+const DataUIDisplay = (t0) => {
+	const $ = c(9);
+	let Fallback;
+	let props;
+	if ($[0] !== t0) {
+		({Fallback, ...props} = t0);
+		$[0] = t0;
+		$[1] = Fallback;
+		$[2] = props;
+	} else {
+		Fallback = $[1];
+		props = $[2];
+	}
+	let t1;
+	if ($[3] !== Fallback || $[4] !== props.name) {
+		t1 = (s) => getDataRenderer(s.dataRenderers, props.name, Fallback);
+		$[3] = Fallback;
+		$[4] = props.name;
+		$[5] = t1;
+	} else t1 = $[5];
+	const Render = useAuiState(t1);
 	if (!Render) return null;
-	return /* @__PURE__ */ jsx(Render, { ...props });
+	let t2;
+	if ($[6] !== Render || $[7] !== props) {
+		t2 = /* @__PURE__ */ jsx(Render, { ...props });
+		$[6] = Render;
+		$[7] = props;
+		$[8] = t2;
+	} else t2 = $[8];
+	return t2;
 };
 /**
 * Platform-agnostic no-op default components.
@@ -120,54 +225,187 @@ const defaultComponents = {
 	ToolGroup: ({ children }) => children,
 	ReasoningGroup: ({ children }) => children
 };
-const MessagePartComponent = ({ components: { Text = defaultComponents.Text, Reasoning = defaultComponents.Reasoning, Image = defaultComponents.Image, Source = defaultComponents.Source, File = defaultComponents.File, Unstable_Audio: Audio = defaultComponents.Unstable_Audio, tools = {}, data, generativeUI } = {} }) => {
+const MessagePartComponent = (t0) => {
+	const $ = c(47);
+	const { components: t1 } = t0;
+	let t2;
+	if ($[0] !== t1) {
+		t2 = t1 === void 0 ? {} : t1;
+		$[0] = t1;
+		$[1] = t2;
+	} else t2 = $[1];
+	const { Text: t3, Reasoning: t4, Image: t5, Source: t6, File: t7, Unstable_Audio: t8, tools: t9, data, generativeUI } = t2;
+	const Text = t3 === void 0 ? defaultComponents.Text : t3;
+	const Reasoning = t4 === void 0 ? defaultComponents.Reasoning : t4;
+	const Image = t5 === void 0 ? defaultComponents.Image : t5;
+	const Source = t6 === void 0 ? defaultComponents.Source : t6;
+	const File = t7 === void 0 ? defaultComponents.File : t7;
+	const Audio = t8 === void 0 ? defaultComponents.Unstable_Audio : t8;
+	let t10;
+	if ($[2] !== t9) {
+		t10 = t9 === void 0 ? {} : t9;
+		$[2] = t9;
+		$[3] = t10;
+	} else t10 = $[3];
+	const tools = t10;
 	const aui = useAui();
-	const part = useAuiState((s) => s.part);
+	const part = useAuiState(_temp5);
 	const type = part.type;
 	if (type === "tool-call") {
-		const addResult = aui.part().addToolResult;
-		const resume = aui.part().resumeToolCall;
-		const respondToApproval = aui.part().respondToToolApproval;
-		if ("Override" in tools) return /* @__PURE__ */ jsx(tools.Override, {
-			...part,
-			addResult,
-			resume,
-			respondToApproval
-		});
+		let t11;
+		if ($[4] !== aui) {
+			t11 = aui.part();
+			$[4] = aui;
+			$[5] = t11;
+		} else t11 = $[5];
+		const addResult = t11.addToolResult;
+		let t12;
+		if ($[6] !== aui) {
+			t12 = aui.part();
+			$[6] = aui;
+			$[7] = t12;
+		} else t12 = $[7];
+		const resume = t12.resumeToolCall;
+		let t13;
+		if ($[8] !== aui) {
+			t13 = aui.part();
+			$[8] = aui;
+			$[9] = t13;
+		} else t13 = $[9];
+		const respondToApproval = t13.respondToToolApproval;
+		if ("Override" in tools) {
+			let t14;
+			if ($[10] !== addResult || $[11] !== part || $[12] !== respondToApproval || $[13] !== resume || $[14] !== tools.Override) {
+				t14 = /* @__PURE__ */ jsx(tools.Override, {
+					...part,
+					addResult,
+					resume,
+					respondToApproval
+				});
+				$[10] = addResult;
+				$[11] = part;
+				$[12] = respondToApproval;
+				$[13] = resume;
+				$[14] = tools.Override;
+				$[15] = t14;
+			} else t14 = $[15];
+			return t14;
+		}
 		const Tool = tools.by_name?.[part.toolName] ?? tools.Fallback;
-		return /* @__PURE__ */ jsx(ToolUIDisplay, {
-			...part,
-			Fallback: Tool,
-			addResult,
-			resume,
-			respondToApproval
-		});
+		let t14;
+		if ($[16] !== Tool || $[17] !== addResult || $[18] !== part || $[19] !== respondToApproval || $[20] !== resume) {
+			t14 = /* @__PURE__ */ jsx(ToolUIDisplay, {
+				...part,
+				Fallback: Tool,
+				addResult,
+				resume,
+				respondToApproval
+			});
+			$[16] = Tool;
+			$[17] = addResult;
+			$[18] = part;
+			$[19] = respondToApproval;
+			$[20] = resume;
+			$[21] = t14;
+		} else t14 = $[21];
+		return t14;
 	}
 	if (part.status?.type === "requires-action") throw new Error("Encountered unexpected requires-action status");
 	switch (type) {
-		case "text": return /* @__PURE__ */ jsx(Text, { ...part });
-		case "reasoning": return /* @__PURE__ */ jsx(Reasoning, { ...part });
-		case "source": return /* @__PURE__ */ jsx(Source, { ...part });
-		case "image": return /* @__PURE__ */ jsx(Image, { ...part });
-		case "file": return /* @__PURE__ */ jsx(File, { ...part });
-		case "audio": return /* @__PURE__ */ jsx(Audio, { ...part });
+		case "text": {
+			let t11;
+			if ($[22] !== Text || $[23] !== part) {
+				t11 = /* @__PURE__ */ jsx(Text, { ...part });
+				$[22] = Text;
+				$[23] = part;
+				$[24] = t11;
+			} else t11 = $[24];
+			return t11;
+		}
+		case "reasoning": {
+			let t11;
+			if ($[25] !== Reasoning || $[26] !== part) {
+				t11 = /* @__PURE__ */ jsx(Reasoning, { ...part });
+				$[25] = Reasoning;
+				$[26] = part;
+				$[27] = t11;
+			} else t11 = $[27];
+			return t11;
+		}
+		case "source": {
+			let t11;
+			if ($[28] !== Source || $[29] !== part) {
+				t11 = /* @__PURE__ */ jsx(Source, { ...part });
+				$[28] = Source;
+				$[29] = part;
+				$[30] = t11;
+			} else t11 = $[30];
+			return t11;
+		}
+		case "image": {
+			let t11;
+			if ($[31] !== Image || $[32] !== part) {
+				t11 = /* @__PURE__ */ jsx(Image, { ...part });
+				$[31] = Image;
+				$[32] = part;
+				$[33] = t11;
+			} else t11 = $[33];
+			return t11;
+		}
+		case "file": {
+			let t11;
+			if ($[34] !== File || $[35] !== part) {
+				t11 = /* @__PURE__ */ jsx(File, { ...part });
+				$[34] = File;
+				$[35] = part;
+				$[36] = t11;
+			} else t11 = $[36];
+			return t11;
+		}
+		case "audio": {
+			let t11;
+			if ($[37] !== Audio || $[38] !== part) {
+				t11 = /* @__PURE__ */ jsx(Audio, { ...part });
+				$[37] = Audio;
+				$[38] = part;
+				$[39] = t11;
+			} else t11 = $[39];
+			return t11;
+		}
 		case "data": {
 			const Data = data?.by_name?.[part.name] ?? data?.Fallback;
-			return /* @__PURE__ */ jsx(DataUIDisplay, {
-				...part,
-				Fallback: Data
-			});
+			let t11;
+			if ($[40] !== Data || $[41] !== part) {
+				t11 = /* @__PURE__ */ jsx(DataUIDisplay, {
+					...part,
+					Fallback: Data
+				});
+				$[40] = Data;
+				$[41] = part;
+				$[42] = t11;
+			} else t11 = $[42];
+			return t11;
 		}
-		case "generative-ui":
+		case "generative-ui": {
 			if (!generativeUI?.components) {
 				if (typeof process !== "undefined" && process.env?.NODE_ENV !== "production") console.warn("MessagePrimitive.Parts received a generative-ui part but no `components.generativeUI.components` allowlist was provided. Pass an allowlist or render with <MessagePrimitive.GenerativeUI />.");
 				return null;
 			}
-			return /* @__PURE__ */ jsx(GenerativeUIRender, {
-				spec: part.spec,
-				components: generativeUI.components,
-				Fallback: generativeUI.Fallback
-			});
+			const t11 = part;
+			let t12;
+			if ($[43] !== generativeUI.Fallback || $[44] !== generativeUI.components || $[45] !== t11.spec) {
+				t12 = /* @__PURE__ */ jsx(GenerativeUIRender, {
+					spec: t11.spec,
+					components: generativeUI.components,
+					Fallback: generativeUI.Fallback
+				});
+				$[43] = generativeUI.Fallback;
+				$[44] = generativeUI.components;
+				$[45] = t11.spec;
+				$[46] = t12;
+			} else t12 = $[46];
+			return t12;
+		}
 		default:
 			console.warn(`Unknown message part type: ${type}`);
 			return null;
@@ -176,53 +414,128 @@ const MessagePartComponent = ({ components: { Text = defaultComponents.Text, Rea
 /**
 * Renders a single message part at the specified index.
 */
-const MessagePrimitivePartByIndex = memo(({ index, components }) => {
-	return /* @__PURE__ */ jsx(PartByIndexProvider, {
-		index,
-		children: /* @__PURE__ */ jsx(MessagePartComponent, { components })
-	});
+const MessagePrimitivePartByIndex = memo((t0) => {
+	const $ = c(5);
+	const { index, components } = t0;
+	let t1;
+	if ($[0] !== components) {
+		t1 = /* @__PURE__ */ jsx(MessagePartComponent, { components });
+		$[0] = components;
+		$[1] = t1;
+	} else t1 = $[1];
+	let t2;
+	if ($[2] !== index || $[3] !== t1) {
+		t2 = /* @__PURE__ */ jsx(PartByIndexProvider, {
+			index,
+			children: t1
+		});
+		$[2] = index;
+		$[3] = t1;
+		$[4] = t2;
+	} else t2 = $[4];
+	return t2;
 }, (prev, next) => prev.index === next.index && prev.components?.Text === next.components?.Text && prev.components?.Reasoning === next.components?.Reasoning && prev.components?.Source === next.components?.Source && prev.components?.Image === next.components?.Image && prev.components?.File === next.components?.File && prev.components?.Unstable_Audio === next.components?.Unstable_Audio && prev.components?.tools === next.components?.tools && prev.components?.data === next.components?.data && prev.components?.generativeUI === next.components?.generativeUI && prev.components?.ToolGroup === next.components?.ToolGroup && prev.components?.ReasoningGroup === next.components?.ReasoningGroup);
 MessagePrimitivePartByIndex.displayName = "MessagePrimitive.PartByIndex";
-const EmptyPartFallback = ({ status, component: Component }) => {
-	return /* @__PURE__ */ jsx(TextMessagePartProvider, {
-		text: "",
-		isRunning: status.type === "running",
-		children: /* @__PURE__ */ jsx(Component, {
+const EmptyPartFallback = (t0) => {
+	const $ = c(6);
+	const { status, component: Component } = t0;
+	const t1 = status.type === "running";
+	let t2;
+	if ($[0] !== Component || $[1] !== status) {
+		t2 = /* @__PURE__ */ jsx(Component, {
 			type: "text",
 			text: "",
 			status
-		})
-	});
+		});
+		$[0] = Component;
+		$[1] = status;
+		$[2] = t2;
+	} else t2 = $[2];
+	let t3;
+	if ($[3] !== t1 || $[4] !== t2) {
+		t3 = /* @__PURE__ */ jsx(TextMessagePartProvider, {
+			text: "",
+			isRunning: t1,
+			children: t2
+		});
+		$[3] = t1;
+		$[4] = t2;
+		$[5] = t3;
+	} else t3 = $[5];
+	return t3;
 };
 const COMPLETE_STATUS = Object.freeze({ type: "complete" });
 const RUNNING_STATUS = Object.freeze({ type: "running" });
-const EmptyPartsImpl = ({ components }) => {
-	const status = useAuiState((s) => s.message.status ?? COMPLETE_STATUS);
-	if (components?.Empty) return /* @__PURE__ */ jsx(components.Empty, { status });
+const EmptyPartsImpl = (t0) => {
+	const $ = c(6);
+	const { components } = t0;
+	const status = useAuiState(_temp6);
+	if (components?.Empty) {
+		let t1;
+		if ($[0] !== components.Empty || $[1] !== status) {
+			t1 = /* @__PURE__ */ jsx(components.Empty, { status });
+			$[0] = components.Empty;
+			$[1] = status;
+			$[2] = t1;
+		} else t1 = $[2];
+		return t1;
+	}
 	if (status.type !== "running") return null;
-	return /* @__PURE__ */ jsx(EmptyPartFallback, {
-		status,
-		component: components?.Text ?? defaultComponents.Text
-	});
+	const t1 = components?.Text ?? defaultComponents.Text;
+	let t2;
+	if ($[3] !== status || $[4] !== t1) {
+		t2 = /* @__PURE__ */ jsx(EmptyPartFallback, {
+			status,
+			component: t1
+		});
+		$[3] = status;
+		$[4] = t1;
+		$[5] = t2;
+	} else t2 = $[5];
+	return t2;
 };
 const EmptyParts = memo(EmptyPartsImpl, (prev, next) => prev.components?.Empty === next.components?.Empty && prev.components?.Text === next.components?.Text);
-const ConditionalEmptyImpl = ({ components, enabled }) => {
-	if (!useAuiState((s) => {
-		if (!enabled) return false;
-		if (s.message.parts.length === 0) return false;
-		const lastPart = s.message.parts[s.message.parts.length - 1];
-		return lastPart?.type !== "text" && lastPart?.type !== "reasoning";
-	})) return null;
-	return /* @__PURE__ */ jsx(EmptyParts, { components });
+const ConditionalEmptyImpl = (t0) => {
+	const $ = c(4);
+	const { components, enabled } = t0;
+	let t1;
+	if ($[0] !== enabled) {
+		t1 = (s) => {
+			if (!enabled) return false;
+			if (s.message.parts.length === 0) return false;
+			const lastPart = s.message.parts[s.message.parts.length - 1];
+			return lastPart?.type !== "text" && lastPart?.type !== "reasoning";
+		};
+		$[0] = enabled;
+		$[1] = t1;
+	} else t1 = $[1];
+	if (!useAuiState(t1)) return null;
+	let t2;
+	if ($[2] !== components) {
+		t2 = /* @__PURE__ */ jsx(EmptyParts, { components });
+		$[2] = components;
+		$[3] = t2;
+	} else t2 = $[3];
+	return t2;
 };
 const ConditionalEmpty = memo(ConditionalEmptyImpl, (prev, next) => prev.enabled === next.enabled && prev.components?.Empty === next.components?.Empty && prev.components?.Text === next.components?.Text);
-const QuoteRendererImpl = ({ Quote }) => {
+const QuoteRendererImpl = (t0) => {
+	const $ = c(4);
+	const { Quote } = t0;
 	const quoteInfo = useAuiState(getMessageQuote);
 	if (!quoteInfo) return null;
-	return /* @__PURE__ */ jsx(Quote, {
-		text: quoteInfo.text,
-		messageId: quoteInfo.messageId
-	});
+	let t1;
+	if ($[0] !== Quote || $[1] !== quoteInfo.messageId || $[2] !== quoteInfo.text) {
+		t1 = /* @__PURE__ */ jsx(Quote, {
+			text: quoteInfo.text,
+			messageId: quoteInfo.messageId
+		});
+		$[0] = Quote;
+		$[1] = quoteInfo.messageId;
+		$[2] = quoteInfo.text;
+		$[3] = t1;
+	} else t1 = $[3];
+	return t1;
 };
 const QuoteRenderer = memo(QuoteRendererImpl);
 function resolveToolRender(toolsState, part) {
@@ -236,26 +549,66 @@ function resolveToolRender(toolsState, part) {
 * current part context. Reads tool registry and part state from context.
 */
 const RegisteredToolUI = () => {
+	const $ = c(12);
 	const aui = useAui();
-	const part = useAuiState((s) => s.part);
-	const Render = useAuiState((s) => s.part.type === "tool-call" ? resolveToolRender(s.tools, s.part) : null);
+	const part = useAuiState(_temp7);
+	const Render = useAuiState(_temp8);
 	if (!Render || part.type !== "tool-call") return null;
-	return /* @__PURE__ */ jsx(Render, {
-		...part,
-		addResult: aui.part().addToolResult,
-		resume: aui.part().resumeToolCall,
-		respondToApproval: aui.part().respondToToolApproval
-	});
+	let t0;
+	if ($[0] !== aui) {
+		t0 = aui.part();
+		$[0] = aui;
+		$[1] = t0;
+	} else t0 = $[1];
+	const t1 = t0.addToolResult;
+	let t2;
+	if ($[2] !== aui) {
+		t2 = aui.part();
+		$[2] = aui;
+		$[3] = t2;
+	} else t2 = $[3];
+	const t3 = t2.resumeToolCall;
+	let t4;
+	if ($[4] !== aui) {
+		t4 = aui.part();
+		$[4] = aui;
+		$[5] = t4;
+	} else t4 = $[5];
+	let t5;
+	if ($[6] !== Render || $[7] !== part || $[8] !== t0.addToolResult || $[9] !== t2.resumeToolCall || $[10] !== t4.respondToToolApproval) {
+		t5 = /* @__PURE__ */ jsx(Render, {
+			...part,
+			addResult: t1,
+			resume: t3,
+			respondToApproval: t4.respondToToolApproval
+		});
+		$[6] = Render;
+		$[7] = part;
+		$[8] = t0.addToolResult;
+		$[9] = t2.resumeToolCall;
+		$[10] = t4.respondToToolApproval;
+		$[11] = t5;
+	} else t5 = $[11];
+	return t5;
 };
 /**
 * Stable propless component that renders the registered data renderer UI
 * for the current part context.
 */
 const RegisteredDataRendererUI = () => {
-	const part = useAuiState((s) => s.part);
-	const Render = useAuiState((s) => s.part.type === "data" ? getDataRenderer(s.dataRenderers, s.part.name, void 0) ?? null : null);
+	const $ = c(3);
+	const part = useAuiState(_temp9);
+	const Render = useAuiState(_temp0);
 	if (!Render || part.type !== "data") return null;
-	return /* @__PURE__ */ jsx(Render, { ...part });
+	const t0 = part;
+	let t1;
+	if ($[0] !== Render || $[1] !== t0) {
+		t1 = /* @__PURE__ */ jsx(Render, { ...t0 });
+		$[0] = Render;
+		$[1] = t0;
+		$[2] = t1;
+	} else t1 = $[2];
+	return t1;
 };
 /**
 * Fallback component rendered when the children render function returns null.
@@ -271,9 +624,24 @@ const RegisteredDataRendererUI = () => {
 * To explicitly render nothing (suppressing registered UIs), return <></>.
 */
 const DefaultPartFallback = () => {
-	const partType = useAuiState((s) => s.part.type);
-	if (partType === "tool-call") return /* @__PURE__ */ jsx(RegisteredToolUI, {});
-	if (partType === "data") return /* @__PURE__ */ jsx(RegisteredDataRendererUI, {});
+	const $ = c(2);
+	const partType = useAuiState(_temp1);
+	if (partType === "tool-call") {
+		let t0;
+		if ($[0] === Symbol.for("react.memo_cache_sentinel")) {
+			t0 = /* @__PURE__ */ jsx(RegisteredToolUI, {});
+			$[0] = t0;
+		} else t0 = $[0];
+		return t0;
+	}
+	if (partType === "data") {
+		let t0;
+		if ($[1] === Symbol.for("react.memo_cache_sentinel")) {
+			t0 = /* @__PURE__ */ jsx(RegisteredDataRendererUI, {});
+			$[1] = t0;
+		} else t0 = $[1];
+		return t0;
+	}
 	return null;
 };
 const EMPTY_RUNNING_TEXT_PART = Object.freeze({
@@ -295,7 +663,7 @@ const MessagePartChildren = ({ index, children }) => {
 	return /* @__PURE__ */ jsx(PartByIndexProvider, {
 		index,
 		children: /* @__PURE__ */ jsx(RenderChildrenWithAccessor, {
-			getItemState: (aui) => aui.message().part({ index }).getState(),
+			getItemState: (aui_0) => aui_0.message().part({ index }).getState(),
 			children: (getItem) => children({ get part() {
 				const state = getItem();
 				if (state.type === "tool-call") {
@@ -321,22 +689,49 @@ const MessagePartChildren = ({ index, children }) => {
 		})
 	});
 };
-const MessagePrimitivePartsInner = ({ children }) => {
-	const contentLength = useAuiState((s) => s.message.parts.length);
-	const isRunning = useAuiState((s) => (s.message.status?.type ?? "complete") === "running");
+const MessagePrimitivePartsInner = (t0) => {
+	const $ = c(9);
+	const { children } = t0;
+	const contentLength = useAuiState(_temp10);
+	const isRunning = useAuiState(_temp11);
 	const isEmptyRunning = contentLength === 0 && isRunning;
 	if (contentLength === 0) {
 		if (!isEmptyRunning) return null;
-		return /* @__PURE__ */ jsx(TextMessagePartProvider, {
-			text: "",
-			isRunning: true,
-			children: children({ part: EMPTY_RUNNING_TEXT_PART })
-		});
+		let t1;
+		if ($[0] !== children) {
+			t1 = children({ part: EMPTY_RUNNING_TEXT_PART });
+			$[0] = children;
+			$[1] = t1;
+		} else t1 = $[1];
+		let t2;
+		if ($[2] !== t1) {
+			t2 = /* @__PURE__ */ jsx(TextMessagePartProvider, {
+				text: "",
+				isRunning: true,
+				children: t1
+			});
+			$[2] = t1;
+			$[3] = t2;
+		} else t2 = $[3];
+		return t2;
 	}
-	return /* @__PURE__ */ jsx(Fragment$1, { children: Array.from({ length: contentLength }, (_, index) => /* @__PURE__ */ jsx(MessagePartChildren, {
-		index,
-		children: (value) => children(value) ?? /* @__PURE__ */ jsx(DefaultPartFallback, {})
-	}, index)) });
+	let t1;
+	if ($[4] !== children || $[5] !== contentLength) {
+		let t2;
+		if ($[7] !== children) {
+			t2 = (_, index) => /* @__PURE__ */ jsx(MessagePartChildren, {
+				index,
+				children: (value) => children(value) ?? /* @__PURE__ */ jsx(DefaultPartFallback, {})
+			}, index);
+			$[7] = children;
+			$[8] = t2;
+		} else t2 = $[8];
+		t1 = /* @__PURE__ */ jsx(Fragment$1, { children: Array.from({ length: contentLength }, t2) });
+		$[4] = children;
+		$[5] = contentLength;
+		$[6] = t1;
+	} else t1 = $[6];
+	return t1;
 };
 /**
 * Renders the parts of a message with support for multiple content types.
@@ -344,70 +739,177 @@ const MessagePrimitivePartsInner = ({ children }) => {
 * This is the platform-agnostic base. Each platform wraps this with its own
 * default components (web uses `<p>`, `<span>`; RN would use `<Text>`, etc.).
 */
-const MessagePrimitiveParts = ({ components, unstable_showEmptyOnNonTextEnd = true, children }) => {
-	if (children) return /* @__PURE__ */ jsx(MessagePrimitivePartsInner, { children });
-	return /* @__PURE__ */ jsx(MessagePrimitivePartsCompat, {
-		components,
-		unstable_showEmptyOnNonTextEnd
-	});
+const MessagePrimitiveParts = (t0) => {
+	const $ = c(5);
+	const { components, unstable_showEmptyOnNonTextEnd: t1, children } = t0;
+	const unstable_showEmptyOnNonTextEnd = t1 === void 0 ? true : t1;
+	if (children) {
+		let t2;
+		if ($[0] !== children) {
+			t2 = /* @__PURE__ */ jsx(MessagePrimitivePartsInner, { children });
+			$[0] = children;
+			$[1] = t2;
+		} else t2 = $[1];
+		return t2;
+	}
+	let t2;
+	if ($[2] !== components || $[3] !== unstable_showEmptyOnNonTextEnd) {
+		t2 = /* @__PURE__ */ jsx(MessagePrimitivePartsCompat, {
+			components,
+			unstable_showEmptyOnNonTextEnd
+		});
+		$[2] = components;
+		$[3] = unstable_showEmptyOnNonTextEnd;
+		$[4] = t2;
+	} else t2 = $[4];
+	return t2;
 };
 MessagePrimitiveParts.displayName = "MessagePrimitive.Parts";
-const MessagePrimitivePartsCompat = ({ components, unstable_showEmptyOnNonTextEnd }) => {
-	const contentLength = useAuiState((s) => s.message.parts.length);
-	const messageRanges = useMessagePartsGroups(!!components?.ChainOfThought);
-	const partsElements = useMemo(() => {
-		if (contentLength === 0) return /* @__PURE__ */ jsx(EmptyParts, { components });
-		return messageRanges.map((range) => {
-			if (range.type === "single") return /* @__PURE__ */ jsx(MessagePrimitivePartByIndex, {
-				index: range.index,
-				components
-			}, range.index);
-			else if (range.type === "chainOfThoughtGroup") {
-				const ChainOfThoughtComponent = components?.ChainOfThought;
-				if (!ChainOfThoughtComponent) return null;
-				return /* @__PURE__ */ jsx(ChainOfThoughtByIndicesProvider, {
+const MessagePrimitivePartsCompat = (t0) => {
+	const $ = c(15);
+	const { components, unstable_showEmptyOnNonTextEnd } = t0;
+	const contentLength = useAuiState(_temp12);
+	const { ranges: messageRanges, partIds } = useMessagePartsGroups(!!components?.ChainOfThought);
+	let t1;
+	bb0: {
+		if (contentLength === 0) {
+			let t2;
+			if ($[0] !== components) {
+				t2 = /* @__PURE__ */ jsx(EmptyParts, { components });
+				$[0] = components;
+				$[1] = t2;
+			} else t2 = $[1];
+			t1 = t2;
+			break bb0;
+		}
+		let t2;
+		if ($[2] !== components || $[3] !== messageRanges || $[4] !== partIds) {
+			const claimed = /* @__PURE__ */ new Set();
+			const toolLeafKey = (partIndex) => {
+				const id = partIds[partIndex];
+				if (id !== void 0 && !claimed.has(id)) {
+					claimed.add(id);
+					return `part-id:${id}`;
+				}
+				return `part-${partIndex}`;
+			};
+			t2 = messageRanges.map((range) => {
+				if (range.type === "single") return /* @__PURE__ */ jsx(MessagePrimitivePartByIndex, {
+					index: range.index,
+					components
+				}, range.index);
+				else if (range.type === "chainOfThoughtGroup") {
+					const ChainOfThoughtComponent = components?.ChainOfThought;
+					if (!ChainOfThoughtComponent) return null;
+					return /* @__PURE__ */ jsx(ChainOfThoughtByIndicesProvider, {
+						startIndex: range.startIndex,
+						endIndex: range.endIndex,
+						children: /* @__PURE__ */ jsx(ChainOfThoughtComponent, {})
+					}, `chainOfThought-${range.idKey ?? range.startIndex}`);
+				} else if (range.type === "toolGroup") return /* @__PURE__ */ jsx(components?.ToolGroup ?? defaultComponents.ToolGroup, {
 					startIndex: range.startIndex,
 					endIndex: range.endIndex,
-					children: /* @__PURE__ */ jsx(ChainOfThoughtComponent, {})
-				}, `chainOfThought-${range.startIndex}`);
-			} else if (range.type === "toolGroup") return /* @__PURE__ */ jsx(components?.ToolGroup ?? defaultComponents.ToolGroup, {
-				startIndex: range.startIndex,
-				endIndex: range.endIndex,
-				children: Array.from({ length: range.endIndex - range.startIndex + 1 }, (_, i) => {
-					const partIndex = range.startIndex + i;
-					return /* @__PURE__ */ jsx(MessagePrimitivePartByIndex, {
-						index: partIndex,
-						components
-					}, `part-${partIndex}`);
-				})
-			}, `tool-${range.startIndex}`);
-			else return /* @__PURE__ */ jsx(components?.ReasoningGroup ?? defaultComponents.ReasoningGroup, {
-				startIndex: range.startIndex,
-				endIndex: range.endIndex,
-				children: Array.from({ length: range.endIndex - range.startIndex + 1 }, (_, i) => {
-					const partIndex = range.startIndex + i;
-					return /* @__PURE__ */ jsx(MessagePrimitivePartByIndex, {
-						index: partIndex,
-						components
-					}, `part-${partIndex}`);
-				})
-			}, `reasoning-${range.startIndex}`);
-		});
-	}, [
-		messageRanges,
-		components,
-		contentLength
-	]);
-	return /* @__PURE__ */ jsxs(Fragment$1, { children: [
-		components?.Quote && /* @__PURE__ */ jsx(QuoteRenderer, { Quote: components.Quote }),
-		partsElements,
-		/* @__PURE__ */ jsx(ConditionalEmpty, {
+					children: Array.from({ length: range.endIndex - range.startIndex + 1 }, (_, i) => {
+						const partIndex_0 = range.startIndex + i;
+						return /* @__PURE__ */ jsx(MessagePrimitivePartByIndex, {
+							index: partIndex_0,
+							components
+						}, toolLeafKey(partIndex_0));
+					})
+				}, `tool-${range.idKey ?? range.startIndex}`);
+				else return /* @__PURE__ */ jsx(components?.ReasoningGroup ?? defaultComponents.ReasoningGroup, {
+					startIndex: range.startIndex,
+					endIndex: range.endIndex,
+					children: Array.from({ length: range.endIndex - range.startIndex + 1 }, (__0, i_0) => {
+						const partIndex_1 = range.startIndex + i_0;
+						return /* @__PURE__ */ jsx(MessagePrimitivePartByIndex, {
+							index: partIndex_1,
+							components
+						}, `part-${partIndex_1}`);
+					})
+				}, `reasoning-${range.startIndex}`);
+			});
+			$[2] = components;
+			$[3] = messageRanges;
+			$[4] = partIds;
+			$[5] = t2;
+		} else t2 = $[5];
+		t1 = t2;
+	}
+	const partsElements = t1;
+	let t2;
+	if ($[6] !== components) {
+		t2 = components?.Quote && /* @__PURE__ */ jsx(QuoteRenderer, { Quote: components.Quote });
+		$[6] = components;
+		$[7] = t2;
+	} else t2 = $[7];
+	let t3;
+	if ($[8] !== components || $[9] !== unstable_showEmptyOnNonTextEnd) {
+		t3 = /* @__PURE__ */ jsx(ConditionalEmpty, {
 			components,
 			enabled: unstable_showEmptyOnNonTextEnd
-		})
-	] });
+		});
+		$[8] = components;
+		$[9] = unstable_showEmptyOnNonTextEnd;
+		$[10] = t3;
+	} else t3 = $[10];
+	let t4;
+	if ($[11] !== partsElements || $[12] !== t2 || $[13] !== t3) {
+		t4 = /* @__PURE__ */ jsxs(Fragment$1, { children: [
+			t2,
+			partsElements,
+			t3
+		] });
+		$[11] = partsElements;
+		$[12] = t2;
+		$[13] = t3;
+		$[14] = t4;
+	} else t4 = $[14];
+	return t4;
 };
+function _temp(c) {
+	return c.type;
+}
+function _temp2(s) {
+	return s.message.parts.map(_temp);
+}
+function _temp3(c_0) {
+	return c_0.type === "tool-call" ? c_0.toolCallId : void 0;
+}
+function _temp4(s_0) {
+	return s_0.message.parts.map(_temp3);
+}
+function _temp5(s) {
+	return s.part;
+}
+function _temp6(s) {
+	return s.message.status ?? COMPLETE_STATUS;
+}
+function _temp7(s) {
+	return s.part;
+}
+function _temp8(s_0) {
+	return s_0.part.type === "tool-call" ? resolveToolRender(s_0.tools, s_0.part) : null;
+}
+function _temp9(s) {
+	return s.part;
+}
+function _temp0(s_0) {
+	return s_0.part.type === "data" ? getDataRenderer(s_0.dataRenderers, s_0.part.name, void 0) ?? null : null;
+}
+function _temp1(s) {
+	return s.part.type;
+}
+function _temp10(s) {
+	return s.message.parts.length;
+}
+function _temp11(s_0) {
+	return (s_0.message.status?.type ?? "complete") === "running";
+}
+function _temp12(s) {
+	return s.message.parts.length;
+}
 //#endregion
-export { MessagePartChildren, MessagePartComponent, MessagePrimitivePartByIndex, MessagePrimitiveParts, defaultComponents };
+export { MessagePartChildren, MessagePartComponent, MessagePrimitivePartByIndex, MessagePrimitiveParts, defaultComponents, groupMessageParts };
 
 //# sourceMappingURL=MessageParts.js.map