Eric Xu commited on
Commit
cc9ff58
·
unverified ·
1 Parent(s): 41b60b4

Add SSE keepalive pings every 15s to prevent proxy timeout disconnects

Browse files
Files changed (1) hide show
  1. web/app.py +3 -3
web/app.py CHANGED
@@ -646,7 +646,7 @@ async def evaluate_stream(sid: str, request: Request, parallel: int = 5,
646
  }
647
  yield {"event": "complete", "data": json.dumps(summary)}
648
 
649
- return EventSourceResponse(event_generator())
650
 
651
 
652
  class CounterfactualRequest(BaseModel):
@@ -789,7 +789,7 @@ async def counterfactual_stream(sid: str, ticket: str, request: Request):
789
  "goal": goal if has_goal else None,
790
  })}
791
 
792
- return EventSourceResponse(event_generator())
793
 
794
 
795
  @app.get("/api/bias-audit/stream/{sid}")
@@ -876,7 +876,7 @@ async def bias_audit_stream(
876
  "report": report,
877
  })}
878
 
879
- return EventSourceResponse(event_generator())
880
 
881
 
882
  @app.get("/api/results/{sid}")
 
646
  }
647
  yield {"event": "complete", "data": json.dumps(summary)}
648
 
649
+ return EventSourceResponse(event_generator(), ping=15)
650
 
651
 
652
  class CounterfactualRequest(BaseModel):
 
789
  "goal": goal if has_goal else None,
790
  })}
791
 
792
+ return EventSourceResponse(event_generator(), ping=15)
793
 
794
 
795
  @app.get("/api/bias-audit/stream/{sid}")
 
876
  "report": report,
877
  })}
878
 
879
+ return EventSourceResponse(event_generator(), ping=15)
880
 
881
 
882
  @app.get("/api/results/{sid}")