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

Refactor CustomComponent's data conversion method to use Record objects

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-01 09:33:58 -03:00
commit 9d696230bc
1 changed files with 11 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/backend/langflow/interface/custom/custom_component/custom_component.py
View file

@ -124,15 +124,21 @@ class CustomComponent(Component):
self, data: Any, text_key: str = "text", data_key: str = "data" self, data: Any, text_key: str = "text", data_key: str = "data"
) -> List[Record]: ) -> List[Record]:
""" """
Convert data into a list of records. Converts input data into a list of Record objects.
Args: Args:
data (Any): The input data to be converted. data (Any): The input data to be converted. It can be a single item or a sequence of items.
text_key (str, optional): The key to extract the text from a dictionary item. Defaults to "text". If the input data is a Langchain Document, text_key and data_key are ignored.
data_key (str, optional): The key to extract the data from a dictionary item. Defaults to "data".
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: 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 = [] records = []
if not isinstance(data, Sequence): if not isinstance(data, Sequence):