Outils d'agents d'IA pour la récupération structurée

Cet article explique comment créer un outil d'agent IA pour la récupération de données structurées en utilisant le Mosaic AI Agent Framework. Pour permettre aux agents d’interroger des sources de données structurées telles que des tables SQL, vous pouvez utiliser l’une des méthodes suivantes :

• Fonctions catalogue SQL Unity : définissez une requête SQL avec des paramètres que l’agent peut remplir. Utilisez cette méthode lorsque les requêtes ont un format connu et fixe.

Interroger des données à l’aide de l’outil de fonction SQL du catalogue Unity

Créez un outil de récupération structurée à l’aide de la fonction SQL catalogue Unity lorsque la requête est connue à l’avance et que l’agent fournit les paramètres.

L’exemple suivant crée une fonction De catalogue Unity appelée lookup_customer_info, qui permet à un agent IA de récupérer des données structurées à partir d’une table hypothétique customer_data .

Exécutez le code suivant dans un éditeur 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;

Après avoir créé un outil de catalogue Unity, ajoutez-le à votre agent. Consultez Créer un outil d’agent.

Système multi-agent Genie

Créez un système multi-agent qui inclut un agent Genie lorsque vous ne savez pas quel type de requêtes votre agent doit répondre et que vous avez besoin de la flexibilité nécessaire pour interroger plusieurs tables.

Pour plus d’informations, consultez Utiliser Genie dans les systèmes multi-agents.