Dela via


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.