subhdotsol commited on
Commit
0e662d7
·
1 Parent(s): 574fde3

feat(app): add POST /reset endpoint — starts new episode, returns ResetResponse

Browse files
Files changed (1) hide show
  1. server/app.py +10 -1
server/app.py CHANGED
@@ -1,6 +1,7 @@
1
  from contextlib import asynccontextmanager
2
- from fastapi import FastAPI
3
  from fastapi.middleware.cors import CORSMiddleware
 
4
  from server.environment import RedTeamEnvironment
5
  from server.config import get_settings
6
 
@@ -37,3 +38,11 @@ app.add_middleware(
37
  @app.get("/health")
38
  async def health_check():
39
  return {"status": "healthy", "version": "0.1.0"}
 
 
 
 
 
 
 
 
 
1
  from contextlib import asynccontextmanager
2
+ from fastapi import FastAPI, HTTPException
3
  from fastapi.middleware.cors import CORSMiddleware
4
+ from models import ResetResponse
5
  from server.environment import RedTeamEnvironment
6
  from server.config import get_settings
7
 
 
38
  @app.get("/health")
39
  async def health_check():
40
  return {"status": "healthy", "version": "0.1.0"}
41
+
42
+ @app.post("/reset", response_model=ResetResponse)
43
+ async def reset_episode():
44
+ try:
45
+ observation = await env.reset()
46
+ return ResetResponse(observation=observation, episode_id=observation.episode_id)
47
+ except Exception as e:
48
+ raise HTTPException(status_code=500, detail=str(e))