fix: do not ignore all RuntimeErrors, only for websocket closed.

A runtime error also occurs when a send is triggered from the same
event loop thread.

Author: maartenbreddels

solara/server/starlette.py

@@ -133,16 +133,21 @@ async def _send_bytes_exc(self, data: bytes):
             await self.ws.send_bytes(data)
         except (websockets.exceptions.ConnectionClosed, starlette.websockets.WebSocketDisconnect, RuntimeError) as e:
             # starlette throws a RuntimeError once you call send after the connection is closed
-            raise websocket.WebSocketDisconnect() from e
+            if isinstance(e, RuntimeError) and "close message" in repr(e):
+                raise websocket.WebSocketDisconnect() from e
+            else:
+                raise
 
     async def _send_text_exc(self, data: str):
         # make sures we catch the starlette/websockets specific exception
         # and re-raise it as a websocket.WebSocketDisconnect
         try:
             await self.ws.send_text(data)
         except (websockets.exceptions.ConnectionClosed, starlette.websockets.WebSocketDisconnect, RuntimeError) as e:
-            # starlette throws a RuntimeError once you call send after the connection is closed
-            raise websocket.WebSocketDisconnect() from e
+            if isinstance(e, RuntimeError) and "close message" in repr(e):
+                raise websocket.WebSocketDisconnect() from e
+            else:
+                raise
 
     def close(self):
         if self.portal is None: