diff --git a/tools/prompt_injection_tester/pit/config/schema.py b/tools/prompt_injection_tester/pit/config/schema.py index 9ca4098..f016267 100644 --- a/tools/prompt_injection_tester/pit/config/schema.py +++ b/tools/prompt_injection_tester/pit/config/schema.py @@ -53,6 +53,12 @@ class AttackConfig(BaseModel): le=10.0, description="Requests per second", ) + timeout_per_test: int = Field( + default=30, + ge=1, + le=300, + description="Timeout per individual test in seconds", + ) class ReportingConfig(BaseModel):