Tutorial: Controla el acceso al modelo de un agente de codificación con Unity Gateway

En este tutorial, conectas un agente de codificación externo, como Claude Code, Codex, Cursor o Gemini CLI, a grandes modelos de lenguaje (LLMs) gobernados por Unity Gateway. A continuación, controlar quién puede acceder a esos modelos y cuánto puede consumir y gastar el agente. En lugar de que cada desarrollador pegue una clave API de proveedor en su herramienta, cada solicitud pasa por un servicio modelo Unity Gateway. Unity Catalog controla quién puede usar qué modelos, y Unity Gateway impone límites y registra el uso y coste.

Al final de este tutorial, habrás:

  • Un agente de codificación que envía cada solicitud de modelo a través de Unity Gateway usando tus credenciales de Azure Databricks, sin una clave API de proveedor en la máquina del desarrollador.
  • Registros centrales del uso y el coste del agente, sujetos a los límites de tasa y los permisos que hayas establecido en el servicio de modelo.

Prerequisites

Paso 1: Conexión del agente de codificación

Puede conectar el agente de una de estas dos maneras:

  • Unity Gateway CLI (recomendado): Un lanzador de código abierto que gestiona OAuth y escribe el archivo de configuración de cada agente, incluyendo la URL de la base de servicio del modelo y el modelo. No administras manualmente las claves de API ni los puntos de conexión.
  • Configuración manual: Establece la URL de la base del servicio del modelo y modela tú mismo en la configuración integrada de la herramienta (Claude Code, Codex, Cursor o Gemini CLI). Consulta Agentes de codificación soportados.

Esta sección utiliza la CLI de Unity Gateway. Instálelo y, a continuación, ejecute el agente que desee:

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

Claude Code

ug claude

Codex

ug codex

Para ejecutarse en modo no interactivo, pasa la propia bandera del agente a través de la CLI de Unity Gateway:

ug codex --full-auto

Gemini CLI

ug gemini

OpenCode

ug opencode

Copilot

ug copilot

En la primera ejecución, la CLI de Unity Gateway solicita la URL de tu espacio de trabajo y se autentica; las ejecuciones posteriores van directamente al agente.

Paso 2: Gestionar el acceso al modelo del agente

Puedes gobernar un servicio de modelo como cualquier otro elemento protegible de Unity Catalog: otorga EXECUTE para controlar quién puede usarlo y usa Unity Gateway para poner límites al volumen de solicitudes y al gasto. Configure cualquiera de las siguientes opciones:

  • Administrar quién puede acceder al modelo: en el Explorador de catálogos, abra el servicio de modelo y conceda EXECUTE a usuarios o grupos en la pestaña Permisos .
  • Limitar el volumen de solicitudes: en el Explorador de catálogos, en la pestaña Información general del servicio del modelo, haga clic en Configurar en Límites de tasa. Establezca un límite de velocidad de solicitudes por minuto (QPM) o tokens por minuto (TPM) para todas las solicitudes del punto de conexión o por usuario.
  • Establece un presupuesto de gastos: Un administrador de cuentas puede crear un presupuesto que alcance todos los endpoints de Unity Gateway. El presupuesto agrupa el gasto de todos tus servicios de modelos; configura un umbral de alerta para recibir una notificación o bloquear el uso antes de que los costes lo superen.

Paso 3: Compruebe que el tráfico está controlado

Confirma que las solicitudes del agente pasan por Unity Gateway y están registradas:

  • Desde el agente: Envía una solicitud. El agente responde con el modelo que configuró, sin ninguna clave de API de proveedor en la máquina.

  • En la tabla del sistema de uso: consulte el seguimiento del uso para confirmar que el servicio de modelo registra las solicitudes:

    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;

Para analizar el gasto con más detalle, consulta Analizar el coste de Unity Gateway.

Pasos siguientes

  • Última actualización el