From 2fcbfa25a5fcf6f6708bb6bb556f3a6de2497d52 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 5 Aug 2023 23:18:31 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=89=20feat(utils.py):=20add=20new=20ut?= =?UTF-8?q?ility=20function=20`get=5Fsettings=5Fmanager()`=20to=20retrieve?= =?UTF-8?q?=20the=20settings=20manager=20from=20the=20service=20manager=20?= =?UTF-8?q?=F0=9F=8E=89=20feat(utils.py):=20add=20new=20utility=20function?= =?UTF-8?q?=20`get=5Fsession()`=20to=20retrieve=20a=20session=20from=20the?= =?UTF-8?q?=20database=20manager?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/utils.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 src/backend/langflow/services/utils.py diff --git a/src/backend/langflow/services/utils.py b/src/backend/langflow/services/utils.py new file mode 100644 index 000000000..07c67dfbe --- /dev/null +++ b/src/backend/langflow/services/utils.py @@ -0,0 +1,14 @@ +from langflow.services import ServiceType, service_manager +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from langflow.services.settings.manager import SettingsManager + + +def get_settings_manager() -> "SettingsManager": + return service_manager.get(ServiceType.SETTINGS_MANAGER) + + +def get_session(): + db_manager = service_manager.get(ServiceType.DATABASE_MANAGER) + yield from db_manager.get_session()