From 163c70e027f9d57d2b8fb78bbd5e40362f5d07e6 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 26 Jun 2024 16:11:39 -0300 Subject: [PATCH] feat: Add flow_id assignment in MessageBase constructor --- .../base/langflow/services/database/models/message/model.py | 2 ++ 1 file changed, 2 insertions(+) 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 5a775d3d9..128088398 100644 --- a/src/backend/base/langflow/services/database/models/message/model.py +++ b/src/backend/base/langflow/services/database/models/message/model.py @@ -34,6 +34,8 @@ class MessageBase(SQLModel): timestamp = datetime.fromisoformat(message.timestamp) else: timestamp = message.timestamp + if not flow_id and message.flow_id: + flow_id = message.flow_id return cls( sender=message.sender, sender_name=message.sender_name,