Udostępnij przez


Narzędzia agenta sztucznej inteligencji do wyszukiwania opartego na strukturach

W tym artykule pokazano, jak utworzyć narzędzia agenta AI do pobierania danych ustrukturyzowanych przy użyciu Framework Mosaic AI Agent. Aby umożliwić agentom wykonywanie zapytań o ustrukturyzowane źródła danych, takie jak tabele SQL, można użyć jednej z następujących metod:

  • funkcje katalogu SQL Unity: Zdefiniuj zapytanie SQL z parametrami, które agent może wypełnić. Użyj tej metody, gdy zapytania mają znany, stały format.

Zapytanie danych z wykorzystaniem funkcji SQL Katalogu Unity

Utwórz strukturalne narzędzie wyszukiwania przy użyciu funkcji SQL Unity Catalog, gdy zapytanie jest znane z wyprzedzeniem, a agent udostępnia parametry.

Poniższy przykład tworzy funkcję Katalogu Unity o nazwie lookup_customer_info, która umożliwia agentowi sztucznej inteligencji pobieranie danych ustrukturyzowanych z hipotetycznej tabeli customer_data .

Uruchom następujący kod w edytorze SQL.

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;

Po utworzeniu narzędzia Unity Catalog dodaj je do agenta. Zobacz Tworzenie narzędzia agenta.

Genie system wieloagentowy

Ważne

Ta funkcja jest dostępna w publicznej wersji testowej.

Utwórz system z wieloma agentami, który zawiera agenta Genie, gdy nie wiesz, jakiego rodzaju zapytania agent musi odpowiedzieć, i potrzebujesz elastyczności w zakresie wykonywania zapytań dotyczących wielu tabel.

Aby uzyskać więcej informacji, zobacz Użycie Genie w systemach multi-agentowych.