scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 152

feat(perplexity): add Agent API Nemotron models to cost map

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
shivam 2026-09-09 23:52:07 +00:00
parent 2000642592
commit a7aef366c1
3 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/model_prices_and_context_window_backup.json
View file

@ -61230,6 +61230,7 @@
"litellm_provider": "perplexity",
"mode": "responses",
"supports_web_search": true,
"supports_reasoning": true,
"supports_function_calling": true,
"input_cost_per_token": 1.15e-08,
"output_cost_per_token": 1.7e-07,
@ -61240,6 +61241,7 @@
"litellm_provider": "perplexity",
"mode": "responses",
"supports_web_search": true,
"supports_reasoning": true,
"supports_function_calling": true,
"input_cost_per_token": 2.5e-07,
"output_cost_per_token": 2.5e-06,

2
model_prices_and_context_window.json
View file

@ -61230,6 +61230,7 @@
"litellm_provider": "perplexity",
"mode": "responses",
"supports_web_search": true,
"supports_reasoning": true,
"supports_function_calling": true,
"input_cost_per_token": 1.15e-08,
"output_cost_per_token": 1.7e-07,
@ -61240,6 +61241,7 @@
"litellm_provider": "perplexity",
"mode": "responses",
"supports_web_search": true,
"supports_reasoning": true,
"supports_function_calling": true,
"input_cost_per_token": 2.5e-07,
"output_cost_per_token": 2.5e-06,

14
tests/test_litellm/test_utils.py
View file

@ -188,6 +188,8 @@ def test_get_model_info_resolves_provider_prefixed_model_ids(local_model_cost_ma
("perplexity/perplexity/kimi-k3", True),
("perplexity/perplexity/deepseek-v4-flash-0731", True),
("perplexity/perplexity/kimi-k2.7-code", False),
("perplexity/perplexity/nemotron-3.5-lightning-30b-a3b", True),
("perplexity/perplexity/nemotron-3-ultra-550b-a55b", True),
):
assert litellm.supports_reasoning(model=model) is reasoning, model
@ -199,6 +201,18 @@ def test_get_model_info_resolves_provider_prefixed_model_ids(local_model_cost_ma
assert via_provider["output_cost_per_token"] == 4.4e-06
assert via_provider["mode"] == "responses"
lightning = litellm.get_model_info(
model="perplexity/nemotron-3.5-lightning-30b-a3b", custom_llm_provider="perplexity"
)
assert lightning["key"] == "perplexity/perplexity/nemotron-3.5-lightning-30b-a3b"
assert lightning["input_cost_per_token"] == 1.15e-08
assert lightning["output_cost_per_token"] == 1.7e-07
assert lightning["cache_read_input_token_cost"] == 1.15e-09
assert lightning["mode"] == "responses"
ultra = litellm.get_model_info(model="perplexity/perplexity/nemotron-3-ultra-550b-a55b")
assert ultra["key"] == "perplexity/perplexity/nemotron-3-ultra-550b-a55b"
def test_get_model_info_strips_openai_finetune_ids_without_a_custom_suffix(local_model_cost_map):
info = litellm.get_model_info(model="ft:gpt-4o-2024-08-06:my-org::abc123", custom_llm_provider="openai")