This commit is contained in:
hailin 2025-05-29 19:55:49 +08:00
parent e1d22386a1
commit fa170db990
1 changed files with 43 additions and 2 deletions

View File

@ -62,7 +62,9 @@ export const Message: FC<MessageProps> = ({
models
} = useContext(ChatbotUIContext)
const { handleSendMessage } = useChatHandler()
const [showThink, setShowThink] = useState(false) //添加Think的状态
const { handleSendMessage } = useChatHandler()
const editInputRef = useRef<HTMLTextAreaElement>(null)
@ -305,7 +307,46 @@ export const Message: FC<MessageProps> = ({
maxRows={20}
/>
) : (
<MessageMarkdown content={message.content} />
// <MessageMarkdown content={message.content} />
(() => {
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()
return (
<div className="space-y-2">
<MessageMarkdown content={visibleContent} />
{thinkMatch && (
<div className="mt-2 text-sm text-muted-foreground">
<button
className="text-blue-500 hover:underline"
onClick={() => setShowThink(prev => !prev)}
>
{showThink ? "隐藏思考过程 ▲" : "显示思考过程 ▼"}
</button>
{showThink && (
<div className="mt-2 rounded bg-gray-100 p-3 text-sm dark:bg-gray-800">
<MessageMarkdown content={thinkContent.trim()} />
</div>
)}
</div>
)}
</div>
)
})()
)}
</div>