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)