scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

style(guardrails): satisfy ruff format and annotate the new tests

Browse source 
`ruff format` wanted the widened `_collect_responses_fields` signature on one
line, and the three tests added with the split-vault fix needed return
annotations to keep ANN201 level with the base.
This commit is contained in:
Ninad Phalak 2026-09-05 06:25:12 -05:00
parent 6a4fc88d21
commit 8b2ee5ac7a
No known key found for this signature in database
GPG key ID: 59119ED515433744
2 changed files with 4 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/guardrails/guardrail_hooks/llm_shield_proxy/llm_shield_proxy.py
View file

@ -192,9 +192,7 @@ def _collect_system(data: MutableRequest, slots: _SlotSink) -> None:
_collect(part, "text", slots)
def _collect_responses_fields(
data: MutableRequest, slots: _SlotSink, privileged: _SlotSink
) -> None:
def _collect_responses_fields(data: MutableRequest, slots: _SlotSink, privileged: _SlotSink) -> None:
"""The Responses API sends text outside `messages`, in `instructions` and `input`.
`instructions` is written by the application, not by the caller, so it is

6
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_llm_shield_proxy.py
View file

@ -624,7 +624,7 @@ class TestVaultIsolation:
pytest.param({"instructions": "S", "input": "U"}, id="responses-instructions"),
],
)
def test_server_authored_text_is_split_from_the_callers(self, data: dict):
def test_server_authored_text_is_split_from_the_callers(self, data: dict) -> None:
"""Every request shape must sort its server-authored spans out of the caller's."""
caller, privileged = LLMShieldProxyGuardrail._locate_request_texts(data)
@ -632,7 +632,7 @@ class TestVaultIsolation:
assert [text for text, _ in privileged] == ["S"]
@pytest.mark.asyncio
async def test_a_system_prompt_gets_a_vault_of_its_own(self):
async def test_a_system_prompt_gets_a_vault_of_its_own(self) -> None:
"""The reply is restored against the caller's vault, so the two cannot be one."""
guardrail = _guardrail()
mock = _mock_post(guardrail, {"texts": ["[EMAIL_1]"]}, {"texts": ["[EMAIL_2]"]})
@ -654,7 +654,7 @@ class TestVaultIsolation:
assert guardrail._session_id(data) == caller_id
@pytest.mark.asyncio
async def test_the_system_prompt_vault_id_is_never_stored(self):
async def test_the_system_prompt_vault_id_is_never_stored(self) -> None:
"""Nothing can restore against the system vault later, because its id is not kept.
This is what stops a caller from having the model echo a placeholder out of a