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

Don't scan images

Browse source
This commit is contained in:
Ashton Sidhu 2026-05-28 16:15:34 -04:00 • committed by Yucheng Zhu
parent d6cce13308
commit d7ca5c4fae
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/proxy/guardrails/guardrail_hooks/hiddenlayer/hiddenlayer.py
View file

@ -263,6 +263,18 @@ class HiddenlayerGuardrail(CustomGuardrail):
if scan_params := inputs.get("structured_messages"):
last_msg: Final = scan_params[-1]
content = last_msg.get("content")
scan_contents = []
# Remove images from multimodal content before sending to HiddenLayer v1
if content and isinstance(content, list):
for item in content:
if isinstance(item, dict) and not item.get("type") == "image_url":
scan_contents.append(item)
effective_content: Final = str(scan_contents) if scan_contents else ""
else:
effective_content: Final = str(content or "")
result: _HiddenlayerResponse = await self._call_hiddenlayer(
project_id,
hl_request_metadata,
@ -270,7 +282,7 @@ class HiddenlayerGuardrail(CustomGuardrail):
"messages": [
{
"role": last_msg.get("role", "user"),
"content": str(last_msg.get("content", "")),
"content": effective_content
}
]
},