KChad commited on
Commit
00af297
·
1 Parent(s): ea78b45

Add root route for HF Space

Browse files
Files changed (1) hide show
  1. server/app.py +10 -0
server/app.py CHANGED
@@ -41,6 +41,16 @@ def health():
41
  return {"status": "ok", "env": "idpi-exfil-env", "version": "0.1.0"}
42
 
43
 
 
 
 
 
 
 
 
 
 
 
44
  @app.post("/reset", response_model=Observation)
45
  def reset(request: ResetRequest):
46
  # Validates difficulty before passing to env
 
41
  return {"status": "ok", "env": "idpi-exfil-env", "version": "0.1.0"}
42
 
43
 
44
+ @app.get("/")
45
+ def root():
46
+ return {
47
+ "name": "IDPI Exfil Environment",
48
+ "status": "ok",
49
+ "description": "OpenEnv-compatible prompt injection defense environment.",
50
+ "routes": ["/health", "/reset", "/step", "/state", "/result"],
51
+ }
52
+
53
+
54
  @app.post("/reset", response_model=Observation)
55
  def reset(request: ResetRequest):
56
  # Validates difficulty before passing to env