From 6056d8d82d6bd3ebb421304e6750baea94f1e8ca Mon Sep 17 00:00:00 2001
From: Praveen11558 <44603409+Praveen11558@users.noreply.github.com>
Date: Mon, 23 Mar 2026 17:30:47 +0530
Subject: [PATCH] Update
 tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
---
 .../llm_cost_calc/test_llm_cost_calc_utils.py                   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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