Merge pull request #322 from thivy/main

close stream on error and abort
microsoft · Feb 12, 2024 · 55dbe1e · 55dbe1e
2 parents 6ff8264 + 6bfaba9
commit 55dbe1e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/features/chat-page/chat-services/chat-api/open-ai-stream.ts b/src/features/chat-page/chat-services/chat-api/open-ai-stream.ts
@@ -70,6 +70,7 @@ export const OpenAIStream = (props: {
             response: "Chat aborted",
           };
           streamResponse(response.type, JSON.stringify(response));
+          controller.close();
         })
         .on("error", async (error) => {
           console.log("🔴 error", error);
@@ -87,6 +88,7 @@ export const OpenAIStream = (props: {
           });
 
           streamResponse(response.type, JSON.stringify(response));
+          controller.close();
         })
         .on("finalContent", async (content: string) => {
           await CreateChatMessage({