From 52d5fd1a67c03e791a0008182da862178863817f Mon Sep 17 00:00:00 2001
From: teeverc <72298507+teeverc@users.noreply.github.com>
Date: Sat, 13 Dec 2025 13:17:53 -0800
Subject: [PATCH] fix: streaming for amp cli

---
 internal/api/modules/amp/response_rewriter.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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