From 343aac6326063ab5f0b5c2c4a16988c989854682 Mon Sep 17 00:00:00 2001 From: Yuwei Ba Date: Mon, 18 Nov 2024 22:16:38 +1100 Subject: [PATCH] fixed model mapping --- relay/controller/audio.go | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/relay/controller/audio.go b/relay/controller/audio.go index 83040662..54b97a30 100644 --- a/relay/controller/audio.go +++ b/relay/controller/audio.go @@ -110,15 +110,10 @@ func RelayAudioHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatus }() // map model name - modelMapping := c.GetString(ctxkey.ModelMapping) - if modelMapping != "" { - modelMap := make(map[string]string) - 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] + modelMapping := c.GetStringMapString(ctxkey.ModelMapping) + if modelMapping != nil { + if modelMapping[audioModel] != "" { + audioModel = modelMapping[audioModel] } }