wehub-resource-sync
f36e2104d8
sep-tests / ragflow_preflight (push) Waiting to run
sep-tests / ragflow_tests_infinity (push) Blocked by required conditions
sep-tests / ragflow_tests_elasticsearch (push) Blocked by required conditions
tests / ragflow_preflight (push) Waiting to run
tests / ragflow_tests_infinity (push) Blocked by required conditions
tests / ragflow_tests_elasticsearch (push) Blocked by required conditions
21 lines
684 B
Python
21 lines
684 B
Python
import time
|
|
import logging
|
|
from datetime import datetime
|
|
from datetime import timedelta
|
|
from datetime import timezone
|
|
|
|
from github import Github
|
|
|
|
|
|
def sleep_after_rate_limit_exception(github_client: Github) -> None:
|
|
"""
|
|
Sleep until the GitHub rate limit resets.
|
|
|
|
Args:
|
|
github_client: The GitHub client that hit the rate limit
|
|
"""
|
|
sleep_time = github_client.get_rate_limit().core.reset.replace(tzinfo=timezone.utc) - datetime.now(tz=timezone.utc)
|
|
sleep_time += timedelta(minutes=1) # add an extra minute just to be safe
|
|
logging.info("Ran into Github rate-limit. Sleeping %s seconds.", sleep_time.seconds)
|
|
time.sleep(sleep_time.total_seconds())
|