update job status fields

brighter-ai · Jan 7, 2022 · fac8073 · fac8073
1 parent ec41c47
commit fac8073
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/redact/data_models.py b/redact/data_models.py
@@ -1,6 +1,6 @@
 from enum import Enum
 from httpx import Response
-from pydantic import BaseModel, Field, PositiveInt
+from pydantic import BaseModel, Field, PositiveInt, conint, confloat
 from typing import Optional, List, Tuple
 from uuid import UUID
 
@@ -64,6 +64,8 @@ class JobStatus(BaseModel):
     start_timestamp: Optional[float] = None
     end_timestamp: Optional[float] = None
     estimated_time_to_completion: Optional[float] = None
+    progress: confloat(ge=0.0, le=1.0)
+    total_frames: conint(ge=1)
     warnings: List[str] = Field(default_factory=list)
 
     def is_running(self):