From d4d371fbbcb096e0c7e26db2c039744c52bb6cc9 Mon Sep 17 00:00:00 2001 From: lixiangwuxian Date: Fri, 10 Jan 2025 02:24:10 +0800 Subject: [PATCH] refactor: extend user message history retention and increase maximum history size for improved context management in headmasterHandler --- handler/headmaster/headmaster.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/handler/headmaster/headmaster.go b/handler/headmaster/headmaster.go index 3ef66dc..6404894 100644 --- a/handler/headmaster/headmaster.go +++ b/handler/headmaster/headmaster.go @@ -87,7 +87,7 @@ func enterFormatter(msgIn string) string { } func GenRequestFromUsr(from string, question string) []openai.ChatCompletionMessage { - if _, ok := histories[from]; !ok || histories_time[from].Add(2*time.Minute).Before(time.Now()) { + if _, ok := histories[from]; !ok || histories_time[from].Add(10*time.Minute).Before(time.Now()) { histories[from] = make([]openai.ChatCompletionMessage, 0) histories[from] = append(histories[from], openai.ChatCompletionMessage{ Role: openai.ChatMessageRoleSystem, @@ -108,7 +108,7 @@ func AppendReplyToHistory(from string, reply string) { Content: reply, }) histories_time[from] = time.Now() - for len(histories[from]) > 10 { + for len(histories[from]) > 20 { histories[from] = histories[from][1:] histories[from][0] = openai.ChatCompletionMessage{ Role: openai.ChatMessageRoleSystem,