xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-05 16:53:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

optimize ChatPlus page, fixed bug for websocket reconnection

Browse Source
This commit is contained in:
RockYang
2024-09-02 16:35:15 +08:00
parent a16ef6476d
commit d70fcbb9f3
15 changed files with 64 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -270,7 +270,8 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session *types.ChatSessio
tks, _ := utils.CalcTokens(utils.JsonEncode(req.Tools), req.Model)
tokens += tks + promptTokens
for _, v := range messages {
for i := len(messages) - 1; i >= 0; i-- {
v := messages[i]
tks, _ := utils.CalcTokens(v.Content, req.Model)
// 上下文 token 超出了模型的最大上下文长度
if tokens+tks >= session.Model.MaxContext {