commit

environment/models.py

@@ -1,49 +1,29 @@
-from pydantic import BaseModel
-from typing import Optional, Dict, Any, List
-
-
-class Issue(BaseModel):
-    line_number: Optional[int] = None
-    issue_type: str  # "syntax_error", "logic_bug", "security_vulnerability"
-    description: str
-    severity: str = "medium"  # "low", "medium", "high", "critical"
-
-
-class CodeReviewAction(BaseModel):
-    identified_issues: List[Issue] = []
-    suggested_fix: Optional[str] = None  # corrected code as a string
-    explanation: str = ""
-    done: bool = False
-
-
-class CodeReviewObservation(BaseModel):
-    task_id: str
-    task_name: str
-    difficulty: str
-    language: str
-    code_snippet: str
-    context: str
-    step_number: int
-    max_steps: int
-    previous_feedback: Optional[str] = None
-
-
-class StepResult(BaseModel):
-    observation: CodeReviewObservation
-    reward: float
-    done: bool
-    info: Dict[str, Any] = {}
-
-
-class ResetResult(BaseModel):
-    observation: CodeReviewObservation
-    info: Dict[str, Any] = {}
-
-
-class StateResult(BaseModel):
-    task_id: str
-    step_number: int
-    total_reward: float
-    actions_history: List[Dict] = []
-    done: bool
-    initialized: bool
+from typing import List, Optional
+from pydantic import Field
+from openenv.core.env_server.types import Action, Observation
+
+
+class Issue(Action):
+    line_number: Optional[int] = Field(default=None)
+    issue_type: str = Field(default="syntax_error")
+    description: str = Field(default="")
+    severity: str = Field(default="medium")
+
+
+class CodeReviewAction(Action):
+    identified_issues: List[Issue] = Field(default_factory=list)
+    suggested_fix: Optional[str] = Field(default=None)
+    explanation: str = Field(default="")
+    submit: bool = Field(default=False)
+
+
+class CodeReviewObservation(Observation):
+    task_id: str = Field(default="")
+    task_name: str = Field(default="")
+    difficulty: str = Field(default="easy")
+    language: str = Field(default="python")
+    code_snippet: str = Field(default="")
+    context: str = Field(default="")
+    step_number: int = Field(default=0)
+    max_steps: int = Field(default=5)
+    previous_feedback: Optional[str] = Field(default=None)