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(openai): pass the provider explicitly so the gpt-6-astra tier rows survive a remote cost map

Browse source
This commit is contained in:
mateo-berri 2026-09-03 12:15:15 -07:00
parent b22bf99cc7
commit 7564993f3e
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/test_openai_service_tier_long_context_pricing.py
View file

@ -131,6 +131,7 @@ def test_cost_per_token_bills_long_context_at_the_tier_rate(
"""A prompt over 272K on flex or priority must bill at that tier's long-context rate."""
input_cost, output_cost = litellm.cost_per_token(
model=model,
custom_llm_provider="openai",
prompt_tokens=LONG_CONTEXT_PROMPT_TOKENS,
completion_tokens=COMPLETION_TOKENS,
service_tier=tier,
@ -148,6 +149,7 @@ def test_cost_per_token_tier_differs_from_the_standard_long_context_cost(
standard = sum(
litellm.cost_per_token(
model=model,
custom_llm_provider="openai",
prompt_tokens=LONG_CONTEXT_PROMPT_TOKENS,
completion_tokens=COMPLETION_TOKENS,
)
@ -155,6 +157,7 @@ def test_cost_per_token_tier_differs_from_the_standard_long_context_cost(
tiered = sum(
litellm.cost_per_token(
model=model,
custom_llm_provider="openai",
prompt_tokens=LONG_CONTEXT_PROMPT_TOKENS,
completion_tokens=COMPLETION_TOKENS,
service_tier=tier,