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

Refactor RecordsAsTextComponent to improve formatting and update status

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-21 16:02:06 -03:00
commit 7fcf3a4cc2
1 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/backend/langflow/components/utilities/RecordsAsText.py
View file

@ -9,7 +9,10 @@ class RecordsAsTextComponent(CustomComponent):
def build_config(self): def build_config(self):
return { return {
"records": {"display_name": "Records", "info": "The records to convert to text."}, "records": {
"display_name": "Records",
"info": "The records to convert to text.",
},
"template": { "template": {
"display_name": "Template", "display_name": "Template",
"info": "The template to use for formatting the records. It must contain the keys {text} and {data}.", "info": "The template to use for formatting the records. It must contain the keys {text} and {data}.",
@ -23,5 +26,10 @@ class RecordsAsTextComponent(CustomComponent):
) -> Text: ) -> Text:
if isinstance(records, Record): if isinstance(records, Record):
records = [records] records = [records]
formated_records = [template.format(text=record.text, data=record.data) for record in records]
return "\n".join(formated_records) formated_records = [
template.format(text=record.text, **record.data) for record in records
]
result_string = "\n".join(formated_records)
self.status = result_string
return result_string