ENCOA-311

ielts_be/api/exam/writing.py

@@ -20,10 +20,10 @@ writing_router = APIRouter()
 async def generate_writing_academic(
         task: int = Path(..., ge=1, le=2),
         file: UploadFile = File(...),
-        difficulty: Optional[List[str]] = None,
+        difficulty: List[str] = Query(default=None),
         writing_controller: IWritingController = Depends(Provide[controller])
 ):
-    difficulty = random.choice(EducationalContent.DIFFICULTIES) if not difficulty else difficulty
+    difficulty = [random.choice(EducationalContent.DIFFICULTIES)] if not difficulty else difficulty
     return await writing_controller.get_writing_task_academic_question(task, file, difficulty)
 
 
@@ -34,10 +34,10 @@ async def generate_writing_academic(
 @inject
 async def generate_writing(
     task: int = Path(..., ge=1, le=2),
-    difficulty: Optional[str] = None,
+    difficulty: List[str] = Query(default=None),
     topic: str = Query(default=None),
     writing_controller: IWritingController = Depends(Provide[controller])
 ):
-    difficulty = random.choice(EducationalContent.DIFFICULTIES) if not difficulty else difficulty
+    difficulty = [random.choice(EducationalContent.DIFFICULTIES)] if not difficulty else difficulty
     topic = random.choice(EducationalContent.MTI_TOPICS) if not topic else topic
     return await writing_controller.get_writing_task_general_question(task, topic, difficulty)