From bb3257ed803646a97c6b790cda0e09e04c39506d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 23 Mar 2024 00:21:09 -0300 Subject: [PATCH] Add session_scope context manager to manage session scope --- src/backend/langflow/services/deps.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/backend/langflow/services/deps.py b/src/backend/langflow/services/deps.py index 7d2338b04..685b964a9 100644 --- a/src/backend/langflow/services/deps.py +++ b/src/backend/langflow/services/deps.py @@ -57,6 +57,16 @@ def get_session() -> Generator["Session", None, None]: @contextmanager def session_scope(): + """ + Context manager for managing a session scope. + + Yields: + session: The session object. + + Raises: + Exception: If an error occurs during the session scope. + + """ session = next(get_session()) try: yield session