Refactor dependencies and imports in

custom_component.py and add new deps.py file

src/backend/langflow/interface/custom/custom_component.py

@@ -3,14 +3,13 @@ from uuid import UUID
 
 import yaml
 from fastapi import HTTPException
-
 from langflow.field_typing.constants import CUSTOM_COMPONENT_SUPPORTED_TYPES
 from langflow.interface.custom.component import Component
 from langflow.interface.custom.directory_reader import DirectoryReader
 from langflow.interface.custom.utils import extract_inner_type, extract_union_types
 from langflow.services.database.models.flow import Flow
 from langflow.services.database.utils import session_getter
-from langflow.services.getters import get_db_service
+from langflow.services.deps import get_db_service
 from langflow.utils import validate
 
 

src/backend/langflow/services/deps.py

@@ -0,0 +1,54 @@
+from typing import TYPE_CHECKING, Generator
+
+from langflow.services import ServiceType, service_manager
+
+if TYPE_CHECKING:
+    from sqlmodel import Session
+
+    from langflow.services.cache.service import BaseCacheService
+    from langflow.services.chat.service import ChatService
+    from langflow.services.database.service import DatabaseService
+    from langflow.services.session.service import SessionService
+    from langflow.services.settings.service import SettingsService
+    from langflow.services.store.service import StoreService
+    from langflow.services.task.service import TaskService
+
+
+def get_settings_service() -> "SettingsService":
+    try:
+        return service_manager.get(ServiceType.SETTINGS_SERVICE)
+    except ValueError:
+        # initialize settings service
+        from langflow.services.manager import initialize_settings_service
+
+        initialize_settings_service()
+        return service_manager.get(ServiceType.SETTINGS_SERVICE)
+
+
+def get_db_service() -> "DatabaseService":
+    return service_manager.get(ServiceType.DATABASE_SERVICE)
+
+
+def get_session() -> Generator["Session", None, None]:
+    db_service = service_manager.get(ServiceType.DATABASE_SERVICE)
+    yield from db_service.get_session()
+
+
+def get_cache_service() -> "BaseCacheService":
+    return service_manager.get(ServiceType.CACHE_SERVICE)
+
+
+def get_session_service() -> "SessionService":
+    return service_manager.get(ServiceType.SESSION_SERVICE)
+
+
+def get_task_service() -> "TaskService":
+    return service_manager.get(ServiceType.TASK_SERVICE)
+
+
+def get_chat_service() -> "ChatService":
+    return service_manager.get(ServiceType.CHAT_SERVICE)
+
+
+def get_store_service() -> "StoreService":
+    return service_manager.get(ServiceType.STORE_SERVICE)

tests/test_setup_superuser.py

@@ -6,7 +6,7 @@ from langflow.services.utils import teardown_superuser
 
 # @patch("langflow.services.deps.get_session")
 # @patch("langflow.services.utils.create_super_user")
-# @patch("langflow.services.getters.get_settings_service")
+# @patch("langflow.services.deps.get_settings_service")
 # # @patch("langflow.services.utils.verify_password")
 # def test_setup_superuser(
 #     mock_get_session, mock_create_super_user, mock_get_settings_service
@@ -87,8 +87,8 @@ from langflow.services.utils import teardown_superuser
 #     assert str(actual_expr) == str(expected_expr)
 
 
-@patch("langflow.services.getters.get_settings_service")
-@patch("langflow.services.getters.get_session")
+@patch("langflow.services.deps.get_settings_service")
+@patch("langflow.services.deps.get_session")
 def test_teardown_superuser_default_superuser(mock_get_session, mock_get_settings_service):
     mock_settings_service = MagicMock()
     mock_settings_service.auth_settings.AUTO_LOGIN = True
@@ -113,8 +113,8 @@ def test_teardown_superuser_default_superuser(mock_get_session, mock_get_setting
     mock_session.commit.assert_called_once()
 
 
-@patch("langflow.services.getters.get_settings_service")
-@patch("langflow.services.getters.get_session")
+@patch("langflow.services.deps.get_settings_service")
+@patch("langflow.services.deps.get_session")
 def test_teardown_superuser_no_default_superuser(mock_get_session, mock_get_settings_service):
     ADMIN_USER_NAME = "admin_user"
     mock_settings_service = MagicMock()