Files
browser-use__browser-use/browser_use/tokens/custom_pricing.py
Saurav Panda b1d1794a93 fix(tokens): bill bu-1-0 at bu-2-0 rates
bu-1-0 is redirected to bu-2-0 at the gateway, so requests naming it are
served and billed as bu-2-0. The pricing table still carried the retired
bu-1-0 rates, which under-reported cost by 3x on input and 3x on cached
reads for anyone still passing that id.

Aliases it to the bu-2-0 entry rather than duplicating the numbers, same
as bu-latest, so the three cannot drift apart.
2026-08-13 11:16:52 -07:00

98 lines
3.6 KiB
Python

"""
Custom model pricing for models not available in LiteLLM's pricing data.
Prices are per token (not per 1M tokens).
"""
from typing import Any
# Custom model pricing data
# Format matches LiteLLM's model_prices_and_context_window.json structure
CUSTOM_MODEL_PRICING: dict[str, dict[str, Any]] = {
'bu-2-0': {
'input_cost_per_token': 0.60 / 1_000_000, # $0.60 per 1M tokens
'output_cost_per_token': 3.50 / 1_000_000, # $3.50 per 1M tokens
'cache_read_input_token_cost': 0.06 / 1_000_000, # $0.06 per 1M tokens
'cache_creation_input_token_cost': None, # Not specified
'max_tokens': None, # Not specified
'max_input_tokens': None, # Not specified
'max_output_tokens': None, # Not specified
},
'bu-2-0-mini-preview': {
'input_cost_per_token': 0.15 / 1_000_000, # $0.15 per 1M tokens
'output_cost_per_token': 1.50 / 1_000_000, # $1.50 per 1M tokens
# No cache discount on this model: cached reads bill at the input rate.
'cache_read_input_token_cost': 0.15 / 1_000_000, # $0.15 per 1M tokens
'cache_creation_input_token_cost': None, # Not specified
'max_tokens': None, # Not specified
'max_input_tokens': None, # Not specified
'max_output_tokens': None, # Not specified
},
'claude-sonnet-4-6': {
'input_cost_per_token': 3.00 / 1_000_000,
'output_cost_per_token': 15.00 / 1_000_000,
'cache_read_input_token_cost': 0.30 / 1_000_000,
'cache_creation_input_token_cost': 3.75 / 1_000_000,
'cache_creation_1h_input_token_cost': 6.00 / 1_000_000,
'max_tokens': None,
'max_input_tokens': None,
'max_output_tokens': None,
},
'anthropic/claude-sonnet-4.6': {
'input_cost_per_token': 3.00 / 1_000_000,
'output_cost_per_token': 15.00 / 1_000_000,
'cache_read_input_token_cost': 0.30 / 1_000_000,
'cache_creation_input_token_cost': 3.75 / 1_000_000,
'cache_creation_1h_input_token_cost': 6.00 / 1_000_000,
'max_tokens': None,
'max_input_tokens': None,
'max_output_tokens': None,
},
'claude-opus-4-6': {
'input_cost_per_token': 5.00 / 1_000_000,
'output_cost_per_token': 25.00 / 1_000_000,
'cache_read_input_token_cost': 0.50 / 1_000_000,
'cache_creation_input_token_cost': 6.25 / 1_000_000,
'cache_creation_1h_input_token_cost': 10.00 / 1_000_000,
'max_tokens': None,
'max_input_tokens': None,
'max_output_tokens': None,
},
'anthropic/claude-opus-4.6': {
'input_cost_per_token': 5.00 / 1_000_000,
'output_cost_per_token': 25.00 / 1_000_000,
'cache_read_input_token_cost': 0.50 / 1_000_000,
'cache_creation_input_token_cost': 6.25 / 1_000_000,
'cache_creation_1h_input_token_cost': 10.00 / 1_000_000,
'max_tokens': None,
'max_input_tokens': None,
'max_output_tokens': None,
},
'claude-fable-5': {
'input_cost_per_token': 10.00 / 1_000_000,
'output_cost_per_token': 50.00 / 1_000_000,
'cache_read_input_token_cost': 1.00 / 1_000_000,
'cache_creation_input_token_cost': 12.50 / 1_000_000,
'cache_creation_1h_input_token_cost': 20.00 / 1_000_000,
'max_tokens': 1_000_000,
'max_input_tokens': 1_000_000,
'max_output_tokens': 128_000,
},
'anthropic/claude-fable-5': {
'input_cost_per_token': 10.00 / 1_000_000,
'output_cost_per_token': 50.00 / 1_000_000,
'cache_read_input_token_cost': 1.00 / 1_000_000,
'cache_creation_input_token_cost': 12.50 / 1_000_000,
'cache_creation_1h_input_token_cost': 20.00 / 1_000_000,
'max_tokens': 1_000_000,
'max_input_tokens': 1_000_000,
'max_output_tokens': 128_000,
},
}
CUSTOM_MODEL_PRICING['bu-latest'] = CUSTOM_MODEL_PRICING['bu-2-0']
# bu-1-0 is redirected to bu-2-0 at the gateway, so it bills at bu-2-0 rates.
CUSTOM_MODEL_PRICING['bu-1-0'] = CUSTOM_MODEL_PRICING['bu-2-0']
CUSTOM_MODEL_PRICING['smart'] = CUSTOM_MODEL_PRICING['bu-2-0']