Add CredentialServiceFactory to utils.py

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-24 12:45:56 -03:00
commit d61df94849
2 changed files with 17 additions and 0 deletions

View file

@ -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)

View file

@ -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]),
]