diff --git a/litellm/integrations/custom_logger.py b/litellm/integrations/custom_logger.py index 0b198ed8d1..cccabf53e5 100644 --- a/litellm/integrations/custom_logger.py +++ b/litellm/integrations/custom_logger.py @@ -178,9 +178,7 @@ class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callbac async def async_log_failure_event(self, kwargs, response_obj, start_time, end_time): pass - async def async_log_audit_log_event( - self, audit_log: "StandardAuditLogPayload" - ): + async def async_log_audit_log_event(self, audit_log: "StandardAuditLogPayload"): """Called when an audit log is created. Override in subclasses to handle.""" pass diff --git a/litellm/proxy/management_helpers/audit_logs.py b/litellm/proxy/management_helpers/audit_logs.py index a26a38bc4c..08d92c3e6a 100644 --- a/litellm/proxy/management_helpers/audit_logs.py +++ b/litellm/proxy/management_helpers/audit_logs.py @@ -101,9 +101,7 @@ async def _dispatch_audit_log_to_callbacks( continue if isinstance(resolved, CustomLogger): - task = asyncio.create_task( - resolved.async_log_audit_log_event(payload) - ) + task = asyncio.create_task(resolved.async_log_audit_log_event(payload)) task.add_done_callback(_audit_log_task_done_callback) except Exception as e: verbose_proxy_logger.error(