From 4ba436572a293f401655c18b8480c2bbca1ef361 Mon Sep 17 00:00:00 2001 From: "ben.hansen" Date: Thu, 1 Oct 2026 20:02:40 +0000 Subject: [PATCH 1/4] air: document local agent skill installation in config help --- acceptance/air/config-help/output.txt | 13 +++++++++++++ cmd/air/runconfig.go | 17 +++++++++++++++-- 2 files changed, 28 insertions(+), 2 deletions(-) diff --git a/acceptance/air/config-help/output.txt b/acceptance/air/config-help/output.txt index c0949f97a43..d4a43efcb60 100644 --- a/acceptance/air/config-help/output.txt +++ b/acceptance/air/config-help/output.txt @@ -36,6 +36,19 @@ Global Flags: config The run YAML schema. Pass a field path for details, e.g. config.compute.accelerator_type. + AI coding agents can use the databricks-ai-runtime skill to write workload + YAML, submit runs, and monitor progress. The skill is experimental and not + officially supported. + + Set the DATABRICKS_SKILLS_REF environment variable to main to get the updated + skill, then install it locally for your coding agent: + + databricks aitools install --skills-only --skills databricks-ai-runtime --experimental + + Follow the prompts to choose your coding agent and installation scope. + Use --scope project to install into the current project. + See: https://github.com/databricks/databricks-agent-skills/tree/main/experimental/databricks-ai-runtime + Fields: experiment_name (required) Name of the experiment. compute (required) Which accelerators to run on and how many. diff --git a/cmd/air/runconfig.go b/cmd/air/runconfig.go index b466f2784f0..1e74a041aca 100644 --- a/cmd/air/runconfig.go +++ b/cmd/air/runconfig.go @@ -530,8 +530,21 @@ type configField struct { // `-h config.` and --override path validation resolve against it. func configSchema() configField { return configField{ - path: configHelpRoot, - help: "The run YAML schema. Pass a field path for details, e.g. " + configHelpRoot + ".compute.accelerator_type.", + path: configHelpRoot, + help: "The run YAML schema. Pass a field path for details, e.g. " + configHelpRoot + `.compute.accelerator_type. + + AI coding agents can use the databricks-ai-runtime skill to write workload + YAML, submit runs, and monitor progress. The skill is experimental and not + officially supported. + + Set the DATABRICKS_SKILLS_REF environment variable to main to get the updated + skill, then install it locally for your coding agent: + + databricks aitools install --skills-only --skills databricks-ai-runtime --experimental + + Follow the prompts to choose your coding agent and installation scope. + Use --scope project to install into the current project. + See: https://github.com/databricks/databricks-agent-skills/tree/main/experimental/databricks-ai-runtime`, children: describeStruct(reflect.TypeFor[runConfig](), configHelpRoot), } } From df17647059092fb719551c5b021c3b3a69017790 Mon Sep 17 00:00:00 2001 From: "ben.hansen" Date: Thu, 1 Oct 2026 20:49:08 +0000 Subject: [PATCH 2/4] air: make the skill install command copy-pasteable --- acceptance/air/config-help/output.txt | 8 +++++--- cmd/air/runconfig.go | 8 +++++--- 2 files changed, 10 insertions(+), 6 deletions(-) diff --git a/acceptance/air/config-help/output.txt b/acceptance/air/config-help/output.txt index d4a43efcb60..bacaf142322 100644 --- a/acceptance/air/config-help/output.txt +++ b/acceptance/air/config-help/output.txt @@ -40,10 +40,12 @@ config YAML, submit runs, and monitor progress. The skill is experimental and not officially supported. - Set the DATABRICKS_SKILLS_REF environment variable to main to get the updated - skill, then install it locally for your coding agent: + Install the updated skill from the repository's main branch (Bash/Zsh): - databricks aitools install --skills-only --skills databricks-ai-runtime --experimental + DATABRICKS_SKILLS_REF=main databricks aitools install \ + --skills-only --skills databricks-ai-runtime --experimental + + The default skills release still contains the older Python CLI skill. Follow the prompts to choose your coding agent and installation scope. Use --scope project to install into the current project. diff --git a/cmd/air/runconfig.go b/cmd/air/runconfig.go index 1e74a041aca..e7d869a3d21 100644 --- a/cmd/air/runconfig.go +++ b/cmd/air/runconfig.go @@ -537,10 +537,12 @@ func configSchema() configField { YAML, submit runs, and monitor progress. The skill is experimental and not officially supported. - Set the DATABRICKS_SKILLS_REF environment variable to main to get the updated - skill, then install it locally for your coding agent: + Install the updated skill from the repository's main branch (Bash/Zsh): - databricks aitools install --skills-only --skills databricks-ai-runtime --experimental + DATABRICKS_SKILLS_REF=main databricks aitools install \ + --skills-only --skills databricks-ai-runtime --experimental + + The default skills release still contains the older Python CLI skill. Follow the prompts to choose your coding agent and installation scope. Use --scope project to install into the current project. From 138f0785f193cbae0350bfdfba5d79b8914b30a5 Mon Sep 17 00:00:00 2001 From: "ben.hansen" Date: Thu, 1 Oct 2026 20:57:30 +0000 Subject: [PATCH 3/4] air: simplify agent skill help wording --- acceptance/air/config-help/output.txt | 5 +---- cmd/air/runconfig.go | 5 +---- 2 files changed, 2 insertions(+), 8 deletions(-) diff --git a/acceptance/air/config-help/output.txt b/acceptance/air/config-help/output.txt index bacaf142322..5a93805e623 100644 --- a/acceptance/air/config-help/output.txt +++ b/acceptance/air/config-help/output.txt @@ -37,16 +37,13 @@ config The run YAML schema. Pass a field path for details, e.g. config.compute.accelerator_type. AI coding agents can use the databricks-ai-runtime skill to write workload - YAML, submit runs, and monitor progress. The skill is experimental and not - officially supported. + YAML, submit runs, and monitor progress. Install the updated skill from the repository's main branch (Bash/Zsh): DATABRICKS_SKILLS_REF=main databricks aitools install \ --skills-only --skills databricks-ai-runtime --experimental - The default skills release still contains the older Python CLI skill. - Follow the prompts to choose your coding agent and installation scope. Use --scope project to install into the current project. See: https://github.com/databricks/databricks-agent-skills/tree/main/experimental/databricks-ai-runtime diff --git a/cmd/air/runconfig.go b/cmd/air/runconfig.go index e7d869a3d21..76f98f5aa9f 100644 --- a/cmd/air/runconfig.go +++ b/cmd/air/runconfig.go @@ -534,16 +534,13 @@ func configSchema() configField { help: "The run YAML schema. Pass a field path for details, e.g. " + configHelpRoot + `.compute.accelerator_type. AI coding agents can use the databricks-ai-runtime skill to write workload - YAML, submit runs, and monitor progress. The skill is experimental and not - officially supported. + YAML, submit runs, and monitor progress. Install the updated skill from the repository's main branch (Bash/Zsh): DATABRICKS_SKILLS_REF=main databricks aitools install \ --skills-only --skills databricks-ai-runtime --experimental - The default skills release still contains the older Python CLI skill. - Follow the prompts to choose your coding agent and installation scope. Use --scope project to install into the current project. See: https://github.com/databricks/databricks-agent-skills/tree/main/experimental/databricks-ai-runtime`, From 3477e9f1f501311b71247385513b264dfa09c45d Mon Sep 17 00:00:00 2001 From: "ben.hansen" Date: Thu, 1 Oct 2026 21:26:52 +0000 Subject: [PATCH 4/4] air: use released skills in the install help example --- acceptance/air/config-help/output.txt | 5 ++--- cmd/air/runconfig.go | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/acceptance/air/config-help/output.txt b/acceptance/air/config-help/output.txt index 5a93805e623..000f02f9513 100644 --- a/acceptance/air/config-help/output.txt +++ b/acceptance/air/config-help/output.txt @@ -39,10 +39,9 @@ config AI coding agents can use the databricks-ai-runtime skill to write workload YAML, submit runs, and monitor progress. - Install the updated skill from the repository's main branch (Bash/Zsh): + Install the skill locally for your coding agent: - DATABRICKS_SKILLS_REF=main databricks aitools install \ - --skills-only --skills databricks-ai-runtime --experimental + databricks aitools install --skills-only --skills databricks-ai-runtime --experimental Follow the prompts to choose your coding agent and installation scope. Use --scope project to install into the current project. diff --git a/cmd/air/runconfig.go b/cmd/air/runconfig.go index 76f98f5aa9f..3f7e09e64ad 100644 --- a/cmd/air/runconfig.go +++ b/cmd/air/runconfig.go @@ -536,10 +536,9 @@ func configSchema() configField { AI coding agents can use the databricks-ai-runtime skill to write workload YAML, submit runs, and monitor progress. - Install the updated skill from the repository's main branch (Bash/Zsh): + Install the skill locally for your coding agent: - DATABRICKS_SKILLS_REF=main databricks aitools install \ - --skills-only --skills databricks-ai-runtime --experimental + databricks aitools install --skills-only --skills databricks-ai-runtime --experimental Follow the prompts to choose your coding agent and installation scope. Use --scope project to install into the current project.