Partilhar via

Ferramentas de agente de IA de recuperação estruturada

  • Artigo

Importante

Este recurso está em Public Preview.

Este artigo mostra como criar agentes de IA 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 tabelas SQL.

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

Ferramenta de consulta de tabela

O exemplo a seguir cria uma ferramenta que permite que um agente consulte dados estruturados do cliente a partir de uma tabela do Unity Catalog.

Ele define uma função de UC chamada lookup_customer_info, que permite que um agente de IA 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 de criar uma ferramenta de agente, adicione a ferramenta a um agente de IA. Consulte Adicionar ferramentas do Catálogo Unity aos agentes.