Function ImageHelper

relay/image_handler.go:111–247

(c *gin.Context)

}

func ImageHelper(c *gin.Context) (newAPIError *types.NewAPIError) {
	relayInfo := relaycommon.GenRelayInfoImage(c)

	imageRequest, err := getAndValidImageRequest(c, relayInfo)
	if err != nil {
		common.LogError(c, fmt.Sprintf("getAndValidImageRequest failed: %s", err.Error()))
		return types.NewError(err, types.ErrorCodeInvalidRequest)
	}

	err = helper.ModelMappedHelper(c, relayInfo, imageRequest)
	if err != nil {
		return types.NewError(err, types.ErrorCodeChannelModelMappedError)
	}

	priceData, err := helper.ModelPriceHelper(c, relayInfo, len(imageRequest.Prompt), 0)
	if err != nil {
		return types.NewError(err, types.ErrorCodeModelPriceError)
	}
	var preConsumedQuota int
	var quota int
	var userQuota int
	if !priceData.UsePrice {
		// modelRatio 16 = modelPrice $0.04
		// per 1 modelRatio = $0.04 / 16
		// priceData.ModelPrice = 0.0025 * priceData.ModelRatio
		preConsumedQuota, userQuota, newAPIError = preConsumeQuota(c, priceData.ShouldPreConsumedQuota, relayInfo)
		if newAPIError != nil {
			return newAPIError
		}
		defer func() {
			if newAPIError != nil {
				returnPreConsumedQuota(c, relayInfo, userQuota, preConsumedQuota)
			}
		}()

	} else {
		sizeRatio := 1.0
		qualityRatio := 1.0

		if strings.HasPrefix(imageRequest.Model, "dall-e") {
			// Size
			if imageRequest.Size == "256x256" {
				sizeRatio = 0.4
			} else if imageRequest.Size == "512x512" {
				sizeRatio = 0.45
			} else if imageRequest.Size == "1024x1024" {
				sizeRatio = 1
			} else if imageRequest.Size == "1024x1792" || imageRequest.Size == "1792x1024" {
				sizeRatio = 2
			}

			if imageRequest.Model == "dall-e-3" && imageRequest.Quality == "hd" {
				qualityRatio = 2.0
				if imageRequest.Size == "1024x1792" || imageRequest.Size == "1792x1024" {
					qualityRatio = 1.5
				}
			}
		}