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

Fix validation logic for emoji in attributes.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-22 17:12:36 -03:00
commit bb732f5264
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/langflow/interface/custom/attributes.py
View file

@ -6,9 +6,14 @@ import emoji
def validate_icon(value: str, *args, **kwargs): def validate_icon(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(":"):
warnings.warn("Invalid emoji. Please use the :emoji_name: syntax.") if not value.startswith(":") and not value.endswith(":"):
return value return value
elif not value.startswith(":") or not value.endswith(":"):
# emoji should have both starting and ending colons
# so if one of them is missing, we will raise
raise ValueError(f"Invalid emoji. {value} is not a valid emoji.")
emoji_value = emoji.emojize(value, variant="emoji_type") emoji_value = emoji.emojize(value, variant="emoji_type")
if value == emoji_value: if value == emoji_value:
warnings.warn(f"Invalid emoji. {value} is not a valid emoji.") warnings.warn(f"Invalid emoji. {value} is not a valid emoji.")
@ -28,7 +33,7 @@ def getattr_return_bool(value):
ATTR_FUNC_MAPPING = { ATTR_FUNC_MAPPING = {
"display_name": getattr_return_str, "display_name": getattr_return_str,
"description": getattr_return_str, "description": getattr_return_str,
"beta": getattr_return_str, "beta": getattr_return_bool,
"documentation": getattr_return_str, "documentation": getattr_return_str,
"icon": validate_icon, "icon": validate_icon,
"pinned": getattr_return_bool, "pinned": getattr_return_bool,