From 78c371d2e88afd7d1b0ef18d8d77e22469824931 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 22 Mar 2025 13:08:57 -0700 Subject: [PATCH] search_context_cost_per_query test --- tests/local_testing/test_get_model_info.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/local_testing/test_get_model_info.py b/tests/local_testing/test_get_model_info.py index f885a61e78d..4cca96481d2 100644 --- a/tests/local_testing/test_get_model_info.py +++ b/tests/local_testing/test_get_model_info.py @@ -518,12 +518,12 @@ def test_aaamodel_prices_and_context_window_json_is_valid(): ], }, }, - "search_context_cost_per_1k_calls": { + "search_context_cost_per_query": { "type": "object", "properties": { - "low_context": {"type": "number"}, - "medium_context": {"type": "number"}, - "high_context": {"type": "number"}, + "search_context_size_low": {"type": "number"}, + "search_context_size_medium": {"type": "number"}, + "search_context_size_high": {"type": "number"}, }, "additionalProperties": False, },