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

fix(guardrails): satisfy strict comparison lint

Browse source
This commit is contained in:
zhengguangzhuo 2026-10-08 17:28:45 +08:00
parent 654bd3d0d6
commit e013a40b57
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/integrations/custom_guardrail.py
View file

@ -1539,15 +1539,15 @@ class CustomGuardrail(CustomLogger):
):
return data.get("messages")
if (
call_type == CallTypes.embedding.value
or call_type == CallTypes.aembedding.value
or call_type == CallTypes.responses.value
or call_type == CallTypes.aresponses.value
if call_type in (
CallTypes.embedding.value,
CallTypes.aembedding.value,
CallTypes.responses.value,
CallTypes.aresponses.value,
):
input_data: Final = data.get("input")
if call_type == CallTypes.embedding.value or call_type == CallTypes.aembedding.value:
if call_type in (CallTypes.embedding.value, CallTypes.aembedding.value):
if isinstance(input_data, str):
return [{"role": "user", "content": input_data}]
if isinstance(input_data, list):