fix: args for pipeline (#860)

epam · Jun 4, 2024 · 11b6447 · 11b6447
1 parent 2c3afde
commit 11b6447
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/annotation/annotation/revisions/resources.py b/annotation/annotation/revisions/resources.py
@@ -39,6 +39,7 @@ def get_revisions_without_annotation(
             AnnotatedDoc.file_id == file_id,
             AnnotatedDoc.tenant == x_current_tenant,
         )
+        .order_by(AnnotatedDoc.date)
         .all()
     )
     return revisions or []
diff --git a/jobs/jobs/utils.py b/jobs/jobs/utils.py
@@ -183,7 +183,8 @@ async def convert_previous_jobs_for_inference(
             )
             if not revisions:
                 continue
-            file_data["revision"] = revisions[-1]
+            file_data["revision"] = revisions[-1]["revision"]
+            file_data["file_id"] = file_id
             result.append(file_data)
 
     return result
@@ -749,7 +750,7 @@ async def enrich_annotators_with_usernames(
 
 async def get_annotation_revisions(
     job_id: int, file_id: int, current_tenant: Optional[str], jw_token: str
-) -> Optional[List[int]]:
+) -> Optional[List[dict]]:
     """Get progress of the job with 'job_id' from Pipelines
     or Annotation Manager depending on 'job_mode'."""