diff --git a/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_common/rai_service.py b/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_common/rai_service.py index dab41a6fe050..b0ec4e340efe 100644 --- a/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_common/rai_service.py +++ b/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_common/rai_service.py @@ -290,7 +290,7 @@ async def submit_request_onedp( payload = generate_payload(normalized_user_text, metric, annotation_task=annotation_task) headers = get_common_headers(token, evaluator_name) if scan_session_id: - headers["client_request_id"] = scan_session_id + headers["x-ms-client-request-id"] = scan_session_id response = client.evaluations.submit_annotation(payload, headers=headers) result = json.loads(response) operation_id = result["location"].split("/")[-1] diff --git a/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/simulator/_model_tools/_generated_rai_client.py b/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/simulator/_model_tools/_generated_rai_client.py index 33cf909219df..6e6068de900f 100644 --- a/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/simulator/_model_tools/_generated_rai_client.py +++ b/sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/simulator/_model_tools/_generated_rai_client.py @@ -124,7 +124,7 @@ async def get_attack_objectives( risk_category=risk_category, lang="en", strategy=strategy, - headers={"client_request_id": scan_session_id}, + headers={"x-ms-client-request-id": scan_session_id}, ) return response @@ -146,7 +146,7 @@ async def get_jailbreak_prefixes(self, scan_session_id: Optional[str] = None) -> try: # Send the request using the autogenerated client response = self._client.get_jail_break_dataset_with_type( - type="upia", headers={"client_request_id": scan_session_id} + type="upia", headers={"x-ms-client-request-id": scan_session_id} ) if isinstance(response, list): return response