fix: add exception handler

Author: giancarloromeo

services/api-server/src/simcore_service_api_server/api/routes/tasks.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Annotated, Any
 
-from celery_library.errors import TaskNotFoundError
 from common_library.error_codes import create_error_code
 from fastapi import APIRouter, Depends, FastAPI, HTTPException, status
 from models_library.api_schemas_long_running_tasks.base import TaskProgress
@@ -110,16 +109,10 @@ async def get_task_status(
 ):
     task_manager = get_task_manager(app)
 
-    try:
-        task_status = await task_manager.get_task_status(
-            task_filter=_get_task_filter(user_id, product_name),
-            task_uuid=TaskUUID(f"{task_id}"),
-        )
-    except TaskNotFoundError as err:
-        raise HTTPException(
-            status_code=status.HTTP_404_NOT_FOUND,
-            detail="Task not found",
-        ) from err
+    task_status = await task_manager.get_task_status(
+        task_filter=_get_task_filter(user_id, product_name),
+        task_uuid=TaskUUID(f"{task_id}"),
+    )
 
     return TaskStatus(
         task_progress=TaskProgress(
@@ -184,16 +177,10 @@ async def get_task_result(
     task_manager = get_task_manager(app)
     task_filter = _get_task_filter(user_id, product_name)
 
-    try:
-        task_status = await task_manager.get_task_status(
-            task_filter=task_filter,
-            task_uuid=TaskUUID(f"{task_id}"),
-        )
-    except TaskNotFoundError as err:
-        raise HTTPException(
-            status_code=status.HTTP_404_NOT_FOUND,
-            detail="Task not found",
-        ) from err
+    task_status = await task_manager.get_task_status(
+        task_filter=task_filter,
+        task_uuid=TaskUUID(f"{task_id}"),
+    )
 
     if not task_status.is_done:
         raise HTTPException(

services/api-server/src/simcore_service_api_server/exceptions/handlers/__init__.py

@@ -1,6 +1,9 @@
+from os import error
+
 from celery.exceptions import (  # type: ignore[import-untyped] #pylint: disable=no-name-in-module
     CeleryError,
 )
+from celery_library.errors import TaskNotFoundError
 from fastapi import FastAPI
 from fastapi.exceptions import RequestValidationError
 from httpx import HTTPError as HttpxException
@@ -41,6 +44,15 @@ def setup(app: FastAPI, *, is_debug: bool = False):
         ),
     )
 
+    app.add_exception_handler(
+        TaskNotFoundError,
+        make_handler_for_exception(
+            TaskNotFoundError,
+            status.HTTP_404_NOT_FOUND,
+            error_message="The requested task was not found",
+        ),
+    )
+
     app.add_exception_handler(
         CeleryError,
         make_handler_for_exception(