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

Refactor component attributes and add is_composition flag

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-20 21:33:41 -03:00
commit 17dee05b78
3 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/langflow/interface/custom/custom_component/component.py
View file

@ -20,7 +20,9 @@ class ComponentFunctionEntrypointNameNullError(HTTPException):
class Component:
ERROR_CODE_NULL: ClassVar[str] = "Python code must be provided."
ERROR_FUNCTION_ENTRYPOINT_NAME_NULL: ClassVar[str] = "The name of the entrypoint function must be provided."
ERROR_FUNCTION_ENTRYPOINT_NAME_NULL: ClassVar[str] = (
"The name of the entrypoint function must be provided."
)
code: Optional[str] = None
_function_entrypoint_name: str = "build"
@ -67,6 +69,10 @@ class Component:
return str(value) if value else ""
def getattr_return_bool(self, value):
if isinstance(value, bool):
return value
def build_template_config(self) -> dict:
if not self.code:
return {}
@ -80,6 +86,7 @@ class Component:
"beta": self.getattr_return_str,
"documentation": self.getattr_return_str,
"icon": self.validate_icon,
"is_composition": self.getattr_return_bool,
}
for attribute, func in attributes_func_mapping.items():