Merge pull request #9 from wolfsilver/master

fix: 过滤空字符串导致格式错误
LLM-Red-Team · Jun 20, 2024 · 81d2dff · 81d2dff
2 parents eae48d3 + 19ce3d0
commit 81d2dff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts
@@ -411,7 +411,7 @@ async function receiveStream(model: string, stream: any): Promise<any> {
         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 == ' ')
+        if (!result.choices || !result.choices[0] || !result.choices[0].delta || !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 == ' ')
+      if (!result.choices || !result.choices[0] || !result.choices[0].delta || !result.choices[0].delta.content)
         return;
       result.model = model;
       transStream.write(`data: ${JSON.stringify({