fix tls errors for async client stream

requirements.txt

@@ -10,4 +10,4 @@ typer
 uvicorn
 wheel
 apispec
-httpx
+httpx[http2]

runhouse/servers/http/http_client.py

@@ -98,7 +98,7 @@ def __init__(
             self.verify = self.cert_path if self._certs_are_self_signed() else True
 
         self.async_session = httpx.AsyncClient(
-            auth=self.auth, verify=self.verify, timeout=None
+            auth=self.auth, verify=self.verify, timeout=None, http2=True
         )
 
         self.log_formatter = ClusterLogsFormatter(self.system)
@@ -573,7 +573,9 @@ async def _alogs_request(
         # When running this in another thread, we need to explicitly create an async client here. When running within
         # the main thread, we can use the client that was passed in.
         if create_async_client:
-            client = httpx.AsyncClient(auth=self.auth, verify=self.verify, timeout=None)
+            client = httpx.AsyncClient(
+                auth=self.auth, verify=self.verify, timeout=None, http2=True
+            )
         else:
             client = self.async_session
 

setup.py

@@ -75,7 +75,7 @@ def parse_readme(readme: str) -> str:
     "uvicorn",
     "wheel",
     "apispec",
-    "httpx",
+    "httpx[http2]",
 ]
 
 # NOTE: Change the templates/spot-controller.yaml.j2 file if any of the following

tests/fixtures/docker_cluster_fixtures.py

@@ -372,7 +372,7 @@ def docker_cluster_pk_ssh(request, test_org_rns_folder):
         reqs=[
             "ray==2.30.0",
             "pytest",
-            "httpx",
+            "httpx[http2]",
             "pytest_asyncio",
             "pandas",
             "numpy<=1.26.4",
@@ -477,7 +477,7 @@ def docker_cluster_pk_http_exposed(request, test_rns_folder):
         reqs=[
             "ray==2.30.0",
             "pytest",
-            "httpx",
+            "httpx[http2]",
             "pytest_asyncio",
             "pandas",
             "numpy<=1.26.4",

tests/requirements.txt

@@ -1,6 +1,6 @@
 pytest
 pytest-mock
-httpx
+httpx[http2]
 pytest_asyncio
 datasets
 dask

tests/utils.py

@@ -101,7 +101,7 @@ def friend_account_in_org():
 
 def test_env(logged_in=False):
     return rh.env(
-        reqs=["pytest", "httpx", "pytest_asyncio", "pandas", "numpy<=1.26.4"],
+        reqs=["pytest", "httpx[http2]", "pytest_asyncio", "pandas", "numpy<=1.26.4"],
         working_dir=None,
         setup_cmds=[
             f"mkdir -p ~/.rh; touch ~/.rh/config.yaml; "