From 72418ce4d7ed5fda6e2425a0443b062d458efd32 Mon Sep 17 00:00:00 2001 From: tongkunming Date: Wed, 8 Nov 2023 16:51:36 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E6=B7=BB=E5=8A=A0messages=E4=B8=BA?= =?UTF-8?q?=E7=A9=BA=E6=A0=A1=E9=AA=8C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- api/handler/chatimpl/chat_handler.go | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/api/handler/chatimpl/chat_handler.go b/api/handler/chatimpl/chat_handler.go index d56d00aa..ebbf9f2a 100644 --- a/api/handler/chatimpl/chat_handler.go +++ b/api/handler/chatimpl/chat_handler.go @@ -252,15 +252,17 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session *types.ChatSessio // loading the role context var messages []types.Message - err := utils.JsonDecode(role.Context, &messages) - if err == nil { - for _, v := range messages { - tks, _ := utils.CalcTokens(v.Content, req.Model) - if tokens+tks >= types.ModelToTokens[req.Model] { - break + if len(messages) > 0 { + err := utils.JsonDecode(role.Context, &messages) + if err == nil { + for _, v := range messages { + tks, _ := utils.CalcTokens(v.Content, req.Model) + if tokens+tks >= types.ModelToTokens[req.Model] { + break + } + tokens += tks + chatCtx = append(chatCtx, v) } - tokens += tks - chatCtx = append(chatCtx, v) } }