fix!: connection issue with GW from new arch

Author: i0bs

interactions/api/__init__.py

@@ -9,6 +9,7 @@
 from .cache import *  # noqa: F401 F403
 from .enums import *  # noqa: F401 F403
 from .error import *  # noqa: F401 F403
-from .gateway import *  # noqa: F401 F403
+from .gateway.client import *  # noqa: F401 F403
+from .gateway.heartbeat import *  # noqa: F401 F403
 from .http import *  # noqa: F401 F403
 from .models import *  # noqa: F401 F403

interactions/api/gateway/client.py

@@ -33,8 +33,6 @@
 
 log = get_logger("gateway")
 
-__all__ = "WebSocketClient"
-
 
 class WebSocketClient:
     """

interactions/client/bot.py

@@ -9,10 +9,10 @@
 from types import ModuleType
 from typing import Any, Callable, Coroutine, Dict, List, Optional, Union
 
-from ..api.cache import Cache
-from ..api.cache import Item as Build
+from ..api import Cache
+from ..api import Item as Build
+from ..api import WebSocketClient as WSClient
 from ..api.error import InteractionException, JSONException
-from ..api.gateway.client import WebSocketClient
 from ..api.http.client import HTTPClient
 from ..api.models.flags import Intents
 from ..api.models.guild import Guild
@@ -78,7 +78,7 @@ def __init__(
         self._loop = get_event_loop()
         self._http = HTTPClient(token=token)
         self._intents = kwargs.get("intents", Intents.DEFAULT)
-        self._websocket = WebSocketClient(token=token, intents=self._intents)
+        self._websocket = WSClient(token=token, intents=self._intents)
         self._shard = kwargs.get("shards", [])
         self._presence = kwargs.get("presence")
         self._token = token