Spaces:

matthoffner
/

serp-chat

Paused

matt HOFFNER commited on Jun 23, 2023

Commit

b446577

1 Parent(s): bf8ef54

debug

Files changed (1) hide show

src/pages/api/stream.js CHANGED Viewed

@@ -73,12 +73,8 @@ export const LLMStream = async (
           const data = event.data;
           try {
-            if (data === '[DONE]') {
-              controller.close();
-              return;
-            }
             const json = JSON.parse(data);
-            if (json.choices[0].finish_reason === "stop") {
               controller.close();
               return;
             } else if (json.choices[0].finish_reason === "function_call") {

           const data = event.data;
           try {
             const json = JSON.parse(data);
+            if (json.choices[0].finish_reason === "stop" || data === '[DONE]') {
               controller.close();
               return;
             } else if (json.choices[0].finish_reason === "function_call") {