From 94952c052481ae1b634e2ffb3ad4c2dbd9d2662f Mon Sep 17 00:00:00 2001 From: Eric Date: Tue, 18 Feb 2025 00:14:51 +0000 Subject: [PATCH] fix: update store_message to avoid crash (#6117) --- .../base/langflow/components/helpers/store_message.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/helpers/store_message.py b/src/backend/base/langflow/components/helpers/store_message.py index ae779b399..18fea64e6 100644 --- a/src/backend/base/langflow/components/helpers/store_message.py +++ b/src/backend/base/langflow/components/helpers/store_message.py @@ -61,10 +61,10 @@ class MessageStoreComponent(Component): self.memory.session_id = message.session_id lc_message = message.to_lc_message() await self.memory.aadd_messages([lc_message]) - stored_message = await self.memory.aget_messages() - stored_message = [Message.from_lc_message(m) for m in stored_message] + stored_messages = await self.memory.aget_messages() + stored_messages = [Message.from_lc_message(m) for m in stored_messages] if message.sender: - stored_message = [m for m in stored_message if m.sender == message.sender] + stored_messages = [m for m in stored_messages if m.sender == message.sender] else: await astore_message(message, flow_id=self.graph.flow_id) stored_messages = await aget_messages(