Ferramentas de IA para recuperação estruturada

Este artigo mostra como criar ferramentas de agente de IA para recuperação de dados estruturados usando o Mosaic AI Agent Framework. Para permitir que os agentes consultem fontes de dados estruturadas, como tabelas SQL, você pode usar um dos seguintes métodos:

• Funções do Catálogo SQL Unity: defina uma consulta SQL com parâmetros que o agente pode preencher. Use esse método quando as consultas tiverem um formato conhecido e fixo.

Consultar dados usando a ferramenta de função SQL do Unity Catalog

Crie uma ferramenta de recuperação estruturada usando a função SQL do Unity Catalog quando a consulta for conhecida com antecedência 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 de IA recupere dados estruturados de uma tabela hipotética customer_data .

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'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 de catálogo Unity, adicione-a ao seu agente. Consulte Criar uma ferramenta de agente.

Sistema multiagente Genie

Crie um sistema multiagente que inclua um agente Genie quando você não souber que tipo de consultas seu agente precisa responder e precisar de flexibilidade para consultar várias tabelas.

Para obter mais informações, consulte Usar o Genie em sistemas multiagentes.