admin/Akern-Genai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

handle websocket disconnect

Browse Source
This commit is contained in:
Matteo Rosati
2026-01-22 10:22:36 +01:00
parent 48d8ce9276
commit 867b029e1c
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
app.py
View File

@@ -12,6 +12,7 @@ from fastapi import (
FastAPI, FastAPI,
Request, Request,
WebSocket, WebSocket,
WebSocketDisconnect,
Depends, Depends,
HTTPException, HTTPException,
status, status,
@@ -95,10 +96,14 @@ async def websocket_endpoint(websocket: WebSocket):
websocket: The WebSocket connection. websocket: The WebSocket connection.
""" """
await websocket.accept() await websocket.accept()
while True: try:
data = await websocket.receive_text() while True:
data = await websocket.receive_text()
async for chunk in generate(data): async for chunk in generate(data):
await websocket.send_text(chunk) await websocket.send_text(chunk)
await websocket.send_text("<<END>>") await websocket.send_text("<<END>>")
except WebSocketDisconnect:
logger.info("Client disconnected")
await websocket.close()