ljydlv
/
saw-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修复模型tokens使用量返回，修改sql

This commit is contained in:
junleea 2025-03-30 14:56:53 +08:00
parent eb4c0f9404
commit 6a45d28ec9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dao/model.go
View File

@ -225,9 +225,9 @@ type ModelTokensModel struct {
func FindModelTotalTokensByModelID(userID int) []ModelTokensModel { func FindModelTotalTokensByModelID(userID int) []ModelTokensModel {
var modelTokens []ModelTokensModel var modelTokens []ModelTokensModel
if proto.Config.SERVER_SQL_LOG { 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 { } 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 return modelTokens
} }