xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-09-18 01:26:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: update request URL formatting in Adaptor for proper query string handling

Browse Source
This commit is contained in:
Laisky.Cai 2024-12-15 02:16:56 +00:00
parent 33493aa8b3
commit 61d5f0eccf
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
relay/adaptor/gemini/adaptor.go
View File

@ -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 {