phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix lakera ai tests

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-19 07:49:54 -07:00
parent 2ad1c0455a
commit e66a735304
1 changed files with 16 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

32
enterprise/enterprise_hooks/lakera_ai.py
View file

@ -110,28 +110,28 @@ class _ENTERPRISE_lakeraAI_Moderation(CustomLogger):
"content": content, "content": content,
} }
lakera_input = [ lakera_input = [
v v
for k, v in sorted( for k, v in sorted(
lakera_input_dict.items(), key=lambda x: INPUT_POSITIONING_MAP[x[0]] lakera_input_dict.items(), key=lambda x: INPUT_POSITIONING_MAP[x[0]]
) )
if v is not None if v is not None
] ]
if len(lakera_input) == 0: if len(lakera_input) == 0:
verbose_proxy_logger.debug( verbose_proxy_logger.debug(
"Skipping lakera prompt injection, no roles with messages found" "Skipping lakera prompt injection, no roles with messages found"
) )
return return
data = {"input": lakera_input}
_json_data = json.dumps(data)
elif "input" in data and isinstance(data["input"], str): elif "input" in data and isinstance(data["input"], str):
text = data["input"] text = data["input"]
_json_data = json.dumps({"input": text})
elif "input" in data and isinstance(data["input"], list): elif "input" in data and isinstance(data["input"], list):
text = "\n".join(data["input"]) text = "\n".join(data["input"])
_json_data = json.dumps({"input": text})
# https://platform.lakera.ai/account/api-keys # https://platform.lakera.ai/account/api-keys
data = {"input": lakera_input}
_json_data = json.dumps(data)
""" """
export LAKERA_GUARD_API_KEY=<your key> export LAKERA_GUARD_API_KEY=<your key>