From 4736aaed8cd57508580c27f0093af5e91b43a03e Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Mon, 16 Dec 2024 17:05:54 -0300 Subject: [PATCH] fix: update message update logic and add comment (#5295) * refactor: Simplify message update logic in aupdate_messages function * add comment --- src/backend/base/langflow/memory.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/base/langflow/memory.py b/src/backend/base/langflow/memory.py index 3efdc59c4..ebf8fac3d 100644 --- a/src/backend/base/langflow/memory.py +++ b/src/backend/base/langflow/memory.py @@ -135,10 +135,10 @@ async def aupdate_messages(messages: Message | list[Message]) -> list[Message]: for message in messages: msg = await session.get(MessageTable, message.id) if msg: - if hasattr(message, "data"): - msg = msg.sqlmodel_update(message.data) - else: - msg = msg.sqlmodel_update(message.model_dump(exclude_unset=True, exclude_none=True)) + msg = msg.sqlmodel_update(message.model_dump(exclude_unset=True, exclude_none=True)) + # Convert flow_id to UUID if it's a string preventing error when saving to database + if msg.flow_id and isinstance(msg.flow_id, str): + msg.flow_id = UUID(msg.flow_id) session.add(msg) await session.commit() await session.refresh(msg)