"""OpenEnv client wrapper for PM-Ops environment."""
import os
from openenv.core.env_client import EnvClient
from models import PMOpsAction, PMOpsObservation


class PMOpsEnv(EnvClient):
    action_class = PMOpsAction
    observation_class = PMOpsObservation

    def __init__(self, base_url: str | None = None, token: str | None= None):
        url = base_url or os.getenv("API_BASE_URL", "https://adityaguntur-pm-ops.hf.space")
        super().__init__(base_url=url)