Thought I had staged it

app/dtos/speaking.py

@@ -1,9 +1,9 @@
 import random
-from typing import List, Dict
+from typing import List, Dict, Optional
 
 from pydantic import BaseModel
 
-from app.configs.constants import MinTimers, ELAIAvatars
+from app.configs.constants import MinTimers
 
 
 class SaveSpeakingDTO(BaseModel):
@@ -21,22 +21,7 @@ class GradeSpeakingAnswersDTO(BaseModel):
 
 
 class GenerateVideo1DTO(BaseModel):
-    avatar: str = (random.choice(list(ELAIAvatars))).name
+    avatar: str = Optional[str]
     questions: List[str]
     first_topic: str
     second_topic: str
-
-
-class GenerateVideo2DTO(BaseModel):
-    avatar: str = (random.choice(list(ELAIAvatars))).name
-    prompts: List[str] = []
-    suffix: str = ""
-    question: str
-    topic: str
-
-
-class GenerateVideo3DTO(BaseModel):
-    avatar: str = (random.choice(list(ELAIAvatars))).name
-    questions: List[str]
-    topic: str
-