Skip to content

Commit 2741dce

Browse files
behnam-oBehnam Ousat
andauthored
FIX Add an additional Microsoft target that supports api-version-sensitive Azure AI endpoints (#1730)
Co-authored-by: Behnam Ousat <behnamousat@microsoft.com>
1 parent e677567 commit 2741dce

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

pyrit/setup/initializers/components/targets.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,15 @@ class TargetConfig:
167167
model_var="AZURE_OPENAI_GPT4O_UNSAFE_CHAT_MODEL2",
168168
underlying_model_var="AZURE_OPENAI_GPT4O_UNSAFE_CHAT_UNDERLYING_MODEL2",
169169
),
170+
TargetConfig(
171+
registry_name="mai_target",
172+
target_class=OpenAIChatTarget,
173+
endpoint_var="MAI_CHAT_ENDPOINT",
174+
key_var="MAI_CHAT_KEY",
175+
model_var="MAI_CHAT_MODEL",
176+
underlying_model_var="MAI_CHAT_UNDERLYING_MODEL",
177+
extra_kwargs={"httpx_client_kwargs": {"default_query": {"api-version": "2024-05-01-preview"}}},
178+
),
170179
# ============================================
171180
# Adversarial Chat Target (for scenario attack techniques)
172181
# ============================================

0 commit comments

Comments
 (0)