From a5b84ba524982e64a6b736ee4d19818acf8fc220 Mon Sep 17 00:00:00 2001 From: HowieWu <98788152+utopeadia@users.noreply.github.com> Date: Wed, 15 May 2024 10:25:01 +0800 Subject: [PATCH 1/3] Update adaptor.go --- relay/channel/gemini/adaptor.go | 1 + 1 file changed, 1 insertion(+) diff --git a/relay/channel/gemini/adaptor.go b/relay/channel/gemini/adaptor.go index daaadc52..d372d82b 100644 --- a/relay/channel/gemini/adaptor.go +++ b/relay/channel/gemini/adaptor.go @@ -21,6 +21,7 @@ func (a *Adaptor) Init(info *relaycommon.RelayInfo, request dto.GeneralOpenAIReq // 定义一个映射,存储模型名称和对应的版本 var modelVersionMap = map[string]string{ "gemini-1.5-pro-latest": "v1beta", + "gemini-1.5-flash-latest": "v1beta", "gemini-ultra": "v1beta", } From 0f11461af3b7a660f0167c1b374b9de6e740c2c0 Mon Sep 17 00:00:00 2001 From: HowieWu <98788152+utopeadia@users.noreply.github.com> Date: Wed, 15 May 2024 10:27:30 +0800 Subject: [PATCH 2/3] Update model-ratio.go --- common/model-ratio.go | 1 + 1 file changed, 1 insertion(+) diff --git a/common/model-ratio.go b/common/model-ratio.go index 7302731d..8b66f499 100644 --- a/common/model-ratio.go +++ b/common/model-ratio.go @@ -79,6 +79,7 @@ var DefaultModelRatio = map[string]float64{ "gemini-1.0-pro-vision-001": 1, "gemini-1.0-pro-001": 1, "gemini-1.5-pro-latest": 1, + "gemini-1.5-flash-latest": 1, "gemini-1.0-pro-latest": 1, "gemini-1.0-pro-vision-latest": 1, "gemini-ultra": 1, From a3a6733fb5518e47257f15f5d25bd16657090fdc Mon Sep 17 00:00:00 2001 From: HowieWu <98788152+utopeadia@users.noreply.github.com> Date: Wed, 15 May 2024 10:28:30 +0800 Subject: [PATCH 3/3] Update constant.go --- relay/channel/gemini/constant.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/relay/channel/gemini/constant.go b/relay/channel/gemini/constant.go index 5e833bca..621336b1 100644 --- a/relay/channel/gemini/constant.go +++ b/relay/channel/gemini/constant.go @@ -5,7 +5,7 @@ const ( ) var ModelList = []string{ - "gemini-1.0-pro-latest", "gemini-1.0-pro-001", "gemini-1.5-pro-latest", "gemini-ultra", + "gemini-1.0-pro-latest", "gemini-1.0-pro-001", "gemini-1.5-pro-latest", "gemini-1.5-flash-latest", "gemini-ultra", "gemini-1.0-pro-vision-latest", "gemini-1.0-pro-vision-001", }