diff --git a/litellm/litellm_core_utils/llm_cost_calc/google.py b/litellm/litellm_core_utils/llm_cost_calc/google.py index a9a04ad00..b42376884 100644 --- a/litellm/litellm_core_utils/llm_cost_calc/google.py +++ b/litellm/litellm_core_utils/llm_cost_calc/google.py @@ -48,6 +48,7 @@ def cost_router( "claude" in model or "llama" in model or "mistral" in model + or "jamba" in model or "codestral" in model ): return "cost_per_token"