Update models.py

models.py

@@ -6,7 +6,6 @@
 
 """
 Data models for the FixOS Environment.
-
 FixOS is a deterministic OS troubleshooting simulation where AI agents
 diagnose and fix system issues through commands.
 """
@@ -81,17 +80,17 @@ class FixOSAction(Action):
 
 
 class FixOSObservation(Observation):
-    command_output: str = Field(default="")
-    processes: List[ProcessInfo] = Field(default_factory=list)
-    services: List[ServiceInfo] = Field(default_factory=list)
-    filesystem: List[FileInfo] = Field(default_factory=list)
-    resources: Dict[str, float] = Field(default_factory=dict)
-    logs: List[LogEntry] = Field(default_factory=list)
-    history: List[str] = Field(default_factory=list)
-    task_id: str = Field(default="")
-    task_difficulty: str = Field(default="")
-    task_score: float = Field(default=0.0)
-    is_success_step: bool = Field(default=False)
-    remaining_steps: int = Field(default=0)
-    done: bool = Field(default=False)
-    reward: float = Field(default=0.0)
+    """Observation from FixOS environment."""
+    
+    command_output: str = Field(default="", description="Output from the last command")
+    processes: List[ProcessInfo] = Field(default_factory=list, description="Current processes")
+    services: List[ServiceInfo] = Field(default_factory=list, description="Current services")
+    filesystem: List[FileInfo] = Field(default_factory=list, description="Tracked files")
+    resources: Dict[str, float] = Field(default_factory=dict, description="Resource aggregates (cpu%, memory%, disk%)")
+    logs: List[LogEntry] = Field(default_factory=list, description="Recent log entries")
+    history: List[str] = Field(default_factory=list, description="Command history")
+    task_id: str = Field(default="", description="Current task ID")
+    task_difficulty: str = Field(default="", description="Task difficulty: easy, medium, hard")
+    task_score: float = Field(default=0.0, description="Current task progress score in [0.0, 1.0]")
+    is_success_step: bool = Field(default=False, description="Whether task is solved")
+    remaining_steps: int = Field(default=0, description="Steps remaining in episode")