次の方法で共有

構造化取得 AI エージェント ツール

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

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

Unity カタログ SQL 関数ツールを使用してデータのクエリを する

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

次の例では、lookup_customer_infoと呼ばれる Unity Catalog 関数を作成します。これにより、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'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;

Genie マルチエージェント システム

重要

この機能はパブリック プレビュー段階にあります。

エージェントが応答する必要があるクエリの種類がわからない場合や、複数のテーブルに対して柔軟にクエリを実行できる必要がある場合に、Genie エージェントを含むマルチエージェント システムを作成します。

詳細については、「マルチエージェント システムでの Genie の使用」を参照してください。

次のステップ

ツールを作成したら、エージェントに追加します。 「エージェントに Unity カタログ ツールを追加する」をご覧ください。