From 81a9e6c1fe0d7c59d87832604e8f14a379e82f90 Mon Sep 17 00:00:00 2001
From: MyPrototypeWhat <43230886+MyPrototypeWhat@users.noreply.github.com>
Date: Sat, 3 May 2025 20:39:40 +0800
Subject: [PATCH] =?UTF-8?q?fix(messageThunk):=20enhance=20reset=20message?=
 =?UTF-8?q?=20logic=20to=20include=20model=20for=20s=E2=80=A6=20(#5632)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

fix(messageThunk): enhance reset message logic to include model for single message resets
---
 src/renderer/src/store/thunk/messageThunk.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/renderer/src/store/thunk/messageThunk.ts b/src/renderer/src/store/thunk/messageThunk.ts
index 3cd8be0b3c..78bb375f27 100644
--- a/src/renderer/src/store/thunk/messageThunk.ts
+++ b/src/renderer/src/store/thunk/messageThunk.ts
@@ -901,7 +901,8 @@ export const resendMessageThunk =
       for (const originalMsg of assistantMessagesToReset) {
         const blockIdsToDelete = [...(originalMsg.blocks || [])]
         const resetMsg = resetAssistantMessage(originalMsg, {
-          status: AssistantMessageStatus.PENDING
+          status: AssistantMessageStatus.PENDING,
+          ...(assistantMessagesToReset.length === 1 ? { model: assistant.model } : {})
         })
 
         resetDataList.push({ resetMsg })