From b7941a45b9b7e8c295a11783e56e0da6121b7e1c Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Thu, 29 May 2025 20:34:59 +0800
Subject: [PATCH] .

---
 chatdesk-ui/components/messages/message.tsx | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/chatdesk-ui/components/messages/message.tsx b/chatdesk-ui/components/messages/message.tsx
index f20193d..be7763e 100644
--- a/chatdesk-ui/components/messages/message.tsx
+++ b/chatdesk-ui/components/messages/message.tsx
@@ -314,15 +314,26 @@ export const Message: FC<MessageProps> = ({
 
 
             (() => {
+              const [showThink, setShowThink] = useState(false)
+
+              // 动态提取 <think> 内容块
               const thinkMatch = message.content.match(/<think>([\s\S]*?)<\/think>/i)
-              const thinkContent = thinkMatch?.[1] || ""
-              const visibleContent = message.content.replace(/<think>[\s\S]*?<\/think>/i, "").trim()
+
+              // 思考区域（如果匹配到了）
+              const thinkContent = thinkMatch?.[1]?.trim() || ""
+
+              // 可见区域 = 原始内容去除 <think> 区域
+              const visibleContent = thinkMatch
+                ? message.content.replace(thinkMatch[0], "").trim()
+                : message.content.trim()
 
               return (
                 <div className="space-y-2">
+                  {/* 主内容显示 */}
                   <MessageMarkdown content={visibleContent} />
 
-                  {thinkMatch && (
+                  {/* 有 <think> 匹配时，才展示按钮 */}
+                  {thinkMatch && thinkContent && (
                     <div className="mt-2 text-sm text-muted-foreground">
                       <button
                         className="text-blue-500 hover:underline"
@@ -333,7 +344,7 @@ export const Message: FC<MessageProps> = ({
 
                       {showThink && (
                         <div className="mt-2 rounded bg-gray-100 p-3 text-sm dark:bg-gray-800">
-                          <MessageMarkdown content={thinkContent.trim()} />
+                          <MessageMarkdown content={thinkContent} />
                         </div>
                       )}
                     </div>