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

Add validation for list items in records_to_text function

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-30 16:42:12 -03:00
commit c5fb239852
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/helpers/record.py
View file

@ -29,6 +29,11 @@ def records_to_text(template: str, records: list[Record]) -> str:
if isinstance(records, Record): if isinstance(records, Record):
records = [records] records = [records]
# Check if there are any format strings in the template # Check if there are any format strings in the template
if any(not isinstance(record, Record) for record in records):
item_that_is_not_record = next((record for record in records if not isinstance(record, Record)), None)
raise ValueError(
f"All items in the list must be of type Record. Found: {item_that_is_not_record} of type {type(item_that_is_not_record)}."
)
formated_records = [template.format(data=record.data, **record.data) for record in records] formated_records = [template.format(data=record.data, **record.data) for record in records]
return "\n".join(formated_records) return "\n".join(formated_records)