xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-09-18 01:06:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed bug for calculate chat message tokens

Browse Source
This commit is contained in:
RockYang 2025-01-19 13:08:38 +08:00
parent cf817fd8ea
commit 5213bdf08b
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
api/handler/chatimpl/chat_handler.go
View File

@ -379,17 +379,17 @@ func (h *ChatHandler) Tokens(c *gin.Context) {
} }
// 如果没有传入 text 字段，则说明是获取当前 reply 总的 token 消耗（带上下文） // 如果没有传入 text 字段，则说明是获取当前 reply 总的 token 消耗（带上下文）
if data.Text == "" && data.ChatId != "" { //if data.Text == "" && data.ChatId != "" {
var item model.ChatMessage // var item model.ChatMessage
userId, _ := c.Get(types.LoginUserID) // userId, _ := c.Get(types.LoginUserID)
res := h.DB.Where("user_id = ?", userId).Where("chat_id = ?", data.ChatId).Last(&item) // res := h.DB.Where("user_id = ?", userId).Where("chat_id = ?", data.ChatId).Last(&item)
if res.Error != nil { // if res.Error != nil {
resp.ERROR(c, res.Error.Error()) // resp.ERROR(c, res.Error.Error())
return // return
} // }
resp.SUCCESS(c, item.Tokens) // resp.SUCCESS(c, item.Tokens)
return // return
} //}
tokens, err := utils.CalcTokens(data.Text, data.Model) tokens, err := utils.CalcTokens(data.Text, data.Model)
if err != nil { if err != nil {