scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

test: type the pricing test helpers

Browse source
This commit is contained in:
mateo-berri 2026-08-26 15:21:24 -07:00
parent d565860f60
commit fb13b47ee5
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_litellm/test_gemini_tts_native_audio_pricing.py
View file

@ -1,4 +1,5 @@
import json
from collections.abc import Iterator
from pathlib import Path
from typing import Final
@ -66,13 +67,13 @@ LONG_CONTEXT_TIER_FIELDS: Final = (
)
def _load(path: Path) -> dict:
def _load(path: Path) -> dict[str, dict[str, object]]:
with open(path, encoding="utf-8") as f:
return json.load(f)
@pytest.fixture
def local_model_cost_map(monkeypatch):
def local_model_cost_map(monkeypatch: pytest.MonkeyPatch) -> Iterator[None]:
original_model_cost = litellm.model_cost
monkeypatch.setenv("LITELLM_LOCAL_MODEL_COST_MAP", "True")
litellm.model_cost = litellm.get_model_cost_map(url="")