mirror of
https://github.com/BerriAI/litellm.git
synced 2026-11-11 08:38:24 +00:00
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>
This commit is contained in:
parent
97f07af044
commit
525df3f9a1
1 changed files with 1 additions and 1 deletions
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue