scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

test(proxy): check the guardrail header size budget by its result

Browse source
This commit is contained in:
yucheng 2026-11-02 14:03:51 -08:00
parent c4767ccf2b
commit bdfb8d5847
1 changed files with 2 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/unit/proxy/pass_through_endpoints/test_pass_through_endpoints.py
View file

@ -13790,16 +13790,10 @@ class TestGuardrailResponseHeaders:
assert headers == {"x-litellm-applied-policies": "tag-policy"}
assert h11.Response(status_code=200, headers=[*headers.items()]).status_code == 200
def test_guardrail_header_past_the_size_budget_is_left_out_without_scanning_its_characters(self):
def test_guardrail_header_past_the_size_budget_is_left_out(self):
names: Final = [f"g{index}" for index in range(2_000)]
with patch(
"litellm.proxy.pass_through_endpoints.pass_through_endpoints._is_sendable_header_value"
) as is_sendable:
headers: Final = _guardrail_response_headers({"metadata": {"applied_guardrails": names}})
assert headers == {}
is_sendable.assert_not_called()
assert _guardrail_response_headers({"metadata": {"applied_guardrails": names}}) == {}
@pytest.mark.parametrize("control", ["\x01", "\x1f", "\x7f", "\x00", "\n"])
def test_guardrail_name_with_a_control_character_stays_out_of_the_header(self, control):