added bot prediction for assessments

2024-09-14 01:50:41 +00:00
parent 45bc62c745
commit cd8f499f97
14 changed files with 698 additions and 22 deletions
@@ -4,4 +4,21 @@ 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]