fix(cohere): Catch attribute error

Author: isaacbmiller

dsp/modules/cohere.py

@@ -6,10 +6,14 @@
 
 try:
     import cohere
+
     cohere_api_error = cohere.errors.UnauthorizedError
 except ImportError:
     cohere_api_error = Exception
     # print("Not loading Cohere because it is not installed.")
+except AttributeError:
+    cohere_api_error = Exception
+
 
 def backoff_hdlr(details):
     """Handler from https://pypi.org/project/backoff/"""
@@ -84,13 +88,14 @@ def basic_request(self, prompt: str, **kwargs):
             kwargs.pop("n")
         response = self.co.chat(**kwargs)
 
-
-        self.history.append({
-            "prompt": prompt,
-            "response": response,
-            "kwargs": kwargs,
-            "raw_kwargs": raw_kwargs,
-        })
+        self.history.append(
+            {
+                "prompt": prompt,
+                "response": response,
+                "kwargs": kwargs,
+                "raw_kwargs": raw_kwargs,
+            }
+        )
 
         return response