From 4b01641190beee06c6cc1031e248a08b57deb6a3 Mon Sep 17 00:00:00 2001
From: shrey kharbanda <shreshth@berri.ai>
Date: Thu, 24 Sep 2026 00:33:16 +0000
Subject: [PATCH] fix(ci): regenerate pricing schema and narrow raises match

---
 model_prices_and_context_window.schema.json                   | 4 ++--
 .../llms/azure_ai/image_generation/test_cost_calculator.py    | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/model_prices_and_context_window.schema.json b/model_prices_and_context_window.schema.json
index 6b8ce824f69..1a93cdb504a 100644
--- a/model_prices_and_context_window.schema.json
+++ b/model_prices_and_context_window.schema.json
@@ -314,11 +314,11 @@
           "type": "number",
           "minimum": 0
         },
-        "input_cost_per_reference_pixel": {
+        "input_cost_per_query": {
           "type": "number",
           "minimum": 0
         },
-        "input_cost_per_query": {
+        "input_cost_per_reference_pixel": {
           "type": "number",
           "minimum": 0
         },
diff --git a/tests/test_litellm/llms/azure_ai/image_generation/test_cost_calculator.py b/tests/test_litellm/llms/azure_ai/image_generation/test_cost_calculator.py
index 608a04959fc..d9f93e19b9f 100644
--- a/tests/test_litellm/llms/azure_ai/image_generation/test_cost_calculator.py
+++ b/tests/test_litellm/llms/azure_ai/image_generation/test_cost_calculator.py
@@ -180,5 +180,5 @@ def test_unlisted_model_bills_deployment_rates() -> None:
 
 
 def test_non_image_response_raises() -> None:
-    with pytest.raises(ValueError):
+    with pytest.raises(ValueError, match="must be of type ImageResponse"):
         cost_calculator(model="FLUX.2-flex", image_response=object())