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

(fix): token_count calculation during inline data sent during multimodal inputs

Browse source 
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
Praveen11558 2026-03-10 17:13:46 +05:30 • committed by GitHub
parent 6cb9353dfb
commit 69a6eb6256
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -44,10 +44,12 @@ from litellm.types.utils import CacheCreationTokenDetails, Usage
@pytest.fixture(autouse=True)
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():
model = "o1-mini"