Skip to content

Commit f7af18d

Browse files
test(test_amazing_vertex_completion.py): fix test
1 parent 069704c commit f7af18d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

litellm/tests/test_amazing_vertex_completion.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,8 @@ def test_vertex_ai_stream():
159159
"code-gecko@latest",
160160
"code-bison@001",
161161
"text-bison@001",
162+
"gemini-1.5-pro",
163+
"gemini-1.5-pro-vision",
162164
]:
163165
# our account does not have access to this model
164166
continue

0 commit comments

Comments
 (0)