From 00414c304add21b0204fb18fb2f3ac1b5882014f Mon Sep 17 00:00:00 2001
From: Alexander Fritsch <35202909+loudar@users.noreply.github.com>
Date: Wed, 19 Jun 2024 14:36:04 +0200
Subject: [PATCH] fix user message history

---
 lib/features/context/Context.mjs | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/lib/features/context/Context.mjs b/lib/features/context/Context.mjs
index 2e3563e..bd30d33 100644
--- 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() {