Tools für KI-Agenten zur strukturierten Informationsabfrage

In diesem Artikel wird gezeigt, wie Sie KI-Agent-Tools für den strukturierten Datenabruf mithilfe des Mosaik AI Agent Frameworks erstellen. Damit Agents strukturierte Datenquellen wie SQL-Tabellen abfragen können, können Sie eine der folgenden Methoden verwenden:

• SQL Unity-Katalogfunktionen: Definieren Sie eine SQL-Abfrage mit Parametern, die der Agent ausfüllen kann. Verwenden Sie diese Methode, wenn Abfragen ein bekanntes, festes Format aufweisen.

Abfragen von Daten mithilfe des SQL-Funktionstools "Unity Catalog"

Erstellen Sie ein strukturiertes Abruftool mithilfe der Unity Catalog SQL-Funktion, wenn die Abfrage rechtzeitig bekannt ist und der Agent die Parameter bereitstellt.

Im folgenden Beispiel wird die Funktion lookup_customer_info im Unity Catalog erstellt, die es einem KI-Agenten 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'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;

Nachdem Sie ein Unity-Katalogtool erstellt haben, fügen Sie es Ihrem Agent hinzu. Siehe Erstellen eines Agent-Tools.

Genie Multi-Agent-System

Erstellen Sie ein Multi-Agent-System, das einen Genie-Agent enthält, wenn Sie nicht wissen, welche Art von Abfragen Ihr Agent beantworten muss, und Sie benötigen die Flexibilität, mehrere Tabellen abzufragen.

Weitere Informationen finden Sie unter Verwendung von Genie in Multi-Agent-Systemen.