ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

vtx test fix gemini-2.5-flash-lite

Browse Source
This commit is contained in:
Ishaan Jaff 2025-08-07 00:11:10 -07:00
parent 9c5e9d7362
commit dfada882f1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/local_testing/test_amazing_vertex_completion.py
View File

@ -518,7 +518,7 @@ async def test_gemini_pro_vision(provider, sync_mode):
litellm.num_retries = 3
if sync_mode:
resp = litellm.completion(
model="{}/gemini-2.5-flash-lite-preview-0514".format(provider),
model="{}/gemini-2.5-flash-lite".format(provider),
messages=[
{"role": "system", "content": "Be a good bot"},
{
@ -537,7 +537,7 @@ async def test_gemini_pro_vision(provider, sync_mode):
)
else:
resp = await litellm.acompletion(
model="{}/gemini-2.5-flash-lite-preview-0514".format(provider),
model="{}/gemini-2.5-flash-lite".format(provider),
messages=[
{"role": "system", "content": "Be a good bot"},
{
@ -605,7 +605,7 @@ def test_completion_function_plus_pdf(load_pdf):
image_message = {"role": "user", "content": image_content}
response = completion(
model="vertex_ai_beta/gemini-2.5-flash-lite-preview-0514",
model="vertex_ai_beta/gemini-2.5-flash-lite",
messages=[image_message],
stream=False,
)