From 525df3f9a113d750e49e1ec865104c5c5d40bc67 Mon Sep 17 00:00:00 2001 From: "devin-ai-integration[bot]" <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 11 Nov 2026 06:27:58 +0000 Subject: [PATCH] test(spend): expect token billing for Vertex image edits (#51148) 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> --- tests/integration/spend/test_vertex_regional_uplift_spend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration/spend/test_vertex_regional_uplift_spend.py b/tests/integration/spend/test_vertex_regional_uplift_spend.py index fc592ba2a99..40b6aaeb96b 100644 --- a/tests/integration/spend/test_vertex_regional_uplift_spend.py +++ b/tests/integration/spend/test_vertex_regional_uplift_spend.py @@ -393,7 +393,7 @@ def test_image_edit_spend_follows_the_vertex_location(gateway: Gateway, location images: Final = raw.parse() assert images.data is not None and [image.b64_json for image in images.data] == ["aW1n"], images prices: Final = _published_prices(gateway, model) - expected: Final = _rate(prices, "output_cost_per_image") * _uplift(prices, placed) + expected: Final = _standard_cost(prices) * _uplift(prices, placed) assert float(raw.headers["x-litellm-response-cost"]) == pytest.approx(expected), raw.headers (sent,) = _only_generate_calls(wire, 1) forwarded: Final = sent.body.decode()