Eric Xu commited on
Add SSE keepalive pings every 15s to prevent proxy timeout disconnects
Browse files- 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}")
|