Spaces:
Paused
Paused
File size: 515 Bytes
96dc6e1 233feb1 ac443a4 96dc6e1 233feb1 96dc6e1 8fc4ed2 233feb1 96dc6e1 8e4c2dd 96dc6e1 ac443a4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from typing import Optional
from fastapi.websockets import WebSocket
from websockets import ConnectionClosedError
class Accelerator:
def __del__(self): self.ws.close()
ws: Optional[WebSocket] = None
def connected(self): return self.ws != None
async def connect(self, ws: WebSocket):
await ws.accept()
self.ws = ws
async def accelerate(self, input):
try:
await self.ws.send_text(input)
return await self.ws.receive_text()
except:
self.ws = None
return None
|