Explorar o código

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

CaIon hai 7 meses
pai
achega
eaee89f77a
Modificáronse 1 ficheiros con 4 adicións e 0 borrados
  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 {