From 8e733ce580827ea08e8e95bb6aca8fe71b412769 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=8Dtalo=20Johnny?= Date: Fri, 23 Aug 2024 09:15:54 -0300 Subject: [PATCH] fix: build version notification function (#3527) fix: added try/except block to version notification function --- src/backend/base/langflow/__main__.py | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/src/backend/base/langflow/__main__.py b/src/backend/base/langflow/__main__.py index f94f96e4d..505f24000 100644 --- a/src/backend/base/langflow/__main__.py +++ b/src/backend/base/langflow/__main__.py @@ -318,12 +318,19 @@ def is_prerelease(version: str) -> bool: def fetch_latest_version(package_name: str, include_prerelease: bool) -> Optional[str]: - response = httpx.get(f"https://pypi.org/pypi/{package_name}/json") - versions = response.json()["releases"].keys() - valid_versions = [v for v in versions if include_prerelease or not is_prerelease(v)] - if not valid_versions: - return None # Handle case where no valid versions are found - return max(valid_versions, key=lambda v: pkg_version.parse(v)) + valid_versions = [] + try: + response = httpx.get(f"https://pypi.org/pypi/{package_name}/json") + versions = response.json()["releases"].keys() + valid_versions = [v for v in versions if include_prerelease or not is_prerelease(v)] + + except Exception as e: + logger.exception(e) + + finally: + if not valid_versions: + return None # Handle case where no valid versions are found + return max(valid_versions, key=lambda v: pkg_version.parse(v)) def build_version_notice(current_version: str, package_name: str) -> str: