diff --git a/litellm/litellm_core_utils/litellm_logging.py b/litellm/litellm_core_utils/litellm_logging.py index eebd61b5b86..0e230d0d710 100644 --- a/litellm/litellm_core_utils/litellm_logging.py +++ b/litellm/litellm_core_utils/litellm_logging.py @@ -3194,9 +3194,9 @@ class Logging(LiteLLMLoggingBaseClass): ) if isinstance(callback, CustomLogger): # custom logger class - callback_model_call_details: Final[dict] = ( + callback_model_call_details: Final[dict] = ( # mutable-ok: logger hook payload self.model_call_details - ) # mutable-ok: callback payload API + ) ################################## # call redaction hook for custom logger standard_redacted_model_call_details: Final[dict] = ( # mutable-ok: callback payload API @@ -3204,12 +3204,12 @@ class Logging(LiteLLMLoggingBaseClass): model_call_details=callback_model_call_details ) ) - streaming_redacted_model_call_details: Final[dict] = ( + streaming_redacted_model_call_details: Final[dict] = ( # mutable-ok: logger hook payload redact_streaming_responses_for_custom_logger( # mutable-ok: callback payload API model_call_details=standard_redacted_model_call_details, custom_logger=callback ) ) - redacted_model_call_details: Final[dict] = ( + redacted_model_call_details: Final[dict] = ( # mutable-ok: logger hook payload redact_model_call_details_for_custom_logger( # mutable-ok: callback payload API model_call_details=streaming_redacted_model_call_details, custom_logger=callback ) diff --git a/litellm/litellm_core_utils/redact_messages.py b/litellm/litellm_core_utils/redact_messages.py index fa216c53faa..f9cbb51b893 100644 --- a/litellm/litellm_core_utils/redact_messages.py +++ b/litellm/litellm_core_utils/redact_messages.py @@ -81,8 +81,9 @@ def redact_response_for_custom_logger(result: object, custom_logger: CustomLogge def redact_model_call_details_for_custom_logger( - model_call_details: dict[str, object], custom_logger: CustomLogger -) -> dict[str, object]: + model_call_details: dict[str, object], # mutable-ok: logger hook payload + custom_logger: CustomLogger, +) -> dict[str, object]: # mutable-ok: logger hook payload opted_out: Final = ( getattr(custom_logger, "message_logging", True) is not True or getattr(custom_logger, "turn_off_message_logging", False) is True