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

fix(guardrails): narrow prompt and input to a list before iterating

Browse source 
Guarding with a conditional iterable left the value un-narrowed, so passing
it on was an argument-type error and the element checks read as unreachable.
An early return narrows it properly and reads better.
This commit is contained in:
Ninad Phalak 2026-09-03 20:26:57 -05:00
parent 46f13807a5
commit a35c502819
No known key found for this signature in database
GPG key ID: 59119ED515433744
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/guardrails/guardrail_hooks/llm_shield/llm_shield.py
View file

@ -102,7 +102,9 @@ def _collect_prompt(data: MutableRequest, slots: _SlotSink) -> None:
if isinstance(prompt, str):
_collect(data, "prompt", slots)
return
for index in range(len(prompt)) if isinstance(prompt, list) else ():
if not isinstance(prompt, list):
return
for index in range(len(prompt)):
_collect_entry(prompt, index, slots)
@ -146,7 +148,9 @@ def _collect_responses_fields(data: MutableRequest, slots: _SlotSink) -> None:
if isinstance(request_input, str):
_collect(data, "input", slots)
return
for index, item in enumerate(request_input if isinstance(request_input, list) else ()):
if not isinstance(request_input, list):
return
for index, item in enumerate(request_input):
if isinstance(item, str):
# The embeddings and moderations shape: `input` as an array of strings.
_collect_entry(request_input, index, slots)