diff --git a/tests/test_o3_pro_output_text_fix.py b/tests/test_o3_pro_output_text_fix.py index b465425..2709305 100644 --- a/tests/test_o3_pro_output_text_fix.py +++ b/tests/test_o3_pro_output_text_fix.py @@ -41,6 +41,12 @@ class TestO3ProOutputTextFix: # Use the new public API for registry cleanup ModelProviderRegistry.reset_for_testing() # Provider registration is now handled by inject_transport helper + + # Clear restriction service to ensure it re-reads environment + # This is necessary because previous tests may have set restrictions + # that are cached in the singleton + import utils.model_restrictions + utils.model_restrictions._restriction_service = None def teardown_method(self): """Clean up after test to ensure no state pollution."""