azure ai usage tracing

Author: linglingye001

src/appConfigurationImpl.ts

@@ -43,7 +43,7 @@ import {
     CONDITIONS_KEY_NAME,
     CLIENT_FILTERS_KEY_NAME
 } from "./featureManagement/constants.js";
-import { FM_PACKAGE_NAME, AI_MIME_PROFILE, AI_CHAT_COMPLETION_MIME_PROFILE } from "./requestTracing/constants.js";
+import { FM_PACKAGE_NAME, AI_MIME_PROFILE, AI_CHAT_COMPLETION_MIME_PROFILE, AZURE_AI_PACKAGE_NAMES } from "./requestTracing/constants.js";
 import { parseContentType, isJsonContentType, isFeatureFlagContentType, isSecretReferenceContentType } from "./common/contentType.js";
 import { AzureKeyVaultKeyValueAdapter } from "./keyvault/keyVaultKeyValueAdapter.js";
 import { RefreshTimer } from "./refresh/refreshTimer.js";
@@ -86,6 +86,7 @@ export class AzureAppConfigurationImpl implements AzureAppConfiguration {
     #fmVersion: string | undefined;
     #aiConfigurationTracing: AIConfigurationTracingOptions | undefined;
     #useSnapshotReference: boolean = false;
+    #useAzureAI: boolean = false;
 
     // Refresh
     #refreshInProgress: boolean = false;
@@ -218,7 +219,8 @@ export class AzureAppConfigurationImpl implements AzureAppConfiguration {
             featureFlagTracing: this.#featureFlagTracing,
             fmVersion: this.#fmVersion,
             aiConfigurationTracing: this.#aiConfigurationTracing,
-            useSnapshotReference: this.#useSnapshotReference
+            useSnapshotReference: this.#useSnapshotReference,
+            useAzureAI: this.#useAzureAI
         };
     }
 
@@ -384,6 +386,7 @@ export class AzureAppConfigurationImpl implements AzureAppConfiguration {
     async #initializeWithRetryPolicy(abortSignal: AbortSignal): Promise<void> {
         if (!this.#isInitialLoadCompleted) {
             await this.#inspectFmPackage();
+            await this.#inspectAzureAIPackages();
             const startTimestamp = Date.now();
             let postAttempts = 0;
             do { // at least try to load once
@@ -437,6 +440,23 @@ export class AzureAppConfigurationImpl implements AzureAppConfiguration {
         }
     }
 
+    /**
+     * Inspects whether Azure AI packages are installed.
+     */
+    async #inspectAzureAIPackages() {
+        if (this.#requestTracingEnabled && !this.#useAzureAI) {
+            for (const packageName of AZURE_AI_PACKAGE_NAMES) {
+                try {
+                    await import(/* @vite-ignore */packageName);
+                    this.#useAzureAI = true;
+                    break; // Found one package, no need to check others
+                } catch {
+                    // Package not installed, continue checking
+                }
+            }
+        }
+    }
+
     async #refreshTasks(): Promise<void> {
         const refreshTasks: Promise<boolean>[] = [];
         if (this.#refreshEnabled || this.#secretRefreshEnabled) {

src/requestTracing/constants.ts

@@ -56,6 +56,7 @@ export const SNAPSHOT_REFERENCE_TAG = "SnapshotRef";
 // Compact feature tags
 export const FEATURES_KEY = "Features";
 export const LOAD_BALANCE_CONFIGURED_TAG = "LB";
+export const AZURE_AI_SDK_TAG = "AzureAI";
 
 // Feature management package
 export const FM_PACKAGE_NAME = "@microsoft/feature-management";
@@ -79,4 +80,7 @@ export const AI_CHAT_COMPLETION_CONFIGURATION_TAG = "AICC";
 export const AI_MIME_PROFILE = "https://azconfig.io/mime-profiles/ai";
 export const AI_CHAT_COMPLETION_MIME_PROFILE = "https://azconfig.io/mime-profiles/ai/chat-completion";
 
+// Azure AI SDK tracing
+export const AZURE_AI_PACKAGE_NAMES = ["@azure/ai-agents", "@azure/ai-projects", "@azure/openai", "@azure-rest/ai-inference"];
+
 export const DELIMITER = "+";

src/requestTracing/utils.ts

@@ -42,7 +42,8 @@ import {
     DELIMITER,
     AI_CONFIGURATION_TAG,
     AI_CHAT_COMPLETION_CONFIGURATION_TAG,
-    SNAPSHOT_REFERENCE_TAG
+    SNAPSHOT_REFERENCE_TAG,
+    AZURE_AI_SDK_TAG
 } from "./constants.js";
 
 export interface RequestTracingOptions {
@@ -55,6 +56,7 @@ export interface RequestTracingOptions {
     fmVersion: string | undefined;
     aiConfigurationTracing: AIConfigurationTracingOptions | undefined;
     useSnapshotReference: boolean;
+    useAzureAI: boolean;
 }
 
 // Utils
@@ -200,6 +202,9 @@ function createFeaturesString(requestTracingOptions: RequestTracingOptions): str
     if (requestTracingOptions.useSnapshotReference) {
         tags.push(SNAPSHOT_REFERENCE_TAG);
     }
+    if (requestTracingOptions.useAzureAI) {
+        tags.push(AZURE_AI_SDK_TAG);
+    }
     return tags.join(DELIMITER);
 }