jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

chore: Fix issue with message timestamp conversion in MessageBase model

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-25 17:03:41 -03:00
commit 95b0b91e18
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/services/database/models/message/model.py
View file

@ -23,14 +23,16 @@ class MessageBase(SQLModel):
if message.text is None or not message.sender or not message.sender_name: 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).") raise ValueError("The message does not have the required fields (text, sender, sender_name).")
if isinstance(message.timestamp, str): if isinstance(message.timestamp, str):
message.timestamp = datetime.fromisoformat(message.timestamp) timestamp = datetime.fromisoformat(message.timestamp)
else:
timestamp = message.timestamp
return cls( return cls(
sender=message.sender, sender=message.sender,
sender_name=message.sender_name, sender_name=message.sender_name,
text=message.text, text=message.text,
session_id=message.session_id, session_id=message.session_id,
files=message.files or [], files=message.files or [],
timestamp=message.timestamp, timestamp=timestamp,
flow_id=flow_id, flow_id=flow_id,
) )