From f1ab141b20d5f40c15eed0158045db264db06b90 Mon Sep 17 00:00:00 2001
From: Eric Del Orefice <114540347+ericdelorefice@users.noreply.github.com>
Date: Wed, 30 Sep 2026 13:06:05 -0400
Subject: [PATCH] Python: Cosmos NoSQL collection_exists reports a throttled
 request as a missing container

collection_exists caught CosmosHttpResponseError and returned False. That
is the parent of every Cosmos HTTP error, so a 401, 403, 429 or 5xx all
reported that the container does not exist.

_does_database_exist in the same connector already answers the
equivalent question correctly: CosmosResourceNotFoundError returns False
and anything else raises VectorStoreOperationException.
collection_exists now mirrors it.

Three tests, none existed for collection_exists on this collection. A
missing container returns False, a 429 raises, a successful read returns
True. Against main only the 429 test fails.
---
 .../connectors/azure_cosmos_db.py             |  6 ++-
 .../test_azure_cosmos_db_no_sql_collection.py | 45 +++++++++++++++++++
 2 files changed, 50 insertions(+), 1 deletion(-)

diff --git a/python/semantic_kernel/connectors/azure_cosmos_db.py b/python/semantic_kernel/connectors/azure_cosmos_db.py
index 2d1bb53d9630..6e78e78e6d2f 100644
--- a/python/semantic_kernel/connectors/azure_cosmos_db.py
+++ b/python/semantic_kernel/connectors/azure_cosmos_db.py
@@ -1036,8 +1036,12 @@ async def collection_exists(self, **kwargs) -> bool:
         try:
             await container_proxy.read(**kwargs)
             return True
-        except CosmosHttpResponseError:
+        except CosmosResourceNotFoundError:
             return False
+        except Exception as e:
+            raise VectorStoreOperationException(
+                f"Failed to check if container '{self.collection_name}' exists, with message {e}"
+            ) from e
 
     @override
     async def ensure_collection_deleted(self, **kwargs) -> None:
diff --git a/python/tests/unit/connectors/memory/azure_cosmos_db/test_azure_cosmos_db_no_sql_collection.py b/python/tests/unit/connectors/memory/azure_cosmos_db/test_azure_cosmos_db_no_sql_collection.py
index 717c9ee65e7f..b853bff944e2 100644
--- a/python/tests/unit/connectors/memory/azure_cosmos_db/test_azure_cosmos_db_no_sql_collection.py
+++ b/python/tests/unit/connectors/memory/azure_cosmos_db/test_azure_cosmos_db_no_sql_collection.py
@@ -676,3 +676,48 @@ async def test_client_is_closed(
         assert collection.cosmos_client is not None
 
     mock_cosmos_client_close.assert_called()
+
+
+# -- collection_exists: only a missing container is "absent" --
+#
+# CosmosHttpResponseError is the parent of every Cosmos HTTP error: 401, 403, 429 and 5xx as
+# well as 404. _does_database_exist in the same connector already narrows to
+# CosmosResourceNotFoundError and raises VectorStoreOperationException for the rest;
+# collection_exists now does the same.
+
+
+def _collection(record_type, collection_name, container_proxy) -> CosmosNoSqlCollection:
+    collection = CosmosNoSqlCollection(record_type=record_type, collection_name=collection_name)
+    collection._get_container_proxy = AsyncMock(return_value=container_proxy)
+    return collection
+
+
+async def test_azure_cosmos_db_no_sql_collection_exists_is_false_when_the_container_is_missing(
+    azure_cosmos_db_no_sql_unit_test_env, record_type, collection_name: str
+):
+    container_proxy = MagicMock()
+    container_proxy.read = AsyncMock(side_effect=CosmosResourceNotFoundError)
+
+    assert await _collection(record_type, collection_name, container_proxy).collection_exists() is False
+
+
+async def test_azure_cosmos_db_no_sql_collection_exists_raises_on_a_throttled_request(
+    azure_cosmos_db_no_sql_unit_test_env, record_type, collection_name: str
+):
+    """A 429 is a failure to look, not an answer of "absent"."""
+    container_proxy = MagicMock()
+    container_proxy.read = AsyncMock(
+        side_effect=CosmosHttpResponseError(status_code=429, message="Request rate is large")
+    )
+
+    with pytest.raises(VectorStoreOperationException):
+        await _collection(record_type, collection_name, container_proxy).collection_exists()
+
+
+async def test_azure_cosmos_db_no_sql_collection_exists_is_true_when_the_read_succeeds(
+    azure_cosmos_db_no_sql_unit_test_env, record_type, collection_name: str
+):
+    container_proxy = MagicMock()
+    container_proxy.read = AsyncMock(return_value={"id": collection_name})
+
+    assert await _collection(record_type, collection_name, container_proxy).collection_exists() is True