scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-05 07:37:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 233

test(proxy): type the unpriceable stream bills as floats

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-10-27 21:54:42 +00:00
parent 9ef0b5569e
commit 68af9f061c
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/unit/proxy/test_usage_cost_egress.py
View file

@ -65,8 +65,8 @@ def test_a_zero_bill_replaces_the_upstream_cost() -> None:
assert billed.usage is not None and billed.usage.cost == 0.0
@pytest.mark.parametrize("bill", [None, -0.001, float("nan"), float("inf"), True])
def test_an_unpriceable_stream_keeps_the_upstream_cost(bill: float | bool | None) -> None:
@pytest.mark.parametrize("bill", [None, -0.001, float("nan"), float("inf")])
def test_an_unpriceable_stream_keeps_the_upstream_cost(bill: float | None) -> None:
chunk: Final = _usage_chunk(UPSTREAM_COST)
assert with_client_stream_usage_cost(chunk, lambda _reply: bill) is chunk