scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

fix(tests): stabilize gpt-image-2 cost tests against local map

Browse source 
Force gpt-image cost tests to use the bundled model cost map to avoid remote map drift in CI, and update gpt-image-2 output token expectations to the current official pricing.
This commit is contained in:
root 2026-04-24 11:49:03 +08:00
parent 909c5e1302
commit 6554633e11
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/test_litellm/test_gpt_image_cost_calculator.py
View file

@ -11,13 +11,14 @@ gpt-image-1 uses token-based pricing:
gpt-image-2 uses token-based pricing:
- Image Input: $8.00/1M tokens
- Image Output: $32.00/1M tokens
- Image Output: $30.00/1M tokens
"""
import os
import sys
sys.path.insert(0, os.path.abspath("../.."))
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
import pytest
@ -316,7 +317,7 @@ class TestGPTImage2CostCalculator:
gpt-image-2 pricing (https://openai.com/api/pricing/):
- Image Input: $8.00 / 1M tokens (8e-6 per token)
- Image Output: $32.00 / 1M tokens (3.2e-5 per token)
- Image Output: $30.00 / 1M tokens (3e-5 per token)
"""
def test_gpt_image_2_token_cost(self):
@ -347,9 +348,9 @@ class TestGPTImage2CostCalculator:
# Expected cost:
# Image input: 1000 * $8/1M = 0.008
# Image output: 5000 * $32/1M = 0.16
# Total: 0.168
expected_cost = 1000 * 8e-6 + 5000 * 3.2e-5
# Image output: 5000 * $30/1M = 0.15
# Total: 0.158
expected_cost = 1000 * 8e-6 + 5000 * 3e-5
assert abs(cost - expected_cost) < 1e-6, f"Expected {expected_cost}, got {cost}"
def test_gpt_image_2_in_model_prices(self):
@ -359,7 +360,7 @@ class TestGPTImage2CostCalculator:
assert model_info.get("litellm_provider") == "openai"
assert model_info.get("mode") == "image_generation"
assert model_info.get("input_cost_per_image_token") == 8e-6
assert model_info.get("output_cost_per_image_token") == 3.2e-5
assert model_info.get("output_cost_per_image_token") == 3e-5
def test_gpt_image_2_per_quality_entries(self):
"""Test that per-quality per-size entries exist for gpt-image-2."""