diff --git a/src/components/chat/ChatMessage.tsx b/src/components/chat/ChatMessage.tsx index 1210f35..eab4b26 100644 --- a/src/components/chat/ChatMessage.tsx +++ b/src/components/chat/ChatMessage.tsx @@ -9,9 +9,10 @@ import { CheckCircle, XCircle } from "lucide-react"; interface ChatMessageProps { message: Message; + isLastMessage: boolean; } -const ChatMessage = ({ message }: ChatMessageProps) => { +const ChatMessage = ({ message, isLastMessage }: ChatMessageProps) => { const { isStreaming } = useStreamChat(); return (
{ : "bg-(--sidebar-accent)" }`} > - {message.role === "assistant" && !message.content && isStreaming ? ( + {message.role === "assistant" && + !message.content && + isStreaming && + isLastMessage ? (
{ suppressHydrationWarning > {message.role === "assistant" ? ( - + <> + + {isLastMessage && isStreaming && ( +
+
+
+
+
+ )} + ) : ( )} diff --git a/src/components/chat/MessagesList.tsx b/src/components/chat/MessagesList.tsx index 10a23e5..69a5997 100644 --- a/src/components/chat/MessagesList.tsx +++ b/src/components/chat/MessagesList.tsx @@ -36,7 +36,11 @@ export const MessagesList = forwardRef(
{messages.length > 0 ? ( messages.map((message, index) => ( - + )) ) : (