Merge pull request #1330 from b4s36t4/fix/azure-ai-entra-scope

fix: accept scope from request headers to allow azure serverless infe…

Author: VisargD

src/handlers/handlerUtils.ts

@@ -866,6 +866,7 @@ export function constructConfigFromRequestHeaders(
     azureEntraClientSecret:
       requestHeaders[`x-${POWERED_BY}-azure-entra-client-secret`],
     azureEntraTenantId: requestHeaders[`x-${POWERED_BY}-azure-entra-tenant-id`],
+    azureEntraScope: requestHeaders[`x-${POWERED_BY}-azure-entra-scope`],
   };
 
   const awsConfig = {

src/providers/azure-ai-inference/api.ts

@@ -65,10 +65,15 @@ const AzureAIInferenceAPI: ProviderAPIConfig = {
     }
 
     if (azureAuthMode === 'entra') {
-      const { azureEntraTenantId, azureEntraClientId, azureEntraClientSecret } =
-        providerOptions;
+      const {
+        azureEntraTenantId,
+        azureEntraClientId,
+        azureEntraClientSecret,
+        azureEntraScope,
+      } = providerOptions;
       if (azureEntraTenantId && azureEntraClientId && azureEntraClientSecret) {
-        const scope = 'https://cognitiveservices.azure.com/.default';
+        const scope =
+          azureEntraScope ?? 'https://cognitiveservices.azure.com/.default';
         const accessToken = await getAccessTokenFromEntraId(
           azureEntraTenantId,
           azureEntraClientId,

src/types/requestBody.ts

@@ -157,6 +157,9 @@ export interface Options {
 
   /** Cortex specific fields */
   snowflakeAccount?: string;
+
+  /** Azure entra scope */
+  azureEntraScope?: string;
 }
 
 /**