refactor

Author: Andrei Neagu

packages/service-library/src/servicelib/fastapi/long_running_tasks/_context_manager.py

@@ -119,18 +119,18 @@ async def _wait_for_task_result() -> Any:
         _logger.debug("%s, %s", f"{task_id=}", f"{result=}")
 
         yield result
-    except Exception as e:
-        error = TaskExceptionError(
-            task_id=task_id,
-            exception=e,
-            traceback=f"check remote side for logs, HINT: service replying to: '{client._base_url}' for '{task_id=}'",
-        )
-        _logger.warning(f"{error}")
-        raise error from e
     except TimeoutError as e:
         await client.cancel_and_delete_task(task_id)
         raise TaskClientTimeoutError(
             task_id=task_id,
             timeout=task_timeout,
             exception=e,
         ) from e
+    except Exception as e:
+        error = TaskExceptionError(
+            task_id=task_id,
+            exception=e,
+            traceback=f"check remote side for logs, HINT: service replying to: '{client._base_url}' for '{task_id=}'",  # noqa: SLF001  # pylint:disable=protected-access
+        )
+        _logger.warning(f"{error}")
+        raise error from e