From af15eda10dca4af811466ea0d4f5547e5919e9b5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Thu, 23 Apr 2026 16:11:25 -0700 Subject: [PATCH] fix(guardrail-registry): hardcode llm_as_a_judge in initializer registry so it loads regardless of package install path --- litellm/proxy/guardrails/guardrail_registry.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/proxy/guardrails/guardrail_registry.py b/litellm/proxy/guardrails/guardrail_registry.py index 96175877d65..063f431d2a2 100644 --- a/litellm/proxy/guardrails/guardrail_registry.py +++ b/litellm/proxy/guardrails/guardrail_registry.py @@ -34,6 +34,7 @@ from .guardrail_initializers import ( initialize_presidio, initialize_tool_permission, ) +from .guardrail_hooks.llm_as_a_judge import initialize_guardrail as initialize_llm_as_a_judge guardrail_initializer_registry = { SupportedGuardrailIntegrations.BEDROCK.value: initialize_bedrock, @@ -43,6 +44,7 @@ guardrail_initializer_registry = { SupportedGuardrailIntegrations.HIDE_SECRETS.value: initialize_hide_secrets, SupportedGuardrailIntegrations.TOOL_PERMISSION.value: initialize_tool_permission, SupportedGuardrailIntegrations.GRAYSWAN.value: initialize_grayswan, + SupportedGuardrailIntegrations.LLM_AS_A_JUDGE.value: initialize_llm_as_a_judge, } guardrail_class_registry: Dict[str, Type[CustomGuardrail]] = {