from pydantic import BaseModel
from typing import List, Dict

class ValidateWorker(BaseModel):
    result:str
    
class Result(BaseModel):
    response:str

class Cases(BaseModel):
    open_items: int
    red_flags: int

class AssessmentsFrequency(BaseModel):
    weekly: Cases
    biweekly: Cases
    quarterly: Cases

class AssessmentPrediction(BaseModel):
    AssessmentN: AssessmentsFrequency

class AssessmentPredictionsResponse(BaseModel):
    predictions: List[AssessmentPrediction]