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

Merge pull request #570 from leezhuuuuu/main

增加对于gemini-exp-1114模型的支持,映射到v1beta
Calcium-Ion 1 год назад
Родитель
Сommit
c3e6b2408e
2 измененных файлов с 2 добавлено и 0 удалено
  1. 1 0
      constant/env.go
  2. 1 0
      relay/channel/gemini/constant.go

+ 1 - 0
constant/env.go

@@ -30,6 +30,7 @@ var GeminiModelMap = map[string]string{
 	"gemini-1.5-flash-001":      "v1beta",
 	"gemini-1.5-flash-001":      "v1beta",
 	"gemini-1.5-flash":          "v1beta",
 	"gemini-1.5-flash":          "v1beta",
 	"gemini-ultra":              "v1beta",
 	"gemini-ultra":              "v1beta",
+	"gemini-exp-1114":           "v1beta",
 }
 }
 
 
 func InitEnv() {
 func InitEnv() {

+ 1 - 0
relay/channel/gemini/constant.go

@@ -7,6 +7,7 @@ const (
 var ModelList = []string{
 var ModelList = []string{
 	"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-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.5-pro-exp-0827", "gemini-1.5-flash-exp-0827",
 	"gemini-1.0-pro-vision-latest", "gemini-1.0-pro-vision-001", "gemini-1.5-pro-exp-0827", "gemini-1.5-flash-exp-0827",
+	"gemini-exp-1114",
 }
 }
 
 
 var ChannelName = "google gemini"
 var ChannelName = "google gemini"