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 new Gemini 2.0 flash models to adapter and billing ratio

Browse Source
This commit is contained in:
Laisky.Cai 2025-02-06 12:57:57 +00:00
parent f1f8aa92c5
commit a31a58700d
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -16,10 +16,12 @@ import (
var ModelList = []string{ var ModelList = []string{
"gemini-pro", "gemini-pro-vision", "gemini-pro", "gemini-pro-vision",
"gemini-1.5-pro-001", "gemini-1.5-flash-001", "gemini-exp-1206",
"gemini-1.5-pro-002", "gemini-1.5-flash-002", "gemini-1.5-pro-001", "gemini-1.5-pro-002",
"gemini-2.0-flash-exp", "gemini-1.5-flash-001", "gemini-1.5-flash-002",
"gemini-2.0-flash-thinking-exp", "gemini-2.0-flash-thinking-exp-01-21", "gemini-2.0-flash-exp", "gemini-2.0-flash-001",
"gemini-2.0-flash-lite-preview-02-05",
"gemini-2.0-flash-thinking-exp-01-21",
} }
type Adaptor struct { type Adaptor struct {

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

@ -126,6 +126,7 @@ var ModelRatio = map[string]float64{
"gemini-1.5-flash-8b": 0.0375 * MILLI_USD, "gemini-1.5-flash-8b": 0.0375 * MILLI_USD,
"gemini-2.0-flash-exp": 0.075 * MILLI_USD, "gemini-2.0-flash-exp": 0.075 * MILLI_USD,
"gemini-2.0-flash": 0.15 * MILLI_USD, "gemini-2.0-flash": 0.15 * MILLI_USD,
"gemini-2.0-flash-001": 0.15 * MILLI_USD,
"gemini-2.0-flash-lite-preview-02-05": 0.075 * MILLI_USD, "gemini-2.0-flash-lite-preview-02-05": 0.075 * MILLI_USD,
"gemini-2.0-flash-thinking-exp-01-21": 0.075 * MILLI_USD, "gemini-2.0-flash-thinking-exp-01-21": 0.075 * MILLI_USD,
"gemini-2.0-pro-exp-02-05": 1.25 * MILLI_USD, "gemini-2.0-pro-exp-02-05": 1.25 * MILLI_USD,