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

lakera run on /embeddings

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-17 18:27:05 -07:00
parent 9753c3676a
commit b2bf5ad3d0
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
enterprise/enterprise_hooks/lakera_ai.py
View file

@ -56,13 +56,18 @@ class _ENTERPRISE_lakeraAI_Moderation(CustomLogger):
is False is False
): ):
return return
text = ""
if "messages" in data and isinstance(data["messages"], list): if "messages" in data and isinstance(data["messages"], list):
text = "" text = ""
for m in data["messages"]: # assume messages is a list for m in data["messages"]: # assume messages is a list
if "content" in m and isinstance(m["content"], str): if "content" in m and isinstance(m["content"], str):
text += m["content"] text += m["content"]
elif "input" in data and isinstance(data["input"], str):
text = data["input"]
elif "input" in data and isinstance(data["input"], list):
text = "\n".join(data["input"])
# https://platform.lakera.ai/account/api-keys # https://platform.lakera.ai/account/api-keys
data = {"input": text} data = {"input": text}