fix: #2111 make "Attached Messages Count" working

2023-06-24 05:53:39 +08:00 · 2023-06-24 05:53:39 +08:00 · 91d8f9d73e
commit 91d8f9d73e
parent c636993989
1 changed files with 3 additions and 4 deletions
--- a/app/store/chat.ts
+++ b/app/store/chat.ts
@ -405,10 +405,9 @@ export const useChatStore = create<ChatStore>()(
        // 2. pre-defined in-context prompts
        // 3. short term memory: latest n messages
        // 4. newest input message
-        const memoryStartIndex = Math.min(
-          longTermMemoryStartIndex,
-          shortTermMemoryStartIndex,
-        );
+        const memoryStartIndex = shouldSendLongTermMemory
+          ? Math.min(longTermMemoryStartIndex, shortTermMemoryStartIndex)
+          : shortTermMemoryStartIndex;
        // and if user has cleared history messages, we should exclude the memory too.
        const contextStartIndex = Math.max(clearContextIndex, memoryStartIndex);
        const maxTokenThreshold = modelConfig.max_tokens;