From 25cdf5416d8298b3d42cca7589f13e59b9bf7b36 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Sun, 30 Mar 2025 15:00:25 +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 e26a31f..1a15f3f 100644
--- a/dao/model.go
+++ b/dao/model.go
@@ -210,9 +210,9 @@ type ModelTokensSession struct {
 func FindModelTotalTokensBySessionID(userID int) []ModelTokensSession {
 	var modelTokens []ModelTokensSession
 	if proto.Config.SERVER_SQL_LOG {
-		DB.Debug().Table("model_tokens").Select("session_id, count(*) as token").Where("user_id = ?", userID).Group("session_id").Scan(&modelTokens)
+		DB.Debug().Table("model_tokens").Select("session_id, sum(token) as token").Where("user_id = ?", userID).Group("session_id").Scan(&modelTokens)
 	} else {
-		DB.Table("model_tokens").Select("session_id, count(*) as token").Where("user_id = ?", userID).Group("session_id").Scan(&modelTokens)
+		DB.Table("model_tokens").Select("session_id, sum(token) as token").Where("user_id = ?", userID).Group("session_id").Scan(&modelTokens)
 	}
 	return modelTokens
 }