From d4ae7035a112e9eeb5d91868810202ce53b3b115 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 18 Jun 2024 13:07:16 -0300 Subject: [PATCH] refactor: Simplify retrieving messages as text in MemoryComponent --- src/backend/base/langflow/components/helpers/Memory.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/helpers/Memory.py b/src/backend/base/langflow/components/helpers/Memory.py index 2c04b4589..8c2231d79 100644 --- a/src/backend/base/langflow/components/helpers/Memory.py +++ b/src/backend/base/langflow/components/helpers/Memory.py @@ -76,8 +76,6 @@ class MemoryComponent(Component): def retrieve_messages_as_text(self) -> Message: messages = self.retrieve_messages() - messages_text = "\n".join( - [f"{message.data.get('sender_name')}: {message.data.get('text')}" for message in messages] - ) + messages_text = "\n".join(["{sender_name}: {text}".format(**message.data) for message in messages]) self.status = messages_text return Message(text=messages_text)