Partager via

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

  • Article

Important

Cette fonctionnalité est disponible en préversion publique.

Cet article explique comment créer des agents d'IA pour la récupération de données structurées à l'aide du Mosaic AI Agent Framework. Les récupérateurs structurés permettent aux agents d’interroger des sources de données structurées telles que des tables SQL.

Pour en savoir plus sur les outils d’agent, consultez outils d'IA pour agents.

Outil de requête de table

L’exemple suivant crée un outil qui permet à un agent d’interroger des données client structurées à partir d’une table de catalogue Unity.

Il définit une fonction UC appelée lookup_customer_info, qui permet à un agent IA de récupérer des données structurées à partir d’une table customer_data hypothétique.

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 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;

Étapes suivantes

Après avoir créé un outil d’assistant, ajoutez l’outil à un assistant IA. Consultez Ajouter des outils de catalogue Unity aux agents.