From 681424e0d2332e9a828b533bdffd626135c8c8ce Mon Sep 17 00:00:00 2001
From: billytrend-cohere <144115527+billytrend-cohere@users.noreply.github.com>
Date: Wed, 20 Mar 2024 19:23:40 -0500
Subject: [PATCH] Silence tests in staging (#420)

---
 tests/test_async_client.py | 1 +
 tests/test_client.py       | 1 +
 2 files changed, 2 insertions(+)

diff --git a/tests/test_async_client.py b/tests/test_async_client.py
index 059331eab..0b22f3fb5 100644
--- a/tests/test_async_client.py
+++ b/tests/test_async_client.py
@@ -240,6 +240,7 @@ async def test_connectors_crud(self) -> None:
 
         await self.co.connectors.delete(created_connector.connector.id)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_tool_use(self) -> None:
         tools = [
             Tool(
diff --git a/tests/test_client.py b/tests/test_client.py
index babd5a554..3ba192c69 100644
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -238,6 +238,7 @@ def test_connectors_crud(self) -> None:
 
         co.connectors.delete(created_connector.connector.id)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     def test_tool_use(self) -> None:
         tools = [
             Tool(