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

✨ feat: complete some issues (#98)

Browse Source 
* 🎨 improve: SparkDesk support system roles

* 🎨 feat: Update Baidu model name

* ✨ feat: claude support cloudflare gateway
This commit is contained in:
Buer
2024-03-11 11:53:04 +08:00
committed by GitHub
parent fa89a3d60b
commit 27738aa3c7
5 changed files with 30 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
providers/baidu/base.go
View File

@@ -73,6 +73,14 @@ func (p *BaiduProvider) GetFullRequestURL(requestURL string, modelName string) s
"ERNIE-Bot-4": "completions_pro",
"BLOOMZ-7B": "bloomz_7b1",
"Embedding-V1": "embedding-v1",
"ERNIE-4.0": "completions_pro",
"ERNIE-3.5-8K": "completions",
"ERNIE-Bot-8K": "ernie_bot_8k",
"ERNIE-Speed": "ernie_speed",
}
if modelNameConvert, ok := modelNameMap[modelName]; ok {
modelName = modelNameConvert
}
baseURL := strings.TrimSuffix(p.GetBaseURL(), "/")
@@ -81,7 +89,7 @@ func (p *BaiduProvider) GetFullRequestURL(requestURL string, modelName string) s
return ""
}
return fmt.Sprintf("%s%s/%s?access_token=%s", baseURL, requestURL, modelNameMap[modelName], apiKey)
return fmt.Sprintf("%s%s/%s?access_token=%s", baseURL, requestURL, modelName, apiKey)
}
// 获取请求头