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

Merge pull request #47 from KunMingStar/fix/sendMessage

Browse Source 
fix: 添加messages为空校验
This commit is contained in:
RockYang 2023-11-09 08:21:24 +08:00 committed by GitHub
commit c584b82ddb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -252,15 +252,17 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session *types.ChatSessio
// loading the role context // loading the role context
var messages []types.Message var messages []types.Message
err := utils.JsonDecode(role.Context, &messages) if len(messages) > 0 {
if err == nil { err := utils.JsonDecode(role.Context, &messages)
for _, v := range messages { if err == nil {
tks, _ := utils.CalcTokens(v.Content, req.Model) for _, v := range messages {
if tokens+tks >= types.ModelToTokens[req.Model] { tks, _ := utils.CalcTokens(v.Content, req.Model)
break if tokens+tks >= types.ModelToTokens[req.Model] {
break
}
tokens += tks
chatCtx = append(chatCtx, v)
} }
tokens += tks
chatCtx = append(chatCtx, v)
} }
} }