diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index 77572f69b8b..bf5b36f526b 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -50222,6 +50222,25 @@ "supports_tool_choice": true, "supports_vision": true }, + "bedrock_mantle/google.gemma-4-e4b": { + "input_cost_per_token": 0.0, + "output_cost_per_token": 0.0, + "litellm_provider": "bedrock_mantle", + "max_input_tokens": 128000, + "max_output_tokens": 128000, + "max_tokens": 128000, + "mode": "chat", + "use_openai_responses_path": true, + "supported_endpoints": [ + "/v1/chat/completions", + "/v1/responses" + ], + "supports_function_calling": true, + "supports_parallel_function_calling": false, + "supports_reasoning": true, + "supports_tool_choice": true, + "supports_vision": true + }, "bedrock_mantle/xai.grok-4.3": { "use_openai_responses_path": true, "input_cost_per_token": 1.25e-06, diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index 77572f69b8b..bf5b36f526b 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -50222,6 +50222,25 @@ "supports_tool_choice": true, "supports_vision": true }, + "bedrock_mantle/google.gemma-4-e4b": { + "input_cost_per_token": 0.0, + "output_cost_per_token": 0.0, + "litellm_provider": "bedrock_mantle", + "max_input_tokens": 128000, + "max_output_tokens": 128000, + "max_tokens": 128000, + "mode": "chat", + "use_openai_responses_path": true, + "supported_endpoints": [ + "/v1/chat/completions", + "/v1/responses" + ], + "supports_function_calling": true, + "supports_parallel_function_calling": false, + "supports_reasoning": true, + "supports_tool_choice": true, + "supports_vision": true + }, "bedrock_mantle/xai.grok-4.3": { "use_openai_responses_path": true, "input_cost_per_token": 1.25e-06, diff --git a/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_transformation.py b/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_transformation.py index cd775abf136..2c70348b868 100644 --- a/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_transformation.py +++ b/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_transformation.py @@ -174,7 +174,7 @@ class TestBedrockMantleConfig: @pytest.mark.parametrize( "model_id", - ["google.gemma-4-31b", "google.gemma-4-26b-a4b", "google.gemma-4-e2b"], + ["google.gemma-4-31b", "google.gemma-4-26b-a4b", "google.gemma-4-e2b", "google.gemma-4-e4b"], ) def test_chat_base_for_gemma_4_uses_openai_v1( self, monkeypatch, local_cost_map, model_id @@ -641,6 +641,7 @@ class TestBedrockMantlePricing: ("google.gemma-4-31b", 1.4e-07, 4e-07, 256000), ("google.gemma-4-26b-a4b", 1.3e-07, 4e-07, 256000), ("google.gemma-4-e2b", 4e-08, 8e-08, 128000), + ("google.gemma-4-e4b", 0.0, 0.0, 128000), ], ) def test_gemma_4_bedrock_mantle_model_metadata( @@ -672,6 +673,7 @@ def test_gemma_4_bedrock_mantle_model_metadata( "google.gemma-4-31b", "google.gemma-4-26b-a4b", "google.gemma-4-e2b", + "google.gemma-4-e4b", ], ) def test_gemma_4_models_register_under_bedrock_mantle(local_cost_map, model_id):