debug

src/pages/api/stream.js

@@ -73,6 +73,10 @@ 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();
@@ -82,16 +86,19 @@ export const LLMStream = async (
                 const args = json.choices[0].message.function_call.arguments;
             
                 const fn = functions[fnName];
-                const result = await fn(...Object.values(JSON.parse(args)));
+                const functionResult = await fn(...Object.values(JSON.parse(args)));
             
                 console.log(`Function call: ${fnName}, Arguments: ${args}`);
-                console.log(`Calling Function ${fnName} Result: ` + result);
-            
+                console.log(`Calling Function ${fnName} Result: ` + functionResult);
                 
-              }
-            const text = json.choices[0].delta.content;
-            const queue = encoder.encode(text);
-            controller.enqueue(queue);
+                const queue = encoder.encode(functionResult);
+                controller.enqueue(queue);
+            } else {
+              const text = json.choices[0].delta.content;
+              const queue = encoder.encode(text);
+              controller.enqueue(queue);
+            }
+            
           } catch (e) {
             console.log(e);
             controller.error(e);