From dab7bebaf2bc34be1f095a77636531111fe71d8e Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 11 Jan 2025 21:08:15 -0800
Subject: [PATCH] use _get_model_info_helper (#7703)

---
 litellm/cost_calculator.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/litellm/cost_calculator.py b/litellm/cost_calculator.py
index 9690a4d8b4..93961b1372 100644
--- a/litellm/cost_calculator.py
+++ b/litellm/cost_calculator.py
@@ -60,6 +60,7 @@ from litellm.utils import (
     ModelResponse,
     TextCompletionResponse,
     TranscriptionResponse,
+    _get_model_info_helper,
     print_verbose,
     token_counter,
 )
@@ -278,7 +279,7 @@ def cost_per_token(  # noqa: PLR0915
     elif custom_llm_provider == "deepseek":
         return deepseek_cost_per_token(model=model, usage=usage_block)
     else:
-        model_info = litellm.get_model_info(
+        model_info = _get_model_info_helper(
             model=model, custom_llm_provider=custom_llm_provider
         )