diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 4219c8f1a7..be28ae92a6 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -3773,7 +3773,7 @@ def test_vertex_schema_test(): } response = litellm.completion( - model="vertex_ai/gemini-2.5-flash-preview", + model="vertex_ai/gemini-2.5-flash", messages=[{"role": "user", "content": "call the tool"}], tools=[tool], tool_choice="required",