From 50478c600f735090086b53eccaeeb947a51baf30 Mon Sep 17 00:00:00 2001
From: ousugo <dkzyxh@gmail.com>
Date: Sat, 22 Feb 2025 12:57:24 +0800
Subject: [PATCH] fix: Regenerat messages don't use @ specified models

---
 src/renderer/src/pages/home/Messages/MessageMenubar.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/renderer/src/pages/home/Messages/MessageMenubar.tsx b/src/renderer/src/pages/home/Messages/MessageMenubar.tsx
index 17d51e4f..4831c4b8 100644
--- a/src/renderer/src/pages/home/Messages/MessageMenubar.tsx
+++ b/src/renderer/src/pages/home/Messages/MessageMenubar.tsx
@@ -263,7 +263,7 @@ const MessageMenubar: FC<Props> = (props) => {
   const onRegenerate = async (e: React.MouseEvent | undefined) => {
     e?.stopPropagation?.()
     await modelGenerating()
-    const selectedModel = isGrouped ? model : assistantModel
+    const selectedModel = message.model || (isGrouped ? model : assistantModel)
     const _message = resetAssistantMessage(message, selectedModel)
     onEditMessage?.(_message)
   }