From 13b7c2e60251fdb28211ca7ef4da04bc36ec83b2 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 30 Mar 2026 17:48:15 -0700 Subject: [PATCH] test: update testing --- tests/local_testing/test_amazing_vertex_completion.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 123dfc095c..a1684e2376 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -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)