From 9ae3c752ddcfd83009e87b3184bf40ceb1dac29f Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Thu, 11 Jul 2024 23:08:55 +0800
Subject: [PATCH] fix: regenerate message logic

---
 src/renderer/src/pages/home/components/Message.tsx | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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<Props> = ({ 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<Props> = ({ message, index, showMenu, onDeleteMessage }) =
             <Tooltip title="Delete" mouseEnterDelay={0.8}>
               <DeleteOutlined onClick={onDelete} />
             </Tooltip>
-            {isLastMessage && (
+            {canRegenerate && (
               <Tooltip title="Regenerate" mouseEnterDelay={0.8}>
                 <SyncOutlined onClick={onRegenerate} />
               </Tooltip>