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

Add session_id check in Vertex class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-27 13:45:02 -03:00
commit 31fc5c4be0
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/langflow/graph/vertex/base.py
View file

@ -53,6 +53,7 @@ class Vertex:
output_component_name in self.id output_component_name in self.id
for output_component_name in OUTPUT_COMPONENTS for output_component_name in OUTPUT_COMPONENTS
) )
self.has_session_id = None
self._custom_component = None self._custom_component = None
self.has_external_input = False self.has_external_input = False
self.has_external_output = False self.has_external_output = False
@ -223,6 +224,8 @@ class Vertex:
if isinstance(value, dict) if isinstance(value, dict)
} }
self.has_session_id = "session_id" in template_dicts
self.required_inputs = [ self.required_inputs = [
template_dicts[key]["type"] template_dicts[key]["type"]
for key, value in template_dicts.items() for key, value in template_dicts.items()