fix: flush stream response when done event is sent (#172)

stainless-app[bot] · stainless-app[bot] · commit cf1a6a5b384f · 2025-01-20T14:52:35.000Z
diff --git 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")

Original file line number	Diff line number	Diff line change
`@@ -138,12 +138,13 @@ func (s *Stream[T]) Next() bool {`
`138`	`138`
`139`	`139`	`for s.decoder.Next() {`
`140`	`140`	`if s.done {`
`141`		`- return false`
	`141`	`+ continue`
`142`	`142`	`}`
`143`	`143`
`144`	`144`	`if bytes.HasPrefix(s.decoder.Event().Data, []byte("[DONE]")) {`
	`145`	`+ // In this case we don't break because we still want to iterate through the full stream.`
`145`	`146`	`s.done = true`
`146`		`- return false`
	`147`	`+ continue`
`147`	`148`	`}`
`148`	`149`
`149`	`150`	`ep := gjson.GetBytes(s.decoder.Event().Data, "error")`