scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 35

test: update testing

Browse source
This commit is contained in:
Krrish Dholakia 2026-03-30 17:48:15 -07:00
parent 6c3de43fcf
commit 13b7c2e602
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/local_testing/test_amazing_vertex_completion.py
View file

@ -2758,7 +2758,7 @@ async def test_partner_models_httpx_ai21():
assert (
url[0]
== "https://us-central1-aiplatform.googleapis.com/v1beta1/projects/pathrise-convert-1606954137718/locations/us-central1/publishers/ai21/models/jamba-1.5-mini@001:rawPredict"
== "https://us-central1-aiplatform.googleapis.com/v1beta1/projects/litellm-ci-cd/locations/us-central1/publishers/ai21/models/jamba-1.5-mini@001:rawPredict"
)
# json loads kwargs
@ -3637,8 +3637,9 @@ def test_gemini_tool_calling_not_working():
},
}
],
"vertex_location": "global",
}
response = completion(model="vertex_ai/gemini-2.0-flash", **args)
response = completion(model="vertex_ai/gemini-3-flash-preview", **args)
print(response)