From c0de83f96c8a76dee956c71538b8f2c3cc916710 Mon Sep 17 00:00:00 2001 From: "Jingjing Jia (from Dev Box)" Date: Mon, 16 Mar 2026 09:20:42 -0700 Subject: [PATCH] Add extra_headers support to OpenAIModelConfiguration Add extra_headers field to OpenAIModelConfiguration TypedDict and merge it into per-call extra_headers in construct_prompty_model_config. This enables prompty-based evaluators to send custom HTTP headers (e.g., X-ModelType) required by OpenAI-compatible endpoints that need custom routing headers. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> --- .../azure-ai-evaluation/azure/ai/evaluation/_common/utils.py | 3 +++ .../azure/ai/evaluation/_model_configurations.py | 1 + 2 files changed, 4 insertions(+) diff --git a/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_common/utils.py b/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_common/utils.py index 3a5b999d0f81..993f6aa5e686 100644 --- a/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_common/utils.py +++ b/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_common/utils.py @@ -212,6 +212,9 @@ def construct_prompty_model_config( if _is_aoi_model_config(model_config) and user_agent: prompty_model_config["parameters"]["extra_headers"].update({"x-ms-useragent": user_agent}) + if "extra_headers" in model_config: + prompty_model_config["parameters"]["extra_headers"].update(model_config["extra_headers"]) + return prompty_model_config diff --git a/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_model_configurations.py b/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_model_configurations.py index 916dec2984e4..6e4efb84206b 100644 --- a/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_model_configurations.py +++ b/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_model_configurations.py @@ -78,6 +78,7 @@ class OpenAIModelConfiguration(TypedDict): model: str base_url: NotRequired[str] organization: NotRequired[str] + extra_headers: NotRequired[Dict[str, str]] class AzureAIProject(TypedDict):