代理 databricks-ai-runtime 技能 为 AI 编码代理（如 Claude Code、Cursor 和 Codex）提供了通过 databricks air Databricks CLI 命令操作 AI 运行时的指令。 安装了该技能后，你可以让智能体编写工作负载 YAML、提交 GPU 训练任务，并查看其进度。
Requirements
-
Databricks CLI已通过身份验证并连接到你的工作区。 该技能示例使用扁平
code_source格式，这需要Databricks CLI版本1.20.0及以上。 - 一个启用了 AI 运行环境的工作空间。 参见 要求。
安装技能
运行下面的命令：
databricks aitools install --skills-only --skills databricks-ai-runtime
按照提示选择你的 AI 助手和安装范围。 有关安装选项，请参阅aitools命令组。
技能涵盖的内容
| 能力 | 代理可以执行的操作 |
|---|---|
| 编写工作负载 | 将请求转换为工作负载 YAML，其中包含计算资源、环境、代码源和框架的启动命令，例如 torchrun、Accelerate 或 DeepSpeed。 |
| 配置运行 | 设置环境变量、机密、超参数、重试和权限。 |
| 提交运行 | 使用 --dry-run 验证配置，然后使用 databricks air run 和幂等键提交。 |
| 检查并监视运行 | 用 databricks air get 和 databricks air list 获取状态，用 databricks air logs 流式传输日志，并指引你查看 MLflow 运行以获取指标。 |
| 取消运行 | 使用databricks air cancel取消一个、多个或所有活动运行。 |
| 调试故障 | 从失败的运行中提取日志并显示错误。 |
| 使用自定义 Docker 映像 | 构建并推送一个自定义镜像，然后在工作负载中引用它。 |