Quota validation - added pwsh versions to make deploying from windows easier

Author: Seth

azure.yaml

@@ -11,7 +11,13 @@ hooks:
       interactive: true
       continueOnError: false
   preprovision:
-    shell: sh
-    run: ./scripts/validate_model_deployment_quotas.sh --subscription $AZURE_SUBSCRIPTION_ID --location $AZURE_LOCATION --models-parameter "aiModelDeployments"
-    interactive: false
-    continueOnError: false
+    posix:
+      shell: sh
+      run: ./scripts/validate_model_deployment_quotas.sh --subscription $AZURE_SUBSCRIPTION_ID --location $AZURE_LOCATION --models-parameter "aiModelDeployments"
+      interactive: false
+      continueOnError: false
+    windows:
+      shell: pwsh
+      run: ./scripts/validate_model_deployment_quotas.ps1 -Subscription $env:AZURE_SUBSCRIPTION_ID -Location $env:AZURE_LOCATION -ModelsParameter "aiModelDeployments"
+      interactive: false
+      continueOnError: false

scripts/validate_model_deployment_quotas.ps1

@@ -0,0 +1,72 @@
+param (
+    [string]$SubscriptionId,
+    [string]$Location,
+    [string]$ModelsParameter
+)
+
+# Verify all required parameters are provided
+$MissingParams = @()
+
+if (-not $SubscriptionId) {
+    $MissingParams += "subscription"
+}
+
+if (-not $Location) {
+    $MissingParams += "location"
+}
+
+if (-not $ModelsParameter) {
+    $MissingParams += "models-parameter"
+}
+
+if ($MissingParams.Count -gt 0) {
+    Write-Error "❌ ERROR: Missing required parameters: $($MissingParams -join ', ')"
+    Write-Host "Usage: .\validate_model_deployment_quotas.ps1 -SubscriptionId <SUBSCRIPTION_ID> -Location <LOCATION> -ModelsParameter <MODELS_PARAMETER>"
+    exit 1
+}
+
+$JsonContent = Get-Content -Path "./infra/main.parameters.json" -Raw | ConvertFrom-Json
+
+if (-not $JsonContent) {
+    Write-Error "❌ ERROR: Failed to parse main.parameters.json. Ensure the JSON file is valid."
+    exit 1
+}
+
+$aiModelDeployments = $JsonContent.parameters.$ModelsParameter.value
+
+if (-not $aiModelDeployments -or -not ($aiModelDeployments -is [System.Collections.IEnumerable])) {
+    Write-Error "❌ ERROR: The specified property $ModelsParameter does not exist or is not an array."
+    exit 1
+}
+
+# Set the Azure subscription
+az account set --subscription $SubscriptionId
+Write-Host "🎯 Active Subscription: $(az account show --query '[name, id]' --output tsv)"
+
+$QuotaAvailable = $true
+
+# Iterate over each deployment in the aiModelDeployments array
+foreach ($deployment in $aiModelDeployments) {
+    $name = $deployment.name
+    $model = $deployment.model.name
+    $type = $deployment.sku.name
+    $capacity = $deployment.sku.capacity
+
+    # Call the validate_model_quota.ps1 script
+    Write-Host "🔍 Validating model deployment: $name ..."
+    & .\scripts\validate_model_quota.ps1 -Location $Location -Model $model -Capacity $capacity -DeploymentType $type
+
+    # Check if the script failed
+    if ($LASTEXITCODE -ne 0) {
+        Write-Error "❌ ERROR: Quota validation failed for model deployment: $name"
+        $QuotaAvailable = $false
+    }
+}
+
+if (-not $QuotaAvailable) {
+    Write-Error "❌ ERROR: One or more model deployments failed validation."
+    exit 1
+} else {
+    Write-Host "✅ All model deployments passed quota validation successfully."
+    exit 0
+}

scripts/validate_model_quota.ps1

@@ -0,0 +1,66 @@
+param (
+    [string]$Location,
+    [string]$Model,
+    [string]$DeploymentType = "Standard",
+    [int]$Capacity
+)
+
+# Verify all required parameters are provided
+$MissingParams = @()
+
+if (-not $Location) {
+    $MissingParams += "location"
+}
+
+if (-not $Model) {
+    $MissingParams += "model"
+}
+
+if (-not $Capacity) {
+    $MissingParams += "capacity"
+}
+
+if (-not $DeploymentType) {
+    $MissingParams += "deployment-type"
+}
+
+if ($MissingParams.Count -gt 0) {
+    Write-Error "❌ ERROR: Missing required parameters: $($MissingParams -join ', ')"
+    Write-Host "Usage: .\validate_model_quota.ps1 -Location <LOCATION> -Model <MODEL> -Capacity <CAPACITY> [-DeploymentType <DEPLOYMENT_TYPE>]"
+    exit 1
+}
+
+if ($DeploymentType -ne "Standard" -and $DeploymentType -ne "GlobalStandard") {
+    Write-Error "❌ ERROR: Invalid deployment type: $DeploymentType. Allowed values are 'Standard' or 'GlobalStandard'."
+    exit 1
+}
+
+$ModelType = "OpenAI.$DeploymentType.$Model"
+
+Write-Host "🔍 Checking quota for $ModelType in $Location ..."
+
+# Get quota information using Azure CLI
+$ModelInfo = az cognitiveservices usage list --location $Location --query "[?name.value=='$ModelType']" --output json | ConvertFrom-Json
+
+if (-not $ModelInfo) {
+    Write-Error "❌ ERROR: No quota information found for model: $Model in location: $Location for model type: $ModelType."
+    exit 1
+}
+
+if ($ModelInfo) {
+    $CurrentValue = ($ModelInfo | Where-Object { $_.name.value -eq $ModelType }).currentValue
+    $Limit = ($ModelInfo | Where-Object { $_.name.value -eq $ModelType }).limit
+
+    $CurrentValue = [int]($CurrentValue -replace '\.0+$', '') # Remove decimals
+    $Limit = [int]($Limit -replace '\.0+$', '') # Remove decimals
+
+    $Available = $Limit - $CurrentValue
+    Write-Host "✅ Model available - Model: $ModelType | Used: $CurrentValue | Limit: $Limit | Available: $Available"
+
+    if ($Available -lt $Capacity) {
+        Write-Error "❌ ERROR: Insufficient quota for model: $Model in location: $Location. Available: $Available, Requested: $Capacity."
+        exit 1
+    } else {
+        Write-Host "✅ Sufficient quota for model: $Model in location: $Location. Available: $Available, Requested: $Capacity."
+    }
+}