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-23 17:21:40 +05:30 • committed by GitHub
parent 8297eb89ad
commit 2fa8f3dfff
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1456,12 +1456,12 @@ def test_character_count_billing_does_not_fill_prompt_token_gap():
)
prompt_cost, completion_cost = generic_cost_per_token(
model="gemini-2.0-pro",
model="gemini-2.0-flash-001",
usage=usage,
custom_llm_provider="vertex_ai",
)
model_info = litellm.model_cost["gemini-2.0-pro"]
model_info = litellm.model_cost["gemini-2.0-flash-001"]
expected_prompt_cost = (
100 * model_info["input_cost_per_token"]
+ 1000 * model_info.get("input_cost_per_character", 0)
@ -1498,12 +1498,12 @@ def test_video_length_billing_does_not_fill_prompt_token_gap():
)
prompt_cost, completion_cost = generic_cost_per_token(
model="gemini-2.0-pro",
model="gemini-2.0-flash-001",
usage=usage,
custom_llm_provider="vertex_ai",
)
model_info = litellm.model_cost["gemini-2.0-pro"]
model_info = litellm.model_cost["gemini-2.0-flash-001"]
expected_prompt_cost = (
50 * model_info.get("input_cost_per_token", 0)
+ 12.0 * model_info.get("input_cost_per_video_per_second", 0)