1<#
2.SYNOPSIS
3    Discovers available capacity for an Azure OpenAI model across all regions,
4    cross-references with existing projects and subscription quota, and outputs a ranked table.
5.PARAMETER ModelName
6    The model name (e.g., "gpt-4o", "o3-mini")
7.PARAMETER ModelVersion
8    The model version (e.g., "2025-01-31")
9.PARAMETER MinCapacity
10    Minimum required capacity in K TPM units (default: 0, shows all)
11.EXAMPLE
12    .\discover_and_rank.ps1 -ModelName o3-mini -ModelVersion 2025-01-31 -MinCapacity 200
13#>
14param(
15    [Parameter(Mandatory)][string]$ModelName,
16    [Parameter(Mandatory)][string]$ModelVersion,
17    [int]$MinCapacity = 0
18)
19 
20$ErrorActionPreference = "Stop"
21 
22$subId = az account show --query id -o tsv
23 
24# Query model capacity across all regions
25$capRaw = az rest --method GET `
26    --url "https://management.azure.com/subscriptions/$subId/providers/Microsoft.CognitiveServices/modelCapacities" `
27    --url-parameters api-version=2024-10-01 modelFormat=OpenAI modelName=$ModelName modelVersion=$ModelVersion `
28    2>$null | Out-String | ConvertFrom-Json
29 
30# Query all AI Foundry projects (AIProject kind)
31$projRaw = az rest --method GET `
32    --url "https://management.azure.com/subscriptions/$subId/providers/Microsoft.CognitiveServices/accounts" `
33    --url-parameters api-version=2024-10-01 `
34    --query "value[?kind=='AIProject'].{Name:name, Location:location}" `
35    2>$null | Out-String | ConvertFrom-Json
36 
37# Build capacity map (GlobalStandard only, pick max per region)
38$capMap = @{}
39foreach ($item in $capRaw.value) {
40    $sku = $item.properties.skuName
41    $avail = [int]$item.properties.availableCapacity
42    $region = $item.location
43    if ($sku -eq "GlobalStandard" -and $avail -gt 0) {
44        if (-not $capMap[$region] -or $avail -gt $capMap[$region]) {
45            $capMap[$region] = $avail
46        }
47    }
48}
49 
50# Build project map
51$projMap = @{}
52$projSample = @{}
53foreach ($p in $projRaw) {
54    $loc = $p.Location
55    if (-not $projMap[$loc]) { $projMap[$loc] = 0 }
56    $projMap[$loc]++
57    if (-not $projSample[$loc]) { $projSample[$loc] = $p.Name }
58}
59 
60# Check subscription quota per region
61$quotaMap = @{}
62$checkedRegions = @{}
63foreach ($region in $capMap.Keys) {
64    if ($checkedRegions[$region]) { continue }
65    $checkedRegions[$region] = $true
66    try {
67        $usageData = az cognitiveservices usage list --location $region --subscription $subId -o json 2>$null | Out-String | ConvertFrom-Json
68        $usageEntry = $usageData | Where-Object { $_.name.value -eq "OpenAI.GlobalStandard.$ModelName" }
69        if ($usageEntry) {
70            $quotaMap[$region] = [int]$usageEntry.limit - [int]$usageEntry.currentValue
71        } else {
72            $quotaMap[$region] = 0
73        }
74    } catch {
75        $quotaMap[$region] = -1  # Unable to check
76    }
77}
78 
79# Combine and rank
80$results = foreach ($region in $capMap.Keys) {
81    $avail = $capMap[$region]
82    $meets = $avail -ge $MinCapacity
83    $quota = if ($quotaMap[$region]) { $quotaMap[$region] } else { 0 }
84    $quotaDisplay = if ($quota -eq -1) { "?" } elseif ($quota -gt 0) { "${quota}K" } else { "0" }
85    $quotaOk = $quota -gt 0 -or $quota -eq -1
86    [PSCustomObject]@{
87        Region         = $region
88        AvailableTPM   = "${avail}K"
89        AvailableRaw   = $avail
90        MeetsTarget    = if ($meets) { "YES" } else { "no" }
91        Projects       = if ($projMap[$region]) { $projMap[$region] } else { 0 }
92        SampleProject  = if ($projSample[$region]) { $projSample[$region] } else { "(none)" }
93        QuotaAvailable = $quotaDisplay
94        QuotaOk        = $quotaOk
95    }
96}
97 
98$results = $results | Sort-Object @{Expression={$_.MeetsTarget -eq "YES"}; Descending=$true},
99                                   @{Expression={$_.QuotaOk}; Descending=$true},
100                                   @{Expression={$_.Projects}; Descending=$true},
101                                   @{Expression={$_.AvailableRaw}; Descending=$true}
102 
103# Output summary
104$total = ($results | Measure-Object).Count
105$matching = ($results | Where-Object { $_.MeetsTarget -eq "YES" } | Measure-Object).Count
106$withQuota = ($results | Where-Object { $_.MeetsTarget -eq "YES" -and $_.QuotaOk } | Measure-Object).Count
107$withProjects = ($results | Where-Object { $_.MeetsTarget -eq "YES" -and $_.Projects -gt 0 } | Measure-Object).Count
108 
109Write-Host "Model: $ModelName v$ModelVersion | SKU: GlobalStandard | Min Capacity: ${MinCapacity}K TPM"
110Write-Host "Regions with capacity: $total | Meets target: $matching | With quota: $withQuota | With projects: $withProjects"
111Write-Host ""
112 
113$results | Select-Object Region, AvailableTPM, MeetsTarget, QuotaAvailable, Projects, SampleProject | Format-Table -AutoSize
114