diff --git a/router/classifiers/tests.py b/router/classifiers/tests.py index 9aadc9bd..a12955a2 100644 --- a/router/classifiers/tests.py +++ b/router/classifiers/tests.py @@ -76,11 +76,10 @@ def setUp(self) -> None: choice = MockChoice(message) self.mock_response = MockResponse([choice]) self.mock_client = MockOpenAIClient(response=self.mock_response) - self.chatgpt_model = "gpt-3.5-turbo" self.classifier = ChatGPTFunctionClassifier( client=self.mock_client, - chatgpt_model=self.chatgpt_model, + agent_goal="Answer user questions" ) def test_predict(self):