scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-30 01:52:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

refactor(azure_ai): pick the generated image meter with early returns

Browse source
This commit is contained in:
shrey kharbanda 2026-09-24 00:35:21 +00:00
parent afff814038
commit e9b603f77a
1 changed files with 13 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
litellm/llms/azure_ai/image_generation/cost_calculator.py
View file

@ -56,15 +56,17 @@ def cost_calculator(
return usage_cost
num_images: Final = n if n is not None else len(image_response.data or ())
generated_meters: Final = (
("output_cost_per_image", num_images),
("input_cost_per_pixel", _generated_pixels(optional_params, size) * num_images),
)
generated_cost: Final = next(
(rate * units for key, units in generated_meters if (rate := _rate(resolved, key)) is not None),
0.0,
)
reference_cost: Final = (_rate(resolved, "input_cost_per_reference_pixel") or 0.0) * (
image_response.reference_pixels or 0
)
generated_cost: Final = _generated_cost(resolved, num_images, _generated_pixels(optional_params, size))
reference_rate: Final = _rate(resolved, "input_cost_per_reference_pixel") or 0.0
reference_cost: Final = reference_rate * (image_response.reference_pixels or 0)
return generated_cost + reference_cost
def _generated_cost(resolved: ModelInfo, num_images: int, pixels_per_image: int) -> float:
per_image: Final = _rate(resolved, "output_cost_per_image")
if per_image is not None:
return per_image * num_images
per_pixel: Final = _rate(resolved, "input_cost_per_pixel")
if per_pixel is not None:
return per_pixel * pixels_per_image * num_images
return 0.0