feat: extra fields in reset and step request models for custom params

Author: rycerzes

src/core/env_server/http_server.py

@@ -96,11 +96,8 @@ async def reset(
         ) -> ResetResponse:
             """Reset endpoint - returns initial observation."""
             # Handle optional parameters
-            kwargs = {}
-            if request.seed is not None:
-                kwargs["seed"] = request.seed
-            if request.episode_id is not None:
-                kwargs["episode_id"] = request.episode_id
+            # Start with all fields from the request, including extra ones
+            kwargs = request.model_dump(exclude_unset=True)
 
             # Pass arguments only if environment accepts them
             sig = inspect.signature(self.env.reset)
@@ -132,9 +129,8 @@ async def step(request: StepRequest) -> StepResponse:
                 )
 
             # Handle optional parameters
-            kwargs = {}
-            if request.timeout_s is not None:
-                kwargs["timeout_s"] = request.timeout_s
+            # Start with all fields from the request, including extra ones, but exclude 'action'
+            kwargs = request.model_dump(exclude_unset=True, exclude={'action'})
 
             # Pass arguments only if environment accepts them
             sig = inspect.signature(self.env.step)

src/core/env_server/types.py

@@ -56,7 +56,7 @@ class ResetRequest(BaseModel):
     """Request model for environment reset."""
 
     model_config = ConfigDict(
-        extra="forbid",
+        extra="allow",  # Allow extra fields for custom reset parameters
         json_schema_extra={"examples": [{"seed": 42, "episode_id": "episode-001"}, {}]},
     )
 
@@ -87,7 +87,15 @@ class ResetResponse(BaseModel):
 class StepRequest(BaseModel):
     """Request model for environment step."""
 
-    model_config = ConfigDict(extra="forbid")
+    model_config = ConfigDict(
+        extra="allow",  # Allow extra fields for custom step parameters
+        json_schema_extra={
+            "examples": [
+                {"action": {"value": 1}, "timeout_s": 30.0},
+                {"action": {"value": 1}, "render": True, "verbose": False},
+            ]
+        },
+    )
 
     action: Dict[str, Any] = Field(
         ...,