-
Notifications
You must be signed in to change notification settings - Fork 505
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Perform reachability check once blocks are loaded to avoid delays
- Loading branch information
Showing
2 changed files
with
47 additions
and
36 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,37 @@ | ||
import time | ||
|
||
import requests | ||
from hivemind.utils.logging import get_logger | ||
|
||
logger = get_logger(__file__) | ||
|
||
|
||
def check_reachability(peer_id, wait_time: float = 600, retry_delay: float = 15) -> None: | ||
for attempt_no in range(math.floor(wait_time / retry_delay) + 1): | ||
try: | ||
r = requests.get(f"http://health.petals.ml/api/v1/is_reachable/{peer_id}", timeout=10) | ||
r.raise_for_status() | ||
response = r.json() | ||
|
||
if response["success"]: | ||
logger.info("Server is reachable from the Internet. It will appear at http://health.petals.ml soon") | ||
return | ||
|
||
if attempt_no == 0: | ||
# If health.petals.ml didn't manage to connect right away, we need to wait for libp2p to set up relays | ||
logger.info("Detected a NAT or a firewall, connecting to libp2p relays. This takes a few minutes") | ||
time.sleep(retry_delay) | ||
except Exception as e: | ||
logger.warning(f"Skipping reachability check because health.petals.ml is down: {repr(e)}") | ||
return | ||
|
||
raise RuntimeError( | ||
f"Server has not become reachable from the Internet:\n\n" | ||
f"{response['message']}\n\n" | ||
f"You need to fix your port forwarding and/or firewall settings. How to do that:\n\n" | ||
f" 1. Choose a specific port for the Petals server, for example, 31337.\n" | ||
f" 2. Ensure that this port is accessible from the Internet and not blocked by your firewall.\n" | ||
f" 3. Add these arguments to explicitly announce your IP address and port to other peers:\n" | ||
f" python -m petals.cli.run_server ... --public_ip {response['your_ip']} --port 31337\n" | ||
f" 4. If it does not help, ask for help in our Discord: https://discord.gg/Wuk8BnrEPH\n" | ||
) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters