チュートリアル:Unity Gatewayでコーディングエージェントのモデルアクセスを制御する方法

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

このチュートリアルの終わりまでに、次の内容が得られました。

  • 開発者のマシンにはプロバイダーAPIキーを持たずに、Azure Databricksの認証情報を使ってすべてのモデルリクエストをUnity Gateway経由で送信するコーディングエージェントです。
  • モデル サービスで設定したレート制限とアクセス許可に従って、エージェントの使用状況とコストの一元的な記録。

[前提条件]

手順 1: コーディング エージェントを接続する

エージェントは、次の 2 つの方法のいずれかで接続できます。

  • Unity Gateway CLI (推奨):OAuthを処理し、各エージェントの設定ファイル(モデルサービスベースURLやモデルを含む)を書き込むオープンソースランチャーです。 API キーまたはエンドポイントは手動で管理しません。
  • 手動設定:モデルサービスベースのURLを設定し、ツールの組み込み設定(Claude Code、Codex、カーソル、またはGemini CLI)で自分でモデル化します。 「サポートされたコーディングエージェント」を参照してください。

このセクションではUnity Gateway CLIを使用しています。 それをインストールし、必要なエージェントを実行します。

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

クロード符号

ug claude

コーデックス

ug codex

非インタラクティブモードで動作するには、エージェント自身のフラグをUnity Gateway CLIに通します:

ug codex --full-auto

Gemini CLI

ug gemini

OpenCode

ug opencode

Copilot

ug copilot

最初の実行では、Unity Gateway CLIがワークスペースのURLをプロンプトし認証します。その後の実行では直接エージェントに送られます。

手順 2: エージェントのモデル アクセスを管理する

他のUnityカタログのセキュアブルと同様にモデルサービスを管理できます。誰が使用できるかを EXECUTE し、Unityゲートウェイを使ってリクエストボリュームや支出を制限できます。 次のいずれかを設定します。

  • モデルにアクセスできるユーザーを管理する: カタログ エクスプローラーでモデル サービスを開き、[EXECUTE] タブでユーザーまたはグループにを付与します。
  • 上限要求ボリューム: カタログ エクスプローラーのモデル サービスの [概要] タブで、[レート制限] の [設定] をクリックします。 エンドポイント上のすべての要求またはユーザーごとに、1 分あたりの要求数 (QPM) またはトークン /分 (TPM) のレート制限 を設定します。
  • 支出予算を設定する:アカウント管理者はすべてのUnity Gatewayエンドポイントにスコープをかけた 予算 を作成できます。 予算は、モデル サービス全体の支出を集計します。アラートのしきい値を設定して通知するか、コストを超える前に使用状況をブロックします。

手順 3: トラフィックが管理されていることを確認する

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

  • エージェントから: プロンプトを送信します。 エージェントは、構成したモデルを使用して応答します。プロバイダー API キーはマシン上にありません。

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

    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ゲートウェイコストの分析」をご覧ください。

次のステップ

  • 最終更新: