Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛 fix: Model cost calculation error (#74)

Browse Source
This commit is contained in:
Buer 2024-02-26 19:05:10 +08:00 committed by GitHub
parent 9c58190db4
commit ec64bf1ad4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
controller/quota.go
View File

@ -100,7 +100,7 @@ func (q *QuotaInfo) completedQuotaConsumption(usage *types.Usage, tokenName stri
completionRatio := q.modelRatio[1] * q.groupRatio completionRatio := q.modelRatio[1] * q.groupRatio
promptTokens := usage.PromptTokens promptTokens := usage.PromptTokens
completionTokens := usage.CompletionTokens completionTokens := usage.CompletionTokens
quota = int(math.Ceil((float64(promptTokens) + float64(completionTokens)*completionRatio))) quota = int(math.Ceil(((float64(promptTokens) * q.ratio) + (float64(completionTokens) * completionRatio))))
if q.ratio != 0 && quota <= 0 { if q.ratio != 0 && quota <= 0 {
quota = 1 quota = 1
} }