Преглед изворни кода

fix(distributor): add validation for model name in channel selection

CaIon пре 7 месеци
родитељ
комит
eaee89f77a
1 измењених фајлова са 4 додато и 0 уклоњено
  1. 4 0
      middleware/distributor.go

+ 4 - 0
middleware/distributor.go

@@ -100,6 +100,10 @@ func Distribute() func(c *gin.Context) {
 			}
 
 			if shouldSelectChannel {
+				if modelRequest.Model == "" {
+					abortWithOpenAiMessage(c, http.StatusBadRequest, "未指定模型名称,模型名称不能为空")
+					return
+				}
 				var selectGroup string
 				channel, selectGroup, err = model.CacheGetRandomSatisfiedChannel(c, userGroup, modelRequest.Model, 0)
 				if err != nil {