enforece guardrails per API Key as enterprise

2026-07-11 03:04:02 +00:00 · 2024-08-20 17:34:28 -07:00
parent 225e75cafb
commit cea7b73015
1 changed files with 7 additions and 0 deletions
@@ -354,6 +354,13 @@ def move_guardrails_to_metadata(
    """
    if user_api_key_dict.metadata:
        if "guardrails" in user_api_key_dict.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}"
+                )
+
            data[_metadata_variable_name]["guardrails"] = user_api_key_dict.metadata[
                "guardrails"
            ]