Használd az ügynökkészségeket az AI Runtime-mal

Az databricks-ai-runtime ügynöki készség utasításokat ad az AI kódoló ügynököknek, mint például a Claude Code, a Cursor és a Codex, az AI Runtime-mal való munkához a databricks air Databricks CLI parancsokon keresztül. A képesség telepítésével megkérheted az ügynöködet, hogy írjon egy workload YAML-fájlt, küldjön be egy GPU-s betanítási futtatást, és ellenőrizze annak előrehaladását.

Requirements

A képesség telepítése

Futtassa a következő parancsot:

databricks aitools install --skills-only --skills databricks-ai-runtime

Kövesse az utasításokat, hogy válassza ki az AI ügynököt és a telepítési hatókört. A telepítési lehetőségekért lásd a aitools parancscsoportot.

Mit tartalmaz a képesség

Capability Az ügynök által elvégezhető műveletek
Írási számítási feladatok Alakíts egy kérést olyan workload YAML-fájllá, amely tartalmazza a számítási erőforrásokat, a környezetet, a kódforrást és a keretrendszered indítási parancsát, például torchrun, Accelerate vagy DeepSpeed.
Futtatások konfigurálása Állíts be környezeti változókat, titkokat, hiperparamétereket, újrapróbálkozásokat és jogosultságokat.
Futamok beküldése Validáld a konfigurációt --dry-run, majd küldd be databricks air run és egy idempotenciakulccsal.
Ellenőrizd és figyeld a futamokat Kérd le az állapotot a databricks air get és databricks air list segítségével, streameld a naplókat a databricks air logs segítségével, és az MLflow-futtatáshoz irányít a metrikákhoz.
Futtatások megszakítása Szakíts meg egyet, többet vagy az összes aktív futásodat a(z) databricks air cancel segítségével.
Hibakeresési sikertelenségek Töltsd le a naplókat egy sikertelen futásból, és jelenítsd meg a hibát.
Egyéni Docker-rendszerképek használata Építs és push-olj egy egyedi lemezképet, majd hivatkozz rá a workloadban.

További erőforrások