scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

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
Show stats 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()
_results = _json_response.get("results", [])
flagged = _results.get("flagged", False)
if len(_results) <= 0:
return
flagged = _results[0].get("flagged", False)
if flagged == True:
raise HTTPException(