phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: update test to not use gemini-pro

Browse source 
google removed it
This commit is contained in:
Krrish Dholakia 2025-04-23 11:31:09 -07:00
parent 7b82144fd0
commit 9965e4a7de
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/local_testing/test_amazing_vertex_completion.py
View file

@ -1784,7 +1784,7 @@ async def test_gemini_pro_function_calling_streaming(sync_mode):
load_vertex_ai_credentials() load_vertex_ai_credentials()
litellm.set_verbose = True litellm.set_verbose = True
data = { data = {
"model": "vertex_ai/gemini-pro", "model": "vertex_ai/gemini-1.5-flash",
"messages": [ "messages": [
{ {
"role": "user", "role": "user",