diff --git a/litellm/proxy/guardrails/guardrail_registry.py b/litellm/proxy/guardrails/guardrail_registry.py index d325c7d6ee6..1675a92af32 100644 --- a/litellm/proxy/guardrails/guardrail_registry.py +++ b/litellm/proxy/guardrails/guardrail_registry.py @@ -8,6 +8,8 @@ from datetime import datetime, timezone from itertools import chain, count from typing import TYPE_CHECKING, Final, Literal, Optional, Protocol, TypeAlias, cast +from pydantic import ValidationError + import litellm from litellm import Router from litellm._logging import verbose_proxy_logger @@ -50,7 +52,6 @@ from litellm.types.guardrails import ( LitellmParams, SupportedGuardrailIntegrations, ) -from pydantic import ValidationError from .guardrail_hooks.llm_as_a_judge import ( initialize_guardrail as initialize_llm_as_a_judge, diff --git a/litellm/types/guardrails.py b/litellm/types/guardrails.py index d514a28b210..6b67aa4edc3 100644 --- a/litellm/types/guardrails.py +++ b/litellm/types/guardrails.py @@ -4,6 +4,9 @@ from enum import Enum from types import MappingProxyType from typing import Final, Literal +from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator +from typing_extensions import ReadOnly, Required, TypedDict + from litellm.constants import BEDROCK_APPLY_GUARDRAIL_CHUNK_BUDGET_CHARS from litellm.types.proxy.guardrails.guardrail_hooks.agent_365 import ( Agent365GuardrailConfigModel, @@ -68,8 +71,6 @@ from litellm.types.proxy.guardrails.guardrail_hooks.vigil_guard import ( from litellm.types.proxy.guardrails.guardrail_hooks.xecguard import ( XecGuardConfigModel, ) -from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator -from typing_extensions import ReadOnly, Required, TypedDict """ Pydantic object defining how to set guardrails on litellm proxy