From 74fe7852896a36129d71ebb8d6a056cebb4bb634 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 22 Jun 2024 16:57:15 -0300 Subject: [PATCH] fix: make optional import happen only if used --- src/backend/base/langflow/services/variable/factory.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/services/variable/factory.py b/src/backend/base/langflow/services/variable/factory.py index b884a1ac3..d9d7c2fb0 100644 --- a/src/backend/base/langflow/services/variable/factory.py +++ b/src/backend/base/langflow/services/variable/factory.py @@ -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)