scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-11 08:38:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 280

test(spend): expect token billing for Vertex image edits with usage (#51109)

Browse source 
Co-authored-by: mateo <277851410+mateo-berri@users.noreply.github.com>
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
devin-ai-integration[bot] 2026-11-11 03:32:19 +00:00 • committed by GitHub
parent 461b56b14b
commit cb024047ac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/integration/spend/test_vertex_image_regional_uplift.py
View file

@ -242,7 +242,7 @@ def test_image_generation_bills_the_deployment_location(
[
pytest.param("global", False, _GLOBAL_IMAGE_COST, id="global-edit"),
pytest.param(_REGION, False, _REGIONAL_IMAGE_COST, id="regional-edit"),
pytest.param(_REGION, True, _REGIONAL_IMAGE_COST, id="regional-edit-usage-is-per-image"),
pytest.param(_REGION, True, _REGIONAL_TOKEN_COST, id="regional-edit-usage-bills-tokens"),
],
)
def test_image_edit_bills_the_deployment_location(
@ -489,8 +489,8 @@ def _burst_calls(global_model: str, regional_model: str) -> tuple[_Call, ...]:
kinds: Final = (
("/v1/images/generations", global_model, _GLOBAL_TOKEN_COST),
("/v1/images/generations", regional_model, _REGIONAL_TOKEN_COST),
("/v1/images/edits", global_model, _GLOBAL_IMAGE_COST),
("/v1/images/edits", regional_model, _REGIONAL_IMAGE_COST),
("/v1/images/edits", global_model, _GLOBAL_TOKEN_COST),
("/v1/images/edits", regional_model, _REGIONAL_TOKEN_COST),
)
return tuple(
_Call(call_id=uuid.uuid4().hex, route=route, model=model, expected=expected)