From 46ebf425d56b6369f61188b64e26c8daad87a373 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Thu, 4 Dec 2025 21:39:42 +0530 Subject: [PATCH] Fix : test_vertexai_model_garden_model_completion --- litellm/llms/vertex_ai/common_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/litellm/llms/vertex_ai/common_utils.py b/litellm/llms/vertex_ai/common_utils.py index 836234f6f13..c0dfda00abe 100644 --- a/litellm/llms/vertex_ai/common_utils.py +++ b/litellm/llms/vertex_ai/common_utils.py @@ -34,6 +34,7 @@ class VertexAIModelRoute(str, Enum): BGE = "bge" MODEL_GARDEN = "model_garden" NON_GEMINI = "non_gemini" + OPENAI_COMPATIBLE = "openai" VERTEX_AI_MODEL_ROUTES = [f"{route.value}/" for route in VertexAIModelRoute] @@ -179,8 +180,6 @@ def get_vertex_base_model_name(model: str) -> str: """ # Derive routing prefixes from VertexAIModelRoute enum # Map specific routes to their prefixes (some routes like PARTNER_MODELS, GEMINI don't have prefixes) - - for route in VERTEX_AI_MODEL_ROUTES: if model.startswith(route): return model.replace(route, "", 1)