Просмотр исходного кода

Merge pull request #252 from utopeadia/main

add gemini-1.5-flash-latest support
Calcium-Ion 1 год назад
Родитель
Сommit
8faf5d2517
3 измененных файлов с 3 добавлено и 1 удалено
  1. 1 0
      common/model-ratio.go
  2. 1 0
      relay/channel/gemini/adaptor.go
  3. 1 1
      relay/channel/gemini/constant.go

+ 1 - 0
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,

+ 1 - 0
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",
 }
 }
 
 

+ 1 - 1
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",
 }
 }