Herramientas para agentes de IA de búsqueda estructurada

En este artículo se muestra cómo crear herramientas de agente de IA para la recuperación de datos estructurados mediante el marco del agente de IA de Mosaico. Para permitir que los agentes consulten orígenes de datos estructurados, como tablas SQL, puede usar uno de los métodos siguientes:

• Funciones del catálogo de SQL Unity: defina una consulta SQL con parámetros que el agente pueda rellenar. Use este método cuando las consultas tengan un formato fijo conocido.

Consultar datos usando la herramienta de funciones SQL del catálogo de Unity

Cree una herramienta de recuperación estructurada con la función SQL del catálogo de Unity cuando la consulta se conozca con antelación y el agente proporcione los parámetros.

En el ejemplo siguiente se crea una función de catálogo de Unity denominada lookup_customer_info, que permite a un agente de IA recuperar datos estructurados de una tabla hipotética customer_data .

Ejecute el código siguiente en un editor de 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;

Después de crear una herramienta de catálogo de Unity, agréguela al agente. Consulte Creación de una herramienta de agente.

Sistema multiagente de Genie

Importante

Esta característica está en versión preliminar pública.

Cree un sistema multiagente que incluya un agente de Genie cuando no sepa qué tipo de consultas necesita responder el agente y necesita la flexibilidad de consultar varias tablas.

Para obtener más información, consulte Uso de Genie en sistemas multiagente.