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)