From 03ec8219a4e94d945ff9e300a074af32e0046f1d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 23 May 2024 15:25:26 -0700 Subject: [PATCH] fix - lakera ai integration --- enterprise/enterprise_hooks/lakera_ai.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/enterprise/enterprise_hooks/lakera_ai.py b/enterprise/enterprise_hooks/lakera_ai.py index bef009dc5..dd37ae2c1 100644 --- a/enterprise/enterprise_hooks/lakera_ai.py +++ b/enterprise/enterprise_hooks/lakera_ai.py @@ -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(