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 }