// Ported from vercel/ai-elements @ main (packages/elements/src/message.tsx). // Customizations applied on import: import-alias rewrites, RTL physical→logical // class swap, Vercel theme-token sweep. See scripts/registry/port-ai-elements.mjs // for the deterministic transforms — re-run to re-sync against upstream. "use client"; import { cjk } from "@streamdown/cjk"; import { code } from "@streamdown/code"; import { math } from "@streamdown/math"; import { mermaid } from "@streamdown/mermaid"; import type { UIMessage } from "ai"; import { ChevronLeftIcon, ChevronRightIcon } from "lucide-react"; import type { ComponentProps, HTMLAttributes, ReactElement } from "react"; import { createContext, memo, useCallback, useContext, useEffect, useMemo, useState, } from "react"; import { type PluginConfig, Streamdown } from "streamdown"; import { Button } from "@/components/registry/primitives/core/button"; import { ButtonGroup, ButtonGroupText, } from "@/components/registry/primitives/core/button-group"; import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger, } from "@/components/registry/primitives/core/tooltip"; import { cn } from "@/lib/registry/cn"; export type MessageProps = HTMLAttributes & { from: UIMessage["role"]; }; export const Message = ({ className, from, ...props }: MessageProps) => (

); export type MessageContentProps = HTMLAttributes; export const MessageContent = ({ children, className, ...props }: MessageContentProps) => (

{children}

); export type MessageActionsProps = ComponentProps<"div">; export const MessageActions = ({ className, children, ...props }: MessageActionsProps) => (

{children}

); export type MessageActionProps = ComponentProps & { tooltip?: string; label?: string; }; export const MessageAction = ({ tooltip, children, label, variant = "ghost", size = "icon-sm", ...props }: MessageActionProps) => { const button = ( ); if (tooltip) { return ( {button}

{tooltip}

); } return button; }; interface MessageBranchContextType { currentBranch: number; totalBranches: number; goToPrevious: () => void; goToNext: () => void; branches: ReactElement[]; setBranches: (branches: ReactElement[]) => void; } const MessageBranchContext = createContext( null, ); const useMessageBranch = () => { const context = useContext(MessageBranchContext); if (!context) { throw new Error( "MessageBranch components must be used within MessageBranch", ); } return context; }; export type MessageBranchProps = HTMLAttributes & { defaultBranch?: number; onBranchChange?: (branchIndex: number) => void; }; export const MessageBranch = ({ defaultBranch = 0, onBranchChange, className, ...props }: MessageBranchProps) => { const [currentBranch, setCurrentBranch] = useState(defaultBranch); const [branches, setBranches] = useState([]); const handleBranchChange = useCallback( (newBranch: number) => { setCurrentBranch(newBranch); onBranchChange?.(newBranch); }, [onBranchChange], ); const goToPrevious = useCallback(() => { const newBranch = currentBranch > 0 ? currentBranch - 1 : branches.length - 1; handleBranchChange(newBranch); }, [currentBranch, branches.length, handleBranchChange]); const goToNext = useCallback(() => { const newBranch = currentBranch < branches.length - 1 ? currentBranch + 1 : 0; handleBranchChange(newBranch); }, [currentBranch, branches.length, handleBranchChange]); const contextValue = useMemo( () => ({ branches, currentBranch, goToNext, goToPrevious, setBranches, totalBranches: branches.length, }), [branches, currentBranch, goToNext, goToPrevious], ); return (

div]:pb-0", className)} {...props} /> ); }; export type MessageBranchContentProps = HTMLAttributes; export const MessageBranchContent = ({ children, ...props }: MessageBranchContentProps) => { const { currentBranch, setBranches, branches } = useMessageBranch(); const childrenArray = useMemo( () => (Array.isArray(children) ? children : [children]), [children], ); // Use useEffect to update branches when they change useEffect(() => { if (branches.length !== childrenArray.length) { setBranches(childrenArray); } }, [childrenArray, branches, setBranches]); return childrenArray.map((branch, index) => (

div]:pb-0", index === currentBranch ? "block" : "hidden", )} key={branch.key} {...props} > {branch}

)); }; export type MessageBranchSelectorProps = ComponentProps; export const MessageBranchSelector = ({ className, ...props }: MessageBranchSelectorProps) => { const { totalBranches } = useMessageBranch(); // Don't render if there's only one branch if (totalBranches <= 1) { return null; } return ( *:not(:first-child)]:rounded-s-md [&>*:not(:last-child)]:rounded-e-md", className, )} orientation="horizontal" {...props} /> ); }; export type MessageBranchPreviousProps = ComponentProps; export const MessageBranchPrevious = ({ children, ...props }: MessageBranchPreviousProps) => { const { goToPrevious, totalBranches } = useMessageBranch(); return ( ); }; export type MessageBranchNextProps = ComponentProps; export const MessageBranchNext = ({ children, ...props }: MessageBranchNextProps) => { const { goToNext, totalBranches } = useMessageBranch(); return ( ); }; export type MessageBranchPageProps = HTMLAttributes; export const MessageBranchPage = ({ className, ...props }: MessageBranchPageProps) => { const { currentBranch, totalBranches } = useMessageBranch(); return ( {currentBranch + 1} of {totalBranches} ); }; export type MessageResponseProps = ComponentProps; const streamdownPlugins = { cjk, code, math, mermaid } as PluginConfig; export const MessageResponse = memo( ({ className, ...props }: MessageResponseProps) => ( *:first-child]:mt-0 [&>*:last-child]:mb-0", className, )} plugins={streamdownPlugins} {...props} /> ), (prevProps, nextProps) => prevProps.children === nextProps.children && nextProps.isAnimating === prevProps.isAnimating, ); MessageResponse.displayName = "MessageResponse"; export type MessageToolbarProps = ComponentProps<"div">; export const MessageToolbar = ({ className, children, ...props }: MessageToolbarProps) => (

{children}

);

Message

Showcase

Message