xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-10 10:33:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: update .gitignore and improve model handling in various adapters and billing ratios

Browse Source 
closes #37
This commit is contained in:
Laisky.Cai
2025-03-13 01:47:41 +00:00
parent d30b6840ad
commit a20f57a923
5 changed files with 28 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -9,7 +9,7 @@ var ModelList = []string{
"gemini-1.5-pro", "gemini-1.5-pro-experimental",
"text-embedding-004", "aqa",
"gemini-2.0-flash", "gemini-2.0-flash-exp",
"gemini-2.0-flash-lite-preview-02-05",
"gemini-2.0-flash-lite",
"gemini-2.0-flash-thinking-exp-01-21",
"gemini-2.0-pro-exp-02-05",
}

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

@@ -13,14 +13,20 @@ import (
"github.com/songquanpeng/one-api/relay/relaymode"
)
// ModelList is the list of models supported by Vertex AI.
//
// https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models
var ModelList = []string{
"gemini-pro", "gemini-pro-vision",
"gemini-exp-1206",
"gemini-1.5-pro-001", "gemini-1.5-pro-002",
"gemini-1.5-flash-001", "gemini-1.5-flash-002",
"gemini-2.0-flash-exp", "gemini-2.0-flash-001",
"gemini-2.0-flash-lite-preview-02-05",
"gemini-1.0-pro",
"gemini-1.0-pro-vision",
"gemini-1.5-pro", "gemini-1.5-pro-001", "gemini-1.5-pro-002",
"gemini-1.5-flash", "gemini-1.5-flash-001", "gemini-1.5-flash-002",
"gemini-2.0-flash", "gemini-2.0-flash-exp", "gemini-2.0-flash-001",
"gemini-2.0-flash-lite", "gemini-2.0-flash-lite-001",
"gemini-2.0-flash-thinking-exp-01-21",
"gemini-2.0-pro-exp-02-05",
}
type Adaptor struct {