From 89767a6e43096ea6a6adca43151dcb5f1b2cb87e Mon Sep 17 00:00:00 2001
From: Sameer Kankute <sameer@berri.ai>
Date: Thu, 16 Apr 2026 19:24:08 +0530
Subject: [PATCH] litellm_staging_03_22_2026

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

diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index 0acbe90130..9101c886ad 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -812,6 +812,10 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                     },
                     "additionalProperties": False,
                 },
+                "web_search_billing_unit": {
+                    "type": "string",
+                    "enum": ["per_prompt", "per_query"],
+                },
                 "citation_cost_per_token": {"type": "number"},
                 "supported_modalities": {
                     "type": "array",