Spaces:
Sleeping
Sleeping
File size: 1,728 Bytes
2eafbc4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
import threading
import time
import requests
from inference.core.env import DISABLE_VERSION_CHECK, VERSION_CHECK_MODE
from inference.core.logger import logger
from inference.core.version import __version__
latest_release = None
last_checked = 0
cache_duration = 86400 # 24 hours
log_frequency = 300 # 5 minutes
def get_latest_release_version():
global latest_release, last_checked
now = time.time()
if latest_release is None or now - last_checked > cache_duration:
try:
logger.debug("Checking for latest inference release version...")
response = requests.get(
"https://api.github.com/repos/roboflow/inference/releases/latest"
)
response.raise_for_status()
latest_release = response.json()["tag_name"].lstrip("v")
last_checked = now
except requests.exceptions.RequestException:
pass
def check_latest_release_against_current():
get_latest_release_version()
if latest_release is not None and latest_release != __version__:
logger.warning(
f"Your inference package version {__version__} is out of date! Please upgrade to version {latest_release} of inference for the latest features and bug fixes by running `pip install --upgrade inference`."
)
def check_latest_release_against_current_continuous():
while True:
check_latest_release_against_current()
time.sleep(log_frequency)
if not DISABLE_VERSION_CHECK:
if VERSION_CHECK_MODE == "continuous":
t = threading.Thread(target=check_latest_release_against_current_continuous)
t.daemon = True
t.start()
else:
check_latest_release_against_current()
|