File size: 538 Bytes
4423d13
 
 
 
 
 
 
06b0d69
 
 
 
 
8eecaf4
06b0d69
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
try:
    from openenv.core.client import EnvClient
except ImportError:
    try:
        from openenv.core.env_client import EnvClient
    except ImportError:
        from openenv.core import EnvClient
from models import APIAction, APIObservation, APIState


class APIDebugEnv(EnvClient[APIAction, APIObservation, APIState]):

    def __init__(self, base_url: str = "http://localhost:7860"):
        super().__init__(
            base_url=base_url,
            action_type=APIAction,
            observation_type=APIObservation,
        )