Recuperação estruturada AI ferramentas de agente

Este artigo mostra como criar ferramentas de agenteAI para recuperação de dados estruturados usando o Mosaic AI Agent Framework. Para permitir que os agentes consultem fontes de dados estruturadas, como as tabelas SQL, o senhor pode usar um dos métodos a seguir:

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

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

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 AI recupere dados estruturados de uma tabela customer_datahipotética.

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;

Próximas etapas

Depois de criar a ferramenta, adicione-a a um agente. Consulte Adicionar ferramentas do Unity Catalog aos agentes.