From e013a40b57e6fc1da253ee27aa575c730607c77f Mon Sep 17 00:00:00 2001 From: zhengguangzhuo <100548634+zhengguangzhuo@users.noreply.github.com> Date: Thu, 8 Oct 2026 17:28:45 +0800 Subject: [PATCH] fix(guardrails): satisfy strict comparison lint --- litellm/integrations/custom_guardrail.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/litellm/integrations/custom_guardrail.py b/litellm/integrations/custom_guardrail.py index b6b88ea7cc0..4ca2bf7b305 100644 --- a/litellm/integrations/custom_guardrail.py +++ b/litellm/integrations/custom_guardrail.py @@ -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):