From f83cb838461c4c1dba7244ff2ed845fa9e7d0f89 Mon Sep 17 00:00:00 2001 From: tbphp Date: Sun, 29 Jun 2025 10:06:27 +0800 Subject: [PATCH] fix: query key --- internal/proxy/server.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/internal/proxy/server.go b/internal/proxy/server.go index 28a1256..6344eb6 100644 --- a/internal/proxy/server.go +++ b/internal/proxy/server.go @@ -262,9 +262,13 @@ func (ps *ProxyServer) executeRequestWithRetry(c *gin.Context, startTime time.Ti if c.GetHeader("Authorization") != "" { req.Header.Set("Authorization", "Bearer "+keyInfo.Key) req.Header.Del("X-Goog-Api-Key") - } else if c.GetHeader("X-Goog-Api-Key") != "" || c.Query("key") != "" { + } else if c.GetHeader("X-Goog-Api-Key") != "" { req.Header.Set("X-Goog-Api-Key", keyInfo.Key) req.Header.Del("Authorization") + } else if c.Query("key") != "" { + q := req.URL.Query() + q.Set("key", keyInfo.Key) + req.URL.RawQuery = q.Encode() } else { c.JSON(http.StatusUnauthorized, gin.H{ "error": "API key required. Please provide a key in 'Authorization' or 'X-Goog-Api-Key' header.",