scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Update test_llm_cost_calc_utils.py

Browse source
This commit is contained in:
Praveen11558 2026-03-22 23:35:19 +05:30 • committed by GitHub
parent c5e5a2c584
commit 1492d46df9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py
View file

@ -28,7 +28,6 @@ from litellm.types.utils import (
StandardBuiltInToolsParams,
)
from litellm.cost_calculator import response_cost_calculator
sys.path.insert(
0, os.path.abspath("../../..")
@ -40,18 +39,8 @@ from litellm.litellm_core_utils.llm_cost_calc.utils import (
calculate_cache_writing_cost,
generic_cost_per_token,
)
from litellm.types.utils import CacheCreationTokenDetails, Usage
@pytest.fixture(autouse=True, scope="module")
def setup_local_model_cost():
"""Use the local model cost map for all tests."""
original_model_cost = litellm.model_cost
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
yield
os.environ.pop("LITELLM_LOCAL_MODEL_COST_MAP", None)
litellm.model_cost = original_model_cost
def test_reasoning_tokens_no_price_set():
# Use o1 - o1-mini was deprecated/renamed; o1 has same reasoning-token semantics