From 3422cbd8e1546b9dc19af3399fc94feb717d83cf Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 18:52:21 -0300 Subject: [PATCH] refactor: Update Vertex class to use type hinting for required_inputs and optional_inputs --- src/backend/base/langflow/graph/vertex/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/graph/vertex/base.py b/src/backend/base/langflow/graph/vertex/base.py index 37714e77d..86a65b9cf 100644 --- a/src/backend/base/langflow/graph/vertex/base.py +++ b/src/backend/base/langflow/graph/vertex/base.py @@ -204,8 +204,8 @@ class Vertex: self.has_session_id = "session_id" in template_dicts - self.required_inputs = [] - self.optional_inputs = [] + self.required_inputs: list[str] = [] + self.optional_inputs: list[str] = [] for value_dict in template_dicts.values(): list_to_append = self.required_inputs if value_dict.get("required") else self.optional_inputs