add search_context_cost_per_1k_calls to model cost map spec

2025-03-22 12:56:21 -07:00 · 2025-03-22 12:56:21 -07:00 · 1bdb94a314
commit 1bdb94a314
parent 792a2d6115
1 changed files with 9 additions and 0 deletions
--- 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": {