scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 114

Update test_llm_cost_calc_utils.py

Browse source
This commit is contained in:
Praveen11558 2026-03-10 17:39:24 +05:30 • committed by GitHub
parent a9e51e6ee7
commit 69fe5b57c1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -209,7 +209,8 @@ def test_inline_pdf_tokens_costed_when_text_both_costed_correctly():
f"Unaccounted PDF tokens alongside audio are not being costed."
)
def test_inline_pdf_with_audio_tokens_both_costed_correctly(self):
def test_inline_pdf_with_audio_tokens_both_costed_correctly():
"""
Scenario: User sends a PDF inline + audio + text.
Provider reports:
@ -261,7 +262,7 @@ def test_inline_pdf_with_audio_tokens_both_costed_correctly(self):
f"Unaccounted PDF tokens alongside audio are not being costed."
)
def test_no_prompt_tokens_details_all_tokens_costed_as_text(self):
def test_no_prompt_tokens_details_all_tokens_costed_as_text():
"""
Scenario: Provider returns no prompt_tokens_details at all.
Expected: All prompt_tokens should default to text_tokens and be costed.
@ -295,7 +296,7 @@ def test_inline_pdf_with_audio_tokens_both_costed_correctly(self):
f"With no prompt_tokens_details, all tokens should be costed as text."
)
def test_fully_accounted_tokens_no_change(self):
def test_fully_accounted_tokens_no_change():
"""
Scenario: All prompt tokens are fully accounted for in details.
Expected: No adjustment needed, cost calculated normally.
@ -336,7 +337,7 @@ def test_inline_pdf_with_audio_tokens_both_costed_correctly(self):
f"Fully accounted tokens should not be adjusted."
)
def test_double_counting_still_handled(self):
def test_double_counting_still_handled():
"""
Scenario: xAI-style double counting where text_tokens includes cached_tokens.
Provider reports:
@ -386,7 +387,7 @@ def test_inline_pdf_with_audio_tokens_both_costed_correctly(self):
f"Double-counting fix should still work."
)
def test_large_pdf_small_text_message(self):
def test_large_pdf_small_text_message():
"""
Scenario: A large PDF (~50 pages) with a tiny instruction.
This is the most common real-world case.