From 5ac57fb02f480a21d9d9c2abc5d036c461debbd0 Mon Sep 17 00:00:00 2001 From: shiva108 Date: Tue, 27 Jan 2026 11:47:55 +0100 Subject: [PATCH] feat: Hardcode LLM client timeout to 60 seconds to accommodate slow LLMs. --- tools/prompt_injection_tester/pit/orchestrator/phases.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/prompt_injection_tester/pit/orchestrator/phases.py b/tools/prompt_injection_tester/pit/orchestrator/phases.py index 59e4971..4c72fae 100644 --- a/tools/prompt_injection_tester/pit/orchestrator/phases.py +++ b/tools/prompt_injection_tester/pit/orchestrator/phases.py @@ -174,7 +174,7 @@ class DiscoveryPhase(Phase): api_type="openai", model=context.config.target.model or "", auth_token=context.config.target.token or "", - timeout=context.config.target.timeout, + timeout=60, # Use 60s timeout for slow LLMs rate_limit=context.config.attack.rate_limit, ) @@ -461,7 +461,7 @@ class AttackPhase(Phase): api_type="openai", model=context.config.target.model or "", auth_token=context.config.target.token or "", - timeout=context.config.target.timeout, + timeout=60, # Use 60s timeout for slow LLMs rate_limit=context.config.attack.rate_limit, )