Tools für strukturiertes Abrufen des KI-Agents
Wichtig
Dieses Feature befindet sich in der Public Preview.
In diesem Artikel wird gezeigt, wie KI-Agents für den strukturierten Datenabruf mithilfe des Mosaic AI Agent Frameworks erstellt werden. Strukturierte Retriever ermöglichen Es Agents, strukturierte Datenquellen wie SQL-Tabellen abzufragen.
Weitere Informationen zu Agenttools finden Sie unter KI-Agent-Tools.
Tabellenabfragetool
Im folgenden Beispiel wird ein Tool erstellt, mit dem ein Agent strukturierte Kundendaten aus einer Unity-Katalogtabelle abfragen kann.
Es definiert eine UC-Funktion namens lookup_customer_info
, die es einem KI-Agent ermöglicht, strukturierte Daten aus einer hypothetischen customer_data
Tabelle abzurufen.
Führen Sie den folgenden Code in einem SQL-Editor aus.
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;
Nächste Schritte
Nachdem Sie ein Agent-Tool erstellt haben, fügen Sie das Tool einem KI-Agent hinzu. Weitere Informationen finden Sie unter Hinzufügen von Unity Catalog-Tools zu Agents.