diff --git a/src/renderer/src/pages/home/components/Message.tsx b/src/renderer/src/pages/home/components/Message.tsx index 3b8e735c..4160deb6 100644 --- a/src/renderer/src/pages/home/components/Message.tsx +++ b/src/renderer/src/pages/home/components/Message.tsx @@ -24,6 +24,7 @@ const MessageItem: FC = ({ message, index, showMenu, onDeleteMessage }) = const avatar = useAvatar() const isLastMessage = index === 0 + const canRegenerate = isLastMessage && message.role === 'assistant' const onCopy = () => { navigator.clipboard.writeText(message.content) @@ -85,7 +86,7 @@ const MessageItem: FC = ({ message, index, showMenu, onDeleteMessage }) = - {isLastMessage && ( + {canRegenerate && (