Trae からDatabricks のデータにリアルタイムでアクセスする方法(CData Connect AI)
Trae は、使い慣れたエディターと Builder や SOLO などのエージェントモードを組み合わせた AI 対応の統合開発環境(IDE)です。Model Context Protocol(MCP) に対応しており、外部ツールやデータソースを追加すると、エージェントがリアルタイムデータにアクセスできるようになります。
Trae と CData Connect AI を組み込みの MCP Server 経由で連携させることで、Trae はガバナンスの効いたリアルタイムの Databricks のデータ にアクセスできるようになります。自然言語プロンプトを使って Databricks のデータ のカタログを一覧表示したり、スキーマを確認したり、レコードをクエリしたりでき、すべてのデータアクセスは許可されたデータソースに対して安全に実行されます。
この記事では、Connect AI で Databricks 接続を設定し、必要なパーソナルアクセストークンを生成し、Trae をインストールして Connect AI MCP Server を追加し、LLM モデルを設定した上で、Trae エージェントからリアルタイムの Databricks のデータ にクエリを実行して連携を検証する方法を説明します。
Databricks とのデータ連携について
CData を使えば、Databricks のリアルタイムデータへのアクセスと統合を簡単に実現できます。CData の接続機能は、主に次のような用途で活用されています。
- Runtime バージョン 9.1 - 13.X から Pro および Classic Databricks SQL バージョンまで、すべてのバージョンの Databricks へのアクセス
- あらゆるホスティングソリューションとの互換性による、お好みの環境での Databricks の利用継続
- パーソナルアクセストークン、Azure サービスプリンシパル、Azure AD など、さまざまな方法での安全な認証
- Databricks ファイルシステム、Azure Blob ストレージ、AWS S3 ストレージを使用した Databricks へのデータアップロード
多くのお客様が、さまざまなシステムから Databricks データレイクハウスにデータを移行するために CData のソリューションを使用していますが、ライブ接続ソリューションを使用して、データベースと Databricks 間の接続をフェデレートしているお客様も多数います。これらのお客様は、SQL Server リンクサーバーまたは Polybase を使用して、既存の RDBMS 内から Databricks へのライブアクセスを実現しています。
関連リンク
Step 1: Trae 向けに Databricks の接続を設定
Trae は Connect AI の Remote MCP Server を介して Databricks に接続します。Trae から Databricks のデータ を操作するには、まず Connect AI で Databricks 接続を作成・設定します。
- Connect AI にログインし、Sources をクリックしてから Add Connection をクリックします。
- Add Connection パネルから Databricks を選択します。
-
Databricks への接続に必要な認証プロパティを入力します。
Databricks 接続プロパティの取得・設定方法
Databricks クラスターに接続するには、以下のプロパティを設定します。
- Database:Databricks データベース名。
- Server:Databricks クラスターのサーバーのホスト名。
- HTTPPath:Databricks クラスターのHTTP パス。
- Token:個人用アクセストークン。この値は、Databricks インスタンスのユーザー設定ページに移動してアクセストークンタブを選択することで取得できます。
Databricks への認証
CData は、次の認証スキームをサポートしています。
- 個人用アクセストークン
- Microsoft Entra ID(Azure AD)
- Azure サービスプリンシパル
- OAuthU2M
- OAuthM2M
個人用アクセストークン
認証するには、次を設定します。
- AuthScheme:PersonalAccessToken。
- Token:Databricks サーバーへの接続に使用するトークン。Databricks インスタンスのユーザー設定ページに移動してアクセストークンタブを選択することで取得できます。
その他の認証方法については、ヘルプドキュメント の「はじめに」セクションを参照してください。
- Save & Test をクリックします。
- Permissions タブに移動し、ユーザーベースの権限を更新します。
Personal Access Token の追加
Personal Access Token(PAT)は、Trae から Connect AI への接続を認証するために使用されます。アクセス制御を細かく管理するために、連携ごとに個別の PAT を作成することをお勧めします。
- Connect AI アプリの右上にある歯車アイコン()をクリックして Settings を開きます。
- Settings ページで Access Tokens セクションに移動し、 Create PAT をクリックします。
- PAT にわかりやすい名前を付けて Create をクリックします。
- トークンが表示されたらコピーして安全に保管してください。再度表示されることはありません。
Databricks 接続の設定と PAT の生成が完了したら、Trae は Connect AI 経由で Databricks のデータ に接続できるようになります。
Step 2: Trae をインストールして Connect AI MCP Server を設定
次に、Trae をインストールし、Connect AI Remote MCP Server を追加して、LLM モデルを設定します。設定が完了すると、エージェントは Connect AI を通じてリアルタイムデータツールを検出・呼び出せるようになります。
- Trae IDE をダウンロードしてインストールし、アプリケーションを起動します。
-
画面左上のトグルを使用するか、Ctrl + Alt + \ を押して SOLO モードに切り替えます。
-
Toggle AI Sidebar をクリックしてチャットパネルを開きます。
-
Settings を開き、左メニューから MCP を選択します。
-
Add Manually をクリックします。
-
Configure Manually ダイアログに以下の設定を貼り付けて Confirm をクリックします:
{ "mcpServers": { "cdata-connect-ai": { "type": "streamable-http", "url": "https://mcp.cloud.cdata.com/mcp", "headers": { "Authorization": "Basic your_base64_encoded_email_PAT" } } } }注意:Trae は Connect AI との認証に Basic 認証を使用します。Connect AI のユーザーメールアドレスと先ほど作成した PAT を email:PAT の形式で結合し、Base64 エンコードした上で Basic を先頭に付けます。たとえば [email protected]:ABC123...XYZ789 の場合、Authorization ヘッダーの値は次のようになります:Basic dXNlckBkb21haW4uY29tOkFCQzEyMy4uLlhZWjc4OQ==
LLM モデルの設定
Trae がエージェントの推論を行うには、少なくとも 1 つの LLM モデルが必要です。エージェントがプロンプトを解釈し、Connect AI 経由で MCP ツールを呼び出せるようにモデルを追加します。
- Settings に戻り、Models を選択します。
-
Add Model をクリックし、OpenAI、Anthropic、Google などのプロバイダーを選択して、モデルを選択し、API キーを入力して Add Model をクリックします。
MCP サーバーの追加と LLM モデルの設定が完了すると、Trae は Connect AI を通じてリアルタイムの Databricks のデータ にクエリを実行する準備が整います。
Step 3: Trae からリアルタイムの Databricks のデータ にクエリ
連携が完了したら、Trae エージェントを使用して、設定した LLM が処理する自然言語プロンプトを通じてリアルタイムの Databricks のデータ を操作します。
-
チャットパネルで @ と入力し、Builder with MCP を選択します。Tools - MCP の下に cdata-connect-ai が表示されていることを確認します。
-
データを操作するプロンプトを入力します。例:
- cdata-connect-ai のカタログを一覧表示して
- Databricks で利用可能なスキーマとテーブルを表示して
- Databricks のデータ のテーブルから上位 5 件のレコードをクエリして
-
Trae は Connect AI MCP Server を呼び出し、Databricks のデータ からリアルタイムの結果を返します。
この時点で、Trae エージェントは Connect AI MCP Server と通信し、リモート MCP ツールを通じて IDE から直接リアルタイムの Databricks のデータ を取得します。
CData Connect AI を入手
クラウドアプリケーションから数百種類の SaaS、ビッグデータ、NoSQL データソースに直接アクセスするには、CData Connect AI をぜひお試しください。 14日間の無料トライアルをダウンロードいただけます。ご不明な点がございましたら、サポートチームまでお気軽にお問い合わせください。