scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

fix(logging): suppress callback payload type warnings

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-14 19:53:16 +00:00
parent 19b44e3e05
commit d3bf96c049
2 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/litellm_core_utils/litellm_logging.py
View file

@ -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
)

5
litellm/litellm_core_utils/redact_messages.py
View file

@ -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