feat: import ws to client

ekkx · Aug 26, 2024 · ec59e9e · ec59e9e
1 parent 79c5a44
commit ec59e9e
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/yaylib/client.py b/yaylib/client.py
@@ -32,7 +32,7 @@
 
 import aiohttp
 
-from . import __version__, config, utils
+from . import __version__, config, utils, ws
 from .api.auth import AuthApi
 from .api.call import CallApi
 from .api.chat import ChatApi
@@ -228,15 +228,13 @@ def generate(self, jwt_required=False) -> Dict[str, str]:
 
 
 # pylint: disable=too-many-public-methods
-class Client(
-    # ws.WebSocketInteractor
-):
+class Client(ws.WebSocketInteractor):
     """yaylib のエントリーポイント"""
 
     def __init__(
         self,
         *,
-        # intents: Optional[ws.Intents] = None,
+        intents: Optional[ws.Intents] = None,
         proxy_url: Optional[str] = None,
         timeout=30,
         max_retries=3,
@@ -249,6 +247,8 @@ def __init__(
         state: Optional[State] = None,
         loglevel=logging.INFO,
     ) -> None:
+        super().__init__(self, intents)
+
         self.__proxy_url = proxy_url
         self.__timeout = timeout
         self.__session = None