Políticas de uso da Pesquisa de AI
Visualização
Esse recurso está em Prévia Pública.
Você pode aplicar uma política de uso a um endpoint ou índice de Pesquisa de IA para rastrear seus gastos serverless por equipe ou projeto.
Políticas de uso permitem aos administradores agrupar e filtrar registros de cobrança em todos os produtos serverless do Databricks e fornecem uma interface de usuário dedicada para acompanhamento de gastos. Políticas de uso são criadas por administradores de workspace e podem ser aplicadas a endpoints e índices de Pesquisa de AI. Para obter informações gerais e detalhes sobre como criar e gerenciar políticas de uso, consulte Uso de atributos com políticas de uso serverless.
Para obter informações gerais sobre como gerenciar seus custos ao usar a Pesquisa de IA, consulte guia de gerenciamento de custos da Pesquisa de IA.
Aplicar uma política de uso a um endpoint
Você pode aplicar uma política de uso ao criar um endpoint, ou editar o endpoint mais tarde para aplicar uma política.
Aplicar política de uso na criação de endpoint
- Databricks UI
- Python SDK
- REST API
Siga estes passos para criar um endpoint de Pesquisa de IA e aplicar uma política de uso:
-
Na barra lateral esquerda, clique em Compute .
-
Clique na **tab AI Search** e clique em **Criar**.
-
Clique na seta à direita de **Configurações avançadas** para abrir essa seção da caixa de diálogo.

-
Selecione uma política no menu **dropdown** **Política de Uso**.
-
Clique em Confirmar .
client.create_endpoint(
name="vector_search_endpoint_name",
endpoint_type="STANDARD",
budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)
client.update_endpoint_budget_policy(
name="vector_search_endpoint_name",
budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)
{
name="vector_search_endpoint_name",
budget_policy_id=”12345678-1234-1234-1234-1234567890ab”
}
Editar uma política de uso para um endpoint
Para editar uma política de uso para um endpoint com a interface do usuário do Databricks:
-
Na página do endpoint, clique no ícone de lápis.

-
Selecione uma política no menu suspenso e clique em Salvar .
Para editar uma política usando o SDK Python ou a API REST, consulte as tabs em Aplicar política de uso na criação do endpoint.
Aplicar uma política de uso a um índice
É possível aplicar uma política de uso ao criar um índice, ou editar o índice posteriormente para aplicar uma política.
Aplicar política de uso na criação do índice
- Databricks UI
- Python SDK
- REST API
Ao criar um índice de Pesquisa de AI na UI do Databricks, é possível selecionar uma política de uso no dropdown **Política de Uso** em **Configurações avançadas** na caixa de diálogo de criação do índice. Consulte Criar índice usando a interface do usuário.
from databricks.ai_search.client import AISearchClient
client = AISearchClient()
client.create_delta_sync_index(
endpoint_name="vector_search_endpoint_name",
index_name="index_name",
source_table_name="catalog.schema.table",
primary_key="id",
pipeline_type="TRIGGERED",
budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)
{
"name": "index_name",
"endpoint_name": "vector_search_endpoint_name",
"primary_key": "id",
"index_type": "DELTA_SYNC",
"delta_sync_index_spec": {
"source_table": "catalog.schema.table",
"pipeline_type": "TRIGGERED",
"budget_policy_id": "12345678-1234-1234-1234-1234567890ab"
}
}
Editar política de uso para um índice
Para editar uma política de uso para um índice usando o SDK para Python:
client.update_index_budget_policy(
index_name="index_name",
budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)
Consultar gastos por política de uso
Para consultar gastos por política de uso, é necessário possuir o ID da política de uso. Para obter o ID da política de uso usando a IU, siga estes passos como um usuário administrador:
- Clique no seu nome de usuário na barra superior do workspace do Databricks e clique em **Configurações**.
- Click compute .
- Ao lado de **políticas de uso serverless**, clique em **Gerenciar**.
- Selecione uma política de uso serverless. O ID da política de uso aparece na seção superior direita, **Sobre esta política**.
Para obter o ID da política de uso usando a API REST, consulte Listar políticas.
A seguinte consulta agrega gastos pelo ID da política de uso. Substituir <INSERT USAGE POLICY ID> pelo ID da política de uso.
WITH vector_search_usage_by_budget_policy (
SELECT *,
CASE WHEN usage_type = "STORAGE_SPACE" THEN 'storage'
ELSE 'serving'
END as workload_type
FROM system.billing.usage
WHERE billing_origin_product = 'VECTOR_SEARCH'
AND usage_metadata.endpoint_name IS NOT NULL
AND usage_metadata.budget_policy_id = '<INSERT USAGE POLICY ID>'
),
daily_usage_by_budget_policy AS (
SELECT workspace_id,
cloud,
usage_date,
workload_type,
usage_metadata.endpoint_name as vector_search_endpoint,
CASE WHEN workload_type = 'serving' THEN SUM(usage_quantity)
ELSE null
END as dbus,
CASE WHEN workload_type = 'storage' THEN SUM(usage_quantity)
ELSE null
END as dsus
FROM vector_search_usage_by_budget_policy
GROUP BY all
ORDER BY 1,2,3,4,5,6 DESC
)
SELECT * FROM daily_usage_by_budget_policy
Limitações
Para Pesquisa de AI, as políticas de uso têm as seguintes limitações:
- Políticas de uso não impõem limites de gastos para AI Pesquisa.