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(