Fix sse bad state close

redevrx · Nov 21, 2023 · 356206b · 356206b
1 parent 2415d53
commit 356206b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/src/client/openai_client.dart b/lib/src/client/openai_client.dart
@@ -273,6 +273,8 @@ class OpenAIClient extends OpenAIWrapper {
                   final mData = data.substring(6);
                   if (mData.startsWith("[DONE]")) {
                     log.log("stream response is done");
+                    controller.done;
+                    controller.close();
 
                     return;
                   }
@@ -283,8 +285,6 @@ class OpenAIClient extends OpenAIWrapper {
                     controller
                       ..sink
                       ..add(complete(jsonMap[jsonMap.keys.last]));
-
-                    controller.close();
                   } else {
                     log.log("stream response invalid try regenerate");
                     log.log("last json error :$mData");