From 6587cd228b2aa18ae1061a2dc13437f1d0a85ce2 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 24 Jan 2026 10:58:35 -0800 Subject: [PATCH] test_partner_models_httpx_streaming --- tests/local_testing/test_amazing_vertex_completion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 1987a7a545..25a85384ca 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -813,7 +813,7 @@ from test_completion import response_format_tests @pytest.mark.parametrize( "model,region", [ - ("vertex_ai/mistral-large-2411", "us-central1"), + ("vertex_ai/mistral-small-2503", "us-central1"), ("vertex_ai/qwen/qwen3-coder-480b-a35b-instruct-maas", "us-south1"), ("vertex_ai/openai/gpt-oss-20b-maas", "us-central1"), ], @@ -886,7 +886,7 @@ async def test_partner_models_httpx(model, region, sync_mode): ("vertex_ai/meta/llama-4-scout-17b-16e-instruct-maas", "us-east5"), ("vertex_ai/qwen/qwen3-coder-480b-a35b-instruct-maas", "us-south1"), ( - "vertex_ai/mistral-large-2411", + "vertex_ai/mistral-small-2503", "us-central1", ), # critical - we had this issue: https://github.com/BerriAI/litellm/issues/13888 ("vertex_ai/openai/gpt-oss-20b-maas", "us-central1"),