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 }