From 7f3cbd41e389d94ea9b390135f92e7eb86de9947 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Wed, 8 Apr 2026 22:22:30 +0530 Subject: [PATCH] Fix greptile review --- tests/test_litellm/test_main.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/tests/test_litellm/test_main.py b/tests/test_litellm/test_main.py index e3386746fb..3cbf309a14 100644 --- a/tests/test_litellm/test_main.py +++ b/tests/test_litellm/test_main.py @@ -811,12 +811,13 @@ def test_responses_api_bridge_check_global_flag_default_false(): """By default, route_all_chat_openai_to_responses is False and doesn't affect routing.""" from litellm.main import responses_api_bridge_check - with patch("litellm.main._get_model_info_helper") as mock_get_model_info: - mock_get_model_info.return_value = {"max_tokens": 4096} - model_info, model = responses_api_bridge_check( - model="gpt-4o", - custom_llm_provider="openai", - ) + with patch.object(litellm, "route_all_chat_openai_to_responses", False): + with patch("litellm.main._get_model_info_helper") as mock_get_model_info: + mock_get_model_info.return_value = {"max_tokens": 4096} + model_info, model = responses_api_bridge_check( + model="gpt-4o", + custom_llm_provider="openai", + ) assert model_info.get("mode") != "responses"