diff --git a/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py b/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py index 37330207bae..18c0dca9315 100644 --- a/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py +++ b/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py @@ -41,7 +41,7 @@ from litellm.litellm_core_utils.llm_cost_calc.utils import ( ) from litellm.types.utils import CacheCreationTokenDetails, Usage -from litellm.cost_calculator import completion_cost, response_cost_calculator +from litellm.types.utils import CacheCreationTokenDetails, Usage def test_reasoning_tokens_no_price_set(): # Use o1 - o1-mini was deprecated/renamed; o1 has same reasoning-token semantics