scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-11 08:38:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 280

test(logging): type the parameters of the per-callback failure redaction tests

Browse source
This commit is contained in:
yucheng 2026-11-04 17:22:12 -08:00
parent fbfd7b3d32
commit 64adf01c96
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/unit/litellm_core_utils/test_litellm_logging.py
View file

@ -12157,7 +12157,9 @@ async def test_failure_with_redaction_off_logs_the_error_as_raised(monkeypatch:
@pytest.mark.asyncio
async def test_failure_redacts_messages_and_error_for_a_callback_that_opted_out_of_message_logging(monkeypatch):
async def test_failure_redacts_messages_and_error_for_a_callback_that_opted_out_of_message_logging(
monkeypatch: pytest.MonkeyPatch,
) -> None:
opted_out: Final = _FailureCapture(message_logging=False)
logging_everything: Final = _FailureCapture()
monkeypatch.setattr(litellm, "turn_off_message_logging", False)
@ -12173,7 +12175,9 @@ async def test_failure_redacts_messages_and_error_for_a_callback_that_opted_out_
@pytest.mark.asyncio
@pytest.mark.parametrize("capture_type", [_FailureCapture, _SelfRedactingFailureCapture])
async def test_failure_scrubs_the_error_for_a_callback_with_its_own_turn_off_message_logging(monkeypatch, capture_type):
async def test_failure_scrubs_the_error_for_a_callback_with_its_own_turn_off_message_logging(
monkeypatch: pytest.MonkeyPatch, capture_type: type[_FailureCapture]
) -> None:
capture: Final = capture_type(turn_off_message_logging=True)
kwargs: Final = await _async_failure_payload(monkeypatch, redact=False, error=_prompt_echo_error(), capture=capture)
@ -12183,7 +12187,7 @@ async def test_failure_scrubs_the_error_for_a_callback_with_its_own_turn_off_mes
@pytest.mark.asyncio
async def test_failure_payload_keeps_the_excluded_fields(monkeypatch):
async def test_failure_payload_keeps_the_excluded_fields(monkeypatch: pytest.MonkeyPatch) -> None:
monkeypatch.setattr(litellm, "standard_logging_payload_excluded_fields", ["metadata"])
kwargs: Final = await _async_failure_payload(monkeypatch, redact=False, error=_prompt_echo_error())