Skip to content

{Search} Add 2026-03-01-preview support#33268

Open
mattgotteiner wants to merge 3 commits intoAzure:devfrom
mattgotteiner:search-2026-03-01-preview
Open

{Search} Add 2026-03-01-preview support#33268
mattgotteiner wants to merge 3 commits intoAzure:devfrom
mattgotteiner:search-2026-03-01-preview

Conversation

@mattgotteiner
Copy link
Copy Markdown
Member

@mattgotteiner mattgotteiner commented Apr 24, 2026

Related command

az search

Description

Adds support for Azure AI Search management API 2026-03-01-preview.

Changes included:

  • Update generated Search AAZ command API versions from 2025-05-01 to 2026-03-01-preview.
  • Add AAZ command az search offering list for the new Offerings_List operation.
  • Add --knowledge-retrieval to az search service create/update for the new knowledgeRetrieval service property.
  • Expose knowledgeRetrieval in service read schemas.
  • Bump azure-mgmt-search to 9.2.0 and pass api_version='2026-03-01-preview' through the Search Python SDK client factory with an overridable default.
  • Update Search help text from Azure Search/Azure Cognitive Search wording to Azure AI Search where touched.
  • Add live recordings for az search offering list and --knowledge-retrieval create/update coverage.

AAZ tooling note: the local AAZ dev-tools fork at mattgotteiner/aaz-dev-tools@51524a31 was fixed so the Search 2026 command model can be generated from package-preview-2026-03-01. The generated raw model uses swagger-derived command names such as search search-service, so this PR preserves the existing public az search service surface while applying the generated 2026 API/schema deltas.

Testing Guide

  • py -3 -m compileall -q src\azure-cli\azure\cli\command_modules\search
  • python scripts\generate_latest_indices.py verify
  • Replay: pytest src\azure-cli\azure\cli\command_modules\search\tests\latest\test_offering.py::AzureSearchOfferingTests::test_offering_list src\azure-cli\azure\cli\command_modules\search\tests\latest\test_service.py::AzureSearchServicesTests::test_service_knowledge_retrieval -q
  • Live recording in 72f tenant, Azure Search POC (magottei) subscription:
    • AzureSearchOfferingTests.test_offering_list
    • AzureSearchServicesTests.test_service_knowledge_retrieval

History Notes

[Search] az search: Add 2026-03-01-preview support.
[Search] az search offering list: Add command to list Azure AI Search offerings by region.
[Search] az search service create/update: Add --knowledge-retrieval to configure the agentic retrieval billing plan.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Copilot AI review requested due to automatic review settings April 24, 2026 23:46
@azure-client-tools-bot-prd
Copy link
Copy Markdown

Validation for Azure CLI Full Test Starting...

Thanks for your contribution!

@azure-client-tools-bot-prd
Copy link
Copy Markdown

Hi @mattgotteiner,
Since the current milestone time is less than 7 days, this pr will be reviewed in the next milestone.

@azure-client-tools-bot-prd
Copy link
Copy Markdown

Validation for Breaking Change Starting...

Thanks for your contribution!

@github-actions
Copy link
Copy Markdown

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>

@yonzhan
Copy link
Copy Markdown
Collaborator

yonzhan commented Apr 24, 2026

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

Copy link
Copy Markdown
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Adds initial Azure AI Search management-plane support for API version 2026-03-01-preview in the az search command module, aligning both generated AAZ (REST) commands and SDK-backed operations with the new preview surface.

Changes:

  • Bump azure-mgmt-search dependency to 9.2.x/9.2.0 and force SDK client api_version='2026-03-01-preview'.
  • Update generated Search AAZ commands to use api-version=2026-03-01-preview and add --knowledge-retrieval to search service create/update.
  • Add new AAZ command az search offering list plus help entries and an implementation-plan markdown document.

Reviewed changes

Copilot reviewed 39 out of 39 changed files in this pull request and generated 9 comments.

Show a summary per file
File Description
src/azure-cli/setup.py Bumps azure-mgmt-search dependency range to ~=9.2.
src/azure-cli/requirements.py3.windows.txt Pins azure-mgmt-search==9.2.0 for Windows runtime.
src/azure-cli/requirements.py3.Linux.txt Pins azure-mgmt-search==9.2.0 for Linux runtime.
src/azure-cli/requirements.py3.Darwin.txt Pins azure-mgmt-search==9.2.0 for macOS runtime.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/usage/_show.py Updates AAZ usage show operation to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/usage/_list.py Updates AAZ usage list operation to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/shared_private_link_resource/_wait.py Updates AAZ SPLR wait to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/shared_private_link_resource/_update.py Updates AAZ SPLR update to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/shared_private_link_resource/_show.py Updates AAZ SPLR show to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/shared_private_link_resource/_list.py Updates AAZ SPLR list to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/shared_private_link_resource/_delete.py Updates AAZ SPLR delete to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/shared_private_link_resource/_create.py Updates AAZ SPLR create to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/query_key/_list.py Updates AAZ query-key list to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/query_key/_delete.py Updates AAZ query-key delete to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/query_key/_create.py Updates AAZ query-key create to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/private_link_resource/_list.py Updates AAZ private-link-resource list to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/private_endpoint_connection/_update.py Updates AAZ PEC update to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/private_endpoint_connection/_show.py Updates AAZ PEC show to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/private_endpoint_connection/_list.py Updates AAZ PEC list to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/private_endpoint_connection/_delete.py Updates AAZ PEC delete to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/network_security_perimeter_configuration/_show.py Updates AAZ NSPC show to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/network_security_perimeter_configuration/_reconcile.py Updates AAZ NSPC reconcile to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/network_security_perimeter_configuration/_list.py Updates AAZ NSPC list to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/admin_key/_regenerate.py Updates AAZ admin-key regenerate to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/admin_key/_list.py Updates AAZ admin-key list to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/_wait.py Updates AAZ service wait to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/_upgrade.py Updates AAZ service upgrade to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/_update.py Updates AAZ service update to 2026-03-01-preview and adds --knowledge-retrieval.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/_show.py Updates AAZ service show to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/_list.py Updates AAZ service list to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/_delete.py Updates AAZ service delete to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/_create.py Updates AAZ service create to 2026-03-01-preview and adds --knowledge-retrieval.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/service/_check_name_availability.py Updates AAZ name-availability check to 2026-03-01-preview.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/offering/_list.py Adds new AAZ command az search offering list.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/offering/init.py Exposes the new offering command group and commands.
src/azure-cli/azure/cli/command_modules/search/aaz/latest/search/offering/__cmd_group.py Adds new AAZ command group search offering.
src/azure-cli/azure/cli/command_modules/search/_help.py Adds help entries for search offering and search offering list.
src/azure-cli/azure/cli/command_modules/search/_client_factory.py Forces SDK-backed Search client to use api_version='2026-03-01-preview'.
search-api-version-update.md Adds implementation plan / API delta notes for reviewers.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Comment thread src/azure-cli/azure/cli/command_modules/search/_help.py Outdated
Comment thread src/azure-cli/azure/cli/command_modules/search/_client_factory.py Outdated
Comment thread search-api-version-update.md Outdated
mattgotteiner and others added 2 commits April 24, 2026 20:15
Add live recordings for knowledge retrieval and offering list, expose knowledge retrieval in read schemas, update Search help text, and make the Search client factory API version configurable.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Add live recordings for knowledge retrieval and offering list, expose knowledge retrieval in read schemas, update Search help text, and make the Search client factory API version configurable.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
@mattgotteiner
Copy link
Copy Markdown
Member Author

Addressed the Copilot review feedback in \�6b50f1983\: added \knowledgeRetrieval\ to read schemas, added live recordings for knowledge retrieval and offering list, regenerated latest help index, updated Azure AI Search wording/help/docstrings, made the Search client factory API version overrideable, and removed \search-api-version-update.md\ from the repo (moved to \C:\Users\magottei\source\scratch\search-api-version-update.md\). Also pushed the local AAZ dev-tools fixes used for generation to \mattgotteiner/aaz-dev-tools@51524a31\.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants