From 5fe50ac1eb8187fcf6875570be4e00f2bde62c79 Mon Sep 17 00:00:00 2001
From: Nikhil Sinha <nikhil@replicate.com>
Date: Mon, 9 Jun 2025 13:52:30 -0700
Subject: [PATCH] Retain query parameters on redirect

---
 pkg/download/buffer.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pkg/download/buffer.go b/pkg/download/buffer.go
index 874f9b3..ddcf94b 100644
--- a/pkg/download/buffer.go
+++ b/pkg/download/buffer.go
@@ -287,6 +287,9 @@ func (m *BufferMode) rewritePrefix(cacheHost, urlString string, parsed *url.URL,
 			Msg("Cache URL Rewrite")
 		return urlString
 	}
+	if parsed.RawQuery != "" {
+		newUrl = fmt.Sprintf("%s?%s", newUrl, parsed.RawQuery)
+	}
 	logger.Info().
 		Str("url", urlString).
 		Str("target_url", newUrl).