diff --git a/tests/llm_translation/test_gemini.py b/tests/llm_translation/test_gemini.py index aa349253e5..c34e59a73e 100644 --- a/tests/llm_translation/test_gemini.py +++ b/tests/llm_translation/test_gemini.py @@ -452,7 +452,7 @@ def test_gemini_finish_reason(): litellm._turn_on_debug() response = completion( - model="gemini/gemini-1.5-pro", + model="gemini/gemini-2.5-flash-lite", messages=[{"role": "user", "content": "give me 3 random words"}], max_tokens=2, )