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