fix gemini

This commit is contained in:
CaIon
2023-12-21 23:08:09 +08:00
parent 676dc95793
commit 6e8ff8c057

View File

@@ -414,13 +414,18 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
} }
case APITypeTencent: case APITypeTencent:
req.Header.Set("Authorization", apiKey) req.Header.Set("Authorization", apiKey)
case APITypeGemini:
req.Header.Set("Content-Type", "application/json")
default: default:
req.Header.Set("Authorization", "Bearer "+apiKey) req.Header.Set("Authorization", "Bearer "+apiKey)
} }
req.Header.Set("Content-Type", c.Request.Header.Get("Content-Type")) if apiType != APITypeGemini {
req.Header.Set("Accept", c.Request.Header.Get("Accept")) // 设置公共头部...
if isStream && c.Request.Header.Get("Accept") == "" { req.Header.Set("Content-Type", c.Request.Header.Get("Content-Type"))
req.Header.Set("Accept", "text/event-stream") req.Header.Set("Accept", c.Request.Header.Get("Accept"))
if isStream && c.Request.Header.Get("Accept") == "" {
req.Header.Set("Accept", "text/event-stream")
}
} }
//req.HeaderBar.Set("Connection", c.Request.HeaderBar.Get("Connection")) //req.HeaderBar.Set("Connection", c.Request.HeaderBar.Get("Connection"))
resp, err = httpClient.Do(req) resp, err = httpClient.Do(req)