fix(server): if cancelling the websocket task fails, websocket might …

…not close
widgetti · Mar 7, 2024 · 9e3cdae · 9e3cdae
1 parent 8dacf8f
commit 9e3cdae
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/solara/server/starlette.py b/solara/server/starlette.py
@@ -228,7 +228,10 @@ async def run():
             await thread_return
     finally:
         if settings.main.experimental_performance:
-            ws_wrapper.task.cancel()
+            try:
+                ws_wrapper.task.cancel()
+            except:  # noqa
+                logger.exception("error cancelling websocket task")
         try:
             await ws.close()
         except:  # noqa