From 61d5f0eccfb943925972d8e3f413c38798b1a242 Mon Sep 17 00:00:00 2001 From: "Laisky.Cai" Date: Sun, 15 Dec 2024 02:16:56 +0000 Subject: [PATCH] fix: update request URL formatting in Adaptor for proper query string handling --- relay/adaptor/gemini/adaptor.go | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/relay/adaptor/gemini/adaptor.go b/relay/adaptor/gemini/adaptor.go index c362d5f3..074e4582 100644 --- a/relay/adaptor/gemini/adaptor.go +++ b/relay/adaptor/gemini/adaptor.go @@ -33,19 +33,16 @@ func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) { action := "" switch meta.Mode { case relaymode.Embeddings: - action = "batchEmbedContents" + action = "batchEmbedContents?" default: - action = "generateContent" + action = "generateContent?" } if meta.IsStream { - if version == "v1" { - action = "streamGenerateContent?alt=sse" - } else { - action = "streamGenerateContent" - } + action = "streamGenerateContent?alt=sse&" } - return fmt.Sprintf("%s/%s/models/%s:%s?key=%s", meta.BaseURL, version, meta.ActualModelName, action, meta.APIKey), nil + + return fmt.Sprintf("%s/%s/models/%s:%skey=%s", meta.BaseURL, version, meta.ActualModelName, action, meta.APIKey), nil } func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {