From bdd04004b39313e668847dd803f8d8f01b6f336c Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 28 Aug 2023 16:21:27 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=80=20chore(utils.py):=20add=20type=20?= =?UTF-8?q?hints=20to=20the=20get=5Fdb=5Fmanager()=20function=20for=20bett?= =?UTF-8?q?er=20code=20readability=20and=20maintainability?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/services/utils.py b/src/backend/langflow/services/utils.py index 049e82c0f..6860f8928 100644 --- a/src/backend/langflow/services/utils.py +++ b/src/backend/langflow/services/utils.py @@ -1,7 +1,9 @@ from langflow.services import ServiceType, service_manager from typing import TYPE_CHECKING + if TYPE_CHECKING: + from langflow.services.database.manager import DatabaseManager from langflow.services.settings.manager import SettingsManager @@ -9,7 +11,7 @@ def get_settings_manager() -> "SettingsManager": return service_manager.get(ServiceType.SETTINGS_MANAGER) -def get_db_manager(): +def get_db_manager() -> "DatabaseManager": return service_manager.get(ServiceType.DATABASE_MANAGER)