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

fix - lakera ai integration

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-23 15:25:26 -07:00
parent ec41226ee8
commit 03ec8219a4
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
enterprise/enterprise_hooks/lakera_ai.py
View file

@ -107,7 +107,10 @@ class _ENTERPRISE_lakeraAI_Moderation(CustomLogger):
""" """
_json_response = response.json() _json_response = response.json()
_results = _json_response.get("results", []) _results = _json_response.get("results", [])
flagged = _results.get("flagged", False) if len(_results) <= 0:
return
flagged = _results[0].get("flagged", False)
if flagged == True: if flagged == True:
raise HTTPException( raise HTTPException(