from pydantic import BaseModel
from typing import Optional

class SimulationCase(BaseModel):
    persona_id: str
    module_id: str
    prompt: str
    emotion_before: str
    desired_ai: Optional[str] = ""
    ai_behavior: Optional[str] = ""
    tone: Optional[str] = ""