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

liangjs · oliang · web-flow · commit 3716e1b0e6aa · 2024-10-27T17:06:41.000+08:00
Co-authored-by: oliang &lt;oliang@tencent.com&gt;
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

Original file line number	Diff line number	Diff line change
`@@ -76,9 +76,9 @@ func testChannel(channel model.Channel, request relaymodel.GeneralOpenAIReques`
`76`	`76`	`if len(modelNames) > 0 {`
`77`	`77`	`modelName = modelNames[0]`
`78`	`78`	`}`
`79`		`- if modelMap != nil && modelMap[modelName] != "" {`
`80`		`- modelName = modelMap[modelName]`
`81`		`- }`
	`79`	`+ }`
	`80`	`+ if modelMap != nil && modelMap[modelName] != "" {`
	`81`	`+ modelName = modelMap[modelName]`
`82`	`82`	`}`
`83`	`83`	`meta.OriginModelName, meta.ActualModelName = request.Model, modelName`
`84`	`84`	`request.Model = modelName`