From ec64bf1ad4ca898fce3d811098b5da9aaaca8ce0 Mon Sep 17 00:00:00 2001
From: Buer <42402987+MartialBE@users.noreply.github.com>
Date: Mon, 26 Feb 2024 19:05:10 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix:=20Model=20cost=20calculatio?=
 =?UTF-8?q?n=20error=20(#74)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controller/quota.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/controller/quota.go b/controller/quota.go
index 5bb9f6d4..c2f96954 100644
--- a/controller/quota.go
+++ b/controller/quota.go
@@ -100,7 +100,7 @@ func (q *QuotaInfo) completedQuotaConsumption(usage *types.Usage, tokenName stri
 	completionRatio := q.modelRatio[1] * q.groupRatio
 	promptTokens := usage.PromptTokens
 	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 {
 		quota = 1
 	}