feat(gemini_tts): adding support to Gemini TTS languageCode parameters

enterprise/litellm_enterprise/enterprise_callbacks/callback_controls.py

@@ -14,53 +14,74 @@
 class EnterpriseCallbackControls:
     @staticmethod
     def is_callback_disabled_dynamically(
-            callback: litellm.CALLBACK_TYPES, 
-            litellm_params: dict,
-            standard_callback_dynamic_params: StandardCallbackDynamicParams
-        ) -> bool:
-            """
-            Check if a callback is disabled via the x-litellm-disable-callbacks header or via `litellm_disabled_callbacks` in standard_callback_dynamic_params.
-
-            Args:
-                callback: The callback to check (can be string, CustomLogger instance, or callable)
-                litellm_params: Parameters containing proxy server request info
-
-            Returns:
-                bool: True if the callback should be disabled, False otherwise
-            """
-            from litellm.litellm_core_utils.custom_logger_registry import (
-                CustomLoggerRegistry,
+        callback: litellm.CALLBACK_TYPES,
+        litellm_params: dict,
+        standard_callback_dynamic_params: StandardCallbackDynamicParams,
+    ) -> bool:
+        """
+        Check if a callback is disabled via the x-litellm-disable-callbacks header or via `litellm_disabled_callbacks` in standard_callback_dynamic_params.
+
+        Args:
+            callback: The callback to check (can be string, CustomLogger instance, or callable)
+            litellm_params: Parameters containing proxy server request info
+
+        Returns:
+            bool: True if the callback should be disabled, False otherwise
+        """
+        from litellm.litellm_core_utils.custom_logger_registry import (
+            CustomLoggerRegistry,
+        )
+
+        try:
+            disabled_callbacks = EnterpriseCallbackControls.get_disabled_callbacks(
+                litellm_params, standard_callback_dynamic_params
+            )
+            verbose_logger.debug(
+                f"Dynamically disabled callbacks from {X_LITELLM_DISABLE_CALLBACKS}: {disabled_callbacks}"
             )
+            verbose_logger.debug(
+                f"Checking if {callback} is disabled via headers. Disable callbacks from headers: {disabled_callbacks}"
+            )
+            if disabled_callbacks is not None:
+                #########################################################
+                # premium user check
+                #########################################################
+                if (
+                    not EnterpriseCallbackControls._should_allow_dynamic_callback_disabling()
+                ):
+                    return False
+                #########################################################
+                if isinstance(callback, str):
+                    if callback.lower() in disabled_callbacks:
+                        verbose_logger.debug(
+                            f"Not logging to {callback} because it is disabled via {X_LITELLM_DISABLE_CALLBACKS}"
+                        )
+                        return True
+                elif isinstance(callback, CustomLogger):
+                    # get the string name of the callback
+                    callback_str = (
+                        CustomLoggerRegistry.get_callback_str_from_class_type(
+                            callback.__class__
+                        )
+                    )
+                    if (
+                        callback_str is not None
+                        and callback_str.lower() in disabled_callbacks
+                    ):
+                        verbose_logger.debug(
+                            f"Not logging to {callback_str} because it is disabled via {X_LITELLM_DISABLE_CALLBACKS}"
+                        )
+                        return True
+            return False
+        except Exception as e:
+            verbose_logger.debug(f"Error checking disabled callbacks header: {str(e)}")
+            return False
 
-            try:
-                disabled_callbacks = EnterpriseCallbackControls.get_disabled_callbacks(litellm_params, standard_callback_dynamic_params)
-                verbose_logger.debug(f"Dynamically disabled callbacks from {X_LITELLM_DISABLE_CALLBACKS}: {disabled_callbacks}")
-                verbose_logger.debug(f"Checking if {callback} is disabled via headers. Disable callbacks from headers: {disabled_callbacks}")
-                if disabled_callbacks is not None:
-                    #########################################################
-                    # premium user check
-                    #########################################################
-                    if not EnterpriseCallbackControls._should_allow_dynamic_callback_disabling():
-                        return False
-                    #########################################################
-                    if isinstance(callback, str):
-                        if callback.lower() in disabled_callbacks:
-                            verbose_logger.debug(f"Not logging to {callback} because it is disabled via {X_LITELLM_DISABLE_CALLBACKS}")
-                            return True
-                    elif isinstance(callback, CustomLogger):
-                        # get the string name of the callback
-                        callback_str = CustomLoggerRegistry.get_callback_str_from_class_type(callback.__class__)
-                        if callback_str is not None and callback_str.lower() in disabled_callbacks:
-                            verbose_logger.debug(f"Not logging to {callback_str} because it is disabled via {X_LITELLM_DISABLE_CALLBACKS}")
-                            return True
-                return False
-            except Exception as e:
-                verbose_logger.debug(
-                    f"Error checking disabled callbacks header: {str(e)}"
-                )
-                return False
     @staticmethod
-    def get_disabled_callbacks(litellm_params: dict, standard_callback_dynamic_params: StandardCallbackDynamicParams) -> Optional[List[str]]:
+    def get_disabled_callbacks(
+        litellm_params: dict,
+        standard_callback_dynamic_params: StandardCallbackDynamicParams,
+    ) -> Optional[List[str]]:
         """
         Get the disabled callbacks from the standard callback dynamic params.
         """
@@ -71,29 +92,39 @@ def get_disabled_callbacks(litellm_params: dict, standard_callback_dynamic_param
         request_headers = get_proxy_server_request_headers(litellm_params)
         disabled_callbacks = request_headers.get(X_LITELLM_DISABLE_CALLBACKS, None)
         if disabled_callbacks is not None:
-            disabled_callbacks = set([cb.strip().lower() for cb in disabled_callbacks.split(",")])
+            disabled_callbacks = set(
+                [cb.strip().lower() for cb in disabled_callbacks.split(",")]
+            )
             return list(disabled_callbacks)
-
 
         #########################################################
         # check if disabled via request body
         #########################################################
-        if standard_callback_dynamic_params.get("litellm_disabled_callbacks", None) is not None:
-            return standard_callback_dynamic_params.get("litellm_disabled_callbacks", None)
-
+        if (
+            standard_callback_dynamic_params.get("litellm_disabled_callbacks", None)
+            is not None
+        ):
+            return standard_callback_dynamic_params.get(
+                "litellm_disabled_callbacks", None
+            )
+
         return None
-    
+
     @staticmethod
     def _should_allow_dynamic_callback_disabling():
         import litellm
         from litellm.proxy.proxy_server import premium_user
 
         # Check if admin has disabled this feature
         if litellm.allow_dynamic_callback_disabling is not True:
-            verbose_logger.debug("Dynamic callback disabling is disabled by admin via litellm.allow_dynamic_callback_disabling")
+            verbose_logger.debug(
+                "Dynamic callback disabling is disabled by admin via litellm.allow_dynamic_callback_disabling"
+            )
             return False
-        
+
         if premium_user:
             return True
-        verbose_logger.warning(f"Disabling callbacks using request headers is an enterprise feature. {CommonProxyErrors.not_premium_user.value}")
-        return False
+        verbose_logger.warning(
+            f"Disabling callbacks using request headers is an enterprise feature. {CommonProxyErrors.not_premium_user.value}"
+        )
+        return False

enterprise/litellm_enterprise/enterprise_callbacks/send_emails/base_email.py

@@ -349,8 +349,10 @@ async def send_max_budget_alert_email(
         )
 
         # Calculate percentage and alert threshold
-        percentage = threshold_pct if threshold_pct is not None else int(
-            EMAIL_BUDGET_ALERT_MAX_SPEND_ALERT_PERCENTAGE * 100
+        percentage = (
+            threshold_pct
+            if threshold_pct is not None
+            else int(EMAIL_BUDGET_ALERT_MAX_SPEND_ALERT_PERCENTAGE * 100)
         )
         threshold_fraction = percentage / 100.0
         alert_threshold_str = (
@@ -609,9 +611,7 @@ async def _handle_multi_threshold_max_budget_alert(
                 continue
 
             _id = user_info.token or user_info.user_id or "default_id"
-            _cache_key = (
-                f"email_budget_alerts:max_budget_alert:{threshold_pct}:{_id}"
-            )
+            _cache_key = f"email_budget_alerts:max_budget_alert:{threshold_pct}:{_id}"
 
             result = await _cache.async_get_cache(key=_cache_key)
             if result is not None:
@@ -630,7 +630,9 @@ async def _handle_multi_threshold_max_budget_alert(
                 continue
             recipient_emails = list(set(emails))
 
-            event_message = f"Max Budget Alert - {threshold_pct}% of Maximum Budget Reached"
+            event_message = (
+                f"Max Budget Alert - {threshold_pct}% of Maximum Budget Reached"
+            )
             webhook_event = WebhookEvent(
                 event="max_budget_alert",
                 event_message=event_message,

enterprise/litellm_enterprise/enterprise_callbacks/send_emails/sendgrid_email.py

@@ -15,7 +15,6 @@
 
 from .base_email import BaseEmailLogger
 
-
 SENDGRID_API_ENDPOINT = "https://api.sendgrid.com/v3/mail/send"
 
 
@@ -79,4 +78,4 @@ async def send_email(
         verbose_logger.debug(
             f"SendGrid response status={response.status_code}, body={response.text}"
         )
-        return
+        return

enterprise/litellm_enterprise/enterprise_callbacks/send_emails/smtp_email.py

@@ -1,6 +1,7 @@
 """
 This is the litellm SMTP email integration
 """
+
 import asyncio
 from typing import List
 

enterprise/litellm_enterprise/litellm_core_utils/litellm_logging.py

@@ -1,6 +1,7 @@
 """
 Enterprise specific logging utils
 """
+
 from litellm.litellm_core_utils.litellm_logging import StandardLoggingMetadata
 
 

enterprise/litellm_enterprise/proxy/audit_logging_endpoints.py

@@ -153,11 +153,11 @@ async def get_audit_logs(
 
     # Return paginated response
     return PaginatedAuditLogResponse(
-        audit_logs=[
-            AuditLogResponse(**audit_log.model_dump()) for audit_log in audit_logs
-        ]
-        if audit_logs
-        else [],
+        audit_logs=(
+            [AuditLogResponse(**audit_log.model_dump()) for audit_log in audit_logs]
+            if audit_logs
+            else []
+        ),
         total=total_count,
         page=page,
         page_size=page_size,

enterprise/litellm_enterprise/proxy/auth/__init__.py

@@ -7,4 +7,4 @@
 
 from .custom_sso_handler import EnterpriseCustomSSOHandler
 
-__all__ = ["EnterpriseCustomSSOHandler"] 
+__all__ = ["EnterpriseCustomSSOHandler"]