ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-19 10:07:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add cloudflare ai gateway support for image & audio (#607)

Browse Source 
* Update channel-test.go

* Update relay-audio.go

* Update relay-image.go

* chore: using a util function

---------

Co-authored-by: JustSong <songquanpeng@foxmail.com>
This commit is contained in:
vc
2023-10-22 17:50:52 +08:00
committed by GitHub
parent 22980b4c44
commit 3b483639a4
5 changed files with 17 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
controller/relay-utils.go
View File

@@ -176,3 +176,13 @@ func relayErrorHandler(resp *http.Response) (openAIErrorWithStatusCode *OpenAIEr
openAIErrorWithStatusCode.OpenAIError = textResponse.Error
return
}
func getFullRequestURL(baseURL string, requestURL string, channelType int) string {
fullRequestURL := fmt.Sprintf("%s%s", baseURL, requestURL)
if channelType == common.ChannelTypeOpenAI {
if strings.HasPrefix(baseURL, "https://gateway.ai.cloudflare.com") {
fullRequestURL = fmt.Sprintf("%s%s", baseURL, strings.TrimPrefix(requestURL, "/v1"))
}
}
return fullRequestURL
}