phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix check on guardrails (#8008)

Browse source
This commit is contained in:
Ishaan Jaff 2025-01-25 17:14:35 -08:00 • committed by GitHub
parent 08b124aeb6
commit 833a268f4b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/litellm_pre_call_utils.py
View file

@ -707,14 +707,12 @@ def _add_guardrails_from_key_or_team_metadata(
metadata_variable_name: The name of the metadata field in data 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]: for _management_object_metadata in [key_metadata, team_metadata]:
if _management_object_metadata and "guardrails" in _management_object_metadata: if _management_object_metadata and "guardrails" in _management_object_metadata:
from litellm.proxy.proxy_server import premium_user if len(_management_object_metadata["guardrails"]) > 0:
_premium_user_check()
if premium_user is not True:
raise ValueError(
f"Using Guardrails on API Key {CommonProxyErrors.not_premium_user}"
)
data[metadata_variable_name]["guardrails"] = _management_object_metadata[ data[metadata_variable_name]["guardrails"] = _management_object_metadata[
"guardrails" "guardrails"