Yaswanth-Bolla commited on
Commit
9ef85f9
·
1 Parent(s): 1175c0b

fix issue

Browse files
Files changed (1) hide show
  1. server/app.py +13 -3
server/app.py CHANGED
@@ -58,15 +58,25 @@ def health() -> Dict[str, str]:
58
  return {"status": "healthy"}
59
 
60
 
 
 
61
  @app.post("/reset")
62
- def reset(request: ResetRequest) -> Dict[str, Any]:
63
  """
64
  Initialize a new incident episode.
65
  POST /reset {"task_name": "memory_leak", "seed": 42}
66
  """
 
 
 
 
 
 
 
 
67
  result = env.reset(
68
- task_name=request.task_name,
69
- seed=request.seed,
70
  )
71
  return result
72
 
 
58
  return {"status": "healthy"}
59
 
60
 
61
+ from fastapi import Request
62
+
63
  @app.post("/reset")
64
+ async def reset(request: Request) -> Dict[str, Any]:
65
  """
66
  Initialize a new incident episode.
67
  POST /reset {"task_name": "memory_leak", "seed": 42}
68
  """
69
+ try:
70
+ body = await request.json()
71
+ except Exception:
72
+ body = {}
73
+
74
+ if not isinstance(body, dict):
75
+ body = {}
76
+
77
  result = env.reset(
78
+ task_name=body.get("task_name"),
79
+ seed=body.get("seed"),
80
  )
81
  return result
82