From 6a45d28ec9664891dfc4b0506c4b4459d1590656 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Sun, 30 Mar 2025 14:56:53 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=A8=A1=E5=9E=8Btokens?=
 =?UTF-8?q?=E4=BD=BF=E7=94=A8=E9=87=8F=E8=BF=94=E5=9B=9E=EF=BC=8C=E4=BF=AE?=
 =?UTF-8?q?=E6=94=B9sql?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dao/model.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dao/model.go b/dao/model.go
index 3fae14b..e26a31f 100644
--- a/dao/model.go
+++ b/dao/model.go
@@ -225,9 +225,9 @@ type ModelTokensModel struct {
 func FindModelTotalTokensByModelID(userID int) []ModelTokensModel {
 	var modelTokens []ModelTokensModel
 	if proto.Config.SERVER_SQL_LOG {
-		DB.Debug().Table("model_tokens").Select("model_id, count(*) as token").Where("user_id = ?", userID).Group("model_id").Scan(&modelTokens)
+		DB.Debug().Table("model_tokens").Select("model_id, sum(token) as token").Where("user_id = ?", userID).Group("model_id").Scan(&modelTokens)
 	} else {
-		DB.Table("model_tokens").Select("model_id, count(*) as token").Where("user_id = ?", userID).Group("model_id").Scan(&modelTokens)
+		DB.Table("model_tokens").Select("model_id, sum(token) as token").Where("user_id = ?", userID).Group("model_id").Scan(&modelTokens)
 	}
 	return modelTokens
 }