From 34a4310124122a564ece67fadc5f349345c9774e Mon Sep 17 00:00:00 2001
From: billytrend-cohere <144115527+billytrend-cohere@users.noreply.github.com>
Date: Tue, 30 Jul 2024 06:21:11 -0500
Subject: [PATCH] Update test_client.py

---
 tests/test_client.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_client.py b/tests/test_client.py
index 775603a51..bf0581529 100644
--- 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",