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

Refactor getattr_return_str method in Component class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-09 09:06:15 -03:00
commit 98f739db99
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -68,9 +68,9 @@ class Component:
return validate.create_function(self.code, self._function_entrypoint_name) return validate.create_function(self.code, self._function_entrypoint_name)
def getattr_return_str(self, component, attribute): def getattr_return_str(self, component, value):
attribute = getattr(component, attribute) value = getattr(component, value)
return str(attribute) if attribute else "" return str(value) if value else ""
def build_template_config(self) -> dict: def build_template_config(self) -> dict:
if not self.code: if not self.code:
@ -89,11 +89,11 @@ class Component:
for attribute, func in attributes_func_mapping.items(): for attribute, func in attributes_func_mapping.items():
if hasattr(component_instance, attribute): if hasattr(component_instance, attribute):
template_config[attribute] = func(component_instance, attribute) template_config[attribute] = func(component=component_instance, value=attribute)
return template_config return template_config
def validate_icon(self, _, value: str): def validate_icon(self, value: str, *args, **kwargs):
# we are going to use the emoji library to validate the emoji # we are going to use the emoji library to validate the emoji
# emojis can be defined using the :emoji_name: syntax # emojis can be defined using the :emoji_name: syntax
if not value.startswith(":") or not value.endswith(":"): if not value.startswith(":") or not value.endswith(":"):