diff --git a/litellm/proxy/auth/litellm_license.py b/litellm/proxy/auth/litellm_license.py index ec51f904c6..0310dcaf58 100644 --- a/litellm/proxy/auth/litellm_license.py +++ b/litellm/proxy/auth/litellm_license.py @@ -5,9 +5,6 @@ import json import os from datetime import datetime -from cryptography.hazmat.primitives import hashes, serialization -from cryptography.hazmat.primitives.asymmetric import padding, rsa - from litellm._logging import verbose_proxy_logger from litellm.llms.custom_httpx.http_handler import HTTPHandler @@ -27,16 +24,22 @@ class LicenseCheck: self.read_public_key() def read_public_key(self): - # current dir - current_dir = os.path.dirname(os.path.realpath(__file__)) + try: + from cryptography.hazmat.primitives import hashes, serialization + from cryptography.hazmat.primitives.asymmetric import padding, rsa - # check if public_key.pem exists - _path_to_public_key = os.path.join(current_dir, "public_key.pem") - if os.path.exists(_path_to_public_key): - with open(_path_to_public_key, "rb") as key_file: - self.public_key = serialization.load_pem_public_key(key_file.read()) - else: - self.public_key = None + # current dir + current_dir = os.path.dirname(os.path.realpath(__file__)) + + # check if public_key.pem exists + _path_to_public_key = os.path.join(current_dir, "public_key.pem") + if os.path.exists(_path_to_public_key): + with open(_path_to_public_key, "rb") as key_file: + self.public_key = serialization.load_pem_public_key(key_file.read()) + else: + self.public_key = None + except Exception as e: + verbose_proxy_logger.error(f"Error reading public key: {str(e)}") def _verify(self, license_str: str) -> bool: url = "{}/verify_license/{}".format(self.base_url, license_str) @@ -76,6 +79,9 @@ class LicenseCheck: def verify_license_without_api_request(self, public_key, license_key): try: + from cryptography.hazmat.primitives import hashes, serialization + from cryptography.hazmat.primitives.asymmetric import padding, rsa + # Decode the license key decoded = base64.b64decode(license_key) message, signature = decoded.split(b".", 1) @@ -107,4 +113,5 @@ class LicenseCheck: return True except Exception as e: + verbose_proxy_logger.error(str(e)) return False