From e55c0cdcef09aaa7afbab725e7d75c6fb15c413d Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Fri, 6 Sep 2024 14:17:22 +0800
Subject: [PATCH] feat: Update context count logic

- Updated logic for determining context count based on clear messages.
---
 src/renderer/src/services/messages.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/services/messages.ts b/src/renderer/src/services/messages.ts
index 6689e05b..44f5dc73 100644
--- a/src/renderer/src/services/messages.ts
+++ b/src/renderer/src/services/messages.ts
@@ -23,13 +23,15 @@ export function filterContextMessages(messages: Message[]): Message[] {
 
 export function getContextCount(assistant: Assistant, messages: Message[]) {
   const contextCount = assistant?.settings?.contextCount ?? DEFAULT_CONEXTCOUNT
-  const clearIndex = takeRight(messages, contextCount).findLastIndex((message) => message.type === 'clear')
+  const _messages = takeRight(messages, contextCount)
+  const clearIndex = _messages.findLastIndex((message) => message.type === 'clear')
+  const messagesCount = _messages.length
 
   if (clearIndex === -1) {
     return contextCount
   }
 
-  return contextCount - (clearIndex + 1)
+  return messagesCount - (clearIndex + 1)
 }
 
 export function estimateInputTokenCount(text: string) {