Support ssl context

javascript/package.json

@@ -1,6 +1,6 @@
 {
  "name": "hypha-rpc",
- "version": "0.20.27",
+ "version": "0.20.28",
  "description": "Hypha RPC client for connecting to Hypha server for data management and AI model serving.",
  "main": "index.js",
  "types": "index.d.ts",

python/hypha_rpc/VERSION

@@ -1,3 +1,3 @@
 {
- "version": "0.20.27"
+ "version": "0.20.28"
 }

python/hypha_rpc/pyodide_websocket.py

@@ -129,6 +129,7 @@ def __init__(
  token=None,
  reconnection_token=None,
  timeout=5,
+ ssl=None,
  ):
  assert server_url and client_id, "server_url and client_id are required"
  self._server_url = server_url
@@ -147,6 +148,7 @@ def __init__(
  self.connection_info = None
  self._enable_reconnect = False
  self.manager_id = None
+ assert ssl is None, "SSL is not supported in Pyodide"
  if self._server_url.startswith("wss://local-hypha-server:"):
  self._WebSocketClass = js.eval(
  "("

python/hypha_rpc/websocket_client.py

@@ -52,6 +52,7 @@ def __init__(
  token=None,
  reconnection_token=None,
  timeout=60,
+ ssl=None,
  ):
  """Set up instance."""
  self._websocket = None
@@ -69,6 +70,7 @@ def __init__(
  self._legacy_auth = None
  self.connection_info = None
  self._enable_reconnect = False
+ self._ssl = ssl
  self.manager_id = None
 
  def on_message(self, handler):
@@ -92,7 +94,7 @@ async def _attempt_connection(self, server_url, attempt_fallback=True):
  try:
  self._legacy_auth = False
  websocket = await asyncio.wait_for(
- websockets.connect(server_url), self._timeout
+ websockets.connect(server_url, ssl=self._ssl), self._timeout
  )
  return websocket
  except websockets.exceptions.InvalidStatusCode as e:
@@ -128,7 +130,7 @@ async def _attempt_connection_with_query_params(self, server_url):
  full_url = f"{server_url}?{query_string}" if query_string else server_url
 
  # Attempt to establish the WebSocket connection with the constructed URL
- return await websockets.connect(full_url)
+ return await websockets.connect(full_url, ssl=self._ssl)
 
  async def open(self):
  """Open the connection with fallback logic for backward compatibility."""
@@ -431,6 +433,7 @@ async def _connect_to_server(config):
  token=config.get("token"),
  reconnection_token=config.get("reconnection_token"),
  timeout=config.get("method_timeout", 30),
+ ssl=config.get("ssl"),
  )
  connection_info = await connection.open()
  assert connection_info, (

python/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "hypha_rpc"
-version = "0.20.27"
+version = "0.20.28"
 description = "Hypha RPC client for connecting to Hypha server for data management and AI model serving"
 readme = "README.md"
 requires-python = ">=3.6"