From 4e1c2e645624836cb9d9b74c5d21fdff23a8e47e Mon Sep 17 00:00:00 2001
From: "caijialin.626" <caijialin.626@bytedance.com>
Date: Fri, 27 Feb 2026 11:00:59 +0800
Subject: [PATCH] fix decode bug

---
 internal/stream/sse.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/internal/stream/sse.go b/internal/stream/sse.go
index 3f7c872..011b064 100755
--- a/internal/stream/sse.go
+++ b/internal/stream/sse.go
@@ -52,9 +52,13 @@ func (d *SSEDecoder) Decode(ctx context.Context) <-chan SSEEvent {
 
 		for {
 			event, err := d.DecodeEvent()
-			ch <- SSEEvent{
-				Event: event,
-				Error: err,
+			select {
+			case <-ctx.Done():
+				return
+			case ch <- SSEEvent{Event: event, Error: err}:
+			}
+			if err != nil {
+				return
 			}
 		}
 	})