From af2b7a214f295cd04a59978fc15701e3f96c8afd Mon Sep 17 00:00:00 2001 From: Himanshupdt09 Date: Wed, 25 Mar 2026 20:28:30 +0530 Subject: [PATCH] fixed --- .../litellm_core_utils/llm_cost_calc/test_utils.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_utils.py b/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_utils.py index eb0ae563a1f..f3d70be14b8 100644 --- a/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_utils.py +++ b/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_utils.py @@ -1,18 +1,14 @@ -from unittest.mock import patch import os import sys -from litellm.types.utils import ( - CompletionTokensDetailsWrapper, - PromptTokensDetailsWrapper -) - sys.path.insert( 0, os.path.abspath("../../..") ) # Adds the parent directory to the system path - -from litellm.litellm_core_utils.llm_cost_calc.utils import ( - generic_cost_per_token +from unittest.mock import patch +from litellm.types.utils import ( + CompletionTokensDetailsWrapper, + PromptTokensDetailsWrapper, ) +from litellm.litellm_core_utils.llm_cost_calc.utils import generic_cost_per_token from litellm.types.utils import Usage def test_completion_breakdown_ignores_inflated_text_tokens():