diff --git a/packages/components/nodes/memory/MongoDBMemory/MongoDBMemory.ts b/packages/components/nodes/memory/MongoDBMemory/MongoDBMemory.ts index 44832466e..df70c4949 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 }, {