refactor: remove consumeQuota

relay/relay-image.go

@@ -24,17 +24,14 @@ func RelayImageHelper(c *gin.Context, relayMode int) *dto.OpenAIErrorWithStatusC
 	channelType := c.GetInt("channel")
 	channelType := c.GetInt("channel")
 	channelId := c.GetInt("channel_id")
 	channelId := c.GetInt("channel_id")
 	userId := c.GetInt("id")
 	userId := c.GetInt("id")
-	consumeQuota := c.GetBool("consume_quota")
 	group := c.GetString("group")
 	group := c.GetString("group")
 	startTime := time.Now()
 	startTime := time.Now()
 
 
 	var imageRequest dto.ImageRequest
 	var imageRequest dto.ImageRequest
-	if consumeQuota {
 	err := common.UnmarshalBodyReusable(c, &imageRequest)
 	err := common.UnmarshalBodyReusable(c, &imageRequest)
 	if err != nil {
 	if err != nil {
 		return service.OpenAIErrorWrapper(err, "bind_request_body_failed", http.StatusBadRequest)
 		return service.OpenAIErrorWrapper(err, "bind_request_body_failed", http.StatusBadRequest)
 	}
 	}
-	}
 
 
 	if imageRequest.Model == "" {
 	if imageRequest.Model == "" {
 		imageRequest.Model = "dall-e-2"
 		imageRequest.Model = "dall-e-2"
@@ -136,7 +133,7 @@ func RelayImageHelper(c *gin.Context, relayMode int) *dto.OpenAIErrorWithStatusC
 
 
 	quota := int(ratio*sizeRatio*qualityRatio*1000) * imageRequest.N
 	quota := int(ratio*sizeRatio*qualityRatio*1000) * imageRequest.N
 
 
-	if consumeQuota && userQuota-quota < 0 {
+	if userQuota-quota < 0 {
 		return service.OpenAIErrorWrapper(errors.New("user quota is not enough"), "insufficient_user_quota", http.StatusForbidden)
 		return service.OpenAIErrorWrapper(errors.New("user quota is not enough"), "insufficient_user_quota", http.StatusForbidden)
 	}
 	}
 
 
@@ -176,7 +173,6 @@ func RelayImageHelper(c *gin.Context, relayMode int) *dto.OpenAIErrorWithStatusC
 	var textResponse dto.ImageResponse
 	var textResponse dto.ImageResponse
 	defer func(ctx context.Context) {
 	defer func(ctx context.Context) {
 		useTimeSeconds := time.Now().Unix() - startTime.Unix()
 		useTimeSeconds := time.Now().Unix() - startTime.Unix()
-		if consumeQuota {
 		if resp.StatusCode != http.StatusOK {
 		if resp.StatusCode != http.StatusOK {
 			return
 			return
 		}
 		}
@@ -196,10 +192,8 @@ func RelayImageHelper(c *gin.Context, relayMode int) *dto.OpenAIErrorWithStatusC
 			channelId := c.GetInt("channel_id")
 			channelId := c.GetInt("channel_id")
 			model.UpdateChannelUsedQuota(channelId, quota)
 			model.UpdateChannelUsedQuota(channelId, quota)
 		}
 		}
-		}
 	}(c.Request.Context())
 	}(c.Request.Context())
 
 
-	if consumeQuota {
 	responseBody, err := io.ReadAll(resp.Body)
 	responseBody, err := io.ReadAll(resp.Body)
 
 
 	if err != nil {
 	if err != nil {
@@ -215,7 +209,6 @@ func RelayImageHelper(c *gin.Context, relayMode int) *dto.OpenAIErrorWithStatusC
 	}
 	}
 
 
 	resp.Body = io.NopCloser(bytes.NewBuffer(responseBody))
 	resp.Body = io.NopCloser(bytes.NewBuffer(responseBody))
-	}
 
 
 	for k, v := range resp.Header {
 	for k, v := range resp.Header {
 		c.Writer.Header().Set(k, v[0])
 		c.Writer.Header().Set(k, v[0])