chore: explicitly set All in count items request.

momentohq · Jan 19, 2024 · 42b18ee · 42b18ee
1 parent b83617b
commit 42b18ee
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/momento/internal/aio/_vector_index_data_client.py b/src/momento/internal/aio/_vector_index_data_client.py
@@ -60,7 +60,9 @@ async def count_items(
             self._log_issuing_request("CountItems", {"index_name": index_name})
             _validate_index_name(index_name)
 
-            request = vectorindex_pb._CountItemsRequest(index_name=index_name)
+            request = vectorindex_pb._CountItemsRequest(
+                index_name=index_name, all=vectorindex_pb._CountItemsRequest.All()
+            )
             response: vectorindex_pb._CountItemsResponse = await self._build_stub().CountItems(
                 request, timeout=self._default_deadline_seconds
             )

diff --git a/src/momento/internal/synchronous/_vector_index_data_client.py b/src/momento/internal/synchronous/_vector_index_data_client.py
@@ -60,7 +60,9 @@ def count_items(
             self._log_issuing_request("CountItems", {"index_name": index_name})
             _validate_index_name(index_name)
 
-            request = vectorindex_pb._CountItemsRequest(index_name=index_name)
+            request = vectorindex_pb._CountItemsRequest(
+                index_name=index_name, all=vectorindex_pb._CountItemsRequest.All()
+            )
             response: vectorindex_pb._CountItemsResponse = self._build_stub().CountItems(
                 request, timeout=self._default_deadline_seconds
             )