Upload folder using huggingface_hub

client.py

@@ -12,11 +12,11 @@ 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
+from models import MazeAction, MazeObservation, MazeState
 
 
 class MazeEnv(
-    EnvClient[MazeAction, MazeObservation]
+    EnvClient[MazeAction, MazeObservation, MazeState]
 ):
     """
     Client for the Maze Env Environment.
@@ -54,9 +54,7 @@ class MazeEnv(
         Returns:
             Dictionary representation suitable for JSON encoding
         """
-        return {
-            "message": action.message,
-        }
+        return {"direction": action.direction}
 
     def _parse_result(self, payload: Dict) -> StepResult[MazeObservation]:
         """
@@ -69,16 +67,13 @@ class MazeEnv(
             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),
         )