Merge pull request #99 from Murugurugan/main

Updating to websockets 14.2

Author: gorantosicpf

requirements.txt

@@ -1,3 +1,3 @@
 aiofiles==23.2.1
 python-dotenv==1.0.1
-websockets==12.0
+websockets

runware/base.py

@@ -10,6 +10,7 @@
 import inspect
 from typing import List, Union, Optional, Callable, Any, Dict
 from urllib.parse import urlparse
+from websockets.protocol import State
 
 from .utils import (
     BASE_RUNWARE_URLS,
@@ -102,7 +103,7 @@ def __init__(
         self._sdkType: SdkType = SdkType.SERVER
 
     def isWebsocketReadyState(self) -> bool:
-        return self._ws and self._ws.open
+        return self._ws and self._ws.state is State.OPEN
 
     def isAuthenticated(self):
         return self._connectionSessionUUID is not None
@@ -1134,7 +1135,7 @@ async def ensureConnection(self) -> None:
 
         :raises: An error message if the connection cannot be established due to an invalid API key or other reasons.
         """
-        isConnected = self.connected() and self._ws.open
+        isConnected = self.connected() and self._ws.state is State.OPEN
         # print(f"Is connected: {isConnected}")
 
         try:

runware/server.py

@@ -2,6 +2,7 @@
 import json
 import logging
 import websockets
+from websockets.protocol import State
 import inspect
 import pprint
 from typing import Any, Callable, Dict, List, Union, Optional, TypeVar
@@ -207,7 +208,7 @@ async def send(self, msg: Dict[str, Any]):
         # print(
         #     f"=============================================== End sending message ===============================================================\n\n\n"
         # )
-        if self._ws and self._ws.open:
+        if self._ws and self._ws.state is State.OPEN:
             await self._ws.send(json.dumps(msg))
 
     def _get_task_by_name(self, name):

setup.py

@@ -36,7 +36,7 @@
         # Add your SDK's dependencies here
         "aiofiles==23.2.1",
         "python-dotenv==1.0.1",
-        "websockets==12.0",
+        "websockets",
         # ...
     ],
 )