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

✨ feat: support other OpenAI APIs (#165)

Browse Source 
* ✨ feat: support other OpenAI APIs

* 🔖 chore: Update English translation
This commit is contained in:
Buer
2024-04-23 19:57:14 +08:00
committed by GitHub
parent f91b9856d4
commit 628df97f96
9 changed files with 152 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
providers/openai/base.go
View File

@@ -85,22 +85,28 @@ func (p *OpenAIProvider) GetFullRequestURL(requestURL string, modelName string)
if p.IsAzure {
apiVersion := p.Channel.Other
// 检测模型是是否包含 . 如果有则直接去掉
modelName = strings.Replace(modelName, ".", "", -1)
if modelName != "" {
// 检测模型是是否包含 . 如果有则直接去掉
modelName = strings.Replace(modelName, ".", "", -1)
if modelName == "dall-e-2" {
// 因为dall-e-3需要api-version=2023-12-01-preview，但是该版本
// 已经没有dall-e-2了，所以暂时写死
requestURL = fmt.Sprintf("/openai/%s:submit?api-version=2023-09-01-preview", requestURL)
if modelName == "dall-e-2" {
// 因为dall-e-3需要api-version=2023-12-01-preview，但是该版本
// 已经没有dall-e-2了，所以暂时写死
requestURL = fmt.Sprintf("/openai/%s:submit?api-version=2023-09-01-preview", requestURL)
} else {
requestURL = fmt.Sprintf("/openai/deployments/%s%s?api-version=%s", modelName, requestURL, apiVersion)
}
} else {
requestURL = fmt.Sprintf("/openai/deployments/%s%s?api-version=%s", modelName, requestURL, apiVersion)
requestURL = strings.TrimPrefix(requestURL, "/v1")
requestURL = fmt.Sprintf("/openai%s?api-version=%s", requestURL, apiVersion)
}
}
if strings.HasPrefix(baseURL, "https://gateway.ai.cloudflare.com") {
if p.IsAzure {
requestURL = strings.TrimPrefix(requestURL, "/openai/deployments")
requestURL = strings.TrimPrefix(requestURL, "/openai")
requestURL = strings.TrimPrefix(requestURL, "/deployments")
} else {
requestURL = strings.TrimPrefix(requestURL, "/v1")
}