Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新百度模型名称，并保留旧版兼容以及修正单价

Browse Source
This commit is contained in:
ManJieqi 2024-04-01 20:06:21 +08:00
parent 6aebabbd7c
commit 3cf5ec9e6f
2 changed files with 29 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
common/model-ratio.go
View File

@ -72,17 +72,23 @@ var ModelRatio = map[string]float64{
"claude-3-sonnet-20240229": 3.0 / 1000 * USD,
"claude-3-opus-20240229": 15.0 / 1000 * USD,
// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/hlrk4akp7
"ERNIE-4.0-8K": 0.150 * RMB,
"ERNIE-3.5-8K": 0.012 * RMB,
"ERNIE-Speed-8K": 0.004 * RMB,
"ERNIE-Speed-128K":0.004 * RMB,
"ERNIE-Lite-8K": 0.003 * RMB,
"ERNIE-Tiny-8K": 0.001 * RMB,
"BLOOMZ-7B": 0.004 * RMB,
"Embedding-V1": 0.002 * RMB,
"bge-large-zh": 0.002 * RMB,
"bge-large-en": 0.002 * RMB,
"tao-8k": 0.002 * RMB,
"ERNIE-4.0": 0.120 * RMB, //旧版兼容 使用completions_pro 等同于ERNIE-4.0-8K
"ERNIE-Bot-4": 0.120 * RMB, //旧版兼容 使用completions_pro 等同于ERNIE-4.0-8K
"ERNIE-Bot": 0.012 * RMB, //旧版兼容 使用completions 等同于ERNIE-3.5-8K
"ERNIE-Bot-turbo": 0.003 * RMB, //旧版兼容 使用eb-instant 等同于ERNIE-Lite-8K
"ERNIE-Bot-8K": 0.024 * RMB, //即将弃用 使用ernie_bot_8k
"ERNIE-4.0-8K": 0.120 * RMB,
"ERNIE-3.5-8K": 0.012 * RMB,
"ERNIE-Speed-8K": 0.004 * RMB,
"ERNIE-Speed-128K": 0.004 * RMB,
"ERNIE-Lite-8K": 0.003 * RMB,
"ERNIE-Tiny-8K": 0.001 * RMB,
"BLOOMZ-7B": 0.004 * RMB,
"Embedding-V1": 0.002 * RMB,
"bge-large-zh": 0.002 * RMB,
"bge-large-en": 0.002 * RMB,
"tao-8k": 0.002 * RMB,
// https://ai.google.dev/pricing
"PaLM-2": 1,
"gemini-pro": 1, // $0.00025 / 1k characters -> $0.001 / 1k tokens

12
relay/channel/baidu/adaptor.go
View File

@ -34,6 +34,18 @@ func (a *Adaptor) GetRequestURL(meta *util.RelayMeta) (string, error) {
suffix = "embeddings/"
}
switch meta.ActualModelName {
case "ERNIE-4.0":
suffix += "completions_pro" //旧版兼容
case "ERNIE-Bot-4":
suffix += "completions_pro" //旧版兼容
case "ERNIE-Bot":
suffix += "completions" //旧版兼容
case "ERNIE-Bot-turbo":
suffix += "eb-instant" //旧版兼容
case "ERNIE-Speed":
suffix += "ernie_speed" //旧版兼容
case "ERNIE-Bot-8K":
suffix += "ernie_bot_8k" //即将弃用
case "ERNIE-4.0-8K":
suffix += "completions_pro"
case "ERNIE-3.5-8K":