fix(litellm_license.py): add better error logs

This commit is contained in:
Krrish Dholakia 2024-07-04 21:06:39 -07:00
parent 166eb24c54
commit 3ece295bf7

View file

@ -3,6 +3,7 @@
import base64 import base64
import json import json
import os import os
import traceback
from datetime import datetime from datetime import datetime
from litellm._logging import verbose_proxy_logger from litellm._logging import verbose_proxy_logger
@ -54,9 +55,13 @@ class LicenseCheck:
premium = response_json["verify"] premium = response_json["verify"]
assert isinstance(premium, bool) assert isinstance(premium, bool)
return premium return premium
except Exception as e: except Exception as e:
verbose_proxy_logger.error(
"litellm.proxy.auth.litellm_license.py::_verify - Unable to verify License via api. - {}".format(
str(e)
)
)
return False return False
def is_premium(self) -> bool: def is_premium(self) -> bool:
@ -116,5 +121,9 @@ class LicenseCheck:
return True return True
except Exception as e: except Exception as e:
verbose_proxy_logger.error(str(e)) verbose_proxy_logger.debug(
"litellm.proxy.auth.litellm_license.py::verify_license_without_api_request - Unable to verify License locally. - {}".format(
str(e)
)
)
return False return False