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

fix: make optional import happen only if used

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-22 16:57:15 -03:00
commit 74fe785289
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/services/variable/factory.py
View file

@ -1,7 +1,7 @@
from typing import TYPE_CHECKING
from langflow.services.factory import ServiceFactory
from langflow.services.variable.service import VariableService, DatabaseVariableService, KubernetesSecretService
from langflow.services.variable.service import DatabaseVariableService, VariableService
if TYPE_CHECKING:
from langflow.services.settings.service import SettingsService
@ -16,6 +16,9 @@ class VariableServiceFactory(ServiceFactory):
# based on the settings_service
if settings_service.settings.variable_store == "kubernetes":
# Keep it here to avoid import errors
from langflow.services.variable.service import KubernetesSecretService
return KubernetesSecretService(settings_service)
else:
return DatabaseVariableService(settings_service)