| 34 | |
| 35 | |
| 36 | async def cancel_request( |
| 37 | req_id: str, |
| 38 | logger: logging.Logger = Depends(get_logger), |
| 39 | request_queue: Queue = Depends(get_request_queue), |
| 40 | ): |
| 41 | set_request_id(req_id) |
| 42 | logger.info("Received cancellation request.") |
| 43 | if await cancel_queued_request(req_id, request_queue, logger): |
| 44 | return JSONResponse( |
| 45 | content={ |
| 46 | "success": True, |
| 47 | "message": f"Request {req_id} marked as cancelled.", |
| 48 | } |
| 49 | ) |
| 50 | else: |
| 51 | return JSONResponse( |
| 52 | status_code=404, |
| 53 | content={ |
| 54 | "success": False, |
| 55 | "message": f"Request {req_id} not found in queue.", |
| 56 | }, |
| 57 | ) |
| 58 | |
| 59 | |
| 60 | async def get_queue_status( |