fix(deps): upgrade to websockets 14.x

ApeWorX · Nov 24, 2024 · d4be238 · d4be238
1 parent b95f147
commit d4be238
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -71,6 +71,7 @@
         "taskiq[metrics]>=0.11.9,<0.12",
         "tomlkit>=0.12,<1",  # For reading/writing global platform profile
         "fief-client[cli]>=0.19,<1",  # for platform auth/cluster login
+        "websockets>=14.1,<15",  # For subscriptions
     ],
     entry_points={
         "console_scripts": ["silverback=silverback._cli:cli"],

diff --git a/silverback/subscriptions.py b/silverback/subscriptions.py
@@ -5,7 +5,7 @@
 
 from ape.logging import logger
 from websockets import ConnectionClosedError
-from websockets import client as ws_client
+from websockets.asyncio import client as ws_client
 
 
 class SubscriptionType(Enum):
@@ -26,7 +26,7 @@ def __init__(self, ws_provider_uri: str):
         self._ws_provider_uri = ws_provider_uri
 
         # Stateful
-        self._connection: ws_client.WebSocketClientProtocol | None = None
+        self._connection: ws_client.ClientConnection | None = None
         self._last_request: int = 0
         self._subscriptions: dict[str, asyncio.Queue] = {}
         self._rpc_msg_buffer: list[dict] = []