From eca231575d08b029a7fc91e5ad2489b3a6bef422 Mon Sep 17 00:00:00 2001
From: Your Name <your.email@example.com>
Date: Sat, 31 Jan 2026 22:29:00 +0800
Subject: [PATCH] fix: show loading placeholder when streaming but last message
 is user message

Co-authored-by: Cursor <cursoragent@cursor.com>
---
 components/messages.tsx | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/components/messages.tsx b/components/messages.tsx
index 6d39e48..f408282 100644
--- a/components/messages.tsx
+++ b/components/messages.tsx
@@ -126,8 +126,6 @@ function PureMessages({
 
   // 控制等待动画延迟消失（Gemini 图片模型延迟 4 秒，其他模型 6 秒）
   const [showThinkingWithDelay, setShowThinkingWithDelay] = useState(false);
-  // 🆕 记录占位图开始显示的时间，用于超时保护
-  const thinkingStartTimeRef = useRef<number | null>(null);
   
   useEffect(() => {
     const lastMessage = messages.length > 0 ? messages[messages.length - 1] : null;
@@ -202,6 +200,13 @@ function PureMessages({
       setShowThinkingWithDelay(true);
       thinkingStartTimeRef.current = Date.now();
     }
+    // 🆕 如果是 streaming 状态但最后一条是用户消息（还没有 AI 回复），也显示占位图
+    else if (status === "streaming" && lastMessage?.role === "user") {
+      setShowThinkingWithDelay(true);
+      if (!thinkingStartTimeRef.current) {
+        thinkingStartTimeRef.current = Date.now();
+      }
+    }
     // 如果是 streaming 状态且没有内容，显示
     else if (status === "streaming" && lastMessage?.role === "assistant" && !hasContent) {
       setShowThinkingWithDelay(true);