fix user message history

targoninc · Jun 19, 2024 · 00414c3 · 00414c3
1 parent 228d25f
commit 00414c3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lib/features/context/Context.mjs b/lib/features/context/Context.mjs
@@ -80,9 +80,8 @@ export class Context {
 
     static getMessageHistoryWithOnlyLastUserMessage(context) {
         const history = Context.getMessageHistory(context);
-        const userMessages = history.filter(h => h.type === MessageTypes.userMessage);
-        const excludeMessages = userMessages.slice(0, -1);
-        return history.filter(h => !excludeMessages.includes(h));
+        const userMessages = history.filter(h => h.role === "user");
+        return history.filter(h => !userMessages.includes(h));
     }
 
     static getToolFunction() {