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 15:00:25 +08:00
parent 6a45d28ec9
commit 25cdf5416d
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

@ -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
}