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

Fix CustomComponent get_build_method return value

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-14 09:39:19 -03:00
commit 5de717c770
1 changed files with 7 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -76,7 +76,7 @@ class CustomComponent(Component):
@property @property
def tree(self): def tree(self):
return self.get_code_tree(self.code) return self.get_code_tree(self.code or "")
@property @property
def get_function_entrypoint_args(self) -> list: def get_function_entrypoint_args(self) -> list:
@ -104,11 +104,11 @@ class CustomComponent(Component):
@cachedmethod(operator.attrgetter("cache")) @cachedmethod(operator.attrgetter("cache"))
def get_build_method(self): def get_build_method(self):
if not self.code: if not self.code:
return [] return {}
component_classes = [cls for cls in self.tree["classes"] if self.code_class_base_inheritance in cls["bases"]] component_classes = [cls for cls in self.tree["classes"] if self.code_class_base_inheritance in cls["bases"]]
if not component_classes: if not component_classes:
return [] return {}
# Assume the first Component class is the one we're interested in # Assume the first Component class is the one we're interested in
component_class = component_classes[0] component_class = component_classes[0]
@ -116,19 +116,13 @@ class CustomComponent(Component):
method for method in component_class["methods"] if method["name"] == self.function_entrypoint_name method for method in component_class["methods"] if method["name"] == self.function_entrypoint_name
] ]
if not build_methods: return build_methods[0] if build_methods else {}
return []
return build_methods[0]
@property @property
def get_function_entrypoint_return_type(self) -> List[Any]: def get_function_entrypoint_return_type(self) -> List[Any]:
build_method = self.get_build_method() build_method = self.get_build_method()
if not build_method: if not build_method or not build_method.get("has_return"):
return [] return []
elif not build_method["has_return"]:
return []
return_type = build_method["return_type"] return_type = build_method["return_type"]
# If list or List is in the return type, then we remove it and return the inner type # If list or List is in the return type, then we remove it and return the inner type
@ -137,10 +131,7 @@ class CustomComponent(Component):
# If the return type is not a Union, then we just return it as a list # If the return type is not a Union, then we just return it as a list
if not hasattr(return_type, "__origin__") or return_type.__origin__ != Union: if not hasattr(return_type, "__origin__") or return_type.__origin__ != Union:
if isinstance(return_type, list): return return_type if isinstance(return_type, list) else [return_type]
return return_type
return [return_type]
# If the return type is a Union, then we need to parse itx # If the return type is a Union, then we need to parse itx
return_type = extract_union_types_from_generic_alias(return_type) return_type = extract_union_types_from_generic_alias(return_type)
return return_type return return_type
@ -206,9 +197,7 @@ class CustomComponent(Component):
"""Returns a function that returns the value at the given index in the iterable.""" """Returns a function that returns the value at the given index in the iterable."""
def get_index(iterable: List[Any]): def get_index(iterable: List[Any]):
if iterable: return iterable[value] if iterable else iterable
return iterable[value]
return iterable
return get_index return get_index