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