|
| 1 | +#!/bin/bash |
| 2 | + |
| 3 | +# List of Azure regions to check for quota (update as needed) |
| 4 | +REGIONS=("eastus" "westus" "northcentralus" "uksouth" "swedencentral") |
| 5 | + |
| 6 | +SUBSCRIPTION_ID="${AZURE_SUBSCRIPTION_ID}" |
| 7 | +GPT_MIN_CAPACITY="${GPT_MIN_CAPACITY}" |
| 8 | +TEXT_EMBEDDING_MIN_CAPACITY="${TEXT_EMBEDDING_MIN_CAPACITY}" |
| 9 | +AZURE_CLIENT_ID="${AZURE_CLIENT_ID}" |
| 10 | +AZURE_TENANT_ID="${AZURE_TENANT_ID}" |
| 11 | +AZURE_CLIENT_SECRET="${AZURE_CLIENT_SECRET}" |
| 12 | + |
| 13 | +# Authenticate using Managed Identity |
| 14 | +echo "Authentication using Managed Identity..." |
| 15 | +if ! az login --service-principal -u "$AZURE_CLIENT_ID" -p "$AZURE_CLIENT_SECRET" --tenant "$AZURE_TENANT_ID"; then |
| 16 | + echo "❌ Error: Failed to login using Managed Identity." |
| 17 | + exit 1 |
| 18 | +fi |
| 19 | + |
| 20 | +echo "🔄 Validating required environment variables..." |
| 21 | +if [[ -z "$SUBSCRIPTION_ID" || -z "$GPT_MIN_CAPACITY" || -z "$TEXT_EMBEDDING_MIN_CAPACITY" ]]; then |
| 22 | + echo "❌ ERROR: Missing required environment variables." |
| 23 | + exit 1 |
| 24 | +fi |
| 25 | + |
| 26 | +echo "🔄 Setting Azure subscription..." |
| 27 | +if ! az account set --subscription "$SUBSCRIPTION_ID"; then |
| 28 | + echo "❌ ERROR: Invalid subscription ID or insufficient permissions." |
| 29 | + exit 1 |
| 30 | +fi |
| 31 | +echo "✅ Azure subscription set successfully." |
| 32 | + |
| 33 | +# Define models and their minimum required capacities |
| 34 | +declare -A MIN_CAPACITY=( |
| 35 | + ["OpenAI.Standard.gpt-4o"]=$GPT_MIN_CAPACITY |
| 36 | + ["OpenAI.Standard.text-embedding-ada-002"]=$TEXT_EMBEDDING_MIN_CAPACITY |
| 37 | +) |
| 38 | + |
| 39 | +VALID_REGION="" |
| 40 | +for REGION in "${REGIONS[@]}"; do |
| 41 | + echo "----------------------------------------" |
| 42 | + echo "🔍 Checking region: $REGION" |
| 43 | + |
| 44 | + QUOTA_INFO=$(az cognitiveservices usage list --location "$REGION" --output json) |
| 45 | + if [ -z "$QUOTA_INFO" ]; then |
| 46 | + echo "⚠️ WARNING: Failed to retrieve quota for region $REGION. Skipping." |
| 47 | + continue |
| 48 | + fi |
| 49 | + |
| 50 | + INSUFFICIENT_QUOTA=false |
| 51 | + for MODEL in "${!MIN_CAPACITY[@]}"; do |
| 52 | + MODEL_INFO=$(echo "$QUOTA_INFO" | awk -v model="\"value\": \"$MODEL\"" ' |
| 53 | + BEGIN { RS="},"; FS="," } |
| 54 | + $0 ~ model { print $0 } |
| 55 | + ') |
| 56 | + |
| 57 | + if [ -z "$MODEL_INFO" ]; then |
| 58 | + echo "⚠️ WARNING: No quota information found for model: $MODEL in $REGION. Skipping." |
| 59 | + continue |
| 60 | + fi |
| 61 | + |
| 62 | + CURRENT_VALUE=$(echo "$MODEL_INFO" | awk -F': ' '/"currentValue"/ {print $2}' | tr -d ',' | tr -d ' ') |
| 63 | + LIMIT=$(echo "$MODEL_INFO" | awk -F': ' '/"limit"/ {print $2}' | tr -d ',' | tr -d ' ') |
| 64 | + |
| 65 | + CURRENT_VALUE=${CURRENT_VALUE:-0} |
| 66 | + LIMIT=${LIMIT:-0} |
| 67 | + |
| 68 | + CURRENT_VALUE=$(echo "$CURRENT_VALUE" | cut -d'.' -f1) |
| 69 | + LIMIT=$(echo "$LIMIT" | cut -d'.' -f1) |
| 70 | + |
| 71 | + AVAILABLE=$((LIMIT - CURRENT_VALUE)) |
| 72 | + |
| 73 | + echo "✅ Model: $MODEL | Used: $CURRENT_VALUE | Limit: $LIMIT | Available: $AVAILABLE" |
| 74 | + |
| 75 | + if [ "$AVAILABLE" -lt "${MIN_CAPACITY[$MODEL]}" ]; then |
| 76 | + echo "❌ ERROR: $MODEL in $REGION has insufficient quota." |
| 77 | + INSUFFICIENT_QUOTA=true |
| 78 | + break |
| 79 | + fi |
| 80 | + done |
| 81 | + |
| 82 | + if [ "$INSUFFICIENT_QUOTA" = false ]; then |
| 83 | + VALID_REGION="$REGION" |
| 84 | + break |
| 85 | + fi |
| 86 | + |
| 87 | +done |
| 88 | + |
| 89 | +if [ -z "$VALID_REGION" ]; then |
| 90 | + echo "❌ No region with sufficient quota found. Blocking deployment." |
| 91 | + echo "QUOTA_FAILED=true" >> "$GITHUB_ENV" |
| 92 | + exit 0 |
| 93 | +else |
| 94 | + echo "✅ Suggested Region: $VALID_REGION" |
| 95 | + echo "VALID_REGION=$VALID_REGION" >> "$GITHUB_ENV" |
| 96 | + exit 0 |
| 97 | +fi |
0 commit comments