.

accelerator.py

@@ -1,6 +1,5 @@
 from typing import Union
 from fastapi.websockets import WebSocket, WebSocketState
-from websockets import ConnectionClosedError
 
 class Accelerator:
   ws: Union[WebSocket, None] = None
@@ -19,8 +18,5 @@ class Accelerator:
     self.ws = ws
 
   async def accelerate(self, input):    
-    try:
-      await self.ws.send_text(input)
-      return await self.ws.receive_text()
-    except ConnectionClosedError:
-      self.ws = None
+    await self.ws.send_text(input)
+    return await self.ws.receive_text()