feat: 增加用户 token 消耗统计功能

This commit is contained in:
RockYang
2023-07-01 23:29:24 +08:00
parent 4603de0e70
commit 32484bd32e
4 changed files with 8 additions and 11 deletions

View File

@@ -319,6 +319,10 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
if res.Error != nil {
logger.Error("failed to save reply history message: ", res.Error)
}
// 统计用户 token 数量
h.db.Model(&user).UpdateColumn("tokens", gorm.Expr("tokens + ?",
historyUserMsg.Tokens+historyReplyMsg.Tokens))
}
// 保存当前会话

View File

@@ -240,7 +240,7 @@ type userProfile struct {
Avatar string `json:"avatar"`
ChatConfig types.ChatConfig `json:"chat_config"`
Calls int `json:"calls"`
Tokens int `json:"tokens"`
Tokens int64 `json:"tokens"`
}
func (h *UserHandler) Profile(c *gin.Context) {