diff --git a/src/backend/langflow/api/v1/chat.py b/src/backend/langflow/api/v1/chat.py
index 9a2048444..59fe7ac9e 100644
--- a/src/backend/langflow/api/v1/chat.py
+++ b/src/backend/langflow/api/v1/chat.py
@@ -4,6 +4,7 @@ from fastapi import APIRouter, Depends, HTTPException, Query, WebSocket, WebSock
 from fastapi.responses import StreamingResponse
 from langflow.api.utils import build_input_keys_response, format_elapsed_time
 from langflow.api.v1.schemas import BuildStatus, BuiltResponse, InitResponse, ResultDict, StreamData, VerticesOrderResponse
+from langflow.graph.vertex.base import StatelessVertex
 from langflow.processing.process import process_tweaks_on_graph
 from langflow.services.database.models.flow.flow import Flow
 from langflow.graph.graph.base import Graph
diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py
index bd38186e1..e34e0a082 100644
--- a/src/backend/langflow/graph/vertex/base.py
+++ b/src/backend/langflow/graph/vertex/base.py
@@ -376,3 +376,10 @@ class Vertex:
     def _built_object_repr(self):
         # Add a message with an emoji, stars for sucess,
         return "Built sucessfully ✨" if self._built_object is not None else "Failed to build 😵‍💫"
+
+class StatefulVertex(Vertex):
+    pass
+
+
+class StatelessVertex(Vertex):
+    pass