diff --git a/src/backend/langflow/core/celery_app.py b/src/backend/langflow/core/celery_app.py
index 85223491e..699fc3e14 100644
--- a/src/backend/langflow/core/celery_app.py
+++ b/src/backend/langflow/core/celery_app.py
@@ -1,4 +1,4 @@
-from celery import Celery
+from celery import Celery  # type: ignore
 
 
 def make_celery(app_name: str):
diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py
index ccab84179..d8de30ca9 100644
--- a/src/backend/langflow/graph/vertex/base.py
+++ b/src/backend/langflow/graph/vertex/base.py
@@ -76,7 +76,7 @@ class Vertex:
     def get_task(self):
         # using the task_id, get the task from celery
         # and return it
-        from celery.result import AsyncResult
+        from celery.result import AsyncResult  # type: ignore
 
         return AsyncResult(self.task_id)
 
diff --git a/src/backend/langflow/worker.py b/src/backend/langflow/worker.py
index 2b66b160c..2077bc3ef 100644
--- a/src/backend/langflow/worker.py
+++ b/src/backend/langflow/worker.py
@@ -1,9 +1,8 @@
 from langflow.core.celery_app import celery_app
 from typing import Any, Dict, Optional, Tuple
 from typing import TYPE_CHECKING
-from celery.exceptions import SoftTimeLimitExceeded
+from celery.exceptions import SoftTimeLimitExceeded  # type: ignore
 from langflow.processing.process import (
-    clear_caches_if_needed,
     generate_result,
     process_inputs,
 )
@@ -42,8 +41,9 @@ def process_graph_cached_task(
     session_id=None,
 ) -> Tuple[Any, str]:
     initialize_session_manager()
-    clear_caches_if_needed(clear_cache)
     session_manager = get_session_manager()
+    if clear_cache:
+        session_manager.clear_session(session_id)
     # Load the graph using SessionManager
     langchain_object, artifacts = session_manager.load_session(session_id, data_graph)
     processed_inputs = process_inputs(inputs, artifacts)