This commit is contained in:
Atharva Nair 2026-08-27 18:26:30 -05:00 committed by GitHub
commit 60e5e2ee14
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 41 additions and 1 deletions

View file

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

View file

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

View file

@ -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):