scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test(test_amazing_vertex_completion.py): update test to not pick experimental gemini models

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-28 14:27:07 -07:00
parent d928220ed2
commit 4ce59f1a96
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_amazing_vertex_completion.py
View file

@ -50,6 +50,8 @@ VERTEX_MODELS_TO_NOT_TEST = [
"text-bison@001",
"gemini-1.5-pro",
"gemini-1.5-pro-preview-0215",
"gemini-pro-flash",
"gemini-pro-experimental",
]
@ -444,7 +446,9 @@ async def test_async_vertexai_response():
test_models = random.sample(test_models, 1)
test_models += litellm.vertex_language_models # always test gemini-pro
for model in test_models:
print(f"model being tested in async call: {model}")
print(
f"model being tested in async call: {model}, litellm.vertex_language_models: {litellm.vertex_language_models}"
)
if model in VERTEX_MODELS_TO_NOT_TEST or (
"gecko" in model or "32k" in model or "ultra" in model or "002" in model
):