From 3fc7911c970fe475d94a61e3adda127ede4450f6 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Thu, 2 Jan 2025 13:41:51 +0800
Subject: [PATCH] feat: add new branch option to message menubar

---
 .../src/pages/home/Messages/MessageMenubar.tsx    | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/src/renderer/src/pages/home/Messages/MessageMenubar.tsx b/src/renderer/src/pages/home/Messages/MessageMenubar.tsx
index 13ec2a0f..2a7883ce 100644
--- a/src/renderer/src/pages/home/Messages/MessageMenubar.tsx
+++ b/src/renderer/src/pages/home/Messages/MessageMenubar.tsx
@@ -156,9 +156,15 @@ const MessageMenubar: FC<Props> = (props) => {
         key: 'edit',
         icon: <EditOutlined />,
         onClick: onEdit
+      },
+      {
+        label: t('chat.message.new.branch'),
+        key: 'new-branch',
+        icon: <ForkOutlined />,
+        onClick: onNewBranch
       }
     ],
-    [message, onEdit, t]
+    [message, onEdit, onNewBranch, t]
   )
 
   const onAtModelRegenerate = async () => {
@@ -212,13 +218,6 @@ const MessageMenubar: FC<Props> = (props) => {
           </ActionButton>
         </Tooltip>
       )}
-      {isAssistantMessage && (
-        <Tooltip title={t('chat.message.new.branch')} mouseEnterDelay={0.8}>
-          <ActionButton className="message-action-button" onClick={onNewBranch}>
-            <ForkOutlined />
-          </ActionButton>
-        </Tooltip>
-      )}
       {!isUserMessage && (
         <Dropdown
           menu={{