@assistant-ui/react 0.14.18 → 0.14.19

package/dist/primitives/message/MessagePartsGrouped.js

@@ -5,7 +5,8 @@ import { MessagePartPrimitiveText } from "../messagePart/MessagePartText.js";
 import { MessagePartPrimitiveImage } from "../messagePart/MessagePartImage.js";
 import { MessagePartPrimitiveInProgress } from "../messagePart/MessagePartInProgress.js";
 import { useAui, useAuiState } from "@assistant-ui/store";
-import { memo, useMemo } from "@assistant-ui/tap/react-shim";
+import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { memo } from "@assistant-ui/tap/react-shim";
 import { Fragment, jsx, jsxs } from "react/jsx-runtime";
 //#region src/primitives/message/MessagePartsGrouped.tsx
 /**
@@ -32,29 +33,99 @@ const groupMessagePartsByParentId = (parts) => {
 	return groups;
 };
 const useMessagePartsGrouped = (groupingFunction) => {
-	const parts = useAuiState((s) => s.message.parts);
-	return useMemo(() => {
-		if (parts.length === 0) return [];
-		return groupingFunction(parts);
-	}, [parts, groupingFunction]);
+	const $ = c(4);
+	const parts = useAuiState(_temp);
+	let t0;
+	bb0: {
+		if (parts.length === 0) {
+			let t1;
+			if ($[0] === Symbol.for("react.memo_cache_sentinel")) {
+				t1 = [];
+				$[0] = t1;
+			} else t1 = $[0];
+			t0 = t1;
+			break bb0;
+		}
+		let t1;
+		if ($[1] !== groupingFunction || $[2] !== parts) {
+			t1 = groupingFunction(parts);
+			$[1] = groupingFunction;
+			$[2] = parts;
+			$[3] = t1;
+		} else t1 = $[3];
+		t0 = t1;
+	}
+	return t0;
 };
-const ToolUIDisplay = ({ Fallback, ...props }) => {
-	const Render = useAuiState((s) => {
-		const Render = s.tools.tools[props.toolName] ?? Fallback;
-		if (Array.isArray(Render)) return Render[0] ?? Fallback;
-		return Render;
-	});
-	if (!Render) return null;
-	return /* @__PURE__ */ jsx(Render, { ...props });
+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) => {
+			const Render = s.tools.tools[props.toolName] ?? Fallback;
+			if (Array.isArray(Render)) return Render[0] ?? Fallback;
+			return Render;
+		};
+		$[3] = Fallback;
+		$[4] = props.toolName;
+		$[5] = t1;
+	} else t1 = $[5];
+	const Render_0 = useAuiState(t1);
+	if (!Render_0) return null;
+	let t2;
+	if ($[6] !== Render_0 || $[7] !== props) {
+		t2 = /* @__PURE__ */ jsx(Render_0, { ...props });
+		$[6] = Render_0;
+		$[7] = props;
+		$[8] = t2;
+	} else t2 = $[8];
+	return t2;
 };
-const DataUIDisplay = ({ Fallback, ...props }) => {
-	const Render = useAuiState((s) => {
-		const Render = s.dataRenderers.renderers[props.name] ?? Fallback;
-		if (Array.isArray(Render)) return Render[0] ?? Fallback;
-		return Render;
-	});
-	if (!Render) return null;
-	return /* @__PURE__ */ jsx(Render, { ...props });
+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) => {
+			const Render = s.dataRenderers.renderers[props.name] ?? Fallback;
+			if (Array.isArray(Render)) return Render[0] ?? Fallback;
+			return Render;
+		};
+		$[3] = Fallback;
+		$[4] = props.name;
+		$[5] = t1;
+	} else t1 = $[5];
+	const Render_0 = useAuiState(t1);
+	if (!Render_0) return null;
+	let t2;
+	if ($[6] !== Render_0 || $[7] !== props) {
+		t2 = /* @__PURE__ */ jsx(Render_0, { ...props });
+		$[6] = Render_0;
+		$[7] = props;
+		$[8] = t2;
+	} else t2 = $[8];
+	return t2;
 };
 const defaultComponents = {
 	Text: () => /* @__PURE__ */ jsxs("p", {
@@ -71,75 +142,249 @@ const defaultComponents = {
 	Unstable_Audio: () => null,
 	Group: ({ 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 } = {} }) => {
+const MessagePartComponent = (t0) => {
+	const $ = c(43);
+	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 } = 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(_temp2);
 	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;
 		}
 		default:
 			console.warn(`Unknown message part type: ${type}`);
 			return null;
 	}
 };
-const MessagePartImpl = ({ partIndex, components }) => {
-	return /* @__PURE__ */ jsx(PartByIndexProvider, {
-		index: partIndex,
-		children: /* @__PURE__ */ jsx(MessagePartComponent, { components })
-	});
+const MessagePartImpl = (t0) => {
+	const $ = c(5);
+	const { partIndex, components } = t0;
+	let t1;
+	if ($[0] !== components) {
+		t1 = /* @__PURE__ */ jsx(MessagePartComponent, { components });
+		$[0] = components;
+		$[1] = t1;
+	} else t1 = $[1];
+	let t2;
+	if ($[2] !== partIndex || $[3] !== t1) {
+		t2 = /* @__PURE__ */ jsx(PartByIndexProvider, {
+			index: partIndex,
+			children: t1
+		});
+		$[2] = partIndex;
+		$[3] = t1;
+		$[4] = t2;
+	} else t2 = $[4];
+	return t2;
 };
 const MessagePart = memo(MessagePartImpl, (prev, next) => prev.partIndex === next.partIndex && 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?.Group === next.components?.Group);
-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 EmptyPartsImpl = ({ components }) => {
-	const status = useAuiState((s) => s.message.status ?? COMPLETE_STATUS);
-	if (components?.Empty) return /* @__PURE__ */ jsx(components.Empty, { status });
-	return /* @__PURE__ */ jsx(EmptyPartFallback, {
-		status,
-		component: components?.Text ?? defaultComponents.Text
-	});
+const EmptyPartsImpl = (t0) => {
+	const $ = c(6);
+	const { components } = t0;
+	const status = useAuiState(_temp3);
+	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;
+	}
+	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);
 /**
@@ -175,26 +420,55 @@ const EmptyParts = memo(EmptyPartsImpl, (prev, next) => prev.components?.Empty =
 * />
 * ```
 */
-const MessagePrimitiveUnstable_PartsGrouped = ({ groupingFunction, components }) => {
-	const contentLength = useAuiState((s) => s.message.parts.length);
+const MessagePrimitiveUnstable_PartsGrouped = (t0) => {
+	const $ = c(9);
+	const { groupingFunction, components } = t0;
+	const contentLength = useAuiState(_temp4);
 	const messageGroups = useMessagePartsGrouped(groupingFunction);
-	return /* @__PURE__ */ jsx(Fragment, { children: useMemo(() => {
-		if (contentLength === 0) return /* @__PURE__ */ jsx(EmptyParts, { components });
-		return messageGroups.map((group, groupIndex) => {
-			return /* @__PURE__ */ jsx(components?.Group ?? defaultComponents.Group, {
-				groupKey: group.groupKey,
-				indices: group.indices,
-				children: group.indices.map((partIndex) => /* @__PURE__ */ jsx(MessagePart, {
-					partIndex,
-					components
-				}, partIndex))
-			}, `group-${groupIndex}-${group.groupKey ?? "ungrouped"}`);
-		});
-	}, [
-		messageGroups,
-		components,
-		contentLength
-	]) });
+	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] !== messageGroups) {
+			let t3;
+			if ($[5] !== components) {
+				t3 = (group, groupIndex) => {
+					return /* @__PURE__ */ jsx(components?.Group ?? defaultComponents.Group, {
+						groupKey: group.groupKey,
+						indices: group.indices,
+						children: group.indices.map((partIndex) => /* @__PURE__ */ jsx(MessagePart, {
+							partIndex,
+							components
+						}, partIndex))
+					}, `group-${groupIndex}-${group.groupKey ?? "ungrouped"}`);
+				};
+				$[5] = components;
+				$[6] = t3;
+			} else t3 = $[6];
+			t2 = messageGroups.map(t3);
+			$[2] = components;
+			$[3] = messageGroups;
+			$[4] = t2;
+		} else t2 = $[4];
+		t1 = t2;
+	}
+	const partsElements = t1;
+	let t2;
+	if ($[7] !== partsElements) {
+		t2 = /* @__PURE__ */ jsx(Fragment, { children: partsElements });
+		$[7] = partsElements;
+		$[8] = t2;
+	} else t2 = $[8];
+	return t2;
 };
 MessagePrimitiveUnstable_PartsGrouped.displayName = "MessagePrimitive.Unstable_PartsGrouped";
 /**
@@ -203,14 +477,45 @@ MessagePrimitiveUnstable_PartsGrouped.displayName = "MessagePrimitive.Unstable_P
 *
 * @deprecated Use MessagePrimitive.Unstable_PartsGrouped instead for more flexibility
 */
-const MessagePrimitiveUnstable_PartsGroupedByParentId = ({ components, ...props }) => {
-	return /* @__PURE__ */ jsx(MessagePrimitiveUnstable_PartsGrouped, {
-		...props,
-		components,
-		groupingFunction: groupMessagePartsByParentId
-	});
+const MessagePrimitiveUnstable_PartsGroupedByParentId = (t0) => {
+	const $ = c(6);
+	let components;
+	let props;
+	if ($[0] !== t0) {
+		({components, ...props} = t0);
+		$[0] = t0;
+		$[1] = components;
+		$[2] = props;
+	} else {
+		components = $[1];
+		props = $[2];
+	}
+	let t1;
+	if ($[3] !== components || $[4] !== props) {
+		t1 = /* @__PURE__ */ jsx(MessagePrimitiveUnstable_PartsGrouped, {
+			...props,
+			components,
+			groupingFunction: groupMessagePartsByParentId
+		});
+		$[3] = components;
+		$[4] = props;
+		$[5] = t1;
+	} else t1 = $[5];
+	return t1;
 };
 MessagePrimitiveUnstable_PartsGroupedByParentId.displayName = "MessagePrimitive.Unstable_PartsGroupedByParentId";
+function _temp(s) {
+	return s.message.parts;
+}
+function _temp2(s) {
+	return s.part;
+}
+function _temp3(s) {
+	return s.message.status ?? COMPLETE_STATUS;
+}
+function _temp4(s) {
+	return s.message.parts.length;
+}
 //#endregion
 export { MessagePrimitiveUnstable_PartsGrouped, MessagePrimitiveUnstable_PartsGroupedByParentId };