AIR CLI Integration: Scaffold experimental AIR CLI command package by riddhibhagwat-db · Pull Request #5564 · databricks/cli

riddhibhagwat-db · 2026-06-11T21:29:34Z

Changes

New cmd/experimental/air/ package containing an air parent command plus 7 stub subcommands: run, status, list, logs, cancel, register-image
currently, all subcommands return an air is not implemented yet error with representative flags mapped from the Python CLI. Registered under the hidden experimental group.
tools/list_embeds.py: text=True was changed to universal_newlines=True so the acceptance harness runs on Python 3.6. General tooling fix.

Why

The AI runtime CLI ships today as a separately installed Python wheel with its own auth, output, and packaging. Folding it into the main Go CLI gives users one databricks install with consistent profiles, authentication, and -o json output, and removes a parallel toolchain to maintain. Landing the package scaffold first lets the individual commands be ported in small, reviewable PRs (status is next) instead of one large drop. Every stub is wired and navigable, so the command tree and registration are reviewable now without functional code.

Tests

Unit (cmd/experimental/air/): New() registers all six subcommands; each stub returns the not-implemented error.
Acceptance (acceptance/experimental/air/unimplemented/): runs every stub end-to-end and asserts the message + non-zero exit.

test with:
go test ./cmd/experimental/air/...
go test ./acceptance -run 'TestAccept/experimental/air'

github-actions · 2026-06-11T21:30:09Z

Waiting for approval

Based on git history, these people are best suited to review:

@pietern -- recent work in ./
@simonfaltum -- recent work in cmd/experimental/
@janniklasrose -- recent work in ./

Eligible reviewers: @andrewnester, @anton-107, @denik, @renaudhartert-db, @shreyas-goenka

_{Suggestions based on git history. See OWNERS for ownership rules.}

eng-dev-ecosystem-bot · 2026-06-11T22:18:16Z

Integration test report

Commit: 059bd61

Run: 27448495793

	Env	🟨KNOWN	🔄flaky	💚RECOVERED	🙈SKIP	✅pass	🙈skip	Time
🟨	aws linux	7			15	264	979	8:17
🟨	aws windows	7			15	266	977	16:00
💚	aws-ucws linux			7	15	360	893	8:01
💚	aws-ucws windows			7	15	362	891	10:19
💚	azure linux			1	17	267	977	6:53
💚	azure windows			1	17	269	975	9:29
💚	azure-ucws linux			1	17	365	889	7:30
💚	azure-ucws windows			1	17	367	887	10:31
🔄	gcp linux		2	1	17	261	980	10:10
🔄	gcp windows		2	1	17	263	978	11:57

24 interesting tests: 15 SKIP, 7 KNOWN, 2 flaky

	Test Name	aws linux	aws windows	aws-ucws linux	aws-ucws windows	azure linux	azure windows	azure-ucws linux	azure-ucws windows	gcp linux	gcp windows
🟨	TestAccept	🟨K	🟨K	💚R	💚R	💚R	💚R	💚R	💚R	💚R	💚R
🙈	TestAccept/bundle/invariant/no_drift	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/permissions	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🟨	TestAccept/bundle/resources/permissions/jobs/destroy_without_mgmtperms/with_permissions	🟨K	🟨K	💚R	💚R	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🟨	TestAccept/bundle/resources/permissions/jobs/destroy_without_mgmtperms/with_permissions/DATABRICKS_BUNDLE_ENGINE=direct	🟨K	🟨K	💚R	💚R
🟨	TestAccept/bundle/resources/permissions/jobs/destroy_without_mgmtperms/with_permissions/DATABRICKS_BUNDLE_ENGINE=terraform	🟨K	🟨K	💚R	💚R
🟨	TestAccept/bundle/resources/permissions/jobs/destroy_without_mgmtperms/without_permissions	🟨K	🟨K	💚R	💚R	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🟨	TestAccept/bundle/resources/permissions/jobs/destroy_without_mgmtperms/without_permissions/DATABRICKS_BUNDLE_ENGINE=direct	🟨K	🟨K	💚R	💚R
🟨	TestAccept/bundle/resources/permissions/jobs/destroy_without_mgmtperms/without_permissions/DATABRICKS_BUNDLE_ENGINE=terraform	🟨K	🟨K	💚R	💚R
🙈	TestAccept/bundle/resources/postgres_branches/basic	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/postgres_branches/recreate	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/postgres_branches/replace_existing	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/postgres_branches/update_protected	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/postgres_branches/without_branch_id	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/postgres_endpoints/basic	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/postgres_endpoints/recreate	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/postgres_projects/update_display_name	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/synced_database_tables/basic	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/vector_search_endpoints/drift/recreated_same_name	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/vector_search_indexes/basic	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/bundle/resources/vector_search_indexes/grants/select	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🙈	TestAccept/ssh/connection	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S	🙈S
🔄	TestFetchRepositoryInfoAPI_FromRepo/root	✅p	✅p	✅p	✅p	✅p	✅p	✅p	✅p	🔄f	🔄f
🔄	TestFetchRepositoryInfoAPI_FromRepo/subdir	✅p	✅p	✅p	✅p	✅p	✅p	✅p	✅p	🔄f	🔄f

Top 28 slowest tests (at least 2 minutes):

duration	env	testname
5:13	azure-ucws windows	TestAccept
5:06	aws-ucws windows	TestAccept
5:05	azure windows	TestAccept
5:03	gcp windows	TestAccept
4:26	gcp linux	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=direct
4:17	gcp linux	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=terraform
4:15	gcp windows	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=direct
4:13	gcp windows	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=terraform
3:10	aws-ucws linux	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=direct
3:08	aws linux	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=terraform
2:54	azure linux	TestAccept
2:52	azure-ucws windows	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=direct
2:52	gcp linux	TestAccept
2:51	azure-ucws linux	TestAccept
2:51	aws linux	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=direct
2:50	aws-ucws linux	TestAccept
2:46	azure-ucws windows	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=terraform
2:43	aws windows	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=terraform
2:42	aws windows	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=direct
2:39	azure-ucws linux	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=terraform
2:39	azure linux	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=terraform
2:39	aws-ucws windows	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=terraform
2:37	azure windows	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=terraform
2:35	azure linux	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=direct
2:29	aws-ucws linux	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=terraform
2:29	aws-ucws windows	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=direct
2:27	azure-ucws linux	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=direct
2:25	azure windows	TestAccept/bundle/resources/apps/inline_config/DATABRICKS_BUNDLE_ENGINE=direct

simonfaltum

Thanks for the scaffold. The Go code itself is in good shape and the acceptance setup follows the existing patterns (the EnvMatrix override and the committed out.test.toml are exactly right). Requesting changes for three things:

The tools/list_embeds.py change needs to come out, see the inline comment. It's unrelated to this PR and the stated rationale doesn't match the repo.
Package location should follow the established experimental layout (experimental/air/cmd), see the inline comment.
The description needs fixing: it says 7 stub subcommands but lists and implements 6 (your own Tests section says six). Also drop the list_embeds.py bullet together with that change.

Two questions:

Which Python air CLI source/version were the flags mapped from? Please link it in the description; flag parity isn't verifiable from this repo.
Was a seventh subcommand intentionally dropped, or is "7" a typo?

The remaining inline comments (test assertion, help case, --retry help text, cancel args, test.toml defaults) are all quick fixes.

Add the experimental `air` command group as the Go port surface for the Python `air` CLI. Every subcommand (run, status, list, logs, cancel, register-image) is registered as a stub that returns a not-implemented error; the real implementations land in later milestones. The package lives under experimental/air/cmd (imported as aircmd), matching the layout of the other experimental features (aitools, genie, postgres); cmd/experimental/ keeps only the dispatcher. TEST_PACKAGES in Taskfile.yml gains ./experimental/air/... so the unit tests keep running after the move. Includes unit tests for the command-tree wiring and the not-implemented stubs, plus an acceptance test exercising the stubs end-to-end. Co-authored-by: Isaac

riddhibhagwat-db · 2026-06-12T22:58:20Z

Thanks for the review, I have made the changes and updated the branch with the changes. Regarding your questions:

Which Python air CLI source/version were the flags mapped from? Please link it in the description; flag parity isn't verifiable from this repo.

Individual command flags:

Global flags:

Was a seventh subcommand intentionally dropped, or is "7" a typo?

Yes, the seventh subcommand in the original air CLI is the version command that we decided to drop. This is why the description said 7 rather than 6, which was a mistype on my part while opening this PR.

One more thing I updated: the latest name for the status command is get, so just adding in that renaming in the latest commit.

Rename the run-details subcommand from `status` to `get`, matching the Python air CLI's current `air get run` naming (it replaced `get status`). Renames the file, constructor, command name, and updates the stub/help/unimplemented tests and goldens accordingly. Co-authored-by: Isaac

riddhibhagwat-db requested review from maggiewang-db and simonfaltum June 11, 2026 21:29

riddhibhagwat-db temporarily deployed to test-trigger-is June 11, 2026 21:30 — with GitHub Actions Inactive

maggiewang-db approved these changes Jun 12, 2026

View reviewed changes

simonfaltum requested changes Jun 12, 2026

View reviewed changes

riddhibhagwat-db force-pushed the air-integration-m0 branch from 0474207 to 3ff5a77 Compare June 12, 2026 18:18

riddhibhagwat-db temporarily deployed to test-trigger-is June 12, 2026 18:18 — with GitHub Actions Inactive

riddhibhagwat-db force-pushed the air-integration-m0 branch from 3ff5a77 to 30d284d Compare June 12, 2026 18:52

riddhibhagwat-db temporarily deployed to test-trigger-is June 12, 2026 18:53 — with GitHub Actions Inactive

riddhibhagwat-db force-pushed the air-integration-m0 branch from 30d284d to d8652c2 Compare June 12, 2026 19:03

riddhibhagwat-db temporarily deployed to test-trigger-is June 12, 2026 19:04 — with GitHub Actions Inactive

riddhibhagwat-db force-pushed the air-integration-m0 branch from d8652c2 to 8462709 Compare June 12, 2026 19:07

riddhibhagwat-db temporarily deployed to test-trigger-is June 12, 2026 19:08 — with GitHub Actions Inactive

riddhibhagwat-db force-pushed the air-integration-m0 branch from 8462709 to 80144e6 Compare June 12, 2026 19:08

riddhibhagwat-db temporarily deployed to test-trigger-is June 12, 2026 19:09 — with GitHub Actions Inactive

riddhibhagwat-db force-pushed the air-integration-m0 branch from 80144e6 to d96e64f Compare June 12, 2026 20:35

riddhibhagwat-db temporarily deployed to test-trigger-is June 12, 2026 20:36 — with GitHub Actions Inactive

riddhibhagwat-db force-pushed the air-integration-m0 branch from d96e64f to 6c80f23 Compare June 12, 2026 20:41

riddhibhagwat-db temporarily deployed to test-trigger-is June 12, 2026 20:41 — with GitHub Actions Inactive

riddhibhagwat-db requested a review from simonfaltum June 12, 2026 21:53

riddhibhagwat-db temporarily deployed to test-trigger-is June 12, 2026 23:14 — with GitHub Actions Inactive

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

AIR CLI Integration: Scaffold experimental AIR CLI command package#5564

AIR CLI Integration: Scaffold experimental AIR CLI command package#5564
riddhibhagwat-db wants to merge 2 commits into
mainfrom
air-integration-m0

riddhibhagwat-db commented Jun 11, 2026

Uh oh!

github-actions Bot commented Jun 11, 2026 •

edited

Loading

Uh oh!

eng-dev-ecosystem-bot commented Jun 11, 2026 •

edited

Loading

Uh oh!

simonfaltum left a comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

riddhibhagwat-db commented Jun 12, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

4 participants

Conversation

riddhibhagwat-db commented Jun 11, 2026

Changes

Why

Tests

Uh oh!

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

Uh oh!

Waiting for approval

Uh oh!

eng-dev-ecosystem-bot commented Jun 11, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Integration test report

Uh oh!

simonfaltum left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

riddhibhagwat-db commented Jun 12, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

4 participants

github-actions Bot commented Jun 11, 2026 •

edited

Loading

eng-dev-ecosystem-bot commented Jun 11, 2026 •

edited

Loading