diff --git a/lib/src/client/openai_client.dart b/lib/src/client/openai_client.dart
index 1131409..f692441 100644
--- 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");