Add `az aks list-vm-skus` command to aks-preview extension by CustardTart32 · Pull Request #9698 · Azure/azure-cli-extensions

CustardTart32 · 2026-03-18T01:41:16Z

This checklist is used to make sure that common guidelines for a pull request are followed.

Related command

az aks list-vm-skus

General Guidelines

Have you run azdev style <YOUR_EXT> locally? (pip install azdev required)
Have you run python scripts/ci/test_index.py -q locally? (pip install wheel==0.30.0 required)
My extension version conforms to the Extension version schema

For new extensions:

My extension description/summary conforms to the Extension Summary Guidelines.

About Extension Publish

There is a pipeline to automatically build, upload and publish extension wheels.
Once your pull request is merged into main branch, a new pull request will be created to update src/index.json automatically.
You only need to update the version information in file setup.py and historical information in file HISTORY.rst in your PR but do not modify src/index.json.

azure-client-tools-bot-prd · 2026-03-18T01:41:20Z

⚠️Azure CLI Extensions Breaking Change Test

⚠️aks-preview

rule cmd_name rule_message suggest_message

⚠️ 1001 - CmdAdd aks list-vm-skus cmd aks list-vm-skus added

azure-client-tools-bot-prd · 2026-03-18T01:41:23Z

Hi @CustardTart32,
Please write the description of changes which can be perceived by customers into HISTORY.rst.
If you want to release a new extension version, please update the version in setup.py as well.

yonzhan · 2026-03-18T01:41:27Z

Thank you for your contribution! We will review the pull request and get back to you soon.

github-actions · 2026-03-18T01:41:30Z

The git hooks are available for azure-cli and azure-cli-extensions repos. They could help you run required checks before creating the PR.

Please sync the latest code with latest dev branch (for azure-cli) or main branch (for azure-cli-extensions).
After that please run the following commands to enable git hooks:

pip install azdev --upgrade
azdev setup -c <your azure-cli repo path> -r <your azure-cli-extensions repo path>

github-actions · 2026-03-18T01:42:28Z

CodeGen Tools Feedback Collection

Thank you for using our CodeGen tool. We value your feedback, and we would like to know how we can improve our product. Please take a few minutes to fill our codegen survey

Copilot

Pull request overview

Adds a new az aks list-vm-skus command to the aks-preview extension, backed by the AKS vmSkus ARM endpoint, with client-side filtering and table output formatting.

Changes:

Introduces aks list-vm-skus command wiring (client factory, command registration, args) and implements filtering (--size, --zone, --all).
Adds table formatter for -o table output.
Adds unit tests for SKU availability logic and command filtering, plus a scenario test.

Reviewed changes

Copilot reviewed 9 out of 10 changed files in this pull request and generated 1 comment.

Show a summary per file

File	Description
src/aks-preview/azext_aks_preview/vm_skus_util.py	Adds helper to evaluate subscription/zone availability based on SKU restrictions.
src/aks-preview/azext_aks_preview/custom.py	Implements `aks_list_vm_skus` custom command (fetch + filters).
src/aks-preview/azext_aks_preview/commands.py	Registers `aks list-vm-skus` command and hooks table transformer.
src/aks-preview/azext_aks_preview/_params.py	Defines CLI parameters for `aks list-vm-skus` (`--location`, `--size`, `--zone`, `--all`).
src/aks-preview/azext_aks_preview/_format.py	Adds `aks_list_vm_skus_table_format` for table output summarization.
src/aks-preview/azext_aks_preview/_client_factory.py	Adds `cf_vm_skus` to access `ContainerServiceClient.vm_skus`.
src/aks-preview/azext_aks_preview/tests/latest/test_vm_skus.py	Unit tests for `_aks_is_vm_sku_available`.
src/aks-preview/azext_aks_preview/tests/latest/test_custom.py	Unit tests for `aks_list_vm_skus` filtering behavior.
src/aks-preview/azext_aks_preview/tests/latest/test_aks_commands.py	Adds scenario coverage for `aks list-vm-skus` basic/filtered calls.

github-actions · 2026-03-18T01:45:56Z

Hi @CustardTart32

Release Suggestions

Module: aks-preview

Update VERSION to 19.0.0b28 in src/aks-preview/setup.py

Notes

For more info about extension versioning, please refer to Extension version schema

FumingZhang · 2026-03-18T03:26:18Z

/azp run

azure-pipelines · 2026-03-18T03:26:32Z

Azure Pipelines successfully started running 2 pipeline(s).

FumingZhang · 2026-03-18T03:31:16Z

+            options_list=["--zone", "-z"],
+            arg_type=get_three_state_flag(),


Do we need a three-state flag here? From what I understand, the list API already handles results with this filter by default. Wouldn't a single state be enough to toggle the filter? I'm concerned because we rarely use three-state flags in the aks command group, except for a few cases like the greater monitoring addon. This could affect consistency across all commands. The same applies to the --show-all option

Changed to use a store_true action.

In term of the list API itself, it will filter out VM SKU's that are restricted by region but not by AZ (this might actually be an oversight that should be fixed in the internal API implementation).

The --zone flag is different in that it filters out any VM SKUs that do not have zones associated.

FumingZhang

Queued live test to validate the change.

FumingZhang · 2026-03-18T08:51:22Z

/azp run

azure-pipelines · 2026-03-18T08:51:36Z

Azure Pipelines successfully started running 2 pipeline(s).

FumingZhang · 2026-03-19T02:55:56Z

Re-queued live test.

FumingZhang · 2026-03-19T03:42:38Z

Re-queued live test.

The test failed with an error. Need to queue the test using a sub that has the specific feature flag registered?

  assert len(skus) > 0, "expected at least one SKU in eastus"
E AssertionError: expected at least one SKU in eastus
E assert 0 > 0
E + where 0 = len([])

CustardTart32 · 2026-03-19T06:29:34Z

What subscription is the live test being run against? This seems pretty strange. I run the live tests locally in a loop and didn't get any test failures for test_aks_list_vm_skus

FumingZhang · 2026-03-19T06:40:31Z

This one 79a7390d-3a85-432d-9f6f-a11a703c8b83

FumingZhang · 2026-03-19T06:40:53Z

fixing the CI failure in #9703

CustardTart32 · 2026-03-19T07:00:41Z

Fixed the live test failure. Turns out that subscription didn't have any vm skus available in eastus. Switched over to westus2.

FumingZhang · 2026-03-19T23:38:00Z

/azp run

azure-pipelines · 2026-03-19T23:38:14Z

Azure Pipelines successfully started running 2 pipeline(s).

FumingZhang · 2026-03-20T00:51:33Z

fixing the CI failure in #9703

Please rebase from main to pick up the fix, cc @CustardTart32

The CI GH Azdev Linter / azdev-linter (pull_request_target) sets up the test environment based on your branch's HEAD, not the merge reference, so it won't automatically pick up the fix.

Implements a new CLI command that lists VM SKUs available for AKS node pools in a given location by querying the AKS-specific vmSkus endpoint, mirroring the shape and filtering behavior of 'az vm list-skus'. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

…_aks_list_vm_skus Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

FumingZhang · 2026-03-20T03:20:55Z

/azp run

azure-pipelines · 2026-03-20T03:21:10Z

Azure Pipelines successfully started running 2 pipeline(s).

yanzhudd · 2026-03-23T00:02:06Z

/azp run

azure-pipelines · 2026-03-23T00:02:23Z

Azure Pipelines successfully started running 2 pipeline(s).

CustardTart32 requested review from FumingZhang and andyliuliming as code owners March 18, 2026 01:41

Copilot AI review requested due to automatic review settings March 18, 2026 01:41

azure-client-tools-bot-prd Bot added this to the April 2026 (2026-04-07) milestone Mar 18, 2026

microsoft-github-policy-service Bot requested review from yanzhudd, yonzhan and zhoxing-ms March 18, 2026 01:41

microsoft-github-policy-service Bot assigned yanzhudd and zhoxing-ms Mar 18, 2026

microsoft-github-policy-service Bot added Auto-Assign Auto assign by bot AKS labels Mar 18, 2026

Copilot started reviewing on behalf of CustardTart32 March 18, 2026 01:42 View session

Copilot AI reviewed Mar 18, 2026

View reviewed changes

Comment thread src/aks-preview/azext_aks_preview/vm_skus_util.py Outdated

FumingZhang reviewed Mar 18, 2026

View reviewed changes

CustardTart32 force-pushed the czhong/aks-list-vm-skus branch 3 times, most recently from e7ba3dc to 6b81746 Compare March 18, 2026 07:13

yonzhan unassigned zhoxing-ms Mar 18, 2026

CustardTart32 force-pushed the czhong/aks-list-vm-skus branch from 6b81746 to 0890873 Compare March 19, 2026 06:30

CustardTart32 and others added 7 commits March 20, 2026 12:01

test: add tests for aks list-vm-skus command

570c45f

chore: update HISTORY.rst

3f27c5a

refactor: apply PR comments

2171db4

fix: use store_true action instead of 3 state flag

b5ddf1d

fix: tidy up command description

1c7e4ac

test: update recording and increase AllowLargeResponse limit for test…

688f9f3

…_aks_list_vm_skus Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

CustardTart32 force-pushed the czhong/aks-list-vm-skus branch from 2c8b11f to 688f9f3 Compare March 20, 2026 01:06

FumingZhang approved these changes Mar 22, 2026

View reviewed changes

yanzhudd approved these changes Mar 23, 2026

View reviewed changes

yanzhudd merged commit 7b437ef into Azure:main Mar 23, 2026
24 checks passed

		options_list=["--zone", "-z"],
		arg_type=get_three_state_flag(),

Conversation

CustardTart32 commented Mar 18, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Related command

General Guidelines

About Extension Publish

Uh oh!

azure-client-tools-bot-prd Bot commented Mar 18, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

azure-client-tools-bot-prd Bot commented Mar 18, 2026

Uh oh!

yonzhan commented Mar 18, 2026

Uh oh!

github-actions Bot commented Mar 18, 2026

Uh oh!

github-actions Bot commented Mar 18, 2026

CodeGen Tools Feedback Collection

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

Uh oh!

github-actions Bot commented Mar 18, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Release Suggestions

Module: aks-preview

Notes

Uh oh!

FumingZhang commented Mar 18, 2026

Uh oh!

azure-pipelines Bot commented Mar 18, 2026

Uh oh!

FumingZhang Mar 18, 2026

Choose a reason for hiding this comment

Uh oh!

CustardTart32 Mar 18, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

FumingZhang left a comment

Choose a reason for hiding this comment

Uh oh!

FumingZhang commented Mar 18, 2026

Uh oh!

azure-pipelines Bot commented Mar 18, 2026

Uh oh!

FumingZhang commented Mar 19, 2026

Uh oh!

FumingZhang commented Mar 19, 2026

Uh oh!

CustardTart32 commented Mar 19, 2026

Uh oh!

FumingZhang commented Mar 19, 2026

Uh oh!

FumingZhang commented Mar 19, 2026

Uh oh!

CustardTart32 commented Mar 19, 2026

Uh oh!

FumingZhang commented Mar 19, 2026

Uh oh!

azure-pipelines Bot commented Mar 19, 2026

Uh oh!

FumingZhang commented Mar 20, 2026

Uh oh!

FumingZhang commented Mar 20, 2026

Uh oh!

azure-pipelines Bot commented Mar 20, 2026

Uh oh!

yanzhudd commented Mar 23, 2026

Uh oh!

azure-pipelines Bot commented Mar 23, 2026

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

CustardTart32 commented Mar 18, 2026 •

edited

Loading

azure-client-tools-bot-prd Bot commented Mar 18, 2026 •

edited

Loading

github-actions Bot commented Mar 18, 2026 •

edited

Loading

CustardTart32 Mar 18, 2026 •

edited

Loading