Use as habilidades do agente com o Runtime de IA

A databricks-ai-runtime habilidade de agente fornece aos agentes de codificação de IA, como Claude Code, Cursor e Codex, instruções para trabalhar com o Runtime de IA por meio dos databricks air comandos da CLI do Databricks. Com a habilidade instalada, você pode pedir ao seu agente para escrever um YAML de carga de trabalho, enviar uma execução de treinamento com GPU e verificar seu progresso.

Requirements

Instalar a habilidade

Execute o comando a seguir:

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

Siga as instruções para escolher seu agente de IA e o escopo da instalação. Para opções de instalação, veja aitools o grupo de comandos.

O que a habilidade cobre

Capability O que o agente pode fazer
Gravar cargas de trabalho Transforme uma requisição em um YAML de carga de trabalho com computação, ambiente, código fonte e o comando de lançamento do seu framework, como torchrun, Accelerate ou DeepSpeed.
Configurar execuções Defina variáveis de ambiente, segredos, hiperparâmetros, tentativas e permissões.
Enviar execuções Valide a configuração com --dry-run, então envie com databricks air run e uma chave de idempotência.
Verifique e monitore as execuções Obtenha status com databricks air get e databricks air list, transmita logs com databricks air logs e direcione você para a execução do MLflow para ver as métricas.
Cancelar execuções Cancele uma, várias ou todas as suas execuções ativas com databricks air cancel.
Falhas de depuração Extraia logs de uma execução falhada e mostre o erro.
Usar imagens personalizadas do Docker Construa e faça push de uma imagem personalizada, depois faça referência a ela na carga de trabalho.

Recursos adicionais