|
|
@@ -64,7 +64,21 @@ func testChannel(channel *model.Channel, testModel string) (err error, openaiErr
|
|
|
} else {
|
|
|
testModel = adaptor.GetModelList()[0]
|
|
|
}
|
|
|
+ } else {
|
|
|
+ modelMapping := *channel.ModelMapping
|
|
|
+ if modelMapping != "" && modelMapping != "{}" {
|
|
|
+ modelMap := make(map[string]string)
|
|
|
+ err := json.Unmarshal([]byte(modelMapping), &modelMap)
|
|
|
+ if err != nil {
|
|
|
+ openaiErr := service.OpenAIErrorWrapperLocal(err, "unmarshal_model_mapping_failed", http.StatusInternalServerError).Error
|
|
|
+ return err, &openaiErr
|
|
|
+ }
|
|
|
+ if modelMap[testModel] != "" {
|
|
|
+ testModel = modelMap[testModel]
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|
|
|
+
|
|
|
request := buildTestRequest()
|
|
|
request.Model = testModel
|
|
|
meta.UpstreamModelName = testModel
|