@@ -93,8 +93,32 @@ def test_refiner_name(self):
9393 selector = Selector ()
9494 assert selector .refiner_name == "selector"
9595
96+ def test_default_classifier_model (self ):
97+ """Default classifier_model should be openai/gpt-4o-mini."""
98+ selector = Selector ()
99+ assert selector .classifier_model == "openai/gpt-4o-mini"
100+
101+ def test_custom_classifier_model (self ):
102+ """Should accept custom classifier_model."""
103+ selector = Selector (classifier_model = "openai/gpt-5-nano" )
104+ assert selector .classifier_model == "openai/gpt-5-nano"
105+
96106
97- class TestSelectorSmallChecklist :
107+ class TestSelectorClassifierModel :
108+ """Tests that _classify_feedback honors classifier_model."""
109+
110+ def test_classify_uses_classifier_model_not_self_model (self , small_checklist ):
111+ """_classify_feedback should pass classifier_model, not self.model, to _call_model."""
112+ selector = Selector (
113+ model = "anthropic/claude-sonnet-4.6" ,
114+ classifier_model = "openai/gpt-4o-mini" ,
115+ observations = ["obs A" , "obs B" ],
116+ )
117+ with patch .object (selector , "_call_model" , return_value = "0" ) as mock_call :
118+ selector ._classify_feedback (small_checklist )
119+ assert mock_call .call_count == 2
120+ for call in mock_call .call_args_list :
121+ assert call .kwargs .get ("model" ) == "openai/gpt-4o-mini"
98122 """Tests when checklist is already small enough."""
99123
100124 @patch ("autochecklist.refiners.selector.get_embeddings" )
0 commit comments