代理databricks-ai-runtime技能為 AI 編碼代理（如 Claude Code、Cursor 與 Codex）提供透過 databricks air Databricks CLI 指令操作 AI Runtime 的指令。 安裝了這個技能後，你可以請代理程式撰寫工作負載 YAML、提交 GPU 訓練執行，並檢查其進度。
Requirements
-
Databricks 的 CLI 已向你的工作區完成驗證。 技能範例使用扁平
code_source格式，需 Databricks CLI 版本 1.20.0 或以上。 - 一個啟用 AI 執行環境的工作空間。 請參閱 需求說明。
安裝這個技能
執行以下命令：
databricks aitools install --skills-only --skills databricks-ai-runtime
依照提示選擇你的 AI 助理和安裝範圍選項。 關於安裝選項，請參見 aitools 指令群組。
技能涵蓋的內容
| 能力 | 代理人能做的事 |
|---|---|
| 寫入工作負載 | 將請求轉換成包含運算、環境、程式碼來源及框架啟動指令（如 torchrun、Accelerate 或 DeepSpeed）的工作負載 YAML。 |
| 設定執行 | 設定環境變數、秘密、超參數、重試和權限。 |
| 提交執行 | 使用 --dry-run 驗證設定，然後使用 databricks air run 和冪等性金鑰提交。 |
| 檢查並監控執行情況 | 用 databricks air get 和 databricks air list 取得狀態，串流日誌用 databricks air logs，並將你導向 MLflow 執行頁面查看指標。 |
| 取消執行 | 使用 databricks air cancel 取消一個、多個或所有進行中的執行作業。 |
| 偵錯錯誤 | 從失敗的執行中擷取日誌並顯示錯誤。 |
| 使用自訂的 Docker 映像檔 | 建立並推送一個自訂映像檔，然後在工作負載中參考它。 |