Fix delete message in memory

src/LLMProviders/chainManager.ts

@@ -315,8 +315,6 @@ export default class ChainManager {
                 `system prompt: ${systemPrompt}\n` +
                 `chat context turns: ${chatContextTurns}\n`
             );
-            console.log("chain RunnableSequence:", ChainManager.chain);
-            console.log("Chat memory:", memory);
           }
 
           for await (const chunk of chatStream) {
@@ -377,6 +375,12 @@ export default class ChainManager {
       }
       updateCurrentAiMessage("");
     }
+    if (debug) {
+      console.log(
+        "Chat memory:",
+        (memory as any).chatHistory.messages.map((msg: any) => msg.content)
+      );
+    }
     return fullAIResponse;
   }
 

src/chatUtils.ts

@@ -46,22 +46,20 @@ export async function updateChatMemory(
   messages: ChatMessage[],
   memoryManager: MemoryManager
 ): Promise<void> {
+  // Clear existing memory
   await memoryManager.clearChatMemory();
 
-  let lastUserMessage = "";
-  for (const msg of messages) {
+  // Process each message in sequence
+  for (let i = 0; i < messages.length - 1; i++) {
+    const msg = messages[i];
+
     if (msg.sender === USER_SENDER) {
-      lastUserMessage = msg.message;
-    } else if (msg.sender === AI_SENDER && lastUserMessage) {
-      await memoryManager
-        .getMemory()
-        .saveContext({ input: lastUserMessage }, { output: msg.message });
-      lastUserMessage = ""; // Reset after saving
+      const nextMsg = messages[i + 1];
+      if (nextMsg?.sender === AI_SENDER) {
+        await memoryManager
+          .getMemory()
+          .saveContext({ input: msg.message }, { output: nextMsg.message });
+      }
     }
   }
-
-  // If there's a trailing user message, save it with an empty AI response
-  if (lastUserMessage) {
-    await memoryManager.getMemory().saveContext({ input: lastUserMessage }, { output: "" });
-  }
 }