Nanda Kumar Kondreddy commited on
Commit
a91aa82
·
1 Parent(s): f2a42f5

FIX: Properly type request parameter in /reset endpoint as fastapi.Request

Browse files
Files changed (1) hide show
  1. server/app.py +2 -1
server/app.py CHANGED
@@ -5,6 +5,7 @@ Uses OpenEnv's create_fastapi_app() for standard framework compatibility
5
  """
6
  import json
7
  import gradio as gr
 
8
  from starlette.middleware.base import BaseHTTPMiddleware
9
  from starlette.responses import JSONResponse
10
  from starlette.types import ASGIApp, Receive, Scope, Send
@@ -215,7 +216,7 @@ def metadata():
215
 
216
 
217
  @app.post("/reset")
218
- async def reset_env(request):
219
  """Override /reset endpoint to return correct OpenEnv contract schema.
220
 
221
  Must include: observation, done, reward, metadata, info
 
5
  """
6
  import json
7
  import gradio as gr
8
+ from fastapi import Request
9
  from starlette.middleware.base import BaseHTTPMiddleware
10
  from starlette.responses import JSONResponse
11
  from starlette.types import ASGIApp, Receive, Scope, Send
 
216
 
217
 
218
  @app.post("/reset")
219
+ async def reset_env(request: Request):
220
  """Override /reset endpoint to return correct OpenEnv contract schema.
221
 
222
  Must include: observation, done, reward, metadata, info