From 167cea2074a9ab2b697fc3b43ed63babe276217f Mon Sep 17 00:00:00 2001
From: rebear077 <51772538+rebear077@users.noreply.github.com>
Date: 星期三, 14 八月 2024 16:08:08 +0800
Subject: [PATCH] Go_ws_client客户端V1 (#2011)

---
 funasr/utils/version_checker.py |   13 +++++++++----
 1 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/funasr/utils/version_checker.py b/funasr/utils/version_checker.py
index 1a6970e..d016ed0 100644
--- a/funasr/utils/version_checker.py
+++ b/funasr/utils/version_checker.py
@@ -1,9 +1,10 @@
-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):
+    import requests
+
     url = f"https://pypi.org/pypi/{package_name}/json"
     response = requests.get(url)
     if response.status_code == 200:
@@ -13,12 +14,16 @@
         raise Exception("Failed to retrieve version information from PyPI.")
 
 
-def check_for_update():
+def check_for_update(disable=False):
     current_version = version.parse(__version__)
+    print(f"funasr version: {current_version}.")
+    if disable:
+        return
+
     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(f"New version is available: {pypi_version}.")
         print('Please use the command "pip install -U funasr" to upgrade.')
     else:
-        print("You are using the latest version of funasr.")
+        print(f"You are using the latest version of funasr-{current_version}")

--
Gitblit v1.9.1