diff --git a/litellm/types/guardrails.py b/litellm/types/guardrails.py index b9a04fe5e60..ae65af35275 100644 --- a/litellm/types/guardrails.py +++ b/litellm/types/guardrails.py @@ -40,6 +40,7 @@ from litellm.types.proxy.guardrails.guardrail_hooks.hiddenlayer import ( ) from litellm.types.proxy.guardrails.guardrail_hooks.alice_wonderfence import ( WonderFenceGuardrailConfigModel, +) """ Pydantic object defining how to set guardrails on litellm proxy @@ -100,6 +101,7 @@ class SupportedGuardrailIntegrations(Enum): LLM_AS_A_JUDGE = "llm_as_a_judge" ALICE_WONDERFENCE = "alice_wonderfence" + class Role(Enum): SYSTEM = "system" ASSISTANT = "assistant"