構造化された検索AIエージェントツール

この記事では、Mosaic AI Agent Framework を使用して構造化データを取得するための AI エージェント ツール を作成する方法について説明します。エージェントが SQL テーブルなどの構造化データソースをクエリできるようにするには、次のいずれかの方法を使用できます。

• SQL Unity Catalog関数 : エージェントが入力できるパラメーターを使用してSQLクエリを定義します。この方法は、クエリが既知の固定形式である場合に使用します。

Unity Catalog SQL 関数ツールを使用したデータのクエリ

Unity Catalog SQL Function を使用して構造化取得ツールを作成するには、クエリが事前にわかっていて、エージェントがパラメーターを提供します。

次の例では、 lookup_customer_infoという Unity Catalog 関数を作成し、AI エージェントが架空の customer_dataテーブルから構造化データを取得できるようにします。

SQL エディターで次のコードを実行します。

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;

次のステップ

ツールを作成したら、エージェントに追加します。「 エージェントへの Unity Catalog ツールの追加」を参照してください。