From 833a268f4bcf02e72f101e986f7f47c81c17d9c7 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 25 Jan 2025 17:14:35 -0800 Subject: [PATCH] fix check on guardrails (#8008) --- litellm/proxy/litellm_pre_call_utils.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/litellm/proxy/litellm_pre_call_utils.py b/litellm/proxy/litellm_pre_call_utils.py index 2702da79fe1..1933bfb016e 100644 --- a/litellm/proxy/litellm_pre_call_utils.py +++ b/litellm/proxy/litellm_pre_call_utils.py @@ -707,14 +707,12 @@ def _add_guardrails_from_key_or_team_metadata( metadata_variable_name: The name of the metadata field in data """ + from litellm.proxy.utils import _premium_user_check + for _management_object_metadata in [key_metadata, team_metadata]: if _management_object_metadata and "guardrails" in _management_object_metadata: - from litellm.proxy.proxy_server import premium_user - - if premium_user is not True: - raise ValueError( - f"Using Guardrails on API Key {CommonProxyErrors.not_premium_user}" - ) + if len(_management_object_metadata["guardrails"]) > 0: + _premium_user_check() data[metadata_variable_name]["guardrails"] = _management_object_metadata[ "guardrails"