Tutorial: Controlla l'accesso al modello di un agente di codifica con Unity Gateway

In questo tutorial, colleghi un agente di codifica esterno, come Claude Code, Codex, Cursor o Gemini CLI, a grandi modelli linguistici (LLM) governati da Unity Gateway. È quindi possibile controllare chi può accedere a tali modelli e quanto l'agente può utilizzare e spendere. Invece che ogni sviluppatore incolli una chiave API provider nel proprio strumento, ogni richiesta passa attraverso un servizio modello Unity Gateway. Unity Catalog controlla chi può utilizzare quali modelli, e Unity Gateway applica i limiti e registra l'uso e i costi.

Al termine di questa esercitazione, si dispone di:

  • Un agente di codifica che invia ogni richiesta di modello tramite Unity Gateway usando le tue credenziali Azure Databricks, senza una chiave API provider sulla macchina dello sviluppatore.
  • Record centrali dell'utilizzo e dei costi dell'agente, soggetti ai limiti e alle autorizzazioni impostati nel servizio modello.

Prerequisites

Passaggio 1: Connettere l'agente di codifica

È possibile connettere l'agente in uno dei due modi seguenti:

  • Unity Gateway CLI (consigliato): Un launcher open-source che gestisce OAuth e scrive il file di configurazione di ogni agente, inclusi l'URL della base del servizio modello e il modello. Non si gestiscono le chiavi API o gli endpoint a mano.
  • Configurazione manuale: Imposta l'URL della base del servizio modello e modella tu stesso nelle impostazioni integrate dello strumento (Claude Code, Codex, Cursor o Gemini CLI). Vedi Agenti di codifica supportati.

Questa sezione utilizza la CLI del Gateway Unity. Installarlo, quindi eseguire l'agente desiderato:

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

Claude Code

ug claude

Codex

ug codex

Per eseguire in modalità non interattiva, passa il flag dell'agente tramite la CLI del Gateway Unity:

ug codex --full-auto

Interfaccia della riga di comando gemini

ug gemini

OpenCode

ug opencode

Copilot

ug copilot

Alla prima esecuzione, la CLI Unity Gateway invia i prompt per l'URL del tuo workspace e si autentica; le esecuzioni successive vanno direttamente all'agente.

Passaggio 2: Gestire l'accesso al modello dell'agente

Puoi governare un servizio modello come qualsiasi altro sicurabile del Catalogo Unity: concedere EXECUTE per controllare chi può usarlo, e usare Unity Gateway per limitare il volume delle richieste e la spesa. Configurare una delle opzioni seguenti:

  • Gestire chi può accedere al modello: in Esplora cataloghi aprire il servizio modello e concedere EXECUTE a utenti o gruppi nella scheda Autorizzazioni .
  • Limita il volume di richieste: In Catalog Explorer, nella scheda Panoramica del servizio del modello, fai clic su Configura in corrispondenza di Limiti di frequenza. Impostare un limite di frequenza di richieste al minuto (QPM) o token al minuto (TPM) per tutte le richieste nell'endpoint o per utente.
  • Imposta un budget di spesa: un amministratore dell’account può creare un budget applicabile a tutti gli endpoint di Unity Gateway. Il budget raggruppa i costi relativi ai tuoi servizi modello; imposta una soglia di avviso per ricevere una notifica o per bloccare l'utilizzo prima che i costi superino il budget.

Passaggio 3: Verificare che il traffico sia regolato

Conferma che le richieste dell'agente passino attraverso Unity Gateway e vengano registrate:

  • Dall'agente: inviare un prompt. L'agente risponde usando il modello configurato, senza chiave API del provider nel computer.

  • Dalla tabella del sistema di utilizzo: interroga il tracciamento dell'utilizzo per confermare che il servizio del modello registri le richieste:

    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;

Per analizzare la spesa in modo più dettagliato, vedi Analizza il costo del Gateway Unity.

Passaggi successivi

  • Ultimo aggiornamento: