@assistant-ui/react 0.14.6 → 0.14.8

package/dist/primitives/message/MessagePartsGrouped.js

@@ -1,201 +1,214 @@
 "use client";
-import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
-import { memo, useMemo, } from "react";
-import { useAuiState, useAui } from "@assistant-ui/store";
 import { PartByIndexProvider } from "../../context/providers/PartByIndexProvider.js";
 import { TextMessagePartProvider } from "../../context/providers/TextMessagePartProvider.js";
 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 "react";
+import { Fragment, jsx, jsxs } from "react/jsx-runtime";
+//#region src/primitives/message/MessagePartsGrouped.tsx
 /**
- * Groups message parts by their parent ID.
- * Parts without a parent ID appear in their chronological position as individual groups.
- * Parts with the same parent ID are grouped together at the position of their first occurrence.
- */
+* Groups message parts by their parent ID.
+* Parts without a parent ID appear in their chronological position as individual groups.
+* Parts with the same parent ID are grouped together at the position of their first occurrence.
+*/
 const groupMessagePartsByParentId = (parts) => {
-    // Map maintains insertion order, so groups appear in order of first occurrence
-    const groupMap = new Map();
-    // Process each part in order
-    for (let i = 0; i < parts.length; i++) {
-        const part = parts[i];
-        const parentId = part?.parentId;
-        // For parts without parentId, assign a unique group ID to maintain their position
-        const groupId = parentId ?? `__ungrouped_${i}`;
-        // Get or create the indices array for this group
-        const indices = groupMap.get(groupId) ?? [];
-        indices.push(i);
-        groupMap.set(groupId, indices);
-    }
-    // Convert map to array of groups
-    const groups = [];
-    for (const [groupId, indices] of groupMap) {
-        // Extract parentId (undefined for ungrouped parts)
-        const groupKey = groupId.startsWith("__ungrouped_") ? undefined : groupId;
-        groups.push({ groupKey, indices });
-    }
-    return groups;
+	const groupMap = /* @__PURE__ */ new Map();
+	for (let i = 0; i < parts.length; i++) {
+		const groupId = parts[i]?.parentId ?? `__ungrouped_${i}`;
+		const indices = groupMap.get(groupId) ?? [];
+		indices.push(i);
+		groupMap.set(groupId, indices);
+	}
+	const groups = [];
+	for (const [groupId, indices] of groupMap) {
+		const groupKey = groupId.startsWith("__ungrouped_") ? void 0 : groupId;
+		groups.push({
+			groupKey,
+			indices
+		});
+	}
+	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 parts = useAuiState((s) => s.message.parts);
+	return useMemo(() => {
+		if (parts.length === 0) return [];
+		return groupingFunction(parts);
+	}, [parts, groupingFunction]);
 };
 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 _jsx(Render, { ...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 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 _jsx(Render, { ...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 defaultComponents = {
-    Text: () => (_jsxs("p", { style: { whiteSpace: "pre-line" }, children: [_jsx(MessagePartPrimitiveText, {}), _jsx(MessagePartPrimitiveInProgress, { children: _jsx("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })] })),
-    Reasoning: () => null,
-    Source: () => null,
-    Image: () => _jsx(MessagePartPrimitiveImage, {}),
-    File: () => null,
-    Unstable_Audio: () => null,
-    Group: ({ children }) => children,
+	Text: () => /* @__PURE__ */ jsxs("p", {
+		style: { whiteSpace: "pre-line" },
+		children: [/* @__PURE__ */ jsx(MessagePartPrimitiveText, {}), /* @__PURE__ */ jsx(MessagePartPrimitiveInProgress, { children: /* @__PURE__ */ jsx("span", {
+			style: { fontFamily: "revert" },
+			children: " ●"
+		}) })]
+	}),
+	Reasoning: () => null,
+	Source: () => null,
+	Image: () => /* @__PURE__ */ jsx(MessagePartPrimitiveImage, {}),
+	File: () => null,
+	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 aui = useAui();
-    const part = useAuiState((s) => s.part);
-    const type = part.type;
-    if (type === "tool-call") {
-        const addResult = aui.part().addToolResult;
-        const resume = aui.part().resumeToolCall;
-        if ("Override" in tools)
-            return _jsx(tools.Override, { ...part, addResult: addResult, resume: resume });
-        const Tool = tools.by_name?.[part.toolName] ?? tools.Fallback;
-        return (_jsx(ToolUIDisplay, { ...part, Fallback: Tool, addResult: addResult, resume: resume }));
-    }
-    if (part.status?.type === "requires-action")
-        throw new Error("Encountered unexpected requires-action status");
-    switch (type) {
-        case "text":
-            return _jsx(Text, { ...part });
-        case "reasoning":
-            return _jsx(Reasoning, { ...part });
-        case "source":
-            return _jsx(Source, { ...part });
-        case "image":
-            return _jsx(Image, { ...part });
-        case "file":
-            return _jsx(File, { ...part });
-        case "audio":
-            return _jsx(Audio, { ...part });
-        case "data": {
-            const Data = data?.by_name?.[part.name] ?? data?.Fallback;
-            return _jsx(DataUIDisplay, { ...part, Fallback: Data });
-        }
-        default:
-            console.warn(`Unknown message part type: ${type}`);
-            return null;
-    }
+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 aui = useAui();
+	const part = useAuiState((s) => s.part);
+	const type = part.type;
+	if (type === "tool-call") {
+		const addResult = aui.part().addToolResult;
+		const resume = aui.part().resumeToolCall;
+		if ("Override" in tools) return /* @__PURE__ */ jsx(tools.Override, {
+			...part,
+			addResult,
+			resume
+		});
+		const Tool = tools.by_name?.[part.toolName] ?? tools.Fallback;
+		return /* @__PURE__ */ jsx(ToolUIDisplay, {
+			...part,
+			Fallback: Tool,
+			addResult,
+			resume
+		});
+	}
+	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 "data": {
+			const Data = data?.by_name?.[part.name] ?? data?.Fallback;
+			return /* @__PURE__ */ jsx(DataUIDisplay, {
+				...part,
+				Fallback: Data
+			});
+		}
+		default:
+			console.warn(`Unknown message part type: ${type}`);
+			return null;
+	}
 };
 const MessagePartImpl = ({ partIndex, components }) => {
-    return (_jsx(PartByIndexProvider, { index: partIndex, children: _jsx(MessagePartComponent, { components: components }) }));
+	return /* @__PURE__ */ jsx(PartByIndexProvider, {
+		index: partIndex,
+		children: /* @__PURE__ */ jsx(MessagePartComponent, { components })
+	});
 };
-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 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 (_jsx(TextMessagePartProvider, { text: "", isRunning: status.type === "running", children: _jsx(Component, { type: "text", text: "", status: status }) }));
+	return /* @__PURE__ */ jsx(TextMessagePartProvider, {
+		text: "",
+		isRunning: status.type === "running",
+		children: /* @__PURE__ */ jsx(Component, {
+			type: "text",
+			text: "",
+			status
+		})
+	});
 };
-const COMPLETE_STATUS = Object.freeze({
-    type: "complete",
-});
+const COMPLETE_STATUS = Object.freeze({ type: "complete" });
 const EmptyPartsImpl = ({ components }) => {
-    const status = useAuiState((s) => (s.message.status ?? COMPLETE_STATUS));
-    if (components?.Empty)
-        return _jsx(components.Empty, { status: status });
-    return (_jsx(EmptyPartFallback, { status: status, component: components?.Text ?? defaultComponents.Text }));
+	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 EmptyParts = memo(EmptyPartsImpl, (prev, next) => prev.components?.Empty === next.components?.Empty &&
-    prev.components?.Text === next.components?.Text);
+const EmptyParts = memo(EmptyPartsImpl, (prev, next) => prev.components?.Empty === next.components?.Empty && prev.components?.Text === next.components?.Text);
 /**
- * Renders the parts of a message grouped by a custom grouping function.
- *
- * This component allows you to group message parts based on any criteria you define.
- * The grouping function receives all message parts and returns an array of groups,
- * where each group has a key and an array of part indices.
- *
- * @deprecated Prefer `<MessagePrimitive.GroupedParts>` for adjacent
- * grouping — it dispatches all rendering through one `switch (part.type)`
- * and supports nested group paths. Keep this primitive only for
- * non-adjacent clustering (e.g., gathering parts with the same parent-id
- * across the message).
- *
- * @example
- * ```tsx
- * // Group by parent ID (default behavior)
- * <MessagePrimitive.Unstable_PartsGrouped
- *   components={{
- *     Text: ({ text }) => <p className="message-text">{text}</p>,
- *     Image: ({ image }) => <img src={image} alt="Message image" />,
- *     Group: ({ groupKey, indices, children }) => {
- *       if (!groupKey) return <>{children}</>;
- *       return (
- *         <div className="parent-group border rounded p-4">
- *           <h4>Parent ID: {groupKey}</h4>
- *           {children}
- *         </div>
- *       );
- *     }
- *   }}
- * />
- * ```
- */
-export const MessagePrimitiveUnstable_PartsGrouped = ({ groupingFunction, components }) => {
-    const contentLength = useAuiState((s) => s.message.parts.length);
-    const messageGroups = useMessagePartsGrouped(groupingFunction);
-    const partsElements = useMemo(() => {
-        if (contentLength === 0) {
-            return _jsx(EmptyParts, { components: components });
-        }
-        return messageGroups.map((group, groupIndex) => {
-            const GroupComponent = components?.Group ?? defaultComponents.Group;
-            return (_jsx(GroupComponent, { groupKey: group.groupKey, indices: group.indices, children: group.indices.map((partIndex) => (_jsx(MessagePart, { partIndex: partIndex, components: components }, partIndex))) }, `group-${groupIndex}-${group.groupKey ?? "ungrouped"}`));
-        });
-    }, [messageGroups, components, contentLength]);
-    return _jsx(_Fragment, { children: partsElements });
+* Renders the parts of a message grouped by a custom grouping function.
+*
+* This component allows you to group message parts based on any criteria you define.
+* The grouping function receives all message parts and returns an array of groups,
+* where each group has a key and an array of part indices.
+*
+* @deprecated Prefer `<MessagePrimitive.GroupedParts>` for adjacent
+* grouping — it dispatches all rendering through one `switch (part.type)`
+* and supports nested group paths. Keep this primitive only for
+* non-adjacent clustering (e.g., gathering parts with the same parent-id
+* across the message).
+*
+* @example
+* ```tsx
+* // Group by parent ID (default behavior)
+* <MessagePrimitive.Unstable_PartsGrouped
+*   components={{
+*     Text: ({ text }) => <p className="message-text">{text}</p>,
+*     Image: ({ image }) => <img src={image} alt="Message image" />,
+*     Group: ({ groupKey, indices, children }) => {
+*       if (!groupKey) return <>{children}</>;
+*       return (
+*         <div className="parent-group border rounded p-4">
+*           <h4>Parent ID: {groupKey}</h4>
+*           {children}
+*         </div>
+*       );
+*     }
+*   }}
+* />
+* ```
+*/
+const MessagePrimitiveUnstable_PartsGrouped = ({ groupingFunction, components }) => {
+	const contentLength = useAuiState((s) => s.message.parts.length);
+	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
+	]) });
 };
-MessagePrimitiveUnstable_PartsGrouped.displayName =
-    "MessagePrimitive.Unstable_PartsGrouped";
+MessagePrimitiveUnstable_PartsGrouped.displayName = "MessagePrimitive.Unstable_PartsGrouped";
 /**
- * Renders the parts of a message grouped by their parent ID.
- * This is a convenience wrapper around Unstable_PartsGrouped with parent ID grouping.
- *
- * @deprecated Use MessagePrimitive.Unstable_PartsGrouped instead for more flexibility
- */
-export const MessagePrimitiveUnstable_PartsGroupedByParentId = ({ components, ...props }) => {
-    return (_jsx(MessagePrimitiveUnstable_PartsGrouped, { ...props, components: components, groupingFunction: groupMessagePartsByParentId }));
+* Renders the parts of a message grouped by their parent ID.
+* This is a convenience wrapper around Unstable_PartsGrouped with parent ID grouping.
+*
+* @deprecated Use MessagePrimitive.Unstable_PartsGrouped instead for more flexibility
+*/
+const MessagePrimitiveUnstable_PartsGroupedByParentId = ({ components, ...props }) => {
+	return /* @__PURE__ */ jsx(MessagePrimitiveUnstable_PartsGrouped, {
+		...props,
+		components,
+		groupingFunction: groupMessagePartsByParentId
+	});
 };
-MessagePrimitiveUnstable_PartsGroupedByParentId.displayName =
-    "MessagePrimitive.Unstable_PartsGroupedByParentId";
+MessagePrimitiveUnstable_PartsGroupedByParentId.displayName = "MessagePrimitive.Unstable_PartsGroupedByParentId";
+//#endregion
+export { MessagePrimitiveUnstable_PartsGrouped, MessagePrimitiveUnstable_PartsGroupedByParentId };
+
 //# sourceMappingURL=MessagePartsGrouped.js.map

package/dist/primitives/message/MessagePartsGrouped.js.map

@@ -1 +1 @@
-{"version":3,"file":"MessagePartsGrouped.js","sourceRoot":"","sources":["../../../src/primitives/message/MessagePartsGrouped.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAEb,OAAO,EAGL,IAAI,EAEJ,OAAO,GACR,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,mBAAmB,EAAE,uDAAoD;AAClF,OAAO,EAAE,uBAAuB,EAAE,2DAAwD;AAC1F,OAAO,EAAE,wBAAwB,EAAE,0CAAuC;AAC1E,OAAO,EAAE,yBAAyB,EAAE,2CAAwC;AAc5E,OAAO,EAAE,8BAA8B,EAAE,gDAA6C;AAUtF;;;;GAIG;AACH,MAAM,2BAA2B,GAAqB,CACpD,KAAqB,EACD,EAAE;IACtB,+EAA+E;IAC/E,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAoB,CAAC;IAE7C,6BAA6B;IAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,QAAQ,GAAG,IAAI,EAAE,QAA8B,CAAC;QAEtD,kFAAkF;QAClF,MAAM,OAAO,GAAG,QAAQ,IAAI,eAAe,CAAC,EAAE,CAAC;QAE/C,iDAAiD;QACjD,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChB,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACjC,CAAC;IAED,iCAAiC;IACjC,MAAM,MAAM,GAAuB,EAAE,CAAC;IACtC,KAAK,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,QAAQ,EAAE,CAAC;QAC1C,mDAAmD;QACnD,MAAM,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;QAC1E,MAAM,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;IACrC,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC;AAEF,MAAM,sBAAsB,GAAG,CAC7B,gBAAkC,EACd,EAAE;IACtB,MAAM,KAAK,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAElD,OAAO,OAAO,CAAC,GAAG,EAAE;QAClB,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC,EAAE,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC,CAAC;AAChC,CAAC,CAAC;AA8IF,MAAM,aAAa,GAAG,CAAC,EACrB,QAAQ,EACR,GAAG,KAAK,EAGkB,EAAE,EAAE;IAC9B,MAAM,MAAM,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE;QAC/B,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC;QACzD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YAAE,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC;QACxD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,CAAC;IACH,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IACzB,OAAO,KAAC,MAAM,OAAK,KAAK,GAAI,CAAC;AAC/B,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,CAAC,EACrB,QAAQ,EACR,GAAG,KAAK,EAGc,EAAE,EAAE;IAC1B,MAAM,MAAM,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE;QAC/B,MAAM,MAAM,GAAG,CAAC,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC;QACjE,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YAAE,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC;QACxD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,CAAC;IACH,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IACzB,OAAO,KAAC,MAAM,OAAK,KAAK,GAAI,CAAC;AAC/B,CAAC,CAAC;AAEF,MAAM,iBAAiB,GAAG;IACxB,IAAI,EAAE,GAAG,EAAE,CAAC,CACV,aAAG,KAAK,EAAE,EAAE,UAAU,EAAE,UAAU,EAAE,aAClC,KAAC,wBAAwB,KAAG,EAC5B,KAAC,8BAA8B,cAC7B,eAAM,KAAK,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,YAAG,SAAS,GAAQ,GAC1B,IAC/B,CACL;IACD,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI;IACrB,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI;IAClB,KAAK,EAAE,GAAG,EAAE,CAAC,KAAC,yBAAyB,KAAG;IAC1C,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI;IAChB,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI;IAC1B,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,QAAQ;CACkC,CAAC;AAMtE,MAAM,oBAAoB,GAAkC,CAAC,EAC3D,UAAU,EAAE,EACV,IAAI,GAAG,iBAAiB,CAAC,IAAI,EAC7B,SAAS,GAAG,iBAAiB,CAAC,SAAS,EACvC,KAAK,GAAG,iBAAiB,CAAC,KAAK,EAC/B,MAAM,GAAG,iBAAiB,CAAC,MAAM,EACjC,IAAI,GAAG,iBAAiB,CAAC,IAAI,EAC7B,cAAc,EAAE,KAAK,GAAG,iBAAiB,CAAC,cAAc,EACxD,KAAK,GAAG,EAAE,EACV,IAAI,GACL,GAAG,EAAE,GACP,EAAE,EAAE;IACH,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IACrB,MAAM,IAAI,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAExC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;IACvB,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;QACzB,MAAM,SAAS,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC,aAAa,CAAC;QAC3C,MAAM,MAAM,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC,cAAc,CAAC;QACzC,IAAI,UAAU,IAAI,KAAK;YACrB,OAAO,KAAC,KAAK,CAAC,QAAQ,OAAK,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAI,CAAC;QAC5E,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC;QAC9D,OAAO,CACL,KAAC,aAAa,OACR,IAAI,EACR,QAAQ,EAAE,IAAI,EACd,SAAS,EAAE,SAAS,EACpB,MAAM,EAAE,MAAM,GACd,CACH,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,iBAAiB;QACzC,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IAEnE,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,MAAM;YACT,OAAO,KAAC,IAAI,OAAK,IAAI,GAAI,CAAC;QAE5B,KAAK,WAAW;YACd,OAAO,KAAC,SAAS,OAAK,IAAI,GAAI,CAAC;QAEjC,KAAK,QAAQ;YACX,OAAO,KAAC,MAAM,OAAK,IAAI,GAAI,CAAC;QAE9B,KAAK,OAAO;YACV,OAAO,KAAC,KAAK,OAAK,IAAI,GAAI,CAAC;QAE7B,KAAK,MAAM;YACT,OAAO,KAAC,IAAI,OAAK,IAAI,GAAI,CAAC;QAE5B,KAAK,OAAO;YACV,OAAO,KAAC,KAAK,OAAK,IAAI,GAAI,CAAC;QAE7B,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,MAAM,IAAI,GAAG,IAAI,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,QAAQ,CAAC;YAC1D,OAAO,KAAC,aAAa,OAAK,IAAI,EAAE,QAAQ,EAAE,IAAI,GAAI,CAAC;QACrD,CAAC;QAED;YACE,OAAO,CAAC,IAAI,CAAC,8BAA8B,IAAI,EAAE,CAAC,CAAC;YACnD,OAAO,IAAI,CAAC;IAChB,CAAC;AACH,CAAC,CAAC;AAOF,MAAM,eAAe,GAAyB,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE,EAAE;IAC1E,OAAO,CACL,KAAC,mBAAmB,IAAC,KAAK,EAAE,SAAS,YACnC,KAAC,oBAAoB,IAAC,UAAU,EAAE,UAAU,GAAI,GAC5B,CACvB,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,IAAI,CACtB,eAAe,EACf,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CACb,IAAI,CAAC,SAAS,KAAK,IAAI,CAAC,SAAS;IACjC,IAAI,CAAC,UAAU,EAAE,IAAI,KAAK,IAAI,CAAC,UAAU,EAAE,IAAI;IAC/C,IAAI,CAAC,UAAU,EAAE,SAAS,KAAK,IAAI,CAAC,UAAU,EAAE,SAAS;IACzD,IAAI,CAAC,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC,UAAU,EAAE,MAAM;IACnD,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,IAAI,CAAC,UAAU,EAAE,KAAK;IACjD,IAAI,CAAC,UAAU,EAAE,IAAI,KAAK,IAAI,CAAC,UAAU,EAAE,IAAI;IAC/C,IAAI,CAAC,UAAU,EAAE,cAAc,KAAK,IAAI,CAAC,UAAU,EAAE,cAAc;IACnE,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,IAAI,CAAC,UAAU,EAAE,KAAK;IACjD,IAAI,CAAC,UAAU,EAAE,IAAI,KAAK,IAAI,CAAC,UAAU,EAAE,IAAI;IAC/C,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,IAAI,CAAC,UAAU,EAAE,KAAK,CACpD,CAAC;AAEF,MAAM,iBAAiB,GAGlB,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,EAAE;IACxC,OAAO,CACL,KAAC,uBAAuB,IAAC,IAAI,EAAC,EAAE,EAAC,SAAS,EAAE,MAAM,CAAC,IAAI,KAAK,SAAS,YACnE,KAAC,SAAS,IAAC,IAAI,EAAC,MAAM,EAAC,IAAI,EAAC,EAAE,EAAC,MAAM,EAAE,MAAM,GAAI,GACzB,CAC3B,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,eAAe,GAAsB,MAAM,CAAC,MAAM,CAAC;IACvD,IAAI,EAAE,UAAU;CACjB,CAAC,CAAC;AAEH,MAAM,cAAc,GAAkC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE;IACvE,MAAM,MAAM,GAAG,WAAW,CACxB,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,IAAI,eAAe,CAAsB,CAClE,CAAC;IAEF,IAAI,UAAU,EAAE,KAAK;QAAE,OAAO,KAAC,UAAU,CAAC,KAAK,IAAC,MAAM,EAAE,MAAM,GAAI,CAAC;IAEnE,OAAO,CACL,KAAC,iBAAiB,IAChB,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,UAAU,EAAE,IAAI,IAAI,iBAAiB,CAAC,IAAI,GACrD,CACH,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,UAAU,GAAG,IAAI,CACrB,cAAc,EACd,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CACb,IAAI,CAAC,UAAU,EAAE,KAAK,KAAK,IAAI,CAAC,UAAU,EAAE,KAAK;IACjD,IAAI,CAAC,UAAU,EAAE,IAAI,KAAK,IAAI,CAAC,UAAU,EAAE,IAAI,CAClD,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,MAAM,CAAC,MAAM,qCAAqC,GAE9C,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAAE,EAAE,EAAE;IACvC,MAAM,aAAa,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACjE,MAAM,aAAa,GAAG,sBAAsB,CAAC,gBAAgB,CAAC,CAAC;IAE/D,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,EAAE;QACjC,IAAI,aAAa,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,KAAC,UAAU,IAAC,UAAU,EAAE,UAAU,GAAI,CAAC;QAChD,CAAC;QAED,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE;YAC7C,MAAM,cAAc,GAAG,UAAU,EAAE,KAAK,IAAI,iBAAiB,CAAC,KAAK,CAAC;YAEpE,OAAO,CACL,KAAC,cAAc,IAEb,QAAQ,EAAE,KAAK,CAAC,QAAQ,EACxB,OAAO,EAAE,KAAK,CAAC,OAAO,YAErB,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAChC,KAAC,WAAW,IAEV,SAAS,EAAE,SAAS,EACpB,UAAU,EAAE,UAAU,IAFjB,SAAS,CAGd,CACH,CAAC,IAVG,SAAS,UAAU,IAAI,KAAK,CAAC,QAAQ,IAAI,WAAW,EAAE,CAW5C,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,aAAa,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC,CAAC;IAE/C,OAAO,4BAAG,aAAa,GAAI,CAAC;AAC9B,CAAC,CAAC;AAEF,qCAAqC,CAAC,WAAW;IAC/C,wCAAwC,CAAC;AAE3C;;;;;GAKG;AACH,MAAM,CAAC,MAAM,+CAA+C,GAExD,CAAC,EAAE,UAAU,EAAE,GAAG,KAAK,EAAE,EAAE,EAAE;IAC/B,OAAO,CACL,KAAC,qCAAqC,OAChC,KAAK,EACT,UAAU,EAAE,UAAU,EACtB,gBAAgB,EAAE,2BAA2B,GAC7C,CACH,CAAC;AACJ,CAAC,CAAC;AAEF,+CAA+C,CAAC,WAAW;IACzD,kDAAkD,CAAC"}
+{"version":3,"file":"MessagePartsGrouped.js","names":[],"sources":["../../../src/primitives/message/MessagePartsGrouped.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  type ComponentType,\n  type FC,\n  memo,\n  type PropsWithChildren,\n  useMemo,\n} from \"react\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\nimport { PartByIndexProvider } from \"../../context/providers/PartByIndexProvider\";\nimport { TextMessagePartProvider } from \"../../context/providers/TextMessagePartProvider\";\nimport { MessagePartPrimitiveText } from \"../messagePart/MessagePartText\";\nimport { MessagePartPrimitiveImage } from \"../messagePart/MessagePartImage\";\nimport type {\n  Unstable_AudioMessagePartComponent,\n  DataMessagePartComponent,\n  DataMessagePartProps,\n  EmptyMessagePartComponent,\n  TextMessagePartComponent,\n  ImageMessagePartComponent,\n  SourceMessagePartComponent,\n  ToolCallMessagePartComponent,\n  ToolCallMessagePartProps,\n  FileMessagePartComponent,\n  ReasoningMessagePartComponent,\n} from \"@assistant-ui/core/react\";\nimport { MessagePartPrimitiveInProgress } from \"../messagePart/MessagePartInProgress\";\nimport type { MessagePartStatus } from \"@assistant-ui/core\";\n\ntype MessagePartGroup = {\n  groupKey: string | undefined;\n  indices: number[];\n};\n\nexport type GroupingFunction = (parts: readonly any[]) => MessagePartGroup[];\n\n/**\n * Groups message parts by their parent ID.\n * Parts without a parent ID appear in their chronological position as individual groups.\n * Parts with the same parent ID are grouped together at the position of their first occurrence.\n */\nconst groupMessagePartsByParentId: GroupingFunction = (\n  parts: readonly any[],\n): MessagePartGroup[] => {\n  // Map maintains insertion order, so groups appear in order of first occurrence\n  const groupMap = new Map<string, number[]>();\n\n  // Process each part in order\n  for (let i = 0; i < parts.length; i++) {\n    const part = parts[i];\n    const parentId = part?.parentId as string | undefined;\n\n    // For parts without parentId, assign a unique group ID to maintain their position\n    const groupId = parentId ?? `__ungrouped_${i}`;\n\n    // Get or create the indices array for this group\n    const indices = groupMap.get(groupId) ?? [];\n    indices.push(i);\n    groupMap.set(groupId, indices);\n  }\n\n  // Convert map to array of groups\n  const groups: MessagePartGroup[] = [];\n  for (const [groupId, indices] of groupMap) {\n    // Extract parentId (undefined for ungrouped parts)\n    const groupKey = groupId.startsWith(\"__ungrouped_\") ? undefined : groupId;\n    groups.push({ groupKey, indices });\n  }\n\n  return groups;\n};\n\nconst useMessagePartsGrouped = (\n  groupingFunction: GroupingFunction,\n): MessagePartGroup[] => {\n  const parts = useAuiState((s) => s.message.parts);\n\n  return useMemo(() => {\n    if (parts.length === 0) {\n      return [];\n    }\n    return groupingFunction(parts);\n  }, [parts, groupingFunction]);\n};\n\nexport namespace MessagePrimitiveUnstable_PartsGrouped {\n  export type Props = {\n    /**\n     * Function that takes an array of message parts and returns an array of groups.\n     * Each group contains a key (for identification) and an array of indices.\n     *\n     * @example\n     * ```tsx\n     * // Group by parent ID (default behavior)\n     * groupingFunction={(parts) => {\n     *   const groups = new Map<string, number[]>();\n     *   parts.forEach((part, i) => {\n     *     const key = part.parentId ?? `__ungrouped_${i}`;\n     *     const indices = groups.get(key) ?? [];\n     *     indices.push(i);\n     *     groups.set(key, indices);\n     *   });\n     *   return Array.from(groups.entries()).map(([key, indices]) => ({\n     *     key: key.startsWith(\"__ungrouped_\") ? undefined : key,\n     *     indices\n     *   }));\n     * }}\n     * ```\n     *\n     * @example\n     * ```tsx\n     * // Group by tool name\n     * import { groupMessagePartsByToolName } from \"@assistant-ui/react\";\n     *\n     * <MessagePrimitive.Unstable_PartsGrouped\n     *   groupingFunction={groupMessagePartsByToolName}\n     *   components={{\n     *     Group: ({ key, indices, children }) => {\n     *       if (!key) return <>{children}</>;\n     *       return (\n     *         <div className=\"tool-group\">\n     *           <h4>Tool: {key}</h4>\n     *           {children}\n     *         </div>\n     *       );\n     *     }\n     *   }}\n     * />\n     * ```\n     */\n    groupingFunction: GroupingFunction;\n\n    /**\n     * Component configuration for rendering different types of message content.\n     *\n     * You can provide custom components for each content type (text, image, file, etc.)\n     * and configure tool rendering behavior. If not provided, default components will be used.\n     */\n    components:\n      | {\n          /** Component for rendering empty messages */\n          Empty?: EmptyMessagePartComponent | undefined;\n          /** Component for rendering text content */\n          Text?: TextMessagePartComponent | undefined;\n          /** Component for rendering reasoning content (typically hidden) */\n          Reasoning?: ReasoningMessagePartComponent | undefined;\n          /** Component for rendering source content */\n          Source?: SourceMessagePartComponent | undefined;\n          /** Component for rendering image content */\n          Image?: ImageMessagePartComponent | undefined;\n          /** Component for rendering file content */\n          File?: FileMessagePartComponent | undefined;\n          /** Component for rendering audio content (experimental) */\n          Unstable_Audio?: Unstable_AudioMessagePartComponent | undefined;\n          /** Configuration for data part rendering */\n          data?:\n            | {\n                /** Map data event names to specific components */\n                by_name?:\n                  | Record<string, DataMessagePartComponent | undefined>\n                  | undefined;\n                /** Fallback component for unmatched data events */\n                Fallback?: DataMessagePartComponent | undefined;\n              }\n            | undefined;\n          /** Configuration for tool call rendering */\n          tools?:\n            | {\n                /** Map of tool names to their specific components */\n                by_name?:\n                  | Record<string, ToolCallMessagePartComponent | undefined>\n                  | undefined;\n                /** Fallback component for unregistered tools */\n                Fallback?: ComponentType<ToolCallMessagePartProps> | undefined;\n              }\n            | {\n                /** Override component that handles all tool calls */\n                Override: ComponentType<ToolCallMessagePartProps>;\n              }\n            | undefined;\n\n          /**\n           * Component for rendering grouped message parts.\n           *\n           * When provided, this component will automatically wrap message parts that share\n           * the same group key as determined by the groupingFunction.\n           *\n           * The component receives:\n           * - `groupKey`: The group key (or undefined for ungrouped parts)\n           * - `indices`: Array of indices for the parts in this group\n           * - `children`: The rendered message part components\n           *\n           * @example\n           * ```tsx\n           * // Collapsible group\n           * Group: ({ groupKey, indices, children }) => {\n           *   if (!groupKey) return <>{children}</>;\n           *   return (\n           *     <details className=\"message-group\">\n           *       <summary>\n           *         Group {groupKey} ({indices.length} parts)\n           *       </summary>\n           *       <div className=\"group-content\">\n           *         {children}\n           *       </div>\n           *     </details>\n           *   );\n           * }\n           * ```\n           *\n           * @param groupKey - The group key (undefined for ungrouped parts)\n           * @param indices - Array of indices for the parts in this group\n           * @param children - Rendered message part components to display within the group\n           */\n          Group?: ComponentType<\n            PropsWithChildren<{\n              groupKey: string | undefined;\n              indices: number[];\n            }>\n          >;\n        }\n      | undefined;\n  };\n}\n\nconst ToolUIDisplay = ({\n  Fallback,\n  ...props\n}: {\n  Fallback: ToolCallMessagePartComponent | undefined;\n} & ToolCallMessagePartProps) => {\n  const Render = useAuiState((s) => {\n    const Render = s.tools.tools[props.toolName] ?? Fallback;\n    if (Array.isArray(Render)) return Render[0] ?? Fallback;\n    return Render;\n  });\n  if (!Render) return null;\n  return <Render {...props} />;\n};\n\nconst DataUIDisplay = ({\n  Fallback,\n  ...props\n}: {\n  Fallback: DataMessagePartComponent | undefined;\n} & DataMessagePartProps) => {\n  const Render = useAuiState((s) => {\n    const Render = s.dataRenderers.renderers[props.name] ?? Fallback;\n    if (Array.isArray(Render)) return Render[0] ?? Fallback;\n    return Render;\n  });\n  if (!Render) return null;\n  return <Render {...props} />;\n};\n\nconst defaultComponents = {\n  Text: () => (\n    <p style={{ whiteSpace: \"pre-line\" }}>\n      <MessagePartPrimitiveText />\n      <MessagePartPrimitiveInProgress>\n        <span style={{ fontFamily: \"revert\" }}>{\" \\u25CF\"}</span>\n      </MessagePartPrimitiveInProgress>\n    </p>\n  ),\n  Reasoning: () => null,\n  Source: () => null,\n  Image: () => <MessagePartPrimitiveImage />,\n  File: () => null,\n  Unstable_Audio: () => null,\n  Group: ({ children }) => children,\n} satisfies MessagePrimitiveUnstable_PartsGrouped.Props[\"components\"];\n\ntype MessagePartComponentProps = {\n  components: MessagePrimitiveUnstable_PartsGrouped.Props[\"components\"];\n};\n\nconst MessagePartComponent: FC<MessagePartComponentProps> = ({\n  components: {\n    Text = defaultComponents.Text,\n    Reasoning = defaultComponents.Reasoning,\n    Image = defaultComponents.Image,\n    Source = defaultComponents.Source,\n    File = defaultComponents.File,\n    Unstable_Audio: Audio = defaultComponents.Unstable_Audio,\n    tools = {},\n    data,\n  } = {},\n}) => {\n  const aui = useAui();\n  const part = useAuiState((s) => s.part);\n\n  const type = part.type;\n  if (type === \"tool-call\") {\n    const addResult = aui.part().addToolResult;\n    const resume = aui.part().resumeToolCall;\n    if (\"Override\" in tools)\n      return <tools.Override {...part} addResult={addResult} resume={resume} />;\n    const Tool = tools.by_name?.[part.toolName] ?? tools.Fallback;\n    return (\n      <ToolUIDisplay\n        {...part}\n        Fallback={Tool}\n        addResult={addResult}\n        resume={resume}\n      />\n    );\n  }\n\n  if (part.status?.type === \"requires-action\")\n    throw new Error(\"Encountered unexpected requires-action status\");\n\n  switch (type) {\n    case \"text\":\n      return <Text {...part} />;\n\n    case \"reasoning\":\n      return <Reasoning {...part} />;\n\n    case \"source\":\n      return <Source {...part} />;\n\n    case \"image\":\n      return <Image {...part} />;\n\n    case \"file\":\n      return <File {...part} />;\n\n    case \"audio\":\n      return <Audio {...part} />;\n\n    case \"data\": {\n      const Data = data?.by_name?.[part.name] ?? data?.Fallback;\n      return <DataUIDisplay {...part} Fallback={Data} />;\n    }\n\n    default:\n      console.warn(`Unknown message part type: ${type}`);\n      return null;\n  }\n};\n\ntype MessagePartProps = {\n  partIndex: number;\n  components: MessagePrimitiveUnstable_PartsGrouped.Props[\"components\"];\n};\n\nconst MessagePartImpl: FC<MessagePartProps> = ({ partIndex, components }) => {\n  return (\n    <PartByIndexProvider index={partIndex}>\n      <MessagePartComponent components={components} />\n    </PartByIndexProvider>\n  );\n};\n\nconst MessagePart = memo(\n  MessagePartImpl,\n  (prev, next) =>\n    prev.partIndex === next.partIndex &&\n    prev.components?.Text === next.components?.Text &&\n    prev.components?.Reasoning === next.components?.Reasoning &&\n    prev.components?.Source === next.components?.Source &&\n    prev.components?.Image === next.components?.Image &&\n    prev.components?.File === next.components?.File &&\n    prev.components?.Unstable_Audio === next.components?.Unstable_Audio &&\n    prev.components?.tools === next.components?.tools &&\n    prev.components?.data === next.components?.data &&\n    prev.components?.Group === next.components?.Group,\n);\n\nconst EmptyPartFallback: FC<{\n  status: MessagePartStatus;\n  component: TextMessagePartComponent;\n}> = ({ status, component: Component }) => {\n  return (\n    <TextMessagePartProvider text=\"\" isRunning={status.type === \"running\"}>\n      <Component type=\"text\" text=\"\" status={status} />\n    </TextMessagePartProvider>\n  );\n};\n\nconst COMPLETE_STATUS: MessagePartStatus = Object.freeze({\n  type: \"complete\",\n});\n\nconst EmptyPartsImpl: FC<MessagePartComponentProps> = ({ components }) => {\n  const status = useAuiState(\n    (s) => (s.message.status ?? COMPLETE_STATUS) as MessagePartStatus,\n  );\n\n  if (components?.Empty) return <components.Empty status={status} />;\n\n  return (\n    <EmptyPartFallback\n      status={status}\n      component={components?.Text ?? defaultComponents.Text}\n    />\n  );\n};\n\nconst EmptyParts = memo(\n  EmptyPartsImpl,\n  (prev, next) =>\n    prev.components?.Empty === next.components?.Empty &&\n    prev.components?.Text === next.components?.Text,\n);\n\n/**\n * Renders the parts of a message grouped by a custom grouping function.\n *\n * This component allows you to group message parts based on any criteria you define.\n * The grouping function receives all message parts and returns an array of groups,\n * where each group has a key and an array of part indices.\n *\n * @deprecated Prefer `<MessagePrimitive.GroupedParts>` for adjacent\n * grouping — it dispatches all rendering through one `switch (part.type)`\n * and supports nested group paths. Keep this primitive only for\n * non-adjacent clustering (e.g., gathering parts with the same parent-id\n * across the message).\n *\n * @example\n * ```tsx\n * // Group by parent ID (default behavior)\n * <MessagePrimitive.Unstable_PartsGrouped\n *   components={{\n *     Text: ({ text }) => <p className=\"message-text\">{text}</p>,\n *     Image: ({ image }) => <img src={image} alt=\"Message image\" />,\n *     Group: ({ groupKey, indices, children }) => {\n *       if (!groupKey) return <>{children}</>;\n *       return (\n *         <div className=\"parent-group border rounded p-4\">\n *           <h4>Parent ID: {groupKey}</h4>\n *           {children}\n *         </div>\n *       );\n *     }\n *   }}\n * />\n * ```\n */\nexport const MessagePrimitiveUnstable_PartsGrouped: FC<\n  MessagePrimitiveUnstable_PartsGrouped.Props\n> = ({ groupingFunction, components }) => {\n  const contentLength = useAuiState((s) => s.message.parts.length);\n  const messageGroups = useMessagePartsGrouped(groupingFunction);\n\n  const partsElements = useMemo(() => {\n    if (contentLength === 0) {\n      return <EmptyParts components={components} />;\n    }\n\n    return messageGroups.map((group, groupIndex) => {\n      const GroupComponent = components?.Group ?? defaultComponents.Group;\n\n      return (\n        <GroupComponent\n          key={`group-${groupIndex}-${group.groupKey ?? \"ungrouped\"}`}\n          groupKey={group.groupKey}\n          indices={group.indices}\n        >\n          {group.indices.map((partIndex) => (\n            <MessagePart\n              key={partIndex}\n              partIndex={partIndex}\n              components={components}\n            />\n          ))}\n        </GroupComponent>\n      );\n    });\n  }, [messageGroups, components, contentLength]);\n\n  return <>{partsElements}</>;\n};\n\nMessagePrimitiveUnstable_PartsGrouped.displayName =\n  \"MessagePrimitive.Unstable_PartsGrouped\";\n\n/**\n * Renders the parts of a message grouped by their parent ID.\n * This is a convenience wrapper around Unstable_PartsGrouped with parent ID grouping.\n *\n * @deprecated Use MessagePrimitive.Unstable_PartsGrouped instead for more flexibility\n */\nexport const MessagePrimitiveUnstable_PartsGroupedByParentId: FC<\n  Omit<MessagePrimitiveUnstable_PartsGrouped.Props, \"groupingFunction\">\n> = ({ components, ...props }) => {\n  return (\n    <MessagePrimitiveUnstable_PartsGrouped\n      {...props}\n      components={components}\n      groupingFunction={groupMessagePartsByParentId}\n    />\n  );\n};\n\nMessagePrimitiveUnstable_PartsGroupedByParentId.displayName =\n  \"MessagePrimitive.Unstable_PartsGroupedByParentId\";\n"],"mappings":";;;;;;;;;;;;;;;AA0CA,MAAM,+BACJ,UACuB;CAEvB,MAAM,2BAAW,IAAI,IAAsB;CAG3C,KAAK,IAAI,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;EAKrC,MAAM,UAJO,MAAM,IACI,YAGK,eAAe;EAG3C,MAAM,UAAU,SAAS,IAAI,OAAO,KAAK,CAAC;EAC1C,QAAQ,KAAK,CAAC;EACd,SAAS,IAAI,SAAS,OAAO;CAC/B;CAGA,MAAM,SAA6B,CAAC;CACpC,KAAK,MAAM,CAAC,SAAS,YAAY,UAAU;EAEzC,MAAM,WAAW,QAAQ,WAAW,cAAc,IAAI,KAAA,IAAY;EAClE,OAAO,KAAK;GAAE;GAAU;EAAQ,CAAC;CACnC;CAEA,OAAO;AACT;AAEA,MAAM,0BACJ,qBACuB;CACvB,MAAM,QAAQ,aAAa,MAAM,EAAE,QAAQ,KAAK;CAEhD,OAAO,cAAc;EACnB,IAAI,MAAM,WAAW,GACnB,OAAO,CAAC;EAEV,OAAO,iBAAiB,KAAK;CAC/B,GAAG,CAAC,OAAO,gBAAgB,CAAC;AAC9B;AA8IA,MAAM,iBAAiB,EACrB,UACA,GAAG,YAG4B;CAC/B,MAAM,SAAS,aAAa,MAAM;EAChC,MAAM,SAAS,EAAE,MAAM,MAAM,MAAM,aAAa;EAChD,IAAI,MAAM,QAAQ,MAAM,GAAG,OAAO,OAAO,MAAM;EAC/C,OAAO;CACT,CAAC;CACD,IAAI,CAAC,QAAQ,OAAO;CACpB,OAAO,oBAAC,QAAD,EAAQ,GAAI,MAAQ,CAAA;AAC7B;AAEA,MAAM,iBAAiB,EACrB,UACA,GAAG,YAGwB;CAC3B,MAAM,SAAS,aAAa,MAAM;EAChC,MAAM,SAAS,EAAE,cAAc,UAAU,MAAM,SAAS;EACxD,IAAI,MAAM,QAAQ,MAAM,GAAG,OAAO,OAAO,MAAM;EAC/C,OAAO;CACT,CAAC;CACD,IAAI,CAAC,QAAQ,OAAO;CACpB,OAAO,oBAAC,QAAD,EAAQ,GAAI,MAAQ,CAAA;AAC7B;AAEA,MAAM,oBAAoB;CACxB,YACE,qBAAC,KAAD;EAAG,OAAO,EAAE,YAAY,WAAW;YAAnC,CACE,oBAAC,0BAAD,CAA2B,CAAA,GAC3B,oBAAC,gCAAD,EAAA,UACE,oBAAC,QAAD;GAAM,OAAO,EAAE,YAAY,SAAS;aAAI;EAAgB,CAAA,EAC1B,CAAA,CAC/B;;CAEL,iBAAiB;CACjB,cAAc;CACd,aAAa,oBAAC,2BAAD,CAA4B,CAAA;CACzC,YAAY;CACZ,sBAAsB;CACtB,QAAQ,EAAE,eAAe;AAC3B;AAMA,MAAM,wBAAuD,EAC3D,YAAY,EACV,OAAO,kBAAkB,MACzB,YAAY,kBAAkB,WAC9B,QAAQ,kBAAkB,OAC1B,SAAS,kBAAkB,QAC3B,OAAO,kBAAkB,MACzB,gBAAgB,QAAQ,kBAAkB,gBAC1C,QAAQ,CAAC,GACT,SACE,CAAC,QACD;CACJ,MAAM,MAAM,OAAO;CACnB,MAAM,OAAO,aAAa,MAAM,EAAE,IAAI;CAEtC,MAAM,OAAO,KAAK;CAClB,IAAI,SAAS,aAAa;EACxB,MAAM,YAAY,IAAI,KAAK,EAAE;EAC7B,MAAM,SAAS,IAAI,KAAK,EAAE;EAC1B,IAAI,cAAc,OAChB,OAAO,oBAAC,MAAM,UAAP;GAAgB,GAAI;GAAiB;GAAmB;EAAS,CAAA;EAC1E,MAAM,OAAO,MAAM,UAAU,KAAK,aAAa,MAAM;EACrD,OACE,oBAAC,eAAD;GACE,GAAI;GACJ,UAAU;GACC;GACH;EACT,CAAA;CAEL;CAEA,IAAI,KAAK,QAAQ,SAAS,mBACxB,MAAM,IAAI,MAAM,+CAA+C;CAEjE,QAAQ,MAAR;EACE,KAAK,QACH,OAAO,oBAAC,MAAD,EAAM,GAAI,KAAO,CAAA;EAE1B,KAAK,aACH,OAAO,oBAAC,WAAD,EAAW,GAAI,KAAO,CAAA;EAE/B,KAAK,UACH,OAAO,oBAAC,QAAD,EAAQ,GAAI,KAAO,CAAA;EAE5B,KAAK,SACH,OAAO,oBAAC,OAAD,EAAO,GAAI,KAAO,CAAA;EAE3B,KAAK,QACH,OAAO,oBAAC,MAAD,EAAM,GAAI,KAAO,CAAA;EAE1B,KAAK,SACH,OAAO,oBAAC,OAAD,EAAO,GAAI,KAAO,CAAA;EAE3B,KAAK,QAAQ;GACX,MAAM,OAAO,MAAM,UAAU,KAAK,SAAS,MAAM;GACjD,OAAO,oBAAC,eAAD;IAAe,GAAI;IAAM,UAAU;GAAO,CAAA;EACnD;EAEA;GACE,QAAQ,KAAK,8BAA8B,MAAM;GACjD,OAAO;CACX;AACF;AAOA,MAAM,mBAAyC,EAAE,WAAW,iBAAiB;CAC3E,OACE,oBAAC,qBAAD;EAAqB,OAAO;YAC1B,oBAAC,sBAAD,EAAkC,WAAa,CAAA;CAC5B,CAAA;AAEzB;AAEA,MAAM,cAAc,KAClB,kBACC,MAAM,SACL,KAAK,cAAc,KAAK,aACxB,KAAK,YAAY,SAAS,KAAK,YAAY,QAC3C,KAAK,YAAY,cAAc,KAAK,YAAY,aAChD,KAAK,YAAY,WAAW,KAAK,YAAY,UAC7C,KAAK,YAAY,UAAU,KAAK,YAAY,SAC5C,KAAK,YAAY,SAAS,KAAK,YAAY,QAC3C,KAAK,YAAY,mBAAmB,KAAK,YAAY,kBACrD,KAAK,YAAY,UAAU,KAAK,YAAY,SAC5C,KAAK,YAAY,SAAS,KAAK,YAAY,QAC3C,KAAK,YAAY,UAAU,KAAK,YAAY,KAChD;AAEA,MAAM,qBAGA,EAAE,QAAQ,WAAW,gBAAgB;CACzC,OACE,oBAAC,yBAAD;EAAyB,MAAK;EAAG,WAAW,OAAO,SAAS;YAC1D,oBAAC,WAAD;GAAW,MAAK;GAAO,MAAK;GAAW;EAAS,CAAA;CACzB,CAAA;AAE7B;AAEA,MAAM,kBAAqC,OAAO,OAAO,EACvD,MAAM,WACR,CAAC;AAED,MAAM,kBAAiD,EAAE,iBAAiB;CACxE,MAAM,SAAS,aACZ,MAAO,EAAE,QAAQ,UAAU,eAC9B;CAEA,IAAI,YAAY,OAAO,OAAO,oBAAC,WAAW,OAAZ,EAA0B,OAAS,CAAA;CAEjE,OACE,oBAAC,mBAAD;EACU;EACR,WAAW,YAAY,QAAQ,kBAAkB;CAClD,CAAA;AAEL;AAEA,MAAM,aAAa,KACjB,iBACC,MAAM,SACL,KAAK,YAAY,UAAU,KAAK,YAAY,SAC5C,KAAK,YAAY,SAAS,KAAK,YAAY,IAC/C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmCA,MAAa,yCAER,EAAE,kBAAkB,iBAAiB;CACxC,MAAM,gBAAgB,aAAa,MAAM,EAAE,QAAQ,MAAM,MAAM;CAC/D,MAAM,gBAAgB,uBAAuB,gBAAgB;CA4B7D,OAAO,oBAAA,UAAA,EAAA,UA1Be,cAAc;EAClC,IAAI,kBAAkB,GACpB,OAAO,oBAAC,YAAD,EAAwB,WAAa,CAAA;EAG9C,OAAO,cAAc,KAAK,OAAO,eAAe;GAG9C,OACE,oBAHqB,YAAY,SAAS,kBAAkB,OAG5D;IAEE,UAAU,MAAM;IAChB,SAAS,MAAM;cAEd,MAAM,QAAQ,KAAK,cAClB,oBAAC,aAAD;KAEa;KACC;IACb,GAHM,SAGN,CACF;GACa,GAXT,SAAS,WAAW,GAAG,MAAM,YAAY,aAWhC;EAEpB,CAAC;CACH,GAAG;EAAC;EAAe;EAAY;CAAa,CAEtB,EAAI,CAAA;AAC5B;AAEA,sCAAsC,cACpC;;;;;;;AAQF,MAAa,mDAER,EAAE,YAAY,GAAG,YAAY;CAChC,OACE,oBAAC,uCAAD;EACE,GAAI;EACQ;EACZ,kBAAkB;CACnB,CAAA;AAEL;AAEA,gDAAgD,cAC9C"}

package/dist/primitives/message/MessageRoot.d.ts

@@ -1,8 +1,10 @@
 import { Primitive } from "../../utils/Primitive.js";
-import { type ComponentRef, type ComponentPropsWithoutRef } from "react";
-export declare namespace MessagePrimitiveRoot {
-    type Element = ComponentRef<typeof Primitive.div>;
-    type Props = ComponentPropsWithoutRef<typeof Primitive.div>;
+import { ComponentPropsWithoutRef, ComponentRef } from "react";
+
+//#region src/primitives/message/MessageRoot.d.ts
+declare namespace MessagePrimitiveRoot {
+  type Element = ComponentRef<typeof Primitive.div>;
+  type Props = ComponentPropsWithoutRef<typeof Primitive.div>;
 }
 /**
  * The root container component for a message.
@@ -27,9 +29,11 @@ export declare namespace MessagePrimitiveRoot {
  * </MessagePrimitive.Root>
  * ```
  */
-export declare const MessagePrimitiveRoot: import("react").ForwardRefExoticComponent<Omit<Omit<import("react").ClassAttributes<HTMLDivElement> & import("react").HTMLAttributes<HTMLDivElement> & {
-    asChild?: boolean;
+declare const MessagePrimitiveRoot: import("react").ForwardRefExoticComponent<Omit<Omit<import("react").ClassAttributes<HTMLDivElement> & import("react").HTMLAttributes<HTMLDivElement> & {
+  asChild?: boolean;
 }, "ref"> & {
-    render?: import("react").ReactElement | undefined;
+  render?: import("react").ReactElement | undefined;
 } & import("react").RefAttributes<HTMLDivElement>, "ref"> & import("react").RefAttributes<HTMLDivElement>>;
+//#endregion
+export { MessagePrimitiveRoot };
 //# sourceMappingURL=MessageRoot.d.ts.map

package/dist/primitives/message/MessageRoot.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageRoot.d.ts","sourceRoot":"","sources":["../../../src/primitives/message/MessageRoot.tsx"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,iCAA8B;AAClD,OAAO,EACL,KAAK,YAAY,EAEjB,KAAK,wBAAwB,EAG9B,MAAM,OAAO,CAAC;AA8Gf,yBAAiB,oBAAoB,CAAC;IACpC,KAAY,OAAO,GAAG,YAAY,CAAC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;IACzD,KAAY,KAAK,GAAG,wBAAwB,CAAC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;CACpE;AAsDD;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,eAAO,MAAM,oBAAoB;;;;0GAkB/B,CAAC"}
+{"version":3,"file":"MessageRoot.d.ts","names":[],"sources":["../../../src/primitives/message/MessageRoot.tsx"],"mappings":";;;;kBAuHiB,oBAAA;EAAA,KACH,OAAA,GAAU,YAAA,QAAoB,SAAA,CAAU,GAAA;EAAA,KACxC,KAAA,GAAQ,wBAAA,QAAgC,SAAA,CAAU,GAAA;AAAA;;;;;;;;;;;;;;;;;AAAG;AA8EnE;;;;;;cAAa,oBAAA,kBAAoB,yBAAA,CAAA,IAAA,CAAA,IAAA,iBAAA,eAAA,CAAA,cAAA,oBAAA,cAAA,CAAA,cAAA"}