app/models/moderation_data.py

from pydantic import BaseModel
from typing import List
from typing import Any, Dict
from app.models.check_type import CheckType


class ModerationData(BaseModel):
    flagged_for: Dict[CheckType, float]
    scores: Dict[CheckType, float]
    original_content: str
    cleaned_content: str = None
    corrected_content: Any