diff --git a/src/llama_stack/providers/utils/job_scheduler/api.py b/src/llama_stack/providers/utils/job_scheduler/api.py index 023e77430..5b3e289e1 100644 --- a/src/llama_stack/providers/utils/job_scheduler/api.py +++ b/src/llama_stack/providers/utils/job_scheduler/api.py @@ -10,10 +10,13 @@ from typing import Protocol class JobStatus(StrEnum): - PENDING = "pending" - RUNNING = "running" - COMPLETED = "completed" + VALIDATING = "validating" FAILED = "failed" + IN_PROGRESS = "in_progress" + FINALIZING = "finalizing" + COMPLETED = "completed" + EXPIRED = "expired" + CANCELLING = "cancelling" CANCELLED = "cancelled"