FunASR/funasr/utils/version_checker.py
2024-06-21 11:19:19 +08:00

25 lines
887 B
Python

import requests
from packaging import version
from funasr import __version__ # Ensure that __version__ is defined in your package's __init__.py
def get_pypi_version(package_name):
url = f"https://pypi.org/pypi/{package_name}/json"
response = requests.get(url)
if response.status_code == 200:
data = response.json()
return version.parse(data["info"]["version"])
else:
raise Exception("Failed to retrieve version information from PyPI.")
def check_for_update():
current_version = version.parse(__version__)
pypi_version = get_pypi_version("funasr")
if current_version < pypi_version:
print(f"New version available: {pypi_version}. Your current version is {current_version}.")
print('Please use the command "pip install -U funasr" to upgrade.')
else:
print("You are using the latest version of funasr.")