From f9024eb07aeeafb08369d6f76a125bdb49cd2087 Mon Sep 17 00:00:00 2001
From: suyao <sy20010504@gmail.com>
Date: Thu, 27 Nov 2025 05:33:21 +0800
Subject: [PATCH] fix: simplify finish reason warning condition

---
 src/renderer/src/pages/home/Messages/MessageContent.tsx | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/renderer/src/pages/home/Messages/MessageContent.tsx b/src/renderer/src/pages/home/Messages/MessageContent.tsx
index 058a525ab0..f3741fbbe5 100644
--- a/src/renderer/src/pages/home/Messages/MessageContent.tsx
+++ b/src/renderer/src/pages/home/Messages/MessageContent.tsx
@@ -19,8 +19,7 @@ const MessageContent: React.FC<Props> = ({ message, onContinueGeneration, onDism
   const showFinishReasonWarning =
     message.role === 'assistant' &&
     message.finishReason &&
-    message.finishReason !== 'stop' &&
-    message.finishReason !== 'tool-calls'
+    !['stop', 'tool-calls', 'error'].includes(message.finishReason)
 
   const handleContinue = () => {
     onContinueGeneration?.(message)