test: update test to not use gemini-pro

krrishdholakia · krrishdholakia · commit a649f10e6386 · 2025-04-23T11:31:09.000-07:00
google removed it
diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py
@@ -1784,7 +1784,7 @@ async def test_gemini_pro_function_calling_streaming(sync_mode):
     load_vertex_ai_credentials()
     litellm.set_verbose = True
     data = {
-        "model": "vertex_ai/gemini-pro",
+        "model": "vertex_ai/gemini-1.5-flash",
         "messages": [
             {
                 "role": "user",

Original file line number	Diff line number	Diff line change
`@@ -1784,7 +1784,7 @@ async def test_gemini_pro_function_calling_streaming(sync_mode):`
`1784`	`1784`	`load_vertex_ai_credentials()`
`1785`	`1785`	`litellm.set_verbose = True`
`1786`	`1786`	`data = {`
`1787`		`- "model": "vertex_ai/gemini-pro",`
	`1787`	`+ "model": "vertex_ai/gemini-1.5-flash",`
`1788`	`1788`	`"messages": [`
`1789`	`1789`	`{`
`1790`	`1790`	`"role": "user",`