scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 455

fix(proxy): recognize the enterprise email loggers without importing litellm_enterprise

Browse source
This commit is contained in:
Rameshwaran Mohan 2026-10-17 13:49:08 +05:30
parent 65b2c1963e
commit 67d4d01911
1 changed files with 16 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
litellm/proxy/common_utils/model_deprecation_notifications.py
View file

@ -288,22 +288,28 @@ def render_model_deprecation_email(
return subject, body + EMAIL_FOOTER
_ENTERPRISE_BASE_EMAIL_LOGGER: Final = (
"litellm_enterprise.enterprise_callbacks.send_emails.base_email",
"BaseEmailLogger",
)
_ENTERPRISE_SMTP_EMAIL_LOGGER: Final = (
"litellm_enterprise.enterprise_callbacks.send_emails.smtp_email",
"SMTPEmailLogger",
)
def _class_identity(cls: type) -> tuple[str, str]:
return (cls.__module__, cls.__qualname__)
def _is_noop_email_logger(email_logger: object) -> bool:
"""The proxy hands out the bare enterprise base logger when no email provider is configured"""
try:
from litellm_enterprise.enterprise_callbacks.send_emails.base_email import BaseEmailLogger
except ImportError:
return False
return type(email_logger) is BaseEmailLogger
return _class_identity(type(email_logger)) == _ENTERPRISE_BASE_EMAIL_LOGGER
def _is_smtp_email_logger(email_logger: object) -> bool:
"""The enterprise SMTP logger fires one task per recipient without awaiting it, so its failures are invisible"""
try:
from litellm_enterprise.enterprise_callbacks.send_emails.smtp_email import SMTPEmailLogger
except ImportError:
return False
return isinstance(email_logger, SMTPEmailLogger)
return _ENTERPRISE_SMTP_EMAIL_LOGGER in map(_class_identity, type(email_logger).__mro__)
def make_email_deliverer(