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

🐛 fix(custom.py): improve error message formatting for invalid entrypoint function name and return type

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-09 11:08:50 -03:00
commit 19b84247cf
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -172,7 +172,10 @@ class CustomComponent(BaseModel):
status_code=400, status_code=400,
detail={ detail={
"error": "Invalid entrypoint function name", "error": "Invalid entrypoint function name",
"traceback": f"There needs to be at least one entrypoint function named '{self.function_entrypoint_name}' and it needs to return one of the types from this list {str(self.return_type_valid_list)}.", "traceback": (
f"There needs to be at least one entrypoint function named '{self.function_entrypoint_name}'"
f" and it needs to return one of the types from this list {str(self.return_type_valid_list)}.",
),
}, },
) )
@ -182,7 +185,10 @@ class CustomComponent(BaseModel):
status_code=400, status_code=400,
detail={ detail={
"error": "Invalid entrypoint function return", "error": "Invalid entrypoint function return",
"traceback": f"The entrypoint function return '{return_type}' needs to be an item from this list {str(self.return_type_valid_list)}.", "traceback": (
f"The entrypoint function return '{return_type}' needs to be an item "
f"from this list {str(self.return_type_valid_list)}."
),
}, },
) )