mirror of
https://github.com/browser-use/browser-use.git
synced 2026-09-14 19:59:47 +08:00
Saurav Panda
34 lines
1.2 KiB
Python
34 lines
1.2 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-1-0': {
|
|
'input_cost_per_token': 0.2 / 1_000_000, # $0.20 per 1M tokens
|
|
'output_cost_per_token': 2.00 / 1_000_000, # $2.00 per 1M tokens
|
|
'cache_read_input_token_cost': 0.02 / 1_000_000, # $0.02 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': {
|
|
'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
|
|
},
|
|
}
|
|
CUSTOM_MODEL_PRICING['bu-latest'] = CUSTOM_MODEL_PRICING['bu-1-0']
|
|
|
|
CUSTOM_MODEL_PRICING['smart'] = CUSTOM_MODEL_PRICING['bu-1-0']
|