add cost_router for vertexai

This commit is contained in:
Ishaan Jaffer 2025-10-09 18:39:29 -07:00
parent be7c24ebf5
commit 5ce27bfad6

View file

@ -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 (