Catching one more exception for client retry

rusiaaman · Nov 21, 2024 · 594b0ac · 594b0ac
1 parent d0a3144
commit 594b0ac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 authors = [{ name = "Aman Rusia", email = "[email protected]" }]
 name = "wcgw"
-version = "1.1.0"
+version = "1.1.1"
 description = "What could go wrong giving full shell access to chatgpt?"
 readme = "README.md"
 requires-python = ">=3.10, <3.13"

diff --git a/src/wcgw/client/tools.py b/src/wcgw/client/tools.py
@@ -783,7 +783,7 @@ def register_client(server_url: str, client_uuid: str = "") -> None:
                 assert isinstance(output, str)
                 websocket.send(output)
 
-        except (websockets.ConnectionClosed, ConnectionError):
+        except (websockets.ConnectionClosed, ConnectionError, OSError):
             print(f"Connection closed for UUID: {client_uuid}, retrying")
             register_client(server_url, client_uuid)