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) }