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

refactor: Update CustomComponent to use data_objects instead of data list

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 19:54:32 -03:00
commit 503437c73a
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/custom/custom_component/custom_component.py
View file

@ -219,7 +219,7 @@ class CustomComponent(BaseComponent):
""" """
if not keys: if not keys:
keys = [] keys = []
data = [] data_objects = []
if not isinstance(data, Sequence): if not isinstance(data, Sequence):
data = [data] data = [data]
for item in data: for item in data:
@ -245,9 +245,9 @@ class CustomComponent(BaseComponent):
else: else:
raise ValueError(f"Invalid data type: {type(item)}") raise ValueError(f"Invalid data type: {type(item)}")
data.append(Data(data=data_dict)) data_objects.append(Data(data=data_dict))
return data return data_objects
def create_references_from_data(self, data: List[Data], include_data: bool = False) -> str: def create_references_from_data(self, data: List[Data], include_data: bool = False) -> str:
""" """