AI エージェントは、多くの場合、質問に回答したり、レコードを更新したり、データ パイプラインを作成したりするために、構造化データのクエリまたは操作を行う必要があります。
Databricks には、Unity カタログ テーブルと外部データ ストア内の構造化データにエージェントを接続するための複数の方法が用意されています。 Genie スペースと SQL ウェアハウスにすぐにアクセスしたり、特殊なワークフロー用のカスタム ツールを構築したりするには、事前に構成された MCP サーバーを使用します。
このページでは、次の方法を示します。
Unity カタログ テーブルのデータに対してクエリを実行する
エージェントが Unity カタログ テーブルのデータに対してクエリを実行する必要がある場合、Databricks では Genie スペースを使用することをお勧めします。 Genie スペースは、Genie が自然言語を使用してコンテキストとクエリを実行できる最大 25 個の Unity カタログ テーブルのコレクションです。 エージェントは、事前に構成された MCP URL を使用して Genie 空間にアクセスできます。
Genie スペースに接続するには:
- クエリを実行するテーブルを含む Genie 空間を作成し、その領域にアクセスする必要があるユーザーまたはサービス プリンシパルと共有します。 AI/BI Genie 空間の設定と管理に関するページを参照してください。
- エージェントを作成し、スペースの事前構成済みのマネージド MCP URL (
https://<workspace-hostname>/api/2.0/mcp/genie/{genie_space_id}) に接続します。
管理対象 MCP サーバーに接続するエージェントを作成する方法については、「 Databricks マネージド MCP サーバーの使用」を参照してください。
注
Genie 用のマネージド MCP サーバーは Genie を MCP ツールとして呼び出します。つまり、Genie API を呼び出すときに履歴が渡されません。
Genie マルチエージェント システム
重要
この機能はパブリック プレビュー段階にあります。
高度なマルチエージェント システムの場合は、MCP を使用して統合するのではなく、Genie をエージェントとして使用することもできます。 Genie をエージェントとして呼び出すと、既存の会話コンテキストを Genie に確定的に渡すことができます。
コード優先のアプローチについては、「 マルチエージェント システムでの Genie の使用 (モデル サービス)」を参照してください。 UI 優先のアプローチについては、「 エージェント ブリック: Supervisor Agent を使用して、調整されたマルチエージェント システムを作成する」を参照してください。
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;
Unity カタログ ツールを作成したら、エージェントに追加します。 エージェント ツールの作成を参照してください。