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

🐛 fix(endpoints.py): add validation to ensure that custom components are in a single category

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-17 22:33:52 -03:00
commit b772bbe976
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/api/v1/endpoints.py
View file

@ -57,6 +57,13 @@ def get_all():
logger.info(f"Loading {len(custom_component_dicts)} category(ies)") logger.info(f"Loading {len(custom_component_dicts)} category(ies)")
for custom_component_dict in custom_component_dicts: for custom_component_dict in custom_component_dicts:
# custom_component_dict is a dict of dicts # custom_component_dict is a dict of dicts
if len(custom_component_dict) > 1:
raise ValueError(
f"Custom components must be in a single category. Found {len(custom_component_dict)} categories"
)
elif len(custom_component_dict) == 0:
continue
category = list(custom_component_dict.keys())[0] category = list(custom_component_dict.keys())[0]
logger.info( logger.info(
f"Loading {len(custom_component_dict[category])} component(s) from category {category}" f"Loading {len(custom_component_dict[category])} component(s) from category {category}"