From 40ef06ceda58877a131dbdbb46038639f4e83632 Mon Sep 17 00:00:00 2001
From: Martial BE <me@xiao5.info>
Date: Mon, 25 Mar 2024 10:50:43 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20feat:=20add=20gemini=20new=20model?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 common/model-ratio.go                | 2 ++
 providers/gemini/base.go             | 2 +-
 web/src/views/Channel/type/Config.js | 2 +-
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/common/model-ratio.go b/common/model-ratio.go
index 6b556ab3..8c3bb11f 100644
--- a/common/model-ratio.go
+++ b/common/model-ratio.go
@@ -118,6 +118,8 @@ func init() {
 		"PaLM-2":            {[]float64{1, 1}, ChannelTypePaLM},
 		"gemini-pro":        {[]float64{1, 1}, ChannelTypeGemini},
 		"gemini-pro-vision": {[]float64{1, 1}, ChannelTypeGemini},
+		"gemini-1.0-pro":    {[]float64{1, 1}, ChannelTypeGemini},
+		"gemini-1.5-pro":    {[]float64{1, 1}, ChannelTypeGemini},
 
 		// ￥0.005 / 1k tokens
 		"glm-3-turbo": {[]float64{0.3572, 0.3572}, ChannelTypeZhipu},
diff --git a/providers/gemini/base.go b/providers/gemini/base.go
index 8b256701..f10fd632 100644
--- a/providers/gemini/base.go
+++ b/providers/gemini/base.go
@@ -61,7 +61,7 @@ func errorHandle(geminiError *GeminiErrorResponse) *types.OpenAIError {
 
 func (p *GeminiProvider) GetFullRequestURL(requestURL string, modelName string) string {
 	baseURL := strings.TrimSuffix(p.GetBaseURL(), "/")
-	version := "v1"
+	version := "v1beta"
 	if p.Channel.Other != "" {
 		version = p.Channel.Other
 	}
diff --git a/web/src/views/Channel/type/Config.js b/web/src/views/Channel/type/Config.js
index 080ac1ef..23df06eb 100644
--- a/web/src/views/Channel/type/Config.js
+++ b/web/src/views/Channel/type/Config.js
@@ -142,7 +142,7 @@ const typeConfig = {
       other: '版本号'
     },
     input: {
-      models: ['gemini-pro', 'gemini-pro-vision'],
+      models: ['gemini-pro', 'gemini-pro-vision', 'gemini-1.0-pro', 'gemini-1.5-pro'],
       test_model: 'gemini-pro'
     },
     prompt: {