Remove augemented chat tests (#266)

tests/sync/test_chat.py

@@ -1,10 +1,8 @@
 import unittest
-from typing import List
 
 from utils import get_api_key
 
 import cohere
-from cohere.responses.chat import Mode
 
 API_KEY = get_api_key()
 co = cohere.Client(API_KEY)
@@ -214,6 +212,9 @@ def test_invalid_logit_bias(self):
             with self.assertRaises(cohere.error.CohereError):
                 _ = co.chat("Yo what up?", logit_bias=logit_bias, max_tokens=5)
 
+
+"""
+    Stop testing augmented generation while we change the API
     def test_search_query_generation(self):
         prediction = co.chat("What are the tallest penguins?", mode="search_query_generation")
         self.assertIsInstance(prediction.is_search_required, bool)
@@ -244,3 +245,4 @@ def test_augmented_generation(self):
         self.assertIsInstance(prediction.citations, List)
         self.assertGreater(len(prediction.text), 0)
         self.assertGreater(len(prediction.citations), 0)
+"""