Skip to content

air: show the Databricks AI environment in air get - #7637

Draft
dimk0 wants to merge 2 commits into
mainfrom
cnxt-2641-air-get-ai-environment
Draft

dimk0 wants to merge 2 commits into
mainfrom
cnxt-2641-air-get-ai-environment

Conversation

@dimk0

@dimk0 dimk0 commented Nov 14, 2026

Copy link
Copy Markdown
Member

Changes

  • air get now shows the environment, for example databricks_ai_v6, for runs submitted with environment.version: databricks_ai_v*, instead of Environment N/A. The air list info pane renders through the same code and is fixed too.
  • air run submits those versions as spec.base_environment = workspace-base-environments/<version> with no environment_version. The reader now strips that exact prefix, shared as workspaceBaseEnvironmentPrefix with buildRunEnvironment and convert-to-dabs. Numeric versions are still read from environment_version.

Why

Every run on the Databricks AI environment, which most AI Runtime cookbook recipes use, showed Environment N/A. Internal ticket: CNXT-2641.

Tests

  • TestAiRuntimeEnvironmentVersion: numeric environment_version and base_environment cases; the second fails without the fix.
  • New acceptance test acceptance/air/get-databricks-ai: air get text output for a databricks_ai_v6 run. Fails without the fix (Environment N/A).
  • go test ./cmd/air/... and go test ./acceptance -run 'TestAccept/air/' pass; ./task lint-q is clean. Unit tests ran with DATABRICKS_WORKSPACE_ID unset: TestLogsCommandRepeatsRunLinkAfterStreaming also fails on main when that variable is set.

This pull request and its description were written by Isaac.

dimk0 and others added 2 commits November 14, 2026 23:42
`air run` submits a databricks_ai_v* version as spec.base_environment
(workspace-base-environments/databricks_ai_v6), but `air get` and the
`air list` info pane read only spec.environment_version, so Environment
showed N/A. Strip the shared prefix and show the configured version.

Co-authored-by: Isaac <[email protected]>
Co-authored-by: Isaac <[email protected]>
@github-actions github-actions Bot added the AIR Databricks AI Runtime CLI label Nov 14, 2026
@github-actions

Copy link
Copy Markdown
Contributor

Commit: 322abe3a4 | Workflow run

Regression test report

Selected 1 changed tests (limit=20, 0 not selected)

1 tests: regression=1 unreleased=0 coverage=0 failing=0
test branch main (2699068) latest (v1.26.0)
TestAccept/air/get-databricks-ai ✅ ❌ ❌
TestAccept/air/get-databricks-ai: branch ✅ | main (2699068) ❌ | latest (v1.26.0) ❌

main (2699068):

acceptance_test.go:1374: Diff:
        --- air/get-databricks-ai/output.txt
        +++ /tmp/TestAcceptairget-databricks-ai1633857982/002/output.txt
        @@ -13,7 +13,7 @@
         │  MLflow Run    N/A                                             │
         │  User          [email protected]                                │
         │  Accelerators  1x A10                                          │
        -│  Environment   databricks_ai_v6                                │
        +│  Environment   N/A                                             │
         │                                                                │
         ╰────────────────────────────────────────────────────────────────╯
         
        
FAIL

latest (v1.26.0): same output as main (2699068)

@eng-dev-ecosystem-bot

Copy link
Copy Markdown
Collaborator

✅ Integration test report

Commit: 322abe3

Run: 43654791240

Env 💚​RECOVERED ✅​pass 🙈​skip Time
💚​ aws linux-2core-8gb 1 270 16 3:34
💚​ aws-windows-latest-4core-16gb 1 272 14 3:14
💚​ azure linux-2core-8gb 1 269 16 3:20
💚​ azure-windows-latest-4core-16gb 1 271 14 3:09
💚​ gcp linux-2core-8gb 1 270 16 3:35
💚​ gcp-windows-latest-4core-16gb 1 272 14 3:23
Test Name aws linux-2core-8gb aws-windows-latest-4core-16gb azure linux-2core-8gb azure-windows-latest-4core-16gb gcp linux-2core-8gb gcp-windows-latest-4core-16gb
💚​ TestAccept 💚​R 💚​R 💚​R 💚​R 💚​R 💚​R
Top 3 slowest tests (at least 2 minutes):
duration env testname
3:17 gcp-windows-latest-4core-16gb TestAccept
3:07 aws-windows-latest-4core-16gb TestAccept
3:03 azure-windows-latest-4core-16gb TestAccept

This branch has not been deployed

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

Labels

AIR Databricks AI Runtime CLI

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants