diff --git a/src/backend/base/langflow/services/variable/base.py b/src/backend/base/langflow/services/variable/base.py index 2cf9a3593..019857996 100644 --- a/src/backend/base/langflow/services/variable/base.py +++ b/src/backend/base/langflow/services/variable/base.py @@ -9,10 +9,12 @@ from langflow.services.base import Service from langflow.services.database.models.variable.model import Variable from langflow.services.deps import get_session + class VariableService(Service): """ Abstract base class for a variable service. """ + name = "variable_service" @abc.abstractmethod @@ -39,7 +41,7 @@ class VariableService(Service): Returns: The value of the variable. """ - + @abc.abstractmethod def list_variables(self, user_id: Union[UUID, str], session: Session) -> list[Optional[str]]: """ @@ -52,7 +54,7 @@ class VariableService(Service): Returns: A list of variable names. """ - + @abc.abstractmethod def update_variable(self, user_id: Union[UUID, str], name: str, value: str, session: Session) -> Variable: """ @@ -66,7 +68,7 @@ class VariableService(Service): Returns: The updated variable. - """ + """ @abc.abstractmethod def delete_variable(self, user_id: Union[UUID, str], name: str, session: Session) -> Variable: @@ -80,8 +82,8 @@ class VariableService(Service): Returns: The deleted variable. - """ - + """ + @abc.abstractmethod def create_variable( self, @@ -106,4 +108,3 @@ class VariableService(Service): Returns: The created variable. """ - diff --git a/src/backend/base/langflow/services/variable/service.py b/src/backend/base/langflow/services/variable/service.py index c6cf2a3f1..1f90ae7f3 100644 --- a/src/backend/base/langflow/services/variable/service.py +++ b/src/backend/base/langflow/services/variable/service.py @@ -17,7 +17,6 @@ if TYPE_CHECKING: class DatabaseVariableService(VariableService, Service): - def __init__(self, settings_service: "SettingsService"): self.settings_service = settings_service @@ -129,9 +128,7 @@ class DatabaseVariableService(VariableService, Service): return variable - class KubernetesSecretService(VariableService, Service): - def __init__(self, settings_service: "SettingsService"): self.settings_service = settings_service @@ -149,7 +146,7 @@ class KubernetesSecretService(VariableService, Service): def list_variables(self, user_id: Union[UUID, str], session: Session = Depends(get_session)) -> list[Optional[str]]: return [] - + def update_variable( self, user_id: Union[UUID, str], @@ -158,15 +155,15 @@ class KubernetesSecretService(VariableService, Service): session: Session = Depends(get_session), ): return - - def delete_variable( + + def delete_variable( self, user_id: Union[UUID, str], name: str, session: Session = Depends(get_session), ): return - + def create_variable( self, user_id: Union[UUID, str],