From 05ceff8391a3e0bb2ab398abdb19926c14f0bd2b Mon Sep 17 00:00:00 2001 From: balazss Date: Wed, 18 Mar 2026 01:34:18 -0700 Subject: [PATCH] test(gemini): fix fallback handler patch targets --- tests/test_litellm/llms/gemini/test_fallback_handler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/llms/gemini/test_fallback_handler.py b/tests/test_litellm/llms/gemini/test_fallback_handler.py index 544f241a99b..1648aa61be0 100644 --- a/tests/test_litellm/llms/gemini/test_fallback_handler.py +++ b/tests/test_litellm/llms/gemini/test_fallback_handler.py @@ -18,7 +18,7 @@ def test_run_gemini_completion_with_code_assist_fallback_disabled(): return_value=True, ), patch( - "litellm.llms.gemini.fallback_handler.GoogleCodeAssistChat.completion" + "litellm.llms.gemini.fallback_handler._google_code_assist_chat.completion" ) as mock_completion, ): with pytest.raises(Exception, match="ACCESS_TOKEN_SCOPE_INSUFFICIENT"): @@ -42,7 +42,7 @@ async def test_run_gemini_acompletion_with_code_assist_fallback_enabled(): return_value=True, ), patch( - "litellm.llms.gemini.fallback_handler.GoogleCodeAssistChat.acompletion", + "litellm.llms.gemini.fallback_handler._google_code_assist_chat.acompletion", new_callable=AsyncMock, ) as mock_acompletion, ):