client.py

"""OpenEnv client for the Astrum environment.



Connects to a running Astrum server (local or HF Space) and provides

sync/async access to reset() and step().



Usage:

    from astrum_env import AstrumEnv, AstrumAction



    with AstrumEnv(base_url="https://abnaidu-hypernoa-astrum.hf.space").sync() as env:

        result = env.reset()

        result = env.step(AstrumAction(

            action_type="allocate_resources",

            params={"stakeholder": "workers", "amount": 10, "resource": "budget"}

        ))

        print(result.reward)

"""

from openenv.core.env_client import EnvClient

from .models import AstrumAction, AstrumObservation


class AstrumEnv(EnvClient[AstrumAction, AstrumObservation]):
    """Client for the Hypernoa Astrum environment."""

    @property
    def action_type(self):
        return AstrumAction

    @property
    def observation_type(self):
        return AstrumObservation