fix(saq-controller): Return a 404 status code when the job is not found

litestar_saq/controllers.py

@@ -4,6 +4,7 @@
 
 from litestar import Controller, MediaType, get, post
 from litestar.exceptions import NotFoundException
+from litestar.status_codes import HTTP_202_ACCEPTED
 
 from litestar_saq import info, urls
 
@@ -77,6 +78,7 @@ async def job_detail(self, task_queues: dict[str, Queue], queue_id: str, job_id:
         cache=False,
         summary="Job Retry",
         description="Retry a failed job..",
+        status_code=HTTP_202_ACCEPTED,
     )
     async def job_retry(self, task_queues: dict[str, Queue], queue_id: str, job_id: str) -> dict:
         """Retry job."""
@@ -96,6 +98,7 @@ async def job_retry(self, task_queues: dict[str, Queue], queue_id: str, job_id:
         cache=False,
         summary="Job Abort",
         description="Abort active job.",
+        status_code=HTTP_202_ACCEPTED,
     )
     async def job_abort(self, task_queues: dict[str, Queue], queue_id: str, job_id: str) -> dict:
         """Abort job."""

litestar_saq/info.py

@@ -2,7 +2,7 @@
 
 from typing import TYPE_CHECKING, cast
 
-from .exceptions import LitestarSaqError
+from litestar.exceptions import NotFoundException
 
 if TYPE_CHECKING:
     from .base import Job, Queue
@@ -11,6 +11,6 @@
 async def job(queue: Queue, job_id: str) -> Job:
     job = await queue._get_job_by_id(job_id)  # noqa: SLF001
     if not job:
-        msg = "Could not find job ID %s"
-        raise LitestarSaqError(msg, job_id)
+        msg = f"Could not find job ID {job_id}"
+        raise NotFoundException(msg)
     return cast("Job", job)