Function execute

python/mirage/server/routers/execute.py:91–133

(
        workspace_id: str,
        request: Request,
        background: bool = Query(False),
)

@router.post("")
async def execute(
        workspace_id: str,
        request: Request,
        background: bool = Query(False),
) -> Response:
    entry = _require_entry(request, workspace_id)
    job_table = request.app.state.jobs
    content_type = request.headers.get("content-type", "")
    req_obj, stdin_bytes = await _parse_execute_body(request, content_type)
    schedule = functools.partial(_schedule_on_runner, entry.runner)
    job = job_table.submit(
        workspace_id=workspace_id,
        command=req_obj.command,
        schedule=schedule,
        coro_factory=_make_coro_factory(
            entry.runner,
            _build_execute_kwargs(req_obj, stdin_bytes),
        ),
    )
    if background:
        return Response(
            content=BackgroundResponse(
                job_id=job.id,
                workspace_id=workspace_id,
                submitted_at=job.submitted_at,
            ).model_dump_json(),
            media_type="application/json",
            status_code=202,
            headers={"X-Mirage-Job-Id": job.id},
        )
    await job_table.wait(job.id)
    if job.status == JobStatus.CANCELED:
        raise HTTPException(status_code=499, detail="job canceled")
    if job.status == JobStatus.FAILED:
        raise HTTPException(status_code=500,
                            detail=job.error or "execute failed")
    result_dict = await io_result_to_dict(job.result)
    return Response(
        content=json.dumps(result_dict),
        media_type="application/json",
        status_code=200,
        headers={"X-Mirage-Job-Id": job.id},
    )


async def _parse_execute_body(