diff --git a/chatdesk-ui/components/messages/message.tsx b/chatdesk-ui/components/messages/message.tsx index d0867d6..f20193d 100644 --- a/chatdesk-ui/components/messages/message.tsx +++ b/chatdesk-ui/components/messages/message.tsx @@ -62,7 +62,9 @@ export const Message: FC = ({ models } = useContext(ChatbotUIContext) - const { handleSendMessage } = useChatHandler() + const [showThink, setShowThink] = useState(false) //添加Think的状态 + + const { handleSendMessage } = useChatHandler() const editInputRef = useRef(null) @@ -305,7 +307,46 @@ export const Message: FC = ({ maxRows={20} /> ) : ( - + + + + // + + + (() => { + const thinkMatch = message.content.match(/([\s\S]*?)<\/think>/i) + const thinkContent = thinkMatch?.[1] || "" + const visibleContent = message.content.replace(/[\s\S]*?<\/think>/i, "").trim() + + return ( +
+ + + {thinkMatch && ( +
+ + + {showThink && ( +
+ +
+ )} +
+ )} +
+ ) + })() + + + + + + )}