Recuperação estruturada AI ferramentas de agente

Prévia

Esse recurso está em Pré-lançamento público.

Este artigo mostra como criar agentes AI para recuperação de dados estruturados usando o Mosaic AI Agent Framework. Os recuperadores estruturados permitem que os agentes consultem fontes de dados estruturadas, como as tabelas SQL.

Para saber mais sobre as ferramentas do agente, consulte Criar ferramentas do agente AI .

Ferramenta de consulta de tabela

O exemplo a seguir cria uma ferramenta que permite a um agente consultar dados estruturados do cliente em uma tabela Unity Catalog.

Ele define uma função UC chamada lookup_customer_info, que permite que um agente AI recupere dados estruturados de uma tabela customer_data hipotética.

Execute o código a seguir em um editor 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 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;

Próximos passos

Depois que o senhor criar uma ferramenta de agente, adicione a ferramenta a um agente AI. Consulte Adicionar ferramentas do Unity Catalog aos agentes.