Ferramentas do agente 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 do SQL Unity: defina uma consulta SQL com parâmetros que o agente pode preencher. Use esse método quando as consultas tiverem um formato fixo conhecido.

Consultar dados usando a ferramenta de função SQL do Catálogo do Unity

Crie uma ferramenta de recuperação de dados estruturada usando a Função SQL do Unity Catalog quando uma consulta é conhecida com antecedência e o agente fornece os parâmetros.

O exemplo a seguir cria uma função de Catálogo do Unity 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 do 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 Catálogo do Unity, adicione-a ao agente. Consulte Criar uma ferramenta de agente.

Sistema multi-agente do Genie

Crie um sistema de vários agentes que inclua um agente do Genie quando você não sabe que tipo de consultas seu agente precisa responder e você precisa da flexibilidade para consultar várias tabelas.

Para obter mais informações, consulte Usar o Genie em sistemas de vários agentes.