From 9d696230bc3e73993df88c0dc8bb916c334e2cdc Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 1 Mar 2024 09:33:58 -0300 Subject: [PATCH] Refactor CustomComponent's data conversion method to use Record objects --- .../custom/custom_component/custom_component.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/interface/custom/custom_component/custom_component.py b/src/backend/langflow/interface/custom/custom_component/custom_component.py index 707ab6611..9364265c2 100644 --- a/src/backend/langflow/interface/custom/custom_component/custom_component.py +++ b/src/backend/langflow/interface/custom/custom_component/custom_component.py @@ -124,15 +124,21 @@ class CustomComponent(Component): self, data: Any, text_key: str = "text", data_key: str = "data" ) -> List[Record]: """ - Convert data into a list of records. + Converts input data into a list of Record objects. Args: - data (Any): The input data to be converted. - text_key (str, optional): The key to extract the text from a dictionary item. Defaults to "text". - data_key (str, optional): The key to extract the data from a dictionary item. Defaults to "data". + data (Any): The input data to be converted. It can be a single item or a sequence of items. + If the input data is a Langchain Document, text_key and data_key are ignored. + + text_key (str, optional): The key to access the text value in each item. Defaults to "text". + data_key (str, optional): The key to access the data value in each item. Defaults to "data". Returns: - List[dict]: A list of records, where each record is a dictionary with 'text' and 'data' keys. + List[Record]: A list of Record objects. + + Raises: + ValueError: If the input data is not of a valid type or if the specified keys are not found in the data. + """ records = [] if not isinstance(data, Sequence):