From 1bdb94a3148698776f2acea65c976d8eac49b33a Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 22 Mar 2025 12:56:21 -0700
Subject: [PATCH] add search_context_cost_per_1k_calls to model cost map spec

---
 tests/local_testing/test_get_model_info.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/local_testing/test_get_model_info.py b/tests/local_testing/test_get_model_info.py
index d71f3f7c24..f885a61e78 100644
--- a/tests/local_testing/test_get_model_info.py
+++ b/tests/local_testing/test_get_model_info.py
@@ -518,6 +518,15 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                         ],
                     },
                 },
+                "search_context_cost_per_1k_calls": {
+                    "type": "object",
+                    "properties": {
+                        "low_context": {"type": "number"},
+                        "medium_context": {"type": "number"},
+                        "high_context": {"type": "number"},
+                    },
+                    "additionalProperties": False,
+                },
                 "supported_modalities": {
                     "type": "array",
                     "items": {