from pydantic import BaseModel
from typing import Optional, Any, Dict, List

class Action(BaseModel):
    action_str: str
    
# Alias for compatibility with the sample inference script
BrowserGymAction = Action

class Observation(BaseModel):
    goal: str
    result: str
    step: int
    last_action_error: bool
    schema_dump: Optional[str] = None
    
class Reward(BaseModel):
    value: float
    reason: Optional[str] = None

class StepResult(BaseModel):
    observation: Observation
    reward: float
    done: bool
    info: Dict[str, Any]

class ResetResult(BaseModel):
    observation: Observation
    info: Dict[str, Any]