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

🔖 chore: Update baidu model

Browse Source
This commit is contained in:
MartialBE 2024-05-22 02:57:40 +08:00
parent 7ddeada284
commit f1b97cecce
No known key found for this signature in database
GPG Key ID: 27C0267EC84B0A5C
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
providers/baidu/base.go
View File

@ -75,8 +75,11 @@ func (p *BaiduProvider) GetFullRequestURL(requestURL string, modelName string) s
"Embedding-V1": "embedding-v1", "Embedding-V1": "embedding-v1",
"ERNIE-4.0": "completions_pro", "ERNIE-4.0": "completions_pro",
"ERNIE-3.5-8K": "completions", "ERNIE-3.5-8K": "completions",
"ERNIE-Bot-8K": "ernie_bot_8k",
"ERNIE-Speed": "ernie_speed", "ERNIE-Speed": "ernie_speed",
"ERNIE-Speed-128K": "ernie_speed_128k",
"ERNIE-Lite-8K": "eb-instant",
"ERNIE-Tiny-8K": "ernie-tiny-8k",
"ERNIE-Functions-8K": "ernie-func-8k",
} }
if modelNameConvert, ok := modelNameMap[modelName]; ok { if modelNameConvert, ok := modelNameMap[modelName]; ok {