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 = ({ (() => { + const [showThink, setShowThink] = useState(false) + + // 动态提取 内容块 const thinkMatch = message.content.match(/([\s\S]*?)<\/think>/i) - const thinkContent = thinkMatch?.[1] || "" - const visibleContent = message.content.replace(/[\s\S]*?<\/think>/i, "").trim() + + // 思考区域(如果匹配到了) + const thinkContent = thinkMatch?.[1]?.trim() || "" + + // 可见区域 = 原始内容去除 区域 + const visibleContent = thinkMatch + ? message.content.replace(thinkMatch[0], "").trim() + : message.content.trim() return (
+ {/* 主内容显示 */} - {thinkMatch && ( + {/* 有 匹配时,才展示按钮 */} + {thinkMatch && thinkContent && (