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

Refactor memory.py to ensure messages have a timestamp before adding them to the database

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 11:31:47 -03:00
commit 080af42802
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/backend/base/langflow/memory.py
View file

@ -74,6 +74,7 @@ def add_messages(messages: Message | list[Message], flow_id: Optional[str] = Non
messages_models: list[MessageModel] = [] messages_models: list[MessageModel] = []
for msg in messages: for msg in messages:
if not msg.timestamp:
msg.timestamp = monitor_service.get_timestamp() msg.timestamp = monitor_service.get_timestamp()
messages_models.append(MessageModel.from_message(msg, flow_id=flow_id)) messages_models.append(MessageModel.from_message(msg, flow_id=flow_id))