From 080af42802fc60d861307307eacac1e47f814b0e Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 17 Jun 2024 11:31:47 -0300 Subject: [PATCH] Refactor memory.py to ensure messages have a timestamp before adding them to the database --- src/backend/base/langflow/memory.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/memory.py b/src/backend/base/langflow/memory.py index 6b2ceca50..e89682969 100644 --- a/src/backend/base/langflow/memory.py +++ b/src/backend/base/langflow/memory.py @@ -74,7 +74,8 @@ def add_messages(messages: Message | list[Message], flow_id: Optional[str] = Non messages_models: list[MessageModel] = [] for msg in messages: - msg.timestamp = monitor_service.get_timestamp() + if not msg.timestamp: + msg.timestamp = monitor_service.get_timestamp() messages_models.append(MessageModel.from_message(msg, flow_id=flow_id)) for message_model in messages_models: