diff --git a/relay/adaptor/gemini/adaptor.go b/relay/adaptor/gemini/adaptor.go index 84083f60..5d87d2d6 100644 --- a/relay/adaptor/gemini/adaptor.go +++ b/relay/adaptor/gemini/adaptor.go @@ -25,7 +25,7 @@ func (a *Adaptor) Init(meta *meta.Meta) { func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) { defaultVersion := config.GeminiVersion - if strings.Contains(meta.ActualModelName, "gemini-2.0") || + if strings.Contains(meta.ActualModelName, "gemini-2") || strings.Contains(meta.ActualModelName, "gemini-1.5") { defaultVersion = "v1beta" } diff --git a/relay/channeltype/helper.go b/relay/channeltype/helper.go index e8adb7df..ea3e6a55 100644 --- a/relay/channeltype/helper.go +++ b/relay/channeltype/helper.go @@ -23,6 +23,8 @@ func ToAPIType(channelType int) int { apiType = apitype.Tencent case Gemini: apiType = apitype.Gemini + case GeminiOpenAICompatible: + apiType = apitype.Gemini case Ollama: apiType = apitype.Ollama case AwsClaude: