From 81142d42e27abe1824657f27705efd6890f2a805 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Sat, 5 Sep 2026 01:02:56 -0700
Subject: [PATCH] test(cost-map): accept any *_above_<N>k_tokens breakpoint in
 the map schema

---
 tests/test_litellm/test_utils.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index 9d90d47e022..5a6aa20657d 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -1195,6 +1195,11 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                     },
                 },
             },
+            "patternProperties": {
+                "^(input_cost_per_token|output_cost_per_token|cache_read_input_token_cost|cache_creation_input_token_cost)_above_[0-9]+k_tokens$": {
+                    "type": "number"
+                },
+            },
             "additionalProperties": False,
         },
     }