scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-21 00:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 74

test(gemini): fix fallback handler patch targets

Browse source
This commit is contained in:
balazss 2026-03-18 01:34:18 -07:00
parent 0872a655fd
commit 05ceff8391
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/llms/gemini/test_fallback_handler.py
View file

@ -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,
):