scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

chore(guardrails): drop unrelated masking change

Browse source
This commit is contained in:
feng.tsai 2026-09-02 11:14:24 +08:00
parent 5b95fd659b
commit f871a79b69
2 changed files with 0 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/guardrails/guardrail_hooks/bedrock_guardrails.py
View file

@ -3325,9 +3325,6 @@ class BedrockGuardrail(CustomGuardrail, BaseAWSLLM):
masking_index += 1
if item is not None:
new_content.append(item)
else:
new_content.append(item) # pyright: ignore[reportArgumentType] # opaque provider parts stay unchanged
return new_content, masking_index
def _apply_masking_to_response(

16
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_bedrock_guardrails.py
View file

@ -6241,22 +6241,6 @@ class TestBedrockGuardrailImageInput:
)
assert sent == [20, 20, 5], sent
def test_masking_keeps_image_parts_in_the_request(self):
"""Masking rewrites text in place; the image must survive to reach the model."""
image_part = {"type": "image_url", "image_url": {"url": self._PNG_DATA_URI}}
messages = [
{
"role": "user",
"content": [{"type": "text", "text": "my ssn is 123-45-6789"}, image_part],
}
]
updated = self._guardrail()._apply_masking_to_messages(messages=messages, masked_texts=["my ssn is {SSN}"])
assert updated[0]["content"] == [
{"type": "text", "text": "my ssn is {SSN}"},
image_part,
]
def test_load_credentials_assumes_role_with_external_id():
"""A trust policy requiring sts:ExternalId must be satisfied by the guardrail's aws_external_id."""