Provide output files in job model

molinfo-vienna · Dec 21, 2024 · 60e22d3 · 60e22d3
1 parent ad92c13
commit 60e22d3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/nerdd_backend/models/job.py b/nerdd_backend/models/job.py
@@ -3,7 +3,12 @@
 
 from pydantic import BaseModel
 
-__all__ = ["Job", "JobCreate", "JobPublic", "JobUpdate", "JobInternal"]
+__all__ = ["Job", "JobCreate", "JobPublic", "JobUpdate", "JobInternal", "OutputFile"]
+
+
+class OutputFile(BaseModel):
+    format: str
+    url: str
 
 
 class Job(BaseModel):
@@ -33,6 +38,7 @@ class JobPublic(Job):
     num_pages_total: Optional[int]
     num_pages_processed: int
     page_size: int
+    output_files: List[OutputFile]
     job_url: str
     results_url: str