Alat agen AI untuk pencarian terstruktur

Artikel ini memperlihatkan cara membuat alat agen AI untuk pengambilan data terstruktur menggunakan Mosaic AI Agent Framework. Untuk mengizinkan agen mengkueri sumber data terstruktur seperti tabel SQL, Anda bisa menggunakan salah satu metode berikut:

• fungsi SQL Unity Catalog: Tentukan kueri SQL dengan parameter yang dapat diisi agen. Gunakan metode ini ketika kueri memiliki format tetap yang diketahui.

Mengkueri data menggunakan alat fungsi SQL Unity Catalog

Buat alat pengambilan terstruktur menggunakan Fungsi SQL Katalog Unity saat kueri diketahui sebelumnya dan agen menyediakan parameter.

Contoh berikut membuat fungsi Katalog Unity yang disebut lookup_customer_info, yang memungkinkan agen AI untuk mengambil data terstruktur dari tabel customer_data hipotetis.

Jalankan kode berikut di editor 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;

Sistem multi-agen Genie

Penting

Fitur ini ada di Pratinjau Publik.

Buat sistem multi-agen yang menyertakan agen Genie saat Anda tidak tahu jenis kueri apa yang perlu dijawab agen Anda, dan Anda memerlukan fleksibilitas untuk mengkueri beberapa tabel.

Untuk informasi selengkapnya, lihat Penggunaan Genie dalam sistem multi-agen.

Langkah berikutnya

Setelah Anda membuat alat tersebut, tambahkan alat tersebut ke agen. Lihat Menambahkan alat Katalog Unity ke agen.