From b49916333fb2c40902417ebc284e16a55d98a376 Mon Sep 17 00:00:00 2001
From: Praveen11558 <44603409+Praveen11558@users.noreply.github.com>
Date: Tue, 10 Mar 2026 17:14:45 +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 c30aa89f113..e22e99b0aeb 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
@@ -149,7 +149,7 @@ def test_inline_pdf_tokens_costed_when_text_tokens_less_than_prompt_tokens():
         usage= usage,
         model = "gemini-2.0-flash-001",
     )
-    cost = response_cost_calculator(
+from litellm.cost_calculator import response_cost_calculator
         response_object= response,
         model = "gemini-2.0-flash-001",
         custom_llm_provider = "vertex_ai",