xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-18 08:26:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: support AI Gateway, close #52

Browse Source
This commit is contained in:
CaIon 2024-02-01 18:11:00 +08:00
parent 4274b925da
commit 0004c1022d
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
controller/channel-test.go
View File

@ -43,14 +43,10 @@ func testChannel(channel *model.Channel, request ChatRequest) (err error, openai
default: default:
request.Model = "gpt-3.5-turbo" request.Model = "gpt-3.5-turbo"
} }
requestURL := common.ChannelBaseURLs[channel.Type] requestURL := getFullRequestURL(channel.GetBaseURL(), "/v1/chat/completions", channel.Type)
if channel.Type == common.ChannelTypeAzure { if channel.Type == common.ChannelTypeAzure {
requestURL = getFullRequestURL(channel.GetBaseURL(), fmt.Sprintf("/openai/deployments/%s/chat/completions?api-version=2023-03-15-preview", request.Model), channel.Type) requestURL = getFullRequestURL(channel.GetBaseURL(), fmt.Sprintf("/openai/deployments/%s/chat/completions?api-version=2023-03-15-preview", request.Model), channel.Type)
} else {
if channel.GetBaseURL() != "" {
requestURL = channel.GetBaseURL()
}
requestURL += "/v1/chat/completions"
} }
jsonData, err := json.Marshal(request) jsonData, err := json.Marshal(request)