ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-17 16:43:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: reorganize billing related package

Browse Source
This commit is contained in:
JustSong
2024-04-06 01:09:23 +08:00
parent 2ab7d25a80
commit cd2707692f
6 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
relay/controller/helper.go
View File

@@ -12,7 +12,6 @@ import (
"github.com/songquanpeng/one-api/relay/billing"
"github.com/songquanpeng/one-api/relay/channel/openai"
"github.com/songquanpeng/one-api/relay/channeltype"
"github.com/songquanpeng/one-api/relay/constant"
relaymodel "github.com/songquanpeng/one-api/relay/model"
"github.com/songquanpeng/one-api/relay/relaymode"
"github.com/songquanpeng/one-api/relay/util"
@@ -61,12 +60,12 @@ func isValidImageSize(model string, size string) bool {
if model == "cogview-3" {
return true
}
_, ok := constant.ImageSizeRatios[model][size]
_, ok := billing.ImageSizeRatios[model][size]
return ok
}
func getImageSizeRatio(model string, size string) float64 {
ratio, ok := constant.ImageSizeRatios[model][size]
ratio, ok := billing.ImageSizeRatios[model][size]
if !ok {
return 1
}
@@ -83,7 +82,7 @@ func validateImageRequest(imageRequest *relaymodel.ImageRequest, meta *util.Rela
if imageRequest.Prompt == "" {
return openai.ErrorWrapper(errors.New("prompt is required"), "prompt_missing", http.StatusBadRequest)
}
if len(imageRequest.Prompt) > constant.ImagePromptLengthLimitations[imageRequest.Model] {
if len(imageRequest.Prompt) > billing.ImagePromptLengthLimitations[imageRequest.Model] {
return openai.ErrorWrapper(errors.New("prompt is too long"), "prompt_too_long", http.StatusBadRequest)
}
// Number of generated images validation