update parser builder

axonzeta · Oct 29, 2024 · 28ba6f3 · 28ba6f3
1 parent c7fd2f0
commit 28ba6f3
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/extension/background/APIHandler/AILanguageModelHandler.ts b/src/extension/background/APIHandler/AILanguageModelHandler.ts
@@ -75,22 +75,22 @@ class AILanguageModelHandler {
     let tokenCount = systemPrompt
       ? await AIPrompter.countTokens(systemPrompt, manifest.tokens.method)
       : 0
-    const messagesWindow = [
-      ...systemPrompt
-        ? [{ content: systemPrompt, role: 'system' }]
-        : []
-    ]
 
     const history = [...initialPrompts, ...messages]
+    const countedMessages=[]
     for (let i = history.length - 1; i >= 0; i--) {
       const message = history[i]
       tokenCount += await AIPrompter.countTokens(message.content, manifest.tokens.method)
       if (tokenCount > manifest.tokens.max) {
         break
       }
-      messagesWindow.unshift(message)
+      countedMessages.unshift(message)
     }
-    messagesWindow.reverse()
+    const messagesWindow = [
+      ...systemPrompt
+        ? [{ content: systemPrompt, role: 'system' }, ...countedMessages]
+        : countedMessages
+    ]
 
     // Send to the template
     const template = new Template(promptConfig.template)