diff --git a/tests/logging_callback_tests/test_built_in_tools_cost_tracking.py b/tests/logging_callback_tests/test_built_in_tools_cost_tracking.py index d4013f2e8db..97cc88414bd 100644 --- a/tests/logging_callback_tests/test_built_in_tools_cost_tracking.py +++ b/tests/logging_callback_tests/test_built_in_tools_cost_tracking.py @@ -115,13 +115,13 @@ async def test_openai_web_search_logging_cost_tracking( "tools_config,expected_context_size,stream", [ ( - [{"type": "web_search_preview", "search_context_size": "high"}], - "search_context_size_high", + [{"type": "web_search_preview", "search_context_size": "low"}], + "search_context_size_low", True, ), ( - [{"type": "web_search_preview", "search_context_size": "high"}], - "search_context_size_high", + [{"type": "web_search_preview", "search_context_size": "low"}], + "search_context_size_low", False, ), ([{"type": "web_search_preview"}], "search_context_size_medium", True),