瀏覽代碼

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

CaIon 8 月之前
父節點
當前提交
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 {