From 31fc5c4be0a3b14f52914c381bafa2064273e598 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 27 Feb 2024 13:45:02 -0300 Subject: [PATCH] Add session_id check in Vertex class --- src/backend/langflow/graph/vertex/base.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index e87389c7b..fde96d9a5 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -53,6 +53,7 @@ class Vertex: output_component_name in self.id for output_component_name in OUTPUT_COMPONENTS ) + self.has_session_id = None self._custom_component = None self.has_external_input = False self.has_external_output = False @@ -223,6 +224,8 @@ class Vertex: if isinstance(value, dict) } + self.has_session_id = "session_id" in template_dicts + self.required_inputs = [ template_dicts[key]["type"] for key, value in template_dicts.items()