From 107a07f6159c9b0ac60ef0d16011e17c6b9fe10a Mon Sep 17 00:00:00 2001 From: Scott Wilson Date: Mon, 28 Sep 2026 16:24:23 -0400 Subject: [PATCH] test(anthropic): expect the per-turn-control beta to reach Bedrock Invoke Generated with AI Co-Authored-By: Claude Code --- .../messages/test_anthropic_messages_per_turn_control.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/unit/llms/anthropic/pass_through/messages/test_anthropic_messages_per_turn_control.py b/tests/unit/llms/anthropic/pass_through/messages/test_anthropic_messages_per_turn_control.py index 4197192e4af..ff5ad701e45 100644 --- a/tests/unit/llms/anthropic/pass_through/messages/test_anthropic_messages_per_turn_control.py +++ b/tests/unit/llms/anthropic/pass_through/messages/test_anthropic_messages_per_turn_control.py @@ -95,15 +95,16 @@ def test_added_per_turn_control_beta_survives_the_anthropic_allowlist(): assert PER_TURN_CONTROL in _betas(filtered) -@pytest.mark.parametrize("provider", ["bedrock", "bedrock_converse", "vertex_ai", "databricks"]) +@pytest.mark.parametrize("provider", ["bedrock_converse", "vertex_ai", "databricks"]) def test_per_turn_control_beta_is_dropped_for_providers_without_it(provider): filtered = update_headers_with_filtered_beta(headers={"anthropic-beta": PER_TURN_CONTROL}, provider=provider) assert "anthropic-beta" not in filtered -def test_per_turn_control_beta_is_forwarded_for_azure_ai(): - filtered = update_headers_with_filtered_beta(headers={"anthropic-beta": PER_TURN_CONTROL}, provider="azure_ai") +@pytest.mark.parametrize("provider", ["azure_ai", "bedrock"]) +def test_per_turn_control_beta_is_forwarded(provider): + filtered = update_headers_with_filtered_beta(headers={"anthropic-beta": PER_TURN_CONTROL}, provider=provider) assert _betas(filtered) == {PER_TURN_CONTROL}