From d810a43e2e59b29ebbb8a0e1b2a9e1ae7c1c6e59 Mon Sep 17 00:00:00 2001
From: Stainless Bot <dev@stainlessapi.com>
Date: Mon, 20 Jan 2025 04:09:30 +0000
Subject: [PATCH] fix: flush stream response when done event is sent

---
 packages/ssestream/streaming.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/packages/ssestream/streaming.go b/packages/ssestream/streaming.go
index 76616d9e..3704f820 100644
--- a/packages/ssestream/streaming.go
+++ b/packages/ssestream/streaming.go
@@ -138,12 +138,13 @@ func (s *Stream[T]) Next() bool {
 
 	for s.decoder.Next() {
 		if s.done {
-			return false
+			continue
 		}
 
 		if bytes.HasPrefix(s.decoder.Event().Data, []byte("[DONE]")) {
+			// In this case we don't break because we still want to iterate through the full stream.
 			s.done = true
-			return false
+			continue
 		}
 
 		ep := gjson.GetBytes(s.decoder.Event().Data, "error")