From f951358db4907c2d192e61d1ef5f32c89e84f527 Mon Sep 17 00:00:00 2001
From: jialin <jialinkuang@126.com>
Date: Tue, 13 May 2025 20:24:17 +0800
Subject: [PATCH] fix: stop thinking when loading end

---
 src/pages/playground/components/multiple-chat/content-item.tsx | 3 ++-
 src/pages/playground/components/multiple-chat/message-body.tsx | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/pages/playground/components/multiple-chat/content-item.tsx b/src/pages/playground/components/multiple-chat/content-item.tsx
index b6a3e48d..445f9bfb 100644
--- a/src/pages/playground/components/multiple-chat/content-item.tsx
+++ b/src/pages/playground/components/multiple-chat/content-item.tsx
@@ -188,7 +188,7 @@ const ContentItem: React.FC<MessageItemProps> = ({
         {roleTitle}
         {data.role === Roles.Assistant && reasoningContent.thought && (
           <ThoughtStatus
-            isThinking={reasoningContent.isThinking}
+            isThinking={loading && reasoningContent.isThinking}
             collapsed={collapsed}
             setCollapsed={setCollapsed}
           ></ThoughtStatus>
@@ -200,6 +200,7 @@ const ContentItem: React.FC<MessageItemProps> = ({
     data.title,
     intl,
     showTitle,
+    loading,
     reasoningContent.thought,
     reasoningContent.isThinking,
     collapsed
diff --git a/src/pages/playground/components/multiple-chat/message-body.tsx b/src/pages/playground/components/multiple-chat/message-body.tsx
index cf7e514c..d9f9560b 100644
--- a/src/pages/playground/components/multiple-chat/message-body.tsx
+++ b/src/pages/playground/components/multiple-chat/message-body.tsx
@@ -228,7 +228,7 @@ const MessageBody: React.FC<MessageBodyProps> = forwardRef(
                     <ThinkContent
                       collapsed={collapsed}
                       content={reasoningContent.thought}
-                      isThinking={reasoningContent.isThinking}
+                      isThinking={loading && reasoningContent.isThinking}
                     ></ThinkContent>
                     <div style={{ paddingInline: 4 }}>
                       <FullMarkdown