From fb13b47ee59552bdd4fef11f1354085d6beae686 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Wed, 26 Aug 2026 15:21:24 -0700
Subject: [PATCH] test: type the pricing test helpers

---
 tests/test_litellm/test_gemini_tts_native_audio_pricing.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_litellm/test_gemini_tts_native_audio_pricing.py b/tests/test_litellm/test_gemini_tts_native_audio_pricing.py
index 803e112d5c9..3679c73aecd 100644
--- a/tests/test_litellm/test_gemini_tts_native_audio_pricing.py
+++ b/tests/test_litellm/test_gemini_tts_native_audio_pricing.py
@@ -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="")