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.database.models.variable.model import Variable
from langflow.services.deps import get_session from langflow.services.deps import get_session
class VariableService(Service): class VariableService(Service):
""" """
Abstract base class for a variable service. Abstract base class for a variable service.
""" """
name = "variable_service" name = "variable_service"
@abc.abstractmethod @abc.abstractmethod
@ -39,7 +41,7 @@ class VariableService(Service):
Returns: Returns:
The value of the variable. The value of the variable.
""" """
@abc.abstractmethod @abc.abstractmethod
def list_variables(self, user_id: Union[UUID, str], session: Session) -> list[Optional[str]]: def list_variables(self, user_id: Union[UUID, str], session: Session) -> list[Optional[str]]:
""" """
@ -52,7 +54,7 @@ class VariableService(Service):
Returns: Returns:
A list of variable names. A list of variable names.
""" """
@abc.abstractmethod @abc.abstractmethod
def update_variable(self, user_id: Union[UUID, str], name: str, value: str, session: Session) -> Variable: def update_variable(self, user_id: Union[UUID, str], name: str, value: str, session: Session) -> Variable:
""" """
@ -66,7 +68,7 @@ class VariableService(Service):
Returns: Returns:
The updated variable. The updated variable.
""" """
@abc.abstractmethod @abc.abstractmethod
def delete_variable(self, user_id: Union[UUID, str], name: str, session: Session) -> Variable: def delete_variable(self, user_id: Union[UUID, str], name: str, session: Session) -> Variable:
@ -80,8 +82,8 @@ class VariableService(Service):
Returns: Returns:
The deleted variable. The deleted variable.
""" """
@abc.abstractmethod @abc.abstractmethod
def create_variable( def create_variable(
self, self,
@ -106,4 +108,3 @@ class VariableService(Service):
Returns: Returns:
The created variable. 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): class DatabaseVariableService(VariableService, Service):
def __init__(self, settings_service: "SettingsService"): def __init__(self, settings_service: "SettingsService"):
self.settings_service = settings_service self.settings_service = settings_service
@ -129,9 +128,7 @@ class DatabaseVariableService(VariableService, Service):
return variable return variable
class KubernetesSecretService(VariableService, Service): class KubernetesSecretService(VariableService, Service):
def __init__(self, settings_service: "SettingsService"): def __init__(self, settings_service: "SettingsService"):
self.settings_service = settings_service 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]]: def list_variables(self, user_id: Union[UUID, str], session: Session = Depends(get_session)) -> list[Optional[str]]:
return [] return []
def update_variable( def update_variable(
self, self,
user_id: Union[UUID, str], user_id: Union[UUID, str],
@ -158,15 +155,15 @@ class KubernetesSecretService(VariableService, Service):
session: Session = Depends(get_session), session: Session = Depends(get_session),
): ):
return return
def delete_variable( def delete_variable(
self, self,
user_id: Union[UUID, str], user_id: Union[UUID, str],
name: str, name: str,
session: Session = Depends(get_session), session: Session = Depends(get_session),
): ):
return return
def create_variable( def create_variable(
self, self,
user_id: Union[UUID, str], user_id: Union[UUID, str],