diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts index 0e9e7c0..53ba25d 100644 --- a/src/api/controllers/chat.ts +++ b/src/api/controllers/chat.ts @@ -161,7 +161,7 @@ async function receiveStream(convId: string, stream: any) { data.choices[0].message.content += result.text; } else if (result.event == 'all_done' || result.event == 'error') { - data.choices[0].message.content += '\n[内容由于不合规被停止生成,我们换个话题吧]' + (refContent ? `\n\n搜索结果来自:\n${refContent}` : ''); + data.choices[0].message.content += (result.event == 'error' ? '\n[内容由于不合规被停止生成,我们换个话题吧]' : '') + (refContent ? `\n\n搜索结果来自:\n${refContent}` : ''); refContent = ''; resolve(data); }