From 95b0b91e1873fbcac9db9b0f10554c76b0689b84 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 25 Jun 2024 17:03:41 -0300 Subject: [PATCH] chore: Fix issue with message timestamp conversion in MessageBase model --- .../base/langflow/services/database/models/message/model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/services/database/models/message/model.py b/src/backend/base/langflow/services/database/models/message/model.py index 26161ac8b..ed36af3c9 100644 --- a/src/backend/base/langflow/services/database/models/message/model.py +++ b/src/backend/base/langflow/services/database/models/message/model.py @@ -23,14 +23,16 @@ class MessageBase(SQLModel): if message.text is None or not message.sender or not message.sender_name: raise ValueError("The message does not have the required fields (text, sender, sender_name).") if isinstance(message.timestamp, str): - message.timestamp = datetime.fromisoformat(message.timestamp) + timestamp = datetime.fromisoformat(message.timestamp) + else: + timestamp = message.timestamp return cls( sender=message.sender, sender_name=message.sender_name, text=message.text, session_id=message.session_id, files=message.files or [], - timestamp=message.timestamp, + timestamp=timestamp, flow_id=flow_id, )