Pular para o conteúdo principal
Última atualização em

Conectar agentes a dados estruturados

Os agentes AI frequentemente precisam consultar ou manipular dados estruturados para responder a perguntas, atualizar registros ou criar fluxos de dados.

O Databricks oferece diversas abordagens para conectar agentes a dados estruturados em tabelas do Unity Catalog e em armazenamentos de dados externos. Utilize servidores MCP pré-configurados para acesso imediato aos Genie spaces e ao banco de dados SQL , ou crie ferramentas personalizadas para fluxos de trabalho especializados.

Esta página mostra como:

Consultar dados nas tabelas Unity Catalog​

Se o seu agente precisar consultar dados em tabelas Unity Catalog , Databricks recomenda o uso Genie spaces. Um espaço Genie é uma coleção de até 25 tabelas do Unity Catalog que o Genie pode manter em contexto e consultar usando linguagem natural. Os agentes podem acessar o espaço Genie usando um URL MCP pré-configurado.

Para se conectar a um espaço Genie:

  1. Crie um espaço Genie com as tabelas que deseja consultar e compartilhe o espaço com os usuários ou a entidade de serviço que precisam acessá-lo. Consulte Configurar e gerenciar um espaço AI/BI Genie.
  2. Crie um agente e conecte-o ao URL MCP pré-configurado para o espaço: https://<workspace-hostname>/api/2.0/mcp/genie/{genie_space_id}.

Para aprender como criar um agente que se conecta a servidores MCP de gerenciamento, consulte Usar servidores MCP de gerenciamento do Databricks.

nota

O servidor MCP de gerenciamento para Genie invoca Genie como uma ferramenta MCP, o que significa que o histórico não é passado ao invocar APIs Genie .

Consultar uso de dados Unity Catalog SQL ferramenta de função​

Crie uma ferramenta de recuperação estruturada usando funções SQL do Unity Catalog quando a consulta for conhecida antecipadamente e o agente fornecer os parâmetros.

O exemplo a seguir cria uma função Unity Catalog chamada lookup_customer_info, que permite que um agente AI recupere dados estruturados de uma tabela hipotética customer_data .

Execute o código a seguir em um editor SQL.

SQL
CREATE OR REPLACE FUNCTION main.default.lookup_customer_info(
  customer_name 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
  LIMIT 1;

Depois de criar uma ferramenta do Unity Catalog, adicione-a ao seu agente. Consulte Criar uma ferramenta de agente.