Ai-agentverktyg för strukturerad hämtning

Den här artikeln visar hur du skapar AI-agentverktyg för strukturerad datahämtning med hjälp av Mosaic AI Agent Framework. Om du vill tillåta agenter att fråga strukturerade datakällor, till exempel SQL-tabeller, kan du använda någon av följande metoder:

• SQL Unity Catalog-funktioner: Definiera en SQL-fråga med parametrar som agenten kan fylla i. Använd den här metoden när frågor har ett känt, fast format.

Hämta data med Unity Catalogs SQL-funktionsverktyg

Skapa ett strukturerat hämtningsverktyg med hjälp av Unity Catalog SQL Function när frågan är känd i förväg och agenten tillhandahåller parametrarna.

I följande exempel skapas en Unity Catalog-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'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;

När du har skapat ett Unity Catalog-verktyg lägger du till det i din agent. Se Skapa ett agentverktyg.

Genie multiagentssystem

Skapa ett system med flera agenter som innehåller en Genie-agent när du inte vet vilken typ av frågor din agent behöver besvara, och du behöver flexibiliteten för att köra frågor mot flera tabeller.

Mer information finns i Använda Genie i system med flera agenter.