From cd8c86c6fb4aaf7ba639ec984d476b87d6f740d0 Mon Sep 17 00:00:00 2001
From: teeverc <72298507+teeverc@users.noreply.github.com>
Date: Sat, 13 Dec 2025 13:31:34 -0800
Subject: [PATCH] refactor: only flush stream response on successful write

---
 internal/api/modules/amp/response_rewriter.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/internal/api/modules/amp/response_rewriter.go b/internal/api/modules/amp/response_rewriter.go
index 04d45398..de6ba137 100644
--- a/internal/api/modules/amp/response_rewriter.go
+++ b/internal/api/modules/amp/response_rewriter.go
@@ -40,8 +40,10 @@ func (rw *ResponseRewriter) Write(data []byte) (int, error) {
 
 	if rw.isStreaming {
 		n, err := rw.ResponseWriter.Write(rw.rewriteStreamChunk(data))
-		if flusher, ok := rw.ResponseWriter.(http.Flusher); ok {
-			flusher.Flush()
+		if err == nil {
+			if flusher, ok := rw.ResponseWriter.(http.Flusher); ok {
+				flusher.Flush()
+			}
 		}
 		return n, err
 	}