From 69ef062f5565f9411e8648d758a52da878cc52c6 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 11 Sep 2025 19:56:44 -0700
Subject: [PATCH] fix tiered_pricing test

---
 ...odel_prices_and_context_window_backup.json |  5 -----
 model_prices_and_context_window.json          |  5 -----
 tests/test_litellm/test_utils.py              | 20 +++++++++++++++++++
 3 files changed, 20 insertions(+), 10 deletions(-)

diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json
index 972e212e693..46331c7cff7 100644
--- a/litellm/model_prices_and_context_window_backup.json
+++ b/litellm/model_prices_and_context_window_backup.json
@@ -19132,7 +19132,6 @@
         "supports_function_calling": true,
         "supports_tool_choice": true,
         "supports_reasoning": true,
-        "supports_caching": true,
         "mode": "chat",
         "tiered_pricing": [
             {"range": [0, 256e3], "input_cost_per_token": 5e-08, "output_cost_per_token": 4e-07},
@@ -19161,7 +19160,6 @@
         "supports_function_calling": true,
         "supports_tool_choice": true,
         "supports_reasoning": true,
-        "supports_caching": true,
         "mode": "chat",
         "tiered_pricing": [
             {"range": [0, 32e3], "input_cost_per_token": 1e-06, "output_cost_per_token": 5e-06, "cache_read_input_token_cost": 1e-07},
@@ -19196,7 +19194,6 @@
         "supports_function_calling": true,
         "supports_tool_choice": true,
         "supports_reasoning": true,
-        "supports_caching": true,
         "mode": "chat",
         "tiered_pricing": [
             {"range": [0, 32e3], "input_cost_per_token": 3e-07, "output_cost_per_token": 1.5e-06, "cache_read_input_token_cost": 8e-08},
@@ -19302,7 +19299,6 @@
         "supports_function_calling": true,
         "supports_tool_choice": true,
         "supports_reasoning": true,
-        "supports_caching": true,
         "mode": "chat",
         "tiered_pricing": [
             {"range": [0, 256e3], "input_cost_per_token": 5e-08, "output_cost_per_token": 4e-07},
@@ -19355,7 +19351,6 @@
         "max_tokens": 131072,
         "max_input_tokens": 98304,
         "max_output_tokens": 8192,
-        "max_chain_of_thought_tokens": 32768,
         "input_cost_per_token": 8e-07,
         "output_cost_per_token": 2.4e-06,
         "litellm_provider": "dashscope",
diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index 972e212e693..46331c7cff7 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -19132,7 +19132,6 @@
         "supports_function_calling": true,
         "supports_tool_choice": true,
         "supports_reasoning": true,
-        "supports_caching": true,
         "mode": "chat",
         "tiered_pricing": [
             {"range": [0, 256e3], "input_cost_per_token": 5e-08, "output_cost_per_token": 4e-07},
@@ -19161,7 +19160,6 @@
         "supports_function_calling": true,
         "supports_tool_choice": true,
         "supports_reasoning": true,
-        "supports_caching": true,
         "mode": "chat",
         "tiered_pricing": [
             {"range": [0, 32e3], "input_cost_per_token": 1e-06, "output_cost_per_token": 5e-06, "cache_read_input_token_cost": 1e-07},
@@ -19196,7 +19194,6 @@
         "supports_function_calling": true,
         "supports_tool_choice": true,
         "supports_reasoning": true,
-        "supports_caching": true,
         "mode": "chat",
         "tiered_pricing": [
             {"range": [0, 32e3], "input_cost_per_token": 3e-07, "output_cost_per_token": 1.5e-06, "cache_read_input_token_cost": 8e-08},
@@ -19302,7 +19299,6 @@
         "supports_function_calling": true,
         "supports_tool_choice": true,
         "supports_reasoning": true,
-        "supports_caching": true,
         "mode": "chat",
         "tiered_pricing": [
             {"range": [0, 256e3], "input_cost_per_token": 5e-08, "output_cost_per_token": 4e-07},
@@ -19355,7 +19351,6 @@
         "max_tokens": 131072,
         "max_input_tokens": 98304,
         "max_output_tokens": 8192,
-        "max_chain_of_thought_tokens": 32768,
         "input_cost_per_token": 8e-07,
         "output_cost_per_token": 2.4e-06,
         "litellm_provider": "dashscope",
diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index f9f150b6234..0c7aa527240 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -652,6 +652,26 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                     },
                 },
                 "supports_native_streaming": {"type": "boolean"},
+                "tiered_pricing": {
+                    "type": "array",
+                    "items": {
+                        "type": "object",
+                        "properties": {
+                            "range": {
+                                "type": "array",
+                                "items": {"type": "number"},
+                                "minItems": 2,
+                                "maxItems": 2
+                            },
+                            "input_cost_per_token": {"type": "number"},
+                            "output_cost_per_token": {"type": "number"},
+                            "cache_read_input_token_cost": {"type": "number"},
+                            "output_cost_per_reasoning_token": {"type": "number"}
+                        },
+                        "required": ["range"],
+                        "additionalProperties": False
+                    }
+                },
             },
             "additionalProperties": False,
         },