fix(llama_guard.py): fix merge

2024-02-17 23:38:53 -08:00 · 2024-02-17 23:38:53 -08:00 · 52e425c7ac
commit 52e425c7ac
parent 9522f6a2c1
1 changed files with 1 additions and 0 deletions
--- 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(