phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: testing fixes

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-27 20:37:08 -08:00
parent d70149fe49
commit ce46d11a37
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_amazing_vertex_completion.py
View file

@ -110,6 +110,7 @@ def test_vertex_ai():
"code-bison@001", "code-bison@001",
"text-bison@001", "text-bison@001",
"gemini-1.5-pro", "gemini-1.5-pro",
"gemini-1.5-pro-preview-0215",
"gemini-1.5-pro-vision", "gemini-1.5-pro-vision",
]: ]:
# our account does not have access to this model # our account does not have access to this model
@ -160,6 +161,7 @@ def test_vertex_ai_stream():
"code-bison@001", "code-bison@001",
"text-bison@001", "text-bison@001",
"gemini-1.5-pro", "gemini-1.5-pro",
"gemini-1.5-pro-preview-0215",
"gemini-1.5-pro-vision", "gemini-1.5-pro-vision",
]: ]:
# our account does not have access to this model # our account does not have access to this model
@ -211,6 +213,7 @@ async def test_async_vertexai_response():
"code-bison@001", "code-bison@001",
"text-bison@001", "text-bison@001",
"gemini-1.5-pro", "gemini-1.5-pro",
"gemini-1.5-pro-preview-0215",
"gemini-1.5-pro-vision", "gemini-1.5-pro-vision",
]: ]:
# our account does not have access to this model # our account does not have access to this model
@ -255,6 +258,7 @@ async def test_async_vertexai_streaming_response():
"code-bison@001", "code-bison@001",
"text-bison@001", "text-bison@001",
"gemini-1.5-pro", "gemini-1.5-pro",
"gemini-1.5-pro-preview-0215",
"gemini-1.5-pro-vision", "gemini-1.5-pro-vision",
]: ]:
# our account does not have access to this model # our account does not have access to this model