fix: use modelMap when testing a channel (songquanpeng#1855)

Co-authored-by: oliang <oliang@tencent.com>
InitialXKO · Oct 27, 2024 · 3716e1b · 3716e1b
1 parent a4d6e7a
commit 3716e1b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/controller/channel-test.go b/controller/channel-test.go
@@ -76,9 +76,9 @@ func testChannel(channel *model.Channel, request *relaymodel.GeneralOpenAIReques
 		if len(modelNames) > 0 {
 			modelName = modelNames[0]
 		}
-		if modelMap != nil && modelMap[modelName] != "" {
-			modelName = modelMap[modelName]
-		}
+	}
+	if modelMap != nil && modelMap[modelName] != "" {
+		modelName = modelMap[modelName]
 	}
 	meta.OriginModelName, meta.ActualModelName = request.Model, modelName
 	request.Model = modelName