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

🔧 chore(utils.py): refactor get_factories_and_deps function to improve code organization and readability

Browse source 
🔧 chore(utils.py): refactor initialize_services function to use the updated get_factories_and_deps function for better maintainability
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-04 17:42:26 -03:00
commit 537e1a3e05
1 changed files with 30 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

59
src/backend/langflow/services/utils.py
View file

@ -11,35 +11,36 @@ from .getters import get_db_service, get_session, get_settings_service
from loguru import logger from loguru import logger
from langflow.services.database import factory as database_factory def get_factories_and_deps():
from langflow.services.cache import factory as cache_factory from langflow.services.database import factory as database_factory
from langflow.services.chat import factory as chat_factory from langflow.services.cache import factory as cache_factory
from langflow.services.settings import factory as settings_factory from langflow.services.chat import factory as chat_factory
from langflow.services.auth import factory as auth_factory from langflow.services.settings import factory as settings_factory
from langflow.services.task import factory as task_factory from langflow.services.auth import factory as auth_factory
from langflow.services.session import factory as session_service_factory # type: ignore from langflow.services.task import factory as task_factory
from langflow.services.session import factory as session_service_factory # type: ignore
FACTORIES_AND_DEPS = [ return [
(settings_factory.SettingsServiceFactory(), []), (settings_factory.SettingsServiceFactory(), []),
( (
auth_factory.AuthServiceFactory(), auth_factory.AuthServiceFactory(),
[ServiceType.SETTINGS_SERVICE], [ServiceType.SETTINGS_SERVICE],
), ),
( (
database_factory.DatabaseServiceFactory(), database_factory.DatabaseServiceFactory(),
[ServiceType.SETTINGS_SERVICE], [ServiceType.SETTINGS_SERVICE],
), ),
( (
cache_factory.CacheServiceFactory(), cache_factory.CacheServiceFactory(),
[ServiceType.SETTINGS_SERVICE], [ServiceType.SETTINGS_SERVICE],
), ),
(chat_factory.ChatServiceFactory(), []), (chat_factory.ChatServiceFactory(), []),
(task_factory.TaskServiceFactory(), []), (task_factory.TaskServiceFactory(), []),
( (
session_service_factory.SessionServiceFactory(), session_service_factory.SessionServiceFactory(),
[ServiceType.CACHE_SERVICE], [ServiceType.CACHE_SERVICE],
), ),
] ]
def get_or_create_super_user(session: Session, username, password, is_default): def get_or_create_super_user(session: Session, username, password, is_default):
@ -193,7 +194,7 @@ def initialize_services():
""" """
Initialize all the services needed. Initialize all the services needed.
""" """
for factory, dependencies in FACTORIES_AND_DEPS: for factory, dependencies in get_factories_and_deps():
try: try:
service_manager.register_factory(factory, dependencies=dependencies) service_manager.register_factory(factory, dependencies=dependencies)
except Exception as exc: except Exception as exc: