Pular para o conteúdo principal

Rastrear o uso do modelo

Esta página descreve como monitorar o uso de serviços do Unity Gateway usando a tabela do sistema de acompanhamento de uso.

A tabela de acompanhamento de uso captura detalhes de solicitações e respostas para serviços de modelo, serviços de provedor de modelos e serviços MCP. Para solicitações de modelo, grava métricas como uso de tokens e latência. Para solicitações MCP, registra metadados de chamadas e serviços. Use a tabela para monitorar usuários, acompanhar custos e analisar o uso e o desempenho do serviço.

O acompanhamento de uso também captura solicitações de ai_query para serviços de modelo fornecidos pelo Databricks.

Os administradores de conta e de Workspace podem visualizar uma visão geral consolidada do uso de AI na página de AI no Governance Hub.

Requisitos​

preços​

O acompanhamento de uso é um recurso faturável do Unity Gateway. O Databricks cobra pelo uso que registra na tabela system.ai_gateway.usage. Consulte preços do Unity Gateway.

Consultar a tabela de uso​

O Unity Gateway Logs dados de uso na tabela do sistema system.ai_gateway.usage. Você pode ver a tabela na interface de usuário, ou consultar a tabela a partir do Databricks SQL ou de um Notebook.

nota

Ambas as funções de administrador de account e de metastore são necessárias para view ou query a tabela system.ai_gateway.usage por default. Os administradores podem gerenciar o acesso às tabelas do sistema para controlar as permissões de usuários, grupos e Service Principal.

Para view a tabela na interface do usuário, clique no link da tabela de acompanhamento de uso na página do serviço de modelo para abrir a tabela no Explorador de Catálogos.

Para consultar a tabela do Databricks SQL ou de um Notebook:

SQL
SELECT * FROM system.ai_gateway.usage;
prompt

Genie Code (modo Agente) pode fazer isso por você. Experimente este exemplo de prompt:

Query the system.ai_gateway.usage table to analyze AI Gateway usage showing request count and total tokens, grouped by endpoint name for the last 7 days.

Dashboard de uso integrada​

nota

Alguns workspaces ainda não exibem o dropdown Govern . Nesses workspaces, use os botões independentes Create Dashboard , View Dashboard e Update na página do Unity Gateway.

Criar painel de uso integrado​

Os administradores da account podem criar um painel de uso integrado do Unity Gateway para monitorar o uso, rastrear custos e obter percepções sobre o desempenho e o consumo do serviço de modelo. Na página do Unity Gateway, clique em Governar no canto superior direito e, em seguida, clique em Criar Painel de Uso . O warehouse que executa as queries do painel é selecionado automaticamente.

nota

A criação de dashboards é restrita a administradores de account porque requer permissões SELECT na tabela system.ai_gateway.usage. Os dados do dashboard estão sujeitos às políticas de retenção da tabela usage. Consulte Quais tabelas do sistema estão disponíveis?.

Quando uma versão mais recente do painel de uso integrado estiver disponível, os administradores da account podem clicar em Atualizar na linha da versão do painel no menu suspenso Govern na página do Unity Gateway.

É possível usar as seguintes opções de configuração do painel para gerenciar o painel:

  • Escopo: Selecione se deseja definir o escopo do painel na account ou no workspace.
  • Permissões : Escolha se as consultas serão de execução com base nas permissões do proprietário do painel ou nas permissões de cada visualizador. Consulte O que são permissões de dados compartilhados?.
  • Atualizações automáticas : Quando você habilita essa opção, o dashboard é atualizado automaticamente sempre que uma versão mais recente estiver disponível e um administrador da account visitar a página do Unity Gateway.

Opções de atualização do dashboard do Gateway de AI

Quando o painel é atualizado para a versão 0.3 ou superior, uma programação é criada automaticamente para refresh o painel a cada 6 horas. Se necessário, este programar pode ser desativado no painel Lakeview. Consulte Criar um programar.

view painel de uso​

Para ver o dashboard, clique em Govern no canto superior direito da página do Unity Gateway e, em seguida, clique em Usage Dashboard . O dashboard abre em uma nova tab. O painel integrado tem visibilidade abrangente sobre o uso, desempenho e custo do serviço de modelo do Unity Gateway. Ele inclui várias páginas de acompanhamento de requisições, consumo de tokens, métricas de latência, taxas de erro, detalhamento de custos, tráfego de servidor MCP externo e atividade de agente de codificação.

Painel de uso do gateway de AI

O dashboard fornece analítica entre workspaces por default. Todas as páginas do dashboard podem ser filtradas por intervalo de datas e ID do workspace.

  • Tab Visão geral : mostra métricas de uso de alto nível, incluindo volume diário de solicitações, tendências de uso de tokens ao longo do tempo, principais usuários por consumo de tokens e contagem total de usuários únicos. Use esta tab para obter um Snapshot rápido da atividade geral do Unity Gateway e identificar os usuários e modelos mais ativos.
  • Tab de desempenho : Acompanha as key métricas de desempenho, incluindo percentis de latência (P50, P90, P95, P99), tempo até o primeiro byte, taxas de erro e distribuições de código de status HTTP. Use esta tab para monitorar a integridade do serviço de modelo e identificar gargalos de desempenho ou problemas de confiabilidade.
  • Tab Uso : Mostra detalhamentos de consumo detalhados por serviço de modelo, workspace e solicitante. Esta tab mostra padrões de uso de tokens, distribuições de solicitações e taxas de acerto de cache.
  • Tab de observabilidade de custos : mostra detalhamentos de custos por serviço de modelo, modelo de destino, usuário, tags de serviço e tags de solicitação. Esta tab também inclui o custo estimado para modelos externos. Consulte Analisar custo do Unity Gateway.
  • Tab do Servidor MCP Externo : mostra o volume de solicitações, taxas de erro, usuários e conexões, e tendências de uso diário para o tráfego do servidor MCP externo.
  • Coding Agents tab : Tracks activity from integrated coding agents including Claude Code, Codex CLI, Cursor, and Gemini CLI. This tab shows métricas like active days, coding sessions, commits, and lines of code added or removed to monitor developer tool usage. See Coding agent dashboard for more details.

Esquema da tabela de uso​

A tabela system.ai_gateway.usage tem o seguinte esquema:

Nome da coluna

Tipo

Descrição

Exemplo

account_id

String

O ID da account.

11d77e21-5e05-4196-af72-423257f74974

workspace_id

String

O ID do workspace.

1653573648247579

request_id

String

Um identificador exclusivo para a solicitação.

b4a47a30-0e18-4ae3-9a7f-29bcb07e0f00

invocation_id

String

Um identificador exclusivo para cada chamada de inferência individual. Múltiplas invocações podem compartilhar o mesmo request_id, como verificações de guardrail ou chamadas de agente de várias etapas. Use invocation_id para distingui-los.

c0a8012e-9f3b-4d21-8a7e-1b2c3d4e5f60

schema_version

Integer

A versão do esquema do registro de uso.

1

service_type

String

O tipo de serviço que gerou o registro de uso. Os valores são MODEL_SERVICE, MCP_SERVICE e MODEL_PROVIDER_SERVICE.

MODEL_SERVICE

service_id

String

O ID do Model Service, do serviço MCP ou do serviço de provedor de modelos.

43addf89-d802-3ca2-bd54-fe4d2a60d58a

service_name

String

O nome totalmente qualificado do serviço no Unity Catalog.

main.default.github_tools

service_tags

Mapa

Tags de recurso aplicadas ao objeto protegível do Unity Catalog no momento da criação ou atualização. Eles se aplicam a todas as solicitações ao serviço e são úteis para categorizar o uso por equipe, centro de custos ou projeto.

{"team": "engineering"}

endpoint_id

String

O ID exclusivo do serviço de modelo do Unity Gateway.

43addf89-d802-3ca2-bd54-fe4d2a60d58a

endpoint_name

String

O nome do serviço de modelo do Unity Gateway.

system.ai.gpt-5-2

endpoint_tags

Mapa

Tags configuradas no serviço de modelo no momento da criação ou atualização. Eles se aplicam a todas as solicitações para o serviço de modelo e são úteis para categorizar serviços por equipe, centro de custo ou projeto.

{"team": "engineering"}

endpoint_metadata

struct

Metadados do serviço de modelo, incluindo creator, creation_time, last_updated_time, destinations, inference_table e fallbacks.

{"creator": "user.name@email.com", "creation_time": "2026-01-06T12:00:00.000Z", ...}

event_time

Timestamp

O carimbo de data/hora quando a solicitação foi recebida.

2026-01-20T19:48:08.000+00:00

latency_ms

Long

A latência total em milissegundos.

300

time_to_first_byte_ms

Long

O tempo para o primeiro byte em milissegundos.

300

destination_type

String

O tipo de destino (por exemplo, modelo externo ou modelo básico).

PAY_PER_TOKEN_FOUNDATION_MODEL

destination_name

String

O nome do modelo ou provedor de destino.

system.ai.gpt-5-2

destination_id

String

A ID exclusiva do destino.

507e7456151b3cc89e05ff48161efb87

destination_model

String

O modelo específico utilizado para a solicitação.

GPT-5.2

requester

String

O ID do usuário ou da entidade de serviço que fez a solicitação.

user.name@email.com

requester_type

String

O tipo de solicitante (usuário, entidade de serviço ou grupo de usuários).

USER

ip_address

String

O endereço IP do solicitante.

1.2.3.4

url

String

A URL da solicitação.

https://<workspace-url>/ai-gateway/mlflow/v1/chat/completions

user_agent

String

O agente do usuário do solicitante.

OpenAI/Python 2.13.0

api_type

String

O tipo de chamada de API (por exemplo, chat, conclusões ou incorporações).

mlflow/v1/chat/completions

request_tags

Mapa

Tags fornecidas pelo usuário enviadas com solicitações individuais usando o cabeçalho HTTP Databricks-Ai-Gateway-Request-Tags. Use tags de solicitação para atribuir o uso a projetos, equipes, ambientes ou usuários finais específicos. Consulte Solicitações de tag para acompanhamento de uso e Marcação de solicitações.

{"project": "chatbot", "team": "ml-platform"}

invocation_metadata

struct

Informação sobre onde a solicitação se originou, o nível de serviço relatado pelo provedor do modelo e se a solicitação usou uma inscrição do Claude.

{"source": "EXTERNAL_CLIENT", "service_tier": "priority", "relayed": false}

input_tokens

Long

O número de tokens de entrada.

100

output_tokens

Long

O número de tokens de saída.

100

total_tokens

Long

O número total de tokens (entrada + saída).

200

token_details

struct

Detalhamento do uso de tokens e ferramentas, incluindo cache_read_input_tokens, cache_creation_input_tokens, output_reasoning_tokens, cache_creation_5m_input_tokens, cache_creation_1h_input_tokens, file_search_count e num_web_search_queries.

{"cache_read_input_tokens": 100, ...}

response_content_type

String

O tipo de conteúdo da resposta.

application/json

status_code

INT

O código de status HTTP da resposta.

200

routing_information

struct

Detalhes de roteamento para tentativas de fallback. Contém um array attempts com priority, action, destination, destination_id, status_code, error_code, latency_ms, start_time e end_time para cada modelo tentado durante a solicitação.

{"attempts": [{"priority": "1", ...}]}

mcp_metadata

struct

Detalhes de uma solicitação para um serviço MCP, incluindo a ferramenta invocada, o tipo de servidor e a operação JSON-RPC. Preenchido para MCP_SERVICE linhas.

{"tool_name": "echo", "server_type": "EXTERNAL", "json_rpc_method": "tools/call"}

session_metadata

struct

Contexto de sessão e cliente, incluindo IDs de sessão e subagente, nome e versão do agente de codificação, interface do cliente, esforço de raciocínio e receita de roteamento inteligente. Use estes campos para agrupar solicitações relacionadas e analisar o uso por agente ou sessão.

{"coding_agent": "claude-code", "agent_version": "2.1.282", "reasoning_effort": "high", ...}

auth_mode

String

O tipo de credencial do Databricks usado para autenticar a solicitação: um access token pessoal (PAT) ou um token OAuth (OAUTH).

OAUTH

Nome da coluna

Tipo

Descrição

Exemplo

account_id

String

O ID da account.

11d77e21-5e05-4196-af72-423257f74974

workspace_id

String

O ID do workspace.

1653573648247579

request_id

String

Um identificador exclusivo para a solicitação.

b4a47a30-0e18-4ae3-9a7f-29bcb07e0f00

invocation_id

String

Um identificador exclusivo para cada chamada de inferência individual. Múltiplas invocações podem compartilhar o mesmo request_id, como verificações de guardrail ou chamadas de agente de várias etapas. Use invocation_id para distingui-los.

c0a8012e-9f3b-4d21-8a7e-1b2c3d4e5f60

schema_version

Integer

A versão do esquema do registro de uso.

1

service_type

String

O tipo de serviço que gerou o registro de uso. Os valores são MODEL_SERVICE, MCP_SERVICE e MODEL_PROVIDER_SERVICE.

MODEL_SERVICE

service_id

String

O ID do Model Service, do serviço MCP ou do serviço de provedor de modelos.

43addf89-d802-3ca2-bd54-fe4d2a60d58a

service_name

String

O nome totalmente qualificado do serviço no Unity Catalog.

main.default.github_tools

service_tags

Mapa

Tags de recurso aplicadas ao objeto protegível do Unity Catalog no momento da criação ou atualização. Eles se aplicam a todas as solicitações ao serviço e são úteis para categorizar o uso por equipe, centro de custos ou projeto.

{"team": "engineering"}

endpoint_id

String

O ID exclusivo do serviço de modelo do Unity Gateway.

43addf89-d802-3ca2-bd54-fe4d2a60d58a

endpoint_name

String

O nome do serviço de modelo do Unity Gateway.

system.ai.gpt-5-2

endpoint_tags

Mapa

Tags configuradas no serviço de modelo no momento da criação ou atualização. Eles se aplicam a todas as solicitações para o serviço de modelo e são úteis para categorizar serviços por equipe, centro de custo ou projeto.

{"team": "engineering"}

endpoint_metadata

struct

Metadados do serviço de modelo, incluindo creator, creation_time, last_updated_time, destinations, inference_table e fallbacks.

{"creator": "user.name@email.com", "creation_time": "2026-01-06T12:00:00.000Z", ...}

event_time

Timestamp

O carimbo de data/hora quando a solicitação foi recebida.

2026-01-20T19:48:08.000+00:00

latency_ms

Long

A latência total em milissegundos.

300

time_to_first_byte_ms

Long

O tempo para o primeiro byte em milissegundos.

300

destination_type

String

O tipo de destino (por exemplo, modelo externo ou modelo básico).

PAY_PER_TOKEN_FOUNDATION_MODEL

destination_name

String

O nome do modelo ou provedor de destino.

system.ai.gpt-5-2

destination_id

String

A ID exclusiva do destino.

507e7456151b3cc89e05ff48161efb87

destination_model

String

O modelo específico utilizado para a solicitação.

GPT-5.2

requester

String

O ID do usuário ou da entidade de serviço que fez a solicitação.

user.name@email.com

requester_type

String

O tipo de solicitante (usuário, entidade de serviço ou grupo de usuários).

USER

ip_address

String

O endereço IP do solicitante.

1.2.3.4

url

String

A URL da solicitação.

https://<workspace-url>/ai-gateway/mlflow/v1/chat/completions

user_agent

String

O agente do usuário do solicitante.

OpenAI/Python 2.13.0

api_type

String

O tipo de chamada de API (por exemplo, chat, conclusões ou incorporações).

mlflow/v1/chat/completions

request_tags

Mapa

Tags fornecidas pelo usuário enviadas com solicitações individuais usando o cabeçalho HTTP Databricks-Ai-Gateway-Request-Tags. Use tags de solicitação para atribuir o uso a projetos, equipes, ambientes ou usuários finais específicos. Consulte Solicitações de tag para acompanhamento de uso e Marcação de solicitações.

{"project": "chatbot", "team": "ml-platform"}

invocation_metadata

struct

Informação sobre onde a solicitação se originou, o nível de serviço relatado pelo provedor do modelo e se a solicitação usou uma inscrição do Claude.

{"source": "EXTERNAL_CLIENT", "service_tier": "priority", "relayed": false}

input_tokens

Long

O número de tokens de entrada.

100

output_tokens

Long

O número de tokens de saída.

100

total_tokens

Long

O número total de tokens (entrada + saída).

200

token_details

struct

Detalhamento do uso de tokens e ferramentas, incluindo cache_read_input_tokens, cache_creation_input_tokens, output_reasoning_tokens, cache_creation_5m_input_tokens, cache_creation_1h_input_tokens, file_search_count e num_web_search_queries.

{"cache_read_input_tokens": 100, ...}

response_content_type

String

O tipo de conteúdo da resposta.

application/json

status_code

INT

O código de status HTTP da resposta.

200

routing_information

struct

Detalhes de roteamento para tentativas de fallback. Contém um array attempts com priority, action, destination, destination_id, status_code, error_code, latency_ms, start_time e end_time para cada modelo tentado durante a solicitação.

{"attempts": [{"priority": "1", ...}]}

mcp_metadata

struct

Detalhes de uma solicitação para um serviço MCP, incluindo a ferramenta invocada, o tipo de servidor e a operação JSON-RPC. Preenchido para MCP_SERVICE linhas.

{"tool_name": "echo", "server_type": "EXTERNAL", "json_rpc_method": "tools/call"}

session_metadata

struct

Contexto de sessão e cliente, incluindo IDs de sessão e subagente, nome e versão do agente de codificação, interface do cliente, esforço de raciocínio e receita de roteamento inteligente. Use estes campos para agrupar solicitações relacionadas e analisar o uso por agente ou sessão.

{"coding_agent": "claude-code", "agent_version": "2.1.282", "reasoning_effort": "high", ...}

auth_mode

String

O tipo de credencial do Databricks usado para autenticar a solicitação: um access token pessoal (PAT) ou um token OAuth (OAUTH).

OAUTH

Schemas de colunas aninhadas​

As tabelas a seguir descrevem os campos em colunas aninhadas de STRUCT. A disponibilidade dos campos depende do serviço, modelo e cliente usados para a solicitação.

Metadados de invocação e token​

Caminho do campo

Tipo

Descrição

invocation_metadata.source

String

O aplicativo, serviço ou API que iniciou a solicitação. Use este campo para atribuir o uso à sua origem. Os valores incluem AI_PLAYGROUND, EXTERNAL_CLIENT, AI_QUERY, GUARDRAIL e MANAGED_AGENT.

invocation_metadata.service_tier

String

A camada de serviço relatada pelo provedor do modelo na resposta de inferência, como default ou priority. Use este campo para comparar o uso entre os níveis de preços do provedor.

invocation_metadata.relayed

Booleana

Se a solicitação foi encaminhada para a Anthropic usando a inscrição do Claude do chamador.

token_details.cache_read_input_tokens

Long

O número de tokens lidos do cache de prompts.

token_details.cache_creation_input_tokens

Long

O número de tokens gravados no cache de prompts.

token_details.output_reasoning_tokens

Long

O número de tokens de raciocínio na saída.

token_details.cache_creation_5m_input_tokens

Long

O número de tokens de entrada gravados no cache de prompts com um tempo de vida de 5 minutos.

token_details.cache_creation_1h_input_tokens

Long

O número de tokens de entrada gravados no cache do prompt com um tempo de vida de 1 hora.

token_details.file_search_count

Long

O número de chamadas da ferramenta de pesquisa de arquivos feitas como parte da solicitação.

token_details.num_web_search_queries

Long

O número de web search queries cobráveis feitas como parte da requisição.

Caminho do campo

Tipo

Descrição

invocation_metadata.source

String

O aplicativo, serviço ou API que iniciou a solicitação. Use este campo para atribuir o uso à sua origem. Os valores incluem AI_PLAYGROUND, EXTERNAL_CLIENT, AI_QUERY, GUARDRAIL e MANAGED_AGENT.

invocation_metadata.service_tier

String

A camada de serviço relatada pelo provedor do modelo na resposta de inferência, como default ou priority. Use este campo para comparar o uso entre os níveis de preços do provedor.

invocation_metadata.relayed

Booleana

Se a solicitação foi encaminhada para a Anthropic usando a inscrição do Claude do chamador.

token_details.cache_read_input_tokens

Long

O número de tokens lidos do cache de prompts.

token_details.cache_creation_input_tokens

Long

O número de tokens gravados no cache de prompts.

token_details.output_reasoning_tokens

Long

O número de tokens de raciocínio na saída.

token_details.cache_creation_5m_input_tokens

Long

O número de tokens de entrada gravados no cache de prompts com um tempo de vida de 5 minutos.

token_details.cache_creation_1h_input_tokens

Long

O número de tokens de entrada gravados no cache do prompt com um tempo de vida de 1 hora.

token_details.file_search_count

Long

O número de chamadas da ferramenta de pesquisa de arquivos feitas como parte da solicitação.

token_details.num_web_search_queries

Long

O número de web search queries cobráveis feitas como parte da requisição.

MCP service metadata​

Esses campos são preenchidos em mcp_metadata para MCP_SERVICE linhas.

Caminho do campo

Tipo

Descrição

mcp_metadata.tool_name

String

O nome da ferramenta invocada por uma solicitação MCP tools/call. Use este campo para analisar o uso de ferramentas individuais em um servidor.

mcp_metadata.server_type

String

The category of MCP server handling the request, such as EXTERNAL or SYSTEM.

mcp_metadata.json_rpc_method

String

A operação JSON-RPC solicitada pelo cliente, como tools/call para invocar uma ferramenta, tools/list para descobrir ferramentas ou initialize para iniciar uma sessão.

Caminho do campo

Tipo

Descrição

mcp_metadata.tool_name

String

O nome da ferramenta invocada por uma solicitação MCP tools/call. Use este campo para analisar o uso de ferramentas individuais em um servidor.

mcp_metadata.server_type

String

The category of MCP server handling the request, such as EXTERNAL or SYSTEM.

mcp_metadata.json_rpc_method

String

A operação JSON-RPC solicitada pelo cliente, como tools/call para invocar uma ferramenta, tools/list para descobrir ferramentas ou initialize para iniciar uma sessão.

Metadados da sessão​

Os campos session_metadata ajudam a correlacionar solicitações em uma sessão e a distinguir agentes de codificação, interfaces de cliente e configurações de solicitação. Cada campo é preenchido quando a informação correspondente está disponível no cliente ou na solicitação.

Caminho do campo

Tipo

Descrição

session_metadata.client_session_id

String

O ID da sessão fornecido pelo cliente. Use-o para agrupar solicitações feitas durante a mesma conversa ou sessão de agente de programação.

session_metadata.client_subagent_id

String

O ID do subagente fornecido pelo cliente. Use-o com client_session_id para distinguir solicitações de subagentes em uma sessão pai.

session_metadata.coding_agent

String

The normalized name of the coding agent that sent the request, such as claude-code or codex.

session_metadata.agent_version

String

A versão relatada do agente de programação. Use-o com coding_agent para comparar o uso entre versões do agente.

session_metadata.surface

String

A interface do cliente da qual o agente de codificação enviou a solicitação, como uma interface de linha de comando, IDE ou aplicativo de desktop.

session_metadata.reasoning_effort

String

O esforço de raciocínio especificado na solicitação, como low, medium ou high. Os valores disponíveis dependem do modelo e da API.

session_metadata.smart_router_name

String

The name of the smart-routing recipe selected by the client. Use it to group usage by routing recipe.

Caminho do campo

Tipo

Descrição

session_metadata.client_session_id

String

O ID da sessão fornecido pelo cliente. Use-o para agrupar solicitações feitas durante a mesma conversa ou sessão de agente de programação.

session_metadata.client_subagent_id

String

O ID do subagente fornecido pelo cliente. Use-o com client_session_id para distinguir solicitações de subagentes em uma sessão pai.

session_metadata.coding_agent

String

The normalized name of the coding agent that sent the request, such as claude-code or codex.

session_metadata.agent_version

String

A versão relatada do agente de programação. Use-o com coding_agent para comparar o uso entre versões do agente.

session_metadata.surface

String

A interface do cliente da qual o agente de codificação enviou a solicitação, como uma interface de linha de comando, IDE ou aplicativo de desktop.

session_metadata.reasoning_effort

String

O esforço de raciocínio especificado na solicitação, como low, medium ou high. Os valores disponíveis dependem do modelo e da API.

session_metadata.smart_router_name

String

The name of the smart-routing recipe selected by the client. Use it to group usage by routing recipe.

Solicitações de tag para acompanhamento de uso​

Tags de solicitação são pares key-value personalizados que o chamador anexa a solicitações individuais. Use tags de solicitação para atribuir o uso por projeto, equipe, ambiente, usuário final ou qualquer outra dimensão relevante para sua organização. Tags de solicitação são registradas na tabela system.ai_gateway.usage e podem ser usadas para filtrar, agregar e analisar dados de uso.

Para adicionar tag a solicitações individuais, inclua o cabeçalho HTTP Databricks-Ai-Gateway-Request-Tags com um objeto JSON que mapeia key de strings para valores de strings. As tags de solicitação são registradas na coluna request_tags na tabela de uso e em tabelas de inferência.

Para obter exemplos que mostram como definir tags de solicitação com a API REST, o SDK da OpenAI e o SDK da Anthropic, consulte Marcação de tag.

Por exemplo, pode agregar o uso por projeto usando tags de solicitação:

SQL
SELECT
request_tags['project'] AS project,
COUNT(*) AS request_count,
SUM(total_tokens) AS total_tokens
FROM system.ai_gateway.usage
WHERE request_tags['project'] IS NOT NULL
GROUP BY request_tags['project']
ORDER BY total_tokens DESC;

Limitações​

  • O Unity Gateway não rastreia o uso de tokens para respostas que não sejam de transmissão ou de embedding maiores que 1 MiB.

Recursos adicionais​