Merge pull request #94 from Priyanka-Microsoft/update-quota-check-script

chore: update quota check script

Author: Roopan-Microsoft

docs/quota_check.md

@@ -10,11 +10,11 @@ az login
 
 ## 📌 Default Models & Capacities:
 ```
-gpt-4o:30, gpt-4o-mini:30, gpt-4:30, text-embedding-ada-002:80
+gpt-4o:150, gpt-4o-mini:150, gpt-4:150, text-embedding-3-small:100
 ```
 ## 📌 Default Regions:
 ```
-eastus, uksouth, eastus2, northcentralus, swedencentral, westus, westus2, southcentralus, canadacentral
+eastus, uksouth, eastus2, northcentralus, swedencentral, westus, westus2, southcentralus, canadacentral, australiaeast, japaneast, norwayeast
 ```
 ## Usage Scenarios:
 - No parameters passed → Default models and capacities will be checked in default regions.
@@ -36,19 +36,23 @@ eastus, uksouth, eastus2, northcentralus, swedencentral, westus, westus2, southc
    ```
 ✔️ Check specific model(s) in default regions:
   ```
-  ./quota_check.sh --models gpt-4o:30,text-embedding-ada-002:80
+  ./quota_check.sh --models gpt-4o:150,text-embedding-3-small:100
   ```
 ✔️ Check default models in specific region(s):
   ```
 ./quota_check.sh --regions eastus,westus
   ```
 ✔️ Passing Both models and regions:  
   ```
-  ./quota_check.sh --models gpt-4o:30 --regions eastus,westus2
+  ./quota_check.sh --models gpt-4o:150 --regions eastus,westus2
   ```
 ✔️ All parameters combined:
   ```
- ./quota_check.sh --models gpt-4:30,text-embedding-ada-002:80 --regions eastus,westus --verbose
+ ./quota_check.sh --models gpt-4:150,text-embedding-3-small:100 --regions eastus,westus --verbose
+  ```
+✔️ Multiple models with single region:
+  ```
+ ./quota_check.sh --models gpt-4:150,text-embedding-3-small:100 --regions eastus2 --verbose
   ```
 
 ## **Sample Output**

scripts/quota_check.sh

@@ -47,7 +47,7 @@ log_verbose() {
 }
 
 # Default Models and Capacities (Comma-separated in "model:capacity" format)
-DEFAULT_MODEL_CAPACITY="gpt-4o:30,gpt-4o-mini:30,gpt-4:30,text-embedding-ada-002:80"
+DEFAULT_MODEL_CAPACITY="gpt-4o:150,gpt-4o-mini:150,gpt-4:150,text-embedding-3-small:100"
 
 # Convert the comma-separated string into an array
 IFS=',' read -r -a MODEL_CAPACITY_PAIRS <<< "$DEFAULT_MODEL_CAPACITY"
@@ -93,7 +93,7 @@ az account set --subscription "$AZURE_SUBSCRIPTION_ID"
 echo "🎯 Active Subscription: $(az account show --query '[name, id]' --output tsv)"
 
 # Default Regions to check (Comma-separated, now configurable)
-DEFAULT_REGIONS="eastus,uksouth,eastus2,northcentralus,swedencentral,westus,westus2,southcentralus,canadacentral"
+DEFAULT_REGIONS="eastus,uksouth,eastus2,australiaeast,norwayeast,japaneast,northcentralus,swedencentral,westus,westus2,southcentralus,canadacentral"
 IFS=',' read -r -a DEFAULT_REGION_ARRAY <<< "$DEFAULT_REGIONS"
 
 # Read parameters (if any)
@@ -165,7 +165,7 @@ for REGION in "${REGIONS[@]}"; do
         FOUND=false
         INSUFFICIENT_QUOTA=false
 
-        if [ "$MODEL_NAME" = "text-embedding-ada-002" ]; then
+        if [ "$MODEL_NAME" = "text-embedding-3-small" ]; then
             MODEL_TYPES=("openai.standard.$MODEL_NAME")
         else
             MODEL_TYPES=("openai.standard.$MODEL_NAME" "openai.globalstandard.$MODEL_NAME")
@@ -203,7 +203,7 @@ for REGION in "${REGIONS[@]}"; do
 
                 if [ "$AVAILABLE" -ge "$REQUIRED_CAPACITY" ]; then
                     FOUND=true
-                    if [ "$MODEL_NAME" = "text-embedding-ada-002" ]; then
+                    if [ "$MODEL_NAME" = "text-embedding-3-small" ]; then
                         TEXT_EMBEDDING_AVAILABLE=true
                     fi
                     AT_LEAST_ONE_MODEL_AVAILABLE=true