diff --git a/common/model-ratio.go b/common/model-ratio.go index 7302731d..8b66f499 100644 --- a/common/model-ratio.go +++ b/common/model-ratio.go @@ -79,6 +79,7 @@ var DefaultModelRatio = map[string]float64{ "gemini-1.0-pro-vision-001": 1, "gemini-1.0-pro-001": 1, "gemini-1.5-pro-latest": 1, + "gemini-1.5-flash-latest": 1, "gemini-1.0-pro-latest": 1, "gemini-1.0-pro-vision-latest": 1, "gemini-ultra": 1, diff --git a/relay/channel/gemini/adaptor.go b/relay/channel/gemini/adaptor.go index daaadc52..d372d82b 100644 --- a/relay/channel/gemini/adaptor.go +++ b/relay/channel/gemini/adaptor.go @@ -21,6 +21,7 @@ func (a *Adaptor) Init(info *relaycommon.RelayInfo, request dto.GeneralOpenAIReq // 定义一个映射,存储模型名称和对应的版本 var modelVersionMap = map[string]string{ "gemini-1.5-pro-latest": "v1beta", + "gemini-1.5-flash-latest": "v1beta", "gemini-ultra": "v1beta", } diff --git a/relay/channel/gemini/constant.go b/relay/channel/gemini/constant.go index 5e833bca..621336b1 100644 --- a/relay/channel/gemini/constant.go +++ b/relay/channel/gemini/constant.go @@ -5,7 +5,7 @@ const ( ) 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", }