ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-15 15:30:26 +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/image.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"
"github.com/songquanpeng/one-api/relay/helper"
relaymodel "github.com/songquanpeng/one-api/relay/model"
"github.com/songquanpeng/one-api/relay/util"
@@ -21,11 +20,11 @@ import (
)
func isWithinRange(element string, value int) bool {
if _, ok := constant.ImageGenerationAmounts[element]; !ok {
if _, ok := billing.ImageGenerationAmounts[element]; !ok {
return false
}
min := constant.ImageGenerationAmounts[element][0]
max := constant.ImageGenerationAmounts[element][1]
min := billing.ImageGenerationAmounts[element][0]
max := billing.ImageGenerationAmounts[element][1]
return value >= min && value <= max
}