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

プレビュー

この機能はパブリックプレビュー段階です。

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

エージェント ツールの詳細については、「 AI エージェント ツールを作成する」を参照してください。

テーブル クエリ ツール

次の例では、エージェントが Unity Catalog テーブルから構造化された顧客データをクエリできるツールを作成します。

これは、 lookup_customer_infoと呼ばれる UC 関数を定義し、AI エージェントが架空の customer_data テーブルから構造化データを取得できるようにします。

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 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;

次のステップ

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