Add session_id check in Vertex class

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-27 13:45:02 -03:00
commit 31fc5c4be0

View file

@ -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()