メインコンテンツまでスキップ
に最終更新

チュートリアル: Unity Gateway を使用したコーディングエージェントのモデルアクセスの管理

このチュートリアルでは、Claude Code、Codex、Cursor、Gemini CLIなどの外部コーディングエージェントを、Unity Gatewayによって管理される大規模言語モデル（LLM）に接続します。その後、それらのモデルにアクセスできるユーザーや、エージェントが消費して費やすことのできる量を制御します。各開発者がプロバイダーのAPIキーをツールに貼り付ける代わりに、すべてのリクエストがUnity Gatewayのモデルサービスを経由してルーティングされます。Unity Catalogはどのモデルを使用できるかを制御し、Unity Gatewayは制限の適用、使用量およびコストの記録を行います。

このチュートリアルを終えるまでに、以下を達成しました:

  • 開発者のマシン上にプロバイダーのAPIキーを置くことなく、Databricksの認証情報を使用してすべてのモデルリクエストをUnity Gateway経由で送信するコーディングエージェント。
  • モデルサービスで設定したレート制限と権限に従って、エージェントの使用状況とコストの集中記録。

前提条件​

  • Unity Catalog が有効になっているワークスペース。Unity Catalog の概要を参照してください。
  • トラフィックのルーティング先となるモデルサービス、およびその上での EXECUTE。Databricksでホストされるモデルは、すぐに使用できるモデルサービスとして提供されています。モデルAPI（モデルサービス）のガバナンスを参照してください。自身で管理するモデルをガバナンスするには、まずモデルを作成してください。モデルAPI（モデルサービス）の作成と管理を参照してください。
  • ワークスペースに認証されたDatabricks CLI。
  • DatabricksワークスペースのURL（形式はhttps://<workspace-url>）です。

ステップ 1: コーディング エージェントを接続する​

エージェントを接続する方法は2つあります。

  • Unity Gateway CLI (推奨) : OAuthを処理し、モデルサービスベースURLやモデルを含む各エージェントの設定ファイルを書き込むオープンソースのランチャー。APIキーやEndpointを手動で管理することはありません。
  • 手動構成 ：モデル サービス ベース URL とモデルご自身でツールの組み込み設定（Claude Code、Codex、Cursor、または Gemini CLI）で設定します。コーディングエージェントとの統合を参照してください。

このセクションでは、Unity Gateway CLIを使用します。インストールしてから、目的のエージェントを実行します：

Bash
uv tool install git+https://github.com/databricks/unity-gateway
Bash
ug claude

初回**ラン**時に、Unity Gateway CLI は**ワークスペース** URL の入力を求め、認証を行います。2 回目以降の**ラン**では直接エージェントに接続されます。

ステップ 2: エージェントのモデルアクセスを管理する​

モデルサービスは、他のUnity Catalogのセキュリティ保護可能なアイテムと同様に管理できます。EXECUTEを付与して使用できるユーザーを制御し、Unity Gatewayを使用してリクエスト量と支出に上限を設定します。以下のいずれかを設定します。

  • モデルにアクセスできるユーザーを管理する ：Catalog Explorerでモデルサービスを開き、「 権限 」tabでユーザーまたはグループにEXECUTEを付与します。
  • リクエストボリュームの制限 : Catalog Explorer で、モデルサービスの 概要 tab の レート制限 の下にある 設定 をクリックします。Endpointでのすべてのリクエスト、またはユーザーごとの1分あたりのリクエスト数 (QPM) または1分あたりのトークン数 (TPM) のレート制限を設定します。
  • 支出予算の設定 : アカウント管理者は、すべてのUnity Gateway Endpointを対象とした予算を作成できます。予算はモデルサービス全体の支出を集計します。コストが超過する前に通知を受け取る、または使用をブロックするためのアラートthresholdを設定してください。

ステップ 3：トラフィックが管理されていることを確認する​

エージェントのリクエストが Unity Gateway を経由してルーティングされ、記録されていることを確認します：

  • エージェントから ：プロンプトを送信します。エージェントは、マシンにプロバイダーのAPIキーがなくても、構成したモデルを使用して応答します。

  • 使用状況システムテーブルから : 使用状況追跡をクエリーして、モデルサービスがリクエストを記録していることを確認します。

    SQL
    SELECT service_name, requester, status_code, COUNT(*) AS calls
    FROM system.ai_gateway.usage
    WHERE service_type = 'MODEL_SERVICE'
    GROUP BY service_name, requester, status_code
    ORDER BY calls DESC;

支出をより詳細に分析するには、「Unity Gateway コストの分析」を参照してください。

次のステップ​

このページの見出し