Ai-agentverktyg för strukturerad hämtning
Viktig
Den här funktionen finns i offentlig förhandsversion.
Den här artikeln visar hur du skapar AI-agenter för strukturerad datahämtning med hjälp av Mosaic AI Agent Framework. Med strukturerade hämtningar kan agenter köra frågor mot strukturerade datakällor, till exempel SQL-tabeller.
Mer information om agentverktyg finns i Skapa AI-agentverktyg.
Frågeverktyg för tabell
I följande exempel skapas ett verktyg som gör att en agent kan fråga efter strukturerade kunddata från en Unity Catalog-tabell.
Den definierar en UC-funktion med namnet lookup_customer_info
, som gör att en AI-agent kan hämta strukturerade data från en hypotetisk customer_data
tabell.
Kör följande kod i en SQL-redigerare.
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ästa steg
När du har skapat ett agentverktyg lägger du till verktyget i en AI-agent. Se lägg till verktyg för Unity Catalog i agenter.