Merge pull request #252 from utopeadia/main

add gemini-1.5-flash-latest support

common/model-ratio.go

@@ -79,6 +79,7 @@ var DefaultModelRatio = map[string]float64{
 	"gemini-1.0-pro-vision-001":    1,
 	"gemini-1.0-pro-vision-001":    1,
 	"gemini-1.0-pro-001":           1,
 	"gemini-1.0-pro-001":           1,
 	"gemini-1.5-pro-latest":        1,
 	"gemini-1.5-pro-latest":        1,
+	"gemini-1.5-flash-latest":      1,
 	"gemini-1.0-pro-latest":        1,
 	"gemini-1.0-pro-latest":        1,
 	"gemini-1.0-pro-vision-latest": 1,
 	"gemini-1.0-pro-vision-latest": 1,
 	"gemini-ultra":                 1,
 	"gemini-ultra":                 1,

relay/channel/gemini/adaptor.go

@@ -21,6 +21,7 @@ func (a *Adaptor) Init(info *relaycommon.RelayInfo, request dto.GeneralOpenAIReq
 // 定义一个映射，存储模型名称和对应的版本
 // 定义一个映射，存储模型名称和对应的版本
 var modelVersionMap = map[string]string{
 var modelVersionMap = map[string]string{
     "gemini-1.5-pro-latest": "v1beta",
     "gemini-1.5-pro-latest": "v1beta",
+    "gemini-1.5-flash-latest": "v1beta",
     "gemini-ultra":   "v1beta",
     "gemini-ultra":   "v1beta",
 }
 }
 
 

relay/channel/gemini/constant.go

@@ -5,7 +5,7 @@ const (
 )
 )
 
 
 var ModelList = []string{
 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",
 	"gemini-1.0-pro-vision-latest", "gemini-1.0-pro-vision-001",
 }
 }