jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor dependencies and imports in

Browse source 
custom_component.py and add new deps.py file
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-14 22:01:37 -03:00
commit d93df82607
3 changed files with 15 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/langflow/interface/custom/custom_component.py
View file

@ -3,14 +3,13 @@ from uuid import UUID
import yaml import yaml
from fastapi import HTTPException from fastapi import HTTPException
from langflow.field_typing.constants import CUSTOM_COMPONENT_SUPPORTED_TYPES from langflow.field_typing.constants import CUSTOM_COMPONENT_SUPPORTED_TYPES
from langflow.interface.custom.component import Component from langflow.interface.custom.component import Component
from langflow.interface.custom.directory_reader import DirectoryReader from langflow.interface.custom.directory_reader import DirectoryReader
from langflow.interface.custom.utils import extract_inner_type, extract_union_types from langflow.interface.custom.utils import extract_inner_type, extract_union_types
from langflow.services.database.models.flow import Flow from langflow.services.database.models.flow import Flow
from langflow.services.database.utils import session_getter 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 from langflow.utils import validate

17
src/backend/langflow/services/getters.py → src/backend/langflow/services/deps.py
View file

@ -1,17 +1,18 @@
from langflow.services import ServiceType, service_manager
from typing import TYPE_CHECKING, Generator from typing import TYPE_CHECKING, Generator
from langflow.services import ServiceType, service_manager
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.services.database.service import DatabaseService
from langflow.services.settings.service import SettingsService
from langflow.services.cache.service import BaseCacheService
from langflow.services.session.service import SessionService
from langflow.services.task.service import TaskService
from langflow.services.chat.service import ChatService
from langflow.services.store.service import StoreService
from sqlmodel import Session 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": def get_settings_service() -> "SettingsService":
try: try:

10
tests/test_setup_superuser.py
View file

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