Merge pull request #79 from gamestap99/fix-sse

Fix sse bad state close
redevrx · Nov 22, 2023 · dd34388 · dd34388
2 parents 2415d53 + 356206b
commit dd34388
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");