PR suggestions

Author: thewhaleking

bittensor_cli/cli.py

@@ -6354,11 +6354,13 @@ def best_connection(
         additional_networks = additional_networks or []
         if any(not x.startswith("ws") for x in additional_networks):
             err_console.print(
-                "Invalid network endpoint. Ensure you are specifying a valid websocket endpoint.",
+                "Invalid network endpoint. Ensure you are specifying a valid websocket endpoint"
+                f" (starting with [{COLORS.G.LINKS}]ws://[/{COLORS.G.LINKS}] or "
+                f"[{COLORS.G.LINKS}]wss://[/{COLORS.G.LINKS}]).",
             )
             return False
         results: dict[str, list[float]] = self._run_command(
-            best_connection(Constants.finney_nodes + (additional_networks or []))
+            best_connection(Constants.lite_nodes + additional_networks)
         )
         sorted_results = {
             k: v for k, v in sorted(results.items(), key=lambda item: item[1][0])

bittensor_cli/src/__init__.py

@@ -23,7 +23,7 @@ class Constants:
     dev_entrypoint = "wss://dev.chain.opentensor.ai:443"
     local_entrypoint = "ws://127.0.0.1:9944"
     latent_lite_entrypoint = "wss://lite.sub.latent.to:443"
-    finney_nodes = [finney_entrypoint, subvortex_entrypoint, latent_lite_entrypoint]
+    lite_nodes = [finney_entrypoint, subvortex_entrypoint, latent_lite_entrypoint]
     network_map = {
         "finney": finney_entrypoint,
         "test": finney_test_entrypoint,

bittensor_cli/src/bittensor/subtensor_interface.py

@@ -1669,15 +1669,18 @@ async def best_connection(networks: list[str]):
     """
     results = {}
     for network in networks:
-        t1 = time.monotonic()
-        async with websockets.connect(network) as websocket:
-            pong = await websocket.ping()
-            latency = await pong
-            pt1 = time.monotonic()
-            await websocket.send(
-                "{'jsonrpc': '2.0', 'method': 'chain_getHead', 'params': [], 'id': '82'}"
-            )
-            await websocket.recv()
-            t2 = time.monotonic()
-        results[network] = [t2 - t1, latency, t2 - pt1]
+        try:
+            t1 = time.monotonic()
+            async with websockets.connect(network) as websocket:
+                pong = await websocket.ping()
+                latency = await pong
+                pt1 = time.monotonic()
+                await websocket.send(
+                    "{'jsonrpc': '2.0', 'method': 'chain_getHead', 'params': [], 'id': '82'}"
+                )
+                await websocket.recv()
+                t2 = time.monotonic()
+            results[network] = [t2 - t1, latency, t2 - pt1]
+        except Exception as e:
+            err_console.print(f"Error attempting network {network}: {e}")
     return results