update job status fields (#47)

brighter-ai · Jan 10, 2022 · fb2e4be · fb2e4be
1 parent ec41c47
commit fb2e4be
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):