diff --git a/api/server/routes/ask.js b/api/server/routes/ask.js index 6fa1589fbd..1f18082dd4 100644 --- a/api/server/routes/ask.js +++ b/api/server/routes/ask.js @@ -77,6 +77,10 @@ router.post('/', async (req, res) => { sendMessage(res, { ...partial, message: true }); } else { tokens += partial === text ? '' : partial; + if (tokens.match(/^\n/)) { + tokens = tokens.replace(/^\n/, ''); + } + if (tokens.includes('[DONE]')) { tokens = tokens.replace('[DONE]', ''); }