Basic option selection

src/documents/serialisers.py

@@ -2410,6 +2410,13 @@ class TasksViewSerializer(OwnedObjectSerializer):
         duplicates = _get_viewable_duplicates(document, user)
         return list(duplicates.values("id", "title", "deleted_at"))
 
+class RetryTaskSerializer(serializers.Serializer):
+    clean = serializers.BooleanField(
+        default=False,
+        write_only=True,
+        required=False,
+    )
+
 
 class RunTaskViewSerializer(serializers.Serializer[dict[str, Any]]):
     task_name = serializers.ChoiceField(

src/documents/views.py

@@ -189,6 +189,7 @@ from documents.serialisers import NotesSerializer
 from documents.serialisers import PostDocumentSerializer
 from documents.serialisers import RemovePasswordDocumentsSerializer
 from documents.serialisers import ReprocessDocumentsSerializer
+from documents.serialisers import RetryTaskSerializer
 from documents.serialisers import RotateDocumentsSerializer
 from documents.serialisers import RunTaskViewSerializer
 from documents.serialisers import SavedViewSerializer
@@ -3471,9 +3472,16 @@ class TasksViewSet(ReadOnlyModelViewSet):
     @action(methods=["post"], detail=True)
     def retry(self, request, pk=None):
         task = self.get_object()
+
+        serializer = RetryTaskSerializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        clean = serializer.validated_data.get("clean")
+
         try:
-            new_task_id = retry_failed_file(task.task_id, True)
+            new_task_id = retry_failed_file(task.task_id, clean)
             return Response({"task_id": new_task_id})
+        except FileNotFoundError:
+            return HttpResponseBadRequest("Original file not found")
         except Exception as e:
             logger.warning(f"An error occurred retrying task: {e!s}")
             return HttpResponseBadRequest(