@assistant-ui/core

import { jsx as _jsx } from "react/jsx-runtime"; import { memo, useMemo, } from "react"; import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store"; import { MessageByIndexProvider } from "../../providers/MessageByIndexProvider.js"; const isComponentsSame = (prev, next) => { return (prev.Message === next.Message && prev.EditComposer === next.EditComposer && prev.UserEditComposer === next.UserEditComposer && prev.AssistantEditComposer === next.AssistantEditComposer && prev.SystemEditComposer === next.SystemEditComposer && prev.UserMessage === next.UserMessage && prev.AssistantMessage === next.AssistantMessage && prev.SystemMessage === next.SystemMessage); }; const DEFAULT_SYSTEM_MESSAGE = () => null; const getComponent = (components, role, isEditing) => { switch (role) { case "user": if (isEditing) { return (components.UserEditComposer ?? components.EditComposer ?? components.UserMessage ?? components.Message); } else { return components.UserMessage ?? components.Message; } case "assistant": if (isEditing) { return (components.AssistantEditComposer ?? components.EditComposer ?? components.AssistantMessage ?? components.Message); } else { return (components.AssistantMessage ?? components.Message); } case "system": if (isEditing) { return (components.SystemEditComposer ?? components.EditComposer ?? components.SystemMessage ?? components.Message); } else { return (components.SystemMessage ?? components.Message ?? DEFAULT_SYSTEM_MESSAGE); } default: const _exhaustiveCheck = role; throw new Error(Unknown message role: ${_exhaustiveCheck}); } }; const ThreadMessageComponent = ({ components, }) => { const role = useAuiState((s) => s.message.role); const isEditing = useAuiState((s) => s.message.composer.isEditing); const Component = getComponent(components, role, isEditing); return _jsx(Component, {}); }; /**

• Renders a single message at the specified index in the current thread. / export const ThreadPrimitiveMessageByIndex = memo(({ index, components }) => { return (_jsx(MessageByIndexProvider, { index: index, children: jsx(ThreadMessageComponent, { components: components }) })); }, (prev, next) => prev.index === next.index && isComponentsSame(prev.components, next.components)); ThreadPrimitiveMessageByIndex.displayName = "ThreadPrimitive.MessageByIndex"; const ThreadPrimitiveMessagesInner = ({ children }) => { const messagesLength = useAuiState((s) => s.thread.messages.length); return useMemo(() => { if (messagesLength === 0) return null; return Array.from({ length: messagesLength }, (, index) => (_jsx(MessageByIndexProvider, { index: index, children: _jsx(RenderChildrenWithAccessor, { getItemState: (aui) => aui.thread().message({ index }).getState(), children: (getItem) => children({ get message() { return getItem(); }, }) }) }, index))); }, [messagesLength, children]); }; /*
• Renders all messages in the current thread.
• @example
• <ThreadPrimitive.Messages>
• {({ message }) => {

• if (message.role === "user") return <MyUserMessage />;
  

• return <MyAssistantMessage />;
  

• }}
• </ThreadPrimitive.Messages>

*/ export const ThreadPrimitiveMessagesImpl = ({ components, children, }) => { if (components) { return (_jsx(ThreadPrimitiveMessagesInner, { children: () => _jsx(ThreadMessageComponent, { components: components }) })); } return (_jsx(ThreadPrimitiveMessagesInner, { children: children })); }; ThreadPrimitiveMessagesImpl.displayName = "ThreadPrimitive.Messages"; export const ThreadPrimitiveMessages = memo(ThreadPrimitiveMessagesImpl, (prev, next) => { if (prev.children || next.children) { return prev.children === next.children; } return isComponentsSame(prev.components, next.components); }); //# sourceMappingURL=ThreadMessages.js.map