Implement fix

graphql-python · Jul 14, 2024 · 3a1594b · 3a1594b
1 parent 3c91b2f
commit 3a1594b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gql/transport/websockets_base.py b/gql/transport/websockets_base.py
@@ -512,7 +512,11 @@ async def connect(self) -> None:
                 await self._initialize()
             except ConnectionClosed as e:
                 raise e
-            except (TransportProtocolError, asyncio.TimeoutError) as e:
+            except (
+                TransportProtocolError,
+                TransportServerError,
+                asyncio.TimeoutError,
+            ) as e:
                 await self._fail(e, clean_close=False)
                 raise e