From 9f73ae23fffd0b3a0c9a6fa483910d58a3d7d8d0 Mon Sep 17 00:00:00 2001
From: shiva108 <esmurf@gmail.com>
Date: Tue, 27 Jan 2026 10:15:51 +0100
Subject: [PATCH] chore: Added optional 'model' parameter to schema.py in
 prompt_injection_tester

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

diff --git a/tools/prompt_injection_tester/pit/config/schema.py b/tools/prompt_injection_tester/pit/config/schema.py
index 30e215a..9ca4098 100644
--- a/tools/prompt_injection_tester/pit/config/schema.py
+++ b/tools/prompt_injection_tester/pit/config/schema.py
@@ -111,6 +111,7 @@ class Config(BaseModel):
         cls,
         target_url: str,
         token: Optional[str] = None,
+        model: Optional[str] = None,
         api_type: str = "openai",
         timeout: int = 30,
         categories: Optional[List[str]] = None,
@@ -128,6 +129,7 @@ class Config(BaseModel):
             target=TargetConfig(
                 url=target_url,
                 token=token,
+                model=model,
                 api_type=api_type,
                 timeout=timeout,
             ),