From 1587ea565b1bd98879f9b7bcfd0d4ba0f9a440b6 Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Mon, 25 Mar 2024 22:33:46 +0800
Subject: [PATCH] feat: support gemini-1.5

---
 common/model-ratio.go            | 7 +++++--
 relay/channel/gemini/constant.go | 4 ++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/common/model-ratio.go b/common/model-ratio.go
index 63d32474..f618af85 100644
--- a/common/model-ratio.go
+++ b/common/model-ratio.go
@@ -71,8 +71,11 @@ var DefaultModelRatio = map[string]float64{
 	"ERNIE-Bot-4":               8.572,  // ￥0.12 / 1k tokens
 	"Embedding-V1":              0.1429, // ￥0.002 / 1k tokens
 	"PaLM-2":                    1,
-	"gemini-pro":                1,      // $0.00025 / 1k characters -> $0.001 / 1k tokens
-	"gemini-pro-vision":         1,      // $0.00025 / 1k characters -> $0.001 / 1k tokens
+	"gemini-pro":                1, // $0.00025 / 1k characters -> $0.001 / 1k tokens
+	"gemini-pro-vision":         1, // $0.00025 / 1k characters -> $0.001 / 1k tokens
+	"gemini-1.0-pro-vision-001": 1,
+	"gemini-1.0-pro-001":        1,
+	"gemini-1.5-pro":            1,
 	"chatglm_turbo":             0.3572, // ￥0.005 / 1k tokens
 	"chatglm_pro":               0.7143, // ￥0.01 / 1k tokens
 	"chatglm_std":               0.3572, // ￥0.005 / 1k tokens
diff --git a/relay/channel/gemini/constant.go b/relay/channel/gemini/constant.go
index 24e85a87..582f4c0a 100644
--- a/relay/channel/gemini/constant.go
+++ b/relay/channel/gemini/constant.go
@@ -5,8 +5,8 @@ const (
 )
 
 var ModelList = []string{
-	"gemini-pro",
-	"gemini-pro-vision",
+	"gemini-pro", "gemini-1.0-pro-001", "gemini-1.5-pro",
+	"gemini-pro-vision", "gemini-1.0-pro-vision-001",
 }
 
 var ChannelName = "google gemini"