Update quota check script to list regions with available models and capacities, even if not all models are available

Author: Priyanka-Microsoft

scripts/quota_check_params.sh

@@ -1,17 +1,35 @@
 #!/bin/bash
 
-# Parameters
+# Default Models and Capacities (Comma-separated)
+DEFAULT_MODELS="gpt-4o,text-embedding-ada-002,gpt-4"
+DEFAULT_CAPACITY="30,80,30"  # Each model will be paired with a capacity in order
+
+# Convert the comma-separated strings into arrays
+IFS=',' read -r -a MODEL_NAMES <<< "$DEFAULT_MODELS"
+IFS=',' read -r -a CAPACITIES <<< "$DEFAULT_CAPACITY"
+
+# Default Regions to check (Comma-separated, now configurable)
+DEFAULT_REGIONS="eastus,uksouth,eastus2,northcentralus,swedencentral,westus,westus2,southcentralus,canadacentral"
+IFS=',' read -r -a DEFAULT_REGION_ARRAY <<< "$DEFAULT_REGIONS"  # Split into an array
+
+# Read parameters (if any)
 IFS=',' read -r -a MODEL_CAPACITY_PAIRS <<< "$1"  # Split the comma-separated model and capacity pairs into an array
 USER_REGION="$2"
 
+# If no parameters are passed, use default models and regions
 if [ ${#MODEL_CAPACITY_PAIRS[@]} -lt 1 ]; then
-    echo "❌ ERROR: At least one model and capacity pairs must be provided as arguments."
-    exit 1
+    echo "No parameters provided, using default models: ${MODEL_NAMES[*]} with respective capacities: ${CAPACITIES[*]}"
+    # Use default models and their respective capacities
+    for i in "${!MODEL_NAMES[@]}"; do
+        MODEL_CAPACITY_PAIRS+=("${MODEL_NAMES[$i]}:${CAPACITIES[$i]}")
+    done
+else
+    echo "Using provided model and capacity pairs: ${MODEL_CAPACITY_PAIRS[*]}"
 fi
 
 # Extract model names and required capacities into arrays
-declare -a MODEL_NAMES
-declare -a CAPACITIES
+declare -a FINAL_MODEL_NAMES
+declare -a FINAL_CAPACITIES
 
 for PAIR in "${MODEL_CAPACITY_PAIRS[@]}"; do
     MODEL_NAME=$(echo "$PAIR" | cut -d':' -f1)
@@ -22,11 +40,11 @@ for PAIR in "${MODEL_CAPACITY_PAIRS[@]}"; do
         exit 1
     fi
 
-    MODEL_NAMES+=("$MODEL_NAME")
-    CAPACITIES+=("$CAPACITY")
+    FINAL_MODEL_NAMES+=("$MODEL_NAME")
+    FINAL_CAPACITIES+=("$CAPACITY")
 done
 
-echo "🔄 Using Models: ${MODEL_NAMES[*]} with respective Capacities: ${CAPACITIES[*]}"
+echo "🔄 Using Models: ${FINAL_MODEL_NAMES[*]} with respective Capacities: ${FINAL_CAPACITIES[*]}"
 
 echo "🔄 Fetching available Azure subscriptions..."
 SUBSCRIPTIONS=$(az account list --query "[?state=='Enabled'].{Name:name, ID:id}" --output tsv)
@@ -58,20 +76,20 @@ fi
 
 # Set the selected subscription
 az account set --subscription "$AZURE_SUBSCRIPTION_ID"
-echo "🎯 Active Subscription: $(az account show --query '[name, id]' --output table)"
-
-# List of regions to check
-DEFAULT_REGIONS=("eastus" "uksouth" "eastus2" "northcentralus" "swedencentral" "westus" "westus2" "southcentralus" "canadacentral")
+echo "🎯 Active Subscription: $(az account show --query '[name, id]' --output tsv)"
 
-# Prioritize user-provided region if given
+# Check if the user provided a region, if not, use the default regions
 if [ -n "$USER_REGION" ]; then
-    # Ensure the user-provided region is checked first
-    REGIONS=("$USER_REGION" "${DEFAULT_REGIONS[@]}")
+    echo "🔍 User provided region: $USER_REGION"
+    IFS=',' read -r -a REGIONS <<< "$USER_REGION"  # Split into an array using comma
 else
-    REGIONS=("${DEFAULT_REGIONS[@]}")
+    echo "No region specified, using default regions: ${DEFAULT_REGION_ARRAY[*]}"
+    REGIONS=("${DEFAULT_REGION_ARRAY[@]}")
 fi
 
 echo "✅ Retrieved Azure regions. Checking availability..."
+declare -a TABLE_ROWS
+INDEX=1
 
 VALID_REGIONS=()
 for REGION in "${REGIONS[@]}"; do
@@ -85,13 +103,13 @@ for REGION in "${REGIONS[@]}"; do
         continue
     fi
 
-    # Initialize a flag to track if both models have sufficient quota in the region
-    BOTH_MODELS_AVAILABLE=true
+    # Initialize a flag to track if all models have sufficient quota in the region
+    ALL_MODELS_AVAILABLE=true
+
+    for index in "${!FINAL_MODEL_NAMES[@]}"; do
+        MODEL_NAME="${FINAL_MODEL_NAMES[$index]}"
+        REQUIRED_CAPACITY="${FINAL_CAPACITIES[$index]}"
 
-    for index in "${!MODEL_NAMES[@]}"; do
-        MODEL_NAME="${MODEL_NAMES[$index]}"
-        REQUIRED_CAPACITY="${CAPACITIES[$index]}"
-        
         echo "🔍 Checking model: $MODEL_NAME with required capacity: $REQUIRED_CAPACITY"
 
         # Extract model quota information
@@ -102,7 +120,7 @@ for REGION in "${REGIONS[@]}"; do
 
         if [ -z "$MODEL_INFO" ]; then
             echo "⚠️ WARNING: No quota information found for model: OpenAI.Standard.$MODEL_NAME in $REGION. Skipping."
-            BOTH_MODELS_AVAILABLE=false
+            ALL_MODELS_AVAILABLE=false
             break  # If any model is not available, no need to check further for this region
         fi
 
@@ -123,37 +141,30 @@ for REGION in "${REGIONS[@]}"; do
         if [ "$AVAILABLE" -lt "$REQUIRED_CAPACITY" ]; then
             echo "❌ ERROR: 'OpenAI.Standard.$MODEL_NAME' in $REGION has insufficient quota. Required: $REQUIRED_CAPACITY, Available: $AVAILABLE"
             echo "➡️  To request a quota increase, visit: https://aka.ms/oai/stuquotarequest"
-            BOTH_MODELS_AVAILABLE=false
-            break
+            ALL_MODELS_AVAILABLE=false
+        else
+            TABLE_ROWS+=("$(printf "| %-4s | %-20s | %-35s | %-10s | %-10s | %-10s |" "$INDEX" "$REGION" "$MODEL_NAME" "$LIMIT" "$CURRENT_VALUE" "$AVAILABLE")")
+
+            INDEX=$((INDEX + 1))
         fi
     done
 
-    # If both models have sufficient quota, add region to valid regions
-    if [ "$BOTH_MODELS_AVAILABLE" = true ]; then
+    # If all models have sufficient quota, add region to valid regions
+    if [ "$ALL_MODELS_AVAILABLE" = true ]; then
         echo "✅ All models have sufficient quota in $REGION."
         VALID_REGIONS+=("$REGION")
     fi
 done
 
-# Determine final result and display in table format
-if [ ${#VALID_REGIONS[@]} -eq 0 ]; then
-    echo "----------------------------------------"
-    echo "❌ No region with sufficient quota found for all models. Blocking deployment."
-    echo "----------------------------------------"
-    exit 0
-else
-    echo "----------------------------------------"
-    echo "✅ Suggested Regions with Sufficient Quota"
-    echo "----------------------------------------"
-    printf "| %-5s | %-20s |\n" "No." "Region"
-    echo "----------------------------------------"
-    
-    INDEX=1
-    for REGION in "${VALID_REGIONS[@]}"; do
-        printf "| %-5s | %-20s |\n" "$INDEX" "$REGION"
-        INDEX=$((INDEX + 1))
-    done
-    
-    echo "----------------------------------------"
-    exit 0
-fi
+# Print table header
+echo "----------------------------------------------------------------------------------------------------------"
+printf "| %-4s | %-20s | %-35s | %-10s | %-10s | %-10s |\n" "No." "Region" "Model Name" "Limit" "Used" "Available"
+echo "----------------------------------------------------------------------------------------------------------"
+
+for ROW in "${TABLE_ROWS[@]}"; do
+    echo "$ROW"
+done
+
+echo "----------------------------------------------------------------------------------------------------------"
+echo "➡️  To request a quota increase, visit: https://aka.ms/oai/stuquotarequest"
+echo "✅ Script completed."