From 416e57380ea7ce2e66f89aded61b249ff3eef3b2 Mon Sep 17 00:00:00 2001
From: Arif Romadhan <arifromadhan19@gmail.com>
Date: Tue, 15 Apr 2025 01:39:37 +0700
Subject: [PATCH] add history chat timestamp in mongodb (#4296)

---
 .../nodes/memory/MongoDBMemory/MongoDBMemory.ts        | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/packages/components/nodes/memory/MongoDBMemory/MongoDBMemory.ts b/packages/components/nodes/memory/MongoDBMemory/MongoDBMemory.ts
index 44832466..df70c494 100644
--- a/packages/components/nodes/memory/MongoDBMemory/MongoDBMemory.ts
+++ b/packages/components/nodes/memory/MongoDBMemory/MongoDBMemory.ts
@@ -155,7 +155,10 @@ class BufferMemoryExtended extends FlowiseMemory implements MemoryMethods {
 
         if (input) {
             const newInputMessage = new HumanMessage(input.text)
-            const messageToAdd = [newInputMessage].map((msg) => msg.toDict())
+            const messageToAdd = [newInputMessage].map((msg) => ({
+                ...msg.toDict(),
+                timestamp: new Date() // Add timestamp to the message
+            }))
             await collection.updateOne(
                 { sessionId: id },
                 {
@@ -167,7 +170,10 @@ class BufferMemoryExtended extends FlowiseMemory implements MemoryMethods {
 
         if (output) {
             const newOutputMessage = new AIMessage(output.text)
-            const messageToAdd = [newOutputMessage].map((msg) => msg.toDict())
+            const messageToAdd = [newOutputMessage].map((msg) => ({
+                ...msg.toDict(),
+                timestamp: new Date() // Add timestamp to the message
+            }))
             await collection.updateOne(
                 { sessionId: id },
                 {