From 5ce27bfad6dc429cfcaf6852078aca3521e553f4 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Thu, 9 Oct 2025 18:39:29 -0700 Subject: [PATCH] add cost_router for vertexai --- litellm/llms/vertex_ai/cost_calculator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/llms/vertex_ai/cost_calculator.py b/litellm/llms/vertex_ai/cost_calculator.py index 119ba2b0366..e98dc75915d 100644 --- a/litellm/llms/vertex_ai/cost_calculator.py +++ b/litellm/llms/vertex_ai/cost_calculator.py @@ -44,6 +44,7 @@ def cost_router( or "mistral" in model or "jamba" in model or "codestral" in model + or "gemma" in model ): return "cost_per_token" elif custom_llm_provider == "vertex_ai" and (