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

poetry run ruff format .

Browse source
This commit is contained in:
ming luo 2024-06-19 20:59:47 -04:00 • committed by Gabriel Luiz Freitas Almeida
commit c2b9f22172
2 changed files with 11 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/base/langflow/services/variable/base.py
View file

@ -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.
"""

11
src/backend/base/langflow/services/variable/service.py
View file

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