From b99ff9f88ee3f9e7abc4c74d18c0733b149c50b2 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 18:59:14 -0300 Subject: [PATCH] refactor: Add input validation for None values in Component class --- .../base/langflow/custom/custom_component/component.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/backend/base/langflow/custom/custom_component/component.py b/src/backend/base/langflow/custom/custom_component/component.py index 4d4bdc225..127ba63b7 100644 --- a/src/backend/base/langflow/custom/custom_component/component.py +++ b/src/backend/base/langflow/custom/custom_component/component.py @@ -66,6 +66,8 @@ class Component(CustomComponent): def map_inputs(self, inputs: List[InputTypes]): self.inputs = inputs for input_ in inputs: + if input_.name is None: + raise ValueError("Input name cannot be None.") self._inputs[input_.name] = input_ def validate(self, params: dict): @@ -112,6 +114,8 @@ class Component(CustomComponent): # Build the output if it's connected to some other vertex # or if it's not connected to any vertex if not vertex.outgoing_edges or output.name in vertex.edges_source_names: + if output.method is None: + raise ValueError(f"Output {output.name} does not have a method defined.") method: Callable = getattr(self, output.method) if output.cache and output.value != UNDEFINED: _results[output.name] = output.value