immortalindeed commited on
Commit
09576c0
·
1 Parent(s): bdb64b6

Ensure /step returns info object perfectly matching OpenEnv spec

Browse files
Files changed (1) hide show
  1. server/app.py +2 -0
server/app.py CHANGED
@@ -226,6 +226,7 @@ async def step(request: Request):
226
  'reward': round(float(result.get('reward', 0.0)), 4),
227
  'done': bool(result.get('done', False)),
228
  'observation': step_obs,
 
229
  }
230
  except Exception as e:
231
  return JSONResponse(status_code=200, content={
@@ -233,6 +234,7 @@ async def step(request: Request):
233
  'done': True,
234
  'error': str(e),
235
  'observation': {},
 
236
  })
237
 
238
 
 
226
  'reward': round(float(result.get('reward', 0.0)), 4),
227
  'done': bool(result.get('done', False)),
228
  'observation': step_obs,
229
+ 'info': {'validation_failed': step_obs.get('validation_failed', False)},
230
  }
231
  except Exception as e:
232
  return JSONResponse(status_code=200, content={
 
234
  'done': True,
235
  'error': str(e),
236
  'observation': {},
237
+ 'info': {'error': str(e)},
238
  })
239
 
240