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.

• Sistemas multiagentes Genie : Crie um sistema multiagente que inclua um agente Genie que possa usar a linguagem natural para raciocinar e consultar tabelas. Use esse método quando precisar de flexibilidade para responder a vários tipos de consultas em várias tabelas e formatos variados.

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;

Sistema multiagente Genie

info

Visualização

Esse recurso está em Public Preview.

Crie um sistema multiagente que inclua um agente Genie quando o senhor 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 Use Genie in multi-agent systems.

Próximas etapas

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