Uso de tags de requisição em políticas de serviço no Unity Gateway

Para que serve

Imagine que você precisa garantir que chamadas a um modelo de IA ou ferramenta MCP incluam um identificador de projeto válido antes de prosseguir. Com essa melhoria, é possível inspecionar tags fornecidas pelo chamador diretamente nas políticas de serviço baseadas em SQL.

• Validar tags de projeto ou custo antes de executar um modelo ou ferramenta.
• Bloquear requisições que não contenham as tags obrigatórias.
• Aplicar regras de controle baseadas no contexto da chamada.

Como usar

Acesse o Unity Gateway para criar uma função UDF em SQL que avalia o objeto event e retorna uma decisão de acesso.

CREATE OR REPLACE FUNCTION meu_catalogo.meu_schema.minha_politica(
  event VARIANT
)
RETURNS VARIANT
LANGUAGE SQL
RETURN CASE 
  WHEN event:context.request_tags.project::string IS NULL THEN 
    to_variant_object(named_struct('result', 'DENY', 'reason', 'Tag de projeto obrigatória ausente.'))
  ELSE 
    to_variant_object(named_struct('result', 'ALLOW', 'reason', 'Acesso permitido.'))
END;

O que acontece nesse código:

1. A função UDF recebe o argumento event do tipo VARIANT contendo o contexto da requisição.
2. O caminho event:context.request_tags.project extrai a tag enviada no cabeçalho HTTP Databricks-Ai-Gateway-Request-Tags.
3. O comando verifica se a tag está presente e retorna um objeto estruturado indicando ALLOW ou DENY através de to_variant_object.

Limitações e requisitos

• Recurso em estágio Beta.
• Disponível para Model Services, Model Provider Services e chamadas MCP (tools/call).
• O tamanho combinado das chaves e valores decodificados é limitado a 10 KiB de dados UTF-8.
• Administradores da conta devem habilitar o recurso na página de Previews do console da conta.