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


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

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