def build_jobs_router(deployment_manager: DeploymentManager) -> APIRouter:
"""Build the tracking-jobs router bound to *deployment_manager*."""
router = APIRouter(tags=["jobs"])
tracking = get_capability("tracking")
def _submit(model: str, input_obj: Any) -> dict[str, Any]:
get_validated_handle(deployment_manager, model, tracking.name)
job_id = deployment_manager.submit_tracking_job(model, input_obj)
job = deployment_manager.get_tracking_job(job_id)
if job is None:
raise HTTPException(status_code=500, detail=f"Job '{job_id}' was not created.")
return serialize_job(job_id, job)
@router.post("/inference/tracking/jobs", status_code=202, response_model=JobStatus)
async def submit_tracking_job(request: TrackingRequestV1) -> dict[str, Any]:
return _submit(request.model, request.to_input())
@router.post("/inference/tracking/jobs/binary", status_code=202, response_model=JobStatus)
async def submit_tracking_job_binary(request: Request) -> dict[str, Any]:
parsed = await build_capability_binary_request(request, tracking)
return _submit(parsed.model, parsed.to_input())
@router.get("/jobs/{job_id}", response_model=JobStatus)
async def get_job(job_id: str) -> dict[str, Any]:
job = deployment_manager.get_tracking_job(job_id)
if job is None:
raise HTTPException(status_code=404, detail=f"Job '{job_id}' not found")
return serialize_job(job_id, job)
@router.delete("/jobs/{job_id}", response_model=JobStatus)
async def cancel_job(job_id: str) -> dict[str, Any]:
job = deployment_manager.cancel_tracking_job(job_id)
if job is None:
raise HTTPException(status_code=404, detail=f"Job '{job_id}' not found")
return serialize_job(job_id, job)
return router