xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-09 02:03:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: update image model ratios and adjust quota calculations for billing

Browse Source
This commit is contained in:
Laisky.Cai
2025-03-16 13:07:36 +00:00
parent 4bdc2835bd
commit ff963e819f
2 changed files with 76 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
relay/controller/image.go
View File

@@ -193,12 +193,12 @@ func RelayImageHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatus
switch meta.ChannelType {
case channeltype.Replicate:
// replicate always return 1 image
quota = int64(ratio * imageCostRatio)
quota = int64(ratio*imageCostRatio) * 1000
default:
quota = int64(ratio*imageCostRatio) * int64(imageRequest.N)
quota = int64(ratio*imageCostRatio) * int64(imageRequest.N) * 1000
}
if userQuota-quota < 0 {
if userQuota < quota {
return openai.ErrorWrapper(errors.New("user quota is not enough"), "insufficient_user_quota", http.StatusForbidden)
}