From fb0696c52610738155431c95eb75edf4c5a741b0 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 28 Jan 2025 09:35:05 -0300 Subject: [PATCH] feat: Add text key option to CSVToDataComponent (#5935) Enhance CSV to Data component by introducing a configurable text key parameter, allowing users to specify the column used for text extraction when converting CSV data to Data objects. --- src/backend/base/langflow/components/data/csv_to_data.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/data/csv_to_data.py b/src/backend/base/langflow/components/data/csv_to_data.py index e4fb7e6f0..866062a1e 100644 --- a/src/backend/base/langflow/components/data/csv_to_data.py +++ b/src/backend/base/langflow/components/data/csv_to_data.py @@ -31,6 +31,12 @@ class CSVToDataComponent(Component): display_name="CSV String", info="Paste a CSV string directly to convert to a list of Data objects", ), + MessageTextInput( + name="text_key", + display_name="Text Key", + info="The key to use for the text column. Defaults to 'text'.", + value="text", + ), ] outputs = [ @@ -66,7 +72,7 @@ class CSVToDataComponent(Component): if csv_data: csv_reader = csv.DictReader(io.StringIO(csv_data)) - result = [Data(data=row) for row in csv_reader] + result = [Data(data=row, text_key=self.text_key) for row in csv_reader] if not result: self.status = "The CSV data is empty."