From e9779083c15fe4b408bf5cda71f124d827081664 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Sat, 7 Nov 2026 04:36:26 -0800
Subject: [PATCH] test(count_tokens): read the local beta config in the runtime
 invoke betas test

---
 .../unit/llms/bedrock/count_tokens/test_bedrock_token_counter.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/unit/llms/bedrock/count_tokens/test_bedrock_token_counter.py b/tests/unit/llms/bedrock/count_tokens/test_bedrock_token_counter.py
index ddc6421e9fb..1e552ba07ea 100644
--- a/tests/unit/llms/bedrock/count_tokens/test_bedrock_token_counter.py
+++ b/tests/unit/llms/bedrock/count_tokens/test_bedrock_token_counter.py
@@ -625,6 +625,7 @@ _CALLER_BETAS: Final = (ANTHROPIC_BETA_HEADER_VALUES.PER_TURN_CONTROL_2026_07_01
 
 
 @pytest.mark.asyncio
+@pytest.mark.usefixtures("local_beta_headers_config")
 async def test_runtime_count_sends_the_caller_betas_invoke_inference_keeps_for_the_model() -> None:
     deployment_model: Final = "bedrock/invoke/us.anthropic.claude-opus-4-5-20251101-v1:0"
     model_info: Final[dict[str, JsonValue]] = {"supports_per_turn_control": True}