From 73cd2b768553a37540e487ae316168fb1d1f8226 Mon Sep 17 00:00:00 2001
From: Buer <42402987+MartialBE@users.noreply.github.com>
Date: Wed, 29 May 2024 20:49:18 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=96=20chore:=20Modify=20PreConsumedQuo?=
 =?UTF-8?q?ta=20Formula=20(#241)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/relay/relay_util/quota.go b/relay/relay_util/quota.go
index e1a92cbb..de5032f6 100644
--- a/relay/relay_util/quota.go
+++ b/relay/relay_util/quota.go
@@ -46,7 +46,7 @@ func NewQuota(c *gin.Context, modelName string, promptTokens int) (*Quota, *type
 	if quota.price.Type == model.TimesPriceType {
 		quota.preConsumedQuota = int(1000 * quota.inputRatio)
 	} else {
-		quota.preConsumedQuota = int(float64(quota.promptTokens+config.PreConsumedQuota) * quota.inputRatio)
+		quota.preConsumedQuota = int(float64(quota.promptTokens)*quota.inputRatio) + config.PreConsumedQuota
 	}
 
 	errWithCode := quota.preQuotaConsumption()