From 6377efd63eb74a079218af6debdf56146c04f1f0 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 13 Feb 2026 11:42:19 -0800 Subject: [PATCH] fix: fix linting error --- litellm/integrations/custom_guardrail.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/litellm/integrations/custom_guardrail.py b/litellm/integrations/custom_guardrail.py index af6d651803e..0b7b95e7bbc 100644 --- a/litellm/integrations/custom_guardrail.py +++ b/litellm/integrations/custom_guardrail.py @@ -29,7 +29,11 @@ from litellm.types.utils import ( LLMResponseTypes, StandardLoggingGuardrailInformation, ) -from fastapi.exceptions import HTTPException + +try: + from fastapi.exceptions import HTTPException +except ImportError: + HTTPException = None # type: ignore if TYPE_CHECKING: from litellm.litellm_core_utils.litellm_logging import Logging as LiteLLMLoggingObj @@ -664,7 +668,11 @@ class CustomGuardrail(CustomLogger): if isinstance(e, ModifyResponseException): return True - if isinstance(e, HTTPException) and e.status_code == 400: + if ( + HTTPException is not None + and isinstance(e, HTTPException) + and e.status_code == 400 + ): return True return False