diff --git a/src/backend/langflow/worker.py b/src/backend/langflow/worker.py
index 5a55c3c90..b3872bcd8 100644
--- a/src/backend/langflow/worker.py
+++ b/src/backend/langflow/worker.py
@@ -1,14 +1,13 @@
-from typing import TYPE_CHECKING, Any, Dict, Optional
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
 from asgiref.sync import async_to_sync
 from celery.exceptions import SoftTimeLimitExceeded  # type: ignore
-from loguru import logger
-from rich import print
-
 from langflow.core.celery_app import celery_app
 from langflow.processing.process import Result, generate_result, process_inputs
 from langflow.services.deps import get_session_service
 from langflow.services.manager import initialize_session_service
+from loguru import logger
+from rich import print
 
 if TYPE_CHECKING:
     from langflow.graph.vertex.base import Vertex
@@ -35,7 +34,7 @@ def build_vertex(self, vertex: "Vertex") -> "Vertex":
 @celery_app.task(acks_late=True)
 def process_graph_cached_task(
     data_graph: Dict[str, Any],
-    inputs: Optional[dict] = None,
+    inputs: Optional[Union[dict, List[dict]]] = None,
     clear_cache=False,
     session_id=None,
 ) -> Dict[str, Any]: