From 3c8a19e343755ec34117d3de0bd10f88d1ac999a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 27 Aug 2024 14:18:04 -0700 Subject: [PATCH] use cost per token for jamba --- litellm/litellm_core_utils/llm_cost_calc/google.py | 1 + 1 file changed, 1 insertion(+) 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"