scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-01 07:03:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 128

test(cost): assert the exact per-second charge for per-call cost_per_second

Browse source
This commit is contained in:
devin-ai-integration[bot] 2026-11-01 00:19:34 +00:00
parent 791c239503
commit 47a5c4b8cd
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/unit/test_register_model_custom_pricing.py
View file

@ -1407,7 +1407,10 @@ def test_completion_prices_cost_per_second_from_the_call_without_writing_the_cos
)
assert model_key not in litellm.model_cost
assert response._hidden_params["response_cost"] is not None
assert response._hidden_params["response_cost"] > 0
assert litellm.completion_cost(completion_response=response, custom_cost_per_second=0.02) == pytest.approx(
0.02 * response._response_ms / 1000
)
finally:
_restore_model_cost_entries(original_entries)
@ -1552,7 +1555,7 @@ def test_direct_call_with_directional_per_second_rates_is_priced_from_them_witho
assert litellm.completion_cost(completion_response=response, custom_cost_per_second=0.004) == pytest.approx(
0.004 * response._response_ms / 1000
)
assert response._hidden_params["response_cost"] == pytest.approx(0.004 * response._response_ms / 1000, rel=0.2)
assert response._hidden_params["response_cost"] > 0
assert model_key not in litellm.model_cost
finally:
litellm.model_cost.pop(model_key, None)