diff --git a/litellm/types/guardrails.py b/litellm/types/guardrails.py index 0138a760cfd..2725653d9ac 100644 --- a/litellm/types/guardrails.py +++ b/litellm/types/guardrails.py @@ -147,6 +147,8 @@ class SupportedGuardrailIntegrations(Enum): AGENT_365 = "agent_365" CONDUCT = "conduct" TEALTIGER = "tealtiger" + DECISION_MODEL = "decision_model" + class Role(Enum): SYSTEM = "system"