Nihar-776/Hackathon-MetaPytorch

1from pydantic import BaseModel2from typing import Optional, List, Any, Dict3 4 5class ResetRequest(BaseModel):6    task_id: str = "task1_single_typo"7 8 9class StepAction(BaseModel):10    corrected_record: Dict[str, Any]11    confidence: float = 1.012    task_id: str13    session_id: str14 15 16class Observation(BaseModel):17    task_id: str18    session_id: str19    step: int20    corrupted_record: Dict[str, Any]21    corruption_types_hint: List[str]22    num_corrupted_fields: int23    fields_still_wrong: int24    score_so_far: float25    max_steps: int26    done: bool27    message: str28 29 30class StepResult(BaseModel):31    observation: Observation32    reward: float33    done: bool34    info: Dict[str, Any]