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

test(ollama): cover short and boundary-length repair warning previews

Browse source
This commit is contained in:
mateo-berri 2026-11-07 01:48:18 -08:00
parent 5ef6d62cca
commit 1f71ed377f
1 changed files with 23 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
tests/unit/litellm_core_utils/prompt_templates/test_tool_call_json_repair.py
View file

@ -122,8 +122,9 @@ def test_salvage_concatenated_tool_arguments(raw: str, expected: tuple[dict[str,
def test_parse_tool_call_arguments_repair_warning_previews_the_first_200_characters(
caplog: pytest.LogCaptureFixture,
caplog: pytest.LogCaptureFixture, monkeypatch: pytest.MonkeyPatch
) -> None:
monkeypatch.setattr(litellm, "turn_off_message_logging", False)
arguments: Final = '{"patient": "Jane Doe", "note": "' + "x" * 300 + '"'
with caplog.at_level(logging.WARNING, logger="LiteLLM"):
result: Final = parse_tool_call_arguments(arguments, tool_name="lookup", context="test")
@ -136,6 +137,27 @@ def test_parse_tool_call_arguments_repair_warning_previews_the_first_200_charact
assert arguments not in caplog.text
@pytest.mark.parametrize(
("arguments", "expected"),
(
('{"patient": "Jane Doe"', {"patient": "Jane Doe"}),
('{"note": "' + "x" * 189 + '"', {"note": "x" * 189}),
),
)
def test_parse_tool_call_arguments_repair_warning_logs_arguments_up_to_200_characters_whole(
caplog: pytest.LogCaptureFixture,
monkeypatch: pytest.MonkeyPatch,
arguments: str,
expected: dict[str, str],
) -> None:
monkeypatch.setattr(litellm, "turn_off_message_logging", False)
with caplog.at_level(logging.WARNING, logger="LiteLLM"):
result: Final = parse_tool_call_arguments(arguments, tool_name="lookup", context="test")
assert result == expected
assert f"Original ({len(arguments)} chars): {arguments}\n" in caplog.text
def test_parse_tool_call_arguments_repair_warning_redacts_arguments_when_message_logging_is_off(
caplog: pytest.LogCaptureFixture, monkeypatch: pytest.MonkeyPatch
) -> None: