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(