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

🔖 chore: Modify PreConsumedQuota Formula (#241)

Browse Source
This commit is contained in:
Buer
2024-05-29 20:49:18 +08:00
committed by GitHub
parent ee662fda04
commit 73cd2b7685
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
relay/relay_util/quota.go
View File

@@ -46,7 +46,7 @@ func NewQuota(c *gin.Context, modelName string, promptTokens int) (*Quota, *type
if quota.price.Type == model.TimesPriceType { if quota.price.Type == model.TimesPriceType {
quota.preConsumedQuota = int(1000 * quota.inputRatio) quota.preConsumedQuota = int(1000 * quota.inputRatio)
} else { } else {
quota.preConsumedQuota = int(float64(quota.promptTokens+config.PreConsumedQuota) * quota.inputRatio) quota.preConsumedQuota = int(float64(quota.promptTokens)*quota.inputRatio) + config.PreConsumedQuota
} }
errWithCode := quota.preQuotaConsumption() errWithCode := quota.preQuotaConsumption()