From cbe3a05ebecf7723b3394e338a906f058b2dcc80 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 26 Jun 2024 16:12:12 -0300 Subject: [PATCH] feat: Migrate messages from monitor service to database --- src/backend/base/langflow/memory.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/memory.py b/src/backend/base/langflow/memory.py index 28052525e..f052acdee 100644 --- a/src/backend/base/langflow/memory.py +++ b/src/backend/base/langflow/memory.py @@ -33,7 +33,8 @@ def get_messages( Returns: List[Data]: A list of Data objects representing the retrieved messages. """ - migrate_messages_from_monitor_service_to_database() + with session_scope() as session: + migrate_messages_from_monitor_service_to_database(session) messages_read: list[Message] = [] with session_scope() as session: stmt = select(MessageTable)