erp_ai/src/models/bot_response_schema.py

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]


class AssessmentSuggestion(BaseModel):
    assessment_type:str
    frequency:str
    Justification:str




class GoalAchivementPrediction(BaseModel):
    prediction:str
    reasoning:str


class AchievementPrediction(BaseModel):
    goal_achievement:GoalAchivementPrediction
    goal_achievement_new_assessment:str