Update test_client.py

cohere-ai · Jul 30, 2024 · 34a4310 · 34a4310
1 parent 616defe
commit 34a4310
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -36,6 +36,7 @@ def test_chat(self) -> None:
 
         print(chat)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     def test_response_format(self) -> None:
         chat = co.chat(
             message="imagine a character from the tv show severance",