From 9a6acc8fcedd4b449724eb604b67d16b4eaf86b2 Mon Sep 17 00:00:00 2001
From: shiva108 <esmurf@gmail.com>
Date: Tue, 27 Jan 2026 10:16:53 +0100
Subject: [PATCH] feat: Add optional model configuration to TargetConfig.

---
 tools/prompt_injection_tester/pit/config/schema.py | 6 ++++++
 1 file changed, 6 insertions(+)

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):