diff --git a/packages/v1/react-ui/src/components/chat/Messages.tsx b/packages/v1/react-ui/src/components/chat/Messages.tsx index cae316a0097..ba892f4fdc6 100644 --- a/packages/v1/react-ui/src/components/chat/Messages.tsx +++ b/packages/v1/react-ui/src/components/chat/Messages.tsx @@ -87,11 +87,16 @@ export const Messages = ({ const LoadingIcon = () => {icons.activityIcon}; + const lastVisibleMessageIndex = messages.reduce( + (lastIdx, msg, idx) => (msg.role === "user" || msg.role === "assistant" ? idx : lastIdx), + -1, + ); + return (
{messages.map((message, index) => { - const isCurrentMessage = index === messages.length - 1; + const isCurrentMessage = index === lastVisibleMessageIndex; return ( ); })} - {messages[messages.length - 1]?.role === "user" && inProgress && ( + {messages[lastVisibleMessageIndex]?.role === "user" && inProgress && ( )} {interrupt}