NILESH1003/codeguardian-env

1from pydantic import BaseModel, Field2from typing import List, Literal, Optional3 4class IssueFound(BaseModel):5    line: int6    bug_type: str7    comment: str8 9class CodeObservation(BaseModel):10    code: str11    filename: str12    language: str = "python"13    task_id: str14    task_difficulty: Literal["easy", "medium", "hard"]15    issues_found: List[IssueFound] = Field(default_factory=list)16    step_count: int = 017    done: bool = False18 19class CodeAction(BaseModel):20    action: Literal["flag_bug", "suggest_fix", "approve", "reject"]21    line: Optional[int] = None22    comment: str23    bug_type: Optional[str] = None24 25class RewardDetail(BaseModel):26    step_reward: float27    reason: str28    partial: bool29 30class EnvState(BaseModel):31    episode_id: str32    task_id: str33    step_count: int34    total_reward: float35    done: bool36    task_difficulty: str37