test: update test

2025-07-16 10:27:59 -07:00 · 2025-07-16 10:27:59 -07:00 · 8a1a90bc7a
commit 8a1a90bc7a
parent 9c4b4303d3
1 changed files with 1 additions and 1 deletions
--- 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",