handle error

replicate · Oct 10, 2024 · f1ffce9 · f1ffce9
1 parent 1995b8c
commit f1ffce9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/stream.go b/stream.go
@@ -206,7 +206,11 @@ func (r *Client) streamTextTo(ctx context.Context, writer *io.PipeWriter, url st
 		lastEventID = event.Id()
 		switch event.Event() {
 		case SSETypeOutput:
-			io.WriteString(writer, event.Data())
+			_, err := io.WriteString(writer, event.Data())
+			if err != nil {
+				writer.CloseWithError(err)
+				return
+			}
 		case SSETypeDone:
 			writer.Close()
 			return