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

21
providers/baidu/base.go
View File

@ -68,15 +68,18 @@ func errorHandle(baiduError *BaiduError) *types.OpenAIError {
// 获取完整请求 URL // 获取完整请求 URL
func (p *BaiduProvider) GetFullRequestURL(requestURL string, modelName string) string { func (p *BaiduProvider) GetFullRequestURL(requestURL string, modelName string) string {
var modelNameMap = map[string]string{ var modelNameMap = map[string]string{
"ERNIE-Bot": "completions", "ERNIE-Bot": "completions",
"ERNIE-Bot-turbo": "eb-instant", "ERNIE-Bot-turbo": "eb-instant",
"ERNIE-Bot-4": "completions_pro", "ERNIE-Bot-4": "completions_pro",
"BLOOMZ-7B": "bloomz_7b1", "BLOOMZ-7B": "bloomz_7b1",
"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 {