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

chore: update model mapping implementation for audio (#1932)

Browse Source 
* fixed model mapping

* chore: update implementation

---------

Co-authored-by: JustSong <quanpengsong@gmail.com>
This commit is contained in:
Yuwei Ba
2024-12-22 22:33:11 +11:00
committed by GitHub
parent 95e8c16338
commit ff589b5e4a
1 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
relay/controller/audio.go
View File

@@ -110,16 +110,9 @@ func RelayAudioHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatus
}() }()
// map model name // map model name
modelMapping := c.GetString(ctxkey.ModelMapping) modelMapping := c.GetStringMapString(ctxkey.ModelMapping)
if modelMapping != "" { if modelMapping != nil && modelMapping[audioModel] != "" {
modelMap := make(map[string]string) audioModel = modelMapping[audioModel]
err := json.Unmarshal([]byte(modelMapping), &modelMap)
if err != nil {
return openai.ErrorWrapper(err, "unmarshal_model_mapping_failed", http.StatusInternalServerError)
}
if modelMap[audioModel] != "" {
audioModel = modelMap[audioModel]
}
} }
baseURL := channeltype.ChannelBaseURLs[channelType] baseURL := channeltype.ChannelBaseURLs[channelType]