Skip to content

Add cli commands for AI Manager - #10089

Draft
Ximeng Zhao (xmzhao0822) wants to merge 9 commits into
Azure:mainfrom
xmzhao0822:feature/aks-inference-commands
Draft

Add cli commands for AI Manager#10089
Ximeng Zhao (xmzhao0822) wants to merge 9 commits into
Azure:mainfrom
xmzhao0822:feature/aks-inference-commands

Conversation

@xmzhao0822

@xmzhao0822 Ximeng Zhao (xmzhao0822) commented Jul 8, 2026

Copy link
Copy Markdown
Contributor

Add new 'az aks inference' command group (create/delete/list/show) mapping to the Microsoft.ContainerService/aiManagers resource, plus the 'az aks inference namespace' subgroup for AIManagerNamespace child resources, targeting api-version 2026-04-02-preview.

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

Related command

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 azdev required)
  • My extension version conforms to the Extension version schema

For new extensions:

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.

@xmzhao0822
Add az aks inference commands for AI Manager
36fcd53 
Add new 'az aks inference' command group (create/delete/list/show) mapping
to the Microsoft.ContainerService/aiManagers resource, plus the
'az aks inference namespace' subgroup for AIManagerNamespace child
resources, targeting api-version 2026-04-02-preview.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
@azure-client-tools-bot-prd

azure-client-tools-bot-prd Bot commented Jul 8, 2026 •
edited
Loading

Copy link
Copy Markdown
⚠️Azure CLI Extensions Breaking Change Test
⚠️aks-preview
rule cmd_name rule_message suggest_message
⚠️ 1011 - SubgroupAdd aks inference sub group aks inference added

@azure-client-tools-bot-prd

azure-client-tools-bot-prd Bot commented Jul 8, 2026

Copy link
Copy Markdown

Hi Ximeng Zhao (@xmzhao0822),
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.

@xmzhao0822
Switch az aks inference to classic vendored-SDK approach
c696139 
Replace the AAZ-generated 'az aks inference' command files with a
hand-written classic module (vendored SDK + custom.py + params + help +
explicit command wiring) under azext_aks_preview/aks_inference/.

- AIManager (Microsoft.ContainerService/aiManagers) create/show/delete/list
- AIManagerNamespace child resource create/show/delete/list
- api-version 2026-04-02-preview
- Wired into the extension's commands.py, _params.py and _help.py

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
@yonzhan

Yong Zhang (yonzhan) commented Jul 8, 2026

Copy link
Copy Markdown
Collaborator

aks

Ximeng Zhao (xmzhao0822) and others added 3 commits August 5, 2026 20:45
@xmzhao0822
Switch az aimanager to real vendored containerserviceaimanager SDK
0e1c969 
Rename the aks_inference command group/folder to aimanager and replace the
hand-written stub SDK with the generated azure-mgmt-containerserviceaimanager
package under aimanager/vendored_sdk.

Commands: az aimanager create|update|list|show|delete and
az aimanager namespace add|update|list|show|delete.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
@xmzhao0822
Restructure az aimanager to match managed namespace format
9fb5ffc 
Register the vendored containerserviceaimanager SDK as a CustomResourceType and
resolve models via cmd.get_models. Anchor command groups on the SDK operations
classes, add a construct/update builder module (aimanager.py), and make the
custom handlers do existence checks, raw_parameters snapshots, custom headers
and sdk_no_wait - mirroring the aks namespace (managed namespace) structure.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
@xmzhao0822
Extract aimanager into standalone src/aimanager extension
ce9c804 
Move the `az aimanager` command group out of aks-preview into a new
standalone extension at src/aimanager, mirroring the src/fleet structure:
loader, _client_factory, commands, custom, _params, _help, _helpers,
_validators, constants, packaging files, and a tests scaffold. The
vendored azure-mgmt-containerserviceaimanager (typespec) SDK is moved to
azext_aimanager/vendored_sdks and registered with api_version=None.

Remove the aimanager code and wiring from aks-preview and revert its
HISTORY.rst entry.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
@azure-client-tools-agent

azure-client-tools-agent Bot commented Aug 5, 2026

Copy link
Copy Markdown

Automated sensitive-information remediation ran on this pull request.

  • Detected categories: email address
  • Replaced with typed [REDACTED:category] placeholders in: no writable PR fields
  • Could not edit: file changes on a fork branch
  • Comment/review owners notified because Agent Assist cannot edit another user's text: none

If a credential was exposed, rotate or revoke it immediately. Detected values are never copied into this comment.

✅ Keep the redaction · ❌ Dispute the redaction

GitHub only supports a fixed reaction set, so 👍 represents ✅ and 👎 represents ❌. The bot-created reactions are only poll choices.

Posted by agent-assist (autonomous bug-fix pipeline).

Ximeng Zhao (xmzhao0822) and others added 2 commits August 5, 2026 23:16
@xmzhao0822
Nest aimanager vendored SDK under versioned folder
de15bee 
Move the vendored typespec SDK into
azext_aimanager/vendored_sdks/v2026_05_02_preview to mirror fleet's
vendored_sdks/v<version> layout, add a top-level vendored_sdks/__init__.py
re-exporting the client, and point the CustomResourceType import prefix
and command operation templates at the versioned package.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
@xmzhao0822
Merge remote-tracking branch 'origin/main' into feature/aks-inference…
ef6ac1a 
…-commands
@azure-client-tools-agent

azure-client-tools-agent Bot commented Aug 5, 2026

Copy link
Copy Markdown

Automated sensitive-information remediation ran on this pull request.

  • Detected categories: email address
  • Replaced with typed [REDACTED:category] placeholders in: no writable PR fields
  • Could not edit: file changes on a fork branch
  • Comment/review owners notified because Agent Assist cannot edit another user's text: none

If a credential was exposed, rotate or revoke it immediately. Detected values are never copied into this comment.

✅ Keep the redaction · ❌ Dispute the redaction

GitHub only supports a fixed reaction set, so 👍 represents ✅ and 👎 represents ❌. The bot-created reactions are only poll choices.

Posted by agent-assist (autonomous bug-fix pipeline).

@xmzhao0822 Ximeng Zhao (xmzhao0822) changed the title Add az aks inference commands for AI Manager Add cli commands for AI Manager Aug 6, 2026
@azure-client-tools-agent

azure-client-tools-agent Bot commented Aug 6, 2026

Copy link
Copy Markdown

Automated sensitive-information remediation ran on this pull request.

  • Detected categories: email address
  • Replaced with typed [REDACTED:category] placeholders in: no writable PR fields
  • Could not edit: file changes on a fork branch
  • Comment/review owners notified because Agent Assist cannot edit another user's text: none

If a credential was exposed, rotate or revoke it immediately. Detected values are never copied into this comment.

✅ Keep the redaction · ❌ Dispute the redaction

GitHub only supports a fixed reaction set, so 👍 represents ✅ and 👎 represents ❌. The bot-created reactions are only poll choices.

Posted by agent-assist (autonomous bug-fix pipeline).

@FumingZhang

FumingZhang commented Aug 6, 2026

Copy link
Copy Markdown
Member

/azp run

@azure-pipelines

azure-pipelines Bot commented Aug 6, 2026

Copy link
Copy Markdown
Azure Pipelines:
Successfully started running 2 pipeline(s).

@azure-client-tools-agent

azure-client-tools-agent Bot commented Aug 6, 2026

Copy link
Copy Markdown

Automated sensitive-information remediation ran on this pull request.

  • Detected categories: email address
  • Replaced with typed [REDACTED:category] placeholders in: no writable PR fields
  • Could not edit: file changes on a fork branch
  • Comment/review owners notified because Agent Assist cannot edit another user's text: none

If a credential was exposed, rotate or revoke it immediately. Detected values are never copied into this comment.

✅ Keep the redaction · ❌ Dispute the redaction

GitHub only supports a fixed reaction set, so 👍 represents ✅ and 👎 represents ❌. The bot-created reactions are only poll choices.

Posted by agent-assist (autonomous bug-fix pipeline).

@a0x1ab Aditya Pujara (a0x1ab) added agent-assist On-demand trigger for the agent-assist autonomous bug-fix pipeline azure-client-tools-agent Pull request commented on or reviewed by Azure Client Tools Agent and removed agent-assist On-demand trigger for the agent-assist autonomous bug-fix pipeline labels Aug 6, 2026
@xmzhao0822
Add e2e scenario test for az aimanager
a3aac89 
Add an azure.cli.testsdk ScenarioTest covering the full aimanager and
aimanager namespace command surface (create/wait/show/list/update/delete
and namespace add/wait/show/list/update/delete), modeled on fleet's
hubful scenario test.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
@a0x1ab

Aditya Pujara (a0x1ab) commented Aug 6, 2026

Copy link
Copy Markdown
Member

/azp run

@azure-pipelines

azure-pipelines Bot commented Aug 6, 2026

Copy link
Copy Markdown
Azure Pipelines:
Successfully started running 2 pipeline(s).

@a0x1ab

Aditya Pujara (a0x1ab) commented Aug 6, 2026

Copy link
Copy Markdown
Member

/azp run

@azure-pipelines

azure-pipelines Bot commented Aug 6, 2026

Copy link
Copy Markdown
Azure Pipelines:
Successfully started running 2 pipeline(s).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@yanzhudd Julie Zhu (yanzhudd) Awaiting requested review from yanzhudd
@yonzhan Yong Zhang (yonzhan) Awaiting requested review from yonzhan
@FumingZhang FumingZhang Awaiting requested review from FumingZhang
Copilot code review Copilot Awaiting requested review from Copilot Copilot will automatically review once the pull request is marked ready for review

At least 1 approving review is required to merge this pull request.

Assignees

@yanzhudd Julie Zhu (yanzhudd)

Labels

AKS Auto-Assign Auto assign by bot azure-client-tools-agent Pull request commented on or reviewed by Azure Client Tools Agent

Projects

None yet

Milestone

August 2026 (2026-08-04)

Development

Successfully merging this pull request may close these issues.

5 participants

@xmzhao0822 @yonzhan @FumingZhang @a0x1ab @yanzhudd