SHYAMSATHISH005/data-cleaning-openenv

1from pydantic import BaseModel2from typing import Any, Dict, List, Optional3 4 5class ResetRequest(BaseModel):6    task_id: str = "fix_types"7 8class Action(BaseModel):9    name: str10    params: Optional[Dict[str, Any]] = {}11 12class Observation(BaseModel):13    dataframe_preview: List[Dict[str, Any]]14    markdown_preview: str15    null_counts: Dict[str, int]16    validation_errors: List[str]17    accuracy: float18    step_count: int19    steps_remaining: int20 21class RewardOut(BaseModel):22    reward: float23    raw: Dict[str, Any] = {}24    done: bool = False25    info: Dict[str, Any] = {}26