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

🔧 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

View file

@ -11,6 +11,7 @@ from .getters import get_db_service, get_session, get_settings_service
from loguru import logger from loguru import logger
def get_factories_and_deps():
from langflow.services.database import factory as database_factory from langflow.services.database import factory as database_factory
from langflow.services.cache import factory as cache_factory from langflow.services.cache import factory as cache_factory
from langflow.services.chat import factory as chat_factory from langflow.services.chat import factory as chat_factory
@ -19,7 +20,7 @@ from langflow.services.auth import factory as auth_factory
from langflow.services.task import factory as task_factory from langflow.services.task import factory as task_factory
from langflow.services.session import factory as session_service_factory # type: ignore 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(),
@ -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: