feat : regulization answer format

4m9d · Nov 23, 2023 · ab0b275 · ab0b275
1 parent 59e5967
commit ab0b275
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/app/quiz/quizv2.py b/app/quiz/quizv2.py
@@ -20,6 +20,14 @@ async def generate_quizzes(summaries: list):
     quiz_list = []
     for quiz_chunk in quiz_chunk_list:
         quiz_list.extend(quiz_chunk)
+
+    for quiz in quiz_list:
+        if isinstance(quiz['answer'], list) and len(quiz['answer']) > 0:
+            quiz['answer'] = quiz['answer'][0]
+
+        if quiz['answer'] == 0:
+            quiz['answer'] = 1
+
     return quiz_list