scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

refactor(cost-map): remove openrouter rows, part 4 of 5 (#45668)

Browse source 
* refactor(cost-map): remove openrouter rows, part 1 of 5 (anthropic to google block)

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>

* refactor(cost-map): remove openrouter rows, part 2 of 5

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>

* refactor(cost-map): remove openrouter rows, part 3 of 5

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>

* refactor(cost-map): remove openrouter rows, part 4 of 5

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>

---------

Co-authored-by: kerry <kerry@berri.ai>
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
devin-ai-integration[bot] 2026-10-09 22:40:46 +00:00 • committed by GitHub
parent ed22fd929c
commit fe3ace9a02
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 0 additions and 8331 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4144
litellm/model_prices_and_context_window_backup.json
View file

File diff suppressed because it is too large Load diff

4144
model_prices_and_context_window.json
View file

File diff suppressed because it is too large Load diff

43
tests/unit/test_openrouter_gpt_image_model_metadata.py
View file

@ -1,43 +0,0 @@
from pathlib import Path
from typing import Final
import pytest
from pydantic import TypeAdapter
from litellm import get_model_info
from litellm.litellm_core_utils.get_llm_provider_logic import get_llm_provider
REPO_ROOT: Final = Path(__file__).parents[2]
COST_MAP_ADAPTER: Final = TypeAdapter(dict[str, dict[str, object]])
MODELS: Final = (
"openrouter/openai/gpt-image-2",
"openrouter/openai/gpt-image-2.5-flare",
"openrouter/openai/gpt-image-2.5-sunburst",
)
PRICE_FIELDS: Final = ("input_cost_per_token", "input_cost_per_image_token", "output_cost_per_image_token")
def _cost_map(path: Path) -> dict[str, dict[str, object]]:
return COST_MAP_ADAPTER.validate_json(path.read_bytes())
MAIN_COST_MAP: Final = _cost_map(REPO_ROOT / "model_prices_and_context_window.json")
BACKUP_COST_MAP: Final = _cost_map(REPO_ROOT / "litellm" / "model_prices_and_context_window_backup.json")
@pytest.mark.parametrize("model", MODELS)
def test_openrouter_gpt_image_row_is_identical_in_main_and_backup(model: str) -> None:
assert model in MAIN_COST_MAP, f"{model} is missing from model_prices_and_context_window.json"
assert BACKUP_COST_MAP.get(model) == MAIN_COST_MAP[model]
@pytest.mark.usefixtures("local_model_cost_map")
@pytest.mark.parametrize("model", MODELS)
def test_openrouter_gpt_image_model_info_comes_from_the_openrouter_row(model: str) -> None:
routed_model, provider, _, _ = get_llm_provider(model=model)
assert (routed_model, provider) == (model.removeprefix("openrouter/"), "openrouter")
info = get_model_info(model=routed_model, custom_llm_provider=provider)
row = MAIN_COST_MAP[model]
assert (info["key"], info["litellm_provider"], info["mode"]) == (model, "openrouter", "image_generation")
assert {field: info[field] for field in PRICE_FIELDS} == {field: row[field] for field in PRICE_FIELDS}