mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
add cost_router for vertexai
This commit is contained in:
parent
be7c24ebf5
commit
5ce27bfad6
1 changed files with 1 additions and 0 deletions
|
|
@ -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 (
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue