diff --git a/litellm/proxy/guardrails/guardrail_hooks/unified_guardrail/unified_guardrail.py b/litellm/proxy/guardrails/guardrail_hooks/unified_guardrail/unified_guardrail.py index 0c202c7a54c..3e355a3b95d 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/unified_guardrail/unified_guardrail.py +++ b/litellm/proxy/guardrails/guardrail_hooks/unified_guardrail/unified_guardrail.py @@ -181,9 +181,9 @@ class UnifiedLLMGuardrails(CustomLogger): if user_api_key_dict.request_route is not None: call_types = get_call_types_for_route(user_api_key_dict.request_route) if call_types is not None: - call_type = call_types[0] + call_type = call_types[0] # type: ignore if call_type is None: - call_type = _infer_call_type(call_type=None, completion_response=response) + call_type = _infer_call_type(call_type=None, completion_response=response) # type: ignore if call_type is None: return response @@ -294,7 +294,7 @@ class UnifiedLLMGuardrails(CustomLogger): call_type = call_types[0] if call_type is None: - call_type = _infer_call_type(call_type=None, completion_response=item) + call_type = _infer_call_type(call_type=None, completion_response=item) # type: ignore # If call type not supported, just pass through all chunks if (