track AI usage

Author: linglingye001

src/Microsoft.Extensions.Configuration.AzureAppConfiguration/AzureAppConfigurationProvider.cs

@@ -1396,6 +1396,15 @@ private void EnsureAssemblyInspected()
 
                     _requestTracingOptions.AspireComponentVersion = TracingUtils.GetAssemblyVersion(RequestTracingConstants.AspireComponentAssemblyName);
 
+                    foreach (string assemblyName in RequestTracingConstants.AzureAIAssemblyNames)
+                    {
+                        if (TracingUtils.GetAssemblyVersion(assemblyName) != null)
+                        {
+                            _requestTracingOptions.IsAzureAIProjects = true;
+                            break;
+                        }
+                    }
+
                     if (TracingUtils.GetAssemblyVersion(RequestTracingConstants.SignalRAssemblyName) != null)
                     {
                         _requestTracingOptions.IsSignalRUsed = true;

src/Microsoft.Extensions.Configuration.AzureAppConfiguration/Constants/RequestTracingConstants.cs

@@ -33,6 +33,7 @@ internal class RequestTracingConstants
         public const string LoadBalancingEnabledTag = "LB";
         public const string AIConfigurationTag = "AI";
         public const string AIChatCompletionConfigurationTag = "AICC";
+        public const string AIProjectsTag = "AzureAIProjects";
         public const string SnapshotReferenceTag = "SnapshotRef";
         public const string SignalRUsedTag = "SignalR";
         public const string FailoverRequestTag = "Failover";
@@ -57,6 +58,14 @@ internal class RequestTracingConstants
         public const string AspireComponentAssemblyName = "Aspire.Microsoft.Extensions.Configuration.AzureAppConfiguration";
         public const string SignalRAssemblyName = "Microsoft.AspNetCore.SignalR";
 
+        public static readonly string[] AzureAIAssemblyNames = new string[]
+        {
+            "Azure.AI.Projects",
+            "Azure.AI.Inference",
+            "Azure.AI.OpenAI",
+            "Azure.AI.Projects.OpenAI"
+        };
+
         public const string Delimiter = "+";
 
         public const string AIMimeProfile = "https://azconfig.io/mime-profiles/ai";

src/Microsoft.Extensions.Configuration.AzureAppConfiguration/RequestTracingOptions.cs

@@ -56,6 +56,11 @@ internal class RequestTracingOptions
         /// </summary>
         public string AspireComponentVersion { get; set; }
 
+        /// <summary>
+        /// Flag to indicate whether Azure AI SDK is present in the application.
+        /// </summary>
+        public bool IsAzureAIProjects { get; set; } = false;
+
         /// <summary>
         /// Flag to indicate whether Microsoft.AspNetCore.SignalR assembly is present in the application.
         /// </summary>

src/Microsoft.Extensions.Configuration.AzureAppConfiguration/TracingUtils.cs

@@ -211,6 +211,11 @@ private static string CreateCorrelationContextHeader(RequestType requestType, Re
                 correlationContextTags.Add(RequestTracingConstants.PushRefreshTag);
             }
 
+            if (requestTracingOptions.IsAzureAIProjects)
+            {
+                correlationContextTags.Add(RequestTracingConstants.AIProjectsTag);
+            }
+
             var sb = new StringBuilder();
 
             foreach (KeyValuePair<string, string> kvp in correlationContextKeyValues)