diff --git a/src/components/ChatUI.tsx b/src/components/ChatUI.tsx index 58626af..d68e62b 100644 --- a/src/components/ChatUI.tsx +++ b/src/components/ChatUI.tsx @@ -191,7 +191,23 @@ const ChatUI = () => { while (true) { const { done, value } = await reader.read(); - if (done) break; + if (done) { + //如果completeResponse为空, + if (completeResponse.trim() === "") { + completeResponse = "这个问题难倒我了,下一位。"; + setMessages(prev => { + const newMessages = [...prev]; + const aiMessageIndex = newMessages.findIndex(msg => msg.id === aiMessage.id); + if (aiMessageIndex !== -1) { + newMessages[aiMessageIndex] = { + ...newMessages[aiMessageIndex], + content: completeResponse + }; + } + return newMessages; + });} + break; + } buffer += decoder.decode(value, { stream: true }); @@ -216,13 +232,15 @@ const ChatUI = () => { } return newMessages; }); - } + } + } catch (e) { console.error('解析响应数据失败:', e); } } } } + // 将当前AI的回复添加到消息历史中,供下一个AI使用 messageHistory.push({ role: 'user', @@ -348,7 +366,7 @@ const ChatUI = () => {