feat: add job cost to metadata

mistralai · Jun 28, 2024 · 93d3b2b · 93d3b2b
1 parent af48070
commit 93d3b2b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/examples/dry_run_job.py b/examples/dry_run_job.py
@@ -32,6 +32,7 @@ async def main():
     print(f"Dataset tokens: {dry_run_job.data_tokens}")
     print(f"Epochs number: {dry_run_job.epochs}")
     print(f"Expected duration: {dry_run_job.expected_duration_seconds}")
+    print(f"Cost: {dry_run_job.cost} {dry_run_job.cost_currency}")
 
 
 if __name__ == "__main__":

diff --git a/src/mistralai/models/jobs.py b/src/mistralai/models/jobs.py
@@ -32,6 +32,8 @@ class JobMetadata(BaseModel):
     train_tokens: int
     epochs: float
     expected_duration_seconds: Optional[int]
+    cost: Optional[float] = None
+    cost_currency: Optional[str] = None
 
 
 class Job(BaseModel):