air: group commands under AI Runtime in root help

acceptance/help/output.txt

@@ -178,12 +178,12 @@ Domains
   domains                                *Beta* Manage domains for organizing and discovering data assets.
 
 Developer Tools
+  air                                    *Public Preview* Run and manage AI Runtime training workloads
   bundle                                 Declarative Automation Bundles let you express data/AI/analytics projects as code.
   sync                                   Synchronize a local directory to a workspace directory
 
 Additional Commands:
   account                                Databricks Account Commands
-  air                                    Run and manage AI Runtime training workloads
   aitools                                Databricks skills and plugins for coding agents
   api                                    Perform Databricks API call
   auth                                   Authentication related commands

cmd/air/air.go

@@ -10,8 +10,9 @@ import (
 // New returns the root command for the AI Runtime CLI.
 func New() *cobra.Command {
 	cmd := &cobra.Command{
-		Use:   "air",
-		Short: "Run and manage AI Runtime training workloads",
+		Use:     "air",
+		Short:   "*Public Preview* Run and manage AI Runtime training workloads",
+		GroupID: "development",
 		Long: `Run and manage AI Runtime training workloads on Databricks serverless GPU compute.
 
 These commands are under active development.`,

cmd/air/air_test.go

@@ -23,6 +23,13 @@ func TestNewRegistersAllSubcommands(t *testing.T) {
 	assert.Len(t, registered, len(want), "unexpected number of subcommands")
 }
 
+func TestNewHelpMetadata(t *testing.T) {
+	cmd := New()
+
+	assert.Equal(t, "development", cmd.GroupID)
+	assert.Contains(t, cmd.Short, "*Public Preview*")
+}
+
 func TestRunErrorIncludesDebugTip(t *testing.T) {
 	originalErr := errors.New("failed")
 	runCommand := &cobra.Command{