diff --git a/litecord/gateway/state.py b/litecord/gateway/state.py index b3bc46e..a16f18e 100644 --- a/litecord/gateway/state.py +++ b/litecord/gateway/state.py @@ -12,6 +12,7 @@ class GatewayState: Used to store all information tied to the websocket's session. """ + def __init__(self, **kwargs): self.session_id = kwargs.get('session_id', gen_session_id()) self.seq = kwargs.get('seq', 0) diff --git a/litecord/gateway/websocket.py b/litecord/gateway/websocket.py index 3ee0637..292d8a5 100644 --- a/litecord/gateway/websocket.py +++ b/litecord/gateway/websocket.py @@ -36,6 +36,7 @@ def decode_etf(data): class GatewayWebsocket: """Main gateway websocket logic.""" + def __init__(self, sm, db, ws, **kwargs): self.state_manager = sm self.db = db