diff --git a/src/backend/langflow/services/utils.py b/src/backend/langflow/services/utils.py
index f1b3a82aa..00c0571c5 100644
--- a/src/backend/langflow/services/utils.py
+++ b/src/backend/langflow/services/utils.py
@@ -5,6 +5,10 @@ from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from langflow.services.database.manager import DatabaseManager
     from langflow.services.settings.manager import SettingsManager
+    from langflow.services.cache.manager import BaseCacheManager
+    from langflow.services.session.manager import SessionManager
+    from langflow.services.task.manager import TaskManager
+    from sqlmodel import Session
 
 
 def get_settings_manager() -> "SettingsManager":
@@ -15,14 +19,18 @@ def get_db_manager() -> "DatabaseManager":
     return service_manager.get(ServiceType.DATABASE_MANAGER)
 
 
-def get_session():
+def get_session() -> "Session":
     db_manager = service_manager.get(ServiceType.DATABASE_MANAGER)
     yield from db_manager.get_session()
 
 
-def get_cache_manager():
+def get_cache_manager() -> "BaseCacheManager":
     return service_manager.get(ServiceType.CACHE_MANAGER)
 
 
-def get_session_manager():
+def get_session_manager() -> "SessionManager":
     return service_manager.get(ServiceType.SESSION_MANAGER)
+
+
+def get_task_manager() -> "TaskManager":
+    return service_manager.get(ServiceType.TASK_MANAGER)