From 0707ef7b684c1e461a41617b78edcaf9fc530e76 Mon Sep 17 00:00:00 2001 From: anton-codes-iac <25830178+anton-codes-iac@users.noreply.github.com> Date: Tue, 8 Sep 2026 15:16:47 +0200 Subject: [PATCH] test(bedrock): update inference profile tests for openai endpoint routing --- .../test_litellm/llms/bedrock/test_bedrock_common_utils.py | 2 -- .../bedrock/test_cross_region_inference_profile_mapping.py | 6 +++--- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/tests/test_litellm/llms/bedrock/test_bedrock_common_utils.py b/tests/test_litellm/llms/bedrock/test_bedrock_common_utils.py index 6f3a3272b7b..ffb7d747739 100644 --- a/tests/test_litellm/llms/bedrock/test_bedrock_common_utils.py +++ b/tests/test_litellm/llms/bedrock/test_bedrock_common_utils.py @@ -891,10 +891,8 @@ def test_bedrock_cross_region_openai_routing(model: str): import litellm from litellm.llms.bedrock.common_utils import get_bedrock_chat_config - # 1. Verify route identification assert BedrockModelInfo.get_bedrock_route(model) == "openai" assert BedrockModelInfo._explicit_openai_route(model) is True - # 2. Verify config mapping config = get_bedrock_chat_config(model) assert isinstance(config, litellm.AmazonBedrockOpenAIConfig) diff --git a/tests/test_litellm/llms/bedrock/test_cross_region_inference_profile_mapping.py b/tests/test_litellm/llms/bedrock/test_cross_region_inference_profile_mapping.py index fda3c8ceb8f..6bb1df90f01 100644 --- a/tests/test_litellm/llms/bedrock/test_cross_region_inference_profile_mapping.py +++ b/tests/test_litellm/llms/bedrock/test_cross_region_inference_profile_mapping.py @@ -150,9 +150,9 @@ def test_proxy_cost_calculation_scenario(): @pytest.mark.parametrize("profile", GPT_5_6_PROFILES, ids=lambda p: p.model_id) -def test_bedrock_gpt_5_6_profiles_route_to_converse(profile, local_model_cost_map): - """GPT-5.6 is served by Converse on bedrock-runtime, never by Invoke.""" - assert BedrockModelInfo.get_bedrock_route(f"bedrock/{profile.model_id}") == "converse" +def test_bedrock_gpt_5_6_profiles_route_to_openai(profile, local_model_cost_map): + """GPT-5.6 is served by Openai on bedrock-runtime.""" + assert BedrockModelInfo.get_bedrock_route(f"bedrock/{profile.model_id}") == "openai" def test_bedrock_gpt_5_6_above_272k_tier_applies_to_cost(local_model_cost_map):