scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(cost): cover garbage-string token counts

Browse source
This commit is contained in:
soroush5 2026-09-03 23:39:50 +03:30 • committed by soroush5
parent cc2d9f10df
commit 5573c446fc
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/test_litellm/test_cost_calculator.py
View file

@ -4785,3 +4785,13 @@ def test_cost_per_token_string_token_counts_coerced():
)
assert prompt_cost == expected_prompt
assert completion_cost == expected_completion
def test_cost_per_token_garbage_string_counts_cost_zero():
prompt_cost, completion_cost = cost_per_token(
model="gpt-4o-mini", prompt_tokens="abc", completion_tokens=5
)
expected_prompt, _ = cost_per_token(
model="gpt-4o-mini", prompt_tokens=0, completion_tokens=5
)
assert prompt_cost == expected_prompt