From 7564993f3e592a1849ffd17addd250f67967e48e Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Thu, 3 Sep 2026 12:15:15 -0700
Subject: [PATCH] test(openai): pass the provider explicitly so the gpt-6-astra
 tier rows survive a remote cost map

---
 .../test_openai_service_tier_long_context_pricing.py           | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/test_litellm/test_openai_service_tier_long_context_pricing.py b/tests/test_litellm/test_openai_service_tier_long_context_pricing.py
index 70e9c2720b8..8a5c22f0fbc 100644
--- a/tests/test_litellm/test_openai_service_tier_long_context_pricing.py
+++ b/tests/test_litellm/test_openai_service_tier_long_context_pricing.py
@@ -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,