From 1e19c333c97a4d8571df3e81eefbea4c2674e6d4 Mon Sep 17 00:00:00 2001 From: suziheng Date: Fri, 16 May 2025 10:18:07 +0800 Subject: [PATCH] fix: get gemini adapter bug --- relay/adaptor/gemini/adaptor.go | 2 +- relay/channeltype/helper.go | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) 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: