Pesquisa Vetorial da Mosaico AI: políticas orçamentárias

Este artigo descreve como usar políticas de orçamento para controlar os custos de pesquisa de vetor.

As políticas de orçamento permitem que os administradores agrupem e filtrem registros de cobrança em todos os produtos sem servidor do Azure Databricks e forneçam uma interface do usuário dedicada para controlar os gastos. As políticas de orçamento são criadas por administradores de espaço de trabalho e podem ser aplicadas a endpoints e índices de pesquisa vetorial. Para obter informações gerais e detalhes sobre como criar e gerenciar políticas de orçamento, consulte o uso de atributos com políticas de orçamento sem servidor.

Para obter informações gerais sobre como gerenciar seus custos ao usar o Mosaic AI Vector Search, consulte Mosaic AI Vector Search: Guia de gerenciamento de custos.

Aplicar uma política de orçamento a um endpoint

Você pode aplicar uma política de orçamento ao criar um ponto de extremidade ou editar o ponto de extremidade mais tarde para aplicar uma política.

Aplicar a política de orçamento na criação do ponto de extremidade

Interface do usuário do Databricks

Siga estas etapas para criar um ponto de extremidade de pesquisa de vetor e aplicar uma política de orçamento:

1. Na barra lateral esquerda, clique em Computação.

2. Clique na guia Busca em Vetores e clique em Criar.

3. Clique na seta à direita das Configurações Avançadas para abrir essa seção da caixa de diálogo.

   

4. Selecione uma política no menu suspenso Política de Orçamento .

5. Clique em Confirmar.

SDK do Python

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",
)

API REST

{
	name="vector_search_endpoint_name",
	budget_policy_id=”12345678-1234-1234-1234-1234567890ab”
}

Editar uma política de orçamento para um endpoint

Para editar uma política orçamentária para um endpoint usando a interface do usuário do Databricks:

1. Na página do ponto de extremidade, clique no ícone de lápis.

   

2. Selecione uma política no menu suspenso e clique em Salvar.

Para editar uma política usando o SDK do Python ou a API REST, consulte as guias em Aplicar política de orçamento na criação do endpoint.

Aplicar uma política de orçamento a um índice

Você pode aplicar uma política de orçamento ao criar um índice ou editar o índice mais tarde para aplicar uma política.

Aplicar a política de orçamento na criação do índice

Interface do usuário do Databricks

Ao criar um índice de pesquisa de vetor na interface do usuário do Databricks, você pode selecionar uma política de orçamento na lista suspensa política de orçamento sem servidor em configurações avançadas na caixa de diálogo de criação de índice. Consulte Criar índice usando a interface do usuário.

SDK do Python

from databricks.vector_search.client import VectorSearchClient

client = VectorSearchClient()

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",
)

API REST

{
  "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 uma política de orçamento para um índice

Para editar uma política de orçamento para um índice usando o SDK do Python:

client.update_index_budget_policy(
    index_name="index_name",
    budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)

Consultar gastos por política de orçamento

Para consultar os gastos por política orçamentária, você deve ter a ID da política de orçamento. Para obter a ID da política de orçamento usando a interface do usuário, siga estas etapas como um usuário administrador:

1. Clique no nome de usuário na barra superior do workspace do Databricks e clique em Configurações.
2. Clique em Computação.
3. Ao lado de políticas de orçamento sem servidor, clique em Gerenciar.
4. Selecione uma política de orçamento sem servidor. A ID da política de orçamento aparece na seção superior direita, sobre essa política.

Para obter a ID da política de orçamento usando a API REST, consulte Políticas de lista.

A consulta a seguir agrega os gastos por ID da política orçamentária. Substitua <INSERT BUDGET POLICY ID> pela ID da política de orçamento.

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 BUDGET 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 vetor, as políticas de orçamento têm as seguintes limitações:

• As políticas de orçamento não impõem limites de gastos para pesquisa de vetor.