From f45fd5db3b03ee6bcf9d15c7b3dd40792a1a27cd Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 17 Feb 2024 23:38:53 -0800 Subject: [PATCH] fix(llama_guard.py): fix merge --- enterprise/enterprise_hooks/llama_guard.py | 1 + 1 file changed, 1 insertion(+) diff --git a/enterprise/enterprise_hooks/llama_guard.py b/enterprise/enterprise_hooks/llama_guard.py index fa12258e1c..7d9ad3cb29 100644 --- a/enterprise/enterprise_hooks/llama_guard.py +++ b/enterprise/enterprise_hooks/llama_guard.py @@ -114,6 +114,7 @@ class _ENTERPRISE_LlamaGuard(CustomLogger): model=self.model, messages=[safety_check_messages], hf_model_name="meta-llama/LlamaGuard-7b", + ) if "unsafe" in response.choices[0].message.content: raise HTTPException(