From 2414f8fcf0a24a494094d0d6624379d5635859c2 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Tue, 22 Sep 2026 10:38:09 -0700
Subject: [PATCH] test: add the local_beta_headers_config fixture the
 safeguards tests use

Hand-ported to stable/1.101.x from 47b2479c94c5b00270b74fe4d22aff6be0add6cf on main (fix(bedrock): gate Invoke tool search on the model map's supports_tool_search flag), the one prerequisite the #42288 handler tests need; the rest of that commit stays on main.
---
 tests/test_litellm/conftest.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/tests/test_litellm/conftest.py b/tests/test_litellm/conftest.py
index 62c95cb100b..183a13e3ca7 100644
--- a/tests/test_litellm/conftest.py
+++ b/tests/test_litellm/conftest.py
@@ -203,6 +203,21 @@ def local_model_cost_map(monkeypatch):
         litellm.get_model_info.cache_clear()
 
 
+@pytest.fixture
+def local_beta_headers_config(monkeypatch):
+    """Pin the bundled ``anthropic_beta_headers_config.json`` so beta header assertions
+    do not depend on the network-fetched copy or on what earlier tests left cached."""
+    from litellm.anthropic_beta_headers_manager import reload_beta_headers_config
+
+    monkeypatch.setenv("LITELLM_LOCAL_ANTHROPIC_BETA_HEADERS", "True")
+    reload_beta_headers_config()
+    try:
+        yield
+    finally:
+        monkeypatch.delenv("LITELLM_LOCAL_ANTHROPIC_BETA_HEADERS", raising=False)
+        reload_beta_headers_config()
+
+
 def _run_coroutine_if_needed(result):
     if not asyncio.iscoroutine(result):
         return