skip cosmos test

.github/workflows/build.yml

@@ -75,7 +75,7 @@ jobs:
       - name: Test with Cosmos DB
         run: |
           pip install -e .[test,cosmosdb]
-          coverage run -a -m pytest test/cache/test_cosmos_db_cache.py --skip-openai --durations=10 --durations-min=1.0
+          pytest test/cache/test_cosmos_db_cache.py --skip-openai --durations=10 --durations-min=1.0
       - name: Upload coverage to Codecov
         if: matrix.python-version == '3.10'
         uses: codecov/codecov-action@v3

test/cache/test_cosmos_db_cache.py

@@ -4,18 +4,27 @@
 import unittest
 from unittest.mock import MagicMock, patch
 
-from azure.cosmos.exceptions import CosmosResourceNotFoundError
+try:
+    from azure.cosmos.exceptions import CosmosResourceNotFoundError
+
+    skip_test = False
+except ImportError:
+    skip_test = True
+    CosmosResourceNotFoundError = None
 
 from autogen.cache.cosmos_db_cache import CosmosDBCache
 
 
 class TestCosmosDBCache(unittest.TestCase):
     def setUp(self):
-        self.seed = "42"
-        self.connection_string = "AccountEndpoint=https://example.documents.azure.com:443/;"
-        self.database_id = "autogen_cache"
-        self.container_id = "TestContainer"
-        self.client = MagicMock()
+        if skip_test:
+            self.skipTest("requires azure.cosmos")
+        else:
+            self.seed = "42"
+            self.connection_string = "AccountEndpoint=https://example.documents.azure.com:443/;"
+            self.database_id = "autogen_cache"
+            self.container_id = "TestContainer"
+            self.client = MagicMock()
 
     @patch("autogen.cache.cosmos_db_cache.CosmosClient.from_connection_string", return_value=MagicMock())
     def test_init(self, mock_from_connection_string):