Aller au contenu principal
Dernière mise à jour le

Connectez les agents aux données structurées

Les agents IA doivent souvent query ou manipuler des données structurées pour répondre à des questions, mettre à jour des enregistrements ou créer des pipelines de données.

Databricks propose plusieurs approches pour connecter les agents aux données structurées dans les tables Unity Catalog et les stockages de données externes. Utilisez des serveurs MCP préconfigurés pour un accès immédiat aux Genie Agents et aux SQL Warehouse, ou créez des outils personnalisés pour des workflows spécialisés.

Cette page montre comment :

Interroger les données dans les tables Unity Catalog​

Si votre agent a besoin de query des données dans les tables Unity Catalog, Databricks vous recommande d'utiliser les Genie Agents. Un Genie Agent est une collection de jusqu'à 25 tables Unity Catalog que Genie peut garder en contexte et query en utilisant le langage naturel. Les agents peuvent accéder au Genie Agent en utilisant une URL MCP préconfigurée.

Pour se connecter à un Genie Agent :

  1. Créez un Genie Agent avec les tables que vous souhaitez interroger et partagez l'agent avec les utilisateurs, ou les Service Principal, qui doivent y accéder. Voir Créer et gérer un Genie Agent.
  2. Créez un agent et connectez-le à l'URL MCP gérée et préconfigurée pour l'espace : https://<workspace-hostname>/api/2.0/mcp/genie/{genie_space_id}.
remarque

Le serveur MCP géré pour Genie invoque Genie comme un outil MCP, ce qui signifie que l'historique n'est pas transmis lors de l'invocation des APIs Genie.

Ajouter un outil Genie Agent à votre agent​

Les exemples suivants montrent comment connecter votre agent à un serveur MCP Genie Agent. Remplacez <genie-space-id> par l'ID de votre Genie Agent.

Python
from agents import Agent, Runner
from databricks.sdk import WorkspaceClient
from databricks_openai.agents import McpServer

workspace_client = WorkspaceClient()
host = workspace_client.config.host

async with McpServer(
    url=f"{host}/api/2.0/mcp/genie/<genie-space-id>",
    name="genie-space",
    workspace_client=workspace_client,
) as genie_server:
    agent = Agent(
        name="Data analyst agent",
        instructions="You are a data analyst. Use the Genie tool to query structured data and answer questions.",
        model="databricks-claude-sonnet-4-5",
        mcp_servers=[genie_server],
    )
    result = await Runner.run(agent, "What were the top 10 customers by revenue last quarter?")
    print(result.final_output)

Accorder l'accès à l'application au Genie Agent dans databricks.yml:

YAML
resources:
  apps:
    my_agent_app:
      resources:
        - name: 'my_genie_space'
          genie_space:
            space_id: '<genie-space-id>'
            permission: 'CAN_RUN'

Système multi-agent Genie​

info

Aperçu

Cette fonctionnalité est en Aperçu public.

Pour les systèmes multi-agents avancés, vous pouvez également utiliser Genie comme agent plutôt que de l'intégrer via MCP. Lorsque vous appelez Genie en tant qu'agent, vous pouvez transférer de manière déterministe le contexte de conversation existant à Genie.

Pour une approche axée sur le code, consultez Utiliser Genie dans les systèmes multi-agents (Model Serving). Pour une approche axée sur l'interface utilisateur, consultez Utiliser l'agent superviseur pour créer un système multi-agents coordonné.

Interrogez les données à l'aide de l'outil de fonction SQL de Unity Catalog​

Créez un outil de récupération structuré en utilisant les fonctions SQL de Unity Catalog lorsque la query est connue à l'avance et que l'agent fournit les parameters.

L'exemple suivant crée une fonction Unity Catalog appelée lookup_customer_info, qui permet à un agent d'IA de récupérer des données structurées à partir d'une table hypothétique customer_data.

Exécutez le code suivant dans un éditeur SQL.

SQL
CREATE OR REPLACE FUNCTION main.default.lookup_customer_info(
  customer_name_input STRING COMMENT 'Name of the customer whose info to look up'
)
RETURNS STRING
COMMENT 'Returns metadata about a particular customer, given the customer''s name, including the customer''s email and ID. The
customer ID can be used for other queries.'
RETURN SELECT CONCAT(
    'Customer ID: ', customer_id, ', ',
    'Customer Email: ', customer_email
  )
  FROM main.default.customer_data
  WHERE customer_name = customer_name_input
  LIMIT 1;

Après avoir créé un outil Unity Catalog, ajoutez-le à votre agent. Consultez Créer un outil de fonction Unity Catalog.

Sur cette page