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

---
 chatdesk-ui/components/messages/message.tsx | 22 +++++++++++++++------
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/chatdesk-ui/components/messages/message.tsx b/chatdesk-ui/components/messages/message.tsx
index cc416a0..dc6f702 100644
--- a/chatdesk-ui/components/messages/message.tsx
+++ b/chatdesk-ui/components/messages/message.tsx
@@ -41,17 +41,26 @@ interface MessageProps {
 const MessageContent: React.FC<{ content: string }> = ({ content }) => {
   const [showThink, setShowThink] = useState(false)
 
-  const thinkMatch = content.match(/<think>([\s\S]*?)<\/think>/i)
-  const thinkContent = thinkMatch?.[1]?.trim() || ""
-  const visibleContent = thinkMatch
-    ? content.replace(thinkMatch[0], "").trim()
-    : content.trim()
+  let thinkContent = ""
+  let visibleContent = content.trim()
+
+  const fullMatch = content.match(/<think>([\s\S]*?)<\/think>/i)
+  if (fullMatch) {
+    thinkContent = fullMatch[1].trim()
+    visibleContent = content.replace(fullMatch[0], "").trim()
+  } else {
+    const endOnlyMatch = content.match(/([\s\S]*?)<\/think>/i)
+    if (endOnlyMatch) {
+      thinkContent = endOnlyMatch[1].trim()
+      visibleContent = content.replace(endOnlyMatch[0], "").trim()
+    }
+  }
 
   return (
     <div className="space-y-2">
       <MessageMarkdown content={visibleContent} />
 
-      {thinkMatch && thinkContent && (
+      {thinkContent && (
         <div className="mt-2 text-sm text-muted-foreground">
           <button
             className="text-blue-500 hover:underline"
@@ -71,6 +80,7 @@ const MessageContent: React.FC<{ content: string }> = ({ content }) => {
   )
 }
 
+
 export const Message: FC<MessageProps> = ({
   message,
   fileItems,