From 36d17c23aca69d9f461b4776ac48b0df291b3606 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 24 Mar 2025 15:51:33 -0700 Subject: [PATCH] fix flaky test test_openai_responses_api_web_search_cost_tracking --- .../test_built_in_tools_cost_tracking.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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),