scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

feat(cost): add pricing for azure_ai/gpt-image-2 and gpt-image-to-image

Browse source 
Adds pricing entries for new Azure AI image models.
    Prices:
    - Text Input: /M tokens
    - Image Input: /M tokens
    - Image Output: /M tokens

    Closes #26765
This commit is contained in:
Dotify71 2026-05-03 02:01:51 +05:30
parent c94a8d6514
commit 9f3c5a7fae
3 changed files with 95 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
litellm/model_prices_and_context_window_backup.json
View file

@ -2089,6 +2089,38 @@
"supports_response_schema": true,
"supports_tool_choice": true
},
"azure_ai/gpt-image-2": {
"cache_read_input_image_token_cost": 2e-06,
"cache_read_input_token_cost": 1.25e-06,
"input_cost_per_token": 5e-06,
"input_cost_per_image_token": 8e-06,
"litellm_provider": "azure_ai",
"mode": "image_generation",
"output_cost_per_token": 1e-05,
"output_cost_per_image_token": 3e-05,
"supported_endpoints": [
"/v1/images/generations",
"/v1/images/edits"
],
"supports_vision": true,
"supports_pdf_input": true
},
"azure_ai/gpt-image-to-image": {
"cache_read_input_image_token_cost": 2e-06,
"cache_read_input_token_cost": 1.25e-06,
"input_cost_per_token": 5e-06,
"input_cost_per_image_token": 8e-06,
"litellm_provider": "azure_ai",
"mode": "image_generation",
"output_cost_per_token": 1e-05,
"output_cost_per_image_token": 3e-05,
"supported_endpoints": [
"/v1/images/generations",
"/v1/images/edits"
],
"supports_vision": true,
"supports_pdf_input": true
},
"azure_ai/model_router": {
"input_cost_per_token": 1.4e-07,
"output_cost_per_token": 0,

32
model_prices_and_context_window.json
View file

@ -2103,6 +2103,38 @@
"supports_response_schema": true,
"supports_tool_choice": true
},
"azure_ai/gpt-image-2": {
"cache_read_input_image_token_cost": 2e-06,
"cache_read_input_token_cost": 1.25e-06,
"input_cost_per_token": 5e-06,
"input_cost_per_image_token": 8e-06,
"litellm_provider": "azure_ai",
"mode": "image_generation",
"output_cost_per_token": 1e-05,
"output_cost_per_image_token": 3e-05,
"supported_endpoints": [
"/v1/images/generations",
"/v1/images/edits"
],
"supports_vision": true,
"supports_pdf_input": true
},
"azure_ai/gpt-image-to-image": {
"cache_read_input_image_token_cost": 2e-06,
"cache_read_input_token_cost": 1.25e-06,
"input_cost_per_token": 5e-06,
"input_cost_per_image_token": 8e-06,
"litellm_provider": "azure_ai",
"mode": "image_generation",
"output_cost_per_token": 1e-05,
"output_cost_per_image_token": 3e-05,
"supported_endpoints": [
"/v1/images/generations",
"/v1/images/edits"
],
"supports_vision": true,
"supports_pdf_input": true
},
"azure_ai/model_router": {
"input_cost_per_token": 1.4e-07,
"output_cost_per_token": 0,

31
tests/test_litellm/test_utils.py
View file

@ -3983,3 +3983,34 @@ class TestValidateAndFixThinkingParam:
validate_and_fix_thinking_param(thinking=thinking)
assert "budgetTokens" in thinking
assert "budget_tokens" not in thinking
def test_azure_ai_gpt_image_models_in_cost_map():
"""
Test that azure_ai/gpt-image-2 and azure_ai/gpt-image-to-image entries
are correctly configured in model_prices_and_context_window.json.
Prices:
- Text Input: $5/M tokens
- Image Input: $8/M tokens
- Image Output: $30/M tokens
Closes https://github.com/BerriAI/litellm/issues/26765
"""
import json
from pathlib import Path
json_path = Path(__file__).parents[2] / "model_prices_and_context_window.json"
with open(json_path) as f:
model_cost = json.load(f)
for key in ["azure_ai/gpt-image-2", "azure_ai/gpt-image-to-image"]:
info = model_cost.get(key)
assert info is not None, f"{key} missing from model_prices_and_context_window.json"
assert info["litellm_provider"] == "azure_ai"
assert info["mode"] == "image_generation"
assert info["input_cost_per_token"] == 5e-06
assert info["input_cost_per_image_token"] == 8e-06
assert info["output_cost_per_image_token"] == 3e-05
assert info["cache_read_input_image_token_cost"] == 2e-06
assert info["supports_vision"] is True