From d61df94849db2cc04e4e6e710b0dc811a090785e Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 24 Nov 2023 12:45:56 -0300 Subject: [PATCH] Add CredentialServiceFactory to utils.py --- .../langflow/services/credentials/factory.py | 15 +++++++++++++++ src/backend/langflow/services/utils.py | 2 ++ 2 files changed, 17 insertions(+) create mode 100644 src/backend/langflow/services/credentials/factory.py diff --git a/src/backend/langflow/services/credentials/factory.py b/src/backend/langflow/services/credentials/factory.py new file mode 100644 index 000000000..c44a43da4 --- /dev/null +++ b/src/backend/langflow/services/credentials/factory.py @@ -0,0 +1,15 @@ +from typing import TYPE_CHECKING + +from langflow.services.credentials.service import CredentialService +from langflow.services.factory import ServiceFactory + +if TYPE_CHECKING: + from langflow.services.settings.service import SettingsService + + +class CredentialServiceFactory(ServiceFactory): + def __init__(self): + super().__init__(CredentialService) + + def create(self, settings_service: "SettingsService"): + return CredentialService(settings_service) diff --git a/src/backend/langflow/services/utils.py b/src/backend/langflow/services/utils.py index 31ae609fb..40d2d7ea4 100644 --- a/src/backend/langflow/services/utils.py +++ b/src/backend/langflow/services/utils.py @@ -18,6 +18,7 @@ def get_factories_and_deps(): from langflow.services.settings import factory as settings_factory from langflow.services.store import factory as store_factory from langflow.services.task import factory as task_factory + from langflow.services.credentials import factory as credentials_factory return [ (settings_factory.SettingsServiceFactory(), []), @@ -40,6 +41,7 @@ def get_factories_and_deps(): [ServiceType.CACHE_SERVICE], ), (store_factory.StoreServiceFactory(), [ServiceType.SETTINGS_SERVICE]), + (credentials_factory.CredentialServiceFactory(), [ServiceType.SETTINGS_SERVICE]), ]