diff --git a/tests/sync/test_chat.py b/tests/sync/test_chat.py index f303f2c6d..931b82155 100644 --- a/tests/sync/test_chat.py +++ b/tests/sync/test_chat.py @@ -24,7 +24,7 @@ def test_multi_replies(self): self.assertIsInstance(prediction.text, str) def test_valid_model(self): - prediction = co.chat("Yo what up?", model="medium", max_tokens=5) + prediction = co.chat("Yo what up?", model="command", max_tokens=5) self.assertIsInstance(prediction.text, str) def test_invalid_model(self):