Merge branch 'feature/remove-zeroshot-route' into staging

weni-ai · Nov 21, 2024 · 2c778cd · 2c778cd
2 parents 5c88120 + 9b8276c
commit 2c778cd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/router/classifiers/tests.py b/router/classifiers/tests.py
@@ -78,11 +78,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):