Upload folder using huggingface_hub

client.py

@@ -1,98 +1,29 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-#
-# This source code is licensed under the BSD-style license found in the
-# LICENSE file in the root directory of this source tree.
-
-"""Maze Environment Client."""
-
 from typing import Dict
-
 from openenv.core.client_types import StepResult
 from openenv.core.env_server.types import State
 from openenv.core import EnvClient
+from models import MazeAction, MazeObservation, MazeState
 
-from .models import MazeAction, MazeObservation
-
-
-class MazeEnv(
-    EnvClient[MazeAction, MazeObservation]
-):
-    """
-    Client for the Maze Environment.
-
-    This client maintains a persistent WebSocket connection to the environment server,
-    enabling efficient multi-step interactions with lower latency.
-    Each client instance has its own dedicated environment session on the server.
-
-    Example:
-        >>> # Connect to a running server
-        >>> with MazeEnv(base_url="http://localhost:8000") as client:
-        ...     result = client.reset()
-        ...     print(result.observation.echoed_message)
-        ...
-        ...     result = client.step(MazeAction(message="Hello!"))
-        ...     print(result.observation.echoed_message)
-
-    Example with Docker:
-        >>> # Automatically start container and connect
-        >>> client = MazeEnv.from_docker_image("maze-env:latest")
-        >>> try:
-        ...     result = client.reset()
-        ...     result = client.step(MazeAction(message="Test"))
-        ... finally:
-        ...     client.close()
-    """
+class MazeEnv(EnvClient[MazeAction, MazeObservation, MazeState]):
+    """Client for the Maze Environment."""
 
     def _step_payload(self, action: MazeAction) -> Dict:
-        """
-        Convert MazeAction to JSON payload for step message.
-
-        Args:
-            action: MazeAction instance
-
-        Returns:
-            Dictionary representation suitable for JSON encoding
-        """
-        return {
-            "message": action.message,
-        }
+        return {"direction": action.direction}
 
     def _parse_result(self, payload: Dict) -> StepResult[MazeObservation]:
-        """
-        Parse server response into StepResult[MazeObservation].
-
-        Args:
-            payload: JSON response data from server
-
-        Returns:
-            StepResult with MazeObservation
-        """
         obs_data = payload.get("observation", {})
-        observation = MazeObservation(
-            echoed_message=obs_data.get("echoed_message", ""),
-            message_length=obs_data.get("message_length", 0),
-            done=payload.get("done", False),
-            reward=payload.get("reward"),
-            metadata=obs_data.get("metadata", {}),
-        )
-
         return StepResult(
-            observation=observation,
+            observation=MazeObservation(
+                position=obs_data.get("position", []),
+                grid_view=obs_data.get("grid_view", ""),
+                done=payload.get("done", False),
+                reward=payload.get("reward"),
+            ),
             reward=payload.get("reward"),
             done=payload.get("done", False),
         )
 
     def _parse_state(self, payload: Dict) -> State:
-        """
-        Parse server response into State object.
-
-        Args:
-            payload: JSON response from state request
-
-        Returns:
-            State object with episode_id and step_count
-        """
         return State(
             episode_id=payload.get("episode_id"),
             step_count=payload.get("step_count", 0),

server/requirements.txt

@@ -1,6 +1,4 @@
-openenv[core]>=0.2.0
-fastapi>=0.115.0
-uvicorn>=0.24.0
-
-
-
+openenv[core]>=0.2.0
+fastapi>=0.115.0
+uvicorn>=0.24.0
+mazelib