From 0a235ef8d553279e3ad731d3f88a3839b6837fec Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 21 Aug 2023 16:58:22 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=80=20refactor(manager.py):=20refactor?= =?UTF-8?q?=20service=20initialization=20in=20manager.py=20for=20better=20?= =?UTF-8?q?organization=20and=20readability=20=F0=9F=94=80=20refactor(sche?= =?UTF-8?q?ma.py):=20add=20AUTH=5FMANAGER=20service=20type=20to=20the=20Se?= =?UTF-8?q?rviceType=20enum=20for=20better=20organization=20and=20consiste?= =?UTF-8?q?ncy?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/manager.py | 13 +++++-------- src/backend/langflow/services/schema.py | 1 + 2 files changed, 6 insertions(+), 8 deletions(-) diff --git a/src/backend/langflow/services/manager.py b/src/backend/langflow/services/manager.py index f40ae1f25..1592e7612 100644 --- a/src/backend/langflow/services/manager.py +++ b/src/backend/langflow/services/manager.py @@ -87,21 +87,18 @@ def initialize_services(): from langflow.services.cache import factory as cache_factory from langflow.services.chat import factory as chat_factory from langflow.services.settings import factory as settings_factory - from langflow.services.session import factory as session_manager_factory + from langflow.services.auth import factory as auth_factory service_manager.register_factory(settings_factory.SettingsManagerFactory()) + service_manager.register_factory( + auth_factory.AuthManagerFactory(), dependencies=[ServiceType.SETTINGS_MANAGER] + ) service_manager.register_factory( database_factory.DatabaseManagerFactory(), dependencies=[ServiceType.SETTINGS_MANAGER], ) - service_manager.register_factory( - cache_factory.CacheManagerFactory(), dependencies=[ServiceType.SETTINGS_MANAGER] - ) + service_manager.register_factory(cache_factory.CacheManagerFactory()) service_manager.register_factory(chat_factory.ChatManagerFactory()) - service_manager.register_factory( - session_manager_factory.SessionManagerFactory(), - dependencies=[ServiceType.CACHE_MANAGER], - ) # Test cache connection service_manager.get(ServiceType.CACHE_MANAGER) diff --git a/src/backend/langflow/services/schema.py b/src/backend/langflow/services/schema.py index 695763afc..6291a0d0b 100644 --- a/src/backend/langflow/services/schema.py +++ b/src/backend/langflow/services/schema.py @@ -7,6 +7,7 @@ class ServiceType(str, Enum): registered with the service manager. """ + AUTH_MANAGER = "auth_manager" CACHE_MANAGER = "cache_manager" SETTINGS_MANAGER = "settings_manager" DATABASE_MANAGER = "database_manager"