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

feat(utils.py): add error handling for missing template_config in build_custom_component_template (#2491)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-07-02 18:21:57 -03:00 • committed by GitHub
commit 2eb786cdd9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/base/langflow/custom/utils.py
View file

@ -388,6 +388,17 @@ def build_custom_component_template(
) -> Tuple[Dict[str, Any], CustomComponent | Component]: ) -> Tuple[Dict[str, Any], CustomComponent | Component]:
"""Build a custom component template""" """Build a custom component template"""
try: try:
if not hasattr(custom_component, "template_config"):
raise HTTPException(
status_code=400,
detail={
"error": (
"Please check if you are importing Component correctly."
" It should be `from langflow.custom import Component`"
),
"traceback": traceback.format_exc(),
},
)
if "inputs" in custom_component.template_config: if "inputs" in custom_component.template_config:
return build_custom_component_template_from_inputs(custom_component, user_id=user_id) return build_custom_component_template_from_inputs(custom_component, user_id=user_id)
frontend_node = CustomComponentFrontendNode(**custom_component.template_config) frontend_node = CustomComponentFrontendNode(**custom_component.template_config)