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 84

test: annotate parametrized traffic-type test inputs

Browse source
This commit is contained in:
Bisma Nawaz 2026-08-21 03:42:34 +05:00
parent d317c5621f
commit 909ab23b89
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/llms/gemini/test_cost_calculator.py
View file

@ -318,7 +318,9 @@ def test_gemini_image_generation_cost_no_web_search_when_absent():
("SOMETHING_UNKNOWN", None),
],
)
def test_map_traffic_type_to_service_tier(traffic_type, expected_service_tier):
def test_map_traffic_type_to_service_tier(
traffic_type: str | None, expected_service_tier: str | None
):
"""
Gemini/Vertex usageMetadata.trafficType maps to the LiteLLM service_tier
that selects flex/priority cost keys. ON_DEMAND_FLEX (Vertex's flex opt-in