From 6beee78ce80b77350ba9aaf22f9cbc51b6a52d7b Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Wed, 11 Dec 2024 20:01:52 +0800
Subject: [PATCH] fix: can not delete last message

---
 src/renderer/src/pages/home/Messages/MessageHeader.tsx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/pages/home/Messages/MessageHeader.tsx b/src/renderer/src/pages/home/Messages/MessageHeader.tsx
index 9736da68..cdccd24b 100644
--- a/src/renderer/src/pages/home/Messages/MessageHeader.tsx
+++ b/src/renderer/src/pages/home/Messages/MessageHeader.tsx
@@ -35,9 +35,9 @@ const MessageHeader: FC<Props> = memo(({ assistant, model, message }) => {
 
   const getUserName = useCallback(() => {
     if (isLocalAi && message.role !== 'user') return APP_NAME
-    if (message.role === 'assistant') return model?.name || model?.id || ''
+    if (message.role === 'assistant') return model?.name || model?.id || message.modelId || ''
     return userName || t('common.you')
-  }, [message.role, model?.id, model?.name, t, userName])
+  }, [message.modelId, message.role, model?.id, model?.name, t, userName])
 
   const isAssistantMessage = message.role === 'assistant'