From 52e425c7ace82fe733fe7ce6aa55bb4335193ec1 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 fa12258e1..7d9ad3cb2 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(