diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts index 0da08b9..4e72e32 100644 --- a/src/api/controllers/chat.ts +++ b/src/api/controllers/chat.ts @@ -411,7 +411,7 @@ async function receiveStream(model: string, stream: any): Promise { const result = _.attempt(() => JSON.parse(event.data)); if (_.isError(result)) throw new Error(`Stream response invalid: ${event.data}`); - if (!result.choices || !result.choices[0] || !result.choices[0].delta || !result.choices[0].delta.content || result.choices[0].delta.content.trim() == '') + if (!result.choices || !result.choices[0] || !result.choices[0].delta || !result.choices[0].delta.content || result.choices[0].delta.content == ' ') return; data.choices[0].message.content += result.choices[0].delta.content; if (result.choices && result.choices[0] && result.choices[0].finish_reason === "stop") @@ -465,7 +465,7 @@ function createTransStream(model: string, stream: any, endCallback?: Function) { const result = _.attempt(() => JSON.parse(event.data)); if (_.isError(result)) throw new Error(`Stream response invalid: ${event.data}`); - if (!result.choices || !result.choices[0] || !result.choices[0].delta || !result.choices[0].delta.content || result.choices[0].delta.content.trim() == '') + if (!result.choices || !result.choices[0] || !result.choices[0].delta || !result.choices[0].delta.content || result.choices[0].delta.content == ' ') return; result.model = model; transStream.write(`data: ${JSON.stringify({