fix: assert

Author: giancarloromeo

packages/service-library/src/servicelib/celery/task_manager.py

@@ -1,4 +1,4 @@
-from typing import Any, Protocol
+from typing import Any, Protocol, runtime_checkable
 
 from models_library.progress_bar import ProgressReport
 
@@ -12,6 +12,7 @@
 )
 
 
+@runtime_checkable
 class TaskManager(Protocol):
     async def submit_task(
         self, task_metadata: TaskMetadata, *, task_filter: TaskFilter, **task_param

packages/service-library/src/servicelib/fastapi/celery/app_server.py

@@ -21,8 +21,9 @@ def __init__(self, app: FastAPI):
 
     @property
     def task_manager(self) -> TaskManager:
-        assert self.app.state.task_manager, "Task manager is not initialized"  # nosec
-        task_manager: TaskManager = self.app.state.task_manager
+        task_manager = self.app.state.task_manager
+        assert task_manager, "Task manager is not initialized"  # nosec
+        assert isinstance(task_manager, TaskManager)
         return task_manager
 
     async def start_and_hold(self, startup_completed_event: threading.Event) -> None: