Update upload paths to use custom user agent for AI Foundry integration

Co-authored-by: singankit <[email protected]>

Author: Copilot

sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_evaluate/_utils.py

@@ -16,6 +16,8 @@
 from tqdm import tqdm
 
 from azure.ai.evaluation._legacy._adapters.entities import Run
+from azure.ai.evaluation._context import get_current_user_agent
+from azure.ai.evaluation._user_agent import USER_AGENT
 
 from azure.ai.evaluation._constants import (
     DEFAULT_EVALUATION_RESULTS_FILE_NAME,
@@ -146,9 +148,20 @@ def _log_metrics_and_instance_results_onedp(
     credentials = AzureMLTokenManager(
         TokenScope.COGNITIVE_SERVICES_MANAGEMENT.value, LOGGER, credential=kwargs.get("credential")
     )
+    
+    # Use custom user agent from context if available
+    custom_user_agent = get_current_user_agent()
+    client_kwargs = {}
+    if custom_user_agent:
+        # Create a user agent policy with the custom user agent
+        from azure.core.pipeline.policies import UserAgentPolicy
+        user_agent = f"{USER_AGENT} {custom_user_agent}"
+        client_kwargs["user_agent_policy"] = UserAgentPolicy(base_user_agent=user_agent)
+    
     client = EvaluationServiceOneDPClient(
         endpoint=project_url,
-        credential=credentials
+        credential=credentials,
+        **client_kwargs
     )
 
     # Massaging before artifacts are put on disk

sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_http_utils.py

@@ -8,6 +8,7 @@
 from typing_extensions import Self, Unpack
 
 from azure.ai.evaluation._user_agent import USER_AGENT
+from azure.ai.evaluation._context import get_current_user_agent
 from azure.core.configuration import Configuration
 from azure.core.pipeline import AsyncPipeline, Pipeline
 from azure.core.pipeline.policies import (
@@ -454,7 +455,14 @@ def get_http_client(**kwargs: Any) -> HttpPipeline:
     :returns: An HttpPipeline with a set of applied policies:
     :rtype: HttpPipeline
     """
-    kwargs.setdefault("user_agent_policy", UserAgentPolicy(base_user_agent=USER_AGENT))
+    # Use custom user agent from context if available, otherwise use default
+    custom_user_agent = get_current_user_agent()
+    if custom_user_agent:
+        user_agent = f"{USER_AGENT} {custom_user_agent}"
+    else:
+        user_agent = USER_AGENT
+    
+    kwargs.setdefault("user_agent_policy", UserAgentPolicy(base_user_agent=user_agent))
     return HttpPipeline(**kwargs)
 
 
@@ -464,5 +472,12 @@ def get_async_http_client(**kwargs: Any) -> AsyncHttpPipeline:
     :returns: An AsyncHttpPipeline with a set of applied policies:
     :rtype: AsyncHttpPipeline
     """
-    kwargs.setdefault("user_agent_policy", UserAgentPolicy(base_user_agent=USER_AGENT))
+    # Use custom user agent from context if available, otherwise use default
+    custom_user_agent = get_current_user_agent()
+    if custom_user_agent:
+        user_agent = f"{USER_AGENT} {custom_user_agent}"
+    else:
+        user_agent = USER_AGENT
+    
+    kwargs.setdefault("user_agent_policy", UserAgentPolicy(base_user_agent=user_agent))
     return AsyncHttpPipeline(**kwargs)