scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-15 23:31:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 165

fix(guardrail-registry): hardcode llm_as_a_judge in initializer registry so it loads regardless of package install path

Browse source
This commit is contained in:
Ishaan Jaffer 2026-04-23 16:11:25 -07:00
parent a731d10f19
commit af15eda10d
No known key found for this signature in database
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/guardrails/guardrail_registry.py
View file

@ -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]] = {