Increase timeout and llama guard

langwatch · Feb 19, 2025 · cf3b3ea · cf3b3ea
1 parent 4cdd0f6
commit cf3b3ea
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/evaluators/huggingface/langevals_huggingface/llama_guard.py b/evaluators/huggingface/langevals_huggingface/llama_guard.py
@@ -163,6 +163,7 @@ def evaluate(self, entry: LlamaGuardEntry) -> SingleEvaluationResult:
                 f"https://api.cloudflare.com/client/v4/accounts/{account_id}/ai/run/@hf/thebloke/llamaguard-7b-awq",
                 headers={"Authorization": f"Bearer {api_key}"},
                 json={"prompt": prompt},
+                timeout=600,
             )
             response.raise_for_status()
             result = response.json()