diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index c97934e9471..493c7b38970 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -5127,8 +5127,9 @@ class ProxyConfig: # instance already exists from an earlier registration if callback in litellm.success_callback: litellm.success_callback.remove(callback) - _add_custom_logger_callback_to_specific_event( # pyright: ignore[reportPrivateUsage] # mirrors the DB-config registration path - callback, "success" + _add_custom_logger_callback_to_specific_event( # pyright: ignore[reportPrivateUsage] + callback, + "success", # mirrors the DB-config registration path ) else: litellm.logging_callback_manager.add_litellm_success_callback(callback) @@ -5161,8 +5162,9 @@ class ProxyConfig: if callback in litellm._known_custom_logger_compatible_callbacks: if callback in litellm.failure_callback: litellm.failure_callback.remove(callback) - _add_custom_logger_callback_to_specific_event( # pyright: ignore[reportPrivateUsage] # mirrors the DB-config registration path - callback, "failure" + _add_custom_logger_callback_to_specific_event( # pyright: ignore[reportPrivateUsage] + callback, + "failure", # mirrors the DB-config registration path ) else: litellm.logging_callback_manager.add_litellm_failure_callback(callback)