Intégrez Databricks dans Kiro IDE grâce à la puissance de l'AI Dev Kit

Pourquoi c'est important

Le développement assisté par IA s'effondre dès que l'assistant doit deviner les noms des colonnes, la structure des tables ou les catalogues auxquels vous avez accès. La solution est l'ancrage : connectez l'assistant aux métadonnées actives de l'espace de travail via le Model Context Protocol (MCP), et le SQL qu'il écrit utilisera vos colonnes réelles, les modèles dbt joindront de vraies tables, et chaque requête héritera des autorisations Unity Catalog déjà en place. Rien ne quitte la plateforme. L'IA ne voit que ce que vous pouvez voir.

Deux étapes majeures viennent d'être franchies pour rendre cela concret dans l'IDE Kiro :

Premièrement, le Databricks AI Dev Kit a ajouté le support de Kiro en amont dans la PR #511. L'installateur unifié traite kiro comme une cible de premier ordre aux côtés de claude, cursor, copilot, codex et gemini. Une seule commande, et Kiro intègre l'ensemble de la boîte à outils à l'adresse ~/.kiro/skills/ et ~/.kiro/settings/mcp.json.

Deuxièmement, le Databricks AI Dev Kit Power est désormais disponible dans le catalogue Kiro Powers dans la PR #129. Ouvrez le panneau Powers, cliquez sur Essayer, et le Power exécute l'ensemble du processus d'intégration : installateur, configuration du MCP, détection de l'authentification et chargement des compétences.

En combinant cela avec les quatre serveurs MCP distants gérés par Databricks déjà intégrés à la plateforme, vous disposez de deux méthodes pour connecter Kiro à Databricks. Les deux partagent un résultat commun : les développeurs déploient des analyses, des pipelines et des workflows d'agents plus rapidement lorsque l'assistant hérite des autorisations réelles de l'espace de travail au lieu de deviner les schémas, les colonnes et les accès.

Pourquoi choisir Databricks pour le développement assisté par IA

Les deux étapes clés ci-dessus rendent l'association Kiro × Databricks concrète. Ce qui importe vraiment, c'est ce qui se cache derrière. Trois éléments font de Databricks le socle de choix pour le développement assisté par IA, quel que soit le chemin choisi.

Unity Catalog est la seule couche de gouvernance qui ancre l'IA au niveau des données. Chaque appel MCP — Chemin A ou Chemin B — hérite des autorisations basées sur les lignes, les colonnes et les tags. L'assistant ne dispose pas d'une vue privilégiée sur vos données ; il voit exactement ce que vous pouvez voir. Il n'y a pas de couche de contrôle d'accès distincte à gérer, et aucun risque que l'IA n'écrive des requêtes sur des tables dont elle ne devrait même pas soupçonner l'existence.

Une seule copie des données, un seul ensemble de définitions. Puisque Databricks est un lakehouse, la table que l'assistant interroge via databricks-sql est la même que celle dans laquelle votre modèle dbt écrit, la même que celle que votre espace Genie expose, et la même que celle que votre tableau de bord AI/BI lit. Il n'y a pas de synchronisation entre le warehouse et le lake à interrompre, ni de couche sémantique distincte à maintenir synchronisée. Lorsque l'assistant s'ancre dans samples.tpch.lineitem, il s'ancre dans la même définition que tous les autres outils utilisent.

La pile IA complète est intégrée, pas simplement greffée. Mosaic AI Gateway achemine les appels de modèles. Agent Bricks orchestre les workflows multi-agents. MLflow suit les expériences et les évaluations. Vector Search alimente la recherche sémantique. Lakebase gère l'état transactionnel. Tout cela apparaît dans le Power, le tout sur le même UC. Vous n'assemblez pas tant bien que mal de nombreux produits ; vous utilisez une seule et même plateforme.

Il y a un quatrième point qui mérite d'être mentionné : le Power lui-même est conçu par Databricks. Aucune autre plateforme de données ne propose de Power IDE en un clic pour Kiro, Cursor, Claude, Copilot, Codex et Gemini. La couche MCP est ouverte, le protocole est ouvert, l'intégration est ouverte — mais l'expérience globale est conçue par Databricks spécifiquement pour la façon dont nos clients développent.

Aperçu des deux approches

Aperçu de l'architecture d'intégration

Les deux approches partagent le même back-end : l'application de Unity Catalog et l'identité de l'espace de travail Databricks. Elles diffèrent par leur périmètre et leur modèle d'authentification.

Chemin A : se connecter aux quatre serveurs MCP gérés

C'est la configuration la plus légère. Un seul fichier mcp.json, un jeton d'accès personnel (PAT) Databricks et la modification d'un profil de shell. En moins de 10 minutes, Kiro communique avec Genie, SQL, Unity Catalog Functions et Vector Search.

Prérequis

• Un espace de travail Databricks sur AWS avec Unity Catalog activé.
• Un jeton d'accès personnel (PAT) Databricks ou un jeton OAuth limité aux serveurs MCP que vous prévoyez d'utiliser (sql, unity-catalog, genie, vector-search). Les PAT inutilisés expirent automatiquement après 90 jours.
• Kiro installé et lancé au moins une fois pour que ~/.kiro/ existe.
• Le nom d'hôte de votre espace de travail au format <workspace>.cloud.databricks.com.

Générer un PAT Databricks

Dans l'espace de travail Databricks, accédez à Settings, Developer, Access tokens, Manage, puis Generate new token. Définissez une date d'expiration conforme à la politique de rotation de votre équipe. Sélectionnez uniquement les portées d'API dont vous avez besoin ; le principe du moindre privilège est préférable à la facilité du "tout autoriser". Copiez immédiatement le jeton. Databricks ne l'affichera plus.

Où Kiro stocke la configuration MCP

Kiro lit la configuration MCP à partir de fichiers JSON selon deux portées ; celle de l'espace de travail prévaut sur celle de l'utilisateur.

• Portée utilisateur : ~/.kiro/settings/mcp.json s'applique à chaque espace de travail.
• Portée espace de travail : $PWD/.kiro/settings/mcp.json s'applique uniquement à l'espace de travail actuel et remplace l'entrée de même clé de la portée utilisateur.

Installation en un clic depuis le répertoire du serveur Kiro

Ouvrez kiro.dev/docs/mcp/servers/, trouvez la ligne Databricks et cliquez sur Add to Kiro. Le navigateur lance Kiro et ouvre une boîte de dialogue de confirmation avec une configuration pré-remplie. Confirmez pour écrire l'entrée databricks-sql dans ~/.kiro/settings/mcp.json. L'entrée fait référence à deux variables d'environnement qui n'existent pas encore ; nous les définirons ensuite.

Vérifier (ou ajouter) l'entrée databricks-sql

Définir les variables d'environnement

Dans le profil de shell qui lance Kiro (généralement ~/.zshrc sur macOS) :

Exécutez le profil (source ~/.zshrc) avant de lancer Kiro. Quittez complètement Kiro (Cmd+Q sur macOS) et rouvrez-le. L'action Reload Window ne recharge pas les variables d'environnement ; seul le redémarrage du processus le permet.

Ajouter Genie, UC Functions et Vector Search

Les quatre serveurs gérés par Databricks se connectent en tant que MCP HTTP distants. L'étape d'initialisation (handshake) réussit même avec une URL temporaire ; le serveur ne valide la ressource que lorsqu'un outil est appelé. Un état connecté mais défaillant, où tools/call renvoie RESOURCE_DOES_NOT_EXIST ou PERMISSION_DENIED, est le mode de défaillance le plus courant. Effectuez d'abord ces vérifications préalables :

• Genie : confirmez qu'un espace Genie existe et que vous pouvez l'ouvrir. L'ID de l'espace apparaît dans l'URL.
• Fonctions UC : confirmez que la fonction existe et que vous disposez de EXECUTE. Listez les fonctions avec 
  SELECT * FROM system.information_schema.routines WHERE routine_type = 'FUNCTION'.
• Vector Search : confirmez qu'un point de terminaison avec au moins un index accessible existe sous Catalogue, Vector Search.
• Portée du PAT : un PAT limité à l'espace de travail peut toujours atteindre PERMISSION_DENIED sur les espaces Genie ou les index vectoriels si l'utilisateur ne dispose pas des droits « Can View » sur la ressource spécifique. Les espaces Genie sont partagés individuellement.

Ajoutez les variables d'environnement supplémentaires (DATABRICKS_GENIE_MCP_URL, DATABRICKS_UC_FUNCTIONS_MCP_URL, DATABRICKS_VECTOR_SEARCH_MCP_URL) et mettez à jour mcp.json avec la configuration complète :

Formats d'URL par serveur :

Quittez et relancez à nouveau Kiro. Ouvrez la section MCP SERVERS du panneau Kiro ; les quatre entrées databricks-* apparaissent avec des indicateurs d'état verts. Cliquez sur reconnecter pour tout élément en rouge et relancez la vérification préliminaire. Essayez une première requête simple dans le panneau de discussion : « Listez les catalogues auxquels j'ai accès. »

Voie B : installer le Databricks AI Dev Kit Power

Les quatre serveurs gérés couvrent le SQL, la recherche sémantique et l'analyse en langage naturel, ce qui est suffisant pour de nombreux développeurs. Si votre flux de travail englobe des pipelines, des tâches (jobs), le service de modèles (model serving), Lakebase, les Asset Bundles, Mosaic AI, Agent Bricks, les tableaux de bord AI/BI, MLflow ou les Databricks Apps, la configuration à quatre serveurs vous obligera à faire des copier-coller continus vers l'interface utilisateur (UI) de l'espace de travail.

Le Databricks AI Dev Kit Power résout ce problème. Une seule installation. Tous les outils et compétences essentiels, quatre options d'authentification, le tout chargeable à la demande.

Ce que vous obtenez

Installer en un clic

Dans Kiro, ouvrez le panneau Powers, recherchez databricks et cliquez sur Essayer. Le Power exécute l'installateur officiel du Databricks AI Dev Kit en mode Kiro non interactif :

L'installateur télécharge le serveur MCP, crée un environnement virtuel uv et extrait la bibliothèque de compétences expertes dans ~/.kiro/skills/. Le Power copie les compétences dans son propre répertoire steering/ afin qu'elles se chargent à la demande en fonction de la tâche à accomplir. Aucun contenu n'est intégré dans le Power lui-même ; tout est récupéré en amont, de sorte que les compétences restent à jour. 

Les quatre options d'authentification

Le flux d'intégration du Power détecte les identifiants existants et vous guide vers le bon choix. Les quatre options sont documentées en ligne :

Le mcp.json du Power est livré avec disabled: true jusqu'à ce que vous choisissiez une option ; rien ne se connecte tant que vous n'avez pas explicitement choisi et configuré vos identifiants. Le flux de détection des identifiants est neutre. Si plusieurs identifiants sont détectés, les quatre options sont présentées dans l'ordre, sans option par défaut ni réutilisation silencieuse.

Vérifier l'installation

Redémarrez Kiro, ouvrez le panneau MCP SERVERS et confirmez que l'entrée databricks est connectée (en vert). Demandez au chat : « Obtenir mon utilisateur Databricks actuel. » Ce simple appel teste l'authentification, la résolution des variables d'environnement et l'activation du serveur. Si cela fonctionne, l'ensemble de la chaîne est opérationnel.

Comment choisir entre les deux voies

Un arbre de décision simple :

1. Vous souhaitez simplement exécuter du SQL, poser des questions en langage naturel sur des jeux de données gouvernés via Genie et effectuer des recherches dans des index vectoriels ? Utilisez la voie A. Les quatre serveurs gérés font exactement cela, et la configuration ne prend que 10 minutes.
2. Créer des pipelines, gérer des jobs, déployer des Asset Bundles, travailler avec Lakebase, développer des Databricks Apps ou appeler Mosaic AI / Agent Bricks ? Utilisez la voie B. La boîte à outils complète représente une surface trop vaste pour être greffée sous forme de serveurs MCP distants ponctuels.
3. Hybride ? Exécutez les deux. La voie A et la voie B ne sont pas en conflit. Le Power écrit sa propre entrée mcpServers.databricks, tandis que les quatre serveurs de la voie A (databricks-genie, databricks-sql, databricks-uc-functions, databricks-vector-search) sont des clés distinctes. Kiro les présente tous dans le panneau MCP.

Pour les développeurs qui prévoient d'utiliser Kiro au quotidien pour leurs charges de travail Databricks, la voie B est la meilleure solution à long terme. La voie A est la bonne réponse si vous disposez de 10 minutes et d'un SQL warehouse avec lequel vous souhaitez discuter.

Du point de vue du développeur

Les deux voies sont perçues différemment selon la personne qui utilise l'IDE. Quatre profils, quatre points de friction, quatre réponses.

L'analytics engineer. Vous passez la moitié de votre journée à interroger des tables que vous n'avez jamais vues et l'autre moitié à faire des copiers-coller entre votre éditeur et l'interface utilisateur de l'espace de travail. La voie A résout ce problème en 10 minutes. Les serveurs Genie et SQL basent chaque requête sur les métadonnées réelles du schéma ; l'assistant écrit en fonction de vos colonnes réelles, sans deviner ; et chaque résultat hérite de vos autorisations Unity Catalog. Vous arrêtez de passer d'un onglet à l'autre.

Le data engineer. Votre quotidien est fait de pipelines, de jobs, d'Asset Bundles et des promotions inter-environnements qui vont avec. Rédiger databricks.yml à la main et exécuter databricks bundle deploy depuis la barre latérale d'un terminal est la méthode lente. La voie B est la méthode rapide. Les compétences en pipelines + jobs + Asset Bundles du Power produisent, valident et déploient l'IaC à partir d'une seule conversation. Spark Declarative Pipelines, CDC, SCD Type 2, Auto Loader — tous générés à partir de vos tables UC réelles et prêts à être commités.

Le concepteur d'IA / d'agents. Vous connectez les appels de modèles, l'évaluation, la gouvernance et l'orchestration d'agents à travers trois ou quatre outils qui ne s'accordent pas tout à fait sur le schéma. La voie B couvre l'ensemble de l'offre Databricks AI — Mosaic AI Gateway pour le routage et les solutions de repli, Agent Bricks pour les superviseurs multi-agents et les assistants de connaissances, MLflow pour l'évaluation, Vector Search pour la recherche — le tout gouverné par UC de bout en bout. Votre agent hérite des mêmes autorisations que son appelant, et vos traces d'évaluation arrivent dans le même espace de travail que vos lancements d'entraînement.

Le concepteur de plateforme. Vous gérez les ressources Databricks en tant que code, effectuez des promotions entre dev/stage/prod et répondez à la question « y a-t-il eu une dérive ? » sur un rythme hebdomadaire. La compétence Asset Bundles de la voie B, associée aux compétences de gestion de Unity Catalog, génère le bundle complet, le valide par rapport à l'état réel de votre espace de travail et détecte la dérive avant qu'elle ne pose problème. Vous arrêtez de maintenir un ensemble de fichiers YAML manuellement et un autre dans un document.

Un workflow que vous pouvez exécuter dès aujourd'hui

Quelle que soit la voie choisie, cet exercice ancre l'assistant dans les métadonnées réelles de l'espace de travail à l'aide du catalogue samples.tpch, disponible dans chaque espace de travail Databricks.

Vous demandez : « Quelles sont les colonnes et les types de samples.tpch.lineitem, et quelle est la distribution des données par année l_shipdate ? »

Kiro renvoie le schéma réel et un histogramme à partir d'une seule requête exécutée par MCP. Noms de colonnes réels, distribution réelle, pas d'hallucinations.

Vous demandez : « Rédige un modèle dbt qui joint lineitem à orders et agrège le chiffre d'affaires par pays et par trimestre. »

Kiro produit du SQL en utilisant les noms de colonnes réels (l_extendedprice, l_discount, o_orderdate) au lieu de deviner. Comme il a d'abord interrogé le schéma, il connaît les types exacts et le niveau de granularité.

Vous demandez : « Exécute ma nouvelle agrégation sur samples.tpch et compare le nombre de lignes à l'instantané de la semaine dernière dans poc.gold.revenue_by_nation_qtr. »

Kiro exécute les deux requêtes et affiche la différence. Si un chiffre semble incorrect, « Montre-moi le lignage de gold.revenue_by_nation_qtr » extrait les tables en amont de system.access.table_lineage. Une fois vérifié, Kiro génère le JSON du job Databricks pour le modèle et liste les catalogues et schémas qu'il touche.

Sur la voie B, le même workflow s'étend à « Générer l'Asset Bundle pour ce job et le déployer en staging », ou « Créer un tableau de bord AI/BI basé sur cette agrégation », ou « Connecter cela à un point de terminaison Mosaic AI Gateway avec un modèle de repli », le tout sans quitter l'IDE. 

Bonnes pratiques, quelle que soit la voie

• Authentification par moindre privilège. Générez un PAT distinct par poste de travail et par ensemble de portées. Sur la voie B, préférez OAuth U2M aux PAT pour une utilisation interactive.
• Ne commitez jamais d'identifiants. Stockez DATABRICKS_ACCESS_TOKEN dans votre profil de shell ou dans un gestionnaire de secrets. Ne le placez jamais dans mcp.json soumis au contrôle de version.
• Définir la portée par projet. Conservez les ID d'espace Genie et les chemins d'index Vector Search dans $PWD/.kiro/settings/mcp.json afin que chaque projet contienne ses propres liaisons de ressources.
• Faites confiance aux autorisations UC. Toutes les voies appliquent les autorisations basées sur les lignes, les colonnes et les balises de Unity Catalog. L'IA hérite de vos autorisations effectives à chaque appel. Il n'y a pas de couche de contrôle d'accès distincte à gérer.
• Redémarrez, ne rechargez pas. Kiro lit les variables d'environnement une seule fois au démarrage du processus. Après avoir modifié votre profil de shell ou ajouté une authentification, quittez complètement (Cmd+Q sur macOS) et rouvrez.

Dépannage

« Serveur introuvable » ou statut rouge sur une entrée MCP.

Voie A : vérifiez echo $DATABRICKS_SQL_MCP_URL dans le shell qui a lancé Kiro. Une valeur vide signifie que Kiro ne peut pas résoudre l'URL. Confirmez que le mcp.json de portée espace de travail ne masque pas votre configuration de portée utilisateur. Vérifiez que le PAT est toujours valide dans Paramètres, Développeur, Jetons d'accès.

Voie B : saisissez à nouveau le flux de détection des identifiants depuis l'intégration du Power. Si le serveur MCP renvoie Invalid access token ou 401, le hook de récupération 401 intégré du Power suspend les appels d'outils et affiche à nouveau les options d'authentification.

Connecté au MCP mais les appels d'outils renvoient RESOURCE_DOES_NOT_EXIST ou PERMISSION_DENIED.

L'échec le plus courant de la voie A. La liaison d'initialisation (handshake) réussit avec une URL générique car le serveur diffère la validation des ressources jusqu'à l'appel. Réexécutez les vérifications préalables pour le serveur spécifique (l'espace Genie existe et est partagé avec vous, la fonction existe et vous disposez de EXECUTE, l'index vectoriel existe et vous disposez de l'autorisation d'affichage).

Essayez-le dès aujourd'hui. Le Databricks AI Dev Kit Power est le moyen le plus rapide d'obtenir la plateforme complète — pipelines, jobs, Lakebase, Mosaic AI, Agent Bricks et tout le reste mentionné ci-dessus — au sein de Kiro. Installez-le directement depuis le catalogue Kiro Powers sur github.com/kirodotdev/powers ou visitez github.com/databricks-solutions/ai-dev-kit pour installer la boîte à outils sous-jacente pour Kiro ou tout IDE pris en charge (Claude, Cursor, Copilot, Codex, Gemini). Pour les quatre serveurs MCP gérés par Databricks, l'installation en un clic se trouve sur kiro.dev/docs/mcp/servers/.

Vous avez des commentaires ou vous rencontrez un problème ? Signalez un problème : pour la configuration ou le packaging de Power, utilisez kirodotdev/powers ; pour l'installateur, le serveur MCP ou les compétences, utilisez databricks-solutions/ai-dev-kit. Nous les lisons tous.
 

Les opinions et idées partagées ici sont les nôtres et ne constituent pas une politique officielle de Databricks.

(Cet article de blog a été traduit à l'aide d'outils basés sur l'intelligence artificielle) Article original