From f8d6b646c5fafb3779afb1da7977fb926551692c Mon Sep 17 00:00:00 2001 From: Behnam Ousat Date: Wed, 13 May 2026 21:38:35 -0700 Subject: [PATCH] FIX Add an additional Microsoft target that supports api-versioned Azure AI endpoints --- pyrit/setup/initializers/components/targets.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/pyrit/setup/initializers/components/targets.py b/pyrit/setup/initializers/components/targets.py index 3a8049ca3a..6340880c8e 100644 --- a/pyrit/setup/initializers/components/targets.py +++ b/pyrit/setup/initializers/components/targets.py @@ -167,6 +167,15 @@ class TargetConfig: model_var="AZURE_OPENAI_GPT4O_UNSAFE_CHAT_MODEL2", underlying_model_var="AZURE_OPENAI_GPT4O_UNSAFE_CHAT_UNDERLYING_MODEL2", ), + TargetConfig( + registry_name="mai_target", + target_class=OpenAIChatTarget, + endpoint_var="MAI_CHAT_ENDPOINT", + key_var="MAI_CHAT_KEY", + model_var="MAI_CHAT_MODEL", + underlying_model_var="MAI_CHAT_UNDERLYING_MODEL", + extra_kwargs={"httpx_client_kwargs": {"default_query": {"api-version": "2024-05-01-preview"}}}, + ), # ============================================ # Adversarial Chat Target (for scenario attack techniques) # ============================================