xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-09-18 01:26:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add support for new Gemini model version 'gemini-2.0-flash-thinking-exp-01-21'

Browse Source
This commit is contained in:
Laisky.Cai 2025-01-22 03:27:22 +00:00
parent 13b1b165bd
commit a9310deb65
4 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
relay/adaptor/gemini/adaptor.go
View File

@ -27,7 +27,8 @@ func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
var defaultVersion string var defaultVersion string
switch meta.ActualModelName { switch meta.ActualModelName {
case "gemini-2.0-flash-exp", case "gemini-2.0-flash-exp",
"gemini-2.0-flash-thinking-exp": "gemini-2.0-flash-thinking-exp",
"gemini-2.0-flash-thinking-exp-01-21":
defaultVersion = "v1beta" defaultVersion = "v1beta"
default: default:
defaultVersion = config.GeminiVersion defaultVersion = config.GeminiVersion

2
relay/adaptor/gemini/constants.go
View File

@ -7,5 +7,5 @@ var ModelList = []string{
"gemini-1.5-flash", "gemini-1.5-pro", "gemini-1.5-flash", "gemini-1.5-pro",
"text-embedding-004", "aqa", "text-embedding-004", "aqa",
"gemini-2.0-flash-exp", "gemini-2.0-flash-exp",
"gemini-2.0-flash-thinking-exp", "gemini-2.0-flash-thinking-exp", "gemini-2.0-flash-thinking-exp-01-21",
} }

3
relay/adaptor/vertexai/gemini/adapter.go
View File

@ -18,7 +18,8 @@ var ModelList = []string{
"gemini-pro", "gemini-pro-vision", "gemini-pro", "gemini-pro-vision",
"gemini-1.5-pro-001", "gemini-1.5-flash-001", "gemini-1.5-pro-001", "gemini-1.5-flash-001",
"gemini-1.5-pro-002", "gemini-1.5-flash-002", "gemini-1.5-pro-002", "gemini-1.5-flash-002",
"gemini-2.0-flash-exp", "gemini-2.0-flash-thinking-exp", "gemini-2.0-flash-exp",
"gemini-2.0-flash-thinking-exp", "gemini-2.0-flash-thinking-exp-01-21",
} }
type Adaptor struct { type Adaptor struct {

1
relay/billing/ratio/model.go
View File

@ -125,6 +125,7 @@ var ModelRatio = map[string]float64{
"gemini-1.5-flash-001": 1, "gemini-1.5-flash-001": 1,
"gemini-2.0-flash-exp": 1, "gemini-2.0-flash-exp": 1,
"gemini-2.0-flash-thinking-exp": 1, "gemini-2.0-flash-thinking-exp": 1,
"gemini-2.0-flash-thinking-exp-01-21": 1,
"aqa": 1, "aqa": 1,
// https://open.bigmodel.cn/pricing // https://open.bigmodel.cn/pricing
"glm-4": 0.1 * RMB, "glm-4": 0.1 * RMB,